GLO: K0s optionally separating mass in low Occ and low pt and high

Signed-off-by: Felix Schlepper <felix.schlepper@cern.ch>

Author: f3sch

Modules/GLO/README.md

@@ -31,6 +31,8 @@ Documents the available task in the module and their parameters.
  - `cutK0Mass=0.05f` cut around K0s peak
  - `trackSourcesK0` SVertexer input sources
  - `publishK0s3D=false` publish 3D cycle,integral histograms
+ - `splitK0sMassOccupancy=float` splitting point in TPC occupancy to define low and high region by default off
+ - `splitK0sMassPt=float` splitting point in pt to define low and high region by default off
  + K0Fitter options
 #### ITS-PV
  - `doPVITSQC=false` produce ITS vs PV plots

Modules/GLO/glo-itstpc-mtch-qcmn-test.json

@@ -76,8 +76,12 @@
               "mFractionITSTPCmatch_ITS_Hist",
               "mFractionITSTPCmatchPhi_ITS_Hist",
               "mFractionITSTPCmatchEta_ITS_Hist",
+              "mK0sMassVsPtVsOcc_Integral_pmass",
               "mK0sMassVsPtVsOcc_Cycle_pmass",
-              "mK0sMassVsPtVsOcc_Integral_pmass"
+              "mK0sMassVsPtVsOcc_Cycle_pmass_lowOcc",
+              "mK0sMassVsPtVsOcc_Cycle_pmass_lowPt",
+              "mK0sMassVsPtVsOcc_Cycle_pmass_highOcc",
+              "mK0sMassVsPtVsOcc_Cycle_pmass_highPt"
             ]
           }
         ],

Modules/GLO/include/GLO/ITSTPCMatchingTask.h

@@ -26,6 +26,7 @@
 #include <CommonConstants/PhysicsConstants.h>
 
 #include <memory>
+#include <limits>
 
 #include <TH2F.h>
 #include <TH3F.h>
@@ -49,6 +50,9 @@ class ITSTPCMatchingTask final : public TaskInterface
   void reset() override;
 
  private:
+  template <typename T>
+  static constexpr T OptValue = std::numeric_limits<T>::max();
+
   o2::gloqc::MatchITSTPCQC mMatchITSTPCQC;
 
   bool mIsSync{ false };
@@ -61,6 +65,8 @@ class ITSTPCMatchingTask final : public TaskInterface
 
   bool mDoK0s{ false };
   bool mPublishK0s3D{ false };
+  float mSplitTPCOccupancy{ OptValue<float> };
+  float mSplitPt{ OptValue<float> };
   std::unique_ptr<TH3F> mK0sCycle;
   std::unique_ptr<TH3F> mK0sIntegral;
   helpers::K0sFitter mK0sFitter;

Modules/GLO/src/ITSTPCMatchingTask.cxx

@@ -67,6 +67,8 @@ void ITSTPCMatchingTask::initialize(o2::framework::InitContext& /*ctx*/)
     }
 
     mPublishK0s3D = getFromConfig(mCustomParameters, "publishK0s3D", false);
+    mSplitTPCOccupancy = getFromConfig(mCustomParameters, "splitK0sMassOccupancy", mSplitTPCOccupancy);
+    mSplitPt = getFromConfig(mCustomParameters, "splitK0sMassPt", mSplitPt);
     mK0sFitter.init(mCustomParameters);
   }
   // PV
@@ -203,6 +205,19 @@ void ITSTPCMatchingTask::endOfCycle()
 
         TH1D* h{ nullptr };
         getObjectsManager()->startPublishing((h = mK0sCycle->ProjectionY("mK0sMassVsPtVsOcc_Cycle_pmass")), PublicationPolicy::Once);
+
+        if (mSplitTPCOccupancy != OptValue<float>) {
+          auto splitOccBin = mK0sCycle->GetZaxis()->FindBin(mSplitTPCOccupancy);
+          getObjectsManager()->startPublishing(mK0sCycle->ProjectionY("mK0sMassVsPtVsOcc_Cycle_pmass_lowOcc", 0, -1, 0, splitOccBin - 1), PublicationPolicy::Once);
+          getObjectsManager()->startPublishing(mK0sCycle->ProjectionY("mK0sMassVsPtVsOcc_Cycle_pmass_highOcc", 0, -1, splitOccBin), PublicationPolicy::Once);
+        }
+
+        if (mSplitPt != OptValue<float>) {
+          auto splitPtBin = mK0sCycle->GetXaxis()->FindBin(mSplitPt);
+          getObjectsManager()->startPublishing(mK0sCycle->ProjectionY("mK0sMassVsPtVsOcc_Cycle_pmass_lowPt", 0, splitPtBin - 1), PublicationPolicy::Once);
+          getObjectsManager()->startPublishing(mK0sCycle->ProjectionY("mK0sMassVsPtVsOcc_Cycle_pmass_highPt", splitPtBin), PublicationPolicy::Once);
+        }
+
         if (mK0sFitter.fit(h)) {
           if (mDoPVITS && mPVITSCycle->GetEntries() != 0) {
             mK0sFitter.mSignalAndBackground->SetParameter(helpers::K0sFitter::Parameters::Pol0, mPVITSCycle->GetEntries());

Modules/GLO/src/ITSTPCmatchingCheck.cxx

@@ -481,15 +481,25 @@ void ITSTPCmatchingCheck::beautify(std::shared_ptr<MonitorObject> mo, Quality ch
       msg->AddText("Not-handled Quality flag, don't panic...");
     }
     eff->GetListOfFunctions()->Add(msg);
-  } else if (mShowK0s && (name == "mK0sMassVsPtVsOcc_Cycle_pmass" || name == "mK0sMassVsPtVsOcc_Integral_pmass")) {
+  } else if (mShowK0s && (name.starts_with("mK0sMassVsPtVsOcc_Cycle_pmass") || name.starts_with("mK0sMassVsPtVsOcc_Integral_pmass"))) {
     auto* h = dynamic_cast<TH1*>(mo->getObject());
     if (!h) {
       ILOG(Error) << "Failed cast for " << name << " beautify!" << ENDM;
       return;
     }
     auto isCycle = name.find("Cycle") != std::string::npos;
+    auto isHigh = name.find("high") != std::string::npos;
+    auto isLow = name.find("low") != std::string::npos;
+    auto isOcc = name.ends_with("Occ");
+    auto isPt = name.ends_with("Pt");
     auto msg = new TPaveText(0.6, 0.6, 0.88, 0.88, "NDC;NB");
-    h->SetTitle(Form("K0s invariant mass (integrated over #it{p}_{T} and occupancy, %s);K0s mass (GeV/c^{2});entries", (isCycle) ? "current cycle" : "integrated"));
+    if (!isLow && !isHigh) {
+      h->SetTitle(Form("K0s invariant mass (integrated over #it{p}_{T} and occupancy, %s);K0s mass (GeV/c^{2});entries", (isCycle) ? "last cycle" : "integrated"));
+    } else {
+      h->SetTitle(Form("K0s invariant mass (integrated over%s#it{p}_{T} and%soccupancy, last cycle);K0s mass (GeV/c^{2});entries",
+                       (isPt) ? ((isLow) ? " low " : " high ") : " ",
+                       (isOcc) ? ((isLow) ? " low " : " high ") : " "));
+    }
     if (!mK0sFitter.fit(h, true)) {
       msg->AddText("Fit: Failed");
       msg->SetFillColor(kRed);