def setupRandom(self, name, type=None, seed=None, quiet=True): rndm = DDG4.Action(self.kernel, 'Geant4Random/' + name) if seed: rndm.Seed = seed if type: rndm.Type = type rndm.initialize() if not quiet: rndm.showStatus() return rndm
def run(): import sys import CLICSid import DDG4 from DDG4 import OutputLevel as Output sid = CLICSid.CLICSid() geant4 = sid.geant4 kernel = sid.kernel sid.loadGeometry(str('CLICSiD_geometry.root')) geant4.printDetectors() if len(sys.argv) >= 2 and sys.argv[1] == "batch": kernel.UI = '' geant4.setupCshUI() sid.setupField(quiet=False) DDG4.importConstants(kernel.detectorDescription(), debug=False) prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint') prt.OutputLevel = Output.INFO prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) gen = DDG4.GeneratorAction(kernel, "Geant4GeneratorActionInit/GenerationInit") kernel.generatorAction().adopt(gen) logger.info("# First particle generator: gun") gun = DDG4.GeneratorAction(kernel, "Geant4GeneratorWrapper/Gun") gun.Uses = 'G4ParticleGun' gun.Mask = 1 kernel.generatorAction().adopt(gun) # Merge all existing interaction records merger = DDG4.GeneratorAction(kernel, "Geant4InteractionMerger/InteractionMerger") merger.enableUI() kernel.generatorAction().adopt(merger) # And handle the simulation particles. part = DDG4.GeneratorAction(kernel, "Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) part.OutputLevel = Output.INFO 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) sid.setupDetectors() sid.setupPhysics('QGSP_BERT') sid.test_config() gun.generator( ) # Instantiate gun to be able to set properties from G4 prompt kernel.run() kernel.terminate()
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(): import logging, LHeD, DDG4 from DDG4 import OutputLevel as Output logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.DEBUG) 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) prt = DDG4.EventAction(kernel,'Geant4ParticlePrint/ParticlePrint') prt.OutputLevel = Output.INFO prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) gen = DDG4.GeneratorAction(kernel,"Geant4GeneratorActionInit/GenerationInit") kernel.generatorAction().adopt(gen) logging.info("# First particle generator: gun") gun = DDG4.GeneratorAction(kernel,"Geant4GeneratorWrapper/Gun"); gun.Uses = 'G4ParticleGun' gun.Mask = 1 kernel.generatorAction().adopt(gun) # Merge all existing interaction records merger = DDG4.GeneratorAction(kernel,"Geant4InteractionMerger/InteractionMerger") merger.enableUI() kernel.generatorAction().adopt(merger) # And handle the simulation particles. part = DDG4.GeneratorAction(kernel,"Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) part.OutputLevel = Output.INFO 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) lhed.setupDetectors() lhed.setupPhysics('QGSP_BERT') lhed.test_config() gun.generator() # Instantiate gun to be able to set properties from G4 prompt kernel.run() kernel.terminate()
def run(): from g4units import * import logging, CLICSid, DDG4 from DDG4 import OutputLevel as Output logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.DEBUG) sid = CLICSid.CLICSid() sid.loadGeometry() sid.geant4.printDetectors() kernel = sid.kernel kernel.UI = "UI" ui = sid.geant4.setupCshUI(ui=None) # # Setup the GDML writer action writer = DDG4.Action(kernel,'Geant4GDMLWriteAction/Writer') writer.enableUI() kernel.registerGlobalAction(writer) sid.setupPhysics('QGSP_BERT') # gen = sid.geant4.setupGun('Gun','pi-',10*GeV,Standalone=True) # Now initialize. At the Geant4 command prompt we can write the geometry: # Idle> /ddg4/Writer/write # or by configuring the UI using ui.Commands # # Please note: The Geant4 physics list must be initialized BEFORE # invoking the writer with options. Otherwise the particle definitions # are missing! # If you ONLY want to dump the geometry to GDML you must call # /run/beamOn 0 # before writing the GDML file! # You also need to setup a minimal generation action like: # sid.geant4.setupGun('Gun','pi-',10*GeV,Standalone=True) # ui.Commands = [ '/run/beamOn 0', '/ddg4/Writer/Output CLICSiD.gdml', '/ddg4/Writer/OverWrite 1', '/ddg4/Writer/write' ] kernel.NumEvents = 0 kernel.configure() kernel.initialize() kernel.run() kernel.terminate()
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.__addParametersToRunHeader() 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: simple.buildInputStage(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) #---------------------------------- user = DDG4.Action(kernel, "Geant4TCUserParticleHandler/UserParticleHandler") try: user.TrackingVolume_Zmax = DDG4.tracker_region_zmax user.TrackingVolume_Rmax = DDG4.tracker_region_rmax print " *** definition of tracker region *** " print " tracker_region_zmax = ", user.TrackingVolume_Zmax print " tracker_region_rmax = ", user.TrackingVolume_Rmax print " ************************************ " except AttributeError as e: print "ERROR - attribute of tracker region missing in detector model ", str( e) print " make sure to specify the global constants tracker_region_zmax and tracker_region_rmax " print " this is needed for the MC-truth link of created sim-hits ! " exit(1) # user.enableUI() part.adopt(user) #================================================================================= # 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() #DDG4.setPrintLevel(Output.DEBUG) startUpTime, _sysTime, _cuTime, _csTime, _elapsedTime = os.times() kernel.run() kernel.terminate() totalTimeUser, totalTimeSys, _cuTime, _csTime, _elapsedTime = os.times( ) if self.printLevel <= 3: eventTime = totalTimeUser - startUpTime perEventTime = eventTime / float(self.numberOfEvents) print "DDSim INFO Total Time: %3.2f s (User), %3.2f s (System)" % ( totalTimeUser, totalTimeSys) print "DDSim INFO StartUp Time: %3.2f s, Event Processing: %3.2f s (%3.2f s/Event) " \ % (startUpTime, eventTime, perEventTime)
def run(): geo = None vis = 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 if not geo: help() sys.exit(1) import DDG4 Output = DDG4.OutputLevel 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) logger.info("# Setup random generator") rndm = DDG4.Action(kernel, 'Geant4Random/Random') rndm.Seed = 987654321 rndm.initialize() # # Setup detector seq, act = geant4.setupDetectors() # # Configure I/O geant4.setupROOTOutput('RootOutput', 'CheckShape_' + time.strftime('%Y-%m-%d_%H-%M'), mc_truth=True) # # Setup particle gun geant4.setupGun(name="Gun", particle='e-', energy=100 * 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) # prt = DDG4.EventAction(kernel, str('Geant4ParticlePrint/ParticlePrint')) prt.OutputLevel = Output.INFO prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) 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() # cmds = [] if vis: 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. 11.') cmds.append('/vis/scene/add/axes 0 0 0 1 m') # # 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 setupWorker(geant4): kernel = geant4.kernel() logging.info('#PYTHON: +++ Creating Geant4 worker thread ....') logging.info("#PYTHON: Configure Run actions") run1 = DDG4.RunAction(kernel, 'Geant4TestRunAction/RunInit', shared=True) run1.Property_int = int(12345) run1.Property_double = -5e15 * keV run1.Property_string = 'Startrun: Hello_2' logging.info("%s %f %d", run1.Property_string, run1.Property_double, run1.Property_int) run1.enableUI() kernel.runAction().adopt(run1) logging.info("#PYTHON: 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("\n#PYTHON: Configure I/O\n") 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( "#PYTHON:\n#PYTHON: Generation of isotrope tracks of a given multiplicity with overlay:\n#PYTHON:" ) logging.info("#PYTHON: First particle generator: pi+") gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropPi+") gen.Mask = 1 gen.Particle = 'pi+' gen.Energy = 20 * GeV gen.Multiplicity = 2 kernel.generatorAction().adopt(gen) logging.info("#PYTHON: 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("#PYTHON: Second particle generator: e-") gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropE-") gen.Mask = 2 gen.Particle = 'e-' gen.Energy = 15 * GeV gen.Multiplicity = 3 kernel.generatorAction().adopt(gen) logging.info("#PYTHON: 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("#PYTHON: 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("#PYTHON: Finally generate Geant4 primaries") gen = DDG4.GeneratorAction(kernel, "Geant4PrimaryHandler/PrimaryHandler") gen.OutputLevel = 4 #generator_output_level gen.enableUI() kernel.generatorAction().adopt(gen) logging.info("#PYTHON: ....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( '#PYTHON: +++ Geant4 worker thread configured successfully....') return 1
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(): 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(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry("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 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='e-', energy=2 * GeV, 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('e[+-]', 'G4eMultipleScattering', -1, 1, 1) ph.addParticleProcess('e[+-]', 'G4eIonisation', -1, 2, 2) ph.addParticleProcess('e[+-]', 'G4eBremsstrahlung', -1, 3, 3) ph.addParticleProcess('e+', 'G4eplusAnnihilation', 0, -1, 4) ph.addParticleProcess('mu[+-]', 'G4MuMultipleScattering', -1, 1, 1) ph.addParticleProcess('mu[+-]', 'G4MuIonisation', -1, 2, 2) ph.addParticleProcess('mu[+-]', 'G4MuBremsstrahlung', -1, 3, 3) ph.addParticleProcess('mu[+-]', '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(): 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(): import CLICSid import DDG4 import os import sys from DDG4 import OutputLevel as Output sid = CLICSid.CLICSid(no_physics=False) geant4 = sid.geant4 kernel = sid.kernel sid.loadGeometry() geant4.printDetectors() kernel.UI = 'UI' if len(sys.argv) >= 2 and sys.argv[1] == "batch": DDG4.setPrintLevel(DDG4.OutputLevel.WARNING) kernel.UI = '' geant4.setupCshUI() sid.setupField(quiet=False) DDG4.importConstants(kernel.detectorDescription(), debug=False) prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint') prt.OutputLevel = Output.INFO prt.OutputType = 3 # Print both: table and tree kernel.eventAction().adopt(prt) # First particle file reader gen = DDG4.GeneratorAction(kernel, "Geant4GeneratorActionInit/GenerationInit") kernel.generatorAction().adopt(gen) input = DDG4.GeneratorAction(kernel, "Geant4InputAction/Input") fname = os.environ['DD4hepExamplesINSTALL'] + '/examples/DDG4/data/hepmc_geant4.dat' input.Input = "Geant4EventReaderHepMC|" + fname input.MomentumScale = 1.0 input.Mask = 1 kernel.generatorAction().adopt(input) # Merge all existing interaction records merger = DDG4.GeneratorAction(kernel, "Geant4InteractionMerger/InteractionMerger") merger.enableUI() kernel.generatorAction().adopt(merger) logger.info("# 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.OutputLevel = Output.INFO part.enableUI() 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) user = DDG4.Action(kernel, "Geant4TCUserParticleHandler/UserParticleHandler") user.TrackingVolume_Zmax = DDG4.EcalEndcap_zmin user.TrackingVolume_Rmax = DDG4.EcalBarrel_rmin user.enableUI() part.adopt(user) # sid.setupDetectors() sid.setupPhysics('QGSP_BERT') sid.test_run(have_geo=True, num_events=1)
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) 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 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 # Configure I/O evt_root = geant4.setupROOTOutput( 'RootOutput', 'SiliconBlock_' + time.strftime('%Y-%m-%d_%H-%M')) # Setup particle gun gun = geant4.setupGun("Gun", particle='mu-', energy=20 * GeV, 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('SiliconBlockUpper') geant4.setupTracker('SiliconBlockDown') # Now build the physics list: phys = geant4.setupPhysics('QGSP_BERT') ph = DDG4.PhysicsList(kernel, 'Geant4PhysicsList/Myphysics') ph.addParticleConstructor('G4Geantino') ph.addParticleConstructor('G4BosonConstructor') ph.enableUI() phys.adopt(ph) phys.dump() geant4.execute()
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'] DDG4.Core.setPrintFormat("%-32s %6s %s") kernel.loadGeometry("file:" + install_dir + "/DDDetectors/compact/SiD.xml") DDG4.importConstants(description) kernel.NumberOfThreads = 3 geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') print "# Configure UI" geant4.setupCshUI() print "# Geant4 user initialization action" geant4.addUserInitialization(worker=setupWorker, worker_args=(geant4, ), master=setupMaster, master_args=(geant4, )) print "# Configure G4 geometry setup" seq, act = geant4.addDetectorConstruction( "Geant4DetectorGeometryConstruction/ConstructGeo") print "# Configure G4 sensitive detectors: python setup callback" seq, act = geant4.addDetectorConstruction( "Geant4PythonDetectorConstruction/SetupSD", sensitives=setupSensitives, sensitives_args=(geant4, )) print "# Configure G4 sensitive detectors: atach'em to the sensitive volumes" seq, act = geant4.addDetectorConstruction( "Geant4DetectorSensitivesConstruction/ConstructSD") # allow_threads=True) print "# Configure G4 magnetic field tracking" seq, field = geant4.addDetectorConstruction( "Geant4FieldTrackingConstruction/MagFieldTrackingSetup") field.stepper = "HelixGeant4Runge" field.equation = "Mag_UsualEqRhs" field.eps_min = 5e-05 * mm field.eps_max = 0.001 * mm field.min_chord_step = 0.01 * mm field.delta_chord = 0.25 * mm field.delta_intersection = 1e-05 * mm field.delta_one_step = 0.001 * 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 print "# Setup random generator" rndm = DDG4.Action(kernel, 'Geant4Random/Random') rndm.Seed = 987654321 rndm.initialize() ##rndm.showStatus() print "# Now build the physics list:" phys = geant4.setupPhysics('QGSP_BERT') phys.dump() geant4.run()
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) lcdd = kernel.lcdd() DDG4.importConstants(lcdd) #---------------------------------------------------------------------------------- #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.__addParametersToRunHeader() 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.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 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: simple.buildInputStage(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() user = DDG4.Action(kernel, "Geant4TCUserParticleHandler/UserParticleHandler") try: user.TrackingVolume_Zmax = DDG4.tracker_region_zmax user.TrackingVolume_Rmax = DDG4.tracker_region_rmax except AttributeError as e: print "No Attribute: ", str(e) # user.enableUI() part.adopt(user) #================================================================================= # 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 lcdd trk, cal = self.getDetectorLists(lcdd) # ---- 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) #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() 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() #DDG4.setPrintLevel(Output.DEBUG) kernel.run() kernel.terminate()
def run(): kernel = DDG4.Kernel() install_dir = os.environ['DD4hepExamplesINSTALL'] kernel.loadGeometry("file:" + install_dir + "/examples/DDCodex/compact/CODEX-b-alone.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 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', 'CodexB_' + time.strftime('%Y-%m-%d_%H-%M')) # Setup particle gun #gun = geant4.setupGun("Gun",particle='pi+', gun = geant4.setupGun( "Gun", particle='mu-', energy=1000 * GeV, multiplicity=1, isotrop=False, Standalone=True, direction=(1, 0, 0), #direction=(0.866025,0,0.5), position='(0,0,12650)') #position='(0,0,0)') setattr(gun, 'print', True) """ gen = DDG4.GeneratorAction(kernel,"Geant4InputAction/Input") ##gen.Input = "Geant4EventReaderHepMC|"+ "/afs/cern.ch/work/j/jongho/Project_DD4hep/Test/DD4hep/examples/DDG4/data/hepmc_geant4.dat" gen.Input = "Geant4EventReaderHepMC|"+ "/afs/cern.ch/work/j/jongho/Project_DD4hep/Test/DD4hep/DDG4/examples/MinBias_HepMC.txt" gen.MomentumScale = 1.0 gen.Mask = 1 geant4.buildInputStage([gen],output_level=Output.DEBUG) """ seq, action = geant4.setupTracker('CODEXb') #action.OutputLevel = Output.ERROR #seq,action = geant4.setupTracker('Shield') #action.OutputLevel = Output.ERROR # And handle the simulation particles. part = DDG4.GeneratorAction(kernel, "Geant4ParticleHandler/ParticleHandler") kernel.generatorAction().adopt(part) part.OutputLevel = Output.INFO part.enableUI() user = DDG4.Action(kernel, "Geant4TCUserParticleHandler/UserParticleHandler") user.TrackingVolume_Zmax = 999999. * m # Something big. All is a tracker user.TrackingVolume_Rmax = 999999. * m user.enableUI() part.adopt(user) # Now build the physics list: ##phys = kernel.physicsList() phys = geant4.setupPhysics('QGSP_BERT') ph = DDG4.PhysicsList(kernel, 'Geant4PhysicsList/Myphysics') ph.addParticleConstructor('G4LeptonConstructor') ph.addParticleConstructor('G4BaryonConstructor') ph.addParticleConstructor('G4MesonConstructor') ph.addParticleConstructor('G4BosonConstructor') ph.enableUI() phys.adopt(ph) phys.enableUI() phys.dump() # run kernel.configure() kernel.initialize() kernel.run() kernel.terminate()
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(): hlp = False 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] == '-VIS': vis = True elif c[:4] == '-DUM': dump = True elif c[:2] == '-H': hlp = True if hlp: help() sys.exit(1) kernel = DDG4.Kernel() description = kernel.detectorDescription() install_dir = os.environ['DD4hepExamplesINSTALL'] geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction') # logger.info("# Configure UI") ui = None if batch: geant4.setupCshUI(ui=None, vis=None) kernel.UI = 'UI' else: ui = geant4.setupCshUI(vis=vis) kernel.loadGeometry(str("file:" + install_dir + "/examples/ClientTests/compact/NestedBoxReflection.xml")) DDG4.importConstants(description) geant4.printDetectors() if dump: seq, act = geant4.addDetectorConstruction("Geant4DetectorGeometryConstruction/ConstructGeo") act.DebugReflections = True act.DebugMaterials = False act.DebugElements = False act.DebugVolumes = False act.DebugShapes = False act.DumpHierarchy = ~0x0 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() # logger.info("# Configure Event actions") prt = DDG4.EventAction(kernel, 'Geant4ParticlePrint/ParticlePrint') prt.OutputType = 3 # Print both: table and tree prt.OutputLevel = Output.INFO kernel.eventAction().adopt(prt) # logger.info("# Configure I/O") geant4.setupROOTOutput('RootOutput', 'BoxReflect_' + time.strftime('%Y-%m-%d_%H-%M')) # gen = DDG4.GeneratorAction(kernel, "Geant4GeneratorActionInit/GenerationInit") gen.enableUI() kernel.generatorAction().adopt(gen) # logger.info("# Generation of isotrope tracks of a given multiplicity with overlay:") gen = DDG4.GeneratorAction(kernel, "Geant4ParticleGun/IsotropE+") gen.mask = 4 gen.isotrop = True gen.particle = 'e+' gen.energy = 100 * GeV gen.multiplicity = 200 gen.position = (0 * m, 0 * m, 0 * m) gen.direction = (0, 0, 1.) gen.distribution = 'uniform' gen.standalone = False # gen.PhiMin = 0.0*rad # gen.PhiMax = 2.0*math.pi*rad # gen.ThetaMin = 0.0*math.pi*rad # gen.ThetaMax = 1.0*math.pi*rad gen.enableUI() 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 = 3.0 * m user.TrackingVolume_Rmax = 3.0 * m user.enableUI() part.adopt(user) # logger.info("# Now setup the calorimeters") seq, actions = geant4.setupDetectors() # logger.info("# Now build the physics list:") 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('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 # # 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.configure() kernel.initialize() # DDG4.setPrintLevel(Output.DEBUG) kernel.run() kernel.terminate()
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 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()