python.ParticleGun Class Reference

Inheritance diagram for python.ParticleGun:

Collaboration diagram for python.ParticleGun:

Public Member Functions
def	__init__ (self, name="ParticleGun", randomStream="ParticleGun", randomSeed=None)
def	sampler (self)
def	sampler (self, s)
def	initialize (self)
def	fillEvent (self, evt)

Public Attributes
	samplers
	randomStream
	randomSeed

Detailed Description

A simple but flexible algorithm for generating events from simple distributions.

Definition at line 17 of file Generators/ParticleGun/python/__init__.py.

Constructor & Destructor Documentation

__init__()

def python.ParticleGun.__init__	(		self,
			name = "ParticleGun",
			randomStream = "ParticleGun",
			randomSeed = None
	)

Definition at line 22 of file Generators/ParticleGun/python/__init__.py.

    def __init__(self, name="ParticleGun", randomStream="ParticleGun", randomSeed=None):
        super(ParticleGun, self).__init__(name=name)
        self.samplers = [ParticleSampler()]
        self.randomStream = randomStream
        self.randomSeed = randomSeed
          

Member Function Documentation

fillEvent()

def python.ParticleGun.fillEvent	(		self,
			evt
	)

Sample a list of particle properties, which are then used to create a new GenEvent in StoreGate.

Definition at line 42 of file Generators/ParticleGun/python/__init__.py.

    def fillEvent(self, evt):
        """
        Sample a list of particle properties, which are then used to create a new GenEvent in StoreGate.
        """
        # set the random seed
        offset = self.randomSeed if self.randomSeed is not None else 0
        seed = ROOT.ATHRNG.calculateSeedsPython(self.randomStream, self._ctx.eventID().event_number(), self._ctx.eventID().run_number(), offset)
 
        if seed is None:
            self.msg.warning("Failed to find a seed for the random stream named '%s'.", self.randomStream)
            seed = self.randomSeed
        if seed is not None:
            self.msg.debug("Set random seed to %s.", str(seed))
            random.seed(seed)
        else:
            self.msg.error("Failed to set random seed.")
            return StatusCode.Failure
 
        if HepMCVersion == 2:
          evt.weights().push_back(1.0)
        
          for s in self.samplers:
            particles = s.shoot()
            for p in particles:
                
                pos = HepMC.FourVector(p.pos.X(), p.pos.Y(), p.pos.Z(), p.pos.T())
                gv = HepMC.GenVertex(pos)
                ROOT.SetOwnership(gv, False)
                evt.add_vertex(gv)
 
                
                mom = HepMC.FourVector(p.mom.Px(), p.mom.Py(), p.mom.Pz(), p.mom.E())
                gp = HepMC.GenParticle()
                gp.set_status(1)
                gp.set_pdg_id(p.pid)
                gp.set_momentum(mom)
                if p.mass is not None:
                    gp.set_generated_mass(p.mass)
                ROOT.SetOwnership(gp, False)
                gv.add_particle_out(gp)
 
        if HepMCVersion == 3:
          evt.set_units(HepMC.Units.MEV, HepMC.Units.MM)
          evt.weights().push_back(1.0)
          beamparticle1 = std.shared_ptr['HepMC3::GenParticle'](HepMC.GenParticle(HepMC.FourVector(0,0,-7000,7000),2212,4))
          ROOT.SetOwnership(beamparticle1, False)
          beamparticle2 = std.shared_ptr['HepMC3::GenParticle'](HepMC.GenParticle(HepMC.FourVector(0,0,7000,7000),2212,4))
          ROOT.SetOwnership(beamparticle2, False)
          primary = std.shared_ptr['HepMC3::GenVertex'](HepMC.GenVertex())
          ROOT.SetOwnership(primary, False)
          primary.add_particle_in(beamparticle1)
          primary.add_particle_in(beamparticle2)
          evt.add_vertex(primary)
          evt.add_beam_particle(beamparticle1)
          evt.add_beam_particle(beamparticle2)
          #Create all the needed particles
          for s in self.samplers:
            particles = s.shoot()
            for p in particles:
                # Create the production vertex of the particle
                gv = std.shared_ptr['HepMC3::GenVertex'](HepMC.GenVertex(HepMC.FourVector(p.pos.X(), p.pos.Y(), p.pos.Z(), p.pos.T())))
                ROOT.SetOwnership(gv, False)
                evt.add_vertex(gv)
                # Create a fake particle to connect the production vertex of the particle of interest to the primary
                fakeparticle = std.shared_ptr['HepMC3::GenParticle'](HepMC.GenParticle(HepMC.FourVector(p.mom.Px(), p.mom.Py(), p.mom.Pz(), p.mom.E()),p.pid,11))
                ROOT.SetOwnership(fakeparticle, False)
                gv.add_particle_in(fakeparticle)
                primary.add_particle_out(fakeparticle)
                # Create the particle
                gp = std.shared_ptr['HepMC3::GenParticle'](HepMC.GenParticle(HepMC.FourVector(p.mom.Px(), p.mom.Py(), p.mom.Pz(), p.mom.E()),p.pid,1))
                ROOT.SetOwnership(gp, False)
                if p.mass is not None:
                    gp.set_generated_mass(p.mass)
                gv.add_particle_out(gp)
          for p in evt.particles():
            att = std.shared_ptr['HepMC3::IntAttribute'](HepMC.IntAttribute(p.id()))
            p.add_attribute("barcode",att)
          for v in evt.vertices():
            att = std.shared_ptr['HepMC3::IntAttribute'](HepMC.IntAttribute(v.id()))
            v.add_attribute("barcode",att)
        return StatusCode.Success

initialize()

def python.ParticleGun.initialize

(

self

)

Definition at line 38 of file Generators/ParticleGun/python/__init__.py.

    def initialize(self):
        return StatusCode.Success
 
 

sampler() [1/2]

def python.ParticleGun.sampler

(

self

)

Definition at line 29 of file Generators/ParticleGun/python/__init__.py.

    def sampler(self):
        "Get the first (and presumed only) sampler"
        return self.samplers[0] if self.samplers else None

sampler() [2/2]

def python.ParticleGun.sampler	(		self,
			s
	)

Definition at line 33 of file Generators/ParticleGun/python/__init__.py.

    def sampler(self, s):
        "Set the samplers list to include only a single sampler, s"
        self.samplers = [s]
 
 

Member Data Documentation

randomSeed

python.ParticleGun.randomSeed

Definition at line 26 of file Generators/ParticleGun/python/__init__.py.

randomStream

python.ParticleGun.randomStream

Definition at line 25 of file Generators/ParticleGun/python/__init__.py.

samplers

python.ParticleGun.samplers

Definition at line 24 of file Generators/ParticleGun/python/__init__.py.

---

The documentation for this class was generated from the following file:

• Generators/ParticleGun/python/__init__.py