Detailed Description

It allows to group a number of runs that satisfy given metadata conditions.

This class allows to make a selection of ROOT data files that fulfill certain metadata conditions allowing to create a group of files that define a particular dataset. The files will be searched in a relative or absolute path that is given together the filePattern parameter.

Basic file selection

We will be able to define the dates range where files will be accepted, using startTime and endTime parameters. The run start time and end time stored inside TRestRun will be evaluated to decide if the file should be considered.

A summary of the basic parameters follows:

filePattern: A full path glob pattern to the files that will be considered. It is a first filter considering the path and the filename. Usual wild cards such as * or ? will be allowed to target a given range of files.
startTime: Only files with start run after startTime will be considered.
endTime: Only files with end run before endTime will be considered.

Metadata rules

We may add rules for any metadata class existing inside our ROOT datafiles. For such, we use the filter key where we define the metadata member name where we want to evaluate the rules. We need to define the metadata field where we specify the class name or metadata user given name, together with the metadata member we want to access, the metadata member must be named using the coventions defined inside the methods TRestRun::ReplaceMetadataMember and TRestRun::ReplaceMetadataMembers.

Three optional fields can be used to apply the rule:

contains: It will accept the file if the metadata member contains the string given inside this field, it can also be a number.
greaterThan: It will accept only runs with the corresponding metadata member above the given value. Obviously, the value must be a numeric value.
lowerThan: It will accept only runs with the corresponding metadata member below the given value. Obviously, the value must be a numeric value.

Example of metadata rule:

Branches (or observables) selection

Once the files that fulfill the given dates, filename pattern and metadata rules have been identified, the initialization will produce an instance to a ROOT::RDataFrame and an instance to a ROOT::TTree that will give access to the unified analysis tree. The available columns or branches at those instances will be defined by the user inside this metadata class, through the special keywords observables and processObservables.

observables: This key will allow to add a list of observables present at the analysis tree to be considered for the new compilation. The values must be separated by colons, ":".
processObservables: This key will allow to add a list of process names for which all the observables found inside will be added to the compilation. The values must be separated by colons, ":".

Their use can be seen in the following example:

<TRestDataSet name="DummySet">
 
   <parameter name="startTime" value = "2022/04/28 00:00" />
   <parameter name="endTime" value = "2022/04/28 23:59" />
   <parameter name="filePattern" value="test*.root"/>
 
   <filter metadata="TRestRun::fRunTag" contains="Baby" />
 
   // Will add to the final tree only the specific observables
   <observables list="g4Ana_totalEdep:hitsAna_energy" />
 
   // Will apply a cut to the observables
   <TRestCut>
     <cut name="c1" variable="rawAna_NumberOfGoodSignals" condition=">1" />
     <cut name="c1" variable="rawAna_NumberOfGoodSignals" condition="<100" />
   </TRestCut>
   // Will add all the observables from the process `rawAna`
   <processObservables list="rate:rawAna" />
 
</TRestDataSet>

Basic usage

The basic usage of this class is by loading the metadata class as any other metadata class. After initialization, the user will get access to the internal RDataFrame and TTree data members, as shown in the following example:

restRoot
[0] TRestDataSet d("dataset");
[1] d.GenerateDataSet();
[2] d.GetTree()->GetEntries()
[3] d.GetDataFrame().GetColumnNames()

We can then use our favorite ROOT::RDataFrame or TTree methods.

Exporting datasets

On top of performing a compilation of runs to construct a dataset and access the data in a unified way, we may also save the generated dataset to disk. This feature might be used to generate easier to handle data compilations that have been extracted from an official data repository.

Different output formats are supported:

root: It will store the simplified TTree with the observables selected by the user and compiled with the corresponding file selection. The root file will also contain a TRestDataSet object to allow future users of the output file generated to identify the origin of the data.
txt or csv: It will create an ASCII table where each column will contain the data of a given branch. A header will be written inside the file with all the information found inside the TRestDataSet instance.

Example 1 Generate DataSet from config file:

restRoot
[0] TRestDataSet d("dataset", "dataSetName");
[1] d.GenerateDataSet();
[2] d.Export("mydataset.csv");
[3] d.Export("mydataset.root");

Example 2 Import existing DataSet:

restRoot
[0] TRestDataSet d();
[1] d.Import("myDataSet.root");
[2] d.GetTree()->GetEntries()

Relevant quantities

Sometimes we will be willing that our dataset contains few variables that are extremelly meaningful for the data compilation, and that will be required for further calculations or for the proper interpretation of the data. The key <quantity will allow the user to define relevant quantities that will be stored together with the dataset. These quantitites must be extracted from existing metadata members that are present at the original files. There are different fields allowed inside, such as: name, metadata, strategy and description.

Example:

<quantity name="Nsim" metadata="[TRestProcessRunner::fEventsToProcess]"

strategy="accumulate" description="The total number of simulated events."/>

The name field will be the user given name of the quantity. The metadata field inside the <quantity definition will allow to include a metadata member or a calculation based on a formula where metadata members intervine. The method TRestRun::ReplaceMetadataMembers is the responsible to translate the given metadata formula into a numeric value, check the documentation inside that method to find out the proper format of metadata members inside this field.

