RootCollectionCursor.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ROOTCOLLECTION_ROOTCOLLECTIONCURSOR_H
#define ROOTCOLLECTION_ROOTCOLLECTIONCURSOR_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 "TTree.h"
#include "TEventList.h"
 
namespace pool {
   namespace RootCollection {
      class RootCollectionCursor : public ICollectionCursor,
                                   virtual public IPositionSeek,
                                   virtual public ICollectionSize
      {
     public:
 
        RootCollectionCursor(
           const pool::ICollectionDescription& description,
           const pool::CollectionRowBuffer& collectionRowBuffer,
           TTree *tree,
           const TEventList *evl );
 
        
        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 ~RootCollectionCursor();
 
     protected:
    
        static const unsigned int       c_maxLengthOfStrings = 5000;    
        
        const ICollectionDescription    &m_description;
 
        pool::CollectionRowBuffer       m_collectionRowBuffer;
 
        const TEventList                *m_eventList;
 
        
        char                             m_charBuffer[c_maxLengthOfStrings];
 
        typedef std::vector< std::pair<TBranch*, std::string*> >  AttrBranchVector_t;
        typedef std::vector< std::pair<TBranch*, Token*> >        TokenBranchVector_t;
 
        AttrBranchVector_t              m_attrBranches;
        TokenBranchVector_t             m_tokenBranches;
 
        int                             m_idx;
        Long64_t                        m_entries;
        bool                            m_dummyRef;
      };
   }
}
 
#endif