xAODTestReadCVec.cxx

Go to the documentation of this file.

/*
 * Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration.
 */
 
 
#include "xAODTestReadCVec.h"
#include "DataModelTestDataCommon/CVec.h"
#include "DataModelTestDataCommon/C.h"
#include "DataModelTestDataCommon/CAuxContainer.h"
#include "StoreGate/ReadHandle.h"
#include "StoreGate/WriteHandle.h"
#include "AthContainers/AuxTypeRegistry.h"
#include "AthContainers/ConstAccessor.h"
#include "AthLinks/ElementLink.h"
#include "AthenaKernel/errorcheck.h"
#include "GaudiKernel/System.h"
#include <memory>
#include <sstream>
 
 
namespace DMTest {
 

StatusCode xAODTestReadCVec::initialize()
{
  errorcheck::ReportMessage::hideErrorLocus();
 
  if (m_testDecorSE) {
    SG::ConstAccessor<float> dVar1 ("dVar1");
  }
 
  ATH_CHECK( m_cvecKey.initialize() );
  if (!m_writeKey.key().empty())
    ATH_CHECK( m_writeKey.initialize() );
  return StatusCode::SUCCESS;
}
 

StatusCode xAODTestReadCVec::execute (const EventContext& ctx) const
{
  SG::ReadHandle<DMTest::CVec> cvec (m_cvecKey, ctx);
 
  if (m_brief) {
    std::ostringstream ost;
    ost << m_cvecKey.key() << " ";
    for (const C* c : *cvec) {
      ost << c->anInt() << " ";
    }
    ATH_MSG_INFO (ost.str());
    return StatusCode::SUCCESS;
  }
 
  const static SG::ConstAccessor<int> anInt2 ("anInt2");
  const static SG::ConstAccessor<int> anInt10 ("anInt10");
  const static SG::ConstAccessor<int> dInt1 ("dInt1");
  const static SG::ConstAccessor<int> dInt100 ("dInt100");
  const static SG::ConstAccessor<int> dInt150 ("dInt150");
  const static SG::ConstAccessor<int> dInt200 ("dInt200");
  const static SG::ConstAccessor<int> dInt250 ("dInt250");
  const static SG::ConstAccessor<ElementLink<DMTest::CVec> > cEL ("cEL");
 
   // testing schema evolution
  const static SG::ConstAccessor<float> dVar1SE (m_testDecorSE ? "dVar1" : "dxVar1");
  const static SG::ConstAccessor<int> dVar1NoSE (m_testDecorSE ? "dxVar1" : "dVar1");
 
  // Ordering of auxid is not reliable.  Sort by name.
  const SG::AuxTypeRegistry& r = SG::AuxTypeRegistry::instance();
  std::vector<std::string> names;
  for (SG::auxid_t auxid : cvec->getAuxIDs())
    names.push_back (r.getName(auxid));
  std::sort (names.begin(), names.end());
  std::ostringstream ost1;
  ost1 << m_cvecKey.key() << " aux items: ";
  for (const std::string& n : names)
    ost1 << n << " ";
  ATH_MSG_INFO (ost1.str());
  {
    const SG::IConstAuxStore* auxstore = cvec->getConstStore();
    ATH_MSG_INFO("Type of aux store: "
                 << System::typeinfoName (typeid (*auxstore)));
  }
 
  for (const C* c : *cvec) {
    std::ostringstream ost;
    ost << " anInt1 " << c->anInt()
        << " aFloat: " << c->aFloat();
    if (anInt2.isAvailable(*c))
      ost << " anInt2: " << anInt2(*c);
    if (dInt1.isAvailable(*c))
      ost << " dInt1: " << dInt1(*c);
    if (m_testDecorSE) {
      if (dVar1SE.isAvailable(*c))
        ost << " dVar1: " << dVar1SE(*c);
    }
    else {
      if (dVar1NoSE.isAvailable(*c))
        ost << " dVar1: " << dVar1NoSE(*c);
    }
    if (dInt100.isAvailable(*c))
      ost << " dInt100: " << dInt100(*c);
    if (dInt150.isAvailable(*c))
      ost << " dInt150: " << dInt150(*c);
    if (dInt200.isAvailable(*c))
      ost << " dInt200: " << dInt200(*c);
    if (dInt250.isAvailable(*c))
      ost << " dInt250: " << dInt250(*c);
    if (anInt10.isAvailable(*c))
      ost << " anInt10: " << anInt10(*c);
    if (cEL.isAvailable(*c))
      ost << " cEL: " << cEL(*c).dataID()
                << "[" << cEL(*c).index() << "]";
    ost << "\n";
    
    ATH_MSG_INFO (ost.str());
  }
 
  if (!m_writeKey.key().empty()) {
    auto vecnew = std::make_unique<CVec>();
    auto store = std::make_unique<CAuxContainer>();
    vecnew->setStore (store.get());
    for (size_t i = 0; i < cvec->size(); i++) {
      vecnew->push_back (new C);
      *vecnew->back() = *(*cvec)[i];
    }
    SG::WriteHandle<DMTest::CVec> writevec (m_writeKey, ctx);
    ATH_CHECK( writevec.record(std::move(vecnew), std::move(store)) );
  }
 
  return StatusCode::SUCCESS;
}
 
 
} // namespace DMTest