Skip to content

Commit de773b0

Browse files
authored
Add files via upload
1 parent 283c52a commit de773b0

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

PWGDQ/Tasks/dqEfficiency_withAssoc_direct.cxx

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,7 @@
5656
#include <iostream>
5757
#include <map>
5858
#include <memory>
59+
#include <random>
5960
#include <string>
6061
#include <utility>
6162
#include <vector>
@@ -2791,7 +2792,12 @@ struct AnalysisDileptonTrack {
27912792
if (!event.isEventSelected_bit(0)) {
27922793
continue;
27932794
}
2794-
if (fConfigOptions.fConfigSavelessevents.value > 0 && event.globalIndex() % fConfigOptions.fConfigSavelessevents == 0)
2795+
std::hash<uint64_t> hasher;
2796+
size_t seed = hasher(static_cast<uint64_t>(event.globalIndex()));
2797+
std::mt19937 gen(seed);
2798+
std::uniform_real_distribution<float> dist(0.0, 1.0);
2799+
float randomVal = dist(gen);
2800+
if (randomVal < fConfigOptions.fConfigSavelessevents)
27952801
continue;
27962802
auto groupedBarrelAssocs = assocs.sliceBy(trackAssocsPerCollision, event.globalIndex());
27972803
// groupedBarrelAssocs.bindInternalIndicesTo(&assocs);
@@ -2986,7 +2992,12 @@ struct AnalysisDileptonTrack {
29862992
if (!event.has_mcCollision()) {
29872993
continue;
29882994
}
2989-
if (fConfigOptions.fConfigSavelessevents.value > 0 && event.globalIndex() % fConfigOptions.fConfigSavelessevents == 0)
2995+
std::hash<uint64_t> hasher;
2996+
size_t seed = hasher(static_cast<uint64_t>(event.globalIndex()));
2997+
std::mt19937 gen(seed);
2998+
std::uniform_real_distribution<float> dist(0.0, 1.0);
2999+
float randomVal = dist(gen);
3000+
if (randomVal < fConfigOptions.fConfigSavelessevents)
29903001
continue;
29913002
runEnergyCorrelators<VarManager::kJpsiHadronMass>(event, mcTracks);
29923003
}

0 commit comments

Comments
 (0)