Skip to content

Commit 1ee4cc6

Browse files
dmallick2dmallick2
authored
[PWGDQ] Add MC truth information and configurable tag (#17182)
1 parent 1284ea0 commit 1ee4cc6

3 files changed

Lines changed: 64 additions & 6 deletions

File tree

‎PWGDQ/Core/HistogramsLibrary.cxx‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1268,6 +1268,20 @@ void o2::aod::dqhistograms::DefineHistograms(HistogramManager* hm, const char* h
12681268
hm->AddHistogram(histClass, "Mass_Pt_cosThetaCS_Tauxy", "", 4, varspTCS.data(), bins.data(), xmin.data(), xmax.data(), nullptr, -1, kFALSE);
12691269
hm->AddHistogram(histClass, "Mass_Pt_cosThetaRM_Tauxy", "", 4, varspTRM.data(), bins.data(), xmin.data(), xmax.data(), nullptr, -1, kFALSE);
12701270
}
1271+
if (groupStr.CompareTo("polarization-dielectron-pbpb-midy-he-gen") == 0) {
1272+
std::array<int, 5> varspTHE = {VarManager::kMCMass, VarManager::kMCPt, VarManager::kMCEventCentrFT0C, VarManager::kMCCosThetaHE, VarManager::kMCPhiHE};
1273+
std::array<int, 5> bins = {50, 30, 10, 20, 10};
1274+
std::array<double, 5> xmin = {2., 0., 0, -1., 0.0};
1275+
std::array<double, 5> xmax = {4., 3., 100, 1., 6.28};
1276+
hm->AddHistogram(histClass, "MC_Dielectron_Mass_Pt_Cent_cosThetaHE", "", 5, varspTHE.data(), bins.data(), xmin.data(), xmax.data(), 0, -1, kFALSE);
1277+
}
1278+
if (groupStr.CompareTo("polarization-dielectron-pbpb-midy-cs-gen") == 0) {
1279+
std::array<int, 5> varspTCS = {VarManager::kMCMass, VarManager::kMCPt, VarManager::kMCEventCentrFT0C, VarManager::kMCCosThetaCS, VarManager::kMCPhiCS};
1280+
std::array<int, 5> bins = {50, 30, 10, 20, 10};
1281+
std::array<double, 5> xmin = {2., 0., 0, -1., 0.0};
1282+
std::array<double, 5> xmax = {4., 3., 100, 1., 6.28};
1283+
hm->AddHistogram(histClass, "MC_Dielectron_Mass_Pt_Cent_cosThetaCS", "", 5, varspTCS.data(), bins.data(), xmin.data(), xmax.data(), 0, -1, kFALSE);
1284+
}
12711285
if (groupStr.CompareTo("pair") == 0) {
12721286
if (subGroupStr.Contains("cepf")) {
12731287
hm->AddHistogram(histClass, "Mass", "", false, 300, 0.0, 12.0, VarManager::kMass);

‎PWGDQ/Core/VarManager.cxx‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2400,6 +2400,7 @@ void VarManager::SetDefaultVarNames()
24002400
fgVarNamesMap["kMCPhi"] = kMCPhi;
24012401
fgVarNamesMap["kMCEta"] = kMCEta;
24022402
fgVarNamesMap["kMCY"] = kMCY;
2403+
fgVarNamesMap["kMCMass"] = kMCMass;
24032404
fgVarNamesMap["kMCCosThetaHE"] = kMCCosThetaHE;
24042405
fgVarNamesMap["kMCPhiHE"] = kMCPhiHE;
24052406
fgVarNamesMap["kMCPhiTildeHE"] = kMCPhiTildeHE;

‎PWGDQ/Tasks/dqEfficiency_withAssoc_direct.cxx‎

Lines changed: 49 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1140,7 +1140,10 @@ struct AnalysisSameEventPairing {
11401140
Configurable<std::string> muon{"cfgMuonCuts", "", "Comma separated list of muon cuts"};
11411141
Configurable<std::string> pair{"cfgPairCuts", "", "Comma separated list of pair cuts, !!! Use only if you know what you are doing, otherwise leave empty"};
11421142
Configurable<bool> fConfigQA{"cfgQA", false, "If true, fill QA histograms"};
1143-
Configurable<bool> fConfigPseudoQA{"cfgPseudoQA", false, "If true, fill QA histograms with pseudo-proper decay length"};
1143+
Configurable<bool> fConfigMCtruthQA{"cfgMCtruthQA", false, "If true, fill QA histograms with default"};
1144+
Configurable<bool> fConfigPseudoQA{"cfgPseudoQA", false, "If true, fill QA histograms with pseudo-proper decay length analysis with polarization"};
1145+
Configurable<bool> fConfigTruthPbPbMIDYHE{"cfgTruthPbPbmidyHE", false, "If true, fill QA histograms with dielectron pairs in helicity frame"};
1146+
Configurable<bool> fConfigTruthPbPbMIDYCS{"cfgTruthPbPbmidyCS", false, "If true, fill QA histograms with dielectron pairs in collion-soper frame"};
11441147
Configurable<std::string> fConfigAddSEPHistogram{"cfgAddSEPHistogram", "", "Comma separated list of histograms"};
11451148
Configurable<std::string> fConfigAddJSONHistograms{"cfgAddJSONHistograms", "", "Histograms in JSON format"};
11461149
Configurable<bool> useRemoteField{"cfgUseRemoteField", false, "Chose whether to fetch the magnetic field from ccdb or set it manually"};
@@ -1500,19 +1503,35 @@ struct AnalysisSameEventPairing {
15001503
histNames += Form("MCTruthGen_%s;", sig->GetName()); // TODO: Add these names to a std::vector to avoid using Form in the process function
15011504
histNames += Form("MCTruthGenSel_%s;", sig->GetName());
15021505
} else if (sig->GetNProngs() == 2) {
1503-
histNames += Form("MCTruthGenPairSel_%s;", sig->GetName()); // after event selection
1506+
if (fConfigOptions.fConfigMCtruthQA.value) {
1507+
histNames += Form("MCTruthGenPairSel_%s;", sig->GetName()); // after event selection
1508+
}
15041509
if (fConfigOptions.fConfigPseudoQA.value) {
15051510
histNames += Form("MCTruthGenPseudoPolPairSel_%s;", sig->GetName());
15061511
}
1512+
if (fConfigOptions.fConfigTruthPbPbMIDYHE.value) {
1513+
histNames += Form("MCTruthGenPseudoPolPairSel_%s;", sig->GetName());
1514+
}
1515+
if (fConfigOptions.fConfigTruthPbPbMIDYCS.value) {
1516+
histNames += Form("MCTruthGenPseudoPolPairSel_%s;", sig->GetName());
1517+
}
15071518
fHasTwoProngGenMCsignals = true;
15081519
}
15091520
// for these pair level signals, also add histograms for each MCgenAcc cut if specified
15101521
if (fUseMCGenAccCut) {
15111522
for (auto& cut : fMCGenAccCuts) {
1512-
histNames += Form("MCTruthGenPairSel_%s_%s;", sig->GetName(), cut->GetName()); // after event selection and MCgenAcc cut
1523+
if (fConfigOptions.fConfigMCtruthQA.value) {
1524+
histNames += Form("MCTruthGenPairSel_%s_%s;", sig->GetName(), cut->GetName()); // after event selection and MCgenAcc cut
1525+
}
15131526
if (fConfigOptions.fConfigPseudoQA.value) {
15141527
histNames += Form("MCTruthGenPseudoPolPairSel_%s_%s;", sig->GetName(), cut->GetName());
15151528
}
1529+
if (fConfigOptions.fConfigTruthPbPbMIDYHE.value) {
1530+
histNames += Form("MCTruthGenPoldielectronPbPbPairHESel_%s_%s;", sig->GetName(), cut->GetName());
1531+
}
1532+
if (fConfigOptions.fConfigTruthPbPbMIDYCS.value) {
1533+
histNames += Form("MCTruthGenPseudoPolPairCSSel_%s_%s;", sig->GetName(), cut->GetName());
1534+
}
15161535
}
15171536
}
15181537
}
@@ -2167,17 +2186,35 @@ struct AnalysisSameEventPairing {
21672186
VarManager::FillTrackCollisionMC<TPairType>(motherMCParticle_t1, collVtxPos, pdgDB->Mass(motherMCParticle_t1.pdgCode()));
21682187
}
21692188
}
2170-
fHistMan->FillHistClass(Form("MCTruthGenPairSel_%s", sig->GetName()), VarManager::fgValues);
2171-
if (fConfigOptions.fConfigQA.value) {
2189+
if (fConfigOptions.fConfigMCtruthQA.value) {
2190+
fHistMan->FillHistClass(Form("MCTruthGenPairSel_%s", sig->GetName()), VarManager::fgValues);
2191+
}
2192+
if (fConfigOptions.fConfigPseudoQA.value) {
21722193
fHistMan->FillHistClass(Form("MCTruthGenPseudoPolPairSel_%s", sig->GetName()), VarManager::fgValues);
21732194
}
2195+
if (fConfigOptions.fConfigTruthPbPbMIDYHE.value) {
2196+
fHistMan->FillHistClass(Form("MCTruthGenPoldielectronPbPbPairHESel_%s", sig->GetName()), VarManager::fgValues);
2197+
}
2198+
if (fConfigOptions.fConfigTruthPbPbMIDYCS.value) {
2199+
fHistMan->FillHistClass(Form("MCTruthGenPoldielectronPbPbPairCSSel_%s", sig->GetName()), VarManager::fgValues);
2200+
}
2201+
21742202
if (fUseMCGenAccCut) {
21752203
for (auto& cut : fMCGenAccCuts) {
21762204
if (cut->IsSelected(VarManager::fgValues)) {
2177-
fHistMan->FillHistClass(Form("MCTruthGenPairSel_%s_%s", sig->GetName(), cut->GetName()), VarManager::fgValues);
2205+
if (fConfigOptions.fConfigMCtruthQA.value) {
2206+
fHistMan->FillHistClass(Form("MCTruthGenPairSel_%s_%s", sig->GetName(), cut->GetName()), VarManager::fgValues);
2207+
}
21782208
if (fConfigOptions.fConfigPseudoQA.value) {
21792209
fHistMan->FillHistClass(Form("MCTruthGenPseudoPolPairSel_%s_%s", sig->GetName(), cut->GetName()), VarManager::fgValues);
21802210
}
2211+
2212+
if (fConfigOptions.fConfigTruthPbPbMIDYHE.value) {
2213+
fHistMan->FillHistClass(Form("MCTruthGenPoldielectronPbPbPairHESel_%s_%s", sig->GetName(), cut->GetName()), VarManager::fgValues);
2214+
}
2215+
if (fConfigOptions.fConfigTruthPbPbMIDYCS.value) {
2216+
fHistMan->FillHistClass(Form("MCTruthGenPoldielectronPbPbPairCSSel_%s_%s", sig->GetName(), cut->GetName()), VarManager::fgValues);
2217+
}
21812218
}
21822219
}
21832220
}
@@ -3181,7 +3218,13 @@ void DefineHistograms(HistogramManager* histMan, TString histClasses, const char
31813218
if (classStr.Contains("MCTruthGenPseudoPolPair")) {
31823219
dqhistograms::DefineHistograms(histMan, objArray->At(iclass)->GetName(), "polarization-pseudoproper-gen", histName);
31833220
}
3221+
if (classStr.Contains("MCTruthGenPoldielectronPbPbPairHE")) {
3222+
dqhistograms::DefineHistograms(histMan, objArray->At(iclass)->GetName(), "polarization-dielectron-pbpb-midy-he-gen", histName);
3223+
}
31843224

3225+
if (classStr.Contains("MCTruthGenPoldielectronPbPbPairCS")) {
3226+
dqhistograms::DefineHistograms(histMan, objArray->At(iclass)->GetName(), "polarization-dielectron-pbpb-midy-cs-gen", histName);
3227+
}
31853228
if (classStr.Contains("MCTruthGenSelBR")) {
31863229
dqhistograms::DefineHistograms(histMan, objArray->At(iclass)->GetName(), "mctruth_triple");
31873230
} else if (classStr.Contains("MCTruthGen")) {

0 commit comments

Comments
 (0)