xAODTestWriteHVec.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
#include "xAODTestWriteHVec.h"
#include "xAODTestWriteHelper.h"
#include "DataModelTestDataWrite/HVec.h"
#include "DataModelTestDataWrite/HView.h"
#include "DataModelTestDataWrite/H.h"
#include "DataModelTestDataWrite/HAuxContainer.h"
 
 
namespace DMTest {
 
 
StatusCode xAODTestWriteHVec::initialize()
{
  ATH_CHECK( m_hvecKey.initialize() );
  ATH_CHECK( m_hviewKey.initialize( SG::AllowEmpty ) );
  return StatusCode::SUCCESS;
}
 
 
StatusCode xAODTestWriteHVec::execute (const EventContext& ctx) const
{
  unsigned int count = ctx.eventID().event_number() + 1;
 
  auto hvec = std::make_unique<DMTest::HVec>();
  auto store = std::make_unique<DMTest::HAuxContainer>();
  hvec->setStore (store.get());
 
  for (int i = 0; i < 20; i++) {
    hvec->push_back (new DMTest::H);
    hvec->back()->setAnInt (i+1 + count * 400);
  }
 
  std::unique_ptr<DMTest::HView> hview;
  if (!m_hviewKey.empty()) {
    hview = std::make_unique<DMTest::HView>();
    for (int i = 0; i < 20; i++) {
      hview->push_back (hvec->at (19-i));
    }
  }
 
  ATH_CHECK( SG::makeHandle(m_hvecKey, ctx).record (std::move(hvec),
                                                    std::move(store)) );
 
  if (!m_hviewKey.empty()) {
    SG::WriteHandle<DMTest::HView> hviewH (m_hviewKey, ctx);
    ATH_CHECK( DMTest::recordView2 (hviewH, std::move(hview)) );
  }
 
  return StatusCode::SUCCESS;
}
 
 
} // namespace DMTest