There are also different strategies for extracting the quantity value, which are defined by the user using the field strategy, the different options available are:

accumulate: It will add the value of the metadata definition for each of the selected files that will be included in the dataset.
max: It will register the maximum metadata value found in each of the files included in the dataset.
min: It will register the minimum metadata value found in each of the files included in the dataset.
unique: It will simply register the value of the metadata member. If the metadata member value found at each of the selected files is not exactly the same, then a warning output message will be prompted.
last: It will simply register the value of the metadata member from the last file in the list of selected files.

Adding a new column based on relevant quantities

Using the method TRestDataSet::Define method we can implement a formula based on column names and relevant quantities. Then, the relevant quantities will be sustituted by their dataset value.

dataset.GetColumnNames()
dataset.Define("newColumnName", "QuantityName * column1" )
dataset.GetColumnNames()
dataset.GetDataFrame().Display({"column1", "newColumnName"})->Print();

Adding a new column on-the-fly during the dataset generation

It is also possible to add new column definitions inside the RML so that the new column will be already pre-generated and included in the new dataset when we invoke TRestDataSet::GenerateDataSet.

We can use a valid mathematical expression where we may include any already existing column or observable, and/or any relevant quantity we introduced in our dataset definition.

We may use the addColumn keyword to add new columns as follows:

where SolarFlux,GeneratorArea and Nsim are the given names of the relevant quantities inside the dataset.

REST-for-Physics - Software for Rare Event Searches Toolkit

History of developments:

2022-November: First implementation of TRestDataSet Javier Galan

Author: : Javier Galan (javie.nosp@m.r.ga.nosp@m.lan.l.nosp@m.acar.nosp@m.ra@ce.nosp@m.rn.c.nosp@m.h)

Definition at line 34 of file TRestDataSet.h.

#include <TRestDataSet.h>

Inheritance diagram for TRestDataSet:

Data Structures
struct	RelevantQuantity

Public Member Functions
	ClassDefOverride (TRestDataSet, 7)

ROOT::RDF::RNode	DefineColumn (const std::string &columnName, const std::string &formula)
	This function will add a new column to the RDataFrame using the same scheme as the usual RDF::Define method, but it will on top of that evaluate the values of any relevant quantities used. More...

void	EnableMultiThreading (Bool_t enable=true)

void	Export (const std::string &filename, std::vector< std::string > excludeColumns={})
	It will generate an output file with the dataset compilation. Only the selected branches and the files that fulfill the metadata filter conditions will be included. More...

void	GenerateDataSet ()
	This function generates the data frame with the filelist and column names (or observables) that have been defined by the user. More...

auto	GetAddedColumns () const

auto	GetCut () const

ROOT::RDF::RNode	GetDataFrame () const
	Gives access to the RDataFrame. More...

auto	GetEndTime () const

auto	GetFilePattern () const

std::vector< std::string >	GetFileSelection ()
	It returns a list of the files that have been finally selected. More...

auto	GetFileSelection () const

auto	GetFilterContains () const

auto	GetFilterEndTime () const

auto	GetFilterEqualsTo () const

auto	GetFilterGreaterThan () const

auto	GetFilterLowerThan () const

auto	GetFilterMetadata () const

auto	GetFilterStartTime () const

size_t	GetNumberOfBranches ()
	Number of variables (or observables) More...

size_t	GetNumberOfColumns ()
	Number of variables (or observables) More...

auto	GetObservablesList () const

auto	GetProcessObservablesList () const

auto	GetQuantity () const

auto	GetStartTime () const

Double_t	GetTotalTimeInSeconds () const
	It returns the accumulated run time in seconds. More...

TTree *	GetTree () const
	Gives access to the tree. More...

void	Import (const std::string &fileName)
	This function imports metadata from a root file it import metadata info from the previous dataSet while it opens the analysis tree. More...

void	Import (std::vector< std::string > fileNames)
	This function initializes the chained tree and the RDataFrame using as input several root files that should contain TRestDataSet metadata information. The values of the first dataset will be considered to be stored in this new instance. More...

void	Initialize () override
	This function initialize different parameters from the TRestDataSet. More...

auto	IsMergedDataSet () const

ROOT::RDF::RNode	MakeCut (const TRestCut *cut)
	This function apply a TRestCut to the dataframe and returns a dataframe with the applied cuts. Note that the cuts are not applied directly to the dataframe on TRestDataSet, to do so you should do fDataSet = MakeCut(fCut);. More...

Bool_t	Merge (const TRestDataSet &dS)
	This function merge different TRestDataSet metadata in current dataSet. More...

TRestDataSet &	operator= (TRestDataSet &dS)
	Operator to copy TRestDataSet metadata. More...

void	PrintMetadata () override
	Prints on screen the information about the metadata members of TRestDataSet. More...

void	SetDataFrame (const ROOT::RDF::RNode &dS)

void	SetFilePattern (const std::string &pattern)

void	SetObservablesList (const std::vector< std::string > &obsList)

void	SetQuantity (const std::map< std::string, RelevantQuantity > &quantity)

void	SetTotalTimeInSeconds (Double_t seconds)

	TRestDataSet ()
	Default constructor. More...

	TRestDataSet (const char *cfgFileName, const std::string &name="")
	Constructor loading data from a config file. More...

	~TRestDataSet ()
	Default destructor. More...

