11#ifndef ROOT_RDF_TINTERFACE_UTILS
12#define ROOT_RDF_TINTERFACE_UTILS
52template<
typename T,
typename V>
93struct GraphAsymmErrors{};
108template <
typename T,
bool ISV6HISTO = std::is_base_of<TH1, std::decay_t<T>>::value>
113 auto xaxis =
h.GetXaxis();
114 return !(
xaxis->GetXmin() == 0. &&
xaxis->GetXmax() == 0.);
126std::unique_ptr<RActionBase>
127BuildAction(
const ColumnNames_t &
bl,
const std::shared_ptr<ActionResultType> &
h,
const unsigned int nSlots,
136template <
typename...
ColTypes,
typename PrevNodeType>
137std::unique_ptr<RActionBase>
138BuildAction(
const ColumnNames_t &
bl,
const std::shared_ptr<::TH1D> &
h,
const unsigned int nSlots,
139 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Histo1D,
const RColumnRegister &
colRegister)
156std::unique_ptr<RActionBase>
157BuildAction(
const ColumnNames_t &
bl,
const std::shared_ptr<ActionResultType> &
h,
const unsigned int nSlots,
158 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Histo3D,
const RColumnRegister &
colRegister)
167 if constexpr (
sizeof...(ColTypes) > 3) {
175template <
typename...
ColTypes,
typename PrevNodeType>
176std::unique_ptr<RActionBase>
177BuildAction(
const ColumnNames_t &
bl,
const std::shared_ptr<TGraph> &
g,
const unsigned int nSlots,
178 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Graph,
const RColumnRegister &
colRegister)
185template <
typename...
ColTypes,
typename PrevNodeType>
186std::unique_ptr<RActionBase>
187BuildAction(
const ColumnNames_t &
bl,
const std::shared_ptr<TGraphAsymmErrors> &
g,
const unsigned int nSlots,
188 std::shared_ptr<PrevNodeType> prevNode, ActionTags::GraphAsymmErrors,
const RColumnRegister &
colRegister)
196template <
typename ColType,
typename PrevNodeType,
typename ActionResultType>
197std::unique_ptr<RActionBase>
199 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Min,
const RColumnRegister &
colRegister)
207template <
typename ColType,
typename PrevNodeType,
typename ActionResultType>
208std::unique_ptr<RActionBase>
210 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Max,
const RColumnRegister &
colRegister)
218template <
typename ColType,
typename PrevNodeType,
typename ActionResultType>
219std::unique_ptr<RActionBase>
221 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Sum,
const RColumnRegister &
colRegister)
229template <
typename ColType,
typename PrevNodeType>
230std::unique_ptr<RActionBase>
232 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Mean,
const RColumnRegister &
colRegister)
240template <
typename ColType,
typename PrevNodeType>
241std::unique_ptr<RActionBase>
243 std::shared_ptr<PrevNodeType> prevNode, ActionTags::StdDev,
const RColumnRegister &
colRegister)
253template <
typename...
ColTypes,
typename PrevNodeType>
254std::unique_ptr<RActionBase>
255BuildAction(
const ColumnNames_t &
bl,
const std::shared_ptr<displayHelperArgs_t> &
helperArgs,
const unsigned int,
256 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Display,
const RColumnRegister &
colRegister)
265 std::string fFileName;
266 std::string fDirName;
267 std::string fTreeName;
275template <
typename PrevNodeType>
276std::unique_ptr<RActionBase>
278 const unsigned int nSlots, std::shared_ptr<PrevNodeType> prevNode,
const RColumnRegister &
colRegister,
279 const std::vector<const std::type_info *> &
colTypeIDs)
291 for (
auto i = 0
u; i <
sz; ++i)
298 using Helper_t = UntypedSnapshotRNTupleHelper;
307 throw std::runtime_error(
"Snapshot: Snapshotting to RNTuple with IMT enabled is not supported yet.");
312 using Helper_t = UntypedSnapshotTTreeHelper;
319 using Helper_t = UntypedSnapshotTTreeHelperMT;
332std::unique_ptr<RActionBase>
333BuildAction(
const ColumnNames_t &
bl,
const std::shared_ptr<Helper_t> &
h,
const unsigned int ,
334 std::shared_ptr<PrevNodeType> prevNode, ActionTags::Book,
const RColumnRegister &
colRegister)
342template <
typename Filter>
345 using FilterRet_t =
typename RDF::CallableTraits<Filter>::ret_type;
346 static_assert(std::is_convertible<FilterRet_t, bool>::value,
347 "filter expression returns a type that is not convertible to bool");
366 std::string_view expression,
const RColumnRegister &
colRegister,
377std::shared_ptr<RJittedVariation>
383std::string
JitBuildAction(
const ColumnNames_t &
bl, std::shared_ptr<RDFDetail::RNodeBase> *prevNode,
384 const std::type_info &
art,
const std::type_info &at,
void *
rOnHeap,
TTree *tree,
397 return new std::weak_ptr<T>(
shPtr);
404 return new std::shared_ptr<T>(
shPtr);
411std::shared_ptr<RNodeBase>
UpcastNode(std::shared_ptr<RNodeBase> ptr);
429 if (
lm.HasDataSourceColumnReaders(
colName,
typeid(T)))
433 lm.GetSuppressErrorsForMissingBranches().find(
colName) ==
lm.GetSuppressErrorsForMissingBranches().
end())
437 std::vector<std::unique_ptr<RColumnReaderBase>>
colReaders;
472template <
typename F,
typename PrevNode>
494 using ColTypes_t =
typename TTraits::CallableTraits<Callable_t>::arg_types;
495 constexpr auto nColumns = ColTypes_t::list_size;
499 auto ds =
lm.GetDataSource();
530 return std::unique_ptr<RDefineBase>(
537template <
typename RDefineTypeTag,
typename F>
562 using ColTypes_t =
typename TTraits::CallableTraits<Callable_t>::arg_types;
564 auto ds =
lm->GetDataSource();
572 std::unique_ptr<RDefineBase>
newCol{
579template <
bool IsSingleColumn,
typename F>
611 using ColTypes_t =
typename TTraits::CallableTraits<Callable_t>::arg_types;
613 auto ds =
lm->GetDataSource();
658 constexpr auto nColumns = ColTypes_t::list_size;
683 using type = TTraits::TakeFirstParameter_t<T>;
687template <
typename R,
typename F,
typename... Args>
690 return [
f](
unsigned int, Args...
a)
mutable ->
R {
return f(
a...); };
698template <
typename...
Rest>
700 static constexpr bool value =
true;
705 static constexpr bool value =
false;
710 static constexpr bool value =
true;
713template <
typename ...ColTypes>
720 static constexpr bool value =
false;
727template <
typename R,
typename Merge,
typename U,
typename T,
typename decayedU = std::decay_t<U>,
728 typename mergeArgsNoDecay_t =
typename CallableTraits<Merge>::arg_types_nodecay,
729 typename mergeArgs_t =
typename CallableTraits<Merge>::arg_types,
730 typename mergeRet_t =
typename CallableTraits<Merge>::ret_type>
734 (std::is_same<R, decayedU>::value) || (std::is_same<R, void>::value && std::is_lvalue_reference<U>::value);
735 static_assert(
isAggregatorOk,
"aggregator function must have signature `U(U,T)` or `void(U&,T)`");
737 (std::is_same<TypeList<decayedU, decayedU>,
mergeArgs_t>
::value && std::is_same<decayedU, mergeRet_t>::value) ||
739 std::is_same<void, mergeRet_t>::value);
740 static_assert(
isMergeOk,
"merge function must have signature `U(U,U)` or `void(std::vector<U>&)`");
745template <
typename R,
typename T>
748 static_assert(
sizeof(T) == 0,
"aggregator function must take exactly two arguments");
766template <
typename NodeType>
767std::vector<std::string>
GetFilterNames(
const std::shared_ptr<NodeType> &node)
775 std::string fTreeName;
776 std::string fDirName;
797struct IsList_t : std::false_type {};
800struct IsList_t<std::list<T>> : std::true_type {};
806struct IsDeque_t<std::deque<T>> : std::true_type {};
816template <
typename Elem>
824std::pair<std::vector<std::string>, std::vector<std::string>>
ROOT::Detail::TRangeCast< T, true > TRangeDynCast
TRangeDynCast is an adapter class that allows the typed iteration through a TCollection.
Int_t gErrorIgnoreLevel
errors with level below this value will be ignored. Default is kUnset.
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char filename
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void value
Option_t Option_t TPoint TPoint const char GetTextMagnitude GetFillStyle GetLineColor GetLineWidth GetMarkerStyle GetTextAlign GetTextColor GetTextSize void char Point_t Rectangle_t WindowAttributes_t Float_t Float_t Float_t Int_t Int_t UInt_t UInt_t Rectangle_t Int_t Int_t Window_t TString Int_t GCValues_t GetPrimarySelectionOwner GetDisplay GetScreen GetColormap GetNativeEvent const char const char dpyName wid window const char font_name cursor keysym reg const char only_if_exist regb h Point_t winding char text const char depth char const char Int_t count const char ColorStruct_t color const char Pixmap_t Pixmap_t PictureAttributes_t attr const char char ret_data h unsigned char height h Atom_t Int_t ULong_t ULong_t unsigned char prop_list Atom_t Atom_t Atom_t Time_t type
The head node of a RDF computation graph.
A binder for user-defined columns, variations and aliases.
RDataSource defines an API that RDataFrame can use to read arbitrary data formats.
The public interface to the RDataFrame federation of classes.
const_iterator end() const
A "std::vector"-like collection of values implementing handy operation to analyse them.
A TTree represents a columnar dataset.
const ColumnNames_t SelectColumns(unsigned int nRequiredNames, const ColumnNames_t &names, const ColumnNames_t &defaultNames)
Choose between local column names or default column names, throw in case of errors.
void CheckForNoVariations(const std::string &where, std::string_view definedColView, const RColumnRegister &colRegister)
Throw if the column has systematic variations attached.
ParsedTreePath ParseTreePath(std::string_view fullTreeName)
std::shared_ptr< RJittedDefine > BookDefineJit(std::string_view name, std::string_view expression, RLoopManager &lm, RDataSource *ds, const RColumnRegister &colRegister, std::shared_ptr< RNodeBase > *upcastNodeOnHeap)
Book the jitting of a Define call.
void CheckValidCppVarName(std::string_view var, const std::string &where)
void RemoveDuplicates(ColumnNames_t &columnNames)
ColumnNames_t GetValidatedColumnNames(RLoopManager &lm, const unsigned int nColumns, const ColumnNames_t &columns, const RColumnRegister &colRegister, RDataSource *ds)
Given the desired number of columns and the user-provided list of columns:
std::shared_ptr< RNodeBase > UpcastNode(std::shared_ptr< RNodeBase > ptr)
std::shared_ptr< RDFDetail::RJittedFilter > BookFilterJit(std::shared_ptr< RDFDetail::RNodeBase > *prevNodeOnHeap, std::string_view name, std::string_view expression, const RColumnRegister &colRegister, TTree *tree, RDataSource *ds)
Book the jitting of a Filter call.
void CheckForDefinition(const std::string &where, std::string_view definedColView, const RColumnRegister &colRegister, const ColumnNames_t &dataSourceColumns)
Throw if column definedColView is not already there.
std::vector< std::string > GetFilterNames(const std::shared_ptr< RLoopManager > &loopManager)
std::string PrettyPrintAddr(const void *const addr)
void CheckTypesAndPars(unsigned int nTemplateParams, unsigned int nColumnNames)
std::string DemangleTypeIdName(const std::type_info &typeInfo)
bool AtLeastOneEmptyString(const std::vector< std::string_view > strings)
std::unique_ptr< ROOT::Detail::RDF::RColumnReaderBase > CreateColumnReader(ROOT::RDF::RDataSource &ds, unsigned int slot, std::string_view col, const std::type_info &tid, TTreeReader *treeReader)
std::pair< std::vector< std::string >, std::vector< std::string > > AddSizeBranches(ROOT::RDF::RDataSource *ds, std::vector< std::string > &&colsWithoutAliases, std::vector< std::string > &&colsWithAliases)
Return copies of colsWithoutAliases and colsWithAliases with size branches for variable-sized array b...
void RemoveRNTupleSubFields(ColumnNames_t &columnNames)
ColumnNames_t FilterArraySizeColNames(const ColumnNames_t &columnNames, const std::string &action)
Take a list of column names, return that list with entries starting by '#' filtered out.
std::vector< std::string > GetValidatedArgTypes(const ColumnNames_t &colNames, const RColumnRegister &colRegister, TTree *tree, RDataSource *ds, const std::string &context, bool vector2RVec)
std::shared_ptr< RJittedVariation > BookVariationJit(const std::vector< std::string > &colNames, std::string_view variationName, const std::vector< std::string > &variationTags, std::string_view expression, RLoopManager &lm, RDataSource *ds, const RColumnRegister &colRegister, std::shared_ptr< RNodeBase > *upcastNodeOnHeap, bool isSingleColumn)
Book the jitting of a Vary call.
void CheckForDuplicateSnapshotColumns(const ColumnNames_t &cols)
ColumnNames_t ConvertRegexToColumns(const ColumnNames_t &colNames, std::string_view columnNameRegexp, std::string_view callerName)
ColumnNames_t FindUnknownColumns(const ColumnNames_t &requiredCols, const RColumnRegister &definedCols, const ColumnNames_t &dataSourceColumns)
std::vector< bool > FindUndefinedDSColumns(const ColumnNames_t &requestedCols, const ColumnNames_t &definedCols)
Return a bitset each element of which indicates whether the corresponding element in selectedColumns ...
unsigned int & NThreadPerTH3()
Obtain or set the number of threads that will share a clone of a thread-safe 3D histogram.
std::shared_ptr< RJittedDefine > BookDefinePerSampleJit(std::string_view name, std::string_view expression, RLoopManager &lm, const RColumnRegister &colRegister, std::shared_ptr< RNodeBase > *upcastNodeOnHeap)
Book the jitting of a DefinePerSample call.
void CheckForRedefinition(const std::string &where, std::string_view definedColView, const RColumnRegister &colRegister, const ColumnNames_t &dataSourceColumns)
Throw if column definedColView is already there.
std::string JitBuildAction(const ColumnNames_t &cols, std::shared_ptr< RDFDetail::RNodeBase > *prevNode, const std::type_info &helperArgType, const std::type_info &at, void *helperArgOnHeap, TTree *tree, const unsigned int nSlots, const RColumnRegister &colRegister, RDataSource *ds, std::weak_ptr< RJittedAction > *jittedActionOnHeap, const bool vector2RVec)
ROOT type_traits extensions.
Namespace for new ROOT classes and functions.
Bool_t IsImplicitMTEnabled()
Returns true if the implicit multi-threading in ROOT is enabled.
A collection of options to steer the creation of the dataset on file.
Lightweight storage for a collection of types.