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()