Public Member Functions inherited from TRestMetadata
void	AddLog (std::string log="", bool print=true)
	Add logs to messageBuffer. More...

void	DoNotStore ()
	If this method is called the metadata information will not be stored in disk. More...

TVector2	Get2DVectorParameterWithUnits (std::string parName, TVector2 defaultValue=TVector2(-1, -1))

TVector3	Get3DVectorParameterWithUnits (std::string parName, TVector3 defaultValue=TVector3(-1, -1, -1))

TString	GetCommit ()
	Returns the REST commit value stored in fCommit. More...

std::string	GetConfigBuffer ()
	Returns the config section of this class. More...

std::string	GetDataMemberValue (std::string memberName)
	Get the value of data member as string. More...

std::vector< std::string >	GetDataMemberValues (std::string memberName, Int_t precision=0)
	Get the value of datamember as a vector of strings. More...

TString	GetDataPath ()
	Returns a std::string with the path used for data storage. More...

Double_t	GetDblParameterWithUnits (std::string parName, Double_t defaultValue=PARAMETER_NOT_FOUND_DBL)
	Gets the value of the parameter name parName, after applying unit conversion. More...

Bool_t	GetError () const
	It returns true if an error was identified by a derived metadata class. More...

TString	GetErrorMessage ()
	Returns a std::string containing the error message. More...

TString	GetLibraryVersion ()
	Returns the REST libraty version stored in fLibraryVersion. More...

TString	GetMainDataPath ()
	Gets a std::string with the path used for data storage. More...

Int_t	GetNumberOfErrors () const

Int_t	GetNumberOfWarnings () const

std::string	GetParameter (std::string parName, TString defaultValue=PARAMETER_NOT_FOUND_STR)
	Returns corresponding REST Metadata parameter from multiple sources. More...

std::string	GetSectionName ()
	Returns the section name of this class, defined at the beginning of fSectionName. More...

TRestStringOutput::REST_Verbose_Level	GetVerboseLevel ()
	returns the verboselevel in type of REST_Verbose_Level enumerator More...

TString	GetVerboseLevelString ()
	returns the verbose level in type of TString More...

TString	GetVersion ()
	Returns the REST version stored in fVersion. More...

Int_t	GetVersionCode ()

UInt_t	GetVersionMajor () const

UInt_t	GetVersionMinor () const

UInt_t	GetVersionPatch () const

Bool_t	GetWarning () const
	It returns true if an error was identified by a derived metadata class. More...

TString	GetWarningMessage ()
	Returns a std::string containing the warning message. More...

virtual void	Initialize ()
	Making default settings. More...

TRestMetadata *	InstantiateChildMetadata (int index, std::string pattern="")
	This method will retrieve a new TRestMetadata instance of a child element of the present TRestMetadata instance based on the `index` given by argument, which defines the element order to be retrieved, 0 for first element found, 1 for the second element found, etc. More...

TRestMetadata *	InstantiateChildMetadata (std::string pattern="", std::string name="")
	This method will retrieve a new TRestMetadata instance of a child element of the present TRestMetadata instance based on the `name` given by argument. More...

Bool_t	isCleanState () const

Bool_t	isOfficialRelease () const

Int_t	LoadConfigFromBuffer ()
	Initialize data from a string element buffer. More...

Int_t	LoadConfigFromElement (TiXmlElement eSectional, TiXmlElement eGlobal, std::map< std::string, std::string > envs={})
	Main starter method. More...

Int_t	LoadConfigFromFile (const std::string &configFilename, const std::string &sectionName="")
	Give the file name, find out the corresponding section. Then call the main starter. More...

virtual void	Merge (const TRestMetadata &)

TRestMetadata &	operator= (const TRestMetadata &)

void	Print ()
	Implementing TObject::Print() method. More...

void	PrintConfigBuffer ()
	Print the config xml section stored in the class. More...

void	PrintMessageBuffer ()
	Print the buffered message. More...

virtual void	PrintMetadata ()
	Implemented it in the derived metadata class to print out specific metadata information. More...

void	PrintTimeStamp (Double_t timeStamp)
	Print the current time on local machine. More...

void	SetConfigFile (std::string configFilename)
	set config file path from external More...

void	SetError (std::string message="", bool print=true, int maxPrint=5)
	A metadata class may use this method to signal that something went wrong. More...

void	SetHostmgr (TRestManager *m)
	Set the host manager for this class. More...

void	SetSectionName (std::string sName)
	set the section name, clear the section content More...

void	SetVerboseLevel (TRestStringOutput::REST_Verbose_Level v)
	sets the verbose level More...

void	SetWarning (std::string message="", bool print=true, int maxPrint=5)
	A metadata class may use this method to signal that something went wrong. More...

void	Store ()
	If this method is called the metadata information will be stored in disk. More...

	TRestMetadata (const TRestMetadata &)

virtual void	UpdateMetadataMembers ()
	Method to allow implementation of specific metadata members updates at inherited classes. More...

virtual Int_t	Write (const char *name=nullptr, Int_t option=0, Int_t bufsize=0)
	overwriting the write() method with fStore considered More...

