Skip to content

Commit 1199cba

Browse files
jaimenormanjaimenorman
authored
[PWGJE] add MB gap event rejection for tracks and collisions (#9484)
1 parent 23a1203 commit 1199cba

File tree

3 files changed

+52
-2
lines changed

3 files changed

+52
-2
lines changed

‎PWGJE/Tasks/jetFinderFullQA.cxx‎

Lines changed: 11 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,6 +74,7 @@ struct JetFinderFullQATask {
7474
Configurable<float> pTHatMaxMCD{"pTHatMaxMCD", 999.0, "maximum fraction of hard scattering for jet acceptance in detector MC"};
7575
Configurable<float> pTHatMaxMCP{"pTHatMaxMCP", 999.0, "maximum fraction of hard scattering for jet acceptance in particle MC"};
7676
Configurable<float> pTHatExponent{"pTHatExponent", 6.0, "exponent of the event weight for the calculation of pTHat"};
77+
Configurable<bool> skipMBGapEvents{"skipMBGapEvents", false, "flag to choose to reject min. bias gap events; jet-level rejection applied at the jet finder level, here rejection is applied for collision and track process functions"};
7778

7879
std::vector<bool> filledJetR;
7980
std::vector<double> jetRadiiValues;
@@ -566,7 +567,9 @@ struct JetFinderFullQATask {
566567

567568
void processMCCollisionsWeighted(aod::JetMcCollision const& collision)
568569
{
569-
570+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
571+
return;
572+
}
570573
registry.fill(HIST("h_collision_eventweight_part"), collision.weight());
571574
}
572575
PROCESS_SWITCH(JetFinderFullQATask, processMCCollisionsWeighted, "collision QA for weighted events", false);
@@ -575,6 +578,9 @@ struct JetFinderFullQATask {
575578
soa::Filtered<aod::JetTracks> const& tracks,
576579
soa::Filtered<aod::JetClusters> const& clusters)
577580
{
581+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
582+
return;
583+
}
578584
registry.fill(HIST("h_collisions"), 0.5);
579585
registry.fill(HIST("h_centrality_collisions"), collision.centrality(), 0.5);
580586
if (!jetderiveddatautilities::eventEMCAL(collision)) {
@@ -586,12 +592,15 @@ struct JetFinderFullQATask {
586592
}
587593
PROCESS_SWITCH(JetFinderFullQATask, processTracks, "QA for charged tracks", false);
588594

589-
void processTracksWeighted(soa::Join<aod::JCollisions, aod::JMcCollisionLbs>::iterator const& collision,
595+
void processTracksWeighted(soa::Join<aod::JetCollisions, aod::JMcCollisionLbs>::iterator const& collision,
590596
aod::JMcCollisions const&,
591597
soa::Filtered<aod::JetTracks> const& tracks,
592598
soa::Filtered<aod::JetClusters> const& clusters)
593599
{
594600
float eventWeight = collision.mcCollision().weight();
601+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
602+
return;
603+
}
595604
registry.fill(HIST("h_collisions"), 0.5);
596605
registry.fill(HIST("h_collisions_weighted"), 0.5, eventWeight);
597606
if (!jetderiveddatautilities::eventEMCAL(collision)) {

‎PWGJE/Tasks/jetFinderHFQA.cxx‎

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ struct JetFinderHFQATask {
6969
Configurable<float> pTHatMaxMCP{"pTHatMaxMCP", 999.0, "maximum fraction of hard scattering for jet acceptance in particle MC"};
7070
Configurable<float> pTHatExponent{"pTHatExponent", 6.0, "exponent of the event weight for the calculation of pTHat"};
7171
Configurable<float> randomConeR{"randomConeR", 0.4, "size of random Cone for estimating background fluctuations"};
72+
Configurable<bool> skipMBGapEvents{"skipMBGapEvents", false, "flag to choose to reject min. bias gap events; jet-level rejection applied at the jet finder level, here rejection is applied for collision and track process functions"};
7273

7374
HfHelper hfHelper;
7475
std::vector<bool> filledJetR_Both;
@@ -1173,6 +1174,9 @@ struct JetFinderHFQATask {
11731174

11741175
void processMCCollisionsWeighted(aod::JetMcCollision const& collision)
11751176
{
1177+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1178+
return;
1179+
}
11761180
registry.fill(HIST("h_collision_eventweight_part"), collision.weight());
11771181
}
11781182
PROCESS_SWITCH(JetFinderHFQATask, processMCCollisionsWeighted, "collision QA for weighted events", false);
@@ -1463,6 +1467,9 @@ struct JetFinderHFQATask {
14631467
void processTracks(soa::Filtered<aod::JetCollisions>::iterator const& collision,
14641468
soa::Filtered<aod::JetTracks> const& tracks)
14651469
{
1470+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1471+
return;
1472+
}
14661473
registry.fill(HIST("h_collisions"), 0.5);
14671474
registry.fill(HIST("h2_centrality_collisions"), collision.centrality(), 0.5);
14681475
if (!jetderiveddatautilities::selectCollision(collision, eventSelection)) {
@@ -1479,6 +1486,9 @@ struct JetFinderHFQATask {
14791486
soa::Filtered<aod::JetTracks> const& tracks)
14801487
{
14811488
float eventWeight = collision.mcCollision().weight();
1489+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1490+
return;
1491+
}
14821492
registry.fill(HIST("h_collisions"), 0.5);
14831493
registry.fill(HIST("h_collisions_weighted"), 0.5, eventWeight);
14841494
if (!jetderiveddatautilities::selectCollision(collision, eventSelection)) {
@@ -1494,6 +1504,9 @@ struct JetFinderHFQATask {
14941504
CandidateTableData const& candidates,
14951505
soa::Filtered<JetTracksDataSub> const& tracks)
14961506
{
1507+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1508+
return;
1509+
}
14971510
if (!jetderiveddatautilities::selectCollision(collision, eventSelection)) {
14981511
return;
14991512
}
@@ -1511,6 +1524,9 @@ struct JetFinderHFQATask {
15111524

15121525
void processRho(aod::JetCollision const& collision, soa::Join<CandidateTableData, BkgRhoTable> const& candidates, soa::Filtered<aod::JetTracks> const& tracks)
15131526
{
1527+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1528+
return;
1529+
}
15141530
if (!jetderiveddatautilities::selectCollision(collision, eventSelection)) {
15151531
return;
15161532
}
@@ -1539,13 +1555,19 @@ struct JetFinderHFQATask {
15391555

15401556
void processRandomConeMCD(soa::Filtered<aod::JetCollisions>::iterator const& collision, JetTableMCDJoined const& jets, soa::Join<CandidateTableMCD, BkgRhoTable> const& candidates, soa::Filtered<aod::JetTracks> const& tracks)
15411557
{
1558+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1559+
return;
1560+
}
15421561
randomCone(collision, jets, candidates, tracks);
15431562
}
15441563
PROCESS_SWITCH(JetFinderHFQATask, processRandomConeMCD, "QA for random cone estimation of background fluctuations in mcd", false);
15451564

15461565
void processCandidates(soa::Filtered<aod::JetCollisions>::iterator const& collision, CandidateTableData const& candidates)
15471566
{
15481567

1568+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1569+
return;
1570+
}
15491571
for (auto const& candidate : candidates) {
15501572
registry.fill(HIST("h_candidate_invmass"), jetcandidateutilities::getCandidateInvariantMass(candidate));
15511573
registry.fill(HIST("h_candidate_pt"), candidate.pt());

‎PWGJE/Tasks/jetFinderQA.cxx‎

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ struct JetFinderQATask {
7272
Configurable<bool> checkMcCollisionIsMatched{"checkMcCollisionIsMatched", false, "0: count whole MCcollisions, 1: select MCcollisions which only have their correspond collisions"};
7373
Configurable<int> trackOccupancyInTimeRangeMax{"trackOccupancyInTimeRangeMax", 999999, "maximum occupancy of tracks in neighbouring collisions in a given time range; only applied to reconstructed collisions (data and mcd jets), not mc collisions (mcp jets)"};
7474
Configurable<int> trackOccupancyInTimeRangeMin{"trackOccupancyInTimeRangeMin", -999999, "minimum occupancy of tracks in neighbouring collisions in a given time range; only applied to reconstructed collisions (data and mcd jets), not mc collisions (mcp jets)"};
75+
Configurable<bool> skipMBGapEvents{"skipMBGapEvents", false, "flag to choose to reject min. bias gap events; jet-level rejection applied at the jet finder level, here rejection is applied for collision and track process functions"};
7576

7677
std::vector<bool> filledJetR_Both;
7778
std::vector<bool> filledJetR_Low;
@@ -924,6 +925,9 @@ struct JetFinderQATask {
924925

925926
void processMCCollisionsWeighted(aod::JetMcCollision const& collision)
926927
{
928+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
929+
return;
930+
}
927931
registry.fill(HIST("h_collision_eventweight_part"), collision.weight());
928932
}
929933
PROCESS_SWITCH(JetFinderQATask, processMCCollisionsWeighted, "collision QA for weighted events", false);
@@ -1057,6 +1061,9 @@ struct JetFinderQATask {
10571061
void processTracks(soa::Filtered<aod::JetCollisions>::iterator const& collision,
10581062
soa::Filtered<soa::Join<aod::JetTracks, aod::JTrackExtras>> const& tracks)
10591063
{
1064+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1065+
return;
1066+
}
10601067
registry.fill(HIST("h_collisions"), 0.5);
10611068
registry.fill(HIST("h2_centrality_collisions"), collision.centrality(), 0.5);
10621069
if (!jetderiveddatautilities::selectCollision(collision, eventSelection)) {
@@ -1078,6 +1085,9 @@ struct JetFinderQATask {
10781085
soa::Filtered<soa::Join<aod::JetTracks, aod::JTrackExtras>> const& tracks)
10791086
{
10801087
float eventWeight = collision.mcCollision().weight();
1088+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1089+
return;
1090+
}
10811091
registry.fill(HIST("h_collisions"), 0.5);
10821092
registry.fill(HIST("h_collisions_weighted"), 0.5, eventWeight);
10831093
if (!jetderiveddatautilities::selectCollision(collision, eventSelection)) {
@@ -1097,6 +1107,9 @@ struct JetFinderQATask {
10971107
void processTracksSub(soa::Filtered<aod::JetCollisions>::iterator const& collision,
10981108
soa::Filtered<aod::JetTracksSub> const& tracks)
10991109
{
1110+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1111+
return;
1112+
}
11001113
if (!jetderiveddatautilities::selectCollision(collision, eventSelection)) {
11011114
return;
11021115
}
@@ -1113,6 +1126,9 @@ struct JetFinderQATask {
11131126

11141127
void processRho(soa::Filtered<soa::Join<aod::JetCollisions, aod::BkgChargedRhos>>::iterator const& collision, soa::Filtered<aod::JetTracks> const& tracks)
11151128
{
1129+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1130+
return;
1131+
}
11161132
if (!jetderiveddatautilities::selectCollision(collision, eventSelection)) {
11171133
return;
11181134
}
@@ -1141,6 +1157,9 @@ struct JetFinderQATask {
11411157

11421158
void processRandomConeMCD(soa::Filtered<soa::Join<aod::JetCollisions, aod::BkgChargedRhos>>::iterator const& collision, soa::Join<aod::ChargedMCDetectorLevelJets, aod::ChargedMCDetectorLevelJetConstituents> const& jets, soa::Filtered<aod::JetTracks> const& tracks)
11431159
{
1160+
if (skipMBGapEvents && collision.subGeneratorId() == jetderiveddatautilities::JCollisionSubGeneratorId::mbGap) {
1161+
return;
1162+
}
11441163
randomCone(collision, jets, tracks);
11451164
}
11461165
PROCESS_SWITCH(JetFinderQATask, processRandomConeMCD, "QA for random cone estimation of background fluctuations in mcd", false);

0 commit comments

Comments
 (0)