SCT_RawDataContainerCnv_p4.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
#include "SCT_RawDataContainerCnv_p4.h"
 
 
#include "MsgUtil.h"
#include "SCT1_RawDataCnv_p2.h"
#include "SCT3_RawDataCnv_p4.h"
 
#include "InDetIdentifier/SCT_ID.h"
#include "InDetEventAthenaPool/SCT3_RawData_p4.h"
#include "InDetEventAthenaPool/InDetRawDataCollection_p1.h"
#include "InDetRawData/SCT_RDO_Container.h"
#include "InDetRawData/SCT_RDO_Collection.h"
#include "InDetRawData/SCT3_RawData.h"
 
#include <memory>
#include "AthAllocators/DataPool.h"
 
void SCT_RawDataContainerCnv_p4::transToPers(const SCT_RDO_Container* transCont, SCT_RawDataContainer_p4* persCont, MsgStream& log)
{
 
 
  using TRANS = SCT_RDO_Container;
 
  SCT1_RawDataCnv_p2  chan1Cnv;
  SCT3_RawDataCnv_p4  chan3Cnv(m_sctId);
  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();
  persCont->m_collections.resize(numColl);
 
  for (collIndex = 0; it_Coll != it_CollEnd; ++collIndex, ++it_Coll)  {
    const SCT_RDO_Collection& collection = (**it_Coll);
    chanBegin  = chanEnd;
    chanEnd   += collection.size();
    InDetRawDataCollection_p1& pcollection = persCont->m_collections[collIndex];
    pcollection.m_id    = collection.identify().get_identifier32().get_compact();
    pcollection.m_hashId = static_cast<unsigned int>(collection.identifyHash());
    pcollection.m_begin = chanBegin;
    pcollection.m_end   = chanEnd;
    if (m_type == 1) {
      persCont->m_rawdata.resize(chanEnd);
      for (unsigned int i = 0; i < collection.size(); ++i) {
        InDetRawData_p2* pchan = &(persCont->m_rawdata[i + chanBegin]);
        const SCT1_RawData* chan = dynamic_cast<const SCT1_RawData*>(collection[i]);
        if (nullptr == chan) throw std::runtime_error("SCT_RawDataContainerCnv_p4::transToPers: *** UNABLE TO DYNAMIC CAST TO SCT1_RawData");
        chan1Cnv.transToPers(chan, pchan, log);
      }
    } else if (m_type == 3) {
      persCont->m_sct3data.resize(chanEnd);
      for (unsigned int i = 0; i < collection.size(); ++i) {
        SCT3_RawData_p4* pchan = &(persCont->m_sct3data[i + chanBegin]);
        const SCT3_RawData* chan = dynamic_cast<const SCT3_RawData*>(collection[i]);
        if (nullptr == chan) throw std::runtime_error("SCT_RawDataContainerCnv_p4::transToPers: *** UNABLE TO DYNAMIC CAST TO SCT3_RawData");
        chan3Cnv.transToPers(chan, pchan, log);
      }
    }
  }
}
 
void  SCT_RawDataContainerCnv_p4::persToTrans(const SCT_RawDataContainer_p4* persCont, SCT_RDO_Container* transCont, MsgStream& log)
{
 
  SCT1_RawDataCnv_p2 chan1Cnv;
  SCT3_RawDataCnv_p4 chan3Cnv(m_sctId);
  if (persCont->m_rawdata.size() != 0 && persCont->m_sct3data.size() != 0) {
    log << MSG::FATAL
        << "The collection has mixed SCT1 and SCT3 elements, this is not "
           "allowed "
        << endmsg;
  }
  if (persCont->m_rawdata.size() != 0){
    m_type = 1;
  }
  if (persCont->m_sct3data.size() != 0){
    m_type = 3;
  }
 
  const size_t numCollections = persCont->m_collections.size();
  std::vector<size_t> chans_per_collection{};
  chans_per_collection.reserve(numCollections);
  size_t totalChannels = 0;
  for (unsigned int icoll = 0; icoll < numCollections; ++icoll) {
    const InDetRawDataCollection_p1& pcoll = persCont->m_collections[icoll];
    unsigned int nchans = pcoll.m_end - pcoll.m_begin;
    chans_per_collection.push_back(nchans);
    totalChannels += nchans;
  }
 
  if (m_type == 1) {
    DataPool<SCT1_RawData> dataItems;
    dataItems.reserve(totalChannels);
    for (unsigned int icoll = 0; icoll < numCollections; ++icoll) {
      const InDetRawDataCollection_p1& pcoll = persCont->m_collections[icoll];
      Identifier collID(pcoll.m_id);
      chan3Cnv.setWaferId(collID);
      IdentifierHash collIDHash(pcoll.m_hashId);
      //
      std::unique_ptr<SCT_RDO_Collection> coll =
          std::make_unique<SCT_RDO_Collection>(IdentifierHash(collIDHash));
      coll->setIdentifier(collID);
      coll->clear(SG::VIEW_ELEMENTS);
      unsigned int nchans = chans_per_collection[icoll];
      coll->resize(nchans);
      //
      for (unsigned int ichan = 0; ichan < nchans; ++ichan) {
        const InDetRawData_p2* pchan =
            &(persCont->m_rawdata[ichan + pcoll.m_begin]);
        SCT1_RawData* chan = dataItems.nextElementPtr();
        chan1Cnv.persToTrans(pchan, chan, log);
        (*coll)[ichan] = chan;
      }
      StatusCode sc = transCont->addCollection(coll.release(), collIDHash);
      if (sc.isFailure()) {
        throw std::runtime_error("Failed to add collection to ID Container");
      }
    }
  } // type 1
  else if (m_type == 3) {
    DataPool<SCT3_RawData> dataItems;
    dataItems.prepareToAdd(totalChannels);
    for (unsigned int icoll = 0; icoll < numCollections; ++icoll) {
      const InDetRawDataCollection_p1& pcoll = persCont->m_collections[icoll];
      Identifier collID(pcoll.m_id);
      chan3Cnv.setWaferId(collID);
      IdentifierHash collIDHash(pcoll.m_hashId);
      //
      std::unique_ptr<SCT_RDO_Collection> coll =
          std::make_unique<SCT_RDO_Collection>(IdentifierHash(collIDHash));
      coll->setIdentifier(collID);
      coll->clear(SG::VIEW_ELEMENTS);
      unsigned int nchans = chans_per_collection[icoll];
      coll->resize(nchans);
      //
      for (unsigned int ichan = 0; ichan < nchans; ++ichan) {
        const SCT3_RawData_p4* pchan =
            &(persCont->m_sct3data[ichan + pcoll.m_begin]);
        SCT3_RawData* chan = dataItems.nextElementPtr();
        chan3Cnv.persToTrans(pchan, chan, log);
        (*coll)[ichan] = chan;
      }
      StatusCode sc = transCont->addCollection(coll.release(), collIDHash);
      if (sc.isFailure()) {
        throw std::runtime_error("Failed to add collection to ID Container");
      }
    }
  }  // type 3
}
 
//================================================================
SCT_RDO_Container* SCT_RawDataContainerCnv_p4::createTransient(const SCT_RawDataContainer_p4* persObj, MsgStream& log) {
  std::unique_ptr<SCT_RDO_Container> trans(std::make_unique<SCT_RDO_Container>(m_sctId->wafer_hash_max()));
  persToTrans(persObj, trans.get(), log);
  return trans.release();
}