coordsysdump.py

Go to the documentation of this file.

#!/usr/bin/env python
'''
Dump local-volume coordinate axis

Note to others: this uses a non-standard way to boot up Gaudi
'''

from DybPython import gaudiutil
from GaudiKernel import SystemOfUnits as units
from GaudiPython import gbl

# site in [db, la, far]
# det in [Pool, AD]
# vol in [iws,ows,ade1,sst2,oil4, (etc)]
site_de_pat = '/dd/Structure/DayaBay/%(site)s-rock'
det_de_pat = '/dd/Structure/%(det)s/%(site)s-%(vol)s'


def volumes():
    '''
    Iterate over possible volumes. 

    for v in volumes(): ...
    '''
    for site in ['db', 'la', 'far']:
        det = 'Pool'
        for vol in ['iws','ows']:
            yield locals()
            continue            # vol

        numads = 2
        if site is 'far': numads = 4
        det = 'AD'
        for adnum in range(numads):
            adnum += 1
            for vol in ['%s%d'%(x,adnum) for x in ['ade','sst','oil','lso','gds']]:
                yield locals()
                continue        # vol
            continue            # adnum

        continue                # site
    raise StopIteration
    
def vec2str(vec, unit_conv = 1):
    'Return string representation of given thing with .x()/.y()/.z()'
    return '[%.3f, %.3f, %.3f]' % (vec.x()/unit_conv, vec.y()/unit_conv, vec.z()/unit_conv)

def v2v(vec,de1,de2):
    'Transform vec from de1 to global and global to de2'
    gvec = de1.geometry().toGlobal(vec)
    return de2.geometry().toLocal(gvec)

def d2d(vec1,vec2,de1,de2):
    'Return the vector from vec1 to vec2 local to de1 expressed in de2 coordinates'
    lv1 = v2v(vec1,de1,de2)
    lv2 = v2v(vec2,de1,de2)
    lv1.SetXYZ(lv2.x()-lv1.x(),lv2.y()-lv1.y(),lv2.z()-lv1.z())
    return lv1

#css = gaudiutil.coordSvc()
zero = gbl.ROOT.Math.XYZPoint(0,0,0)
xaxis = gbl.ROOT.Math.XYZPoint(1,0,0)
yaxis = gbl.ROOT.Math.XYZPoint(0,1,0)
zaxis = gbl.ROOT.Math.XYZPoint(0,0,1)

if __name__ == '__main__':
    import os

    last_site = None
    for d in volumes():
        site_name = site_de_pat%d
        det_name = det_de_pat%d
        site_de = gaudiutil.deify(site_name)
        det_de = gaudiutil.deify(det_name)
        
        site_short_name = os.path.split(site_name)[-1]
        det_short_name = os.path.split(det_name)[-1]

        if site_name != last_site:
            last_site = site_name
            print '%s origin in globals %s' % (site_name, vec2str(site_de.geometry().toGlobal(zero)))
            pass

        lorig = v2v(zero,det_de,site_de)
        lxaxis = d2d(zero,xaxis,det_de,site_de)
        lyaxis = d2d(zero,yaxis,det_de,site_de)
        lzaxis = d2d(zero,zaxis,det_de,site_de)

        print '\t%s @ %s (meter) X=%s, Y=%s, Z=%s' % (det_short_name, 
                                                      vec2str(lorig, units.meter),
                                                      vec2str(lxaxis), vec2str(lyaxis), vec2str(lzaxis))