def run(input_file): import DDG4 from DDG4 import OutputLevel as Output kernel = DDG4.Kernel() description = kernel.detectorDescription() gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/Input") kernel.generatorAction().adopt(gen) gen.Input = "Geant4EventReaderHepMC|/home/frankm/SW/data/hepmc_geant4.dat" gen.Input = "Geant4EventReaderHepMC|/home/frankm/SW/data/" gen.Input = "Geant4EventReaderHepMC|" + input_file gen.OutputLevel = Output.DEBUG gen.HaveAbort = False prim_vtx = DDG4.std_vector('dd4hep::sim::Geant4Vertex*')() parts = gen.new_particles() ret = 1 while ret: try: ret = gen.readParticles(0, prim_vtx, parts) except Exception, X: print '\nException: readParticles:', str(X) ret = None if ret: for v in prim_vtx: print 'Mask:%04X (X:%8.2g,Y:%8.2g,Z:%8.2g) time:%8.2g Outgoing:%4d particles'%\ (v.mask,v.x,v.y,v.z,v.time,len(v.out),) for p in parts: print 'Mask:%04X ID:%5d PDG-id:%8d Charge:%1d Mass:%8.3g Momentum:(%8.2g,%8.2g,%8.2g) '\ 'Vertex:(%8.2g,%8.2g,%8.2g) NDau:%2d Status:%08X'%\ (p.mask,p.id,p.pdgID,int(p.charge3()),p.mass,p.psx,p.psy,p.psz,\ p.vsx,p.vsy,p.vsz,len(p.daughters),p.status,) parts.clear() else: print '*** End of recordset or read failure.....' print 132 * '*'
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepINSTALL'] example_dir = install_dir + '/examples/DDG4/examples' kernel.setOutputLevel('Geant4Converter', Output.DEBUG) kernel.setOutputLevel('RootOutput', Output.INFO) kernel.setOutputLevel('ShellHandler', Output.DEBUG) kernel.setOutputLevel('Gun', Output.INFO) kernel.loadGeometry("file:" + install_dir + "/examples/ClientTests/compact/FCC_HcalBarrel.xml") geant4 = DDG4.Geant4(kernel) geant4.printDetectors() geant4.setupCshUI() # Configure field field = geant4.setupTrackingField(prt=True) # Configure I/O evt_root = geant4.setupROOTOutput('RootOutput', 'FCC_' + time.strftime('%Y-%m-%d_%H-%M'), mc_truth=False) # Setup particle gun geant4.setupGun("Gun", particle='pi-', energy=100 * GeV, multiplicity=1) # Now the calorimeters seq, act = geant4.setupTracker('HcalBarrel') seq, act = geant4.setupDetector('ContainmentShell', 'Geant4EscapeCounter') # Now build the physics list: phys = kernel.physicsList() phys.extends = 'QGSP_BERT' phys.enableUI() phys.dump() # and run geant4.execute()
def run(): kernel = DDG4.Kernel() try: install_dir = os.environ['DD4hepINSTALL'] except (KeyError): print " please set the environment variable DD4hepINSTALL " print " to your DD4hep installation path ! " exit(1) kernel.loadGeometry("file:" + compactFile) lcdd = kernel.detectorDescription() DDG4.importConstants(lcdd) #-------- inf = open(paramFile, 'r') outf = open(dictFile, 'w') names = readNames(inf) writeDictionary(names, outf) inf.close() outf.close()
def __init__(self, tracker='Geant4TrackerCombineAction', no_physics=True): self.kernel = DDG4.Kernel() self.description = self.kernel.detectorDescription() self.geant4 = DDG4.Geant4(self.kernel, tracker=tracker) self.kernel.UI = "" if no_physics: self.noPhysics()
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepINSTALL'] kernel.loadGeometry("file:" + install_dir + "/examples/ClientTests/compact/Assemblies.xml") # geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') geant4.printDetectors() # Configure UI geant4.setupCshUI() if len(sys.argv) >= 2 and sys.argv[1] == "batch": kernel.UI = '' # Configure field field = geant4.setupTrackingField(prt=True) # Configure I/O geant4.setupROOTOutput('RootOutput', 'Assemblies_' + time.strftime('%Y-%m-%d_%H-%M'), mc_truth=False) # Setup particle gun geant4.setupGun("Gun", particle='e-', energy=2 * GeV, position=(0.15 * mm, 0.12 * mm, 0.1 * cm), multiplicity=1) # First the tracking detectors seq, act = geant4.setupTracker('VXD') # Now build the physics list: phys = kernel.physicsList() phys.extends = 'QGSP_BERT' phys.enableUI() phys.dump() DDG4.setPrintLevel(DDG4.OutputLevel.DEBUG) geant4.execute()
def run(input_file): import DDG4 from DDG4 import OutputLevel as Output kernel = DDG4.Kernel() lcdd = kernel.lcdd() gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/Input") kernel.generatorAction().adopt(gen) gen.Input = "Geant4EventReaderHepMC|/home/frankm/SW/data/hepmc_geant4.dat" gen.Input = "Geant4EventReaderHepMC|/home/frankm/SW/data/" gen.Input = "Geant4EventReaderHepMC|" + input_file gen.OutputLevel = Output.DEBUG gen.HaveAbort = False prim_vtx = DDG4.Geant4Vertex() prim_vtx.x = 0.0 prim_vtx.y = 0.0 prim_vtx.z = 0.0 parts = gen.new_particles() ret = 1 while ret: try: ret = gen.readParticles(0, prim_vtx, parts) except Exception, X: print '\nException: readParticles:', str(X) ret = None if ret: for p in parts: print 'ID:%5d PDG-id:%8d Charge:%1d Mass:%8.3g Momentum:(%8.2g,%8.2g,%8.2g) '\ 'Vertex:(%8.2g,%8.2g,%8.2g) NDau:%2d Status:%08X'%\ (p.id,p.pdgID,int(p.charge3()),p.mass,p.psx,p.psy,p.psz,\ p.vsx,p.vsy,p.vsz,len(p.daughters),p.status) parts.clear() else: print '*** End of recordset or read failure.....' print 132 * '*'
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry("file:" + install_dir + "/examples/OpticalSurfaces/compact/OpNovice.xml") logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.DEBUG)
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepINSTALL'] kernel.setOutputLevel('Geant4Converter',Output.DEBUG) kernel.setOutputLevel('Gun',Output.INFO) kernel.detectorDescription().fromXML("file:"+install_dir+"/examples/DDCMS/data/dd4hep-config.xml"); kernel.NumEvents = 5 geant4 = DDG4.Geant4(kernel,tracker='Geant4TrackerCombineAction') geant4.printDetectors() geant4.setupCshUI() batch = False test = False for i in xrange(len(sys.argv)): arg = sys.argv[i].lower() if arg == 'batch': batch = True elif arg == 'test': test = True elif arg == 'numevents': kernel.NumEvents = int(sys.argv[i+1]) if batch or test: kernel.UI = '' # Configure field field = geant4.setupTrackingField(prt=True) # Configure I/O evt_root = geant4.setupROOTOutput('RootOutput','CMSTracker_'+time.strftime('%Y-%m-%d_%H-%M'),mc_truth=True) # Setup particle gun generators = [] generators.append(geant4.setupGun("GunPi-",particle='pi-',energy=300*GeV,multiplicity=1,Standalone=False,register=False, Mask=1)) if not test: generators.append(geant4.setupGun("GunPi+",particle='pi+',energy=300*GeV,multiplicity=1,Standalone=False,register=False, Mask=2)) generators.append(geant4.setupGun("GunE-",particle='e-',energy=100*GeV,multiplicity=1,Standalone=False,register=False, Mask=4)) generators.append(geant4.setupGun("GunE+",particle='e+',energy=100*GeV,multiplicity=1,Standalone=False,register=False, Mask=8)) geant4.buildInputStage(generators) # Now setup all tracking detectors for i in geant4.description.detectors(): o = DDG4.DetElement(i.second.ptr()) sd = geant4.description.sensitiveDetector(o.name()) if sd.isValid(): type = geant4.sensitive_types[sd.type()] print 'CMSTracker: Configure subdetector %-24s of type %s'%(o.name(),type,) geant4.setupDetector(o.name(),type) # And handle the simulation particles. part = DDG4.GeneratorAction(kernel,"Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) part.SaveProcesses = ['conv','Decay'] part.MinimalKineticEnergy = 1*MeV part.OutputLevel = 5 # generator_output_level part.enableUI() # Now build the physics list: phys = kernel.physicsList() phys.extends = 'QGSP_BERT' phys.enableUI() # and run geant4.execute()
def run(): kernel = DDG4.Kernel() description = kernel.detectorDescription() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry( str("file:" + install_dir + "/examples/ClientTests/compact/LheD_tracker.xml")) DDG4.importConstants(description, debug=False) geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') geant4.printDetectors() # Configure UI geant4.setupCshUI() # geant4.setupCshUI('csh',True,True) # geant4.setupCshUI('csh',True,True,'vis.mac') if len(sys.argv) >= 2 and sys.argv[1] == "batch": kernel.UI = '' # Configure field geant4.setupTrackingField(prt=True) # Configure Event actions prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint') prt.OutputLevel = Output.WARNING # Output.WARNING prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) # Configure I/O geant4.setupROOTOutput('RootOutput', 'LHeD_tracker_' + time.strftime('%Y-%m-%d_%H-%M')) gen = geant4.setupGun("Gun", particle='geantino', energy=20 * GeV, position=(0 * mm, 0 * mm, 0 * cm), multiplicity=3) gen.isotrop = False gen.direction = (1, 0, 0) gen.OutputLevel = Output.WARNING # seq,act = geant4.setupTracker('SiVertexBarrel') # Now build the physics list: phys = geant4.setupPhysics('QGSP_BERT') ph = DDG4.PhysicsList(kernel, 'Geant4PhysicsList/Myphysics') ph.addParticleConstructor(str('G4Geantino')) ph.addParticleConstructor(str('G4BosonConstructor')) ph.enableUI() phys.adopt(ph) phys.dump() kernel.configure() kernel.initialize() # DDG4.setPrintLevel(Output.DEBUG) kernel.run() logger.info('End of run. Terminating .......') kernel.terminate()
def __init__(self, geometry_file): self.kernel = DDG4.Kernel() self.kernel.setOutputLevel(str('Geant4Converter'), Output.DEBUG) self.kernel.setOutputLevel(str('Gun'), Output.INFO) self.kernel.loadGeometry(str(geometry_file)) self.geant4 = DDG4.Geant4(self.kernel) self.geant4.printDetectors() self.geant4.setupCshUI()
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepINSTALL'] kernel.setOutputLevel('Geant4Converter', Output.DEBUG) kernel.setOutputLevel('Gun', Output.INFO) kernel.loadGeometry("file:" + install_dir + "/examples/ClientTests/compact/MiniTel.xml") geant4 = DDG4.Geant4(kernel) geant4.printDetectors() geant4.setupCshUI() if len(sys.argv) >= 2 and sys.argv[1] == "batch": kernel.NumEvents = 200 kernel.UI = '' # Configure field field = geant4.setupTrackingField(prt=True) # Setup particle gun geant4.setupGun("Gun", particle='pi-', energy=100 * GeV, multiplicity=1) # Now the calorimeters seq, act = geant4.setupTracker('MyLHCBdetector1') seq, act = geant4.setupTracker('MyLHCBdetector2') seq, act = geant4.setupTracker('MyLHCBdetector3') seq, act = geant4.setupTracker('MyLHCBdetector4') act.OutputLevel = 4 seq, act = geant4.setupTracker('MyLHCBdetector5') seq, act = geant4.setupTracker('MyLHCBdetector6') seq, act = geant4.setupTracker('MyLHCBdetector7') seq, act = geant4.setupTracker('MyLHCBdetector8') seq, act = geant4.setupTracker('MyLHCBdetector9') seq, act = geant4.setupTracker('MyLHCBdetector10') # And handle the simulation particles. part = DDG4.GeneratorAction(kernel, "Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) part.SaveProcesses = ['conv', 'Decay'] part.MinimalKineticEnergy = 1 * MeV part.OutputLevel = 5 # generator_output_level part.enableUI() hit_tuple = DDG4.EventAction(kernel, 'HitTupleAction/MiniTelTuple', True) hit_tuple.OutputFile = 'MiniTel_EnergyDeposits_' + time.strftime( '%Y-%m-%d_%H-%M') + '.root' hit_tuple.Collections = ['*'] kernel.eventAction().add(hit_tuple) # Now build the physics list: phys = kernel.physicsList() phys.extends = 'QGSP_BERT' phys.enableUI() # and run geant4.execute()
def run(): kernel = DDG4.Kernel() description = kernel.detectorDescription() install_dir = os.environ['DD4hepINSTALL'] DDG4.Core.setPrintFormat(str("%-32s %6s %s")) kernel.loadGeometry( str("file:" + install_dir + "/DDDetectors/compact/SiD.xml")) DDG4.importConstants(description) kernel.NumberOfThreads = 3 kernel.RunManagerType = 'G4MTRunManager' geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') logger.info("# Configure UI") geant4.setupCshUI() logger.info("# Geant4 user initialization action") geant4.addUserInitialization(worker=setupWorker, worker_args=(geant4, ), master=setupMaster, master_args=(geant4, )) logger.info("# Configure G4 geometry setup") seq, act = geant4.addDetectorConstruction( "Geant4DetectorGeometryConstruction/ConstructGeo") logger.info("# Configure G4 sensitive detectors: python setup callback") seq, act = geant4.addDetectorConstruction( "Geant4PythonDetectorConstruction/SetupSD", sensitives=setupSensitives, sensitives_args=(geant4, )) logger.info( "# Configure G4 sensitive detectors: atach'em to the sensitive volumes" ) seq, act = geant4.addDetectorConstruction( "Geant4DetectorSensitivesConstruction/ConstructSD") # allow_threads=True) logger.info("# Configure G4 magnetic field tracking") geant4.setupTrackingFieldMT() logger.info("# Setup random generator") rndm = DDG4.Action(kernel, 'Geant4Random/Random') rndm.Seed = 987654321 rndm.initialize() # rndm.showStatus() logger.info("# Now build the physics list:") phys = geant4.setupPhysics('QGSP_BERT') phys.dump() geant4.run()
def run(): batch = False kernel = DDG4.Kernel() install_dir = os.environ['DD4hepINSTALL'] geometry = "file:" + install_dir + "/examples/ClientTests/compact/MultiCollections.xml" kernel.setOutputLevel('Geant4Converter', Output.DEBUG) kernel.setOutputLevel('Gun', Output.INFO) for i in xrange(len(sys.argv)): if sys.argv[i] == '-compact': geometry = sys.argv[i + 1] elif sys.argv[i] == '-input': geometry = sys.argv[i + 1] elif sys.argv[i] == '-batch': batch = True elif sys.argv[i] == 'batch': batch = True kernel.loadGeometry(geometry) geant4 = DDG4.Geant4(kernel) geant4.printDetectors() geant4.setupCshUI() if batch: kernel.UI = '' # Configure field field = geant4.setupTrackingField(prt=True) # Configure I/O evt_root = geant4.setupROOTOutput('RootOutput', 'Multi_coll_' + time.strftime('%Y-%m-%d_%H-%M'), mc_truth=True) # Setup particle gun geant4.setupGun("Gun", particle='pi-', energy=10 * GeV, multiplicity=1) # Now the test calorimeter with multiple collections seq, act = geant4.setupCalorimeter('TestCal') # And handle the simulation particles. part = DDG4.GeneratorAction(kernel, "Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) part.MinimalKineticEnergy = 1 * MeV part.enableUI() # Now build the physics list: phys = kernel.physicsList() phys.extends = 'QGSP_BERT' phys.enableUI() phys.dump() # and run geant4.execute()
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry("file:" + install_dir + "/examples/ClientTests/compact/SiliconBlock.xml") DDG4.importConstants(kernel.detectorDescription(), debug=False) # ======================================================================================= # ===> This is actually the ONLY difference to ClientTests/scripts/SiliconBlock.py # ======================================================================================= geant4 = DDG4.Geant4(kernel, tracker='MyTrackerSDAction') geant4.printDetectors() kernel.NumEvents = 5 kernel.UI = '' # Configure field field = geant4.setupTrackingField(prt=True) # Configure Event actions prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint') prt.OutputLevel = Output.WARNING prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) # Configure I/O evt_root = geant4.setupROOTOutput('RootOutput', 'MySD_' + time.strftime('%Y-%m-%d_%H-%M'), mc_truth=False) # Setup particle gun gun = geant4.setupGun("Gun", particle='mu-', energy=5 * GeV, multiplicity=1, Standalone=True, position=(0, 0, 0)) seq, act = geant4.setupTracker('SiliconBlockUpper') act.OutputLevel = Output.INFO seq, act = geant4.setupTracker('SiliconBlockDown') act.OutputLevel = Output.INFO # Now build the physics list: phys = kernel.physicsList() phys.extends = 'QGSP_BERT' phys.enableUI() phys.dump() # run kernel.configure() kernel.initialize() kernel.run() kernel.terminate()
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry(str("file:" + install_dir + "/examples/ClientTests/compact/TrackingRegion.xml")) geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') # Configure field # field = geant4.setupTrackingField(prt=True) # Configure G4 geometry setup seq, act = geant4.addDetectorConstruction("Geant4DetectorGeometryConstruction/ConstructGeo") act.DebugVolumes = True act.DebugPlacements = True geant4.setupTracker('SiliconBlock') kernel.configure() kernel.initialize() kernel.terminate() sys.exit(0)
def run(): cnt = 0 geo = None vis = False batch = False for i in sys.argv: cnt = cnt + 1 c = i.upper() if c.find('BATCH') < 2 and c.find('BATCH') >= 0: batch = True elif c[:4] == '-GEO': geo = sys.argv[cnt] elif c[:4] == '-VIS': vis = True if not geo: help() sys.exit(1) import DDG4 kernel = DDG4.Kernel() # Configure UI geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') if batch: ui = geant4.setupCshUI(typ=None, ui=None, vis=None) kernel.UI = 'UI' else: ui = geant4.setupCshUI(vis=vis) kernel.loadGeometry(geo) # Configure field geant4.setupTrackingField(prt=True) # Now build the physics list: geant4.setupPhysics('') kernel.physicsList().enableUI() DDG4.setPrintLevel(DDG4.OutputLevel.DEBUG) # ui.Commands = [ '/ddg4/ConstructGeometry/printVolume /world_volume_1', '/ddg4/ConstructGeometry/printMaterial Air', '/ddg4/ConstructGeometry/printMaterial Vacuum', '/ddg4/UI/exit' ] kernel.NumEvents = 0 kernel.configure() kernel.initialize() kernel.run() kernel.terminate()
def run(): geo = None 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] == '-GEO': geo = sys.argv[i + 1] elif c[:4] == '-VIS': vis = True if not geo: help() sys.exit(1) import DDG4 kernel = DDG4.Kernel() # Configure UI geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') if batch: ui = geant4.setupCshUI(ui=None, vis=None) kernel.UI = 'UI' else: ui = geant4.setupCshUI(vis=vis) kernel.loadGeometry(geo) # Configure field geant4.setupTrackingField(prt=True) # Now build the physics list: geant4.setupPhysics('') kernel.physicsList().enableUI() DDG4.setPrintLevel(DDG4.OutputLevel.DEBUG) # cmds = [] cmds.append( '/ddg4/ConstructGeometry/printVolume /world_volume_1/Shape_Test_0/Shape_Test_vol_0_0' ) cmds.append('/ddg4/UI/exit') ui.Commands = cmds kernel.NumEvents = 0 kernel.configure() kernel.initialize() kernel.run() kernel.terminate()
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry( str("file:" + install_dir + "/examples/ClientTests/compact/GdmlDetector.xml")) DDG4.importConstants(kernel.detectorDescription(), debug=False) geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') geant4.printDetectors() # Configure UI if len(sys.argv) > 1: geant4.setupCshUI(macro=sys.argv[1]) else: geant4.setupCshUI() # Configure field geant4.setupTrackingField(prt=True) generator_output_level = Output.INFO # Configure G4 geometry setup seq, act = geant4.addDetectorConstruction( "Geant4DetectorGeometryConstruction/ConstructGeo") act.DebugVolumes = True act.DebugShapes = True act.DebugPlacements = True act.DumpHierarchy = True # Setup particle gun gun = geant4.setupGun("Gun", particle='mu-', energy=20 * GeV, multiplicity=1) gun.OutputLevel = generator_output_level # Now build the physics list: phys = geant4.setupPhysics('QGSP_BERT') phys.dump() geant4.execute()
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.setOutputLevel(str('Geant4Converter'), Output.DEBUG) kernel.setOutputLevel(str('Gun'), Output.INFO) kernel.loadGeometry( str("file:" + install_dir + "/examples/ClientTests/compact/NestedDetectors.xml")) geant4 = DDG4.Geant4(kernel) geant4.printDetectors() geant4.setupCshUI() if len(sys.argv) >= 2 and sys.argv[1] == "batch": kernel.UI = '' # Configure field geant4.setupTrackingField(prt=True) # Configure I/O geant4.setupROOTOutput('RootOutput', 'Nested_' + time.strftime('%Y-%m-%d_%H-%M'), mc_truth=True) # Setup particle gun geant4.setupGun("Gun", particle='pi-', energy=100 * GeV, multiplicity=1) # Now the calorimeters seq, act = geant4.setupTracker('SiTrackerBarrel') seq, act = geant4.setupTracker('SiVertexBarrel') # And handle the simulation particles. part = DDG4.GeneratorAction(kernel, "Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) part.SaveProcesses = ['conv', 'Decay'] part.MinimalKineticEnergy = 1 * MeV part.enableUI() # Now build the physics list: phys = kernel.physicsList() phys.extends = 'QGSP_BERT' phys.enableUI() phys.dump() # and run geant4.execute()
def run(): import os import DDG4 import g4units kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry( str("file:" + install_dir + "/examples/LHeD/compact/compact.xml")) DDG4.Core.setPrintFormat(str("%-32s %6s %s")) geant4 = DDG4.Geant4(kernel) # Configure UI geant4.setupCshUI(ui=None) gun = geant4.setupGun("Gun", Standalone=True, particle='geantino', energy=20 * g4units.GeV, position=(0, 0, 0), multiplicity=1, isotrop=False) scan = DDG4.SteppingAction(kernel, 'Geant4MaterialScanner/MaterialScan') kernel.steppingAction().adopt(scan) # Now build the physics list: geant4.setupPhysics('QGSP_BERT') kernel.configure() kernel.initialize() kernel.NumEvents = 1 # 3 shots in different directions: gun.direction = (0, 1, 0) kernel.run() gun.direction = (1, 0, 0) kernel.run() gun.direction = (1, 1, 1) kernel.run() kernel.terminate() logger.info('End of run. Terminating .......') logger.info('TEST_PASSED')
def run(input_file): import DDG4 from DDG4 import OutputLevel as Output kernel = DDG4.Kernel() kernel.detectorDescription() gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/Input") kernel.generatorAction().adopt(gen) gen.Input = "Geant4EventReaderHepMC|" + input_file gen.OutputLevel = Output.DEBUG gen.HaveAbort = False prim_vtx = DDG4.std_vector(str('dd4hep::sim::Geant4Vertex*'))() parts = gen.new_particles() ret = 1 evtid = 0 while ret: try: parts.clear() prim_vtx.clear() ret = gen.readParticles(evtid, prim_vtx, parts) evtid = evtid + 1 except Exception as X: logger.error('\nException: readParticles: %s', str(X)) if evtid > 0: os._exit(0) if ret: for v in prim_vtx: logger.info( 'Mask:%04X (X:%8.2g,Y:%8.2g,Z:%8.2g) time:%8.2g Outgoing:%4d particles', v.mask, v.x, v.y, v.z, v.time, len(v.out)) for p in parts: logger.info( 'Mask:%04X ID:%5d PDG-id:%8d Charge:%1d Mass:%8.3g Momentum:(%8.2g,%8.2g,%8.2g) ' 'Vertex:(%8.2g,%8.2g,%8.2g) NDau:%2d Status:%08X', p.mask, p.id, p.pdgID, int(p.charge3()), p.mass, p.psx, p.psy, p.psz, p.vsx, p.vsy, p.vsz, len(p.daughters), p.status) parts.clear() else: logger.info('*** End of recordset or read failure.....') logger.info(132 * '*') return 0
def run(): import os, sys, logging, DDG4, SystemOfUnits logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.DEBUG) kernel = DDG4.Kernel() install_dir = os.environ['DD4hepINSTALL'] kernel.loadGeometry("file:" + install_dir + "/DDDetectors/compact/SiD.xml") DDG4.Core.setPrintFormat("%-32s %6s %s") geant4 = DDG4.Geant4(kernel) # Configure UI geant4.setupCshUI(ui=None) gun = geant4.setupGun("Gun", Standalone=True, particle='geantino', energy=20 * SystemOfUnits.GeV, position=(0, 0, 0), multiplicity=1, isotrop=False) scan = DDG4.SteppingAction(kernel, 'Geant4MaterialScanner/MaterialScan') kernel.steppingAction().adopt(scan) # Now build the physics list: phys = geant4.setupPhysics('QGSP_BERT') kernel.configure() kernel.initialize() kernel.NumEvents = 1 # 3 shots in different directions: gun.direction = (0, 1, 0) kernel.run() gun.direction = (1, 0, 0) kernel.run() gun.direction = (1, 1, 1) kernel.run() kernel.terminate() logging.info('End of run. Terminating .......') logging.info('TEST_PASSED')
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry( str("file:" + install_dir + "/examples/OpticalSurfaces/compact/OpNovice.xml")) DDG4.importConstants(kernel.detectorDescription(), debug=False) geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') geant4.printDetectors() # Configure UI if len(sys.argv) > 1: geant4.setupCshUI(macro=sys.argv[1]) else: geant4.setupCshUI() # Configure field geant4.setupTrackingField(prt=True) # Configure Event actions prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint') prt.OutputLevel = Output.DEBUG prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) generator_output_level = Output.INFO # Configure G4 geometry setup seq, act = geant4.addDetectorConstruction( "Geant4DetectorGeometryConstruction/ConstructGeo") act.DebugMaterials = True act.DebugElements = False act.DebugVolumes = True act.DebugShapes = True act.DebugSurfaces = True # Configure I/O # evt_root = geant4.setupROOTOutput('RootOutput','OpNovice_'+time.strftime('%Y-%m-%d_%H-%M')) # Setup particle gun gun = geant4.setupGun("Gun", particle='gamma', energy=5 * keV, multiplicity=1) gun.OutputLevel = generator_output_level # And handle the simulation particles. """ part = DDG4.GeneratorAction(kernel,"Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) 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 = 3.0*m user.TrackingVolume_Rmax = 3.0*m user.enableUI() part.adopt(user) """ geant4.setupTracker('BubbleDevice') # Now build the physics list: phys = geant4.setupPhysics('') ph = DDG4.PhysicsList(kernel, 'Geant4OpticalPhotonPhysics/OpticalGammaPhys') ph.VerboseLevel = 2 ph.addParticleGroup('G4BosonConstructor') ph.addParticleGroup('G4LeptonConstructor') ph.addParticleGroup('G4MesonConstructor') ph.addParticleGroup('G4BaryonConstructor') ph.addParticleGroup('G4IonConstructor') ph.addParticleConstructor('G4OpticalPhoton') ph.addDiscreteParticleProcess('gamma', 'G4GammaConversion') ph.addDiscreteParticleProcess('gamma', 'G4ComptonScattering') ph.addDiscreteParticleProcess('gamma', 'G4PhotoElectricEffect') ph.addParticleProcess(str('e[+-]'), str('G4eMultipleScattering'), -1, 1, 1) ph.addParticleProcess(str('e[+-]'), str('G4eIonisation'), -1, 2, 2) ph.addParticleProcess(str('e[+-]'), str('G4eBremsstrahlung'), -1, 3, 3) ph.addParticleProcess(str('e+'), str('G4eplusAnnihilation'), 0, -1, 4) ph.addParticleProcess(str('mu[+-]'), str('G4MuMultipleScattering'), -1, 1, 1) ph.addParticleProcess(str('mu[+-]'), str('G4MuIonisation'), -1, 2, 2) ph.addParticleProcess(str('mu[+-]'), str('G4MuBremsstrahlung'), -1, 3, 3) ph.addParticleProcess(str('mu[+-]'), str('G4MuPairProduction'), -1, 4, 4) ph.enableUI() phys.adopt(ph) ph = DDG4.PhysicsList(kernel, 'Geant4ScintillationPhysics/ScintillatorPhys') ph.ScintillationYieldFactor = 1.0 ph.ScintillationExcitationRatio = 1.0 ph.TrackSecondariesFirst = False ph.VerboseLevel = 2 ph.enableUI() phys.adopt(ph) ph = DDG4.PhysicsList(kernel, 'Geant4CerenkovPhysics/CerenkovPhys') ph.MaxNumPhotonsPerStep = 10 ph.MaxBetaChangePerStep = 10.0 ph.TrackSecondariesFirst = True ph.VerboseLevel = 2 ph.enableUI() phys.adopt(ph) phys.dump() geant4.execute()
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry("file:" + install_dir + "/examples/OpticalSurfaces/compact/OpNovice.xml")
def run(): geo = None vis = False dump = False batch = False install_dir = os.environ['DD4hepINSTALL'] # 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] == '-GEO': geo = sys.argv[i + 1] elif c[:4] == '-VIS': vis = True elif c[:4] == '-DUM': dump = True if not geo: help() sys.exit(1) import DDG4 kernel = DDG4.Kernel() description = kernel.detectorDescription() DDG4.setPrintLevel(DDG4.OutputLevel.INFO) DDG4.importConstants(description) # # Configure UI geant4 = DDG4.Geant4(kernel) ui = None if batch: geant4.setupCshUI(ui=None, vis=None) kernel.UI = 'UI' else: ui = geant4.setupCshUI(vis=vis) Output = DDG4.OutputLevel seq, act = geant4.addDetectorConstruction( "Geant4DetectorGeometryConstruction/ConstructGeo") act.DebugReflections = True act.DebugMaterials = False act.DebugElements = False act.DebugVolumes = False act.DebugShapes = False if dump: act.DumpHierarchy = ~0x0 # kernel.setOutputLevel(str('Geant4Converter'), Output.WARNING) kernel.loadGeometry(geo) # geant4.printDetectors() # Configure field geant4.setupTrackingField(prt=True) logger.info("# Setup random generator") rndm = DDG4.Action(kernel, 'Geant4Random/Random') rndm.Seed = 987654321 rndm.initialize() # # Setup detector seq, act = geant4.setupCalorimeter('NestedBox') # # Configure I/O geant4.setupROOTOutput('RootOutput', 'Reflections_' + time.strftime('%Y-%m-%d_%H-%M'), mc_truth=True) # # Setup particle gun geant4.setupGun(name="Gun", particle='e-', energy=1000 * GeV, isotrop=True, multiplicity=1, position=(0 * m, 0 * m, 0 * m), PhiMin=0.0 * rad, PhiMax=math.pi * 2.0 * rad, ThetaMin=0.0 * rad, ThetaMax=math.pi * rad) logger.info("# ....and handle the simulation particles.") part = DDG4.GeneratorAction(kernel, str('Geant4ParticleHandler/ParticleHandler')) kernel.generatorAction().adopt(part) part.MinimalKineticEnergy = 100 * MeV part.SaveProcesses = ['Decay'] part.OutputLevel = 5 # generator_output_level part.enableUI() user = DDG4.Action(kernel, str('Geant4TCUserParticleHandler/UserParticleHandler')) user.TrackingVolume_Rmax = 3.0 * m user.TrackingVolume_Zmax = 2.0 * m user.enableUI() part.adopt(user) # # prt = DDG4.EventAction(kernel, str('Geant4ParticlePrint/ParticlePrint')) prt.OutputLevel = Output.INFO prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) # # Now build the physics list: phys = geant4.setupPhysics(str('QGSP_BERT')) ph = geant4.addPhysics(str('Geant4PhysicsList/Myphysics')) ph.addPhysicsConstructor(str('G4StepLimiterPhysics')) ph.addParticleConstructor(str('G4Geantino')) ph.addParticleConstructor(str('G4BosonConstructor')) # # Add special particle types from specialized physics constructor part = geant4.addPhysics('Geant4ExtraParticles/ExtraParticles') part.pdgfile = os.path.join(install_dir, 'examples/DDG4/examples/particle.tbl') # # Add global range cut rg = geant4.addPhysics('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 3 m') ui.Commands = cmds kernel.NumEvents = 0 kernel.configure() kernel.initialize() kernel.run() kernel.terminate()
def __init__(self, tracker='Geant4TrackerCombineAction'): self.kernel = DDG4.Kernel() self.lcdd = self.kernel.lcdd() self.geant4 = DDG4.Geant4(self.kernel, tracker=tracker) self.kernel.UI = "" self.noPhysics()
def run(self): """setup the geometry and dd4hep and geant4 and do what was asked to be done""" import ROOT ROOT.PyConfig.IgnoreCommandLineOptions = True import DDG4, dd4hep self.printLevel = getOutputLevel(self.printLevel) kernel = DDG4.Kernel() dd4hep.setPrintLevel(self.printLevel) #kernel.setOutputLevel('Compact',1) kernel.loadGeometry("file:" + self.compactFile) detectorDescription = kernel.detectorDescription() DDG4.importConstants(detectorDescription) #---------------------------------------------------------------------------------- #simple = DDG4.Geant4( kernel, tracker='Geant4TrackerAction',calo='Geant4CalorimeterAction') #simple = DDG4.Geant4( kernel, tracker='Geant4TrackerCombineAction',calo='Geant4ScintillatorCalorimeterAction') simple = DDG4.Geant4(kernel, tracker=self.action.tracker, calo=self.action.calo) simple.printDetectors() if self.runType == "vis": simple.setupUI(typ="csh", vis=True, macro=self.macroFile) elif self.runType == "run": simple.setupUI(typ="csh", vis=False, macro=self.macroFile, ui=False) elif self.runType == "shell": simple.setupUI(typ="csh", vis=False, macro=None, ui=True) elif self.runType == "batch": simple.setupUI(typ="csh", vis=False, macro=None, ui=False) else: print "ERROR: unknown runType" exit(1) #kernel.UI="csh" kernel.NumEvents = self.numberOfEvents #----------------------------------------------------------------------------------- # setup the magnetic field: self.__setMagneticFieldOptions(simple) #---------------------------------------------------------------------------------- # Configure Run actions run1 = DDG4.RunAction(kernel, 'Geant4TestRunAction/RunInit') kernel.registerGlobalAction(run1) kernel.runAction().add(run1) # Configure the random seed, do it before the I/O because we might change the seed! _rndm = self.random.initialize(DDG4, kernel, self.output.random) # Configure I/O if self.outputFile.endswith(".slcio"): lcOut = simple.setupLCIOOutput('LcioOutput', self.outputFile) lcOut.RunHeader = self.meta.addParametersToRunHeader(self) lcOut.EventParametersString, lcOut.EventParametersInt, lcOut.EventParametersFloat = self.meta.parseEventParameters( ) lcOut.RunNumberOffset = self.meta.runNumberOffset if self.meta.runNumberOffset > 0 else 0 lcOut.EventNumberOffset = self.meta.eventNumberOffset if self.meta.eventNumberOffset > 0 else 0 elif self.outputFile.endswith(".root"): simple.setupROOTOutput('RootOutput', self.outputFile) actionList = [] if self.enableGun: gun = DDG4.GeneratorAction(kernel, "Geant4ParticleGun/" + "Gun") self.gun.setOptions(gun) gun.Standalone = False gun.Mask = 1 actionList.append(gun) self.__applyBoostOrSmear(kernel, actionList, 1) print "++++ Adding DD4hep Particle Gun ++++" if self.enableG4Gun: ## GPS Create something self._g4gun = DDG4.GeneratorAction(kernel, "Geant4GeneratorWrapper/Gun") self._g4gun.Uses = 'G4ParticleGun' self._g4gun.Mask = 2 print "++++ Adding Geant4 Particle Gun ++++" actionList.append(self._g4gun) if self.enableG4GPS: ## GPS Create something self._g4gps = DDG4.GeneratorAction(kernel, "Geant4GeneratorWrapper/GPS") self._g4gps.Uses = 'G4GeneralParticleSource' self._g4gps.Mask = 3 print "++++ Adding Geant4 General Particle Source ++++" actionList.append(self._g4gps) for index, inputFile in enumerate(self.inputFiles, start=4): if inputFile.endswith(".slcio"): gen = DDG4.GeneratorAction(kernel, "LCIOInputAction/LCIO%d" % index) gen.Parameters = self.lcio.getParameters() gen.Input = "LCIOFileReader|" + inputFile elif inputFile.endswith(".stdhep"): gen = DDG4.GeneratorAction(kernel, "LCIOInputAction/STDHEP%d" % index) gen.Input = "LCIOStdHepReader|" + inputFile elif inputFile.endswith(".HEPEvt"): gen = DDG4.GeneratorAction( kernel, "Geant4InputAction/HEPEvt%d" % index) gen.Input = "Geant4EventReaderHepEvtShort|" + inputFile elif inputFile.endswith(".hepevt"): gen = DDG4.GeneratorAction( kernel, "Geant4InputAction/hepevt%d" % index) gen.Input = "Geant4EventReaderHepEvtLong|" + inputFile elif inputFile.endswith(".hepmc"): gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/hepmc%d" % index) gen.Input = "Geant4EventReaderHepMC|" + inputFile elif inputFile.endswith(".pairs"): gen = DDG4.GeneratorAction( kernel, "Geant4InputAction/GuineaPig%d" % index) gen.Input = "Geant4EventReaderGuineaPig|" + inputFile gen.Parameters = self.guineapig.getParameters() else: ##this should never happen because we already check at the top, but in case of some LogicError... raise RuntimeError("Unknown input file type: %s" % inputFile) gen.Sync = self.skipNEvents gen.Mask = index actionList.append(gen) self.__applyBoostOrSmear(kernel, actionList, index) if actionList: self._buildInputStage(simple, actionList, output_level=self.output.inputStage, have_mctruth=self._enablePrimaryHandler()) #================================================================================================ # And handle the simulation particles. part = DDG4.GeneratorAction(kernel, "Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) #part.SaveProcesses = ['conv','Decay'] part.SaveProcesses = self.part.saveProcesses part.MinimalKineticEnergy = self.part.minimalKineticEnergy part.KeepAllParticles = self.part.keepAllParticles part.PrintEndTracking = self.part.printEndTracking part.PrintStartTracking = self.part.printStartTracking part.MinDistToParentVertex = self.part.minDistToParentVertex part.OutputLevel = self.output.part part.enableUI() if self.part.enableDetailedHitsAndParticleInfo: self.part.setDumpDetailedParticleInfo(kernel, DDG4) self.part.setupUserParticleHandler(part, kernel, DDG4) #================================================================================= # Setup global filters for use in sensitive detectors try: self.filter.setupFilters(kernel) except RuntimeError as e: print "ERROR", str(e) exit(1) #================================================================================= # get lists of trackers and calorimeters in detectorDescription trk, cal = self.getDetectorLists(detectorDescription) # ---- add the trackers: try: self.__setupSensitiveDetectors(trk, simple.setupTracker, self.filter.tracker) except Exception as e: print "ERROR setting up sensitive detector", str(e) raise # ---- add the calorimeters: try: self.__setupSensitiveDetectors(cal, simple.setupCalorimeter, self.filter.calo) except Exception as e: print "ERROR setting up sensitive detector", str(e) raise #================================================================================= # Now build the physics list: _phys = self.physics.setupPhysics(kernel, name=self.physicsList) ## add the G4StepLimiterPhysics to activate the max step limits in volumes ph = DDG4.PhysicsList(kernel, 'Geant4PhysicsList/Myphysics') ph.addPhysicsConstructor('G4StepLimiterPhysics') _phys.add(ph) dd4hep.setPrintLevel(self.printLevel) kernel.configure() kernel.initialize() ## GPS if self._g4gun is not None: self._g4gun.generator() if self._g4gps is not None: self._g4gps.generator() startUpTime, _sysTime, _cuTime, _csTime, _elapsedTime = os.times() kernel.run() kernel.terminate() totalTimeUser, totalTimeSys, _cuTime, _csTime, _elapsedTime = os.times( ) if self.printLevel <= 3: print "DDSim INFO Total Time: %3.2f s (User), %3.2f s (System)" % ( totalTimeUser, totalTimeSys) if self.numberOfEvents != 0: eventTime = totalTimeUser - startUpTime perEventTime = eventTime / float(self.numberOfEvents) print "DDSim INFO StartUp Time: %3.2f s, Event Processing: %3.2f s (%3.2f s/Event) " \ % (startUpTime, eventTime, perEventTime)
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 setupWorker(): k = DDG4.Kernel() kernel = k.worker() logger.info('PYTHON: +++ Creating Geant4 worker thread ....') # 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 %f %d", 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.DEBUG prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) # Configure Event actions prt = DDG4.EventAction(kernel, 'Geant4SurfaceTest/SurfaceTest') prt.OutputLevel = Output.INFO kernel.eventAction().adopt(prt) # Configure I/O evt_lcio = geant4.setupLCIOOutput('LcioOutput', 'CLICSiD_' + time.strftime('%Y-%m-%d_%H-%M')) evt_lcio.OutputLevel = Output.DEBUG # 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 """ Generation of isotrope tracks of a given multiplicity with overlay: """ # First particle generator: pi+ gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropPi+") gen.Particle = 'pi+' gen.Energy = 100 * GeV gen.Multiplicity = 2 gen.Mask = 1 gen.OutputLevel = Output.DEBUG gen.PhiMin = 0 gen.PhiMax = 0 gen.ThetaMin = 1.61 gen.ThetaMax = 1.61 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 = 25 * GeV gen.Multiplicity = 3 gen.Mask = 2 gen.OutputLevel = Output.DEBUG 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) #^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ """ # 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 = Output.DEBUG # 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 = 100 * MeV part.OutputLevel = Output.DEBUG # 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('PYTHON: +++ Geant4 worker thread configured successfully....') return 1
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