gentools.py

Go to the documentation of this file.

#!/usr/bin/env python

class GenToolsConfig:
    def __init__(self,
                 volume="/dd/Geometry/AD/lvAD",
                 nevents=10,
                 seed=1234567,
                 # Use Co60.exe because it is fast, easy testing
                 hepevt_source="Co60.exe -seed %(seed)s -n %(nevents)s|"):
        import GaudiPython as gm
        self.app = gm.AppMgr()
        #msg = self.app.service("MessageSvc")
        #msg.OutputLevel = 1
        self.nevents = nevents
        self.seed = seed
        self.volume = volume
        self.hepevt_source=hepevt_source
        self.init()
        return

    def init(self):
        self.init_gen_tools()
        self.init_generator()
        return

    def init_gen_tools(self):
        from GaudiKernel import SystemOfUnits as units

        #print " init_gen_tools"

        # Set up timerator
        tim = self.app.property("ToolSvc.GtTimeratorTool")
        #tim.OutputLevel=1
        tim.LifeTime = int(1*units.second)


        # Set up positioner
        #print " Creating poser"
        poser = self.app.property("ToolSvc.GtPositionerTool")
        poser.OutputLevel = 2
        poser.Volume = self.volume
        poser.Strategy = "FullVolume" # also "AvoidDaughters" and "Surface"
        poser.Mode = "Uniform"
        poser.Spread = 10*units.cm
        poser.Position = [0,0,0]

        # Deal with executable
        if self.hepevt_source[-1] == "|":
            exe = self.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(self.hepevt_source.split(' ')[1:])
            if "%" in source:   # Fill in any placemarks
                source = source%{'nevents':str(self.nevents),
                                 'seed':str(self.seed)}
                pass
            pass
        

        # HEPEvt -> HepMC generator
        #hepevt = self.app.property("ToolSvc.GtHepEvtGenTool")
        #hepevt.HepEvtDataSource = source
        #print " HepEvtGenTool with", source

        # Set up gun
        gun = self.app.property("ToolSvc.GtGunGenTool")
        gun.OutputLevel = 2
        gun.ParticleName = "opticalphoton"
        gun.MomentumMode = "Fixed"
        gun.Momentum = 2.5*units.eV
        gun.DirectionMode = "Fixed"
        gun.Direction = [ 0.0,0.0,-1.0 ]


        return

    def init_generator(self):
        #print " init_generator"
#        self.app.TopAlg += [ "Gnrtr/Generator", "GtHepMCDumper/Dumper" ]
        self.app.TopAlg += [ "Gnrtr/Generator" ]
        #print " GtGenerator"
        gen = self.app.algorithm("Generator")
        #gen.OutputLevel = 2
        #gen.GenTools = [ "GtHepEvtGenTool", "GtPositionerTool", "GtTimeratorTool" ]
        gen.GenTools = [ "GtGunGenTool/GunGen", "GtPositionerTool/GunPos", "GtTimeratorTool/GunTim" ]
        gen.GenName = "Bang Bang"
        gen.Location = "/Event/Gen/HepMCEvents" # this is default anyways

        #print " GtDumper"
        dump = self.app.algorithm("Dumper")
        dump.Location = "/Event/Gen/HepMCEvents"  # this is default anyways.
        return

if '__main__' == __name__:
    import os
    loc = os.getenv('XMLDETDESCROOT')
    if not loc:
        raise RuntimeError,"Location of DetDesc XML file not specified"
    loc += "/DDDB/pmtbox.xml"

    import xmldetdesc
    xddc = xmldetdesc.XmlDetDescConfig(loc=loc)

    tg = GenToolsConfig(volume="/dd/Structure/steel-2/water-2")
    from GaudiPython import AppMgr
    app = AppMgr()
    app.EvtSel = "NONE"
    app.run(tg.nevents)

    msv = app.service('MessageSvc')
    msv.OutputLevel = 2