2222#ifndef PWGCF_FEMTODREAM_CORE_FEMTODREAMCASCADESELECTION_H_
2323#define PWGCF_FEMTODREAM_CORE_FEMTODREAMCASCADESELECTION_H_
2424
25- #include < iostream>
26- #include < string>
27- #include < vector>
28-
29- #include < TDatabasePDG.h> // FIXME
30-
3125#include " PWGCF/FemtoDream/Core/femtoDreamObjectSelection.h"
3226#include " PWGCF/FemtoDream/Core/femtoDreamSelection.h"
3327#include " PWGCF/FemtoDream/Core/femtoDreamTrackSelection.h"
3428
3529#include " Common/Core/RecoDecay.h"
30+
3631#include " Framework/HistogramRegistry.h"
3732#include " ReconstructionDataFormats/PID.h"
3833
34+ #include < TDatabasePDG.h> // FIXME
35+
36+ #include < iostream>
37+ #include < string>
38+ #include < vector>
39+
3940using namespace o2 ::framework;
4041using namespace o2 ::analysis::femtoDream::femtoDreamSelection;
4142
@@ -347,8 +348,8 @@ class FemtoDreamCascadeSelection
347348 }; // /< Helper information for the
348349 // /< different selections
349350
350- // static constexpr int kNcutStages = 2;
351- // static constexpr std::string_view mCutStage[kNcutStages] = {"BeforeSel", "AfterSel"};
351+ // static constexpr int kNcutStages = 2;
352+ // static constexpr std::string_view mCutStage[kNcutStages] = {"BeforeSel", "AfterSel"};
352353}; // namespace femtoDream
353354
354355template <o2::aod::femtodreamparticle::ParticleType part, o2::aod::femtodreamparticle::ParticleType daugh, o2::aod::femtodreamparticle::ParticleType bach, typename cutContainerType>
@@ -664,6 +665,8 @@ void FemtoDreamCascadeSelection::fillQA(Col const& col, Casc const& casc, Track
664665 const float cpaCasc = casc.casccosPA (col.posX (), col.posY (), col.posZ ());
665666 const float cpav0 = casc.v0cosPA (col.posX (), col.posY (), col.posZ ());
666667 const float v0dcatopv = casc.dcav0topv (col.posX (), col.posY (), col.posZ ());
668+ const float invMass = isCascOmega ? casc.mOmega () : casc.mXi ();
669+
667670 if (mQAHistogramRegistry ) {
668671 mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hSign" ), casc.sign ());
669672 mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hPt" ), casc.pt ());
@@ -675,7 +678,7 @@ void FemtoDreamCascadeSelection::fillQA(Col const& col, Casc const& casc, Track
675678 mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hDecVtxX" ), decVtx.at (0 ));
676679 mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hDecVtxY" ), decVtx.at (1 ));
677680 mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hDecVtxZ" ), decVtx.at (2 ));
678- mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hInvMass" ), casc. mXi () );
681+ mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hInvMass" ), invMass );
679682 mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hV0DCADaugh" ), casc.dcaV0daughters ());
680683 mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hV0CPA" ), cpav0);
681684 mQAHistogramRegistry ->fill (HIST (o2::aod::femtodreamparticle::ParticleTypeName[part]) + HIST (" /" ) + HIST (mCutStage [cutstage]) + HIST (" /hV0TranRad" ), casc.v0radius ());
0 commit comments