ROBDataProviderSvc.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BYTESTREAMCNVSVCBASE_ROBDATAPROVIDERSVC_H
#define BYTESTREAMCNVSVCBASE_ROBDATAPROVIDERSVC_H
 
#include "ByteStreamCnvSvcBase/IROBDataProviderSvc.h"
 
#include "AthenaBaseComps/AthService.h"
#include "AthenaKernel/SlotSpecificObj.h"
#include "ByteStreamData/RawEvent.h"
 
#include "Gaudi/Property.h"
#include "eformat/SourceIdentifier.h"
 
#include <vector>
#include <memory>
#include <unordered_map>

class ROBDataProviderSvc : public extends<AthService, IROBDataProviderSvc> {
 
public:
   using ROBF = OFFLINE_FRAGMENTS_NAMESPACE::ROBFragment;

   using base_class::base_class;

   virtual StatusCode initialize() override;

   virtual void setNextEvent(const EventContext& ctx, const RawEvent* re) override;

   virtual void getROBData(const EventContext& ctx, const std::vector<uint32_t>& robIds,
                           VROBFRAG& robFragments,
                           const std::string_view callerName="UNKNOWN") override;

   virtual const RawEvent* getEvent(const EventContext& ctx) override;

   virtual void setEventStatus(const EventContext& ctx, uint32_t status) override;

   virtual uint32_t getEventStatus(const EventContext& ctx) override;
 
   virtual void processCachedROBs(const EventContext& ctx,
                                  const std::function< void(const ROBF* )>& fn ) const override;
 
private:
   bool filterRobWithStatus(const ROBF* rob);

   struct EventCache {
      const RawEvent* event = nullptr;
      uint32_t eventStatus = 0;
      uint32_t currentLvl1ID = 0;
      std::unordered_map<uint32_t, std::unique_ptr<const ROBF>> robmap;
   };
   SG::SlotSpecificObj<EventCache> m_eventsCache;

   std::unordered_map<uint32_t, std::vector<uint32_t>> m_filterRobMap;

   std::unordered_map<eformat::SubDetector, std::vector<uint32_t>> m_filterSubDetMap;
 
   // Properties
   Gaudi::Property<std::vector<std::pair<int, int>>> m_filterRobWithStatus{
      this, "filterRobWithStatus", {}, "ROB IDs and status words to filter (full ROB SourceID)"};
   Gaudi::Property<std::vector<std::pair<int, int>>> m_filterSubDetWithStatus{
      this, "filterSubDetWithStatus", {}, "Sub-detector IDs and status words to filter"};
   Gaudi::Property<bool> m_filterEmptyROB{
      this, "filterEmptyROB", false, "Filter out empty ROB fragments"};
};
 
#endif