ClientTests/scripts/DriftChamber.py

0001 # ==========================================================================
0002 #  AIDA Detector description implementation
0003 # --------------------------------------------------------------------------
0004 # Copyright (C) Organisation europeenne pour la Recherche nucleaire (CERN)
0005 # All rights reserved.
0006 #
0007 # For the licensing terms see $DD4hepINSTALL/LICENSE.
0008 # For the list of contributors see $DD4hepINSTALL/doc/CREDITS.
0009 #
0010 # ==========================================================================
0011 #
0012 #
0013 from __future__ import absolute_import, unicode_literals
0014 import os
0015 import time
0016 import DDG4
0017 import logging
0018 from g4units import GeV, MeV, m
0019 logging.basicConfig(format='%(levelname)s: %(message)s', level=logging.INFO)
0020 logger = logging.getLogger(__name__)
0021 #
0022 #
0023 """
0024
0025    dd4hep simulation example setup using the python configuration
0026
0027    @author  M.Frank
0028    @version 1.0
0029
0030 """
0031
0032
0033 def run():
0034   args = DDG4.CommandLine()
0035   kernel = DDG4.Kernel()
0036   install_dir = os.environ['DD4hepExamplesINSTALL']
0037   kernel.loadGeometry(str("file:" + install_dir + "/examples/ClientTests/compact/DriftChamber.xml"))
0038
0039   DDG4.importConstants(kernel.detectorDescription(), debug=False)
0040   geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction')
0041   geant4.registerInterruptHandler()
0042   geant4.printDetectors()
0043   # Configure UI
0044   if args.macro:
0045     ui = geant4.setupCshUI(macro=args.macro)
0046   else:
0047     ui = geant4.setupCshUI()
0048
0049   cmds = []
0050   if args.verbose:
0051     cmds.append('/run/verbose ' + str(args.verbose))
0052
0053   if args.events:
0054     cmds.append('/run/beamOn ' + str(args.events))
0055     cmds.append('/ddg4/UI/terminate')
0056
0057   if len(cmds) > 0:
0058     ui.Commands = cmds
0059
0060   logger.info("#  Configure G4 magnetic field tracking")
0061   geant4.setupTrackingField()
0062
0063   logger.info("#  Setup random generator")
0064   rndm = DDG4.Action(kernel, 'Geant4Random/Random')
0065   rndm.Seed = 987654321
0066   if args.seed_time:
0067     rndm.Seed = int(time.time())
0068   rndm.initialize()
0069
0070   gen = DDG4.GeneratorAction(kernel, "Geant4GeneratorActionInit/GenerationInit")
0071   kernel.generatorAction().adopt(gen)
0072
0073   logger.info("""
0074   Generation of isotrope tracks of a given multiplicity with overlay:
0075   """)
0076   logger.info("#  First particle generator: pi+")
0077   gen = DDG4.GeneratorAction(kernel, "Geant4IsotropeGenerator/IsotropPi+")
0078   gen.Mask = 1
0079   gen.Particle = 'pi+'
0080   gen.Energy = 100 * GeV
0081   gen.Multiplicity = 2
0082   gen.Distribution = 'cos(theta)'
0083   kernel.generatorAction().adopt(gen)
0084   logger.info("#  Install vertex smearing for this interaction")
0085
0086   logger.info("#  Merge all existing interaction records")
0087   gen = DDG4.GeneratorAction(kernel, "Geant4InteractionMerger/InteractionMerger")
0088   gen.OutputLevel = 4  # generator_output_level
0089   gen.enableUI()
0090   kernel.generatorAction().adopt(gen)
0091   #
0092   logger.info("#  Finally generate Geant4 primaries")
0093   gen = DDG4.GeneratorAction(kernel, "Geant4PrimaryHandler/PrimaryHandler")
0094   gen.OutputLevel = 4  # generator_output_level
0095   gen.enableUI()
0096   kernel.generatorAction().adopt(gen)
0097   #
0098   logger.info("#  ....and handle the simulation particles.")
0099   part = DDG4.GeneratorAction(kernel, "Geant4ParticleHandler/ParticleHandler")
0100   kernel.generatorAction().adopt(part)
0101   part.SaveProcesses = ['Decay']
0102   part.MinimalKineticEnergy = 100 * MeV
0103   part.OutputLevel = 5  # generator_output_level
0104   part.enableUI()
0105   user = DDG4.Action(kernel, "Geant4TCUserParticleHandler/UserParticleHandler")
0106   user.TrackingVolume_Zmax = 1.5 * m
0107   user.TrackingVolume_Rmax = 1.5 * m
0108   user.enableUI()
0109   part.adopt(user)
0110   #
0111   seq, act = geant4.setupTracker('DriftChamber')
0112   #
0113   logger.info("#  Now build the physics list:")
0114   phys = geant4.setupPhysics('QGSP_BERT')
0115   ph = geant4.addPhysics(str('Geant4PhysicsList/Myphysics'))
0116   ph.addPhysicsConstructor(str('G4StepLimiterPhysics'))
0117   #
0118   phys.dump()
0119   #
0120   kernel.configure()
0121   kernel.initialize()
0122
0123   # DDG4.setPrintLevel(Output.DEBUG)
0124   kernel.run()
0125   kernel.terminate()
0126
0127
0128 if __name__ == "__main__":
0129   run()