TgcCoinDataContainerCnv_p3.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#include "MuonTrigCoinData/TgcCoinData.h"
#include "MuonTrigCoinData/TgcCoinDataContainer.h"
#include "MuonEventTPCnv/MuonTrigCoinData/TgcCoinData_p3.h"
#include "MuonEventTPCnv/MuonTrigCoinData/MuonCoinDataContainer_p1.h"
#include "MuonIdHelpers/TgcIdHelper.h"
#include "MuonReadoutGeometry/MuonDetectorManager.h"
#include "MuonEventTPCnv/MuonTrigCoinData/TgcCoinDataCnv_p3.h"
#include "MuonEventTPCnv/MuonTrigCoinData/TgcCoinDataContainerCnv_p3.h"
 
// Gaudi
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/Bootstrap.h"
#include "GaudiKernel/StatusCode.h"
#include "GaudiKernel/Service.h"
#include "GaudiKernel/MsgStream.h"
 
// Athena
#include "AthenaKernel/errorcheck.h"
#include "AthAllocators/DataPool.h"
#include "StoreGate/StoreGateSvc.h"
 
 
StatusCode Muon::TgcCoinDataContainerCnv_p3::initialize(MsgStream &log) {
   // Do not initialize again:
    m_isInitialized=true;
 
   // get DetectorStore service
    SmartIF<StoreGateSvc> detStore{Gaudi::svcLocator()->service("DetectorStore")};
    CHECK( detStore.isValid() );
 
   // Get the helper from the detector store
    CHECK( detStore->retrieve(m_TgcId) );
 
    CHECK( m_eventCnvTool.retrieve() );
    log << MSG::DEBUG << "Converter initialized." << endmsg;
    return StatusCode::SUCCESS;
}
 
void Muon::TgcCoinDataContainerCnv_p3::transToPers(const Muon::TgcCoinDataContainer* transCont,  Muon::MuonCoinDataContainer_p1* persCont, MsgStream &log) 
{
 
    // The transient model has a container holding collections and the
    // collections hold channels.
    //
    // The persistent model flattens this so that the persistent
    // container has two vectors:
    //   1) all collections, and
    //   2) all RDO
    //
    // The persistent collections, then only maintain indexes into the
    // container's vector of all channels. 
    //
    // So here we loop over all collection and add their channels
    // to the container's vector, saving the indexes in the
    // collection. 
 
    typedef Muon::TgcCoinDataContainer TRANS;
    typedef ITPConverterFor<Muon::TgcCoinData> CONV;
 
    TgcCoinDataCnv_p3  chanCnv;
    TRANS::const_iterator it_Coll     = transCont->begin();
    TRANS::const_iterator it_CollEnd  = transCont->end();
    unsigned int collIndex;
    unsigned int chanBegin = 0;
    unsigned int chanEnd = 0;
    int numColl = transCont->numberOfCollections();
    // if(numColl == transCont->hashFunc().max() ) { // let's count how many collections we have:
    //  numColl = 0;
    //  for ( ; it_Coll != it_CollEnd; it_Coll++)
    //     numColl++;
    //  it_Coll     = transCont->begin(); // reset the iterator, we used it!
    // }
    persCont->m_collections.resize(numColl);    log << MSG::DEBUG  << " Preparing " << persCont->m_collections.size() << "Collections" << endmsg;
 
    for (collIndex = 0; it_Coll != it_CollEnd; ++collIndex, ++it_Coll)  {
        // Add in new collection
        log << MSG::DEBUG  << " New collection" << endmsg;
        const Muon::TgcCoinDataCollection& collection = (**it_Coll);
        chanBegin  = chanEnd;
        chanEnd   += collection.size();
        Muon::MuonCoinDataCollection_p1& pcollection = persCont->m_collections[collIndex];
        pcollection.m_id    = collection.identify().get_identifier32().get_compact();
        pcollection.m_hashId = (unsigned int) collection.identifyHash();
        pcollection.m_begin = chanBegin;
        pcollection.m_end   = chanEnd;
        // Add in channels
        persCont->m_CoinData.resize(chanEnd);
        for (unsigned int i = 0; i < collection.size(); ++i) {
            const Muon::TgcCoinData* chan = collection[i];
            persCont->m_CoinData[i + chanBegin] = toPersistent((CONV**)nullptr, chan, log );
        }
    }
    log << MSG::DEBUG  << " ***  Writing TgcCoinDataContainer ***" << endmsg;
}
 
