RNTCollectionCursor.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef RNTCOLLECTIONCURSOR_H
#define RNTCOLLECTIONCURSOR_H
 
 
#include "PersistentDataModel/Token.h"
 
#include "CollectionBase/CollectionRowBuffer.h"
#include "CollectionBase/ICollectionDescription.h"
#include "CollectionBase/ICollectionCursor.h"
 
#include "PersistencySvc/IPositionSeek.h"
#include "AthenaKernel/ICollectionSize.h"
 
#include <memory>
#include <map>
 
namespace ROOT::Experimental {
   class RNTupleReader;
   class REntry;
}
 
namespace pool {
   namespace RootCollection {
 
      using ROOT::Experimental::RNTupleReader;
      using ROOT::Experimental::REntry;
 
      class RNTCollectionCursor : public ICollectionCursor,
                                  virtual public IPositionSeek,
                                  virtual public implements<ICollectionSize>
      {
      public:
 
         RNTCollectionCursor(
            const pool::ICollectionDescription& description,
            const pool::CollectionRowBuffer& collectionRowBuffer,
            RNTupleReader* reader );
 
        
         virtual bool next() override;
 
         virtual const pool::CollectionRowBuffer& currentRow() const override;
 
         virtual bool seek(long long int position) override;
 
         virtual int size() override;
 
         virtual const Token& eventRef() const override;
 
         virtual void close() override;
 
         virtual ~RNTCollectionCursor();
 
      protected:
 
         const ICollectionDescription&  m_description;
 
         RNTupleReader*                 m_RNTReader;
 
         std::unique_ptr< REntry >      m_RNTEntry;
 
         pool::CollectionRowBuffer      m_collectionRowBuffer;
 
         std::vector< std::pair< Token*, std::string > >  m_tokens;
 
         int                            m_idx;
         bool                           m_dummyRef;
      };
   }
}
 
#endif