generateMioctEncodingFile Namespace Reference

Classes
class	Entry

Functions
def	writeXML (g)
def	getTopoCellPosition (rois)
def	main ()

Function Documentation

getTopoCellPosition()

def generateMioctEncodingFile.getTopoCellPosition

(

rois

)

Definition at line 112 of file generateMioctEncodingFile.py.

def getTopoCellPosition(rois):
 
    rois = list(rois)
 
    # ETA
    
    # eta min and max are different in the RPC (barrel) and TGC (endcap and forward)
    # in the RPC: abs(etamin)<abs(etamax)
    # in the TGC: etamin<etamax
    # we are going to use the TGC definition: etamin<etamax
 
    cellsEta = [(min(e.etamin,e.etamax), max(e.etamin,e.etamax)) for e in rois]
    etamins, etamaxs = zip(*cellsEta)  # transpose
    etamin = min( [ min(e.etamin,e.etamax) for e in rois] )
    etamax = max( [ max(e.etamin,e.etamax) for e in rois] )
    eta = (etamin+etamax)/2
 
 
    # PHI CONVENTION
 
    
    cellsPhi = [(e.phimin, e.phimax) for e in rois]
 
    # check if there are cells on both sides of the PI boundary
    upperedge = any([e.phi>6.2 for e in rois])
    loweredge = any([e.phi<0.2 for e in rois])
    splitTopoCell = upperedge and loweredge
 
 
    if splitTopoCell:
        maxAtLowerEdge = max([e.phimax for e in rois if e.phi<1])
        minAtUpperEdge = min([e.phimin for e in rois if e.phi>5])
        centerTopoCell = minAtUpperEdge + maxAtLowerEdge
        if centerTopoCell>=2*PI: # shift down
            phimin = minAtUpperEdge - 2 * PI
            phimax = maxAtLowerEdge
        else: # shift up
            phimin = minAtUpperEdge
            phimax = maxAtLowerEdge + 2 * PI
        phi = (phimin+phimax)/2
        #print "JOERG centerTopoCell = ", centerTopoCell, "=>", phi,"[", phimin, "-", phimax, "]      (" , minAtUpperEdge,"  " , maxAtLowerEdge , ")" 
    else:
 
        phimins, phimaxs = zip(*cellsPhi)  # transpose
        phimin = min(phimins)
        phimax = max(phimaxs)
        phi = (phimin+phimax)/2
 
 
    ieta = round(eta*10)
    iphi = round(phi*10)
 
    #if (phimin * phimax) < -100:
    #    print "eta    ", eta
    #    print "phimin ", phimin
    #    print "phimax ", phimax
    #    print "phi    ", phi
    #    print "iphi   ", iphi
    #    print ""
 
 
    #test = (phimin * phimax) < 0
    #if test:
    #    print '\n'.join([str(x) for x in cells])
    #    print "\n\n\n"
 
 
    return (eta, phi, ieta, iphi, etamin, etamax, phimin, phimax)
 
 
 
 
 
 
 
 

main()

def generateMioctEncodingFile.main

(

)

Definition at line 188 of file generateMioctEncodingFile.py.

def main():
    if len(sys.argv)!=2:
        print("Usage: %s <geomety.dat>" % os.path.basename(sys.argv[0]))
        return
 
    # read in the data files containing the ROI information
    data = open(sys.argv[1])
    lines = [Entry(*l.rstrip().split()) for l in data.readlines() if not l.startswith('#')]
    lines = sorted(lines, key = attrgetter('mioct','connector') )
    data.close()
        
    groups = {}
    for k, g in groupby(lines, key=attrgetter('mioct')):
        groups[k] = {}
        for k2, g2 in groupby(g, key = attrgetter('connector')):
            groups[k][k2] = list(g2)
 
    writeXML(groups)
 
    return 0
 
 
 
 

writeXML()

def generateMioctEncodingFile.writeXML

(

g

)

Definition at line 55 of file generateMioctEncodingFile.py.

def writeXML(g):
    filename = os.path.basename(sys.argv[1]).replace('.dat','.xml')
    xml = open(filename,"w")
    print('<?xml version="1.0" ?>\n', file=xml)
    print('<!DOCTYPE MuCTPiGeometry SYSTEM "MUCTPIGeometry.dtd">\n', file=xml)
    print('<MuCTPiGeometry>', file=xml)
    for mioct in sorted(g.keys()):
 
        allSlots = set()
        allSectorNames = set()
        for sector in g[mioct].values():
            allSlots.update( [e.slot for e in sector] )
            allSectorNames.update( [e.sectorname for e in sector] )
            if len(allSlots) != 1:
                raise RuntimeError("More than one slot defined for MIOCT %i" % mioct)
        slot = allSlots.pop()
        allSectorNames = sorted(list( allSectorNames ))
        
        print('    <MIOCT id="%i" slot="%s">' % ( mioct, slot ), file=xml)
        print('        <!-- contains sectors %s -->' % ", ".join(allSectorNames), file=xml)
        for connector in sorted(g[mioct].keys()):
            sector = sorted(g[mioct][connector], key=attrgetter('eta','phi','etacode','phicode'))
            print('        <Sector connector="%s" name="%s">' % (connector, ", ".join(set([str(e.sectorname) for e in sector])) ), file=xml)
            print('            <!-- contains %i ROIs -->' % len(sector), file=xml)
            print('            <!-- mapping from ROI to coding scheme -->', file=xml)
            for e in sector:
                print('            <ROI eta="%f" phi="%f" etacode="%s" phicode="%s" etamin="%f" etamax="%f" phimin="%f" phimax="%f" roiid="%i"/>' % (e.eta, e.phi, e.etacode, e.phicode, e.etamin, e.etamax, e.phimin, e.phimax, e.roi), file=xml)
            print('        </Sector>', file=xml)
 
        allROIsInMIOCT = []
        map(allROIsInMIOCT.extend, g[mioct].values()) # add all ROIs in this MIOCT
 
        print('        <Decode>', file=xml)
 
        for (eta_code,phi_code), rois in groupby(sorted(allROIsInMIOCT,key=attrgetter('etacode','phicode')), key=attrgetter('etacode','phicode')):
 
            (eta, phi, ieta, iphi, etamin, etamax, phimin, phimax) = getTopoCellPosition(rois)
 
            print('            <TopoCell etacode="%s" phicode="%s" eta="%f" phi="%f" ieta="%i" iphi="%i" etamin="%f" etamax="%f" phimin="%f" phimax="%f"/>' % (eta_code, phi_code, eta, phi, ieta, iphi, etamin, etamax, phimin, phimax), file=xml)
        print('        </Decode>', file=xml)
        print('    </MIOCT>', file=xml)
    # hardcoding the PT at the moment
    print('    <PtEncoding>', file=xml)
    print('        <PtCodeElement pt="1" code="0" value="4"/>', file=xml)
    print('        <PtCodeElement pt="2" code="1" value="6"/>', file=xml)
    print('        <PtCodeElement pt="3" code="2" value="10"/>', file=xml)
    print('        <PtCodeElement pt="4" code="2" value="11"/>', file=xml)
    print('        <PtCodeElement pt="5" code="2" value="15"/>', file=xml)
    print('        <PtCodeElement pt="6" code="2" value="20"/>', file=xml)
    print('    </PtEncoding>', file=xml)
    print('</MuCTPiGeometry>', file=xml)
    xml.close()
    print("Wrote %s" % filename)