gen.py

Go to the documentation of this file.

#!/usr/bin/env python

"""
Generate AD Reflector Detector Elements.

This generates the Structure XML for all AD Reflectors into the TDS at
/dd/Structure/AdDetails/ and files in to DDDB/AdReflectorStructure/*.xml.

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

"""

Eight = True
suffix = ''
style = '2-2-4'

class AdReflectorStructure:

    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('AdReflectorStructure');
        adcat.refs = []
        adcat.things = []
        topcat.refs = [ adcat ]
        topcat.things = []

        ees = []
        for site,adcount in zip(AdReflectorStructure.sites,AdReflectors.adcounts):
            cat = Catalog('%s-ad'%site);
            cat.refs=[]
            for iad in range(0,adcount):
                iad += 1
                eename="%s%dreflectors"%(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 AdReflectors:

    lvtopname="/dd/Geometry/AdDetails/lvTopESR"
    lvbotname="/dd/Geometry/AdDetails/lvBotESR"
    siteids = [0x01,0x02,0x04]
    adcounts = [ 2,    2,     4]


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

    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

            reflectors = []

            data = {
                   'site':self.site,
                   'siteid':self.siteid,
                   'adnum':adn,
                   }
                    
            de = DetElem('%(site)s-ad%(adnum)d-topesr'%data,
                        self.lvtopname,
                        npath="pvTopReflector/pvTopRefGap/pvTopESR",
                        support="/dd/Structure/AD/%(site)s-oil%(adnum)d"%data)
            reflectors.append(de)
                    
            de = DetElem('%(site)s-ad%(adnum)d-botesr'%data,
                        self.lvbotname,
                        npath="pvBotReflector/pvBotRefGap/pvBotESR",
                        support="/dd/Structure/AD/%(site)s-oil%(adnum)d"%data)
            reflectors.append(de)

            file = XmlFile()
            file.objects = reflectors
            file.write(outdir+"/%s%d.xml"%(self.site,adn))
            continue
          
def main(): 

    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="."

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

    
    print sys.argv[0],'Configuring for',style

    for site,siteid,adcount in zip(AdReflectorStructure.sites,AdReflectors.siteids,AdReflectors.adcounts):

        aps = AdReflectors(site,siteid,adcount)
        aps.write(outdir)

    aps = AdReflectorStructure()
    aps.write(outdir)


if __name__ == '__main__' :
    main()