A class to process the entries of a TTree in parallel.
By means of its Process method, ROOT::TTreeProcessorMT provides a way to process the entries of a TTree in parallel. When invoking TTreeProcessor::Process, the user passes a function whose only parameter is a TTreeReader. The function iterates on a subrange of entries by using that TTreeReader.
The implementation of ROOT::TTreeProcessorMT parallelizes the processing of the subranges, each corresponding to a cluster in the TTree. This is possible thanks to the use of a ROOT::TThreadedObject, so that each thread works with its own TFile and TTree objects.
Definition at line 80 of file TTreeProcessorMT.hxx.
Public Member Functions | |
TTreeProcessorMT (const std::vector< std::string_view > &filenames, std::string_view treename="", UInt_t nThreads=0u, const std::pair< Long64_t, Long64_t > &globalRange={0, std::numeric_limits< Long64_t >::max()}) | |
TTreeProcessorMT (std::initializer_list< std::string_view > filenames, std::string_view treename="", UInt_t nThreads=0u, const std::pair< Long64_t, Long64_t > &globalRange={0, std::numeric_limits< Long64_t >::max()}) | |
TTreeProcessorMT (std::string_view filename, std::string_view treename="", UInt_t nThreads=0u, const std::pair< Long64_t, Long64_t > &globalRange={0, std::numeric_limits< Long64_t >::max()}) | |
TTreeProcessorMT (TTree &tree, const TEntryList &entries, UInt_t nThreads=0u, const std::vector< std::string > &suppressErrorsForMissingBranches={}) | |
Constructor based on a TTree and a TEntryList. | |
TTreeProcessorMT (TTree &tree, UInt_t nThreads=0u, const std::pair< Long64_t, Long64_t > &globalRange={0, std::numeric_limits< Long64_t >::max()}, const std::vector< std::string > &suppressErrorsForMissingBranches={}) | |
void | Process (std::function< void(TTreeReader &)> func) |
Process the entries of a TTree in parallel. | |
Static Public Member Functions | |
static unsigned int | GetTasksPerWorkerHint () |
Retrieve the current value for the desired number of tasks per worker. | |
static void | SetTasksPerWorkerHint (unsigned int m) |
Set the hint for the desired number of tasks created per worker. | |
Private Member Functions | |
std::vector< std::string > | FindTreeNames () |
Retrieve the names of the TTrees in each of the input files, throw if a TTree cannot be found. | |
Private Attributes | |
TEntryList | fEntryList |
User-defined selection of entry numbers to be processed, empty if none was provided. | |
const std::vector< std::string > | fFileNames |
Names of the files. | |
ROOT::TreeUtils::RFriendInfo | fFriendInfo |
std::pair< Long64_t, Long64_t > | fGlobalRange {0, std::numeric_limits<Long64_t>::max()} |
ROOT::TThreadExecutor | fPool |
! Thread pool for processing. | |
std::vector< std::string > | fSuppressErrorsForMissingBranches {} |
const std::vector< std::string > | fTreeNames |
TTree names (always same size and ordering as fFileNames) | |
ROOT::TThreadedObject< ROOT::Internal::TTreeView > | fTreeView {TNumSlots{ROOT::GetThreadPoolSize()}} |
Thread-local TreeViews. | |
Static Private Attributes | |
static unsigned int | fgTasksPerWorkerHint = 10U |
#include <ROOT/TTreeProcessorMT.hxx>
ROOT::TTreeProcessorMT::TTreeProcessorMT | ( | std::string_view | filename, |
std::string_view | treename = "" , |
||
UInt_t | nThreads = 0u , |
||
const std::pair< Long64_t, Long64_t > & | globalRange = {0, std::numeric_limits< Long64_t >::max()} |
||
) |
ROOT::TTreeProcessorMT::TTreeProcessorMT | ( | const std::vector< std::string_view > & | filenames, |
std::string_view | treename = "" , |
||
UInt_t | nThreads = 0u , |
||
const std::pair< Long64_t, Long64_t > & | globalRange = {0, std::numeric_limits< Long64_t >::max()} |
||
) |
|
inline |
Definition at line 108 of file TTreeProcessorMT.hxx.
TTreeProcessorMT::TTreeProcessorMT | ( | TTree & | tree, |
const TEntryList & | entries, | ||
UInt_t | nThreads = 0u , |
||
const std::vector< std::string > & | suppressErrorsForMissingBranches = {} |
||
) |
Constructor based on a TTree and a TEntryList.
[in] | tree | Tree or chain of files containing the tree to process. |
[in] | entries | List of entry numbers to process. |
[in] | nThreads | Number of threads to create in the underlying thread-pool. The semantics of this argument are the same as for TThreadExecutor. |
Definition at line 414 of file TTreeProcessorMT.cxx.
ROOT::TTreeProcessorMT::TTreeProcessorMT | ( | TTree & | tree, |
UInt_t | nThreads = 0u , |
||
const std::pair< Long64_t, Long64_t > & | globalRange = {0, std::numeric_limits< Long64_t >::max()} , |
||
const std::vector< std::string > & | suppressErrorsForMissingBranches = {} |
||
) |
|
private |
Retrieve the names of the TTrees in each of the input files, throw if a TTree cannot be found.
Definition at line 330 of file TTreeProcessorMT.cxx.
|
static |
Retrieve the current value for the desired number of tasks per worker.
Definition at line 535 of file TTreeProcessorMT.cxx.
void TTreeProcessorMT::Process | ( | std::function< void(TTreeReader &)> | func | ) |
Process the entries of a TTree in parallel.
The user-provided function receives a TTreeReader which can be used to iterate on a subrange of entries
The user needs to be aware that each of the subranges can potentially be processed in parallel. This means that the code of the user function should be thread safe.
[in] | func | User-defined function that processes a subrange of entries |
Definition at line 460 of file TTreeProcessorMT.cxx.
|
static |
Set the hint for the desired number of tasks created per worker.
[in] | tasksPerWorkerHint | Desired number of tasks per worker. |
This allows to create a reasonable number of tasks even if any of the processed files features a bad clustering, for example with a lot of entries and just a few entries per cluster, or to limit the number of tasks spawned when a very large number of files and workers is used.
Definition at line 548 of file TTreeProcessorMT.cxx.
|
private |
User-defined selection of entry numbers to be processed, empty if none was provided.
Definition at line 85 of file TTreeProcessorMT.hxx.
|
private |
Names of the files.
Definition at line 82 of file TTreeProcessorMT.hxx.
|
private |
Definition at line 86 of file TTreeProcessorMT.hxx.
|
private |
Definition at line 96 of file TTreeProcessorMT.hxx.
|
staticprivate |
Definition at line 94 of file TTreeProcessorMT.hxx.
|
private |
! Thread pool for processing.
Definition at line 87 of file TTreeProcessorMT.hxx.
|
private |
Definition at line 100 of file TTreeProcessorMT.hxx.
|
private |
TTree names (always same size and ordering as fFileNames)
Definition at line 83 of file TTreeProcessorMT.hxx.
|
private |
Thread-local TreeViews.
Definition at line 91 of file TTreeProcessorMT.hxx.