def setupFilters( self, kernel):
""" attach all filters to the kernel """
import DDG4
setOfFilters = set()
for name, filt in self.filters.iteritems():
setOfFilters.add(name)
ddFilt = DDG4.Filter(kernel,filt['name'])
for para, value in filt['parameter'].iteritems():
setattr( ddFilt, para, value )
kernel.registerGlobalFilter(ddFilt)
filt['filter'] = ddFilt
from itertools import chain
listOfFilters = []
for val in self.mapDetFilter.values():
listOfFilters += ConfigHelper.makeList(val)
requestedFilter = set(chain( ConfigHelper.makeList(self.tracker), ConfigHelper.makeList(self.calo), listOfFilters))
print "ReqFilt",requestedFilter
if requestedFilter - setOfFilters:
raise RuntimeError(" Filter(s) '%s' are not registered!" % str(requestedFilter - setOfFilters) )
def run():
global geant4
kernel = DDG4.Kernel()
description = kernel.detectorDescription()
install_dir = os.environ['DD4hepINSTALL']
kernel.loadGeometry(str("file:" + install_dir + "/DDDetectors/compact/SiD_Markus.xml"))
DDG4.importConstants(description)
DDG4.Core.setPrintLevel(Output.DEBUG)
DDG4.Core.setPrintFormat(str("%-32s %6s %s"))
kernel.NumberOfThreads = 1
geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerWeightedAction')
geant4.printDetectors()
# Configure UI
geant4.setupCshUI()
# Geant4 user initialization action
geant4.addUserInitialization(worker=setupWorker, master=setupMaster)
# Configure G4 geometry setup
seq, act = geant4.addDetectorConstruction("Geant4DetectorGeometryConstruction/ConstructGeo")
# Configure G4 magnetic field tracking
self.setupTrackingFieldMT() # noqa: F821
seq, act = geant4.addDetectorConstruction("Geant4PythonDetectorConstruction/DummyDet",
geometry=dummy_geom,
sensitives=dummy_sd)
# Configure G4 sensitive detectors
seq, act = geant4.addDetectorConstruction("Geant4PythonDetectorConstruction/SetupSD",
sensitives=setupSensitives)
# Configure G4 sensitive detectors
seq, act = geant4.addDetectorConstruction("Geant4DetectorSensitivesConstruction/ConstructSD",
allow_threads=True)
# Setup random generator
rndm = DDG4.Action(kernel, 'Geant4Random/Random')
rndm.Seed = 987654321
rndm.initialize()
# Setup global filters fur use in sensntive detectors
f1 = DDG4.Filter(kernel, 'GeantinoRejectFilter/GeantinoRejector')
kernel.registerGlobalFilter(f1)
# seq,act = geant4.setupTracker('SiTrackerBarrel')
# seq,act = geant4.setupTracker('SiTrackerEndcap')
# seq,act = geant4.setupTracker('SiTrackerForward')
# Now the calorimeters
# seq,act = geant4.setupCalorimeter('EcalBarrel')
# seq,act = geant4.setupCalorimeter('EcalEndcap')
# seq,act = geant4.setupCalorimeter('HcalBarrel')
# seq,act = geant4.setupCalorimeter('HcalEndcap')
# seq,act = geant4.setupCalorimeter('HcalPlug')
# seq,act = geant4.setupCalorimeter('MuonBarrel')
# seq,act = geant4.setupCalorimeter('MuonEndcap')
# seq,act = geant4.setupCalorimeter('LumiCal')
# seq,act = geant4.setupCalorimeter('BeamCal')
# Now build the physics list:
seq = geant4.setupPhysics('QGSP_BERT')
phys = DDG4.PhysicsList(geant4.master(), 'Geant4PhysicsList/MyPhysics')
part = DDG4.Action(geant4.master(), 'Geant4ExtraParticles/extraparts')
part.pdgfile = 'checkout/DDG4/examples/particle.tbl'
phys.adoptPhysicsConstructor(part.get())
seq.add(phys)
geant4.run()
# kernel.configure()
# kernel.initialize()
# DDG4.setPrintLevel(Output.DEBUG)
# kernel.run()
# kernel.terminate()
return 1
def run():
kernel = DDG4.Kernel()
description = kernel.detectorDescription()
install_dir = os.environ['DD4hepINSTALL']
kernel.loadGeometry("file:" + install_dir + "/DDDetectors/compact/SiD.xml")
DDG4.importConstants(description)
geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction')
geant4.printDetectors()
logging.info("# Configure UI")
geant4.setupCshUI()
logging.info("# Configure G4 magnetic field tracking")
geant4.setupTrackingField()
logging.info("# Setup random generator")
rndm = DDG4.Action(kernel, 'Geant4Random/Random')
rndm.Seed = 987654321
rndm.initialize()
##rndm.showStatus()
logging.info("# Configure Run actions")
run1 = DDG4.RunAction(kernel, 'Geant4TestRunAction/RunInit')
run1.Property_int = 12345
run1.Property_double = -5e15 * keV
run1.Property_string = 'Startrun: Hello_2'
logging.info("%s %s %s", run1.Property_string, str(run1.Property_double),
str(run1.Property_int))
run1.enableUI()
kernel.registerGlobalAction(run1)
kernel.runAction().adopt(run1)
logging.info("# Configure Event actions")
prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint')
prt.OutputLevel = Output.INFO
prt.OutputType = 3 # Print both: table and tree
kernel.eventAction().adopt(prt)
logging.info("""
Configure I/O
""")
#evt_lcio = geant4.setupLCIOOutput('LcioOutput','CLICSiD_'+time.strftime('%Y-%m-%d_%H-%M'))
#evt_lcio.OutputLevel = Output.ERROR
evt_root = geant4.setupROOTOutput(
'RootOutput', 'CLICSiD_' + time.strftime('%Y-%m-%d_%H-%M'))
generator_output_level = Output.INFO
gen = DDG4.GeneratorAction(kernel,
"Geant4GeneratorActionInit/GenerationInit")
kernel.generatorAction().adopt(gen)
#VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV
logging.info("""
Generation of isotrope tracks of a given multiplicity with overlay:
""")
logging.info("# First particle generator: pi+")
gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropPi+")
gen.Mask = 1
gen.Particle = 'pi+'
gen.Energy = 100 * GeV
gen.Multiplicity = 2
gen.Distribution = 'cos(theta)'
kernel.generatorAction().adopt(gen)
logging.info("# Install vertex smearing for this interaction")
gen = DDG4.GeneratorAction(kernel, "Geant4InteractionVertexSmear/SmearPi+")
gen.Mask = 1
gen.Offset = (20 * mm, 10 * mm, 10 * mm, 0 * ns)
gen.Sigma = (4 * mm, 1 * mm, 1 * mm, 0 * ns)
kernel.generatorAction().adopt(gen)
logging.info("# Second particle generator: e-")
gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropE-")
gen.Mask = 2
gen.Particle = 'e-'
gen.Energy = 25 * GeV
gen.Multiplicity = 3
gen.Distribution = 'uniform'
kernel.generatorAction().adopt(gen)
logging.info(" Install vertex smearing for this interaction")
gen = DDG4.GeneratorAction(kernel, "Geant4InteractionVertexSmear/SmearE-")
gen.Mask = 2
gen.Offset = (-20 * mm, -10 * mm, -10 * mm, 0 * ns)
gen.Sigma = (12 * mm, 8 * mm, 8 * mm, 0 * ns)
kernel.generatorAction().adopt(gen)
#^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
logging.info("# Merge all existing interaction records")
gen = DDG4.GeneratorAction(kernel,
"Geant4InteractionMerger/InteractionMerger")
gen.OutputLevel = 4 #generator_output_level
gen.enableUI()
kernel.generatorAction().adopt(gen)
logging.info("# Finally generate Geant4 primaries")
gen = DDG4.GeneratorAction(kernel, "Geant4PrimaryHandler/PrimaryHandler")
gen.OutputLevel = 4 #generator_output_level
gen.enableUI()
kernel.generatorAction().adopt(gen)
logging.info("# ....and handle the simulation particles.")
part = DDG4.GeneratorAction(kernel,
"Geant4ParticleHandler/ParticleHandler")
kernel.generatorAction().adopt(part)
#part.SaveProcesses = ['conv','Decay']
part.SaveProcesses = ['Decay']
part.MinimalKineticEnergy = 100 * MeV
part.OutputLevel = 5 # generator_output_level
part.enableUI()
user = DDG4.Action(kernel,
"Geant4TCUserParticleHandler/UserParticleHandler")
user.TrackingVolume_Zmax = DDG4.EcalEndcap_zmin
user.TrackingVolume_Rmax = DDG4.EcalBarrel_rmin
user.enableUI()
part.adopt(user)
logging.info("# Setup global filters fur use in sensitive detectors")
f1 = DDG4.Filter(kernel, 'GeantinoRejectFilter/GeantinoRejector')
f2 = DDG4.Filter(kernel, 'ParticleRejectFilter/OpticalPhotonRejector')
f2.particle = 'opticalphoton'
f3 = DDG4.Filter(kernel, 'ParticleSelectFilter/OpticalPhotonSelector')
f3.particle = 'opticalphoton'
f4 = DDG4.Filter(kernel, 'EnergyDepositMinimumCut')
f4.Cut = 10 * MeV
f4.enableUI()
kernel.registerGlobalFilter(f1)
kernel.registerGlobalFilter(f2)
kernel.registerGlobalFilter(f3)
kernel.registerGlobalFilter(f4)
logging.info("# First the tracking detectors")
seq, act = geant4.setupTracker('SiVertexBarrel')
seq.adopt(f1)
#seq.adopt(f4)
act.adopt(f1)
seq, act = geant4.setupTracker('SiVertexEndcap')
seq.adopt(f1)
#seq.adopt(f4)
seq, act = geant4.setupTracker('SiTrackerBarrel')
seq, act = geant4.setupTracker('SiTrackerEndcap')
seq, act = geant4.setupTracker('SiTrackerForward')
logging.info("# Now setup the calorimeters")
seq, act = geant4.setupCalorimeter('EcalBarrel')
seq, act = geant4.setupCalorimeter('EcalEndcap')
seq, act = geant4.setupCalorimeter('HcalBarrel')
seq, act = geant4.setupCalorimeter('HcalEndcap')
seq, act = geant4.setupCalorimeter('HcalPlug')
seq, act = geant4.setupCalorimeter('MuonBarrel')
seq, act = geant4.setupCalorimeter('MuonEndcap')
seq, act = geant4.setupCalorimeter('LumiCal')
seq, act = geant4.setupCalorimeter('BeamCal')
logging.info("# Now build the physics list:")
phys = geant4.setupPhysics('QGSP_BERT')
ph = geant4.addPhysics('Geant4PhysicsList/Myphysics')
#ph.addParticleConstructor('G4BosonConstructor')
#ph.addParticleConstructor('G4LeptonConstructor')
#ph.addParticleProcess('e[+-]','G4eMultipleScattering',-1,1,1)
#ph.addPhysicsConstructor('G4OpticalPhysics')
# Add special particle types from specialized physics constructor
part = geant4.addPhysics('Geant4ExtraParticles/ExtraParticles')
part.pdgfile = 'checkout/DDG4/examples/particle.tbl'
# Add global range cut
rg = geant4.addPhysics('Geant4DefaultRangeCut/GlobalRangeCut')
rg.RangeCut = 0.7 * mm
phys.dump()
kernel.configure()
kernel.initialize()
#DDG4.setPrintLevel(Output.DEBUG)
kernel.run()
kernel.terminate()
def run():
kernel = DDG4.Kernel()
install_dir = os.environ['DD4hepINSTALL']
example_dir = install_dir+'/examples/DDG4/examples';
kernel.loadGeometry("file:"+install_dir+"/examples/CLICSiD/compact/compact.xml")
kernel.loadXML("file:"+example_dir+"/DDG4_field.xml")
simple = DDG4.Simple(kernel,tracker='LcioTestTrackerAction')
simple.printDetectors()
# Configure UI
simple.setupCshUI()
# Configure Run actions
run1 = DDG4.RunAction(kernel,'Geant4TestRunAction/RunInit')
run1.Property_int = 12345
run1.Property_double = -5e15*keV
run1.Property_string = 'Startrun: Hello_2'
print run1.Property_string, run1.Property_double, run1.Property_int
run1.enableUI()
kernel.registerGlobalAction(run1)
kernel.runAction().add(run1)
# Configure Event actions
evt2 = DDG4.EventAction(kernel,'Geant4TestEventAction/UserEvent_2')
evt2.Property_int = 123454321
evt2.Property_double = 5e15*GeV
evt2.Property_string = 'Hello_2 from the python setup'
evt2.enableUI()
kernel.registerGlobalAction(evt2)
evt1 = DDG4.EventAction(kernel,'Geant4TestEventAction/UserEvent_1')
evt1.Property_int=01234
evt1.Property_double=1e11
evt1.Property_string='Hello_1'
evt1.enableUI()
kernel.eventAction().add(evt1)
kernel.eventAction().add(evt2)
"""
trk = DDG4.Action(kernel,"Geant4TrackPersistency/MonteCarloTruthHandler")
kernel.registerGlobalAction(trk)
trk.release()
mc = DDG4.Action(kernel,"Geant4MonteCarloRecordManager/MonteCarloRecordManager")
kernel.registerGlobalAction(mc)
mc.release()
"""
# Configure I/O
evt_lcio = simple.setupLCIOOutput('LcioOutput','CLICSiD_'+time.strftime('%Y-%m-%d_%H-%M'))
gen = DDG4.GeneratorAction(kernel,"Geant4TestGeneratorAction/Generate")
kernel.generatorAction().add(gen)
# Setup particle gun
gun = simple.setupGun('Gun','pi-',100*GeV,True)
"""
rdr = DDG4.GeneratorAction(kernel,"LcioGeneratorAction/Reader")
rdr.zSpread = 0.0
rdr.lorentzAngle = 0.0
rdr.OutputLevel = DDG4.OutputLevel.INFO
rdr.Input = "LcioEventReader|test.data"
rdr.enableUI()
kernel.generatorAction().add(rdr)
"""
# Setup global filters fur use in sensintive detectors
f1 = DDG4.Filter(kernel,'GeantinoRejectFilter/GeantinoRejector')
kernel.registerGlobalFilter(f1)
f2 = DDG4.Filter(kernel,'ParticleRejectFilter/OpticalPhotonRejector')
f2.particle = 'opticalphoton'
kernel.registerGlobalFilter(f2)
f3 = DDG4.Filter(kernel,'ParticleSelectFilter/OpticalPhotonSelector')
f3.particle = 'opticalphoton'
kernel.registerGlobalFilter(f3)
f4 = DDG4.Filter(kernel,'EnergyDepositMinimumCut')
f4.Cut = 10*MeV
f4.enableUI()
kernel.registerGlobalFilter(f4)
# First the tracking detectors
seq,act = simple.setupTracker('SiVertexBarrel')
seq.add(f1)
#seq.add(f4)
act.add(f1)
seq,act = simple.setupTracker('SiVertexEndcap')
seq.add(f1)
#seq.add(f4)
seq,act = simple.setupTracker('SiTrackerBarrel')
seq,act = simple.setupTracker('SiTrackerEndcap')
seq,act = simple.setupTracker('SiTrackerForward')
# Now the calorimeters
seq,act = simple.setupCalorimeter('EcalBarrel')
seq,act = simple.setupCalorimeter('EcalEndcap')
seq,act = simple.setupCalorimeter('HcalBarrel')
seq,act = simple.setupCalorimeter('HcalEndcap')
seq,act = simple.setupCalorimeter('HcalPlug')
seq,act = simple.setupCalorimeter('MuonBarrel')
seq,act = simple.setupCalorimeter('MuonEndcap')
seq,act = simple.setupCalorimeter('LumiCal')
seq,act = simple.setupCalorimeter('BeamCal')
# Now build the physics list:
phys = simple.setupPhysics('QGSP_BERT')
ph = DDG4.PhysicsList(kernel,'Geant4PhysicsList/Myphysics')
ph.addParticleConstructor('G4BosonConstructor')
ph.addParticleConstructor('G4LeptonConstructor')
ph.addParticleProcess('e[+-]','G4eMultipleScattering',-1,1,1)
ph.addPhysicsConstructor('G4OpticalPhysics')
ph.enableUI()
phys.add(ph)
phys.dump()
kernel.configure()
kernel.initialize()
#DDG4.setPrintLevel(Output.DEBUG)
kernel.run()
kernel.terminate()
def run():
kernel = DDG4.Kernel()
try:
install_dir = os.environ['DD4hepINSTALL']
# lcgeo_dir = os.environ['LCGEO']
except (KeyError):
print " please set the environment variable DD4hepINSTALL "
print " to your DD4hep installation path ! "
exit(1)
# example_dir = lcgeo_dir+'/example';
kernel.loadGeometry("file:"+ compactFile )
lcdd = kernel.lcdd()
DDG4.importConstants( lcdd )
#----------------------------------------------------------------------------------
simple = DDG4.Geant4( kernel, tracker='Geant4TrackerAction',calo='Geant4CalorimeterAction')
## Apply BirksLaw effect for Scintillator Calorimeter by using 'Geant4ScintillatorCalorimeterAction'.
# simple = DDG4.Geant4( kernel, tracker='Geant4TrackerAction',calo='Geant4ScintillatorCalorimeterAction')
simple.printDetectors()
# Configure UI
#simple.setupCshUI()
simple.setupUI()
kernel.UI=""
kernel.NumEvents=numberOfEvents
#-----------------------------------------------------------------------------------
# setup the magnetic field:
field = simple.addConfig('Geant4FieldTrackingSetupAction/MagFieldTrackingSetup')
field.stepper = cfgDict['field.stepper']
field.equation = cfgDict['field.equation']
field.eps_min = cfgDict['field.eps_min']
field.eps_max = cfgDict['field.eps_max']
field.min_chord_step = cfgDict['field.min_chord_step']
field.delta_chord = cfgDict['field.delta_chord']
field.delta_intersection = cfgDict['field.delta_intersection']
field.delta_one_step = cfgDict['field.delta_one_step']
#----------------------------------------------------------------------------------
# Configure Run actions
run1 = DDG4.RunAction(kernel,'Geant4TestRunAction/RunInit')
kernel.registerGlobalAction(run1)
kernel.runAction().add(run1)
# Configure I/O
evt_lcio = simple.setupLCIOOutput('LcioOutput', lcioOutputFile )
gen = DDG4.GeneratorAction(kernel,"LCIOInputAction/LCIO1")
if( lcioInputFile[ (len(lcioInputFile)-6 ) : ] == ".slcio" ):
gen.Input="LCIOFileReader|"+lcioInputFile
else:
gen.Input="LCIOStdHepReader|"+lcioInputFile
simple.buildInputStage( [gen] , output_level=DDG4.OutputLevel.INFO )
#================================================================================================
# And handle the simulation particles.
part = DDG4.GeneratorAction(kernel,"Geant4ParticleHandler/ParticleHandler")
kernel.generatorAction().adopt(part)
#part.SaveProcesses = ['conv','Decay']
part.SaveProcesses = ['Decay']
part.MinimalKineticEnergy = 100*MeV
part.OutputLevel = Output.INFO #generator_output_level
part.enableUI()
user = DDG4.Action(kernel,"Geant4TCUserParticleHandler/UserParticleHandler")
user.TrackingVolume_Zmax = DDG4.tracker_region_zmax
user.TrackingVolume_Rmax = DDG4.tracker_region_rmax
# user.enableUI()
part.adopt(user)
#=================================================================================
# Setup global filters fur use in sensintive detectors
f1 = DDG4.Filter(kernel,'GeantinoRejectFilter/GeantinoRejector')
kernel.registerGlobalFilter(f1)
f4 = DDG4.Filter(kernel,'EnergyDepositMinimumCut')
f4.Cut = 1.*keV
kernel.registerGlobalFilter(f4)
#=================================================================================
# get lists of trackers and calorimeters in lcdd
trk,cal = getDetectorLists( lcdd )
# ---- add the trackers:
# fixme: this assumes the same filters for all trackers ...
for t in trk:
print 'simple.setupTracker( ' , t , ')'
seq,act = simple.setupTracker( t )
seq.add(f1)
act.HitCreationMode = 2
# ---- add the calorimeters:
for c in cal:
print 'simple.setupCalorimeter( ' , c , ')'
seq,act = simple.setupCalorimeter( c )
#=================================================================================
# Now build the physics list:
phys = simple.setupPhysics( physicsList )
#fg: do we need these really ?
#fg: ph = DDG4.PhysicsList(kernel,'Geant4PhysicsList/Myphysics')
#fg: ph.addParticleConstructor('G4BosonConstructor')
#fg: ph.addParticleConstructor('G4LeptonConstructor')
#fg: ph.addParticleProcess('e[+-]','G4eMultipleScattering',-1,1,1)
#fg: ph.addPhysicsConstructor('G4OpticalPhysics')
#fg: ph.enableUI()
#fg: phys.add(ph)
#fg: phys.dump()
kernel.configure()
kernel.initialize()
#DDG4.setPrintLevel(Output.DEBUG)
kernel.run()
kernel.terminate()
def run():
import LHeD, DDG4, os, SystemOfUnits
from DDG4 import OutputLevel as Output
lhed = LHeD.LHeD()
geant4 = lhed.geant4
kernel = lhed.kernel
lhed.loadGeometry()
geant4.printDetectors()
kernel.UI = "UI"
geant4.setupCshUI()
lhed.setupField(quiet=False)
DDG4.importConstants(kernel.detectorDescription(), debug=False)
dd4hep_dir = os.environ['DD4hep']
kernel.loadXML("file:" + dd4hep_dir +
"/examples/LHeD/scripts/DDG4_field.xml")
geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction')
geant4.printDetectors()
# Configure G4 magnetic field tracking
field = geant4.addConfig(
'Geant4FieldTrackingSetupAction/MagFieldTrackingSetup')
field.stepper = "HelixGeant4Runge"
field.equation = "Mag_UsualEqRhs"
field.eps_min = 5e-0520 * SystemOfUnits.mm
field.eps_max = 0.001 * SystemOfUnits.mm
field.min_chord_step = 0.01 * SystemOfUnits.mm
field.delta_chord = 0.25 * SystemOfUnits.mm
field.delta_intersection = 1e-05 * SystemOfUnits.mm
field.delta_one_step = 0.001 * SystemOfUnits.mm
print '+++++> ', field.name, '-> stepper = ', field.stepper
print '+++++> ', field.name, '-> equation = ', field.equation
print '+++++> ', field.name, '-> eps_min = ', field.eps_min
print '+++++> ', field.name, '-> eps_max = ', field.eps_max
print '+++++> ', field.name, '-> delta_one_step = ', field.delta_one_step
"""
# Setup random generator
rndm = DDG4.Action(kernel,'Geant4Random/Random')
rndm.Seed = 987654321
rndm.initialize()
rndm.showStatus()
rndm.Seed = 987654321
"""
# Configure Run actions
run1 = DDG4.RunAction(kernel, 'Geant4TestRunAction/RunInit')
"""
run1.Property_int = 12345
run1.Property_double = -5e15*keV
run1.Property_string = 'Startrun: Hello_LHeD'
print run1.Property_string, run1.Property_double, run1.Property_int
"""
run1.enableUI()
kernel.registerGlobalAction(run1)
kernel.runAction().adopt(run1)
# Configure Event actions
prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint')
prt.OutputLevel = Output.INFO
prt.OutputType = 3 # Print both: table and tree
kernel.eventAction().adopt(prt)
# Configure I/O
#evt_lcio = geant4.setupLCIOOutput('LcioOutput','Lhe_dip_sol_circ-higgs-bb')
#evt_lcio.OutputLevel = Output.ERROR
evt_root = geant4.setupROOTOutput('RootOutput',
'Lhe_dip_sol_circ-higgs-bb')
evt_root.OutputLevel = Output.INFO
gen = DDG4.GeneratorAction(kernel,
"Geant4GeneratorActionInit/GenerationInit")
kernel.generatorAction().adopt(gen)
"""
# First particle generator: e- non-isotropic generation using Gun:
gun = DDG4.GeneratorAction(kernel,"Geant4ParticleGenerator/Gun");
gun.Particle = 'e-'
gun.Energy = 60 * GeV
gun.Multiplicity = 1
gun.Position = (0.*mm,0.*mm,0.*mm)
gun.Direction = (1.,0.,0.)
gun.Mask = 2
gun.enableUI()
kernel.generatorAction().adopt(gun)
# Install vertex smearing for this primary e-
gen = DDG4.GeneratorAction(kernel,"Geant4InteractionVertexSmear/SmearE-");
gen.Mask = 1
gen.Sigma = (0*mm, 0*mm, 0*mm, 0*ns)
kernel.generatorAction().adopt(gen)
"""
#VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV
"""
Generation of isotrope tracks of a given multiplicity with overlay:
"""
#VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV
"""
# First particle generator: pi+
gen = DDG4.GeneratorAction(kernel,"Geant4IsotropeGenerator/IsotropPi+");
gen.Particle = 'pi+'
gen.Energy = 200*GeV
gen.Multiplicity = 1
gen.Mask = 1
kernel.generatorAction().adopt(gen)
# Install vertex smearing for this interaction
gen = DDG4.GeneratorAction(kernel,"Geant4InteractionVertexSmear/SmearPi+");
gen.Mask = 1
gen.Offset = (20*mm, 10*mm, 10*mm, 0*ns)
gen.Sigma = (4*mm, 1*mm, 1*mm, 0*ns)
kernel.generatorAction().adopt(gen)
# Second particle generator: e-
gen = DDG4.GeneratorAction(kernel,"Geant4IsotropeGenerator/IsotropE-");
gen.Particle = 'e-'
gen.Energy = 60 * GeV
gen.Multiplicity = 2
gen.Mask = 2
kernel.generatorAction().adopt(gen)
# Install vertex smearing for this interaction
gen = DDG4.GeneratorAction(kernel,"Geant4InteractionVertexSmear/SmearE-");
gen.Mask = 2
gen.Offset = (-20*mm, -10*mm, -10*mm, 0*ns)
gen.Sigma = (12*mm, 8*mm, 8*mm, 0*ns)
kernel.generatorAction().adopt(gen)
"""
#^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
#VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV
"""
Generation of primary particles from LCIO input files
"""
#VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV
# First particle file reader
gen = DDG4.GeneratorAction(kernel, "LCIOInputAction/LCIO1")
# gen.Input = "LCIOStdHepReader|/afs/.cern.ch/project/lhec/software/aidasoft/DD4hep/DD4hep/files/NC_bb_tag_2_pythia_events.hep"
#gen.Input = "LCIOStdHepReader|/opt/DD4hep/files/NC_bb_tag_2_pythia_events.hep"
gen.Input = "LCIOStdHepReader|/opt/DD4hep/files/lhec_for_peter/tag_2_pythia_events.hep"
# gen.Input = "LCIOStdHepReader|/opt/DD4hep/files/single-top-tag_1_pythia_events.hep"
# gen.Input = "Geant4EventReaderHepMC|/opt/DD4hep/files/ePb-q2-0-i.mc2"
# gen.Input = "LCIOStdHepReader|/opt/DD4hep/files/single-top-tag_1_pythia_events.hep"
# gen.Input = "LCIOStdHepReader|/opt/DD4hep/files/root.hep"
gen.OutputLevel = 2 # generator_output_level
gen.MomentumScale = 1.0
gen.Mask = 1
gen.enableUI()
kernel.generatorAction().adopt(gen)
# Install vertex smearing for this interaction
gen = DDG4.GeneratorAction(kernel, "Geant4InteractionVertexSmear/Smear1")
gen.OutputLevel = 4 #generator_output_level
gen.Mask = 1
gen.Offset = (-20 * SystemOfUnits.mm, -10 * SystemOfUnits.mm,
-10 * SystemOfUnits.mm, 0 * SystemOfUnits.ns)
gen.Sigma = (12 * SystemOfUnits.mm, 8 * SystemOfUnits.mm,
8 * SystemOfUnits.mm, 0 * SystemOfUnits.ns)
gen.enableUI()
kernel.generatorAction().adopt(gen)
#^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
# Merge all existing interaction records
gen = DDG4.GeneratorAction(kernel,
"Geant4InteractionMerger/InteractionMerger")
gen.OutputLevel = 4 #generator_output_level
gen.enableUI()
kernel.generatorAction().adopt(gen)
# Finally generate Geant4 primaries
gen = DDG4.GeneratorAction(kernel, "Geant4PrimaryHandler/PrimaryHandler")
gen.OutputLevel = 4 #generator_output_level
gen.enableUI()
kernel.generatorAction().adopt(gen)
# And handle the simulation particles.
part = DDG4.GeneratorAction(kernel,
"Geant4ParticleHandler/ParticleHandler")
kernel.generatorAction().adopt(part)
#part.SaveProcesses = ['conv','Decay']
part.SaveProcesses = ['Decay']
part.MinimalKineticEnergy = 10 * SystemOfUnits.MeV
part.OutputLevel = 5 # generator_output_level
part.enableUI()
user = DDG4.Action(kernel,
"Geant4TCUserParticleHandler/UserParticleHandler")
user.TrackingVolume_Zmax = DDG4.EcalEndcap_zmin_fwd
user.TrackingVolume_Rmax = DDG4.EcalBarrel_rmin
user.enableUI()
part.adopt(user)
"""
rdr = DDG4.GeneratorAction(kernel,"LcioGeneratorAction/Reader")
rdr.zSpread = 0.0
rdr.lorentzAngle = 0.0
rdr.OutputLevel = DDG4.OutputLevel.INFO
rdr.Input = "LcioEventReader|test.data"
rdr.enableUI()
kernel.generatorAction().adopt(rdr)
"""
# Setup global filters fur use in sensntive detectors
f1 = DDG4.Filter(kernel, 'GeantinoRejectFilter/GeantinoRejector')
kernel.registerGlobalFilter(f1)
f2 = DDG4.Filter(kernel, 'ParticleRejectFilter/OpticalPhotonRejector')
f2.particle = 'opticalphoton'
kernel.registerGlobalFilter(f2)
f3 = DDG4.Filter(kernel, 'ParticleSelectFilter/OpticalPhotonSelector')
f3.particle = 'opticalphoton'
kernel.registerGlobalFilter(f3)
f4 = DDG4.Filter(kernel, 'EnergyDepositMinimumCut')
f4.Cut = 0.5 * SystemOfUnits.MeV
f4.enableUI()
kernel.registerGlobalFilter(f4)
# First the tracking detectors
seq, act = geant4.setupTracker('SiVertexBarrel')
seq.adopt(f1)
act.adopt(f1)
seq, act = geant4.setupTracker('SiTrackerBarrel')
seq.adopt(f1)
act.adopt(f1)
seq, act = geant4.setupTracker('SiTrackerForward')
seq.adopt(f1)
act.adopt(f1)
seq, act = geant4.setupTracker('SiTrackerBackward')
seq.adopt(f1)
act.adopt(f1)
# Now the calorimeters
seq, act = geant4.setupCalorimeter('EcalBarrel')
seq.adopt(f3)
act.adopt(f3)
seq.adopt(f4)
act.adopt(f4)
seq, act = geant4.setupCalorimeter('EcalEndcap_fwd')
seq.adopt(f3)
act.adopt(f3)
seq.adopt(f4)
act.adopt(f4)
seq, act = geant4.setupCalorimeter('EcalEndcap_bwd')
seq.adopt(f3)
act.adopt(f3)
seq.adopt(f4)
act.adopt(f4)
seq, act = geant4.setupCalorimeter('HcalBarrel')
seq.adopt(f3)
act.adopt(f3)
seq.adopt(f4)
act.adopt(f4)
seq, act = geant4.setupCalorimeter('HcalEndcap_fwd')
seq.adopt(f3)
act.adopt(f3)
seq.adopt(f4)
act.adopt(f4)
seq, act = geant4.setupCalorimeter('HcalEndcap_bwd')
seq.adopt(f3)
act.adopt(f3)
seq.adopt(f4)
act.adopt(f4)
seq, act = geant4.setupCalorimeter('HcalPlug_fwd')
seq.adopt(f3)
act.adopt(f3)
seq.adopt(f4)
act.adopt(f4)
seq, act = geant4.setupCalorimeter('HcalPlug_bwd')
seq.adopt(f3)
act.adopt(f3)
seq.adopt(f4)
act.adopt(f4)
seq, act = geant4.setupCalorimeter('MuonBarrel')
seq.adopt(f2)
act.adopt(f2)
seq, act = geant4.setupCalorimeter('MuonEndcap_fwd1')
seq.adopt(f2)
act.adopt(f2)
seq, act = geant4.setupCalorimeter('MuonEndcap_fwd2')
seq.adopt(f2)
act.adopt(f2)
seq, act = geant4.setupCalorimeter('MuonEndcap_bwd1')
seq.adopt(f2)
act.adopt(f2)
seq, act = geant4.setupCalorimeter('MuonEndcap_bwd2')
seq.adopt(f2)
act.adopt(f2)
"""
scan = DDG4.SteppingAction(kernel,'Geant4MaterialScanner/MaterialScan')
kernel.steppingAction().adopt(scan)
"""
# Now build the physics list:
phys = geant4.setupPhysics('QGSP_BERT')
ph = DDG4.PhysicsList(kernel, 'Geant4PhysicsList/Myphysics')
ph.addParticleConstructor('G4Geantino')
ph.addParticleConstructor('G4BosonConstructor')
ph.addParticleConstructor('G4LeptonConstructor')
ph.addParticleProcess('e[+-]', 'G4eMultipleScattering', -1, 1, 1)
ph.addPhysicsConstructor('G4OpticalPhysics')
ph.enableUI()
phys.adopt(ph)
phys.dump()
kernel.configure()
kernel.initialize()
#DDG4.setPrintLevel(Output.DEBUG)
kernel.run()
print 'End of run. Terminating .......'
kernel.terminate()
def run():
global geant4
kernel = DDG4.Kernel()
lcdd = kernel.lcdd()
install_dir = os.environ['DD4hepINSTALL']
kernel.loadGeometry("file:"+install_dir+"/DDDetectors/compact/SiD_Markus.xml")
DDG4.importConstants(lcdd)
DDG4.Core.setPrintLevel(Output.DEBUG)
DDG4.Core.setPrintFormat("%-32s %6s %s")
kernel.NumberOfThreads = 1
geant4 = DDG4.Geant4(kernel,tracker='Geant4TrackerWeightedAction')
geant4.printDetectors()
# Configure UI
geant4.setupCshUI()
# Geant4 user initialization action
geant4.addUserInitialization(worker=setupWorker, master=setupMaster)
# Configure G4 geometry setup
seq,act = geant4.addDetectorConstruction("Geant4DetectorGeometryConstruction/ConstructGeo")
# Configure G4 magnetic field tracking
seq,fld = geant4.addDetectorConstruction("Geant4FieldTrackingConstruction/MagFieldTrackingSetup")
fld.stepper = "HelixGeant4Runge"
fld.equation = "Mag_UsualEqRhs"
fld.eps_min = 5e-05 * mm
fld.eps_max = 0.001 * mm
fld.min_chord_step = 0.01 * mm
fld.delta_chord = 0.25 * mm
fld.delta_intersection = 1e-05 * mm
fld.delta_one_step = 0.001 * mm
print '+++++> ',fld.name,'-> stepper = ',fld.stepper
print '+++++> ',fld.name,'-> equation = ',fld.equation
print '+++++> ',fld.name,'-> eps_min = ',fld.eps_min
print '+++++> ',fld.name,'-> eps_max = ',fld.eps_max
print '+++++> ',fld.name,'-> delta_one_step = ',fld.delta_one_step
seq,act = geant4.addDetectorConstruction("Geant4PythonDetectorConstruction/DummyDet",
geometry=dummy_geom,
sensitives=dummy_sd)
# Configure G4 sensitive detectors
seq,act = geant4.addDetectorConstruction("Geant4PythonDetectorConstruction/SetupSD",
sensitives=setupSensitives)
# Configure G4 sensitive detectors
seq,act = geant4.addDetectorConstruction("Geant4DetectorSensitivesConstruction/ConstructSD",
allow_threads=True)
# Setup random generator
rndm = DDG4.Action(kernel,'Geant4Random/Random')
rndm.Seed = 987654321
rndm.initialize()
# Setup global filters fur use in sensntive detectors
f1 = DDG4.Filter(kernel,'GeantinoRejectFilter/GeantinoRejector')
kernel.registerGlobalFilter(f1)
#seq,act = geant4.setupTracker('SiTrackerBarrel')
#seq,act = geant4.setupTracker('SiTrackerEndcap')
#seq,act = geant4.setupTracker('SiTrackerForward')
# Now the calorimeters
#seq,act = geant4.setupCalorimeter('EcalBarrel')
#seq,act = geant4.setupCalorimeter('EcalEndcap')
#seq,act = geant4.setupCalorimeter('HcalBarrel')
#seq,act = geant4.setupCalorimeter('HcalEndcap')
#seq,act = geant4.setupCalorimeter('HcalPlug')
#seq,act = geant4.setupCalorimeter('MuonBarrel')
#seq,act = geant4.setupCalorimeter('MuonEndcap')
#seq,act = geant4.setupCalorimeter('LumiCal')
#seq,act = geant4.setupCalorimeter('BeamCal')
# Now build the physics list:
seq = geant4.setupPhysics('QGSP_BERT')
phys = DDG4.PhysicsList(geant4.master(),'Geant4PhysicsList/MyPhysics')
part = DDG4.Action(geant4.master(),'Geant4ExtraParticles/extraparts')
part.pdgfile = 'checkout/DDG4/examples/particle.tbl'
phys.adoptPhysicsConstructor(part.get())
seq.add(phys)
geant4.run()
#kernel.configure()
#kernel.initialize()
#DDG4.setPrintLevel(Output.DEBUG)
#kernel.run()
#kernel.terminate()
return 1
def run():
hlp = False
vis = False
batch = False
#
for i in list(range(len(sys.argv))):
c = sys.argv[i].upper()
if c.find('BATCH') < 2 and c.find('BATCH') >= 0:
batch = True
elif c[:4] == '-VIS':
vis = True
elif c[:2] == '-H':
hlp = True
if hlp:
help()
sys.exit(1)
kernel = DDG4.Kernel()
description = kernel.detectorDescription()
install_dir = os.environ['DD4hepINSTALL']
kernel.loadGeometry(
str("file:" + install_dir + "/DDDetectors/compact/SiD.xml"))
DDG4.importConstants(description)
geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction')
geant4.printDetectors()
logger.info("# Configure UI")
ui = None
if batch:
geant4.setupCshUI(ui=None, vis=None)
kernel.UI = 'UI'
else:
ui = geant4.setupCshUI(vis=vis)
logger.info("# Configure G4 magnetic field tracking")
geant4.setupTrackingField()
logger.info("# Setup random generator")
rndm = DDG4.Action(kernel, 'Geant4Random/Random')
rndm.Seed = 987654321
rndm.initialize()
# rndm.showStatus()
logger.info("# Configure Run actions")
run1 = DDG4.RunAction(kernel, 'Geant4TestRunAction/RunInit')
run1.Property_int = 12345
run1.Property_double = -5e15 * keV
run1.Property_string = 'Startrun: Hello_2'
logger.info("%s %s %s", run1.Property_string, str(run1.Property_double),
str(run1.Property_int))
run1.enableUI()
kernel.registerGlobalAction(run1)
kernel.runAction().adopt(run1)
logger.info("# Configure Event actions")
prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint')
prt.OutputLevel = Output.INFO
prt.OutputType = 3 # Print both: table and tree
kernel.eventAction().adopt(prt)
logger.info("""
Configure I/O
""")
# evt_lcio = geant4.setupLCIOOutput('LcioOutput','CLICSiD_'+time.strftime('%Y-%m-%d_%H-%M'))
# evt_lcio.OutputLevel = Output.ERROR
geant4.setupROOTOutput('RootOutput',
'CLICSiD_' + time.strftime('%Y-%m-%d_%H-%M'))
gen = DDG4.GeneratorAction(kernel,
"Geant4GeneratorActionInit/GenerationInit")
kernel.generatorAction().adopt(gen)
# VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV
logger.info("""
Generation of isotrope tracks of a given multiplicity with overlay:
""")
logger.info("# First particle generator: pi+")
gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropPi+")
gen.Mask = 1
gen.Particle = 'pi+'
gen.Energy = 100 * GeV
gen.Multiplicity = 2
gen.Distribution = 'cos(theta)'
kernel.generatorAction().adopt(gen)
logger.info("# Install vertex smearing for this interaction")
gen = DDG4.GeneratorAction(kernel, "Geant4InteractionVertexSmear/SmearPi+")
gen.Mask = 1
gen.Offset = (20 * mm, 10 * mm, 10 * mm, 0 * ns)
gen.Sigma = (4 * mm, 1 * mm, 1 * mm, 0 * ns)
kernel.generatorAction().adopt(gen)
logger.info("# Second particle generator: e-")
gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropE-")
gen.Mask = 2
gen.Particle = 'e-'
gen.Energy = 25 * GeV
gen.Multiplicity = 2
gen.Distribution = 'uniform'
kernel.generatorAction().adopt(gen)
logger.info(" Install vertex smearing for this interaction")
gen = DDG4.GeneratorAction(kernel, "Geant4InteractionVertexSmear/SmearE-")
gen.Mask = 2
gen.Offset = (-20 * mm, -10 * mm, -10 * mm, 0 * ns)
gen.Sigma = (12 * mm, 8 * mm, 8 * mm, 0 * ns)
kernel.generatorAction().adopt(gen)
logger.info("# Second particle generator: mu+")
gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropMu+")
gen.Mask = 3
gen.Particle = 'mu+'
gen.Energy = 100 * GeV
gen.Multiplicity = 3
gen.Distribution = 'uniform'
kernel.generatorAction().adopt(gen)
# ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
logger.info("# Merge all existing interaction records")
gen = DDG4.GeneratorAction(kernel,
"Geant4InteractionMerger/InteractionMerger")
gen.OutputLevel = 4 # generator_output_level
gen.enableUI()
kernel.generatorAction().adopt(gen)
#
logger.info("# Finally generate Geant4 primaries")
gen = DDG4.GeneratorAction(kernel, "Geant4PrimaryHandler/PrimaryHandler")
gen.OutputLevel = 4 # generator_output_level
gen.enableUI()
kernel.generatorAction().adopt(gen)
#
logger.info("# ....and handle the simulation particles.")
part = DDG4.GeneratorAction(kernel,
"Geant4ParticleHandler/ParticleHandler")
kernel.generatorAction().adopt(part)
# part.SaveProcesses = ['conv','Decay']
part.SaveProcesses = ['Decay']
part.MinimalKineticEnergy = 100 * MeV
part.OutputLevel = 5 # generator_output_level
part.enableUI()
user = DDG4.Action(kernel,
"Geant4TCUserParticleHandler/UserParticleHandler")
user.TrackingVolume_Zmax = DDG4.EcalEndcap_zmin
user.TrackingVolume_Rmax = DDG4.EcalBarrel_rmin
user.enableUI()
part.adopt(user)
#
logger.info("# Setup global filters fur use in sensitive detectors")
f1 = DDG4.Filter(kernel, 'GeantinoRejectFilter/GeantinoRejector')
f2 = DDG4.Filter(kernel, 'ParticleRejectFilter/OpticalPhotonRejector')
f2.particle = 'opticalphoton'
f3 = DDG4.Filter(kernel, 'ParticleSelectFilter/OpticalPhotonSelector')
f3.particle = 'opticalphoton'
f4 = DDG4.Filter(kernel, 'EnergyDepositMinimumCut')
f4.Cut = 10 * MeV
f4.enableUI()
kernel.registerGlobalFilter(f1)
kernel.registerGlobalFilter(f2)
kernel.registerGlobalFilter(f3)
kernel.registerGlobalFilter(f4)
#
logger.info("# First the tracking detectors")
seq, act = geant4.setupTracker('SiVertexBarrel')
seq.adopt(f1)
act.adopt(f1)
#
seq, act = geant4.setupTracker('SiVertexEndcap')
seq.adopt(f1)
#
seq, act = geant4.setupTracker('SiTrackerBarrel')
seq, act = geant4.setupTracker('SiTrackerEndcap')
seq, act = geant4.setupTracker('SiTrackerForward')
logger.info("# Now setup the calorimeters")
seq, act = geant4.setupCalorimeter('EcalBarrel')
seq, act = geant4.setupCalorimeter('EcalEndcap')
seq, act = geant4.setupCalorimeter('HcalBarrel')
seq, act = geant4.setupCalorimeter('HcalEndcap')
seq, act = geant4.setupCalorimeter('HcalPlug')
seq, act = geant4.setupCalorimeter('MuonBarrel')
seq, act = geant4.setupCalorimeter('MuonEndcap')
seq, act = geant4.setupCalorimeter('LumiCal')
seq, act = geant4.setupCalorimeter('BeamCal')
#
logger.info("# Now build the physics list:")
phys = geant4.setupPhysics('QGSP_BERT')
ph = geant4.addPhysics(str('Geant4PhysicsList/Myphysics'))
ph.addPhysicsConstructor(str('G4StepLimiterPhysics'))
#
# Add special particle types from specialized physics constructor
part = geant4.addPhysics(str('Geant4ExtraParticles/ExtraParticles'))
part.pdgfile = os.path.join(install_dir,
'examples/DDG4/examples/particle.tbl')
#
# Add global range cut
rg = geant4.addPhysics(str('Geant4DefaultRangeCut/GlobalRangeCut'))
rg.RangeCut = 0.7 * mm
#
phys.dump()
#
#
if ui and vis:
cmds = []
cmds.append('/control/verbose 2')
cmds.append('/run/initialize')
cmds.append('/vis/open OGL')
cmds.append('/vis/verbose errors')
cmds.append('/vis/drawVolume')
cmds.append('/vis/viewer/set/viewpointThetaPhi 55. 45.')
cmds.append('/vis/scene/add/axes 0 0 0 10 m')
ui.Commands = cmds
kernel.configure()
kernel.initialize()
# DDG4.setPrintLevel(Output.DEBUG)
kernel.run()
kernel.terminate()
