In This Package:

Ge68PositionStats.py

Go to the documentation of this file.

#!/usr/bin/env python
#
# Makes reconstructed position histograms
#
#  Usage:
#   nuwa.py -n -1 PositionStats reconAndSimData.root

# Load DybPython
from DybPython.DybPythonAlg import DybPythonAlg
from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl, loaddict
from DybPython.Util import irange

# x, y, and z ranges for graphs
z1=-3000
z2=3500
x1=-3000
x2=3000
y1=-3000
y2=3000
r2=5000

from math import sqrt, atan2

# Make shortcuts to any ROOT classes you want to use
TH1F = gbl.TH1F
TH2F = gbl.TH2F

# Make shortcuts to any ROOT classes you want to use

loaddict("libCLHEPRflx")
loaddict("libHepMCRflx")
Detector = gbl.DayaBay.Detector
AdPmtSensor = gbl.DayaBay.AdPmtSensor
ServiceMode = gbl.ServiceMode
ReconStatus = gbl.ReconStatus

# Make your algorithm
class PositionStatsAlg(DybPythonAlg):
    "Algorithm to make Position Statistics file"
    def __init__(self,name):
        DybPythonAlg.__init__(self,name)
        return

    def initialize(self):
        status = DybPythonAlg.initialize(self)
        if status.isFailure(): return status
        self.info("initializing")

        self.cableSvc = self.svc('ICableSvc','StaticCableSvc')
        if self.cableSvc == None:
            self.error("Failed to get StaticCableSvc")
            return FAILURE

        hist = TH2F("reconZvsX",
                    "Reconstructed Z position vs. X position",
                    200,x1,x2,
                    200,z1,z2)
        hist.GetXaxis().SetTitle("X position")
        hist.GetYaxis().SetTitle("Z position")
        self.stats["/file0/position/reconZvsX"] = hist

        hist = TH2F("reconZvsY",
                    "Reconstructed Z position vs. Y position",
                    200,y1,y2,
                    200,z1,z2)
        hist.GetXaxis().SetTitle("Y position")
        hist.GetYaxis().SetTitle("Z position")
        self.stats["/file0/position/reconZvsY"] = hist

        hist = TH2F("reconYvsX",
                    "Reconstructed Y position vs. X position",
                    200,x1,x2,
                    200,y1,y2)
        hist.GetXaxis().SetTitle("X position")
        hist.GetYaxis().SetTitle("Y position")
        self.stats["/file0/position/reconYvsX"] = hist


        hist = TH2F("reconZvsR",
                    "Reconstructed Z position vs. R position",
                    200,0.0,r2,
                    200,z1,z2)
        hist.GetXaxis().SetTitle("R position")
        hist.GetYaxis().SetTitle("Z position")
        self.stats["/file0/position/reconZvsR"] = hist

        return SUCCESS

    def execute(self):
        self.info("executing")
        
        evt = self.evtSvc()

        # Calibrated Readout Data
        recHdr = evt["/Event/Rec/RecHeader"]
        if recHdr == None:
            self.error("Failed to get RecHeader")
            return FAILURE
        recResults = recHdr.recResults()
        recTrigger = recResults["AdSimple"]
        if recTrigger.positionStatus() == ReconStatus.kGood:
            reconPosition = recTrigger.position()
            recPositionR=sqrt(reconPosition.y()**2+reconPosition.x()**2)
            recPositionPhi=atan2(reconPosition.y(),reconPosition.x())
            #PositionDiff=sqrt((reconPosition.y()-yGenPosition)**2+(reconPosition.x()-xGenPosition)**2+(reconPosition.z()-zGenPosition)**2)
            self.stats["/file0/position/reconZvsX"].Fill(
                                                       reconPosition.x(),
                                                       reconPosition.z() )
            self.stats["/file0/position/reconZvsY"].Fill(
                                                       reconPosition.y(),
                                                       reconPosition.z() )
            self.stats["/file0/position/reconYvsX"].Fill(
                                                       reconPosition.x(),
                                                       reconPosition.y() )
            self.stats["/file0/position/reconZvsR"].Fill(
                                                       recPositionR,
                                                       reconPosition.z() )
            
        return SUCCESS
        
    def finalize(self):
        self.info("finalizing")
        status = DybPythonAlg.finalize(self)
        return status


#####  Job Configuration for nuwa.py ########################################

def configure():
    from StatisticsSvc.StatisticsSvcConf import StatisticsSvc
    statsSvc = StatisticsSvc()
    statsSvc.Output ={"file0":"positionStats.root"}
    import DataSvc
    DataSvc.Configure()
    return

def run(app):
    '''
    Configure and add an algorithm to job
    '''
    app.ExtSvc += ["StaticCableSvc", "StatisticsSvc"]
    positionStatsAlg = PositionStatsAlg("MyPositionStats")
    app.addAlgorithm(positionStatsAlg)
    pass

---