Skip to content

Commit 9d147f0

Browse files
committed
[TDF] Showcase jitted filters
1 parent 6cdb5a0 commit 9d147f0

File tree

1 file changed

+12
-14
lines changed

1 file changed

+12
-14
lines changed

tutorials/dataframe/tdf101_h1Analysis.C

Lines changed: 12 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -22,21 +22,19 @@ auto Select = [](ROOT::Experimental::TDataFrame &dataFrame) {
2222
using Iarrayiew_t = std::array_view<int>;
2323

2424
auto ret =
25-
dataFrame.Filter([](float md0_d) { return TMath::Abs(md0_d - 1.8646) < 0.04; }, {"md0_d"})
26-
.Filter([](float ptds_d) { return ptds_d > 2.5; }, {"ptds_d"})
27-
.Filter([](float etads_d) { return TMath::Abs(etads_d) < 1.5; }, {"etads_d"})
28-
.Filter([](int ik, int ipi, std::array_view<int> nhitrp) { return nhitrp[ik - 1] * nhitrp[ipi - 1] > 1; },
25+
dataFrame.Filter("TMath::Abs(md0_d - 1.8646) < 0.04")
26+
.Filter("ptds_d > 2.5")
27+
.Filter("TMath::Abs(etads_d) < 1.5")
28+
.Filter([](int ik, int ipi, Iarrayiew_t nhitrp) { return nhitrp[ik - 1] * nhitrp[ipi - 1] > 1; },
2929
{"ik", "ipi", "nhitrp"})
30-
.Filter([](int ik, std::array_view<float> rstart,
31-
std::array_view<float> rend) { return rend[ik - 1] - rstart[ik - 1] > 22; },
32-
{"ik", "rstart", "rend"})
33-
.Filter([](int ipi, std::array_view<float> rstart,
34-
std::array_view<float> rend) { return rend[ipi - 1] - rstart[ipi - 1] > 22; },
35-
{"ipi", "rstart", "rend"})
36-
.Filter([](int ik, std::array_view<float> nlhk) { return nlhk[ik - 1] > 0.1; }, {"ik", "nlhk"})
37-
.Filter([](int ipi, std::array_view<float> nlhpi) { return nlhpi[ipi - 1] > 0.1; }, {"ipi", "nlhpi"})
38-
.Filter([](int ipis, std::array_view<float> nlhpi) { return nlhpi[ipis - 1] > 0.1; }, {"ipis", "nlhpi"})
39-
.Filter([](int njets) { return njets >= 1; }, {"njets"});
30+
.Filter([](int ik, Farrayiew_t rstart,
31+
Farrayiew_t rend) { return rend[ik - 1] - rstart[ik - 1] > 22; }, {"ik", "rstart", "rend"})
32+
.Filter([](int ipi, Farrayiew_t rstart,
33+
Farrayiew_t rend) { return rend[ipi - 1] - rstart[ipi - 1] > 22; }, {"ipi", "rstart", "rend"})
34+
.Filter([](int ik, Farrayiew_t nlhk) { return nlhk[ik - 1] > 0.1; }, {"ik", "nlhk"})
35+
.Filter([](int ipi, Farrayiew_t nlhpi) { return nlhpi[ipi - 1] > 0.1; }, {"ipi", "nlhpi"})
36+
.Filter([](int ipis, Farrayiew_t nlhpi) { return nlhpi[ipis - 1] > 0.1; }, {"ipis", "nlhpi"})
37+
.Filter("njets >= 1");
4038

4139
return ret;
4240
};

0 commit comments

Comments
 (0)