Muon.py

Go to the documentation of this file.

#!/usr/bin/env python

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

    def __init__(self,
                 stage='null',
                 name='Muon',
                 volume='/dd/Structure/AD/db-ade1',
                 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"]

        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
        # Set up Gen

        musicsite = 'DYB'
                
        nevents=5000

        hepEvtDataSource = "Muon.exe -n 1000 -s %s -v ADE -seed %s -r Yes -music_dir /mnt/sda3/wangzhe/dybinst/data/trunk/NewMuonGenerator/data|"%(musicsite,seed)
        
        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)
        pos.Mode = "Relative"
        pos.Position = [0,0,0]
        
        # Set up timerator
        tim=GtTimeratorTool(name+'Tim')
        tim.LifeTime = 0.001123*units.s
        
        # transform
        tra=GtTransformTool(name+'Tra',Volume=volume)
        tra.Offset = [0., 0., 0.]
        
        pass

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