Inheritance diagram for ParticleGun_FastCalo_Config.MyParticleSampler:

Collaboration diagram for ParticleGun_FastCalo_Config.MyParticleSampler:

Public Member Functions
def	__init__ (self, energy, eta, pid, shift_z=0)

def	shoot (self)

Public Attributes
	pid

	shift_z

	mom1

Detailed Description

Definition at line 10 of file ParticleGun_FastCalo_Config.py.

Constructor & Destructor Documentation

◆ init()

def ParticleGun_FastCalo_Config.MyParticleSampler.__init__	(	self,
		energy,
		eta,
		pid,
		shift_z = `0`
	)

Definition at line 11 of file ParticleGun_FastCalo_Config.py.

     def __init__(self,energy,eta,pid,shift_z=0):
         self.pid = pid
         self.shift_z = shift_z
         pdg_table = ROOT.TDatabasePDG.Instance() #Gives values in GeV
         mass = pdg_table.GetParticle(self.pid()).Mass()*1000.
         self.mom1 = PG.EEtaMPhiSampler(energy=energy,eta=eta,mass=mass)
  

Member Function Documentation

◆ shoot()

def ParticleGun_FastCalo_Config.MyParticleSampler.shoot ( self )

Definition at line 18 of file ParticleGun_FastCalo_Config.py.

     def shoot(self):
         pid = self.pid()
  
         shift_z = self.shift_z
  
         mom = self.mom1.shoot()
         pos_temp = mom.Vect().Unit()
  
         # Define geometry
         barrelR1 = 1148.0
         barrelR2 = 120.0
         barrelR3 = 41.0
         endcapZ1 = 3550.0
         endcapZ2 = 4587.0
  
         # Would it hit the barrel, or the endcap?
         tanTheta = pos_temp.Perp() / abs( pos_temp.Z() );
         if   tanTheta > barrelR1 / endcapZ1:
             pos_temp *= barrelR1 / pos_temp.Perp()
         elif tanTheta > barrelR2 / endcapZ1:
             pos_temp *= endcapZ1 / abs( pos_temp.Z() )
         elif tanTheta > barrelR2 / endcapZ2:
             pos_temp *= barrelR2 / pos_temp.Perp()
         elif tanTheta > barrelR3 / endcapZ2:
             pos_temp *= endcapZ2 / abs( pos_temp.Z() )
         else:
             pos_temp *= barrelR3 / pos_temp.Perp()
  
         # Shift position of vector in the Z direction
         pos_temp_2 = ROOT.TVector3()
         pos_temp_2.SetXYZ(pos_temp.X(), pos_temp.Y(), pos_temp.Z()+shift_z)
         pos_temp_2 *= 1. / pos_temp_2.Mag(); # reduce magnitude of vector
  
         # recalculate; Would it hit the barrel, or the endcap?
         tanTheta_2 = pos_temp_2.Perp() / abs( pos_temp_2.Z() );
         if   tanTheta_2 > barrelR1 / endcapZ1:
             pos_temp_2 *= barrelR1 / pos_temp_2.Perp()
         elif tanTheta_2 > barrelR2 / endcapZ1:
             pos_temp_2 *= endcapZ1 / abs( pos_temp_2.Z() )
         elif tanTheta_2 > barrelR2 / endcapZ2:
             pos_temp_2 *= barrelR2 / pos_temp_2.Perp()
         elif tanTheta_2 > barrelR3 / endcapZ2:
             pos_temp_2 *= endcapZ2 / abs( pos_temp_2.Z() )
         else:
             pos_temp_2 *= barrelR3 / pos_temp_2.Perp()
  
         pos = ROOT.TLorentzVector(pos_temp_2.X(),pos_temp_2.Y(),pos_temp_2.Z(), pos_temp_2.Mag())
  
         #print "pid ",pid
  
         return [ PG.SampledParticle( pid , mom , pos ) ]
  