RNTCollectionQuery.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2025 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef RNTCOLLECTIONQUERY_H
#define RNTCOLLECTIONQUERY_H
 
#include "CoralBase/AttributeList.h"
 
#include "AthenaBaseComps/AthMessaging.h"
 
#include "CollectionBase/TokenList.h"
#include "CollectionBase/ICollectionQuery.h"
#include "CollectionBase/ICollectionDescription.h"
#include "CollectionBase/ICollectionCursor.h"
 
#include "RNTCollectionCursor.h"
 
#include <set>
 
namespace pool::RootCollection {
 
   class RNTCollectionQuery : public ICollectionQuery, public AthMessaging
   {
   public:
      RNTCollectionQuery( const pool::ICollectionDescription& description, ROOT::RNTupleReader *reader );
    
      virtual ~RNTCollectionQuery();
 
      virtual void selectAllAttributes();
 
      virtual void selectAllTokens();
 
      virtual void selectAll();
 
      virtual pool::ICollectionCursor& execute();
 
   protected:
 
      void                addToTokenOutputList( const std::string& columnName );
 
      void                addToAttributeOutputList( const std::string& columnName );
        
        
      const ICollectionDescription   &m_description;
      ROOT::RNTupleReader            *m_reader {nullptr};   // owned by the Collection
 
      RNTCollectionCursor            *m_cursor {nullptr};
 
      pool::TokenList                 m_outputTokenList;
      coral::AttributeList            m_outputAttributeList;
 
      std::set< std::string >         m_selectedColumnNames;
 
      bool                            m_skipEventRef; 
   };
 
} // end namespace 
 
#endif