RpcGeomInfoSvc.cc

Go to the documentation of this file.

#include "RpcGeomInfoSvc.h"
#include "RpcGeomInfo.h"

#include "DetDesc/IDetectorElement.h"
#include "DetDesc/ParamValidDataObject.h"

#include "GaudiKernel/SmartDataPtr.h"
#include "GaudiKernel/MsgStream.h"

using namespace std;

RpcGeomInfoSvc::RpcGeomInfoSvc(const std::string& name, ISvcLocator *svc)
    : Service(name,svc)
    , m_detSvc(0)
    , m_detector(0)
{
    declareProperty("SiteIdUserParameter",m_SiteIdUserParameter="SiteID",
                    "Name of the user parameter attached to Site detector "
                    "elements that gives the packed Site ID number.");
    declareProperty("DetectorIdUserParameter", 
                    m_DetectorIdUserParameter="DetectorID",
                    "Name of the user parameter attached to Detector "
                    "detector elements that gives the fully qualified "
                    "packed Detector ID.");
    declareProperty("RpcIdUserParameter",m_RpcIdUserParameter="RpcID",
                    "Name of the user parameter attached to RPC detector "
                    "elements that gives the fully qualified packed RPC ID");
    declareProperty("StreamItems",m_StreamItems,
                    "List of top level Detector Elements.");
}

RpcGeomInfoSvc::~RpcGeomInfoSvc()
{
}



StatusCode RpcGeomInfoSvc::initialize()
{
    this->Service::initialize();

    MsgStream msg(msgSvc(),name());
    msg << MSG::DEBUG << "RpcGeomInfoSvc::initialize()" << endreq;

    StatusCode sc = service("DetectorDataSvc",m_detSvc,true);
    if (sc.isFailure()) return sc;

    msg << MSG::DEBUG << "Using IDs:"
        << " site: " << m_SiteIdUserParameter
        << " det: " << m_DetectorIdUserParameter
        << " rpc: " << m_RpcIdUserParameter
        << endreq;

    if (! m_StreamItems.size()) {
        msg << MSG::WARNING << "did not get any StreamItems, can not lookup RPCs" << endreq;
        return StatusCode::FAILURE;
    }
    for (size_t ind=0; ind<m_StreamItems.size(); ++ind) {
        string dename = m_StreamItems[ind];
        SmartDataPtr<IDetectorElement> obj(m_detSvc,dename);
        if (!obj) {
            MsgStream msg(msgSvc(),name());
            msg << MSG::WARNING << "Failed to get top Detector Element: \"" 
                << dename << "\", skipping" << endreq;
            sc = StatusCode::FAILURE;
            continue;
        }
        msg << MSG::DEBUG << "Adding top level Detector Element: \"" 
            << dename << endreq;
        m_topDEs.push_back(obj);
    }
    return sc;
}

StatusCode RpcGeomInfoSvc::reinitialize()
{
    return StatusCode::SUCCESS;
}

StatusCode RpcGeomInfoSvc::finalize()
{
    m_detSvc->release();

    // Delete all RpcGeomInfo's.....

    return StatusCode::SUCCESS;
}

IRpcGeomInfo* RpcGeomInfoSvc::get(std::string structure_path)
{
    MsgStream msg(msgSvc(),name());
    msg << MSG::DEBUG << "RpcGeomInfoSvc::get(path "<<structure_path<<")" << endreq;

    IRpcGeomInfo* rpcinfo = m_byPath[structure_path];
    if (rpcinfo) return rpcinfo;

    SmartDataPtr<IDetectorElement> obj(m_detSvc,structure_path);
    if (!obj) return 0;

    unsigned int rpcid = obj->params()->param<int>(m_RpcIdUserParameter);
    if (!rpcid) return 0;
    
    return this->add(rpcid,structure_path,obj);
}

IRpcGeomInfo* RpcGeomInfoSvc::get(IDetectorElement* rpcde)
{
    MsgStream msg(msgSvc(),name());
    msg << MSG::DEBUG << "RpcGeomInfoSvc::get(IDetectorElement "<<rpcde->name()<<")" << endreq;

    IRpcGeomInfo* rpcinfo = m_byDE[rpcde];
    if (rpcinfo) return rpcinfo;

    unsigned int rpcid = rpcde->params()->param<int>(m_RpcIdUserParameter);
    if (!rpcid) return 0;

    return this->add(rpcid,"",rpcde);
}

