xAODTestShallowCopyHVec.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2020 CERN for the benefit of the ATLAS collaboration
*/
#include "xAODTestShallowCopyHVec.h"
#include "DataModelTestDataRead/HVec.h"
#include "DataModelTestDataRead/H.h"
#include "DataModelTestDataRead/HAuxContainer.h"
#include "xAODCore/ShallowCopy.h"
#include "StoreGate/ReadHandle.h"
#include "StoreGate/WriteHandle.h"
#include "StoreGate/WriteDecorHandle.h"
#include "AthContainers/AuxTypeRegistry.h"
#include "AthenaKernel/errorcheck.h"
 
 
namespace DMTest {
 
 
StatusCode xAODTestShallowCopyHVec::initialize()
{
  ATH_CHECK( m_readKey.initialize() );
  ATH_CHECK( m_writeKey.initialize() );
  ATH_CHECK( m_aFloat20Key.initialize() );
  return StatusCode::SUCCESS;
}
 
 
StatusCode xAODTestShallowCopyHVec::execute (const EventContext& ctx) const
{
  size_t count = ctx.evt() + 1;
 
  {
    SG::ReadHandle<DMTest::HVec> vec (m_readKey, ctx);
    auto ret = xAOD::shallowCopyContainer (*vec);
 
    SG::WriteHandle<DMTest::HVec> copy (m_writeKey, ctx);
    ATH_CHECK( copy.record (std::unique_ptr<DMTest::HVec>(ret.first),
                            std::unique_ptr<xAOD::ShallowAuxContainer>(ret.second)) );
 
    SG::WriteDecorHandle<DMTest::HVec, float> aFloat20 (m_aFloat20Key, ctx);
    for (H* h : *ret.first)
      aFloat20(*h) = count * 20000 + h->aFloat();
  }
 
  return StatusCode::SUCCESS;
}
 
 
} // namespace DMTest