DPG: update the k0 perf for MC

Author: njacazio

DPG/Tasks/AOTTrack/V0Cascades/perfK0sResolution.cxx

@@ -35,6 +35,7 @@ struct perfK0sResolution {
   ConfigurableAxis mBins{"mBins", {200, 0.4f, 0.6f}, "Mass binning"};
   ConfigurableAxis pTBins{"pTBins", {200, 0.f, 10.f}, "pT binning"};
   ConfigurableAxis pTResBins{"pTResBins", {200, -1.2f, 1.2f}, "pT resolution binning"};
+  ConfigurableAxis invpTResBins{"invpTResBins", {200, -1.2f, 1.2f}, "inv pT resolution binning"};
   ConfigurableAxis etaBins{"etaBins", {2, -1.f, 1.f}, "eta binning"};
   ConfigurableAxis etaBinsDauthers{"etaBinsDauthers", {2, -1.f, 1.f}, "eta binning"};
   ConfigurableAxis phiBins{"phiBins", {4, 0.f, 6.28f}, "phi binning"};
@@ -46,10 +47,12 @@ struct perfK0sResolution {
     const AxisSpec mAxis{mBins, "#it{m} (GeV/#it{c}^{2})"};
     const AxisSpec pTAxis{pTBins, "#it{p}_{T} (GeV/#it{c})"};
     const AxisSpec pTResAxis{pTResBins, "#Delta#it{p}_{T} (GeV/#it{c})"};
+    const AxisSpec invpTResAxis{pTResBins, "#Delta#it{p}_{T} (GeV/#it{c})"};
     const AxisSpec etaAxis{etaBins, "#eta"};
     const AxisSpec etaAxisPosD{etaBinsDauthers, "#eta pos."};
     const AxisSpec etaAxisNegD{etaBinsDauthers, "#eta neg."};
     const AxisSpec phiAxis{phiBins, "#phi"};
+    const AxisSpec trueK0Axis{2, -0.5, 1.5, "True K0"};
 
     int nProc = 0;
     if (doprocessData) {
@@ -70,7 +73,7 @@ struct perfK0sResolution {
     rK0sResolution.add("h2_massphi", "h2_massphi", {HistType::kTH2F, {mAxis, phiAxis}});
     if (useMultidimHisto) {
       if (doprocessMC) {
-        rK0sResolution.add("thn_mass", "thn_mass", kTHnSparseF, {mAxis, pTAxis, etaAxis, phiAxis, etaAxisPosD, etaAxisNegD, pTResAxis, pTResAxis});
+        rK0sResolution.add("thn_mass", "thn_mass", kTHnSparseF, {mAxis, pTAxis, etaAxis, phiAxis, etaAxisPosD, etaAxisNegD, invpTResAxis, invpTResAxis, trueK0Axis});
       } else {
         rK0sResolution.add("thn_mass", "thn_mass", kTHnSparseF, {mAxis, pTAxis, etaAxis, phiAxis, etaAxisPosD, etaAxisNegD});
       }
@@ -220,7 +223,8 @@ struct perfK0sResolution {
   }
   PROCESS_SWITCH(perfK0sResolution, processData, "Process data", true);
 
-  void processMC(soa::Filtered<SelectedCollisions>::iterator const& collision, soa::Filtered<aod::V0Datas> const& fullV0s, soa::Join<PIDTracks, aod::McTrackLabels> const& tracks, aod::McParticles const&)
+  void processMC(soa::Filtered<SelectedCollisions>::iterator const& collision, soa::Filtered<soa::Join<aod::V0Datas, aod::McV0Labels>> const& fullV0s,
+                 soa::Join<PIDTracks, aod::McTrackLabels> const& tracks, aod::McParticles const&)
   {
     for (auto& v0 : fullV0s) {
       const auto& posTrack = v0.posTrack_as<soa::Join<PIDTracks, aod::McTrackLabels>>();
@@ -236,16 +240,17 @@ struct perfK0sResolution {
       if (posTrack.mcParticle().pdgCode() != 211 || negTrack.mcParticle().pdgCode() != -211) {
         continue;
       }
-
-      rK0sResolution.fill(HIST("h2_massPosPtRes"), v0.mK0Short(), posTrack.pt() - posTrack.mcParticle().pt());
-      rK0sResolution.fill(HIST("h2_massNegPtRes"), v0.mK0Short(), negTrack.pt() - negTrack.mcParticle().pt());
+      const bool isTrueK0s = (v0.has_mcParticle() && v0.mcParticle().pdgCode() == 310);
+      rK0sResolution.fill(HIST("h2_massPosPtRes"), v0.mK0Short(), v0.PositivePt() - posTrack.mcParticle().pt());
+      rK0sResolution.fill(HIST("h2_massNegPtRes"), v0.mK0Short(), v0.NegativePt() - negTrack.mcParticle().pt());
       rK0sResolution.fill(HIST("h2_masspT"), v0.mK0Short(), v0.pt());
       rK0sResolution.fill(HIST("h2_masseta"), v0.mK0Short(), v0.eta());
       rK0sResolution.fill(HIST("h2_massphi"), v0.mK0Short(), v0.phi());
       if (useMultidimHisto) {
         rK0sResolution.fill(HIST("thn_mass"), v0.mK0Short(), v0.pt(), v0.eta(), v0.phi(), posTrack.eta(), negTrack.eta(),
-                            posTrack.pt() - posTrack.mcParticle().pt(),
-                            negTrack.pt() - negTrack.mcParticle().pt());
+                            1./v0.PositivePt() - 1./posTrack.mcParticle().pt(),
+                            1./v0.NegativePt() - 1./negTrack.mcParticle().pt(),
+                            isTrueK0s);
       }
     }
   }