IRpcGeomInfo* RpcGeomInfoSvc::get(unsigned int rpcid)
{
    MsgStream msg(msgSvc(),name());
    msg << MSG::DEBUG << "RpcGeomInfoSvc::get(int "<<(void*)rpcid<<")" << endreq;
    

    /*
      algorithm: 

      * add site/detector values into high level detector elements in
      XML.

      * add isCompatible() method to Detectors.h, Returns true if bits
      * match, but not including fully zero'd bytes.

      * Search down /dd/Structure, check each DetectorElement for a
      * RpcID user parameter.  Failing that look for a DetectorID
      * parameter and if found check if value isCompatible() and
      * abandon branch if it isn't. Quit if DetectorElements are
      * exhausted or if one is found that matches the RPCID.

     */

    // Have we seen you before?
    IRpcGeomInfo* rpcinfo = m_byId[rpcid];
    if (rpcinfo) {
        msg << MSG::DEBUG << "Found "<< (void*)rpcid << ", seen before"<<endreq;
        return rpcinfo;
    }

    // Let's first do a quick check in the last detector we saw
    if (m_detector) {
        msg << MSG::DEBUG << "Looking for "<< (void*)rpcid 
            << " in "<< m_detector->name() << endreq;
        rpcinfo = this->find(rpcid,m_detector);
        if (rpcinfo) {
            msg << MSG::DEBUG << "Found "<< (void*)rpcid
                << " after quick check"<<endreq;
            return rpcinfo;
        }
        m_detector = 0;
    }

    // No luck, do a full search starting from top level DEs
    for (size_t ind=0; ind<m_topDEs.size(); ++ind) {
        msg << MSG::DEBUG << "Looking for "<< (void*)rpcid 
            << " in "<< m_topDEs[ind]->name() << endreq;
        rpcinfo = this->find(rpcid,m_topDEs[ind]);
        if (rpcinfo) {
            msg << MSG::DEBUG << "Found "<< (void*)rpcid
                << " after full search"<<endreq;
            return rpcinfo;
        }
    }
    msg << MSG::DEBUG << "Failed to find RPC ID "<< (void*)rpcid << endreq;
    return 0;
}

IRpcGeomInfo* RpcGeomInfoSvc::find(unsigned int rpcid, IDetectorElement* de)
{
    MsgStream msg(msgSvc(),name());
    msg << MSG::DEBUG << "RpcGeomInfoSvc::find(int "<<(void*)rpcid
        <<","<<de->name()<<")" << endreq;

    const ParamValidDataObject* params = de->params();
    // Check if DE is a RPC.  If current DE has a RpcID we are done
    // for good or bad.
    if (de->params()->exists(m_RpcIdUserParameter)) {
        unsigned int this_rpcid = (unsigned int)(params->param<int>(m_RpcIdUserParameter));
        if (rpcid == this_rpcid) {
            msg << MSG::DEBUG << "found RPC ID " << (void*)rpcid << endreq;
            return this->add(rpcid,"",de);
        }
        msg << MSG::DEBUG << "got RPC ID but wrong one "
            << (void*)this_rpcid << " != " << (void*)rpcid << endreq;
        return 0;
    }

    // Check if DE is a Detector.  If current DE has a DetectorID we
    // check for compatibility and cache or bail out.
    if (de->params()->exists(m_DetectorIdUserParameter)) {
        unsigned int detid = 
            (unsigned int)(params->param<int>(m_DetectorIdUserParameter));
        msg << MSG::DEBUG << "checking Detector ID "
            << (void*)detid << " =?= " << (void*)(rpcid) << endreq;

        if ( (rpcid&0xffff0000) == detid) { 
            msg << MSG::DEBUG << "found Detector ID "
                << (void*)detid << " for " << (void*)(rpcid) << endreq;
            m_detector = de;    // We have a winner!
        }
        else {                  // We have a detector, but the wrong one
            msg << MSG::DEBUG << "got Detector ID but wrong one "
                << (void*)detid << " != " << (void*)(rpcid&0xffff0000) << endreq;
        }
    }

    // Check if DE is a Site.  If current DE has a SiteID we bail out
    // if it isn't compatible.
    if (params->exists(m_SiteIdUserParameter)) {
        unsigned int siteid = 
            (unsigned int)(params->param<int>(m_SiteIdUserParameter));
        if ( (rpcid&0xff000000) != siteid) {
            msg << MSG::DEBUG << "got Site ID but wrong one "
                << (void*)siteid << " != " << (void*)(rpcid&0xff000000) << endreq;
            return 0;
        }
    }

    // We get here we know we are still on the right track
    IDetectorElement::IDEContainer& child = de->childIDetectorElements();
    for (size_t ind=0; ind < child.size(); ++ind) {
        IRpcGeomInfo *pgi = this->find(rpcid,child[ind]);
        if (pgi) return pgi;
    }
    msg << MSG::DEBUG << "failed to find rpcid " << (void*)rpcid << endreq;
    return 0;
}