void	WriteConfigBuffer (std::string fName)
	Writes the config buffer to a file in append mode. More...

	~TRestMetadata ()
	TRestMetadata default destructor. More...

Protected Member Functions
virtual std::vector< std::string >	FileSelection ()
	Function to determine the filenames that satisfy the dataset conditions. More...

Protected Member Functions inherited from TRestMetadata
std::string	ElementToString (TiXmlElement *ele)
	Convert an TiXmlElement object to string. More...

TVector2	Get2DVectorParameterWithUnits (std::string parName, TiXmlElement *e, TVector2 defaultValue=TVector2(-1, -1))

TVector3	Get3DVectorParameterWithUnits (std::string parName, TiXmlElement *e, TVector3 defaultValue=TVector3(-1, -1, -1))

Double_t	GetDblParameterWithUnits (std::string parName, TiXmlElement *e, Double_t defaultVal=PARAMETER_NOT_FOUND_DBL)

TiXmlElement *	GetElement (std::string eleDeclare, TiXmlElement *e=nullptr)
	Get an xml element from a given parent element, according to its declaration. More...

TiXmlElement *	GetElementFromFile (std::string configFilename, std::string NameOrDecalre="")
	Open an xml encoded file and find its element. More...

TiXmlElement *	GetElementWithName (std::string eleDeclare, std::string eleName)
	Get an xml element from the default location, according to its declaration and its field "name". More...

TiXmlElement *	GetElementWithName (std::string eleDeclare, std::string eleName, TiXmlElement *e)
	Get an xml element from a given parent element, according to its declaration and its field "name". More...

std::string	GetFieldValue (std::string fieldName, std::string definition, size_t fromPosition=0)
	Gets field value in an xml element string by parsing it as TiXmlElement. More...

std::string	GetFieldValue (std::string parName, TiXmlElement *e)
	Returns the field value of an xml element which has the specified name. More...

std::string	GetKEYDefinition (std::string keyName)
	Gets the first key definition for keyName found inside buffer starting at fromPosition. More...

std::string	GetKEYDefinition (std::string keyName, size_t &Position)

std::string	GetKEYDefinition (std::string keyName, size_t &Position, std::string buffer)

std::string	GetKEYDefinition (std::string keyName, std::string buffer)

std::string	GetKEYStructure (std::string keyName)
	Gets the first key structure for keyName found inside buffer after fromPosition. More...

std::string	GetKEYStructure (std::string keyName, size_t &Position)

std::string	GetKEYStructure (std::string keyName, size_t &Position, std::string buffer)

std::string	GetKEYStructure (std::string keyName, size_t &Position, TiXmlElement *ele)

std::string	GetKEYStructure (std::string keyName, std::string buffer)

TiXmlElement *	GetNextElement (TiXmlElement *e)
	Get the next sibling xml element of this element, with same eleDeclare. More...

std::string	GetParameter (std::string parName, size_t &pos, std::string inputString)
	Returns the value for the parameter name parName found in inputString. More...

std::string	GetParameter (std::string parName, TiXmlElement *e, TString defaultValue=PARAMETER_NOT_FOUND_STR)
	Returns the value for the parameter named parName in the given section. More...

std::pair< std::string, std::string >	GetParameterAndUnits (std::string parname, TiXmlElement *e=nullptr)
	Returns the unit string of the given parameter of the given xml section. More...

std::map< std::string, std::string >	GetParametersList ()
	It retrieves a map of all parameter:value found in the metadata class. More...

TString	GetSearchPath ()

virtual void	InitFromConfigFile ()
	To make settings from rml file. This method must be implemented in the derived class. More...

virtual void	InitFromRootFile ()
	Method called after the object is retrieved from root file. More...

virtual Int_t	LoadSectionMetadata ()
	This method does some preparation of xml section. More...

void	ReadAllParameters ()
	Reflection methods, Set value of a datamember in class according to TRestMetadata::fElement. More...

void	ReadParametersList (std::map< std::string, std::string > &list)
	It reads a parameter list and associates it to its corresponding metadata member. par0 --> fPar0. More...

std::string	ReplaceConstants (const std::string buffer)
	Identifies "constants" in the input buffer, and replace them with corresponding value. More...

std::string	ReplaceVariables (const std::string buffer)
	Identifies environmental variable replacing marks in the input buffer, and replace them with corresponding value. More...

void	ReSetVersion ()
	Resets the version of TRestRun to REST_RELEASE. Only TRestRun is allowed to update version. More...

std::string	SearchFile (std::string filename)
	Search files in current directory and directories specified in "searchPath" section. More...

void	SetLibraryVersion (TString version)
	Set the library version of this metadata class. More...

TiXmlElement *	StringToElement (std::string definition)
	Parsing a string into TiXmlElement object. More...

	TRestMetadata ()
	TRestMetadata default constructor. More...

	TRestMetadata (const char *configFilename)
	constructor More...

void	UnSetVersion ()
	Resets the version of TRestRun to -1, in case the file is old REST file. Only TRestRun is allowed to update version. More...

Private Member Functions
void	InitFromConfigFile () override
	Initialization of specific TRestDataSet members through an RML file. More...

Private Attributes
std::vector< std::pair< std::string, std::string > >	fColumnNameExpressions
	A list of new columns together with its corresponding expressions added to the dataset. More...

