Beispiel #1
0
    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.meta.addParametersToRunHeader(self)
            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:
            self._buildInputStage(simple,
                                  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)

        self.part.setupUserParticleHandler(part, kernel, DDG4)

        #=================================================================================

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

        startUpTime, _sysTime, _cuTime, _csTime, _elapsedTime = os.times()

        kernel.run()
        kernel.terminate()

        totalTimeUser, totalTimeSys, _cuTime, _csTime, _elapsedTime = os.times(
        )
        if self.printLevel <= 3:
            print "DDSim            INFO  Total Time:   %3.2f s (User), %3.2f s (System)" % (
                totalTimeUser, totalTimeSys)
            if self.numberOfEvents != 0:
                eventTime = totalTimeUser - startUpTime
                perEventTime = eventTime / float(self.numberOfEvents)
                print "DDSim            INFO  StartUp Time: %3.2f s, Event Processing: %3.2f s (%3.2f s/Event) " \
                    % (startUpTime, eventTime, perEventTime)
Beispiel #2
0
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
Beispiel #3
0
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()
Beispiel #4
0
def run():
  kernel = DDG4.Kernel()
  install_dir = os.environ['DD4hepINSTALL']
  example_dir = install_dir+'/examples/DDG4/examples';
  kernel.loadGeometry("file:"+install_dir+"/examples/CLICSiD/compact/compact.xml")
  kernel.loadXML("file:"+example_dir+"/DDG4_field.xml")

  simple = DDG4.Simple(kernel,tracker='LcioTestTrackerAction')
  simple.printDetectors()
  # Configure UI
  simple.setupCshUI()

  # Configure Run actions
  run1 = DDG4.RunAction(kernel,'Geant4TestRunAction/RunInit')
  run1.Property_int    = 12345
  run1.Property_double = -5e15*keV
  run1.Property_string = 'Startrun: Hello_2'
  print run1.Property_string, run1.Property_double, run1.Property_int
  run1.enableUI()
  kernel.registerGlobalAction(run1)
  kernel.runAction().add(run1)

  # Configure Event actions
  evt2 = DDG4.EventAction(kernel,'Geant4TestEventAction/UserEvent_2')
  evt2.Property_int    = 123454321
  evt2.Property_double = 5e15*GeV
  evt2.Property_string = 'Hello_2 from the python setup'
  evt2.enableUI()
  kernel.registerGlobalAction(evt2)

  evt1 = DDG4.EventAction(kernel,'Geant4TestEventAction/UserEvent_1')
  evt1.Property_int=01234
  evt1.Property_double=1e11
  evt1.Property_string='Hello_1'
  evt1.enableUI()

  kernel.eventAction().add(evt1)
  kernel.eventAction().add(evt2)
  """
  trk = DDG4.Action(kernel,"Geant4TrackPersistency/MonteCarloTruthHandler")
  kernel.registerGlobalAction(trk)
  trk.release()
  mc  = DDG4.Action(kernel,"Geant4MonteCarloRecordManager/MonteCarloRecordManager")
  kernel.registerGlobalAction(mc)
  mc.release()
  """
  # Configure I/O
  evt_lcio = simple.setupLCIOOutput('LcioOutput','CLICSiD_'+time.strftime('%Y-%m-%d_%H-%M'))

  gen = DDG4.GeneratorAction(kernel,"Geant4TestGeneratorAction/Generate")
  kernel.generatorAction().add(gen)

  # Setup particle gun
  gun = simple.setupGun('Gun','pi-',100*GeV,True)

  """
  rdr = DDG4.GeneratorAction(kernel,"LcioGeneratorAction/Reader")
  rdr.zSpread = 0.0
  rdr.lorentzAngle = 0.0
  rdr.OutputLevel = DDG4.OutputLevel.INFO
  rdr.Input = "LcioEventReader|test.data"
  rdr.enableUI()
  kernel.generatorAction().add(rdr)
  """

  # Setup global filters fur use in sensintive detectors
  f1 = DDG4.Filter(kernel,'GeantinoRejectFilter/GeantinoRejector')
  kernel.registerGlobalFilter(f1)

  f2 = DDG4.Filter(kernel,'ParticleRejectFilter/OpticalPhotonRejector')
  f2.particle = 'opticalphoton'
  kernel.registerGlobalFilter(f2)

  f3 = DDG4.Filter(kernel,'ParticleSelectFilter/OpticalPhotonSelector') 
  f3.particle = 'opticalphoton'
  kernel.registerGlobalFilter(f3)

  f4 = DDG4.Filter(kernel,'EnergyDepositMinimumCut')
  f4.Cut = 10*MeV
  f4.enableUI()
  kernel.registerGlobalFilter(f4)

  # First the tracking detectors
  seq,act = simple.setupTracker('SiVertexBarrel')
  seq.add(f1)
  #seq.add(f4)
  act.add(f1)

  seq,act = simple.setupTracker('SiVertexEndcap')
  seq.add(f1)
  #seq.add(f4)

  seq,act = simple.setupTracker('SiTrackerBarrel')
  seq,act = simple.setupTracker('SiTrackerEndcap')
  seq,act = simple.setupTracker('SiTrackerForward')
  # Now the calorimeters
  seq,act = simple.setupCalorimeter('EcalBarrel')
  seq,act = simple.setupCalorimeter('EcalEndcap')
  seq,act = simple.setupCalorimeter('HcalBarrel')
  seq,act = simple.setupCalorimeter('HcalEndcap')
  seq,act = simple.setupCalorimeter('HcalPlug')
  seq,act = simple.setupCalorimeter('MuonBarrel')
  seq,act = simple.setupCalorimeter('MuonEndcap')
  seq,act = simple.setupCalorimeter('LumiCal')
  seq,act = simple.setupCalorimeter('BeamCal')

  # Now build the physics list:
  phys = simple.setupPhysics('QGSP_BERT')
  ph = DDG4.PhysicsList(kernel,'Geant4PhysicsList/Myphysics')
  ph.addParticleConstructor('G4BosonConstructor')
  ph.addParticleConstructor('G4LeptonConstructor')
  ph.addParticleProcess('e[+-]','G4eMultipleScattering',-1,1,1)
  ph.addPhysicsConstructor('G4OpticalPhysics')
  ph.enableUI()
  phys.add(ph)

  phys.dump()

  kernel.configure()
  kernel.initialize()

  #DDG4.setPrintLevel(Output.DEBUG)
  kernel.run()
  kernel.terminate()
