In This Package:

FastMuon.py

Go to the documentation of this file.

#!/usr/bin/env python

class FastMuon:
    '''
       configure Muon generator
    '''

    def __init__(self,
                 stage='null',
                 name='Muon',
                 volume='/dd/Structure/Pool/db-ows',
                 start_time=0,
                 seed=1234567):

        # set up Gnrtr itself
        from Gnrtr.GnrtrConf import Gnrtr
        gnrtr = Gnrtr(name);

        gnrtr.GenTools = [ "GtHepEvtGenTool/"+name+"Gen",
                           "GtPositionerTool/"+name+"Pos",
                           "GtTimeratorTool/"+name+"Tim",
                           "GtTransformTool/"+name+"Tra",
                           "MuonProphet/"+name+"Prophet"
                           ]

        gnrtr.ThisStageName = "Kinematic"
        gnrtr.TimeStamp = start_time

        if stage != 'null':
            stage.KinematicSequence.Members.append(gnrtr)

        # set up each tools
        from GaudiKernel import SystemOfUnits as units
        from GenTools.GenToolsConf import GtPositionerTool, GtTransformTool, GtTimeratorTool, GtHepEvtGenTool
        from MuonProphet.MuonProphetConf import MuonProphet
        # Set up Gen

        musicsite = 'DYB'
                
        nevents=5000

        # Get cosmic muon path
        import os
        MuonDataPath = os.getenv('MuonDataPath')
        if MuonDataPath is None:
            print "Muon data path ($MuonDataPath) is not defined."
            import sys
            sys.exit()
        else:
            print "Read muon data from ",MuonDataPath


        # set the muon generator command line
        hepEvtDataSource = "Muon.exe -n 1100 -s %s -v rock -seed %s -r Yes -music_dir %s|"%(musicsite,seed,MuonDataPath)
        
        exe = hepEvtDataSource.split(' ')[0]
        NAME = exe[exe.rfind('/')+1:]
        dot = NAME.rfind('.')
        baseNAME = NAME
        if dot > 0: baseNAME = NAME[:dot]
        sanitized = baseNAME.replace('.','_')
        
        # If we got an executable and it is relative, look for it in the path
        if hepEvtDataSource[-1] == '|' and hepEvtDataSource[0] != '/':
            import os, os.path
            path = os.getenv('PATH')
            for p in path:
                if (os.path.isfile(path+'/'+exe)):
                    hepEvtDataSource = path+'/'+exe + ' ' + ' '.join(hepEvtDataSource.split(' ')[1:])
                    break
                continue
            pass
        
        #generator
        gen=GtHepEvtGenTool(name+'Gen',HepEvtDataSource = hepEvtDataSource)
        #gen.OutputLevel = 2
        gen.HepEvtDataSource = hepEvtDataSource
        
        # Set up positioner
        pos=GtPositionerTool(name+'Pos',Volume=volume)
        # I should use relateiv
        pos.Mode = "Relative"
        #pos.Mode = "Fixed"
        pos.Position = [0,0,0]
        
        # Set up timerator
        tim=GtTimeratorTool(name+'Tim')
        tim.LifeTime = 0.000595*units.s
        
        # transform
        tra=GtTransformTool(name+'Tra',Volume=volume)
        tra.Offset = [0., 0., (0.042)*units.meter]

        ## More surprise from muon starts from here.
        # muon prophet
        prophet=MuonProphet();
        # The can completely turn the tool off.
        #prophet.Active=False;        
        prophet.Site = "DayaBay"
        # Note that the number of parameters of 
        # Tools, Yields, YieldMeasuredAt and Lifetimes must be the same.
        prophet.GenTools     = [ "Li9He8Decayerator/Li9He8" ]
        prophet.GenYields    = [ 0.5*units.cm2/units.gram ]
        prophet.GenYieldMeasuredAt = [260*units.GeV]
        prophet.GenLifetimes = [ 0.002*units.s ]

        # trigger related configuration
        prophet.TrkLengthInWaterThres = 20*units.cm
        prophet.WaterPoolTriggerEff = 0.9999

        #prophet.OutputLevel=2

        # Set false, otherwise all times will be reset so the signal vertex is at t=0.
        from G4DataHelpers.G4DataHelpersConf import HepMCtoG4
        convertor = HepMCtoG4()
        convertor.ZeroTime = False

        pass

if __name__ == "__main__":
    obj=FastMuon()

---