pid/compact/scripts/sim_test.py

Go to the documentation of this file.

#!/usr/bin/env python
#
from __future__ import absolute_import, unicode_literals
import os
import time
import logging
import DDG4
from DDG4 import OutputLevel as Output
from g4units import keV, GeV, mm, ns, MeV
 
logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.WARNING)
logger = logging.getLogger(__name__)
 
def run():
 
  os.environ['G4UI_USE_TCSH'] = "1"
  kernel = DDG4.Kernel()
  description = kernel.detectorDescription()
  kernel.loadGeometry(str("file:" + "GenericRICH_example.xml"))
  DDG4.importConstants(description)
 
  geant4 = DDG4.Geant4(kernel)#, tracker='Geant4TrackerCombineAction')
  geant4.printDetectors()
 
  logger.info("""
  Configure I/O
  """)
 
  rootoutput = geant4.setupROOTOutput('RootOutput', 'derp_' + time.strftime('%Y-%m-%d_%H-%M'), mc_truth=True)
  rootoutput.HandleMCTruth = True
 
  logger.info("#  Configure UI")
  geant4.setupUI('tcsh',vis=True,macro='macro/run.mac')
 
  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.WARNING
  #prt.OutputType = 0  # Print both: table and tree
  #kernel.eventAction().adopt(prt)
 
  geant4.setupGun('electron','e-',3.0*GeV)
 
  #gen = DDG4.GeneratorAction(kernel, "Geant4GeneratorActionInit/GenerationInit")
  #kernel.generatorAction().adopt(gen)
  #gen.enableUI()
 
  #gen = DDG4.GeneratorAction(kernel, "Geant4GeneratorWrapper/GPS")
  #gen.Uses = 'G4GeneralParticleSource'
  #gen.OutputLevel = Output.WARNING
  
 
  #gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropPi+")
  #gen.Mask = 1
  #gen.Particle = 'e-'
  #gen.Energy = 5 * GeV
  #gen.Multiplicity = 2
  #gen.Distribution = 'cos(theta)'
 
  #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 = 3
  #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("#  Merge all existing interaction records")
  #gen = DDG4.GeneratorAction(kernel, "Geant4InteractionMerger/InteractionMerger")
  #gen.OutputLevel = 0  # 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)
  
 
  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 PID detectors")
  seq, act = geant4.setupDetector('ForwardRICH','PhotoMultiplierSDAction')
  act.adopt(f2)
 
  #seq, act = geant4.setupTracker('SiVertexBarrel')
  #seq.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')
  geant4.addPhysics(str('Geant4PhysicsList/Myphysics'))
 
  ph = DDG4.PhysicsList(kernel, 'Geant4OpticalPhotonPhysics/OpticalPhotonPhys')
  ph.VerboseLevel = 0
  ph.enableUI()
  phys.adopt(ph)
 
  ph = DDG4.PhysicsList(kernel, 'Geant4CerenkovPhysics/CerenkovPhys')
  ph.MaxNumPhotonsPerStep = 10
  ph.MaxBetaChangePerStep = 10.0
  ph.TrackSecondariesFirst = True
  ph.VerboseLevel = 0
  ph.enableUI()
  phys.adopt(ph)
 
  
 
  # Add global range cut
  rg = geant4.addPhysics('Geant4DefaultRangeCut/GlobalRangeCut')
  rg.RangeCut = 0.7 * mm
 
  #phys.dump()
 
  #ui_action = dd4hep.sim.createAction(kernel, "Geant4UIManager/UI")
  #ui_action.HaveVIS = True
  #ui_action.HaveUI = True
  #ui_action.SessionType = qt
  #ui_action.SetupUI = macro
  #kernel.registerGlobalAction(ui_action)
 
  #ui = geant4.setupUI("qt",vis=True,macro="vis.mac")
  #ui.enableUI()
 
  kernel.configure()
  kernel.initialize()
 
  # DDG4.setPrintLevel(Output.DEBUG)
  kernel.run()
  kernel.terminate()
 
 
if __name__ == "__main__":
  run()