TRestCut *	fCut = nullptr
	Parameter cuts over the selected dataset. More...

ROOT::RDF::RNode	fDataSet = ROOT::RDataFrame(0)
	The resulting RDF::RNode object after initialization. More...

Double_t	fEndTime = REST_StringHelper::StringToTimeStamp(fFilterStartTime)
	TimeStamp for the end time of the last file. More...

Bool_t	fExternal = false

std::string	fFilePattern = ""
	A glob file pattern that must be satisfied by all files. More...

std::vector< std::string >	fFileSelection
	A list populated by the FileSelection method using the conditions of the dataset. More...

std::vector< std::string >	fFilterContains
	If not empty it will check if the metadata member contains the string. More...

std::string	fFilterEndTime = "3000/12/31"
	All the selected runs will have an ending date before fEndTime. More...

std::vector< Double_t >	fFilterEqualsTo
	If the corresponding element is not empty it will check if the metadata member is equal. More...

std::vector< Double_t >	fFilterGreaterThan
	If the corresponding element is not empty it will check if the metadata member is greater. More...

std::vector< Double_t >	fFilterLowerThan
	If the corresponding element is not empty it will check if the metadata member is lower. More...

std::vector< std::string >	fFilterMetadata
	A list of metadata members where filters will be applied. More...

std::string	fFilterStartTime = "2000/01/01"
	All the selected runs will have a starting date after fStartTime. More...

std::vector< std::string >	fImportedFiles
	The list of dataset files imported. More...

Bool_t	fMergedDataset = false
	It keeps track if the generated dataset is a pure dataset or a merged one. More...

Bool_t	fMT = false
	A flag to enable Multithreading during dataframe generation. More...

std::vector< std::string >	fObservablesList
	It contains a list of the observables that will be added to the final tree or exported file. More...

std::vector< std::string >	fProcessObservablesList
	It contains a list of the process where all observables should be added. More...

std::map< std::string, RelevantQuantity >	fQuantity
	The properties of a relevant quantity that we want to store together with the dataset. More...

Double_t	fStartTime = REST_StringHelper::StringToTimeStamp(fFilterEndTime)
	TimeStamp for the start time of the first file. More...

Double_t	fTotalDuration = 0
	The total integrated run time of selected files. More...

TChain *	fTree = nullptr
	A pointer to the generated tree. More...

Additional Inherited Members
Protected Attributes inherited from TRestMetadata
std::string	configBuffer
	The buffer where the corresponding metadata section is stored. Filled only during Write() More...

std::string	fConfigFileName
	Full name of the rml file. More...

std::map< std::string, std::string >	fConstants
	Saving a list of rml constants. name-value std::pair. Constants are temporary for this class only. More...

TiXmlElement *	fElement
	Saving the sectional element together with global element. More...

TiXmlElement *	fElementGlobal
	Saving the global element, to be passed to the resident class, if necessary. More...

Bool_t	fError = false
	It can be used as a way to identify that something went wrong using SetError method. More...

TString	fErrorMessage = ""
	A std::string to store an optional error message through method SetError. More...

TRestManager *	fHostmgr
	All metadata classes can be initialized and managed by TRestManager. More...

Int_t	fNErrors = 0
	It counts the number of errors notified. More...

Int_t	fNWarnings = 0
	It counts the number of warnings notified. More...

std::string	fSectionName
	Section name given in the constructor of the derived metadata class. More...

Bool_t	fStore
	This variable is used to determine if the metadata structure should be stored in the ROOT file. More...

std::map< std::string, std::string >	fVariables
	Saving a list of rml variables. name-value std::pair. More...

TRestStringOutput::REST_Verbose_Level	fVerboseLevel
	Verbose level used to print debug info. More...

Bool_t	fWarning = false
	It can be used as a way to identify that something went wrong using SetWarning method. More...

TString	fWarningMessage = ""
	It can be used as a way to identify that something went wrong using SetWarning method. More...

std::string	messageBuffer
	The buffer to store the output message through TRestStringOutput in this class. More...

endl_t	RESTendl
	Termination flag object for TRestStringOutput. More...

Constructor & Destructor Documentation

◆ TRestDataSet() [1/2]

TRestDataSet::TRestDataSet ( )

Default constructor.

Definition at line 268 of file TRestDataSet.cxx.

◆ TRestDataSet() [2/2]

TRestDataSet::TRestDataSet	(	const char *	cfgFileName,
		const std::string &	name = `""`
	)

Constructor loading data from a config file.

If no configuration path is defined using TRestMetadata::SetConfigFilePath the path to the config file must be specified using full path, absolute or relative.

The default behaviour is that the config file must be specified with full path, absolute or relative.

Parameters

cfgFileName	A const char* giving the path to an RML file.
name	The name of the specific metadata. It will be used to find the corresponding TRestAxionMagneticField section inside the RML.

Definition at line 284 of file TRestDataSet.cxx.

◆ ~TRestDataSet()

TRestDataSet::~TRestDataSet ( )

Default destructor.

Definition at line 291 of file TRestDataSet.cxx.

Member Function Documentation

◆ DefineColumn()

