da/d83/xAODTestWriteCVec_8cxx_source.html

/*

 * Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration.

 */


#include "xAODTestWriteCVec.h"

#include "DataModelTestDataCommon/CVec.h"

#include "DataModelTestDataCommon/C.h"

#include "DataModelTestDataCommon/CAuxContainer.h"

#include "AthContainersInterfaces/AuxDataOption.h"

#include "AthContainers/Accessor.h"

#include "AthContainers/Decorator.h"

#include "StoreGate/WriteDecorHandle.h"

#include "AthLinks/ElementLink.h"


namespace DMTest {


xAODTestWriteCVec::xAODTestWriteCVec (const std::string &name,

                                      ISvcLocator *pSvcLocator)

  : AthReentrantAlgorithm (name, pSvcLocator)

{

}


StatusCode xAODTestWriteCVec::initialize()

{

  ATH_CHECK( m_cvecKey.initialize() );


  if (m_cvecDecorKey.key().empty())

    m_cvecDecorKey = m_cvecKey.key() + ".dtest";

  ATH_CHECK( m_cvecDecorKey.initialize() );


  return StatusCode::SUCCESS;

}


StatusCode xAODTestWriteCVec::execute (const EventContext& ctx) const

{

  unsigned int count = ctx.eventID().event_number() + 1;


  auto coll = std::make_unique<DMTest::CVec>();

  auto store = std::make_unique<DMTest::CAuxContainer>();

  coll->setStore (store.get());


  const static SG::Accessor<int> anInt2 ("anInt2");

  const static SG::Decorator<int> dVar1 ("dVar1");

  const static SG::Decorator<int> dVar2 ("dVar2");

  const static SG::Accessor<ElementLink<DMTest::CVec> > cEL ("cEL");


  int nent = 10;

  if (count == 5) nent = 0;

  for (int i=0; i < nent; i++) {

    coll->push_back (new DMTest::C);

    C& c = *coll->back();

    c.setAnInt (count * 100 + i+1);

    c.setAFloat (count * 200 + (float)i*0.1);


    anInt2(c) = count*300 + i+1;

    dVar1(c) = count*450 + i+1;

    dVar2(c) = count*750 + i+1;

    cEL(c).toIndexedElement (*coll, 9-i);

  }


  SG::WriteHandle<DMTest::CVec> cvec (m_cvecKey, ctx);

  CHECK( cvec.record (std::move(coll), std::move(store)) );


  SG::WriteDecorHandle<DMTest::CVec, int> dtest (m_cvecDecorKey, ctx);

  for (int i=0; i < nent; i++) {

    dtest (*(*dtest)[i]) = i+123;

  }


  return StatusCode::SUCCESS;

}


} // namespace DMTest


ATH_CHECK
#define ATH_CHECK
Evaluate an expression and check for errors.
Definition AthCheckMacros.h:40

AuxDataOption.h
Hold information about an option setting request.

CAuxContainer.h
Class used for testing xAOD data reading/writing.

CVec.h
Class used for testing xAOD data reading/writing.

C.h
Class used for testing xAOD data reading/writing.

Accessor.h
Helper class to provide type-safe access to aux data.

CHECK
#define CHECK(...)
Evaluate an expression and check for errors.
Definition Control/AthenaKernel/AthenaKernel/errorcheck.h:422

Decorator.h
Helper class to provide type-safe access to aux data.

WriteDecorHandle.h
Handle class for adding a decoration to an object.

AthReentrantAlgorithm
An algorithm that can be simultaneously executed in multiple threads.
Definition AthReentrantAlgorithm.h:74

DMTest::xAODTestWriteCVec::m_cvecDecorKey
SG::WriteDecorHandleKey< DMTest::CVec > m_cvecDecorKey
Definition xAODTestWriteCVec.h:58

DMTest::xAODTestWriteCVec::m_cvecKey
SG::WriteHandleKey< DMTest::CVec > m_cvecKey
Definition xAODTestWriteCVec.h:56

DMTest::xAODTestWriteCVec::execute
virtual StatusCode execute(const EventContext &ctx) const override
Algorithm event processing.
Definition xAODTestWriteCVec.cxx:56

DMTest::xAODTestWriteCVec::initialize
virtual StatusCode initialize() override
Algorithm initialization; called at the beginning of the job.
Definition xAODTestWriteCVec.cxx:41

DMTest::xAODTestWriteCVec::xAODTestWriteCVec
xAODTestWriteCVec(const std::string &name, ISvcLocator *pSvcLocator)
Constructor.
Definition xAODTestWriteCVec.cxx:31

SG::Accessor
Helper class to provide type-safe access to aux data.
Definition Control/AthContainers/AthContainers/Accessor.h:68

SG::Decorator
Helper class to provide type-safe access to aux data.
Definition Decorator.h:59

SG::WriteDecorHandle
Handle class for adding a decoration to an object.
Definition StoreGate/StoreGate/WriteDecorHandle.h:100

SG::WriteHandle
Definition StoreGate/StoreGate/WriteHandle.h:73

SG::WriteHandle::record
StatusCode record(std::unique_ptr< T > data)
Record a const object to the store.

count
int count(std::string s, const std::string &regx)
count how many occurances of a regx are in a string
Definition hcg.cxx:146

DMTest
Definition B.h:23

DMTest::C
C_v1 C
Definition C.h:26

xAODTestWriteCVec.h
Algorithm to test writing xAOD classes with auxiliary data (cvec).