IBD.py

Go to the documentation of this file.

#!/usr/bin/env python

class IBD:
    '''
       configure IBD generator
    '''

    def __init__(self,
                 stage='null',
                 name='IBD',
                 volume='/dd/Structure/AD/far-oil1',
                 start_time=0,
                 seed=12345):

        # 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

        nevents=5000
        hepevt_source="InverseBeta.exe -seed %(seed)s -n %(nevents)s|"
        # Deal with executable
        if hepevt_source[-1] == "|":
            exe = hepevt_source.split(' ')[0]
            if exe[0] != '/':   # Try to find full path
                import os, os.path
                path = os.getenv('PATH')
                for p in path:
                    if (os.path.isfile(path+"/"+exe)):
                        exe = path+"/"+exe
                        break
                    continue
                pass
            source = exe + ' ' + ' '.join(hepevt_source.split(' ')[1:])
            if "%" in source:   # Fill in any placemarks
                source = source%{'nevents':str(nevents),
                                 'seed':str(seed)}
                pass
            pass
        
        
        #generator
        gen=GtHepEvtGenTool(name+'Gen',HepEvtDataSource = source)
        #gen.OutputLevel = 2
        gen.HepEvtDataSource = source
        
        # Set up positioner
        pos=GtPositionerTool(name+'Pos',Volume=volume)
        pos.Strategy = "FullVolume" # also "AvoidDaughters" and "Surface"
        pos.Mode = "Uniform"
        #pos.Mode = "Fixed"
        pos.Spread = 25*units.m
        pos.Position = [0,0,2.5*units.m]
        
        # Set up timerator
        tim=GtTimeratorTool(name+'Tim')
        tim.LifeTime = int(108*units.s)
        
        # transform
        tra=GtTransformTool(name+'Tra',Volume=volume)
        
        pass

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