beamline/compact/scripts/vis_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.INFO)
logger = logging.getLogger(__name__)
"""
 
 dd4hep simulation example setup using the python configuration
 
 @author M.Frank
 @version 1.0
 
"""
 
 
def run():
 os.environ['G4UI_USE_TCSH'] = "1"
 kernel = DDG4.Kernel()
 description = kernel.detectorDescription()
 install_dir = os.environ['DD4hepINSTALL']
 #kernel.loadGeometry(str("file:" + install_dir + "/DDDetectors/compact/SiD.xml"))
 kernel.loadGeometry(str("file:" + "eic_beamline_example.xml"))
 DDG4.importConstants(description)
 
 geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction')
 geant4.printDetectors()
 
 logger.info("# Configure UI")
 geant4.setupUI('tcsh',vis=True,macro='macro/vis.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.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', 'derp_' + time.strftime('%Y-%m-%d_%H-%M'))
 
 gen = DDG4.GeneratorAction(kernel, "Geant4GeneratorActionInit/GenerationInit")
 kernel.generatorAction().adopt(gen)
 
 gen = DDG4.GeneratorAction(kernel, "Geant4GeneratorWrapper/GPS")
 gen.Uses = 'G4GeneralParticleSource'
 gen.OutputLevel = Output.INFO
 gen.Mask = 1
 gen.enableUI()
 kernel.generatorAction().adopt(gen)
 
 #gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropPi+")
 #gen.Mask = 1
 #gen.Particle = 'e-'
 #gen.Energy = 5 * GeV
 #gen.Multiplicity = 2
 #gen.Distribution = 'cos(theta)'
 
 #geant4.setupGun('electron','e-',3.0*GeV)
 
 
 #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 = 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)
 
 
 #logger.info("# Setup global filters fur use in sensitive detectors")
 
 
 #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 = 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)
 
 
 
 
 # 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()