RDataFrame.hxx

Go to the documentation of this file.

// Author: Enrico Guiraud, Danilo Piparo CERN  12/2016
 
/*************************************************************************
 * Copyright (C) 1995-2018, Rene Brun and Fons Rademakers.               *
 * All rights reserved.                                                  *
 *                                                                       *
 * For the licensing terms see $ROOTSYS/LICENSE.                         *
 * For the list of contributors see $ROOTSYS/README/CREDITS.             *
 *************************************************************************/
 
/**
  \defgroup dataframe DataFrame
ROOT's RDataFrame allows to analyse data stored in TTrees with a high level interface.
*/
 
#ifndef ROOT_RDATAFRAME
#define ROOT_RDATAFRAME
 
#include "TROOT.h" // To allow ROOT::EnableImplicitMT without including ROOT.h
#include "ROOT/RDF/RDatasetSpec.hxx"
#include "ROOT/RDF/RInterface.hxx"
#include "ROOT/RDF/Utils.hxx"
#include "ROOT/RStringView.hxx"
#include "RtypesCore.h"
 
#include <memory>
#include <ostream>
#include <string>
#include <vector>
 
class TDirectory;
class TTree;
 
namespace ROOT {
 
class RDataFrame;
 
namespace Internal {
namespace RDF {
ROOT::RDataFrame MakeDataFrameFromSpec(const RDatasetSpec &spec);
} // namespace RDF
} // namespace Internal
 
namespace RDF {
class RDataSource;
}
 
namespace RDFDetail = ROOT::Detail::RDF;
 
class RDataFrame : public ROOT::RDF::RInterface<RDFDetail::RLoopManager> {
public:
   using ColumnNames_t = ROOT::RDF::ColumnNames_t;
   RDataFrame(std::string_view treeName, std::string_view filenameglob, const ColumnNames_t &defaultBranches = {});
   RDataFrame(std::string_view treename, const std::vector<std::string> &filenames,
              const ColumnNames_t &defaultBranches = {});
   RDataFrame(std::string_view treeName, ::TDirectory *dirPtr, const ColumnNames_t &defaultBranches = {});
   RDataFrame(TTree &tree, const ColumnNames_t &defaultBranches = {});
   RDataFrame(ULong64_t numEntries);
   RDataFrame(std::unique_ptr<ROOT::RDF::RDataSource>, const ColumnNames_t &defaultBranches = {});
 
   friend RDataFrame ROOT::Internal::RDF::MakeDataFrameFromSpec(const ROOT::Internal::RDF::RDatasetSpec &spec);
 
private:
   RDataFrame(ROOT::Internal::RDF::RDatasetSpec spec);
};
 
} // ns ROOT
 
/// Print a RDataFrame at the prompt
namespace cling {
std::string printValue(ROOT::RDataFrame *tdf);
} // ns cling
 
#endif // ROOT_RDATAFRAME