ROOT::RDF::RNode TRestDataSet::DefineColumn	(	const std::string &	columnName,
		const std::string &	formula
	)

This function will add a new column to the RDataFrame using the same scheme as the usual RDF::Define method, but it will on top of that evaluate the values of any relevant quantities used.

For example, the following code line would create a new column named test replacing the relevant quantity Nsim and the previously existing column probability.

d.Define("test", "Nsim * probability");

Definition at line 541 of file TRestDataSet.cxx.

◆ EnableMultiThreading()

void TRestDataSet::EnableMultiThreading ( Bool_t enable = true )

inline

Definition at line 133 of file TRestDataSet.h.

◆ Export()

void TRestDataSet::Export	(	const std::string &	filename,
		std::vector< std::string >	excludeColumns = `{}`
	)

It will generate an output file with the dataset compilation. Only the selected branches and the files that fulfill the metadata filter conditions will be included.

For the moment we produce two different output files.

A plain-text (file recognized with csv or txt extension): It will produce a table with observable values, including a header of the dataset conditions.
A root file (recognized through its root extension): It will write to disk an Snapshot of the current dataset, i.e. in standard TTree format, together with a copy of the TRestDataSet instance that contains the conditions used to generate the dataset.

Definition at line 784 of file TRestDataSet.cxx.

◆ FileSelection()

std::vector< std::string > TRestDataSet::FileSelection ( )

protectedvirtual

Function to determine the filenames that satisfy the dataset conditions.

Definition at line 379 of file TRestDataSet.cxx.

◆ GenerateDataSet()

void TRestDataSet::GenerateDataSet ( )

This function generates the data frame with the filelist and column names (or observables) that have been defined by the user.

Definition at line 303 of file TRestDataSet.cxx.

◆ GetAddedColumns()

auto TRestDataSet::GetAddedColumns ( ) const

inline

Definition at line 180 of file TRestDataSet.h.

◆ GetCut()

auto TRestDataSet::GetCut ( ) const

inline

Definition at line 181 of file TRestDataSet.h.

◆ GetDataFrame()

ROOT::RDF::RNode TRestDataSet::GetDataFrame ( ) const

inline

Gives access to the RDataFrame.

Definition at line 127 of file TRestDataSet.h.

◆ GetEndTime()

auto TRestDataSet::GetEndTime ( ) const

inline

Definition at line 169 of file TRestDataSet.h.

◆ GetFilePattern()

auto TRestDataSet::GetFilePattern ( ) const

inline

Definition at line 170 of file TRestDataSet.h.

◆ GetFileSelection() [1/2]

std::vector< std::string > TRestDataSet::GetFileSelection ( )

inline

It returns a list of the files that have been finally selected.

Definition at line 161 of file TRestDataSet.h.

◆ GetFileSelection() [2/2]

auto TRestDataSet::GetFileSelection ( ) const

inline

Definition at line 172 of file TRestDataSet.h.

◆ GetFilterContains()

auto TRestDataSet::GetFilterContains ( ) const

inline

Definition at line 175 of file TRestDataSet.h.

◆ GetFilterEndTime()

auto TRestDataSet::GetFilterEndTime ( ) const

inline

Definition at line 167 of file TRestDataSet.h.

◆ GetFilterEqualsTo()

auto TRestDataSet::GetFilterEqualsTo ( ) const

inline

Definition at line 178 of file TRestDataSet.h.

◆ GetFilterGreaterThan()

auto TRestDataSet::GetFilterGreaterThan ( ) const

inline

Definition at line 176 of file TRestDataSet.h.

◆ GetFilterLowerThan()

auto TRestDataSet::GetFilterLowerThan ( ) const

inline

Definition at line 177 of file TRestDataSet.h.

◆ GetFilterMetadata()

auto TRestDataSet::GetFilterMetadata ( ) const

inline

Definition at line 174 of file TRestDataSet.h.

◆ GetFilterStartTime()

auto TRestDataSet::GetFilterStartTime ( ) const

inline

Definition at line 166 of file TRestDataSet.h.

◆ GetNumberOfBranches()

size_t TRestDataSet::GetNumberOfBranches ( )

inline

Number of variables (or observables)

Definition at line 158 of file TRestDataSet.h.

◆ GetNumberOfColumns()

size_t TRestDataSet::GetNumberOfColumns ( )

inline

Number of variables (or observables)

Definition at line 155 of file TRestDataSet.h.

◆ GetObservablesList()

auto TRestDataSet::GetObservablesList ( ) const

inline

Definition at line 171 of file TRestDataSet.h.

◆ GetProcessObservablesList()

auto TRestDataSet::GetProcessObservablesList ( ) const

inline

Definition at line 173 of file TRestDataSet.h.

◆ GetQuantity()

auto TRestDataSet::GetQuantity ( ) const

inline

Definition at line 179 of file TRestDataSet.h.

◆ GetStartTime()

auto TRestDataSet::GetStartTime ( ) const

inline

Definition at line 168 of file TRestDataSet.h.

◆ GetTotalTimeInSeconds()

Double_t TRestDataSet::GetTotalTimeInSeconds ( ) const

inline

It returns the accumulated run time in seconds.

Definition at line 164 of file TRestDataSet.h.

◆ GetTree()

