In This Package:

test-GeneratorAlg.py

Go to the documentation of this file.

#!/usr/bin/env python

class TestGenerator:
    def __init__(self):
        import gaudimodule as gm
        self.app = gm.AppMgr()
        msv = self.app.service('MessageSvc')
        msv.OutputLevel = 3
        self.nevents = 10
        self.volume = "/dd/Geometry/Pool/lvNearPoolIWS"
        self.init()
        self.app.EvtMax = self.nevents
        self.app.EvtSel = "NONE"
        return

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

    def init_gen_tools(self):
        from units import second

        print " init_gen_tools"
        toolSvc = self.app.toolSvc()

        # Set up timerator
        print " Creating timerator"
        tim = toolSvc.create("GtTimeratorTool","timerator")
        tim.OutputLevel=1
        tim.LifeTime = int(1*second)

        # Set up positioner
        print " Creating poser"
        poser = toolSvc.create("GtPositionerTool", "poser")
        poser.OutputLevel = 3
        poser.Volume = self.volume
        poser.Strategy = "FullVolume" # also "AvoidDaughters" and "Surface"

        # find executable
        exe = "Co60.exe"
        import os, os.path
        path = os.getenv('PATH')
        for p in path:
            if (os.path.isfile(path+"/"+exe)):
                exe = path+"/"+exe
                break
            continue

        # HEPEvt -> HepMC generator
        hepevt = toolSvc.create("GtHepEvtGenTool")
        hepevt.HepEvtDataSource = "%s -n %d -seed 1234567|"%(exe,self.nevents)
        print " HepEvtGenTool with", hepevt.HepEvtDataSource

        return

    def init_generator(self):
        print " init_generator"
        self.app.TopAlg += [ "GtGenerator/Gen", "GtHepMCDumper/Dumper" ]

        print " GtGenerator"
        gen = self.app.algorithm("Gen")
        gen.OutputLevel = 2
        gen.GenTools = [ "GtHepEvtGenTool", "poser", "timerator" ]
        gen.GenName = "Cobalt 60"
        gen.GenEventLoc = "/Event/Gen/HepMCEvents" # this is default anyways

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

    def init_geometry(self):
        print " init_geometry"
        self.app.ExtSvc += [ "XmlCnvSvc" , "XmlParserSvc" ] 
        
        det = self.app.service('DetectorPersistencySvc')
        det.OutputLevel=3
        det.CnvServices = ["XmlCnvSvc"]

        det = self.app.service('XmlCnvSvc')
        #det.OutputLevel=1
        det.AllowGenericConversion = True

        #XmlParserSvc = g.service('XmlParserSvc')
        #XmlParserSvc.EntityResolverSvc = "CondDBEntityResolverSvc"

        det = self.app.service('DetectorDataSvc')
        det.UsePersistency = True
        det.DetDbRootName  = "dd"
        det.DetStorageType = 7
        import os, os.path
        loc = os.getenv('XMLDETDESCROOT')
        if (not loc):
            import sys
            print "Failed to get XMLDETDESCROOT"
            sys.exit(1)
            return
        det.DetDbLocation  = loc + "/DDDB/dayabay.xml"
        print " Using XMLDDDB at",det.DetDbLocation
        return

    def run(self):
        self.app.run(self.nevents)
        return

if '__main__' == __name__:
    tg = TestGenerator()
    tg.run()

---