PrintPMTMap.py

Go to the documentation of this file.

# nuwa.py -n 0 --dbconf="offline_db" -m"Quickstart.PrintPMTMap -s dayabay -d iws"
# chao@bnl.gov 2011-11-08

from DybPython.DybPythonAlg import DybPythonAlg
from GaudiPython import SUCCESS, FAILURE
from GaudiPython import gbl
import PyCintex
from DybPython.Util.decorators import register_svc, initialize_alg, finalize_alg

FeeChannelId = gbl.DayaBay.FeeChannelId
AdPmtSensor = gbl.DayaBay.AdPmtSensor
PoolPmtSensor = gbl.DayaBay.PoolPmtSensor

TimeStamp = gbl.TimeStamp
Detector = gbl.DayaBay.Detector
ServiceMode = gbl.ServiceMode
Site = gbl.Site
SimFlag = gbl.SimFlag
Context = gbl.Context
DetectorId = gbl.DetectorId

class PrintPMTMap(DybPythonAlg):
    
    def __init__(self, name):
        DybPythonAlg.__init__(self, name)
        self.pmtCount = 0
        self.site = 'dayabay'
        self.detector = 'iws'
        self.timestamp = ''
        self.detId_dict = {
            'ad1' : DetectorId.kAD1,
            'ad2' : DetectorId.kAD2,
            'ad3' : DetectorId.kAD3,
            'ad4' : DetectorId.kAD4,
            'iws' : DetectorId.kIWS,
            'ows' : DetectorId.kOWS,
        }
        self.siteId_dict = {
            'dayabay' : Site.kDayaBay,
            'lingao' : Site.kLingAo,
            'far' : Site.kFar,
        }
    
    # -------------------------------------- 
    @register_svc('pmtGeomSvc', ('IPmtGeomInfoSvc','PmtGeomInfoSvc'))
    @register_svc('cableSvc', ('ICableSvc','CableSvc'))
    @initialize_alg
    def initialize(self):
        try:
            self.siteId = self.siteId_dict[self.site]
            self.detId = self.detId_dict[self.detector]
        except IndexError:
            self.error("Site:Detector not found: %s:%s" % (self.site, self.detector))
            return FAILURE
        
        if not self.timestamp:
            import time
            self.t = TimeStamp(time.time()) # now
        else:
            year, month, day = self.timestamp.split(':')
            self.t = TimeStamp(int(year), int(month), int(day), 0, 0, 0)
                        
        if (self.detId in (DetectorId.kIWS, DetectorId.kOWS)):
            self.printPoolPMTMap()                                        
        else:
            self.printAdPMTMap()
        return SUCCESS
        
    # --------------------------------------
    def printPoolPMTMap(self):
        Gaudi = PyCintex.makeNamespace('Gaudi')
        simFlag = SimFlag.kData
        context = Context(self.siteId, simFlag, self.t, self.detId)
        svcMode = ServiceMode(context, 0)
        
        de_dict = {
            'dayabay' : '/dd/Structure/Pool/db-ows',
            'lingao' : '/dd/Structure/Pool/la-ows',
            'far' : '/dd/Structure/Pool/far-ows',
        }  # center is center of ows
        de = self.getDet(de_dict[self.site]) 
        
        print "# id detector wall spot inward-facing board connector x[mm] y[mm] z[mm]"
        
        for pmtId in self.cableSvc.poolPmtSensors(svcMode):
            if pmtId.bogus(): continue
            channelId = self.cableSvc.feeChannelId(pmtId, svcMode)
            pmtGeomInfo = self.pmtGeomSvc.get(pmtId.fullPackedData())
            if (pmtGeomInfo):
                globalVector = pmtGeomInfo.globalPosition()
                globalXYZ = Gaudi.XYZPoint(globalVector.x(), globalVector.y(), globalVector.z())
                localXYZ = de.geometry().toLocal(globalXYZ)
                
                print "%s %4d %4d %4d %4d %4d %10.1f %10.1f %10.1f" % (pmtId.asString(), 
                    pmtId.wallNumber(), pmtId.wallSpot(), pmtId.inwardFacing(), 
                    channelId.board(), channelId.connector(),
                    localXYZ.x(), localXYZ.y(), localXYZ.z())
                self.pmtCount += 1

        # the other way around:
        # 
        # chanId = FeeChannelId(board, connector, self.siteId, self.detId)
        # pmtId = self.cableSvc.poolPmtSensor(chanId, svcMode)


    # --------------------------------------
    def printAdPMTMap(self):
        Gaudi = PyCintex.makeNamespace('Gaudi')
        simFlag = SimFlag.kData
        context = Context(self.siteId, simFlag, self.t, self.detId)
        svcMode = ServiceMode(context, 0)
        
        de_dict = {
            'dayabay' : '/dd/Structure/AD/db-'+self.detector.replace('ad', 'sst'),
            'lingao' : '/dd/Structure/AD/la-'+self.detector.replace('ad', 'sst'),
            'far' : '/dd/Structure/AD/far-'+self.detector.replace('ad', 'sst'),
        } # center is the center of sst
        de = self.getDet(de_dict[self.site]) 
                
        print "# id detector ring column board connector x[mm] y[mm] z[mm]"
        
        for pmtId in self.cableSvc.adPmtSensors(svcMode):
            if pmtId.bogus(): continue
            channelId = self.cableSvc.feeChannelId(pmtId, svcMode)
            pmtGeomInfo = self.pmtGeomSvc.get(pmtId.fullPackedData())
            if (pmtGeomInfo):
                globalVector = pmtGeomInfo.globalPosition()
                globalXYZ = Gaudi.XYZPoint(globalVector.x(), globalVector.y(), globalVector.z())
                localXYZ = de.geometry().toLocal(globalXYZ)
                
                print "%s %4d %4d %4d %4d %10.1f %10.1f %10.1f" % (pmtId.asString(), 
                    pmtId.ring(), pmtId.column(),
                    channelId.board(), channelId.connector(),
                    localXYZ.x(), localXYZ.y(), localXYZ.z())
                self.pmtCount += 1                      
                    
    # --------------------------------------         
    def execute(self):
        self.info("bogus execution")
        return SUCCESS
        
    # -------------------------------------- 
    @finalize_alg
    def finalize(self):
        self.info("%s %s: Total PMT Count: %d" % (self.site.title(), self.detector.title(), self.pmtCount))
        return SUCCESS


#####  Job Configuration for nuwa.py ########################################
options = None

def configure(argv=[]):
    """Configuration with command line arguments"""   
    global options 
    from optparse import OptionParser

    parser = OptionParser()
    parser.add_option("-d", "--detector", type="string",
        default='iws', help="detector name")
    parser.add_option("-s", "--site", type="string",
        default='dayabay', help="site name")
    parser.add_option("-t", "--timestamp", type="string",
        default='', help="timestamp of dbi query [y:m:d]")        
    (options, args) = parser.parse_args(args=argv)
    
def run(app):
    app.ExtSvc += ["CableSvc", "PmtGeomInfoSvc"]
    alg = PrintPMTMap("PrintPMTMap")
    alg.detector = options.detector.lower()
    alg.site = options.site.lower()
    alg.timestamp = options.timestamp
    
    app.addAlgorithm(alg)