TTree * TRestDataSet::GetTree ( ) const

inline

Gives access to the tree.

Definition at line 136 of file TRestDataSet.h.

◆ Import() [1/2]

void TRestDataSet::Import ( const std::string & fileName )

This function imports metadata from a root file it import metadata info from the previous dataSet while it opens the analysis tree.

Definition at line 975 of file TRestDataSet.cxx.

◆ Import() [2/2]

void TRestDataSet::Import ( std::vector< std::string > fileNames )

This function initializes the chained tree and the RDataFrame using as input several root files that should contain TRestDataSet metadata information. The values of the first dataset will be considered to be stored in this new instance.

The metadata member fMergedDataset will be set to true to understand this dataset is the combination of several datasets, and not a pure original one.

Definition at line 1020 of file TRestDataSet.cxx.

◆ InitFromConfigFile()

void TRestDataSet::InitFromConfigFile ( )

overrideprivatevirtual

Initialization of specific TRestDataSet members through an RML file.

Reading filters

Reading observables

Reading process observables

Reading relevant quantities

Reading new dataset columns

Reimplemented from TRestMetadata.

Definition at line 655 of file TRestDataSet.cxx.

◆ Initialize()

void TRestDataSet::Initialize ( )

overridevirtual

This function initialize different parameters from the TRestDataSet.

Reimplemented from TRestMetadata.

Definition at line 297 of file TRestDataSet.cxx.

◆ IsMergedDataSet()

auto TRestDataSet::IsMergedDataSet ( ) const

inline

Definition at line 182 of file TRestDataSet.h.

◆ MakeCut()

ROOT::RDF::RNode TRestDataSet::MakeCut ( const TRestCut * cut )

This function apply a TRestCut to the dataframe and returns a dataframe with the applied cuts. Note that the cuts are not applied directly to the dataframe on TRestDataSet, to do so you should do fDataSet = MakeCut(fCut);.

Definition at line 491 of file TRestDataSet.cxx.

◆ Merge()

Bool_t TRestDataSet::Merge ( const TRestDataSet & dS )

This function merge different TRestDataSet metadata in current dataSet.

Definition at line 950 of file TRestDataSet.cxx.

◆ operator=()

TRestDataSet & TRestDataSet::operator= ( TRestDataSet & dS )

Operator to copy TRestDataSet metadata.

Definition at line 923 of file TRestDataSet.cxx.

◆ PrintMetadata()

void TRestDataSet::PrintMetadata ( )

overridevirtual

Prints on screen the information about the metadata members of TRestDataSet.

Reimplemented from TRestMetadata.

Definition at line 556 of file TRestDataSet.cxx.

◆ SetDataFrame()

void TRestDataSet::SetDataFrame ( const ROOT::RDF::RNode & dS )

inline

Definition at line 189 of file TRestDataSet.h.

◆ SetFilePattern()

void TRestDataSet::SetFilePattern ( const std::string & pattern )

inline

Definition at line 185 of file TRestDataSet.h.

◆ SetObservablesList()

void TRestDataSet::SetObservablesList ( const std::vector< std::string > & obsList )

inline

Definition at line 184 of file TRestDataSet.h.

◆ SetQuantity()

void TRestDataSet::SetQuantity ( const std::map< std::string, RelevantQuantity > & quantity )

inline

Definition at line 186 of file TRestDataSet.h.

◆ SetTotalTimeInSeconds()

void TRestDataSet::SetTotalTimeInSeconds ( Double_t seconds )

inline

Definition at line 188 of file TRestDataSet.h.

Field Documentation

◆ fColumnNameExpressions

std::vector<std::pair<std::string, std::string> > TRestDataSet::fColumnNameExpressions

private

A list of new columns together with its corresponding expressions added to the dataset.

Definition at line 106 of file TRestDataSet.h.

◆ fCut

TRestCut* TRestDataSet::fCut = nullptr

private

Parameter cuts over the selected dataset.

Definition at line 85 of file TRestDataSet.h.

◆ fDataSet

ROOT::RDF::RNode TRestDataSet::fDataSet = ROOT::RDataFrame(0)

private

The resulting RDF::RNode object after initialization.

Definition at line 115 of file TRestDataSet.h.

◆ fEndTime

Double_t TRestDataSet::fEndTime = REST_StringHelper::StringToTimeStamp(fFilterStartTime)

private

TimeStamp for the end time of the last file.

Definition at line 97 of file TRestDataSet.h.

◆ fExternal

Bool_t TRestDataSet::fExternal = false

private

Definition at line 112 of file TRestDataSet.h.

◆ fFilePattern

std::string TRestDataSet::fFilePattern = ""

private

A glob file pattern that must be satisfied by all files.

Definition at line 58 of file TRestDataSet.h.

◆ fFileSelection

std::vector<std::string> TRestDataSet::fFileSelection

private

A list populated by the FileSelection method using the conditions of the dataset.

Definition at line 91 of file TRestDataSet.h.

◆ fFilterContains

std::vector<std::string> TRestDataSet::fFilterContains

private

If not empty it will check if the metadata member contains the string.

Definition at line 70 of file TRestDataSet.h.

◆ fFilterEndTime

std::string TRestDataSet::fFilterEndTime = "3000/12/31"

private

