DMTestRead.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#undef NDEBUG
 
#include "DMTestRead.h"
#include "DataModelAthenaPool/ElementLinkVectorCnv_p1.h"
#include "DataModelAthenaPool/ElementLinkCnv_p3.h"
#include "DataModelAthenaPool/DataLinkCnv_p1.h"
#include "DataModelTestDataRead/BVec.h"
#include "DataModelTestDataRead/BDer.h"
#include "DataModelTestDataRead/DVec.h"
#include "DataModelTestDataRead/DDer.h"
#include "DataModelTestDataRead/ELVec.h"
#include "AthContainers/ClassName.h"
#include "StoreGate/StoreGateSvc.h"
#include "GaudiKernel/MsgStream.h"
#include "AthenaKernel/errorcheck.h"
#include "CxxUtils/checker_macros.h"
#include <iostream>
#include <sstream>
#include <cassert>
 
 
namespace DMTest {
 
 
DMTestRead::DMTestRead (const std::string& name, ISvcLocator* pSvcLocator)
  : AthAlgorithm (name, pSvcLocator)
{
}
 
namespace {
 
 
template <class VEC>
StatusCode print_vec (MsgStream& log,
                      StoreGateSvc* sg,
                      const std::string& key,
                      const std::string& context)
{
  if (!sg->contains<VEC> (key))
  {
    log << MSG::INFO << key << " not in SG; ignored." << endmsg;
    return StatusCode::SUCCESS;
  }
 
  const VEC* vec;
  CHECK_WITH_CONTEXT( sg->retrieve (vec, key), context );
  std::ostringstream ost;
  ost << key << " as " << ClassName<VEC>::name() << ": ";
  for (unsigned i=0; i < vec->size(); i++)
    ost << (*vec)[i]->m_x << " ";
  log << MSG::INFO << ost.str() << endmsg;
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode print_elvec (MsgStream& log,
                        StoreGateSvc* sg,
                        const std::string& key,
                        const std::string& context)
{
  const ELVec* vec;
  CHECK_WITH_CONTEXT( sg->retrieve (vec, key), context );
  std::vector<ElementLink<BVec> > el = vec->m_el;
  std::ostringstream ost;
  ost << key << ": ";
  for (size_t i = 0; i < el.size(); i++) {
    const DMTest::B* b = *el[i];
    el[i].toPersistent();
    ost << b->m_x << " ";
  }
  log << MSG::INFO << ost.str() << endmsg;
  return StatusCode::SUCCESS;
}
 
 
StatusCode remap_test (MsgStream& log, StoreGateSvc* sg)
{
  const ELVec* vec;
  CHECK_WITH_CONTEXT( sg->retrieve (vec, "elv_remap"), "remap_test" );
 
  ElementLinkCnv_p3<ElementLink<BVec> > elcnv;
  std::vector<ElementLink<BVec> > el2; // Transient
  el2.resize (vec->m_el2_p.size());
  for (size_t i=0; i < vec->m_el2_p.size(); i++)
    elcnv.persToTrans (&vec->m_el2_p[i], &el2[i], log);
 
  std::ostringstream ost1;
  ost1 << "elv_remap: ";
  for (size_t i = 0; i < el2.size(); i++) {
    const DMTest::B* b = *el2[i];
    ost1 << b->m_x << " ";
  }
  log << MSG::INFO << ost1.str() << endmsg;
 
  ElementLinkVectorCnv_p1<ElementLinkVector<BVec> > elvcnv;
  ElementLinkVector<BVec> elv2;    // Transient
  elvcnv.persToTrans (&vec->m_elv2_p, &elv2, log);
 
  std::ostringstream ost2;
  ost2 << "elv_remap v2: ";
  for (size_t i = 0; i < elv2.size(); i++) {
    const DMTest::B* b = *elv2[i];
    ost2 << b->m_x << " ";
  }
  log << MSG::INFO << ost2.str() << endmsg;
 
  DataLinkCnv_p1<DataLink<BVec> > dlcnv;
  std::vector<DataLink<BVec> > dl2; // Transient
  dl2.resize (vec->m_dl2_p.size());
  for (size_t i=0; i < vec->m_dl2_p.size(); i++)
    dlcnv.persToTrans (&vec->m_dl2_p[i], &dl2[i], log);
 
  const BVec* b3;
  CHECK_WITH_CONTEXT( sg->retrieve (b3, "b3"), "remap_test" );
  assert (dl2[0].cptr() == b3);
  assert (dl2[1].cptr() == b3);
 
  return StatusCode::SUCCESS;
}
 
 
} // anonymous namespace
 
 
StatusCode DMTestRead::execute()
{
  StoreGateSvc* sg = &*evtStore();
 
  // Test reading our four types.
  CHECK( print_vec<BVec> (msg(), sg, "bvec", name()) );
  CHECK( print_vec<BDer> (msg(), sg, "bder", name()) );
  CHECK( print_vec<DVec> (msg(), sg, "dvec", name()) );
  CHECK( print_vec<DDer> (msg(), sg, "dder", name()) );
 
  // Test using implicit symlinks.
  CHECK( print_vec<BVec> (msg(), sg, "bder", name()) );
  CHECK( print_vec<BVec> (msg(), sg, "dvec", name()) );
  CHECK( print_vec<BVec> (msg(), sg, "dder", name()) );
  CHECK( print_vec<DVec> (msg(), sg, "dder", name()) );
 
  CHECK( print_elvec (msg(), sg, "elvec", name()) );
 
  CHECK( remap_test (msg(), sg) );
 
  return StatusCode::SUCCESS;
}
 
} // namespace DMTest