RNTCollectionQuery.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef RNTCOLLECTIONQUERY_H
#define RNTCOLLECTIONQUERY_H
 
#include "CoralBase/AttributeList.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:
      RNTCollectionQuery( const pool::ICollectionDescription& description, RNTupleReader *reader );
    
      virtual ~RNTCollectionQuery();
 
      virtual void addToOutputList( const std::string& columnName );
 
      virtual void addToOutputList( const std::vector<std::string>& columnNames );
 
      virtual void selectAllAttributes();
 
      virtual void selectAllTokens();
 
      virtual void selectAll();
 
      virtual void setCondition( const std::string& whereClause,
                                 coral::AttributeList* attributeBindData = 0,
                                 TokenList* tokenBindData = 0 ) ;
 
      virtual const std::string& whereClause() const;
 
      virtual void setRowCacheSize( int rowCacheSize );
 
      virtual pool::ICollectionCursor& execute();
 
      virtual void skipEventReference( bool = true );
 
   protected:
 
      void                addToTokenOutputList( const std::string& columnName );
 
      void                addToAttributeOutputList( const std::string& columnName );
        
        
      const ICollectionDescription    &m_description;
      RNTupleReader                  *m_reader {nullptr};   // owned by the Collection
 
      RNTCollectionCursor            *m_cursor {nullptr};
 
      std::string                     m_whereClause;
 
      pool::TokenList                 m_outputTokenList;
      coral::AttributeList            m_outputAttributeList;
 
      std::set< std::string >         m_selectedColumnNames;
 
      bool                            m_skipEventRef; 
   };
 
} // end namespace 
 
#endif