In This Package:

TestCoordSysSvc.py

Go to the documentation of this file.

#!/usr/bin/env python

from GaudiPython.GaudiAlgs import GaudiAlgo
from GaudiPython import SUCCESS, FAILURE

class TesterAlg(GaudiAlgo):

    def __init__(self,myname,npoints,target_de_name):
        GaudiAlgo.__init__(self,myname)
        self.npoints = npoints
        self.target_de_name = target_de_name
        return

    def initialize(self):
        sc = GaudiAlgo.initialize(self)
        if sc.isFailure(): 
            print 'Failed to initialize GaudiAlgo base class'
            return sc
        self.css = self.svc('ICoordSysSvc','CoordSysSvc')
        if not self.css: 
            print 'Failed to get CoordSysSvc'
            return FAILURE
        self.ExecuteOnce = True
        return sc;

    def execute(self):

        import GaudiKernel.SystemOfUnits as units

        import time
        start = time.time()
        for x in range(0,self.npoints):
            print x
            for y in range(0,self.npoints):
                for z in range(0,self.npoints):
                    pt = [x*units.mm,y*units.mm,z*units.mm]
                    self.do_point(pt)
                    if x+y+z == 0: first = time.time()
                    continue
                continue
            continue
        stop = time.time()

        print "\nFirst in %f seconds, remaining %d points in %f seconds"%\
            (first-start,pow(self.npoints,3)-1,stop-first)
        

        return SUCCESS

    def do_point(self,pt):


        # Get underlying DE object
        de = self.getDet(self.target_de_name)
        if not de:
            print 'Failed to get DE',self.target_de_name
            return FAILURE

        # Make a point at the center of the target DE
        import PyCintex
        Gaudi = PyCintex.makeNamespace('Gaudi')
        point = Gaudi.XYZPoint(pt[0],pt[1],pt[2])
        gpoint = de.geometry().toGlobal(point)
        
        # Sanity check
        btde = self.css.belongsToDE(gpoint)
        while btde:
            if btde.name() != de.name(): break
            btde = btde.parentIDetectorElement()
            continue
        if not btde:
            print 'Bogus: center of target DE is not in target DE',\
                self.target_de_name
            return FAILURE
                
        # Find coord system
        csde = self.css.coordSysDE(gpoint)
        if not csde:
            print 'Failed to find coordinate system DE for '\
                '[',gpoint.x(),gpoint.y(),gpoint.z(),']'
            return FAILURE

        #print 'Found coordinate system DE for target',self.target_de_name,\
        #    'as:',csde.name()
                    
        return SUCCESS

class TesterAlg2(GaudiAlgo):

    def __init__(self,myname,global_points,should_fail = False):
        GaudiAlgo.__init__(self,myname)
        self.gp = global_points
        self.should_fail = should_fail
        return

    def initialize(self):
        sc = GaudiAlgo.initialize(self)
        if sc.isFailure(): 
            print 'Failed to initialize GaudiAlgo base class'
            return sc
        self.css = self.svc('ICoordSysSvc','CoordSysSvc')
        if not self.css: 
            print 'Failed to get CoordSysSvc'
            return FAILURE
        self.ExecuteOnce = True
        return sc;

    def execute(self):

        import PyCintex
        Gaudi = PyCintex.makeNamespace('Gaudi')
        gpoint = Gaudi.XYZPoint(self.gp[0],self.gp[1],self.gp[2])

        print 'Checking %f,%f,%f'%(self.gp[0],self.gp[1],self.gp[2])
        btde = self.css.belongsToDE(gpoint)
        print 'Found belong to DE: %s'%btde.name()
        lpoint = btde.geometry().toLocal(gpoint)
        print 'Local point in belongs to DE: %f,%f,%f'%(lpoint.x(),lpoint.y(),lpoint.z())

        # Find coord system
        csde = self.css.coordSysDE(gpoint)
        if csde:
            print 'Found coord sys DE: %s'%csde.name()
            if self.should_fail:
                print '\tbut, I should not have.'
                return FAILURE
            return SUCCESS
        print 'Failed to find coord sys DE for point'
        if self.should_fail:
            print '\twhich is as expected'
            return SUCCESS
        return FAILURE

def configure():
    from DetHelpers.DetHelpersConf import CoordSysSvc
    css = CoordSysSvc()
    css.OutputLevel = 1

    from Gaudi.Configuration import ApplicationMgr
    theApp = ApplicationMgr()
    theApp.ExtSvc.append(css)
    return

def run(app):

    ta = TesterAlg2('TesterAlg21',[-414582.0,814258.0,-1923.73])
    app.addAlgorithm(ta)

    ta = TesterAlg2('TesterAlg22',[-410670.0,813170.0,3100.75])
    app.addAlgorithm(ta)

    ta = TesterAlg2('TesterAlg23',[-417953,747954,-62099.1],True)
    app.addAlgorithm(ta)

    size = 10
    ta1 = TesterAlg('TesterAlg11',size,
                    '/dd/Structure/AD/db-ade1/db-sst1/db-oil1/db-oav1/db-lso1')
    app.addAlgorithm(ta1)
    ta2 = TesterAlg('TesterAlg12',size,
                    '/dd/Structure/AD/db-ade2/db-sst2/db-oil2/db-oav2/db-lso2')
    app.addAlgorithm(ta2)
    return

---