TTreeCollectionCursor.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef ROOTCOLLECTION_TTREECOLLECTIONCURSOR_H
#define ROOTCOLLECTION_TTREECOLLECTIONCURSOR_H
 
 
#include "CollectionSvc/CollectionRowBuffer.h"
#include "CollectionSvc/CollectionDescription.h"
#include "CollectionSvc/ICollectionCursor.h"
 
class Token;
class TTree;
class TBranch;
 
namespace pool {
   namespace RootCollection {
      class TTreeCollectionCursor : public ICollectionCursor
      {
     public:
 
        TTreeCollectionCursor(
           const pool::CollectionDescription& description,
           const pool::CollectionRowBuffer& collectionRowBuffer,
           TTree *tree );

        virtual bool next() override;

        virtual const pool::CollectionRowBuffer& currentRow() const override;

        virtual std::size_t size() override;

        virtual bool seek(std::size_t position) override;

        virtual const Token& eventRef() const override;

        virtual void close() override;
 
        virtual ~TTreeCollectionCursor();
 
     protected:
 
        const CollectionDescription    &m_description;

        pool::CollectionRowBuffer       m_collectionRowBuffer;
 
        static const unsigned int       c_maxLengthOfStrings = 5000;
        char                            m_charBuffer[c_maxLengthOfStrings];
 
        typedef std::vector< std::pair<TBranch*, std::string*> >  AttrBranchVector_t;
        AttrBranchVector_t              m_attrBranches;
        TBranch*                        m_tokenBranch;
 
        std::size_t                     m_idx;
        std::size_t                     m_entries;
      };
   }
}
 
#endif