ParticleGun_SamplingFraction.py

Go to the documentation of this file.

#! -*- python -*-
evgenConfig.description = "Single particle gun for Sampling Fraction event generation"
evgenConfig.keywords = ["singleParticle",]
evgenConfig.generators = ["ParticleGun"]
evgenConfig.contact = ["michael.duehrssen@cern.ch"]
 
import ParticleGun as PG
import ROOT, math, random
 
class MyParticleSampler(PG.ParticleSampler):
    """
    Projective showers starting at entrance of calorimeter, flat in eta, constant energy
    """
 
    def __init__(self,pid=11,momentum=50000.,eta1=0.,eta2=1.4,bec=0,radius=1500.,z=3740.5):
        self.pid = pid
        self.momentum = momentum
        self.eta1 = eta1
        self.eta2 = eta2
        pdg_table = ROOT.TDatabasePDG.Instance() #Gives values in GeV
        self.mass = pdg_table.GetParticle(self.pid()).Mass()*1000.
        self.bec=bec
        self.radius=radius
        self.z=z
 
    def shoot(self):
       rtn=[]
       eta = random.uniform(self.eta1, self.eta2) 
       phi = random.uniform(0, math.tau)  # tau = 2 * pi
       v4 = ROOT.TLorentzVector()
       pt = self.momentum / math.cosh(eta)
       v4.SetPtEtaPhiM(pt, eta, phi, self.mass)
       if self.bec==0:
           radius= self.radius
           x=radius*math.cos(phi)
           y=radius*math.sin(phi)
           z=radius*math.sinh(eta)
       else:
           z=self.z
           radius=z/math.sinh(eta)
           x=radius*math.cos(phi)
           y=radius*math.sin(phi)
       t=math.sqrt(x*x+y*y+z*z)
       vp = ROOT.TLorentzVector(x,y,z,t)
       p = PG.SampledParticle(pid=self.pid(),mom=v4,pos=vp)
       #print "E,eta,phi,mass ",e,eta,phi,self.mass,"  position ",x,y,z," pid=",p.pid
       rtn.append(p)
       return rtn
 
 
 
 
FIRST_DIR = (os.environ['JOBOPTSEARCHPATH']).split(":")[0]
jofiles = [f for f in os.listdir(FIRST_DIR) if (f.startswith('mc') and f.endswith('.py'))]
 
print "================ SETTTINGS ================="
print ("jofiles     = ", jofiles)
 
 
args = jofiles[0].split('.py')[0]
print ("args     = ", args)
 
myMomentum = float(args.split('_Mom')[1].split('_')[0])
print ("Momentum = ", myMomentum,"MeV")
 
myPDGID = int(float(args.split('_pid')[1].split('_')[0].replace('m','-')))
print ("pid      = ", myPDGID)
 
myLowEta  = 0.01*float(args.split('eta_')[1].split('_')[0].replace('m','-'))
print ("etalow   = ", myLowEta)
 
myHighEta = 0.01*float(args.split('eta_')[1].split('_')[1].replace('m','-'))
print ("etahigh   = ", myHighEta)
 
if "_Radius" in args:
  myRadius = 0.001*float(args.split('_Radius')[1].split('_')[0]) #Argument needs to by in mum, since a "." in the filename is not allowed
else: 
  myRadius = 1500.
print ("radius   = ", myRadius,"mm")
 
if "_Z" in args:
  myZ = 0.001*float(args.split('_Z')[1].split('_')[0]) #Argument needs to by in mum, since a "." in the filename is not allowed
else:  
  myZ = 3740.5
print ("Z        = ", myZ,"mm")
  
if "bec" in args:
  bec=1
else:
  bec=0
print ("bec      = ", bec)
print "============================================"
 
genSeq += PG.ParticleGun()
genSeq.ParticleGun.sampler = MyParticleSampler(momentum=myMomentum,eta1=myLowEta,eta2=myHighEta,pid=myPDGID,bec=bec,radius=myRadius,z=myZ)