In This Package:

Helpers.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
Helpers to configure GenMuon.

This is also a job module.  See docs under configure()
'''

import os

class CosmicHelper:
    '''
    A helper for use with GenTools.Configure()
    '''

    def __init__(self, site, volume, coord_de, path):
        '''
        Create a helper to set up kinematics for cosmic muons.  
        Options:

        site            : set which site (DYB, LA, Mid, Far,SAB)
        volume          : keyword for muon generator to determine where to put muon vertices (rock, RPC, ADE)
        coord_de        : detector element providing coordinate system to convert to global (eg, '/dd/Structure/AD/db-ade1')
        path            : set directory holding muon data, overrides $MuonDataPath

        The tools it produces are available via: 
        .muon the GtMuoneratorTool
        .positioner the GtPositionerTool
        .transform the GtTransformTool
        .timerator the GtTimeratorTool
        '''

        # try to find where the input files are
        if not path:
            print 'No muon data path given, checking $MuonDataPath' 
            path = os.getenv('MuonDataPath')
        if not path:
            for where in [os.getenv('NEWMUONGENERATORROOT','data/NewMuonGenerator/data'),
                          os.getenv('NEWMUONGENERATORROOT','data/trunk/NewMuonGenerator/data')]:
                print 'No $MuonDataPath, trying to locate %s/' % where
                sr = os.getenv('SITEROOT')
                if sr:
                    maybe = os.path.join(sr,where)
                    if os.path.exists(maybe):
                        path = maybe
                        break
                    continue
                continue
        if not path:
            raise ValueError('Could not find input muon data anywhere, go fish.')

        flux_file = os.path.join(path,'mountain_%s'%site)
        url = 'http://dayabay.ihep.ac.cn/svn/dybsvn/data/trunk/NewMuonGenerator/data/'
        if not os.path.exists(flux_file):
            raise RuntimeError('No flux file "%s" get it from %s' % (flux_file, url) )

        ratio_file = os.path.join(path,'mu_plus_minus_ratio.root')
        if not os.path.exists(ratio_file):
            print 'No mu+/mu- ratio file found, will rely on paramatrization.'
            #raise RuntimeError('No ratio file "%s" get it from %s' % (ratio_file, url) )

        name = 'Cosmic-%s-%s' % (site,volume)

        from GenMuon.GenMuonConf import GtMuoneratorTool
        muon = GtMuoneratorTool(name+'_muonerator')
        #muon.Rotation = True
        muon.WhichSite = site
        muon.MuonFile = flux_file
        muon.RatioFile = ratio_file
        muon.Volume = volume
        self.muon = muon

        from GenTools.GenToolsConf import GtPositionerTool, GtTransformTool, GtTimeratorTool
        from GaudiKernel import SystemOfUnits as units

        # Set up positioner
        pos = GtPositionerTool(name+'_positioner',Volume=coord_de)
        pos.Mode = "Relative"
        pos.Position = [0,0,0]
        self.positioner = pos
        
        # Set up timerator. note high rate of muons
        tim = GtTimeratorTool(name+'_timerator')
        muonRate = 1000.   # in Hz
        lifet = 1./float(muonRate)
        tim.LifeTime = (lifet)*units.s
        print "Muon: Rate per second is ",muonRate
        self.timerator = tim
        
        # transform
        tra = GtTransformTool(name+'_transformer',Volume=coord_de)
        tra.Offset = [0., 0., (0.042)*units.meter]
        self.transformer = tra

        return

    def tools(self):
        return [self.muon,self.positioner,self.timerator,self.transformer]

    pass

def configure(argv=None):
    '''
    Add a CosmicHelper to the job

    usage: GenMuon.Helpers site volume coord_de [path]
    '''
    if not argv:
        raise ValueError("no args.")
    
    import GenTools

    site = argv[0]
    volume = argv[1]
    coord_de = argv[2]
    try:
        path = argv[3]          # optional
    except IndexError:
        path = None

    cosmic_helper = CosmicHelper(site,volume,coord_de,path)
    res = GenTools.Configure('CosmicsIn%s%s'%(site,volume),
                             'CosmicsIn%s%s'%(site,volume),
                             cosmic_helper)
    

---