IDetectorElement* RpcGeomInfoSvc::findParent(unsigned int rpcid,
                                             IDetectorElement* currentDE)
{
    // Find the parent detector element for this RPC, starting the
    // search at the current detector element.  If currentDE==0, then
    // the entire geometry will be searched.

    MsgStream msg(msgSvc(),name());
    msg << MSG::DEBUG << "RpcGeomInfoSvc::find(int "<<(void*)rpcid;
    if (currentDE) msg <<","<<currentDE->name();
    msg << ")"<< endreq;

    if (!currentDE) {
      // If current detector element is undefined, check the usual suspects
      // First, check the cached detector
      if (m_detector) {
        IDetectorElement* parent = findParent(rpcid, m_detector);
        if(parent) return parent;
        m_detector = 0; // Unset incorrect cached detector
      }
      // Next, check the top-level detector elements
      for (size_t ind=0; ind<m_topDEs.size(); ++ind) {
        if (m_topDEs[ind]){
          IDetectorElement* parent = findParent(rpcid, m_topDEs[ind]);
          if(parent) return parent;
        }
      }
      // Failed to find parent detector
      msg << MSG::DEBUG << "failed to find parent detector for rpcid "
        << (void*)rpcid << endreq;
      return 0;
    }

    msg << MSG::DEBUG << "Looking for "<< (void*)rpcid 
        << " in "<< currentDE->name() << endreq;
    const ParamValidDataObject* params = currentDE->params();

    // Check if current detector element is the rpc parent detector
    if (params->exists(m_DetectorIdUserParameter)) {
      unsigned int detid = 
        (unsigned int)(params->param<int>(m_DetectorIdUserParameter));
      msg << MSG::DEBUG << "checking Detector ID "
          << (void*)detid << " =?= " << (void*)(rpcid) << endreq;
      if ( (rpcid&0xffff0000) == detid) { 
        msg << MSG::DEBUG << "found Detector ID "
            << (void*)detid << " for " << (void*)(rpcid) << endreq;
        return currentDE;    // We have a winner!
      }
    }

    // Check if current detector element is the correct site
    if (params->exists(m_SiteIdUserParameter)) {
      unsigned int siteid = 
        (unsigned int)(params->param<int>(m_SiteIdUserParameter));
      if ( (rpcid&0xff000000) != siteid) {
        msg << MSG::DEBUG << "got Site ID but wrong one "
            << (void*)siteid << " != " << (void*)(rpcid&0xff000000) << endreq;
        return 0;
      }
    }
    
    // Check detector element children
    // We get here we know we are still on the right track
    IDetectorElement::IDEContainer& child = currentDE->childIDetectorElements();
    for (size_t ind=0; ind < child.size(); ++ind) {
        IDetectorElement* parent = this->findParent(rpcid,child[ind]);
        if (parent) return parent;
    }
    msg << MSG::DEBUG 
        << "failed to find parent detector for rpcid "
        << (void*)rpcid << " in current branch" << endreq;
    return 0;    
}


IRpcGeomInfo* RpcGeomInfoSvc::add(unsigned int rpcid, const std::string& path,
                                  IDetectorElement* de)
{
    MsgStream msg(msgSvc(), name());

    if (!de) {
        msg << MSG::ERROR 
            << "Given NULL DetectorElement (rpcid=" 
            << (void*)rpcid << endreq;
        return 0;
    }
    IDetectorElement* parent = findParent(rpcid,0);
    if (!parent) {
        msg << MSG::ERROR 
            << "No parent detector, cannot add DetectorElement with rpcid=" 
            << (void*)rpcid << endreq;
        return 0;
    }

    msg << MSG::DEBUG
        << "Adding rpcid " << (void*)rpcid << " at \"" << path <<"\""
        << endreq;

    RpcGeomInfo* gpi = new RpcGeomInfo(rpcid,de,parent);

    if (path == "") m_byPath[de->name()] = gpi;
    else m_byPath[path] = gpi;
    m_byDE[de] = gpi;
    m_byId[rpcid] = gpi;

    return gpi;
}
StatusCode RpcGeomInfoSvc::queryInterface(const InterfaceID& riid, 
                                          void** ppvInterface)
{
    StatusCode sc = StatusCode::FAILURE;
    if (ppvInterface) {
        *ppvInterface = 0;
    
        if (IRpcGeomInfoSvc::interfaceID().versionMatch(riid)) {
            *ppvInterface = static_cast<IRpcGeomInfoSvc*>(this);
            sc = StatusCode::SUCCESS;
            addRef();
        }
        else sc = Service::queryInterface( riid, ppvInterface );    
    }
    return sc;
}