In This Package:

xmldetdesc.py

Go to the documentation of this file.

#!/usr/bin/env python
"""
  Example of interactive exploration of detector description, following 
  getting into the CMT environment :
      dyb__ dybgaudi/Detector/XmlDetDesc
       
  Using python -i or ipython : 
    
      import xmldetdesc as xdd
      dsv = xdd.detsvc()
      
      In [3]: dsv.dir("/dd", 2 )
        /dd <class 'GaudiPython.Pythonizations.DataObject'> 
        /dd/Geometry
        /dd/Materials
        /dd/Structure
      Out[3]: SUCCESS

      In [4]: dsv.dir("/dd/Geometry", 2 )
        /dd/Geometry <class 'GaudiPython.Pythonizations.DataObject'> 
        /dd/Geometry/DayaBay
        /dd/Geometry/Sites
        /dd/Geometry/Pool
        /dd/Geometry/PoolDetails
        /dd/Geometry/PmtPanel
        /dd/Geometry/AD
        /dd/Geometry/AdDetails
        /dd/Geometry/AdPmts
        /dd/Geometry/RPC
        /dd/Geometry/PMT
     Out[4]: SUCCESS

  CAUTION these will list the entire description:
      dsv.dir("/dd", -1 )   
      dsv.dir("/dd" )

"""



from Gaudi.Configuration import *

def config(xmlfile=None):

    if not xmlfile:
        import sys, os
        loc = os.getenv("XMLDETDESCROOT")
        if not loc:
            print "Warning: no XMLDETDESCROOT variable, no detector description"
            sys.exit(1)
        else:
            loc += "/DDDB/dayabay.xml"
            pass
        xmlfile = loc
        pass



    from XmlTools.XmlToolsConf import XmlCnvSvc, XmlParserSvc
    xmlcnv = XmlCnvSvc()
    xmlcnv.AllowGenericConversion = True
    xmlparser = XmlParserSvc()
    
    app = ApplicationMgr()
    app.ExtSvc += [ xmlcnv , xmlparser ] 
    
    detper = DetectorPersistencySvc()
    detper.CnvServices.append(xmlcnv)
    
    detdat = DetectorDataSvc()
    detdat.UsePersistency = True
    detdat.DetDbRootName  = "dd"
    detdat.DetStorageType = 7
    detdat.DetDbLocation  = xmlfile
    return



def dress_idatasvc():
    import GaudiPython as gaudi

    def _dir_ ( self , node = None , level = -1 ) :
        """
        The simple tool to perform the inspection fo Data Store
        
        Usage:
        
        evtSvc = gaudi.evtSvc()
        evtSvc.dir('MC')
        data = evtSvc['Rec/Calo']
        evtSvc.dir( data)
        
        """
        if 0 == level          : return gaudi.SUCCESS ;
        if   str is type(node) : node = self.retrieveObject( node )
        elif  not node         : return self.dir('', level )
        if not node            : return gaudi.FAILURE
        if     hasattr ( node , 'registry'   ) : node = node.registry()
        if     hasattr ( node , 'identifier' ) :
            obj = node.object ()
            if not obj :
                obj = self.retrieveObject( node.identifier() ) 
                print  node.identifier()
                pass
            else :
                if hasattr( obj , 'size'    ) :
                    size = obj.size()
                    if 0 == size : print "%s (empty) %s" % (  node.identifier() , type( obj ) )
                    elif hasattr ( obj , 'containedObjects' ) :
                        c = obj.containedObjects()[0]
                        print "%s container of %d %s objects " % ( node.identifier() , obj.size() , type(c))
                        pass
                    else : 
                        print  "%s %s " % (  node.identifier() , type( node.object()) )
                        pass
                else :
                    print  "%s %s " % (  node.identifier() , type( node.object()) )
                    pass
        else : print "The node has no 'identifier'"
        # start the recursion 
        for l in self.leaves( node ) :
            if l : self.dir( l , level - 1 )  
            continue
        return gaudi.SUCCESS 
        
    def _get_( self , node = None , cut = lambda x : True ) :
        if   str is type(node) : node = self.retrieveObject( node )
        elif  not node         : return _get_( self , '' , cut  )
        if not node            : return []
        if     hasattr ( node , 'registry'   ) : node = node.registry()
        result = []
        if     hasattr ( node , 'identifier' ) :
            obj = node.object()
            if cut( obj ) : result += [ obj ]
            pass
        for l in self.leaves ( node ) :
            if l : result += _get_( self , l , cut )
            continue
        return result 
    
    gaudi.iDataSvc.dir = _dir_ 
    gaudi.iDataSvc.DIR = _get_ 


def detsvc():
    config()
    dress_idatasvc()
    import GaudiPython as gp
    g = gp.AppMgr()
    dsv = g.detSvc()
    return dsv  

def dump():
    dress_idatasvc()
    import GaudiPython as gaudi
    app = gaudi.AppMgr()
    det = app.detSvc()
    det.dir("/dd")
    det.dump()

    return

if '__main__' == __name__:
    config()
    dump()

---