void  Muon::TgcCoinDataContainerCnv_p3::persToTrans(const Muon::MuonCoinDataContainer_p1* persCont, Muon::TgcCoinDataContainer* transCont, MsgStream &log) 
{
 
    // The transient model has a container holding collections and the
    // collections hold channels.
    //
    // The persistent model flattens this so that the persistent
    // container has two vectors:
    //   1) all collections, and
    //   2) all channels
    //
    // The persistent collections, then only maintain indexes into the
    // container's vector of all channels. 
    //
    // So here we loop over all collection and extract their channels
    // from the vector.
 
 
    Muon::TgcCoinDataCollection* coll = nullptr;
 
    TgcCoinDataCnv_p3  chanCnv;
    typedef ITPConverterFor<Muon::TgcCoinData> CONV;
 
    log << MSG::DEBUG  << " Reading " << persCont->m_collections.size() << "Collections" << endmsg;
    for (unsigned int icoll = 0; icoll < persCont->m_collections.size(); ++icoll) {
 
        // Create trans collection - is NOT owner of TgcCoinData (SG::VIEW_ELEMENTS)
    // IDet collection don't have the Ownership policy c'tor
        const Muon::MuonCoinDataCollection_p1& pcoll = persCont->m_collections[icoll];        
        Identifier collID(Identifier(pcoll.m_id));
        IdentifierHash collIDHash(IdentifierHash(pcoll.m_hashId));
        coll = new Muon::TgcCoinDataCollection(collIDHash);
        coll->setIdentifier(Identifier(pcoll.m_id));
        unsigned int nchans           = pcoll.m_end - pcoll.m_begin;
        coll->resize(nchans);
 
        // Fill with channels
        for (unsigned int ichan = 0; ichan < nchans; ++ ichan) {
            const TPObjRef pchan = persCont->m_CoinData[ichan + pcoll.m_begin];
            Muon::TgcCoinData* chan = dynamic_cast<Muon::TgcCoinData*>(createTransFromPStore((CONV**)nullptr, pchan, log ) );
            if( (chan->type()!=Muon::TgcCoinData::TYPE_TRACKLET_EIFI && (!chan->isInner() )) && (chan->type()!=Muon::TgcCoinData::TYPE_UNKNOWN) ) {
              const MuonGM::TgcReadoutElement * deOut =  getReadOutElement(Identifier(chan->channelIdOut()));
              chan->m_detElOut = deOut;
            }
            if((chan->type()==Muon::TgcCoinData::TYPE_TRACKLET || chan->type()==Muon::TgcCoinData::TYPE_HIPT || 
                chan->type()==Muon::TgcCoinData::TYPE_TRACKLET_EIFI)&& !chan->isInner() ) {
              const MuonGM::TgcReadoutElement * deIn =   getReadOutElement(Identifier(chan->channelIdIn()));
              chan->m_detElIn = deIn;
            }
            else {
              chan->m_detElIn = nullptr;
            }
            (*coll)[ichan] = chan;
        }
 
        // register the rdo collection in IDC with hash - faster addCollection
        StatusCode sc = transCont->addCollection(coll, collIDHash);
        if (sc.isFailure()) {
            throw std::runtime_error("Failed to add collection to ID Container");
        }
        if (log.level() <= MSG::DEBUG) {
            log << MSG::DEBUG << "AthenaPoolTPCnvIDCont::persToTrans, collection, hash_id/coll id = " << (int) collIDHash << " / " << 
                collID.get_compact() << ", added to Identifiable container." << endmsg;
        }
    }
 
    log << MSG::DEBUG  << " ***  Reading TgcCoinDataContainer" << endmsg;
}
 
 
 
//================================================================
Muon::TgcCoinDataContainer* Muon::TgcCoinDataContainerCnv_p3::createTransient(const Muon::MuonCoinDataContainer_p1* persObj, MsgStream& log) 
{
    if(!m_isInitialized) {
        if (this->initialize(log) != StatusCode::SUCCESS) {
            log << MSG::FATAL << "Could not initialize TgcCoinDataContainerCnv_p3 " << endmsg;
            return nullptr;
        } 
    }
    std::unique_ptr<Muon::TgcCoinDataContainer> trans(new Muon::TgcCoinDataContainer(m_TgcId->module_hash_max()));
    persToTrans(persObj, trans.get(), log);
    return(trans.release());
}
 
 
const MuonGM::TgcReadoutElement* Muon::TgcCoinDataContainerCnv_p3::getReadOutElement(const Identifier& id ) const {
    const Trk::ITrkEventCnvTool* cnv_tool = m_eventCnvTool->getCnvTool(id);
    if (!cnv_tool) return nullptr; 
    return dynamic_cast<const MuonGM::TgcReadoutElement*>(cnv_tool->getDetectorElement(id));
}