Beispiel #5
0
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
Beispiel #6
0
    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
            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 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()
Beispiel #7
0
def run():
    kernel = DDG4.Kernel()
    lcdd = kernel.lcdd()
    install_dir = os.environ['DD4hepINSTALL']
    example_dir = install_dir + '/examples/DDG4/examples'
    kernel.loadGeometry("file:" + install_dir +
                        "/DDDetectors/compact/SiD_Markus.xml")
    ##kernel.loadXML("file:"+example_dir+"/DDG4_field.xml")
    DDG4.importConstants(lcdd, debug=False)
    geant4 = DDG4.Geant4(kernel, tracker='Geant4TrackerCombineAction')
    geant4.printDetectors()
    # Configure UI
    #geant4.setupCshUI(macro='run.mac',ui=None)
    geant4.setupCshUI()

    field = geant4.addConfig(
        'Geant4FieldTrackingSetupAction/MagFieldTrackingSetup')
    field.stepper = "HelixSimpleRunge"
    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

    # Configure Run actions
    run1 = DDG4.RunAction(kernel, 'Geant4TestRunAction/RunInit')
    run1.enableUI()
    kernel.registerGlobalAction(run1)
    kernel.runAction().adopt(run1)

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

    generator_output_level = Output.WARNING

    # Configure I/O
    evt_lcio = geant4.setupLCIOOutput(
        'LcioOutput', 'CLICSiD_' + time.strftime('%Y-%m-%d_%H-%M'))
    ##evt_lcio.OutputLevel = generator_output_level
    #evt_root = geant4.setupROOTOutput('RootOutput','CLICSiD_'+time.strftime('%Y-%m-%d_%H-%M'))

    prim = DDG4.GeneratorAction(kernel,
                                "Geant4GeneratorActionInit/GenerationInit")
    #VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV
    """
  Generation of primary particles from LCIO input files
  """
    """
  # First particle file reader
  gen = DDG4.GeneratorAction(kernel,"LCIOInputAction/LCIO1");
  #gen.Input = "LCIOStdHepReader|/home/frankm/SW/data/e2e2nn_gen_1343_1.stdhep"
  #gen.Input = "LCIOStdHepReader|/home/frankm/SW/data/qq_gen_128_999.stdhep"
  #gen.Input = "LCIOStdHepReader|/home/frankm/SW/data/smuonLR_PointK_3TeV_BS_noBkg_run0001.stdhep"
  #gen.Input = "LCIOStdHepReader|/home/frankm/SW/data/bbbb_3TeV.stdhep"
  #gen.Input = "LCIOFileReader|/home/frankm/SW/data/mcparticles_pi-_5GeV.slcio"
  #gen.Input = "LCIOFileReader|/home/frankm/SW/data/mcparticles_mu-_5GeV.slcio"
  #gen.Input = "LCIOFileReader|/home/frankm/SW/data/bbbb_3TeV.slcio"
  #gen.Input = "LCIOStdHepReader|/home/frankm/SW/data/FCC-eh.stdhep"
  #gen.Input = "Geant4EventReaderHepMC|/home/frankm/SW/data/data.hepmc.txt"
  #gen.Input = "Geant4EventReaderHepMC|/home/frankm/SW/data/sherpa-2.1.1_zjets.hepmc2g"
  gen.Input = "LCIOFileReader|/afs/cern.ch/user/n/nikiforo/public/Markus/muons.slcio"
  #gen.Input = "LCIOFileReader|/afs/cern.ch/user/n/nikiforo/public/Markus/geantinos.slcio"
  gen.MomentumScale = 1.0
  gen.Mask = 1
  geant4.buildInputStage([gen],output_level=generator_output_level)
  """
    #^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
    gen = geant4.setupGun("Gun",
                          particle='mu+',
                          energy=20 * GeV,
                          position=(0 * mm, 0 * mm, 0 * cm),
                          multiplicity=3)
    gen.isotrop = True
    gen.direction = (1, 0, 0)
    gen.OutputLevel = generator_output_level
    gen.Standalone = False
    """
  # 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.EcalEndcap_zmin
  user.TrackingVolume_Rmax = DDG4.EcalBarrel_rmin
  user.enableUI()
  part.adopt(user)
  """
    geant4.buildInputStage([prim, gen], Output.ERROR)
    """
  """
    """
  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)
  """

    # First the tracking detectors
    seq, act = geant4.setupTracker('SiTrackerBarrel')
    seq, act = geant4.setupTracker('SiTrackerEndcap')
    seq, act = geant4.setupTracker('SiTrackerForward')
    """
  # Now the calorimeters
  seq,act = geant4.setupTracker('SiVertexBarrel')
  seq,act = geant4.setupTracker('SiVertexEndcap')

  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')
  """
    """
  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.enableUI()
    phys.adopt(ph)
    phys.dump()

    kernel.configure()
    kernel.initialize()

    #DDG4.setPrintLevel(Output.DEBUG)
    kernel.run()
    print 'End of run. Terminating .......'
    kernel.terminate()
Beispiel #8
0
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()
Beispiel #9
0
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()
Beispiel #10
0
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()