RDataSource.h

Go to the documentation of this file.

// Dear emacs, this is -*- c++ -*-
//
// Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
//
#ifndef XAODDATASOURCE_RDATASOURCE_H
#define XAODDATASOURCE_RDATASOURCE_H
 
// Local include(s).
#include "RDataSourceEvent.h"
 
// Framework include(s).
#include "xAODRootAccess/TStore.h"
 
// ROOT include(s).
#include <ROOT/RDataSource.hxx>
#include <ROOT/RStringView.hxx>
#include <TChain.h>
 
// System include(s).
#include <memory>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace xAOD {

   class RDataSource final : public ROOT::RDF::RDataSource {
 
   public:
      RDataSource( std::string_view fileNameGlob,
                   std::string_view treeName = "CollectionTree" );
      RDataSource( const std::vector< std::string >& fileNames,
                   std::string_view treeName = "CollectionTree" );
      ~RDataSource();

      typedef std::vector< std::pair< ULong64_t, ULong64_t > > EntryRanges_t;


      virtual void SetNSlots( unsigned int slots ) override final;
      virtual void Initialize() override final;
      virtual void
      InitSlot( unsigned int slot, ULong64_t firstEntry ) override final;

      virtual void FinalizeSlot( unsigned int slot ) override final;
      virtual void Finalize() override final;

      virtual const std::vector< std::string >&
      GetColumnNames() const override final;
      virtual bool HasColumn( std::string_view name ) const override final;
      virtual std::string
      GetTypeName( std::string_view column ) const override final;

      virtual EntryRanges_t GetEntryRanges() override final;
      virtual bool
      SetEntry( unsigned int slot, ULong64_t entry ) override final;


      void setVerboseOutput( Bool_t value = kTRUE );
      Bool_t isVerboseOutput() const;

      void setAuxMode( TEvent::EAuxMode mode );
      TEvent::EAuxMode auxMode() const;
 
   private:
      virtual Record_t
      GetColumnReadersImpl( std::string_view column,
                            const std::type_info& typeInfo ) override final;
      void readInputMetadata();


      std::vector< std::string > m_fileNames;
      std::string m_treeName;
      Bool_t m_verboseOutput = kFALSE;
      TEvent::EAuxMode m_auxmode = TEvent::kClassAccess;



      std::vector< std::string > m_columnNames;
      std::unordered_map< std::string, std::string > m_classNameMap;



      EntryRanges_t m_entryRanges;



      std::vector< std::unique_ptr< TChain > > m_chains;
      std::vector< std::unique_ptr< RDataSourceEvent > > m_events;
      std::vector< std::unique_ptr< TStore > > m_stores;

 
   }; // class RDataSource
 
} // namespace xAOD
 
#endif // XAODDATASOURCE_RDATASOURCE_H