visdet.py

Go to the documentation of this file.

#!/usr/bin/env python

class VisDetConfig:
    def __init__(self,logvol="",detelem="",which="RawTracerX"):
        self.detelem = detelem
        self.logvol = logvol
        self.which = which

        print "Using vis method: ",self.which

        import gaudimodule as gm
        import units
        app = gm.AppMgr()
        app.ExtSvc += ["GiGa"]

        modularPL = app.property("GiGa.GiGaPhysListModular")
        #modularPL.OutputLevel = 1
        modularPL.CutForElectron = 100*units.micrometer
        modularPL.CutForPositron = 100*units.micrometer
        modularPL.CutForGamma = 1*units.millimeter
        modularPL.PhysicsConstructors = [ 
            "DsPhysConsGeneral",
            "DsPhysConsOptical" 
            ]

        print "\tGetting property GiGaEventActionCommand:"
        viscmd = app.property("GiGa.GiGaEventActionCommand")
        #viscmd.OutputLevel = 1
        #viscmd.EndOfEventCommands = [
        #    "/vis/viewer/update" 
        #    "/vis/viewer/flush",
        #    ]

        print "\tSet EndOfEventCommands"

        if self.which == "DAWNFILE":
            viscmd.BeginOfEventCommands = \
                [ "/vis/scene/create",
                  "/vis/sceneHandler/create DAWNFILE",
                  "/vis/viewer/create",
                  "/vis/viewer/set/viewpointThetaPhi -90 -90",
                  "/vis/scene/add/volume", 
                  "/vis/scene/add/trajectories" ]
        elif self.which == "HepRepFile":
            viscmd.BeginOfEventCommands = [
                "/vis/open HepRepFile",
                #"/vis/scene/create",
                #"/vis/scene/add/volume "+self.detelem,
                #"/vis/DrawVolume",
                "/vis/viewer/flush"
                ]
        elif self.which == "HepRepXML":
            viscmd.BeginOfEventCommands = \
                [ "/vis/open HepRepXML",
                  "/vis/scene/create far.heprep",
                  "/vis/scene/add/volume",
                  "/vis/sceneHandler/attach",
                  #"/vis/drawVolume",
                  "/vis/viewer/flush" 
                  ]
        elif self.which == "OGLIXm" or self.which == "OGLSXm":
            viscmd.BeginOfEventCommands = [
                "/vis/open "+self.which,
                "/vis/drawVolume",
                ]                
        elif self.which == "OGLIX" or self.which == "OGLSX":
            viscmd.BeginOfEventCommands = [
                "/vis/open "+self.which,
                "/vis/drawVolume",
                ]                
        elif self.which == "OIX":
            viscmd.BeginOfEventCommands = [
                "/vis/open OIX",
                "/vis/drawVolume",
                ]                
        elif self.which == "RayTracer" or self.which == "RayTracerX":
            viscmd.BeginOfEventCommands = [
                #"/vis/open "+self.which,
                #"/vis/viewer/reset",
                #"/vis/scene/create",
                #"/vis/scene/add/volume "+self.detelem,
                #"/vis/sceneHandler/attach",
                #"/vis/viewer/set/style surface",
                #"/vis/viewer/set/projection perspective 50 deg",
                #"/vis/viewer/set/viewpointThetaPhi 130 20",
                #"/vis/viewer/zoom 0.5",
                #"/vis/rayTracer/trace foo.jpg",
                #"/vis/viewer/flush",
                ]                
        elif self.which == "ATree":
            viscmd.BeginOfEventCommands = [
                "/vis/open ATree",
                "/vis/ASCIITree/Verbose 15",
                "/vis/scene/create",
                "/vis/scene/add/volume "+self.detelem,
                "/vis/sceneHandler/attach",
                "/vis/drawTree",
                "/vis/viewer/flush",
                ]
        elif self.which == "GAGTree":
            viscmd.BeginOfEventCommands = [
                "/vis/open GAGTree",
                "/vis/scene/create",
                "/vis/scene/add/volume "+self.detelem,
                "/vis/sceneHandler/attach",
                "/vis/drawTree",
                "/vis/viewer/flush",
                ]
        elif self.which == "xml":
            viscmd.BeginOfEventCommands = [
                "/vis/open XMLTree",
                "/vis/scene/create",
                "/vis/scene/add/volume "+self.detelem,
                "/vis/sceneHandler/attach",
                "/vis/drawTree",
                "/vis/viewer/flush",
                ]
        else:
            viscmd.BeginOfEventCommands = []

        #eventseq = app.property("GiGa.GiGaEventActionSequence")
        #eventseq.Members = [# "GiGaEventActionDraw/DrawEvent",
        #"GiGaEventActionCommand" ]


        app.TopAlg += [ "GaudiSequencer/Simulation" ]
        seq = app.algorithm("Simulation")
        seq.Members = [ "GiGaInputStream/GIO" ]

        gio = app.algorithm("GIO")
        gio.ExecuteOnce = True
        gio.ConversionSvcName = "GiGaGeo"
        gio.DataProviderSvcName = "DetectorDataSvc"
        gio.StreamItems = [ self.logvol, self.detelem ]

        import units
        ggg = app.service("GiGaGeo")
        ggg.XsizeOfWorldVolume = 2400*units.meter
        ggg.YsizeOfWorldVolume = 2400*units.meter
        ggg.ZsizeOfWorldVolume = 2400*units.meter
        #ggg.WorldPhysicalVolumeName = "pvFarSiteRock"
        #ggg.WorldLogicalVolumeName = "lvFarSiteRock"

        seq.Members += [ "DsPushKine/PushKine", "DsPullEvent/PullEvent" ]
        push = app.algorithm("PushKine")
        push.Converter = "HepMCtoG4"

        giga = app.service("GiGa")
        giga.PhysicsList = "GiGaPhysListModular"
        giga.EventAction = "GiGaEventActionCommand"
        giga.VisManager = "GiGaVisManager/GiGaVis"
        if self.which[0:3] == 'Ray':
            giga.UIsession = "GiGaUIsession/GiGaUI"
        



if '__main__' == __name__:

    import sys, getopt, os
    det=os.getenv("XMLDETDESCROOT")
    if det == "":
        print "Warning: no XMLDETDESCROOT variable, no default detector description"
    else:
        det += "/DDDB/dayabay.xml"

    optlist,args = getopt.getopt(sys.argv[1:],"-V:v:d:l:p:w:")

    log = "/dd/Geometry/Pool/lvFarPoolOWS"
    phy = "/dd/Structure/daya-bay/far-rock/far-ows"

    verbose = None
    which = "RayTracerX"
    for opt,arg in optlist:
        if opt == "-V":
            verbose = arg
        if opt == "-v":
            vol = arg
        if opt == "-d":
            det = arg
        if opt == "-l":
            log = arg
        if opt == "-p":
            phy = arg
        if opt == "-w":
            which = arg
        pass
    pass
    
    #print "Top volume =",vol
    print "Detector file = ",det
    print "Logical volume = ",log
    print "Physical volume = ",phy

    import gaudimodule as gm
    app = gm.AppMgr()
    if verbose:
        app.service('MessageSvc').OutputLevel = 1
    app.EvtMax = 2
    app.EvtSel = "NONE"


    import xmldetdesc,gentools
    # Order matters
    xddc = xmldetdesc.XmlDetDescConfig(loc=det)
    # there be bug, shouldn't need this but I do:
    tg = gentools.GenToolsConfig(volume=log)
    dsc = VisDetConfig(detelem=phy, logvol=log, which=which)

    app.run(1)