11#ifndef ROOT_RDF_TINTERFACE
12#define ROOT_RDF_TINTERFACE
49#include <initializer_list>
59#include <unordered_set>
82template <
typename Proxied,
typename DataSource>
85using RNode = RInterface<::ROOT::Detail::RDF::RNodeBase, void>;
113template <
typename Proxied,
typename DataSource =
void>
122 template <
typename T,
typename W>
152 template <typename T = Proxied, typename = std::enable_if_t<std::is_same<T, RLoopManager>::value,
int>>
219 template <typename F, std::enable_if_t<!std::is_convertible<F, std::string>::value,
int> = 0>
223 RDFInternal::CheckFilter(
f);
224 using ColTypes_t =
typename TTraits::CallableTraits<F>::arg_types;
225 constexpr auto nColumns = ColTypes_t::list_size;
243 template <typename F, std::enable_if_t<!std::is_convertible<F, std::string>::value,
int> = 0>
259 template <
typename F>
294 using BaseNodeType_t =
typename std::remove_pointer_t<
decltype(upcastNodeOnHeap)>::element_type;
296 const auto jittedFilter =
442 template <typename F, typename std::enable_if_t<!std::is_convertible<F, std::string>::value,
int> = 0>
445 return DefineImpl<F, RDFDetail::ExtraArgsForDefine::None>(
name, std::move(expression), columns,
"Define");
471 template <
typename F>
474 return DefineImpl<F, RDFDetail::ExtraArgsForDefine::Slot>(
name, std::move(expression), columns,
"DefineSlot");
501 template <
typename F>
504 return DefineImpl<F, RDFDetail::ExtraArgsForDefine::SlotAndEntry>(
name, std::move(expression), columns,
532 constexpr auto where =
"Define";
543 newCols.
AddDefine(std::move(jittedDefine));
561 template <typename F, std::enable_if_t<!std::is_convertible<F, std::string>::value,
int> = 0>
564 return DefineImpl<F, RDFDetail::ExtraArgsForDefine::None>(
name, std::move(expression), columns,
"Redefine");
580 template <
typename F>
583 return DefineImpl<F, RDFDetail::ExtraArgsForDefine::Slot>(
name, std::move(expression), columns,
"RedefineSlot");
599 template <
typename F>
602 return DefineImpl<F, RDFDetail::ExtraArgsForDefine::SlotAndEntry>(
name, std::move(expression), columns,
603 "RedefineSlotEntry");
622 constexpr auto where =
"Redefine";
633 newCols.
AddDefine(std::move(jittedDefine));
673 template <
typename T>
676 constexpr auto where{
"DefaultValueFor"};
688 if (retTypeName.empty()) {
692 retTypeName =
"CLING_UNKNOWN_TYPE_" + demangledType;
695 auto newColumn = std::make_shared<ROOT::Internal::RDF::RDefaultValueFor<T>>(
699 newCols.AddDefine(std::move(newColumn));
736 template <typename F, typename RetType_t = typename TTraits::CallableTraits<F>::ret_type>
744 if (retTypeName.empty()) {
748 retTypeName =
"CLING_UNKNOWN_TYPE_" + demangledType;
752 std::make_shared<RDFDetail::RDefinePerSample<F>>(
name, retTypeName, std::move(expression), *
fLoopManager);
755 newCols.AddDefine(std::move(newColumn));
810 newCols.
AddDefine(std::move(jittedDefine));
863 template <
typename F>
865 const std::vector<std::string> &variationTags, std::string_view variationName =
"")
867 std::vector<std::string> colNames{{std::string(colName)}};
868 const std::string theVariationName{variationName.empty() ? colName : variationName};
870 return VaryImpl<true>(std::move(colNames), std::forward<F>(expression), inputColumns, variationTags,
902 template <
typename F>
904 std::size_t nVariations, std::string_view variationName =
"")
906 R__ASSERT(nVariations > 0 &&
"Must have at least one variation.");
908 std::vector<std::string> variationTags;
909 variationTags.reserve(nVariations);
910 for (std::size_t i = 0u; i < nVariations; ++i)
911 variationTags.emplace_back(std::to_string(i));
913 const std::string theVariationName{variationName.empty() ? colName : variationName};
915 return Vary(colName, std::forward<F>(expression), inputColumns, std::move(variationTags), theVariationName);
950 template <
typename F>
953 const std::vector<std::string> &variationTags, std::string_view variationName)
955 return VaryImpl<false>(colNames, std::forward<F>(expression), inputColumns, variationTags, variationName);
972 template <
typename F>
975 const std::vector<std::string> &variationTags, std::string_view variationName)
977 return Vary(std::vector<std::string>(colNames), std::forward<F>(expression), inputColumns, variationTags, variationName);
1012 template <
typename F>
1015 std::size_t nVariations, std::string_view variationName)
1017 R__ASSERT(nVariations > 0 &&
"Must have at least one variation.");
1019 std::vector<std::string> variationTags;
1020 variationTags.reserve(nVariations);
1021 for (std::size_t i = 0u; i < nVariations; ++i)
1022 variationTags.emplace_back(std::to_string(i));
1024 return Vary(colNames, std::forward<F>(expression), inputColumns, std::move(variationTags), variationName);
1043 template <
typename F>
1046 std::size_t nVariations, std::string_view variationName)
1048 return Vary(std::vector<std::string>(colNames), std::forward<F>(expression), inputColumns, nVariations, variationName);
1077 const std::vector<std::string> &variationTags, std::string_view variationName =
"")
1079 std::vector<std::string> colNames{{std::string(colName)}};
1080 const std::string theVariationName{variationName.empty() ? colName : variationName};
1082 return JittedVaryImpl(colNames, expression, variationTags, theVariationName,
true);
1110 std::string_view variationName =
"")
1112 std::vector<std::string> variationTags;
1113 variationTags.reserve(nVariations);
1114 for (std::size_t i = 0u; i < nVariations; ++i)
1115 variationTags.emplace_back(std::to_string(i));
1117 return Vary(colName, expression, std::move(variationTags), variationName);
1147 std::size_t nVariations, std::string_view variationName)
1149 std::vector<std::string> variationTags;
1150 variationTags.reserve(nVariations);
1151 for (std::size_t i = 0u; i < nVariations; ++i)
1152 variationTags.emplace_back(std::to_string(i));
1154 return Vary(colNames, expression, std::move(variationTags), variationName);
1171 std::size_t nVariations, std::string_view variationName)
1173 return Vary(std::vector<std::string>(colNames), expression, nVariations, variationName);
1199 const std::vector<std::string> &variationTags, std::string_view variationName)
1201 return JittedVaryImpl(colNames, expression, variationTags, variationName,
false);
1225 constexpr auto where =
"Alias";
1233 newCols.AddAlias(alias, validColumnName);
1237 return newInterface;
1299 template <
typename... ColumnTypes>
1329 const auto pairOfColumnLists =
1331 std::move(colListNoAliases), std::move(colListNoPoundSizes));
1332 const auto &colListNoAliasesWithSizeBranches = pairOfColumnLists.first;
1333 const auto &colListWithAliasesAndSizeBranches = pairOfColumnLists.second;
1336 const auto fullTreeName = treename;
1338 treename = parsedTreePath.fTreeName;
1339 const auto &dirname = parsedTreePath.fDirName;
1341 auto snapHelperArgs = std::make_shared<RDFInternal::SnapshotHelperArgs>(
1342 RDFInternal::SnapshotHelperArgs{std::string(
filename), std::string(dirname), std::string(treename),
1343 colListWithAliasesAndSizeBranches, options});
1353 fullTreeName,
filename, colListNoAliasesWithSizeBranches,
false));
1355 auto resPtr = CreateAction<RDFInternal::ActionTags::Snapshot, RDFDetail::RInferredType>(
1356 colListNoAliasesWithSizeBranches, newRDF, snapHelperArgs,
fProxiedPtr, colListNoAliasesWithSizeBranches.size(),
1357 options.fVector2RVec);
1378 std::string_view columnNameRegexp =
"",
1387 std::copy_if(dsColumns.begin(), dsColumns.end(), std::back_inserter(dsColumnsWithoutSizeColumns),
1388 [](
const std::string &
name) { return name.size() < 13 || name.substr(0, 13) !=
"R_rdf_sizeof_"; });
1390 columnNames.reserve(definedColumns.size() + treeBranchNames.size() + dsColumnsWithoutSizeColumns.size());
1391 columnNames.insert(columnNames.end(), definedColumns.begin(), definedColumns.end());
1392 columnNames.insert(columnNames.end(), treeBranchNames.begin(), treeBranchNames.end());
1393 columnNames.insert(columnNames.end(), dsColumnsWithoutSizeColumns.begin(), dsColumnsWithoutSizeColumns.end());
1418 std::initializer_list<std::string> columnList,
1460 template <
typename... ColumnTypes>
1463 auto staticSeq = std::make_index_sequence<
sizeof...(ColumnTypes)>();
1464 return CacheImpl<ColumnTypes...>(columnList, staticSeq);
1477 if (columnList.empty()) {
1478 auto nEntries = *this->
Count();
1483 std::stringstream cacheCall;
1490 cacheCall <<
"*reinterpret_cast<ROOT::RDF::RInterface<ROOT::Detail::RDF::RLoopManager>*>("
1492 <<
") = reinterpret_cast<ROOT::RDF::RInterface<ROOT::Detail::RDF::RNodeBase>*>("
1497 const auto validColumnNames =
1501 for (
const auto &colType : colTypes)
1502 cacheCall << colType <<
", ";
1503 if (!columnListWithoutSizeColumns.empty())
1504 cacheCall.seekp(-2, cacheCall.cur);
1505 cacheCall <<
">(*reinterpret_cast<std::vector<std::string>*>("
1526 const auto treeBranchNames =
1531 std::copy_if(dsColumns.begin(), dsColumns.end(), std::back_inserter(dsColumnsWithoutSizeColumns),
1532 [](
const std::string &
name) { return name.size() < 13 || name.substr(0, 13) !=
"R_rdf_sizeof_"; });
1534 columnNames.reserve(definedColumns.size() + treeBranchNames.size() + dsColumns.size());
1535 columnNames.insert(columnNames.end(), definedColumns.begin(), definedColumns.end());
1536 columnNames.insert(columnNames.end(), treeBranchNames.begin(), treeBranchNames.end());
1537 columnNames.insert(columnNames.end(), dsColumns.begin(), dsColumns.end());
1539 return Cache(selectedColumns);
1551 return Cache(selectedColumns);
1575 if (stride == 0 || (end != 0 && end < begin))
1576 throw std::runtime_error(
"Range: stride must be strictly greater than 0 and end must be greater than begin.");
1580 auto rangePtr = std::make_shared<Range_t>(begin, end, stride,
fProxiedPtr);
1582 return newInterface;
1612 template <
typename F>
1615 using arg_types =
typename TTraits::CallableTraits<
decltype(
f)>::arg_types_nodecay;
1616 using ret_type =
typename TTraits::CallableTraits<
decltype(
f)>::ret_type;
1617 ForeachSlot(RDFInternal::AddSlotParameter<ret_type>(
f, arg_types()), columns);
1642 template <
typename F>
1646 constexpr auto nColumns = ColTypes_t::list_size;
1651 using Helper_t = RDFInternal::ForeachSlotHelper<F>;
1689 template <typename F, typename T = typename TTraits::CallableTraits<F>::ret_type>
1693 std::is_default_constructible<T>::value,
1694 "reduce object cannot be default-constructed. Please provide an initialisation value (redIdentity)");
1695 return Reduce(std::move(
f), columnName, T());
1712 template <typename F, typename T = typename TTraits::CallableTraits<F>::ret_type>
1734 auto cSPtr = std::make_shared<ULong64_t>(0);
1735 using Helper_t = RDFInternal::CountHelper;
1739 return MakeResultPtr(cSPtr, *
fLoopManager, std::move(action));
1762 template <
typename T,
typename COLL = std::vector<T>>
1770 using Helper_t = RDFInternal::TakeHelper<T, T, COLL>;
1772 auto valuesPtr = std::make_shared<COLL>();
1777 return MakeResultPtr(valuesPtr, *
fLoopManager, std::move(action));
1805 template <
typename V = RDFDetail::RInferredType>
1812 std::shared_ptr<::TH1D>
h(
nullptr);
1814 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
1815 h = model.GetHistogram();
1816 h->SetDirectory(
nullptr);
1819 if (
h->GetXaxis()->GetXmax() ==
h->GetXaxis()->GetXmin())
1820 RDFInternal::HistoUtils<::TH1D>::SetCanExtendAllAxes(*
h);
1821 return CreateAction<RDFInternal::ActionTags::Histo1D, V>(validatedColumns,
h,
h,
fProxiedPtr);
1841 template <
typename V = RDFDetail::RInferredType>
1844 const auto h_name = std::string(vName);
1845 const auto h_title = h_name +
";" + h_name +
";count";
1846 return Histo1D<V>({h_name.c_str(), h_title.c_str(), 128u, 0., 0.}, vName);
1867 template <
typename V = RDFDetail::RInferredType,
typename W = RDFDetail::RInferredType>
1870 const std::vector<std::string_view> columnViews = {vName, wName};
1874 std::shared_ptr<::TH1D>
h(
nullptr);
1876 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
1879 return CreateAction<RDFInternal::ActionTags::Histo1D, V, W>(userColumns,
h,
h,
fProxiedPtr);
1901 template <
typename V = RDFDetail::RInferredType,
typename W = RDFDetail::RInferredType>
1905 std::string str_vName{vName};
1906 std::string str_wName{wName};
1907 const auto h_name = str_vName +
"_weighted_" + str_wName;
1908 const auto h_title = str_vName +
", weights: " + str_wName +
";" + str_vName +
";count * " + str_wName;
1909 return Histo1D<V, W>({h_name.c_str(), h_title.c_str(), 128u, 0., 0.}, vName, wName);
1921 template <
typename V,
typename W>
1924 return Histo1D<V, W>(model,
"",
"");
1955 template <
typename V1 = RDFDetail::RInferredType,
typename V2 = RDFDetail::RInferredType>
1958 std::shared_ptr<::TH2D>
h(
nullptr);
1960 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
1963 if (!RDFInternal::HistoUtils<::TH2D>::HasAxisLimits(*
h)) {
1964 throw std::runtime_error(
"2D histograms with no axes limits are not supported yet.");
1966 const std::vector<std::string_view> columnViews = {v1Name, v2Name};
1970 return CreateAction<RDFInternal::ActionTags::Histo2D, V1, V2>(userColumns,
h,
h,
fProxiedPtr);
1999 Histo2D(
const TH2DModel &model, std::string_view v1Name, std::string_view v2Name, std::string_view wName)
2001 std::shared_ptr<::TH2D>
h(
nullptr);
2003 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2006 if (!RDFInternal::HistoUtils<::TH2D>::HasAxisLimits(*
h)) {
2007 throw std::runtime_error(
"2D histograms with no axes limits are not supported yet.");
2009 const std::vector<std::string_view> columnViews = {v1Name, v2Name, wName};
2013 return CreateAction<RDFInternal::ActionTags::Histo2D, V1, V2, W>(userColumns,
h,
h,
fProxiedPtr);
2016 template <
typename V1,
typename V2,
typename W>
2019 return Histo2D<V1, V2, W>(model,
"",
"",
"");
2052 std::string_view v3Name =
"")
2054 std::shared_ptr<::TH3D>
h(
nullptr);
2056 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2059 if (!RDFInternal::HistoUtils<::TH3D>::HasAxisLimits(*
h)) {
2060 throw std::runtime_error(
"3D histograms with no axes limits are not supported yet.");
2062 const std::vector<std::string_view> columnViews = {v1Name, v2Name, v3Name};
2066 return CreateAction<RDFInternal::ActionTags::Histo3D, V1, V2, V3>(userColumns,
h,
h,
fProxiedPtr);
2101 std::string_view v3Name, std::string_view wName)
2103 std::shared_ptr<::TH3D>
h(
nullptr);
2105 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2108 if (!RDFInternal::HistoUtils<::TH3D>::HasAxisLimits(*
h)) {
2109 throw std::runtime_error(
"3D histograms with no axes limits are not supported yet.");
2111 const std::vector<std::string_view> columnViews = {v1Name, v2Name, v3Name, wName};
2115 return CreateAction<RDFInternal::ActionTags::Histo3D, V1, V2, V3, W>(userColumns,
h,
h,
fProxiedPtr);
2118 template <
typename V1,
typename V2,
typename V3,
typename W>
2121 return Histo3D<V1, V2, V3, W>(model,
"",
"",
"",
"");
2146 template <
typename FirstColumn,
typename... OtherColumns>
2149 std::shared_ptr<::THnD>
h(
nullptr);
2151 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2154 if (
int(columnList.size()) == (
h->GetNdimensions() + 1)) {
2156 }
else if (
int(columnList.size()) !=
h->GetNdimensions()) {
2157 throw std::runtime_error(
"Wrong number of columns for the specified number of histogram axes.");
2160 return CreateAction<RDFInternal::ActionTags::HistoND, FirstColumn, OtherColumns...>(columnList,
h,
h,
2183 std::shared_ptr<::THnD>
h(
nullptr);
2185 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2188 if (
int(columnList.size()) == (
h->GetNdimensions() + 1)) {
2190 }
else if (
int(columnList.size()) !=
h->GetNdimensions()) {
2191 throw std::runtime_error(
"Wrong number of columns for the specified number of histogram axes.");
2194 return CreateAction<RDFInternal::ActionTags::HistoND, RDFDetail::RInferredType>(columnList,
h,
h,
fProxiedPtr,
2226 template <
typename X = RDFDetail::RInferredType,
typename Y = RDFDetail::RInferredType>
2229 auto graph = std::make_shared<::TGraph>();
2230 const std::vector<std::string_view> columnViews = {
x,
y};
2238 const auto g_name = validatedColumns[1] +
"_vs_" + validatedColumns[0];
2239 const auto g_title = validatedColumns[1] +
" vs " + validatedColumns[0];
2240 graph->SetNameTitle(g_name.c_str(), g_title.c_str());
2241 graph->GetXaxis()->SetTitle(validatedColumns[0].c_str());
2242 graph->GetYaxis()->SetTitle(validatedColumns[1].c_str());
2244 return CreateAction<RDFInternal::ActionTags::Graph, X, Y>(validatedColumns,
graph,
graph,
fProxiedPtr);
2281 std::string_view exh =
"", std::string_view eyl =
"", std::string_view eyh =
"")
2283 auto graph = std::make_shared<::TGraphAsymmErrors>();
2284 const std::vector<std::string_view> columnViews = {
x,
y, exl, exh, eyl, eyh};
2292 const auto g_name = validatedColumns[1] +
"_vs_" + validatedColumns[0];
2293 const auto g_title = validatedColumns[1] +
" vs " + validatedColumns[0];
2294 graph->SetNameTitle(g_name.c_str(), g_title.c_str());
2295 graph->GetXaxis()->SetTitle(validatedColumns[0].c_str());
2296 graph->GetYaxis()->SetTitle(validatedColumns[1].c_str());
2298 return CreateAction<RDFInternal::ActionTags::GraphAsymmErrors, X, Y, EXL, EXH, EYL, EYH>(validatedColumns,
graph,
2325 template <
typename V1 = RDFDetail::RInferredType,
typename V2 = RDFDetail::RInferredType>
2329 std::shared_ptr<::TProfile>
h(
nullptr);
2331 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2335 if (!RDFInternal::HistoUtils<::TProfile>::HasAxisLimits(*
h)) {
2336 throw std::runtime_error(
"Profiles with no axes limits are not supported yet.");
2338 const std::vector<std::string_view> columnViews = {v1Name, v2Name};
2342 return CreateAction<RDFInternal::ActionTags::Profile1D, V1, V2>(userColumns,
h,
h,
fProxiedPtr);
2374 std::shared_ptr<::TProfile>
h(
nullptr);
2376 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2380 if (!RDFInternal::HistoUtils<::TProfile>::HasAxisLimits(*
h)) {
2381 throw std::runtime_error(
"Profile histograms with no axes limits are not supported yet.");
2383 const std::vector<std::string_view> columnViews = {v1Name, v2Name, wName};
2387 return CreateAction<RDFInternal::ActionTags::Profile1D, V1, V2, W>(userColumns,
h,
h,
fProxiedPtr);
2393 template <
typename V1,
typename V2,
typename W>
2396 return Profile1D<V1, V2, W>(model,
"",
"",
"");
2429 std::string_view v2Name =
"", std::string_view v3Name =
"")
2431 std::shared_ptr<::TProfile2D>
h(
nullptr);
2433 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2437 if (!RDFInternal::HistoUtils<::TProfile2D>::HasAxisLimits(*
h)) {
2438 throw std::runtime_error(
"2D profiles with no axes limits are not supported yet.");
2440 const std::vector<std::string_view> columnViews = {v1Name, v2Name, v3Name};
2444 return CreateAction<RDFInternal::ActionTags::Profile2D, V1, V2, V3>(userColumns,
h,
h,
fProxiedPtr);
2477 std::string_view v3Name, std::string_view wName)
2479 std::shared_ptr<::TProfile2D>
h(
nullptr);
2481 ROOT::Internal::RDF::RIgnoreErrorLevelRAII iel(
kError);
2485 if (!RDFInternal::HistoUtils<::TProfile2D>::HasAxisLimits(*
h)) {
2486 throw std::runtime_error(
"2D profiles with no axes limits are not supported yet.");
2488 const std::vector<std::string_view> columnViews = {v1Name, v2Name, v3Name, wName};
2492 return CreateAction<RDFInternal::ActionTags::Profile2D, V1, V2, V3, W>(userColumns,
h,
h,
fProxiedPtr);
2497 template <
typename V1,
typename V2,
typename V3,
typename W>
2500 return Profile2D<V1, V2, V3, W>(model,
"",
"",
"",
"");
2540 auto h = std::make_shared<std::decay_t<T>>(std::forward<T>(model));
2541 if (!RDFInternal::HistoUtils<T>::HasAxisLimits(*
h)) {
2542 throw std::runtime_error(
"The absence of axes limits is not supported yet.");
2563 template <
typename V = RDFDetail::RInferredType>
2567 if (!
value.empty()) {
2568 columns.emplace_back(std::string(
value));
2571 if (std::is_same<V, RDFDetail::RInferredType>::value) {
2574 return Fill<V>(
TStatistic(), validColumnNames);
2595 template <
typename V = RDFDetail::RInferredType,
typename W = RDFDetail::RInferredType>
2599 constexpr auto vIsInferred = std::is_same<V, RDFDetail::RInferredType>::value;
2600 constexpr auto wIsInferred = std::is_same<W, RDFDetail::RInferredType>::value;
2606 if (vIsInferred && wIsInferred) {
2608 }
else if (vIsInferred != wIsInferred) {
2609 std::string error(
"The ");
2610 error += vIsInferred ?
"value " :
"weight ";
2611 error +=
"column type is explicit, while the ";
2612 error += vIsInferred ?
"weight " :
"value ";
2613 error +=
" is specified to be inferred. This case is not supported: please specify both types or none.";
2614 throw std::runtime_error(error);
2616 return Fill<V, W>(
TStatistic(), validColumnNames);
2641 template <
typename T = RDFDetail::RInferredType>
2645 using RetType_t = RDFDetail::MinReturnType_t<T>;
2646 auto minV = std::make_shared<RetType_t>(std::numeric_limits<RetType_t>::max());
2647 return CreateAction<RDFInternal::ActionTags::Min, T>(userColumns, minV, minV,
fProxiedPtr);
2671 template <
typename T = RDFDetail::RInferredType>
2675 using RetType_t = RDFDetail::MaxReturnType_t<T>;
2676 auto maxV = std::make_shared<RetType_t>(std::numeric_limits<RetType_t>::lowest());
2677 return CreateAction<RDFInternal::ActionTags::Max, T>(userColumns, maxV, maxV,
fProxiedPtr);
2700 template <
typename T = RDFDetail::RInferredType>
2704 auto meanV = std::make_shared<double>(0);
2705 return CreateAction<RDFInternal::ActionTags::Mean, T>(userColumns, meanV, meanV,
fProxiedPtr);
2728 template <
typename T = RDFDetail::RInferredType>
2732 auto stdDeviationV = std::make_shared<double>(0);
2733 return CreateAction<RDFInternal::ActionTags::StdDev, T>(userColumns, stdDeviationV, stdDeviationV,
fProxiedPtr);
2759 template <
typename T = RDFDetail::RInferredType>
2761 Sum(std::string_view columnName =
"",
2762 const RDFDetail::SumReturnType_t<T> &initValue = RDFDetail::SumReturnType_t<T>{})
2765 auto sumV = std::make_shared<RDFDetail::SumReturnType_t<T>>(initValue);
2766 return CreateAction<RDFInternal::ActionTags::Sum, T>(userColumns, sumV, sumV,
fProxiedPtr);
2796 bool returnEmptyReport =
false;
2803 returnEmptyReport =
true;
2805 auto rep = std::make_shared<RCutFlowReport>();
2806 using Helper_t = RDFInternal::ReportHelper<Proxied>;
2812 return MakeResultPtr(rep, *
fLoopManager, std::move(action));
2876 template <typename AccFun, typename MergeFun, typename R = typename TTraits::CallableTraits<AccFun>::ret_type,
2877 typename ArgTypes =
typename TTraits::CallableTraits<AccFun>::arg_types,
2878 typename ArgTypesNoDecay =
typename TTraits::CallableTraits<AccFun>::arg_types_nodecay,
2879 typename U = TTraits::TakeFirstParameter_t<ArgTypes>,
2880 typename T = TTraits::TakeFirstParameter_t<TTraits::RemoveFirstParameter_t<ArgTypes>>>
2883 RDFInternal::CheckAggregate<R, MergeFun>(ArgTypesNoDecay());
2889 auto accObjPtr = std::make_shared<U>(aggIdentity);
2890 using Helper_t = RDFInternal::AggregateHelper<AccFun, MergeFun, R, T, U>;
2892 auto action = std::make_unique<Action_t>(
2893 Helper_t(std::move(aggregator), std::move(merger), accObjPtr,
fLoopManager->GetNSlots()), validColumnNames,
2895 return MakeResultPtr(accObjPtr, *
fLoopManager, std::move(action));
2911 template <typename AccFun, typename MergeFun, typename R = typename TTraits::CallableTraits<AccFun>::ret_type,
2912 typename ArgTypes =
typename TTraits::CallableTraits<AccFun>::arg_types,
2913 typename U = TTraits::TakeFirstParameter_t<ArgTypes>,
2914 typename T = TTraits::TakeFirstParameter_t<TTraits::RemoveFirstParameter_t<ArgTypes>>>
2918 std::is_default_constructible<U>::value,
2919 "aggregated object cannot be default-constructed. Please provide an initialisation value (aggIdentity)");
2920 return Aggregate(std::move(aggregator), std::move(merger), columnName, U());
2988 using HelperT = std::decay_t<Helper>;
2991 static_assert(std::is_base_of<AH, HelperT>::value && std::is_convertible<HelperT *, AH *>::value,
2992 "Action helper of type T must publicly inherit from ROOT::Detail::RDF::RActionImpl<T>");
2994 auto hPtr = std::make_shared<HelperT>(std::forward<Helper>(helper));
2995 auto resPtr = hPtr->GetResultPtr();
2997 if (std::is_same<FirstColumn, RDFDetail::RInferredType>::value && columns.empty()) {
3000 return CreateAction<RDFInternal::ActionTags::Book, FirstColumn, OtherColumns...>(columns, resPtr, hPtr,
3030 template <
typename... ColumnTypes>
3034 auto newCols = columnList;
3035 newCols.insert(newCols.begin(),
"rdfentry_");
3036 auto displayer = std::make_shared<RDisplay>(newCols,
GetColumnTypeNamesList(newCols), nMaxCollectionElements);
3037 using displayHelperArgs_t = std::pair<size_t, std::shared_ptr<RDisplay>>;
3040 std::move(newCols), displayer, std::make_shared<displayHelperArgs_t>(nRows, displayer),
fProxiedPtr);
3057 auto newCols = columnList;
3058 newCols.insert(newCols.begin(),
"rdfentry_");
3059 auto displayer = std::make_shared<RDisplay>(newCols,
GetColumnTypeNamesList(newCols), nMaxCollectionElements);
3060 using displayHelperArgs_t = std::pair<size_t, std::shared_ptr<RDisplay>>;
3061 return CreateAction<RDFInternal::ActionTags::Display, RDFDetail::RInferredType>(
3062 std::move(newCols), displayer, std::make_shared<displayHelperArgs_t>(nRows, displayer),
fProxiedPtr,
3063 columnList.size() + 1);
3077 Display(std::string_view columnNameRegexp =
"",
size_t nRows = 5,
size_t nMaxCollectionElements = 10)
3081 return Display(selectedColumns, nRows, nMaxCollectionElements);
3093 Display(std::initializer_list<std::string> columnList,
size_t nRows = 5,
size_t nMaxCollectionElements = 10)
3096 return Display(selectedColumns, nRows, nMaxCollectionElements);
3100 template <typename F, typename DefineType, typename RetType = typename TTraits::CallableTraits<F>::ret_type>
3104 if (where.compare(0, 8,
"Redefine") != 0) {
3114 using ArgTypes_t =
typename TTraits::CallableTraits<F>::arg_types;
3116 std::is_same<DefineType, RDFDetail::ExtraArgsForDefine::Slot>::value, ArgTypes_t>
::type;
3118 std::is_same<DefineType, RDFDetail::ExtraArgsForDefine::SlotAndEntry>::value, ColTypesTmp_t>
::type;
3120 constexpr auto nColumns = ColTypes_t::list_size;
3127 if (retTypeName.empty()) {
3131 retTypeName =
"CLING_UNKNOWN_TYPE_" + demangledType;
3135 auto newColumn = std::make_shared<NewCol_t>(
name, retTypeName, std::forward<F>(expression), validColumnNames,
3139 newCols.
AddDefine(std::move(newColumn));
3143 return newInterface;
3149 template <typename F, typename DefineType, typename RetType = typename TTraits::CallableTraits<F>::ret_type,
3150 bool IsFStringConv = std::is_convertible<F, std::string>::value,
3151 bool IsRetTypeDefConstr = std::is_default_constructible<RetType>::value>
3152 std::enable_if_t<!IsFStringConv && !IsRetTypeDefConstr, RInterface<Proxied, DS_t>>
3155 static_assert(std::is_default_constructible<typename TTraits::CallableTraits<F>::ret_type>
::value,
3156 "Error in `Define`: type returned by expression is not default-constructible");
3160 template <
typename... ColumnTypes>
3168 const auto validCols =
GetValidatedColumnNames(columnListWithoutSizeColumns.size(), columnListWithoutSizeColumns);
3173 const auto &treename = parsedTreePath.fTreeName;
3174 const auto &dirname = parsedTreePath.fDirName;
3176 auto snapHelperArgs = std::make_shared<RDFInternal::SnapshotHelperArgs>(RDFInternal::SnapshotHelperArgs{
3177 std::string(
filename), std::string(dirname), std::string(treename), columnListWithoutSizeColumns, options});
3187 fullTreeName,
filename, columnListWithoutSizeColumns,
false));
3191 auto resPtr =
CreateAction<RDFInternal::ActionTags::Snapshot, ColumnTypes...>(validCols, newRDF, snapHelperArgs,
3201 template <
typename... ColTypes, std::size_t... S>
3207 constexpr bool areCopyConstructible =
3208 RDFInternal::TEvalAnd<std::is_copy_constructible<ColTypes>::value...>
::value;
3209 static_assert(areCopyConstructible,
"Columns of a type which is not copy constructible cannot be cached yet.");
3213 auto colHolders = std::make_tuple(Take<ColTypes>(columnListWithoutSizeColumns[S])...);
3214 auto ds = std::make_unique<
RLazyDS<ColTypes...>>(
3215 std::make_pair(columnListWithoutSizeColumns[S], std::get<S>(colHolders))...);
3222 template <
bool IsSingleColumn,
typename F>
3225 const std::vector<std::string> &variationTags, std::string_view variationName)
3227 using F_t = std::decay_t<F>;
3228 using ColTypes_t =
typename TTraits::CallableTraits<F_t>::arg_types;
3229 using RetType =
typename TTraits::CallableTraits<F_t>::ret_type;
3230 constexpr auto nColumns = ColTypes_t::list_size;
3232 SanityChecksForVary<RetType>(colNames, variationTags, variationName);
3238 if (retTypeName.empty()) {
3242 retTypeName =
"CLING_UNKNOWN_TYPE_" + demangledType;
3245 auto variation = std::make_shared<RDFInternal::RVariation<F_t, IsSingleColumn>>(
3250 newCols.AddVariation(std::move(variation));
3254 return newInterface;
3258 const std::vector<std::string> &variationTags,
3259 std::string_view variationName,
bool isSingleColumn)
3261 R__ASSERT(!variationTags.empty() &&
"Must have at least one variation.");
3262 R__ASSERT(!colNames.empty() &&
"Must have at least one varied column.");
3263 R__ASSERT(!variationName.empty() &&
"Must provide a variation name.");
3265 for (
auto &colName : colNames) {
3273 if (colNames.size() > 1) {
3274 std::set<std::string> uniqueCols(colNames.begin(), colNames.end());
3275 if (uniqueCols.size() != colNames.size())
3276 throw std::logic_error(
"A column name was passed to the same Vary invocation multiple times.");
3280 auto jittedVariation =
3285 newColRegister.
AddVariation(std::move(jittedVariation));
3289 return newInterface;
3292 template <
typename Helper,
typename ActionResultType>
3294 const std::shared_ptr<Helper> &hPtr,
3298 return CreateAction<RDFInternal::ActionTags::Book>({}, resPtr, hPtr,
fProxiedPtr, 0u);
3301 template <
typename Helper,
typename ActionResultType,
typename... Others>
3302 RResultPtr<ActionResultType>
3304 const std::shared_ptr<Helper>& ,
3307 throw std::logic_error(std::string(
"An action was booked with no input columns, but the action requires "
3308 "columns! The action helper type was ") +
3309 typeid(Helper).
name());
unsigned long long ULong64_t
#define R__ASSERT(e)
Checks condition e and reports a fatal error if it's false.
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
Base class for action helpers, see RInterface::Book() for more information.
implementation of FilterAvailable and FilterMissing operations
The head node of a RDF computation graph.
Helper class that provides the operation graph nodes.
A RDataFrame node that produces a result.
A binder for user-defined columns, variations and aliases.
std::vector< std::string_view > GenerateColumnNames() const
Return the list of the names of the defined columns (Defines + Aliases).
void AddVariation(std::shared_ptr< RVariationBase > variation)
Register a new systematic variation.
void AddDefine(std::shared_ptr< RDFDetail::RDefineBase > column)
Add a new defined column.
The dataset specification for RDataFrame.
virtual const std::vector< std::string > & GetColumnNames() const =0
Returns a reference to the collection of the dataset's column names.
ColumnNames_t GetValidatedColumnNames(const unsigned int nColumns, const ColumnNames_t &columns)
ColumnNames_t GetColumnTypeNamesList(const ColumnNames_t &columnList)
std::shared_ptr< ROOT::Detail::RDF::RLoopManager > fLoopManager
< The RLoopManager at the root of this computation graph. Never null.
RResultPtr< ActionResultType > CreateAction(const ColumnNames_t &columns, const std::shared_ptr< ActionResultType > &r, const std::shared_ptr< HelperArgType > &helperArg, const std::shared_ptr< RDFNode > &proxiedPtr, const int=-1)
Create RAction object, return RResultPtr for the action Overload for the case in which all column typ...
RDataSource * fDataSource
Non-owning pointer to a data-source object. Null if no data-source. RLoopManager has ownership of the...
void CheckAndFillDSColumns(ColumnNames_t validCols, TTraits::TypeList< ColumnTypes... > typeList)
void CheckIMTDisabled(std::string_view callerName)
ColumnNames_t GetColumnNames()
Returns the names of the available columns.
RDFInternal::RColumnRegister fColRegister
Contains the columns defined up to this node.
The public interface to the RDataFrame federation of classes.
RResultPtr<::THnD > HistoND(const THnDModel &model, const ColumnNames_t &columnList)
Fill and return an N-dimensional histogram (lazy action).
RInterface(const RInterface &)=default
Copy-ctor for RInterface.
RResultPtr<::TH1D > Histo1D(std::string_view vName, std::string_view wName)
Fill and return a one-dimensional histogram with the weighted values of a column (lazy action).
RInterface(const std::shared_ptr< Proxied > &proxied, RLoopManager &lm, const RDFInternal::RColumnRegister &colRegister)
RResultPtr<::TH1D > Histo1D(const TH1DModel &model={"", "", 128u, 0., 0.})
Fill and return a one-dimensional histogram with the weighted values of a column (lazy action).
RResultPtr<::TH2D > Histo2D(const TH2DModel &model)
RResultPtr<::TProfile > Profile1D(const TProfile1DModel &model, std::string_view v1Name="", std::string_view v2Name="")
Fill and return a one-dimensional profile (lazy action).
RResultPtr<::THnD > HistoND(const THnDModel &model, const ColumnNames_t &columnList)
Fill and return an N-dimensional histogram (lazy action).
std::enable_if_t<!IsFStringConv &&!IsRetTypeDefConstr, RInterface< Proxied, DS_t > > DefineImpl(std::string_view, F, const ColumnNames_t &, const std::string &)
RResultPtr< RInterface< RLoopManager > > Snapshot(std::string_view treename, std::string_view filename, std::string_view columnNameRegexp="", const RSnapshotOptions &options=RSnapshotOptions())
Save selected columns to disk, in a new TTree treename in file filename.
RResultPtr< TStatistic > Stats(std::string_view value="")
Return a TStatistic object, filled once per event (lazy action).
RInterface< Proxied, DS_t > Vary(std::string_view colName, F &&expression, const ColumnNames_t &inputColumns, std::size_t nVariations, std::string_view variationName="")
Register systematic variations for a single existing column using auto-generated variation tags.
RInterface< Proxied, DS_t > Vary(std::string_view colName, std::string_view expression, std::size_t nVariations, std::string_view variationName="")
Register systematic variations for a single existing column using auto-generated variation tags.
RResultPtr<::TGraph > Graph(std::string_view x="", std::string_view y="")
Fill and return a TGraph object (lazy action).
RResultPtr< ActionResultType > CallCreateActionWithoutColsIfPossible(const std::shared_ptr< ActionResultType > &, const std::shared_ptr< Helper > &, Others...)
RInterface< Proxied, DS_t > DefineSlot(std::string_view name, F expression, const ColumnNames_t &columns={})
Define a new column with a value dependent on the processing slot.
RResultPtr< double > StdDev(std::string_view columnName="")
Return the unbiased standard deviation of processed column values (lazy action).
std::enable_if_t< std::is_default_constructible< RetType >::value, RInterface< Proxied, DS_t > > DefineImpl(std::string_view name, F &&expression, const ColumnNames_t &columns, const std::string &where)
RInterface< Proxied, DS_t > DefinePerSample(std::string_view name, F expression)
Define a new column that is updated when the input sample changes.
RInterface & operator=(RInterface &&)=default
Move-assignment operator for RInterface.
RInterface< Proxied, DS_t > Vary(const std::vector< std::string > &colNames, F &&expression, const ColumnNames_t &inputColumns, std::size_t nVariations, std::string_view variationName)
Register systematic variations for multiple existing columns using auto-generated tags.
void ForeachSlot(F f, const ColumnNames_t &columns={})
Execute a user-defined function requiring a processing slot index on each entry (instant action).
RInterface< Proxied, DS_t > Vary(std::string_view colName, std::string_view expression, const std::vector< std::string > &variationTags, std::string_view variationName="")
Register systematic variations for a single existing column using custom variation tags.
RResultPtr< RDisplay > Display(const ColumnNames_t &columnList, size_t nRows=5, size_t nMaxCollectionElements=10)
Provides a representation of the columns in the dataset.
RInterface< RLoopManager > Cache(const ColumnNames_t &columnList)
Save selected columns in memory.
RInterface< Proxied, DS_t > Define(std::string_view name, F expression, const ColumnNames_t &columns={})
Define a new column.
RResultPtr< TStatistic > Stats(std::string_view value, std::string_view weight)
Return a TStatistic object, filled once per event (lazy action).
RInterface< Proxied, DS_t > Redefine(std::string_view name, std::string_view expression)
Overwrite the value and/or type of an existing column.
auto CallCreateActionWithoutColsIfPossible(const std::shared_ptr< ActionResultType > &resPtr, const std::shared_ptr< Helper > &hPtr, TTraits::TypeList< RDFDetail::RInferredType >) -> decltype(hPtr->Exec(0u), RResultPtr< ActionResultType >{})
RInterface< Proxied, DS_t > Vary(const std::vector< std::string > &colNames, std::string_view expression, std::size_t nVariations, std::string_view variationName)
Register systematic variations for multiple existing columns using auto-generated variation tags.
RResultPtr<::TH2D > Histo2D(const TH2DModel &model, std::string_view v1Name="", std::string_view v2Name="")
Fill and return a two-dimensional histogram (lazy action).
RResultPtr< RInterface< RLoopManager > > SnapshotImpl(std::string_view fullTreeName, std::string_view filename, const ColumnNames_t &columnList, const RSnapshotOptions &options)
RInterface< Proxied, DS_t > Vary(std::initializer_list< std::string > colNames, F &&expression, const ColumnNames_t &inputColumns, const std::vector< std::string > &variationTags, std::string_view variationName)
Register systematic variations for multiple existing columns using custom variation tags.
RResultPtr<::TProfile > Profile1D(const TProfile1DModel &model)
Fill and return a one-dimensional profile (lazy action).
RInterface(const std::shared_ptr< RLoopManager > &proxied)
Build a RInterface from a RLoopManager.
RInterface< RDFDetail::RFilter< F, Proxied >, DS_t > Filter(F f, const std::initializer_list< std::string > &columns)
Append a filter to the call graph.
RInterface< Proxied, DS_t > DefinePerSample(std::string_view name, std::string_view expression)
Define a new column that is updated when the input sample changes.
RResultPtr< double > Mean(std::string_view columnName="")
Return the mean of processed column values (lazy action).
RResultPtr< RInterface< RLoopManager > > Snapshot(std::string_view treename, std::string_view filename, std::initializer_list< std::string > columnList, const RSnapshotOptions &options=RSnapshotOptions())
Save selected columns to disk, in a new TTree treename in file filename.
RResultPtr< RDisplay > Display(std::initializer_list< std::string > columnList, size_t nRows=5, size_t nMaxCollectionElements=10)
Provides a representation of the columns in the dataset.
RInterface< Proxied, DS_t > Alias(std::string_view alias, std::string_view columnName)
Allow to refer to a column with a different name.
RInterface< RLoopManager > Cache(const ColumnNames_t &columnList)
Save selected columns in memory.
RInterface< Proxied, DS_t > Redefine(std::string_view name, F expression, const ColumnNames_t &columns={})
Overwrite the value and/or type of an existing column.
RInterface< RLoopManager > Cache(std::string_view columnNameRegexp="")
Save selected columns in memory.
RInterface< Proxied, DS_t > VaryImpl(const std::vector< std::string > &colNames, F &&expression, const ColumnNames_t &inputColumns, const std::vector< std::string > &variationTags, std::string_view variationName)
RResultPtr< typename std::decay_t< Helper >::Result_t > Book(Helper &&helper, const ColumnNames_t &columns={})
Book execution of a custom action using a user-defined helper object.
RResultPtr< RDisplay > Display(std::string_view columnNameRegexp="", size_t nRows=5, size_t nMaxCollectionElements=10)
Provides a representation of the columns in the dataset.
RInterface< RDFDetail::RFilterWithMissingValues< Proxied >, DS_t > FilterAvailable(std::string_view column)
Discard entries with missing values.
friend class RDFInternal::GraphDrawing::GraphCreatorHelper
RResultPtr<::TH2D > Histo2D(const TH2DModel &model, std::string_view v1Name, std::string_view v2Name, std::string_view wName)
Fill and return a weighted two-dimensional histogram (lazy action).
RInterface & operator=(const RInterface &)=default
Copy-assignment operator for RInterface.
RResultPtr< RDFDetail::SumReturnType_t< T > > Sum(std::string_view columnName="", const RDFDetail::SumReturnType_t< T > &initValue=RDFDetail::SumReturnType_t< T >{})
Return the sum of processed column values (lazy action).
RInterface< Proxied, DS_t > Vary(std::string_view colName, F &&expression, const ColumnNames_t &inputColumns, const std::vector< std::string > &variationTags, std::string_view variationName="")
Register systematic variations for a single existing column using custom variation tags.
RResultPtr< ULong64_t > Count()
Return the number of entries processed (lazy action).
RInterface< Proxied, DS_t > Vary(const std::vector< std::string > &colNames, std::string_view expression, const std::vector< std::string > &variationTags, std::string_view variationName)
Register systematic variations for multiple existing columns using custom variation tags.
RInterface< Proxied, DS_t > Define(std::string_view name, std::string_view expression)
Define a new column.
std::shared_ptr< Proxied > fProxiedPtr
Smart pointer to the graph node encapsulated by this RInterface.
RResultPtr<::TH1D > Histo1D(std::string_view vName)
Fill and return a one-dimensional histogram with the values of a column (lazy action).
RInterface< Proxied, DS_t > Vary(const std::vector< std::string > &colNames, F &&expression, const ColumnNames_t &inputColumns, const std::vector< std::string > &variationTags, std::string_view variationName)
Register systematic variations for multiple existing columns using custom variation tags.
RInterface< Proxied, DS_t > RedefineSlotEntry(std::string_view name, F expression, const ColumnNames_t &columns={})
Overwrite the value and/or type of an existing column.
RResultPtr<::TH1D > Histo1D(const TH1DModel &model, std::string_view vName, std::string_view wName)
Fill and return a one-dimensional histogram with the weighted values of a column (lazy action).
RInterface< RLoopManager > CacheImpl(const ColumnNames_t &columnList, std::index_sequence< S... >)
Implementation of cache.
RInterface< RDFDetail::RRange< Proxied >, DS_t > Range(unsigned int end)
Creates a node that filters entries based on range.
RInterface< RDFDetail::RFilterWithMissingValues< Proxied >, DS_t > FilterMissing(std::string_view column)
Keep only the entries that have missing values.
RResultPtr< COLL > Take(std::string_view column="")
Return a collection of values of a column (lazy action, returns a std::vector by default).
RInterface< RLoopManager > Cache(std::initializer_list< std::string > columnList)
Save selected columns in memory.
RResultPtr<::TProfile2D > Profile2D(const TProfile2DModel &model, std::string_view v1Name="", std::string_view v2Name="", std::string_view v3Name="")
Fill and return a two-dimensional profile (lazy action).
const std::shared_ptr< Proxied > & GetProxiedPtr() const
RInterface< Proxied, DS_t > JittedVaryImpl(const std::vector< std::string > &colNames, std::string_view expression, const std::vector< std::string > &variationTags, std::string_view variationName, bool isSingleColumn)
RResultPtr<::TH3D > Histo3D(const TH3DModel &model, std::string_view v1Name="", std::string_view v2Name="", std::string_view v3Name="")
Fill and return a three-dimensional histogram (lazy action).
RInterface< Proxied, DS_t > Vary(std::initializer_list< std::string > colNames, F &&expression, const ColumnNames_t &inputColumns, std::size_t nVariations, std::string_view variationName)
Register systematic variations for for multiple existing columns using custom variation tags.
RResultPtr< std::decay_t< T > > Fill(T &&model, const ColumnNames_t &columnList)
Return an object of type T on which T::Fill will be called once per event (lazy action).
RResultPtr< RInterface< RLoopManager > > Snapshot(std::string_view treename, std::string_view filename, const ColumnNames_t &columnList, const RSnapshotOptions &options=RSnapshotOptions())
Save selected columns to disk, in a new TTree treename in file filename.
RResultPtr< RDisplay > Display(const ColumnNames_t &columnList, size_t nRows=5, size_t nMaxCollectionElements=10)
Provides a representation of the columns in the dataset.
RResultPtr< RCutFlowReport > Report()
Gather filtering statistics.
RInterface< Proxied, DS_t > RedefineSlot(std::string_view name, F expression, const ColumnNames_t &columns={})
Overwrite the value and/or type of an existing column.
RResultPtr<::TProfile2D > Profile2D(const TProfile2DModel &model, std::string_view v1Name, std::string_view v2Name, std::string_view v3Name, std::string_view wName)
Fill and return a two-dimensional profile (lazy action).
RResultPtr<::TGraphAsymmErrors > GraphAsymmErrors(std::string_view x="", std::string_view y="", std::string_view exl="", std::string_view exh="", std::string_view eyl="", std::string_view eyh="")
Fill and return a TGraphAsymmErrors object (lazy action).
RResultPtr< RInterface< RLoopManager > > Snapshot(std::string_view treename, std::string_view filename, const ColumnNames_t &columnList, const RSnapshotOptions &options=RSnapshotOptions())
Save selected columns to disk, in a new TTree treename in file filename.
RResultPtr< U > Aggregate(AccFun aggregator, MergeFun merger, std::string_view columnName="")
Execute a user-defined accumulation operation on the processed column values in each processing slot.
RInterface< Proxied, DS_t > DefineSlotEntry(std::string_view name, F expression, const ColumnNames_t &columns={})
Define a new column with a value dependent on the processing slot and the current entry.
RResultPtr< RDFDetail::MinReturnType_t< T > > Min(std::string_view columnName="")
Return the minimum of processed column values (lazy action).
RResultPtr< T > Reduce(F f, std::string_view columnName="")
Execute a user-defined reduce operation on the values of a column.
void Foreach(F f, const ColumnNames_t &columns={})
Execute a user-defined function on each entry (instant action).
RInterface< RDFDetail::RJittedFilter, DS_t > Filter(std::string_view expression, std::string_view name="")
Append a filter to the call graph.
RResultPtr<::TProfile2D > Profile2D(const TProfile2DModel &model)
Fill and return a two-dimensional profile (lazy action).
RInterface< RDFDetail::RFilter< F, Proxied >, DS_t > Filter(F f, const ColumnNames_t &columns={}, std::string_view name="")
Append a filter to the call graph.
RResultPtr< U > Aggregate(AccFun aggregator, MergeFun merger, std::string_view columnName, const U &aggIdentity)
Execute a user-defined accumulation operation on the processed column values in each processing slot.
RInterface(RInterface &&)=default
Move-ctor for RInterface.
RResultPtr< T > Reduce(F f, std::string_view columnName, const T &redIdentity)
Execute a user-defined reduce operation on the values of a column.
RResultPtr<::TH3D > Histo3D(const TH3DModel &model, std::string_view v1Name, std::string_view v2Name, std::string_view v3Name, std::string_view wName)
Fill and return a three-dimensional histogram (lazy action).
RInterface< Proxied, DS_t > DefaultValueFor(std::string_view column, const T &defaultValue)
In case the value in the given column is missing, provide a default value.
RInterface< RDFDetail::RFilter< F, Proxied >, DS_t > Filter(F f, std::string_view name)
Append a filter to the call graph.
RInterface< RDFDetail::RRange< Proxied >, DS_t > Range(unsigned int begin, unsigned int end, unsigned int stride=1)
Creates a node that filters entries based on range: [begin, end).
std::vector< std::string > GetFilterNames()
Returns the names of the filters created.
RResultPtr<::TH1D > Histo1D(const TH1DModel &model={"", "", 128u, 0., 0.}, std::string_view vName="")
Fill and return a one-dimensional histogram with the values of a column (lazy action).
RResultPtr<::TProfile > Profile1D(const TProfile1DModel &model, std::string_view v1Name, std::string_view v2Name, std::string_view wName)
Fill and return a one-dimensional profile (lazy action).
RResultPtr<::TH3D > Histo3D(const TH3DModel &model)
RResultPtr< RDFDetail::MaxReturnType_t< T > > Max(std::string_view columnName="")
Return the maximum of processed column values (lazy action).
RInterface< Proxied, DS_t > Vary(std::initializer_list< std::string > colNames, std::string_view expression, std::size_t nVariations, std::string_view variationName)
Register systematic variations for multiple existing columns using auto-generated variation tags.
A RDataSource implementation which is built on top of result proxies.
Smart pointer for the return type of actions.
ROOT's RDataFrame offers a modern, high-level interface for analysis of data stored in TTree ,...
typename RemoveFirstParameter< T >::type RemoveFirstParameter_t
TDirectory::TContext keeps track and restore the current directory.
A TGraph is an object made of two arrays X and Y with npoints each.
Statistical variable, defined by its mean and variance (RMS).
std::shared_ptr< ROOT::Detail::RDF::RLoopManager > CreateLMFromTTree(std::string_view datasetName, std::string_view fileNameGlob, const std::vector< std::string > &defaultColumns, bool checkFile=true)
Create an RLoopManager that reads a TChain.
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)
void CheckForRedefinition(const std::string &where, std::string_view definedColView, const RColumnRegister &colRegister, const ColumnNames_t &treeColumns, const ColumnNames_t &dataSourceColumns)
Throw if column definedColView is already there.
void CheckForDefinition(const std::string &where, std::string_view definedColView, const RColumnRegister &colRegister, const ColumnNames_t &treeColumns, const ColumnNames_t &dataSourceColumns)
Throw if column definedColView is not already there.
void ChangeEmptyEntryRange(const ROOT::RDF::RNode &node, std::pair< ULong64_t, ULong64_t > &&newRange)
std::shared_ptr< RJittedDefine > BookDefineJit(std::string_view name, std::string_view expression, RLoopManager &lm, RDataSource *ds, const RColumnRegister &colRegister, const ColumnNames_t &branches, std::shared_ptr< RNodeBase > *upcastNodeOnHeap)
Book the jitting of a Define call.
void CheckValidCppVarName(std::string_view var, const std::string &where)
void ChangeSpec(const ROOT::RDF::RNode &node, ROOT::RDF::Experimental::RDatasetSpec &&spec)
Changes the input dataset specification of an RDataFrame.
void RemoveDuplicates(ColumnNames_t &columnNames)
std::shared_ptr< RNodeBase > UpcastNode(std::shared_ptr< RNodeBase > ptr)
std::string TypeID2TypeName(const std::type_info &id)
Returns the name of a type starting from its type_info An empty string is returned in case of failure...
std::vector< std::string > GetFilterNames(const std::shared_ptr< RLoopManager > &loopManager)
std::string GetDataSourceLabel(const ROOT::RDF::RNode &node)
std::string PrettyPrintAddr(const void *const addr)
void TriggerRun(ROOT::RDF::RNode node)
Trigger the execution of an RDataFrame computation graph.
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::shared_ptr< RDFDetail::RJittedFilter > BookFilterJit(std::shared_ptr< RDFDetail::RNodeBase > *prevNodeOnHeap, std::string_view name, std::string_view expression, const ColumnNames_t &branches, const RColumnRegister &colRegister, TTree *tree, RDataSource *ds)
Book the jitting of a Filter call.
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::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, const ColumnNames_t &branches, 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)
std::pair< std::vector< std::string >, std::vector< std::string > > AddSizeBranches(const std::vector< std::string > &branches, TTree *tree, std::vector< std::string > &&colsWithoutAliases, std::vector< std::string > &&colsWithAliases)
Return copies of colsWithoutAliases and colsWithAliases with size branches for variable-sized array b...
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 ChangeBeginAndEndEntries(const RNode &node, Long64_t begin, Long64_t end)
std::vector< std::string > GetTopLevelBranchNames(TTree &t)
Get all the top-level branches names, including the ones of the friend trees.
RInterface<::ROOT::Detail::RDF::RNodeBase, void > RNode
std::vector< std::string > ColumnNames_t
ROOT type_traits extensions.
tbb::task_arena is an alias of tbb::interface7::task_arena, which doesn't allow to forward declare tb...
void EnableImplicitMT(UInt_t numthreads=0)
Enable ROOT's implicit multi-threading for all objects and methods that provide an internal paralleli...
Bool_t IsImplicitMTEnabled()
Returns true if the implicit multi-threading in ROOT is enabled.
void DisableImplicitMT()
Disables the implicit multi-threading in ROOT (see EnableImplicitMT).
type is TypeList if MustRemove is false, otherwise it is a TypeList with the first type removed
A collection of options to steer the creation of the dataset on file.
bool fLazy
Do not start the event loop when Snapshot is called.
A struct which stores the parameters of a TH1D.
std::shared_ptr<::TH1D > GetHistogram() const
A struct which stores the parameters of a TH2D.
std::shared_ptr<::TH2D > GetHistogram() const
A struct which stores the parameters of a TH3D.
std::shared_ptr<::TH3D > GetHistogram() const
A struct which stores the parameters of a THnD.
std::shared_ptr<::THnD > GetHistogram() const
A struct which stores the parameters of a TProfile.
std::shared_ptr<::TProfile > GetProfile() const
A struct which stores the parameters of a TProfile2D.
std::shared_ptr<::TProfile2D > GetProfile() const
Lightweight storage for a collection of types.