xAODTestReadPVec.cxx

Go to the documentation of this file.

/*
 * Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration.
 */
#include "xAODTestReadPVec.h"
#include "DataModelTestDataCommon/PVec.h"
#include "DataModelTestDataCommon/P.h"
#include "DataModelTestDataCommon/PAuxContainer.h"
#include "StoreGate/ReadHandle.h"
#include "StoreGate/WriteHandle.h"
#include "AthContainers/ConstAccessor.h"
#include <memory>
#include <sstream>
#include <format>
 
 
namespace DMTest {
 
 
StatusCode xAODTestReadPVec::initialize()
{
  ATH_CHECK( m_pvecKey.initialize() );
  if (!m_writeKey.key().empty())
    ATH_CHECK( m_writeKey.initialize() );
  return StatusCode::SUCCESS;
}
 
 
StatusCode xAODTestReadPVec::execute (const EventContext& ctx) const
{
  SG::ReadHandle<DMTest::PVec> pvec (m_pvecKey, ctx);
 
  const static SG::ConstAccessor<unsigned int> dpInt1 ("dpInt1");
  const static SG::ConstAccessor<std::vector<float> > dpvFloat ("dpvFloat");
 
  for (const P* p : *pvec) {
    std::ostringstream ost;
    ost << " pInt: " << p->pInt()
        << " pFloat: " << std::format ("{:.2f}", p->pFloat());
    if (dpInt1.isAvailable(*p))
      ost << " dpInt1: " << dpInt1(*p);
    ost << "\n";
    
    {
      const std::vector<int>& pvi = p->pvInt();
      ost << "  pvInt: [";
      for (auto ii : pvi)
        ost << ii << " ";
      ost << "]\n";
    }
 
    {
      const std::vector<float>& pvf = p->pvFloat();
      ost << "  pvFloat: [";
      for (auto f : pvf)
        ost << std::format ("{:.3f}", f) << " ";
      ost << "]\n";
    }
 
    if (dpvFloat.isAvailable(*p))
    {
      const std::vector<float>& pvf = dpvFloat(*p);
      ost << "  dpvFloat: [";
      for (auto f : pvf)
        ost << std::format ("{:.3f}", f) << " ";
      ost << "]\n";
    }
 
    ATH_MSG_INFO (ost.str());
  }
 
  if (!m_writeKey.key().empty()) {
    auto vecnew = std::make_unique<PVec>();
    auto store = std::make_unique<PAuxContainer>();
    vecnew->setStore (store.get());
    for (size_t i = 0; i < pvec->size(); i++) {
      vecnew->push_back (new P);
      *vecnew->back() = *(*pvec)[i];
    }
    SG::WriteHandle<DMTest::PVec> writevec (m_writeKey, ctx);
    ATH_CHECK( writevec.record(std::move(vecnew), std::move(store)) );
  }
 
  return StatusCode::SUCCESS;
}
 
 
} // namespace DMTest