MCEventInfoByteStreamTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BYTESTREAMCNVSVC_MCEVENTINFOBYTESTREAMTOOL_H
#define BYTESTREAMCNVSVC_MCEVENTINFOBYTESTREAMTOOL_H

 
#include "AthenaBaseComps/AthAlgTool.h"
#include "ByteStreamData/RawEvent.h"
#include "xAODEventInfo/EventInfo.h"
#include "StoreGate/ReadHandleKey.h"
#include "StoreGate/WriteHandleKey.h"
#include "AthenaKernel/SlotSpecificObj.h"
#include "Gaudi/Property.h"
 
#include <vector>
#include <memory>
#include <cstdint>

class IMCEventInfoByteStreamTool : virtual public IAlgTool {
public:
  DeclareInterfaceID(IMCEventInfoByteStreamTool, 1, 0);
  virtual ~IMCEventInfoByteStreamTool() override = default;

  virtual StatusCode convertToBS(std::vector<OFFLINE_FRAGMENTS_NAMESPACE_WRITE::ROBFragment*>& vrobf,
                                 const EventContext& eventContext) = 0;

  virtual StatusCode convertFromBS(const OFFLINE_FRAGMENTS_NAMESPACE::ROBFragment* rob,
                                   xAOD::EventInfo& evtInfo) const = 0;

  virtual const std::vector<uint32_t>& robIds() const = 0;
};

class MCEventInfoByteStreamTool : public extends<AthAlgTool, IMCEventInfoByteStreamTool> {
public:
  MCEventInfoByteStreamTool(const std::string& type, const std::string& name, const IInterface* parent);
  virtual ~MCEventInfoByteStreamTool() override = default;
 
  virtual StatusCode initialize() override;

  virtual StatusCode convertToBS(std::vector<OFFLINE_FRAGMENTS_NAMESPACE_WRITE::ROBFragment*>& vrobf,
                                 const EventContext& eventContext) override;

  virtual StatusCode convertFromBS(const OFFLINE_FRAGMENTS_NAMESPACE::ROBFragment* rob,
                                   xAOD::EventInfo& evtInfo) const override;

  virtual const std::vector<uint32_t>& robIds() const override { return m_robIds.value(); }

  static constexpr uint32_t MC_EVENTINFO_ROB_ID = 0x00ff0001;

  static constexpr uint32_t FORMAT_VERSION = 0x00010000;  // v1.0

  static constexpr uint32_t MIN_DATA_WORDS = 14;
 
private:
  void clearCache(const EventContext& eventContext);

  uint32_t* newRodData(const EventContext& eventContext, size_t size);

  OFFLINE_FRAGMENTS_NAMESPACE_WRITE::ROBFragment* newRobFragment(
    const EventContext& eventContext,
    uint32_t source_id,
    uint32_t ndata,
    const uint32_t* data);
 
  // Properties
  Gaudi::Property<std::vector<uint32_t>> m_robIds {
    this, "ROBIDs", {MC_EVENTINFO_ROB_ID}, "List of ROB IDs for MC EventInfo"};
 
  SG::ReadHandleKey<xAOD::EventInfo> m_eventInfoReadKey {
    this, "EventInfoReadKey", "", "Read handle key for EventInfo (encoding mode)"};
 
  // Cache for ByteStream data representation
  struct Cache {
    std::vector<std::unique_ptr<uint32_t[]>> rodData;
    std::vector<std::unique_ptr<OFFLINE_FRAGMENTS_NAMESPACE_WRITE::ROBFragment>> robFragments;
    ~Cache() { clear(); }
    void clear() {
      rodData.clear();
      robFragments.clear();
    }
  };
  SG::SlotSpecificObj<Cache> m_cache;
};
 
#endif // BYTESTREAMCNVSVC_MCEVENTINFOBYTESTREAMTOOL_H