xAODTestWriteCInfo.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
// $Id$
#include "xAODTestWriteCInfo.h"
#include "DataModelTestDataCommon/C.h"
#include "DataModelTestDataCommon/CInfoAuxContainer.h"
#include "AthLinks/ElementLink.h"
 
 
namespace DMTest {
 
 
xAODTestWriteCInfo::xAODTestWriteCInfo (const std::string &name,
                                        ISvcLocator *pSvcLocator)
  : AthReentrantAlgorithm (name, pSvcLocator),
    m_cvecKey ("cvec"),
    m_cinfoKey ("cinfo")
{
  declareProperty ("Offset", m_offset = 0);
  declareProperty ("CVecKey", m_cvecKey);
  declareProperty ("CInfoKey", m_cinfoKey);
}
  
 
StatusCode xAODTestWriteCInfo::initialize()
{
  ATH_CHECK( m_eventInfoKey.initialize() );
  ATH_CHECK( m_cvecKey.initialize() );
  ATH_CHECK( m_cinfoKey.initialize() );
  return StatusCode::SUCCESS;
}
 
 
StatusCode xAODTestWriteCInfo::execute (const EventContext& ctx) const
{
  SG::ReadHandle<xAOD::EventInfo> eventInfo (m_eventInfoKey, ctx);
  unsigned int count = eventInfo->eventNumber()  + 1;
 
  SG::ReadHandle<DMTest::CVec> cvec (m_cvecKey, ctx);
 
  auto cinfo = std::make_unique<DMTest::C>();
  auto info_store = std::make_unique<DMTest::CInfoAuxContainer>();
  cinfo->setStore (info_store.get());
 
  static const C::Accessor<int> anInt2 ("anInt2");
  static const C::Accessor<ElementLink<DMTest::CVec> > cEL ("cEL");
 
  cinfo->setAnInt (count * 1000 + m_offset);
  cinfo->setAFloat ((float)count * 0.1 + m_offset);
  anInt2(*cinfo) = count * 2000 + m_offset;
 
  if (!cvec->empty()) {
    cEL(*cinfo).toIndexedElement (*cvec, count % cvec->size());
  }
 
  SG::WriteHandle<DMTest::C> cinfoH (m_cinfoKey, ctx);
  std::unique_ptr<const DMTest::C> cinfo_c (std::move (cinfo));
  std::unique_ptr<const DMTest::CInfoAuxContainer> info_store_c (std::move (info_store));
  if (cinfoH.put (std::move(cinfo_c), std::move(info_store_c)) == nullptr) {
    ATH_MSG_ERROR ("Can't record CInfo");
    return StatusCode::FAILURE;
  }
 
  return StatusCode::SUCCESS;
}
 
 
StatusCode xAODTestWriteCInfo::finalize()
{
  return StatusCode::SUCCESS;
}
 
 
} // namespace DMTest