All the selected runs will have an ending date before fEndTime.

Definition at line 55 of file TRestDataSet.h.

◆ fFilterEqualsTo

std::vector<Double_t> TRestDataSet::fFilterEqualsTo

private

If the corresponding element is not empty it will check if the metadata member is equal.

Definition at line 79 of file TRestDataSet.h.

◆ fFilterGreaterThan

std::vector<Double_t> TRestDataSet::fFilterGreaterThan

private

If the corresponding element is not empty it will check if the metadata member is greater.

Definition at line 73 of file TRestDataSet.h.

◆ fFilterLowerThan

std::vector<Double_t> TRestDataSet::fFilterLowerThan

private

If the corresponding element is not empty it will check if the metadata member is lower.

Definition at line 76 of file TRestDataSet.h.

◆ fFilterMetadata

std::vector<std::string> TRestDataSet::fFilterMetadata

private

A list of metadata members where filters will be applied.

Definition at line 67 of file TRestDataSet.h.

◆ fFilterStartTime

std::string TRestDataSet::fFilterStartTime = "2000/01/01"

private

All the selected runs will have a starting date after fStartTime.

Definition at line 52 of file TRestDataSet.h.

◆ fImportedFiles

std::vector<std::string> TRestDataSet::fImportedFiles

private

The list of dataset files imported.

Definition at line 103 of file TRestDataSet.h.

◆ fMergedDataset

Bool_t TRestDataSet::fMergedDataset = false

private

It keeps track if the generated dataset is a pure dataset or a merged one.

Definition at line 100 of file TRestDataSet.h.

◆ fMT

Bool_t TRestDataSet::fMT = false

private

A flag to enable Multithreading during dataframe generation.

Definition at line 109 of file TRestDataSet.h.

◆ fObservablesList

std::vector<std::string> TRestDataSet::fObservablesList

private

It contains a list of the observables that will be added to the final tree or exported file.

Definition at line 61 of file TRestDataSet.h.

◆ fProcessObservablesList

std::vector<std::string> TRestDataSet::fProcessObservablesList

private

It contains a list of the process where all observables should be added.

Definition at line 64 of file TRestDataSet.h.

◆ fQuantity

std::map<std::string, RelevantQuantity> TRestDataSet::fQuantity

private

The properties of a relevant quantity that we want to store together with the dataset.

Definition at line 82 of file TRestDataSet.h.

◆ fStartTime

Double_t TRestDataSet::fStartTime = REST_StringHelper::StringToTimeStamp(fFilterEndTime)

private

TimeStamp for the start time of the first file.

Definition at line 94 of file TRestDataSet.h.

◆ fTotalDuration

Double_t TRestDataSet::fTotalDuration = 0

private

The total integrated run time of selected files.

Definition at line 88 of file TRestDataSet.h.

◆ fTree

TChain* TRestDataSet::fTree = nullptr

private

A pointer to the generated tree.

Definition at line 118 of file TRestDataSet.h.

The documentation for this class was generated from the following files:

Detailed Description

Basic file selection

Metadata rules

Branches (or observables) selection

Basic usage

Exporting datasets

Relevant quantities

Adding a new column based on relevant quantities

Adding a new column on-the-fly during the dataset generation

Data Structures

Public Member Functions

Protected Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Constructor & Destructor Documentation

◆ TRestDataSet() [1/2]

◆ TRestDataSet() [2/2]

◆ ~TRestDataSet()

Member Function Documentation

◆ DefineColumn()

◆ EnableMultiThreading()

◆ Export()

◆ FileSelection()

◆ GenerateDataSet()

◆ GetAddedColumns()

◆ GetCut()

◆ GetDataFrame()

◆ GetEndTime()

◆ GetFilePattern()

◆ GetFileSelection() [1/2]

◆ GetFileSelection() [2/2]

◆ GetFilterContains()

◆ GetFilterEndTime()

◆ GetFilterEqualsTo()

◆ GetFilterGreaterThan()

◆ GetFilterLowerThan()

◆ GetFilterMetadata()

◆ GetFilterStartTime()

◆ GetNumberOfBranches()

◆ GetNumberOfColumns()

◆ GetObservablesList()

◆ GetProcessObservablesList()

◆ GetQuantity()

◆ GetStartTime()

◆ GetTotalTimeInSeconds()

◆ GetTree()

◆ Import() [1/2]

◆ Import() [2/2]

◆ InitFromConfigFile()

◆ Initialize()

◆ IsMergedDataSet()

◆ MakeCut()

◆ Merge()

◆ operator=()

◆ PrintMetadata()

◆ SetDataFrame()

◆ SetFilePattern()

◆ SetObservablesList()

◆ SetQuantity()

◆ SetTotalTimeInSeconds()

Field Documentation

◆ fColumnNameExpressions

◆ fCut

◆ fDataSet

◆ fEndTime

◆ fExternal

◆ fFilePattern

◆ fFileSelection

◆ fFilterContains

◆ fFilterEndTime

◆ fFilterEqualsTo

◆ fFilterGreaterThan

◆ fFilterLowerThan

◆ fFilterMetadata

◆ fFilterStartTime

◆ fImportedFiles

◆ fMergedDataset

◆ fMT

◆ fObservablesList