In This Package:

gen.py

Go to the documentation of this file.

#!/usr/bin/env python

"""
Generate AD PMT Detector Elements.

This generates the Structure XML for all AD PMTs into the TDS at
/dd/Structure/AdPmts and files in to DDDB/AdPmtStructure/*.xml.

This needs to match the hand-written XML in DDDB/AdPmts/geometry.xml.

"""

class AdPmtStructure:

    sites = ['db', 'la', 'far']

    def write(self,outdir):
        from XmlDetDescGen.util import XmlFile, Comment
        from XmlDetDescGen.catalog import Catalog
        from XmlDetDescGen.reference import Reference
        from XmlDetDescGen.util import ExternalEntity

        topcat = Catalog('Structure');
        adcat = Catalog('AdPmtStructure');
        adcat.refs = []
        adcat.things = []
        topcat.refs = [ adcat ]
        topcat.things = []

        ees = []
        for site,adcount in zip(AdPmtStructure.sites,AdPmts.adcounts):
            cat = Catalog('%s-ad'%site);
            cat.refs=[]
            for iad in range(0,adcount):
                iad += 1
                eename="%s%dpmts"%(site,iad)
                ees.append((eename,"%s%d.xml"%(site,iad)))
                cat.refs.append(ExternalEntity(eename))
                continue
            adcat.refs.append(cat)

        file = XmlFile(dtd="../DTD/structure.dtd",
                       external_entities=ees)
        file.objects = [ adcat ]
        file.write(outdir+"/structure.xml")


class AdPmts:

    lvname="/dd/Geometry/PMT/lvPmtHemiwPmtHolder"
    lvname2="/dd/Geometry/PMT/lvHeadonPmtFrame"
    siteids = [0x01,0x02,0x04]
    adcounts = [ 2,    2,     4]

    def __init__(self,site,siteid,adcount):
        self.site = site
        self.siteid = siteid
        self.adcount = adcount
        pass

    # this must match what Conventions/Detectors does
    def pmtid(self,site,adn,icol,iring):
        return (site<<24)|(adn<<16)|(iring<<8)|icol


    def write(self,outdir):

        from XmlDetDescGen.util import XmlFile, Comment
        from XmlDetDescGen.structure import DetElem,UserParameter

        for adn in range(0,self.adcount):
            adn += 1

            pmts = []

            ipmt = 0
            for iring in range(0,8):
                iring += 1

                # 3 cols of PMTs on Ladder #1, 1 through 3
                ladn = 1
                for icol in range(0, 3):
                    icol += 1
                    ipmt = ipmt+1
                    coln = (ladn - 1)*3 + icol
                    pmtid = self.pmtid(siteid,adn,coln,iring)
                    data = {
                        'site':site,
                        'siteid':siteid,
                        'adnum':adn,
                        'ladnum':ladn,
                        'ringnum':iring,
                        'columnnum':icol,
                        'glbcoln':coln,
                        'pmtid':pmtid
                        }
                    
                    de = DetElem('%(site)s-ad%(adnum)d-ring%(ringnum)d-column%(glbcoln)d'%data,
                                 self.lvname,
                                 npath="pvAdPmtArray/pvAdPmtLadder1/pvAdPmtLadderArrayRotated/pvAdPmtLadderArrayRow:%(ringnum)s/pvAdPmtInCol:%(columnnum)d/pvAdPmtUnit"%data,
                                 support="/dd/Structure/AD/%(site)s-oil%(adnum)d"%data)
                    
                    de.refs = [UserParameter("PmtID","int",['0x%x'%pmtid],desc="Packed PMT ID")]
                    
                    pmts.append(de)
                    continue

                # 3 cols of PMTs on Ladder #7
                ladn = 7                
                for icol in range(0, 3):
                    icol += 1
                    ipmt = ipmt+1
                    coln = (ladn - 1)*3 + icol
                    pmtid = self.pmtid(siteid,adn,coln,iring)
                    data = {
                        'site':site,
                        'siteid':siteid,
                        'adnum':adn,
                        'ladnum':ladn,
                        'ringnum':iring,
                        'columnnum':icol,
                        'glbcoln':coln,
                        'pmtid':pmtid
                        }
                    
                    de = DetElem('%(site)s-ad%(adnum)d-ring%(ringnum)d-column%(glbcoln)d'%data,
                                 self.lvname,
                                 npath="pvAdPmtArray/pvAdPmtLadder7/pvAdPmtLadderArrayRotated/pvAdPmtLadderArrayRow:%(ringnum)s/pvAdPmtInCol:%(columnnum)d/pvAdPmtUnit"%data,
                                 support="/dd/Structure/AD/%(site)s-oil%(adnum)d"%data)
                    
                    de.refs = [UserParameter("PmtID","int",['0x%x'%pmtid],desc="Packed PMT ID")]
                    
                    pmts.append(de)
                    continue

                continue

            iring = 0      
            for icol in range(0,6):
                icol += 1
                ipmt = ipmt+1
                pmtid = self.pmtid(siteid,adn,icol,iring)
                data = {
                        'site':site,
                        'siteid':siteid,
                        'adnum':adn,
                        'ringnum':iring,
                        'columnnum':icol,
                        'pmtid':pmtid
                        }
                    
                de = DetElem('%(site)s-ad%(adnum)d-ring%(ringnum)d-column%(columnnum)d'%data,
                             self.lvname2,
                             npath="pvAdCalibrationPmtArray/pvAd2inPmt:%(columnnum)d"%data,
                             support="/dd/Structure/AD/%(site)s-oil%(adnum)d"%data)
                    
                de.refs = [UserParameter("PmtID","int",['0x%x'%pmtid],desc="Packed PMT ID")]
                    
                pmts.append(de)
                continue

            file = XmlFile()
            file.objects = pmts
            file.write(outdir+"/%s%d.xml"%(self.site,adn))
            continue
                             
if '__main__' == __name__:

    import sys,os
    try:
        xddroot = sys.argv[1]
    except IndexError:
        xddroot = os.getenv("XMLDETDESCROOT")

    if not xddroot:
        print "No XMLDETDESCROOT directory given by environment or command line"
        print "Using current working directory"
        xddroot="."

    xddroot = ".."

    # to fix the local vs systematic path problem. --- Wei, Jan, 2010
    xddroot = ".."

    outdir=xddroot + "/DDDB/AdPmtStructure"
    if not os.path.exists(outdir):
        print "Directory does not exist, please make first"
        print outdir
        sys.exit(1)


    for site,siteid,adcount in zip(AdPmtStructure.sites,AdPmts.siteids,AdPmts.adcounts):
        aps = AdPmts(site,siteid,adcount)
        aps.write(outdir)

    aps = AdPmtStructure()
    aps.write(outdir)

---