Skip to content

Commit 57d6c62

Browse files
Oussama3030Oussama3030
committed
added a guard against arm 3d histos
1 parent bbd2fa3 commit 57d6c62

1 file changed

Lines changed: 33 additions & 16 deletions

File tree

‎PWGLF/Tasks/Strangeness/sigmaanalysis.cxx‎

Lines changed: 33 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,7 @@ struct sigmaanalysis {
9191

9292
// Species
9393
Configurable<bool> doLambdaStar{"doLambdaStar", false, "Build Lambda(1520) instead of Sigma0"};
94+
Configurable<bool> doArm{"doArm", true, "Fill the 3D Armenteros histograms"};
9495

9596
// Event level
9697
Configurable<bool> doPPAnalysis{"doPPAnalysis", true, "if in pp, set to true"};
@@ -443,8 +444,10 @@ struct sigmaanalysis {
443444
histos.add(histodir + "/Sigma0/hDCAPairDau", "hDCAPairDau", kTH1D, {axisDCAdau});
444445
histos.add(histodir + "/Sigma0/h3dMass", "h3dMass", kTH3D, {axisCentrality, axisPt, axisSigmaMass});
445446
histos.add(histodir + "/Sigma0/h3dOPAngleVsMass", "h3dOPAngleVsMass", kTH3D, {{140, 0.0f, +7.0f}, axisPt, axisSigmaMass});
446-
histos.add(histodir + "/Sigma0/h3dAlphaVsMass", "h3dAlphaVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
447-
histos.add(histodir + "/Sigma0/h3dQtarmVsMass", "h3dQtarmVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
447+
if (doArm) {
448+
histos.add(histodir + "/Sigma0/h3dAlphaVsMass", "h3dAlphaVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
449+
histos.add(histodir + "/Sigma0/h3dQtarmVsMass", "h3dQtarmVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
450+
}
448451

449452
histos.add(histodir + "/ASigma0/hMass", "hMass", kTH1D, {axisSigmaMass});
450453
histos.add(histodir + "/ASigma0/hPt", "hPt", kTH1D, {axisPt});
@@ -454,8 +457,10 @@ struct sigmaanalysis {
454457
histos.add(histodir + "/ASigma0/hDCAPairDau", "hDCAPairDau", kTH1D, {axisDCAdau});
455458
histos.add(histodir + "/ASigma0/h3dMass", "h3dMass", kTH3D, {axisCentrality, axisPt, axisSigmaMass});
456459
histos.add(histodir + "/ASigma0/h3dOPAngleVsMass", "h3dOPAngleVsMass", kTH3D, {{140, 0.0f, +7.0f}, axisPt, axisSigmaMass});
457-
histos.add(histodir + "/ASigma0/h3dAlphaVsMass", "h3dAlphaVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
458-
histos.add(histodir + "/ASigma0/h3dQtarmVsMass", "h3dQtarmVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
460+
if (doArm) {
461+
histos.add(histodir + "/ASigma0/h3dAlphaVsMass", "h3dAlphaVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
462+
histos.add(histodir + "/ASigma0/h3dQtarmVsMass", "h3dQtarmVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
463+
}
459464

460465
// Process MC
461466
if (doprocessMonteCarlo || doprocessMonteCarloWithEMCal) {
@@ -494,8 +499,10 @@ struct sigmaanalysis {
494499
histos.add(histodir + "/MC/Sigma0/h2dMCProcessVsGenRadius", "h2dMCProcessVsGenRadius", kTH2D, {{50, -0.5f, 49.5f}, axisV0PairRadius});
495500
histos.add(histodir + "/MC/Sigma0/h3dMass", "h3dMass", kTH3D, {axisCentrality, axisPt, axisSigmaMass});
496501
histos.add(histodir + "/MC/Sigma0/h3dMCProcess", "h3dMCProcess", kTH3D, {{50, -0.5f, 49.5f}, axisPt, axisSigmaMass});
497-
histos.add(histodir + "/MC/Sigma0/h3dAlphaVsMass", "h3dAlphaVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
498-
histos.add(histodir + "/MC/Sigma0/h3dQtarmVsMass", "h3dQtarmVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
502+
if (doArm) {
503+
histos.add(histodir + "/MC/Sigma0/h3dAlphaVsMass", "h3dAlphaVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
504+
histos.add(histodir + "/MC/Sigma0/h3dQtarmVsMass", "h3dQtarmVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
505+
}
499506

500507
histos.add(histodir + "/MC/ASigma0/hPt", "hPt", kTH1D, {axisPt});
501508
histos.add(histodir + "/MC/ASigma0/hMCPt", "hMCPt", kTH1D, {axisPt});
@@ -507,8 +514,10 @@ struct sigmaanalysis {
507514
histos.add(histodir + "/MC/ASigma0/h2dMCProcessVsGenRadius", "h2dMCProcessVsGenRadius", kTH2D, {{50, -0.5f, 49.5f}, axisV0PairRadius});
508515
histos.add(histodir + "/MC/ASigma0/h3dMass", "h3dMass", kTH3D, {axisCentrality, axisPt, axisSigmaMass});
509516
histos.add(histodir + "/MC/ASigma0/h3dMCProcess", "h3dMCProcess", kTH3D, {{50, -0.5f, 49.5f}, axisPt, axisSigmaMass});
510-
histos.add(histodir + "/MC/ASigma0/h3dAlphaVsMass", "h3dAlphaVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
511-
histos.add(histodir + "/MC/ASigma0/h3dQtarmVsMass", "h3dQtarmVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
517+
if (doArm) {
518+
histos.add(histodir + "/MC/ASigma0/h3dAlphaVsMass", "h3dAlphaVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
519+
histos.add(histodir + "/MC/ASigma0/h3dQtarmVsMass", "h3dQtarmVsMass", kTH3D, {{100, -5.0f, +5.0f}, axisPt, axisSigmaMass});
520+
}
512521

513522
histos.add(histodir + "/MC/LambdaStar/h3dMCPtvsOPAngle_Sig", "h3dMCPtvsOPAngle_Sig", kTH3D, {{140, 0.f, 7.f}, axisPt, axisSigmaMass});
514523
histos.add(histodir + "/MC/LambdaStar/h3dMCPtvsOPAngle_Bkg", "h3dMCPtvsOPAngle_Bkg", kTH3D, {{140, 0.f, 7.f}, axisPt, axisSigmaMass});
@@ -1210,8 +1219,10 @@ struct sigmaanalysis {
12101219
histos.fill(HIST(MainDir[mode]) + HIST("/Sigma0/hDCAPairDau"), sigma.dcadaughters());
12111220
histos.fill(HIST(MainDir[mode]) + HIST("/Sigma0/h3dMass"), centrality, sigma.pt(), sigma.sigma0Mass());
12121221
histos.fill(HIST(MainDir[mode]) + HIST("/Sigma0/h3dOPAngleVsMass"), sigma.opAngle(), sigma.pt(), sigma.sigma0Mass());
1213-
histos.fill(HIST(MainDir[mode]) + HIST("/Sigma0/h3dAlphaVsMass"), sigma.lStarAlpha(), sigma.pt(), sigma.sigma0Mass());
1214-
histos.fill(HIST(MainDir[mode]) + HIST("/Sigma0/h3dQtarmVsMass"), sigma.lStarQtarm(), sigma.pt(), sigma.sigma0Mass());
1222+
if (doArm) {
1223+
histos.fill(HIST(MainDir[mode]) + HIST("/Sigma0/h3dAlphaVsMass"), sigma.lStarAlpha(), sigma.pt(), sigma.sigma0Mass());
1224+
histos.fill(HIST(MainDir[mode]) + HIST("/Sigma0/h3dQtarmVsMass"), sigma.lStarQtarm(), sigma.pt(), sigma.sigma0Mass());
1225+
}
12151226

12161227
} else {
12171228
if (fillSelhistos) {
@@ -1232,8 +1243,10 @@ struct sigmaanalysis {
12321243
histos.fill(HIST(MainDir[mode]) + HIST("/ASigma0/hDCAPairDau"), sigma.dcadaughters());
12331244
histos.fill(HIST(MainDir[mode]) + HIST("/ASigma0/h3dMass"), centrality, sigma.pt(), sigma.sigma0Mass());
12341245
histos.fill(HIST(MainDir[mode]) + HIST("/ASigma0/h3dOPAngleVsMass"), sigma.opAngle(), sigma.pt(), sigma.sigma0Mass());
1235-
histos.fill(HIST(MainDir[mode]) + HIST("/ASigma0/h3dAlphaVsMass"), sigma.lStarAlpha(), sigma.pt(), sigma.sigma0Mass());
1236-
histos.fill(HIST(MainDir[mode]) + HIST("/ASigma0/h3dQtarmVsMass"), sigma.lStarQtarm(), sigma.pt(), sigma.sigma0Mass());
1246+
if (doArm) {
1247+
histos.fill(HIST(MainDir[mode]) + HIST("/ASigma0/h3dAlphaVsMass"), sigma.lStarAlpha(), sigma.pt(), sigma.sigma0Mass());
1248+
histos.fill(HIST(MainDir[mode]) + HIST("/ASigma0/h3dQtarmVsMass"), sigma.lStarQtarm(), sigma.pt(), sigma.sigma0Mass());
1249+
}
12371250
}
12381251

12391252
//_______________________________________
@@ -1290,8 +1303,10 @@ struct sigmaanalysis {
12901303
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/h2dMCPtVsPhotonMCPt"), sigma.mcpt(), sigma.photonmcpt());
12911304
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/hMass"), sigma.sigma0Mass());
12921305
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/h3dMass"), centrality, sigma.mcpt(), sigma.sigma0Mass());
1293-
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/h3dAlphaVsMass"), sigma.lStarAlpha(), sigma.mcpt(), sigma.sigma0Mass());
1294-
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/h3dQtarmVsMass"), sigma.lStarQtarm(), sigma.mcpt(), sigma.sigma0Mass());
1306+
if (doArm) {
1307+
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/h3dAlphaVsMass"), sigma.lStarAlpha(), sigma.mcpt(), sigma.sigma0Mass());
1308+
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/h3dQtarmVsMass"), sigma.lStarQtarm(), sigma.mcpt(), sigma.sigma0Mass());
1309+
}
12951310

12961311
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/hMCProcess"), sigma.mcprocess());
12971312
histos.fill(HIST(MainDir[mode]) + HIST("/MC/Sigma0/hGenRadius"), sigma.mcradius());
@@ -1309,8 +1324,10 @@ struct sigmaanalysis {
13091324
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/h2dMCPtVsPhotonMCPt"), sigma.mcpt(), sigma.photonmcpt());
13101325
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/hMass"), sigma.sigma0Mass());
13111326
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/h3dMass"), centrality, sigma.mcpt(), sigma.sigma0Mass());
1312-
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/h3dAlphaVsMass"), sigma.lStarAlpha(), sigma.mcpt(), sigma.sigma0Mass());
1313-
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/h3dQtarmVsMass"), sigma.lStarQtarm(), sigma.mcpt(), sigma.sigma0Mass());
1327+
if (doArm) {
1328+
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/h3dAlphaVsMass"), sigma.lStarAlpha(), sigma.mcpt(), sigma.sigma0Mass());
1329+
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/h3dQtarmVsMass"), sigma.lStarQtarm(), sigma.mcpt(), sigma.sigma0Mass());
1330+
}
13141331

13151332
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/hMCProcess"), sigma.mcprocess());
13161333
histos.fill(HIST(MainDir[mode]) + HIST("/MC/ASigma0/hGenRadius"), sigma.mcradius());

0 commit comments

Comments
 (0)