WriteTag.cxx

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#include "WriteTag.h"
 
#include "AthenaPoolUtilities/TagAthenaAttributeList.h"
#include "PersistentDataModel/AthenaAttributeList.h"
#include "AthenaPoolUtilities/AthenaAttributeListSpecification.h"
#include "StoreGate/WriteHandle.h"
 
using namespace AthPoolEx;
 
//___________________________________________________________________________
WriteTag::WriteTag(const std::string& name, ISvcLocator* pSvcLocator)
  : AthReentrantAlgorithm(name, pSvcLocator), m_attribListSpec(0)
{
   declareProperty("Magic", m_magic = 0);
}
//___________________________________________________________________________
WriteTag::~WriteTag() {
   if (m_attribListSpec != 0) {
      m_attribListSpec->release();
   }
}
//___________________________________________________________________________
StatusCode WriteTag::initialize() {
   ATH_MSG_INFO("in initialize()");
 
   ATH_MSG_DEBUG("Defining the attribute list specification.");
   m_attribListSpec = new AthenaAttributeListSpecification();
   m_attribListSpec->extend("RunNumber", "unsigned int");
   m_attribListSpec->extend("EventNumber", "unsigned int");
   if (m_magic > 0) {
      m_attribListSpec->extend("MagicNumber", "unsigned int");
   }
   ATH_MSG_DEBUG("Printing out attribute list specification:");
   for (AthenaAttributeListSpecification::const_iterator first = m_attribListSpec->begin(), last = m_attribListSpec->end(); first != last; ++first) {
      ATH_MSG_DEBUG(" name " << (*first).name() << " type " << (*first).typeName());
   }
 
   ATH_CHECK( m_key.initialize() );
   ATH_CHECK( m_tagKey.initialize() );
   return StatusCode::SUCCESS;
}
//___________________________________________________________________________
StatusCode WriteTag::execute (const EventContext& ctx) const {
   ATH_MSG_DEBUG("in execute()");
 
   unsigned int eventNumber = ctx.eventID().event_number();
   unsigned int runNumber = ctx.eventID().run_number();
   ATH_MSG_INFO("EventInfo event: " << eventNumber << "  run: " << runNumber);
 
   const AthenaAttributeListSpecification* attribListSpec = m_attribListSpec;
   auto tagAttribList = std::make_unique<TagAthenaAttributeList>(*attribListSpec);
   (*tagAttribList)["RunNumber"].data<unsigned int>() = runNumber;
   (*tagAttribList)["EventNumber"].data<unsigned int>() = eventNumber;
   if (m_magic > 0) {
      (*tagAttribList)["MagicNumber"].data<unsigned int>() = m_magic.value();
   }
 
   coral::AttributeListSpecification* cspec = new coral::AttributeListSpecification;
   m_attribListSpec->coralSpec (*cspec);
   auto attribList = std::make_unique<AthenaAttributeList>(*cspec);
   cspec->release();
   (*attribList)["RunNumber"].data<unsigned int>() = runNumber;
   (*attribList)["EventNumber"].data<unsigned int>() = eventNumber;
   if (m_magic > 0) {
      (*attribList)["MagicNumber"].data<unsigned int>() = m_magic.value();
   }
 
   SG::WriteHandle<AthenaAttributeList> attribListH (m_key, ctx);
   SG::WriteHandle<TagAthenaAttributeList> tagAttribListH (m_tagKey, ctx);
   ATH_CHECK( attribListH.record (std::move(attribList)) );
   ATH_CHECK( tagAttribListH.record (std::move(tagAttribList)) );
 
   ATH_MSG_INFO("registered all data");
   return StatusCode::SUCCESS;
}
//___________________________________________________________________________
StatusCode WriteTag::finalize() {
   ATH_MSG_INFO("in finalize()");
   return StatusCode::SUCCESS;
}