reimplement functionality

Author: jjuracka

Modules/MFT/include/MFT/QcMFTClusterTask.h

@@ -26,6 +26,7 @@
 #include <TH2F.h>
 #include <TCanvas.h>
 #include <TString.h>
+#include <TLegend.h>
 #include <DataFormatsITSMFT/TopologyDictionary.h>
 #include "ReconstructionDataFormats/BaseCluster.h"
 #include "MFTBase/GeometryTGeo.h"
@@ -83,6 +84,8 @@ class QcMFTClusterTask /*final*/ : public TaskInterface // todo add back the "fi
   std::vector<std::unique_ptr<TH2FRatio>> mClusterXYinLayer;
   std::vector<std::unique_ptr<TH1FRatio>> mClusterRinLayer;
   std::unique_ptr<TCanvas> mClusterRinAllLayers = nullptr;
+  std::unique_ptr<TH1F> mFrame = nullptr; // dummy histogram to set the axes
+  std::unique_ptr<TLegend> mLegend = nullptr;
 
   std::unique_ptr<TH1FRatio> mClustersROFSize = nullptr;
   std::unique_ptr<TH1FRatio> mClustersBC = nullptr;
@@ -92,6 +95,8 @@ class QcMFTClusterTask /*final*/ : public TaskInterface // todo add back the "fi
   int mOnlineQC;
 
   const TString mColors[10] = { "#1F77B4", "#FF7F0E", "#2CA02C", "#D62728", "#8C564B", "#E377C2", "#9467BD", "#BCBD22", "#7F7F7F", "#17BECF" };
+  TH1F* clonedHistos[10] = { nullptr };
+  bool firstRun = true;
 
   // needed to construct the name and path of some histograms
   int mHalf[936] = { 0 };

Modules/MFT/src/QcMFTClusterTask.cxx

@@ -259,6 +259,11 @@ void QcMFTClusterTask::initialize(o2::framework::InitContext& /*ctx*/)
     // canvas for for cluster R in all layers
     mClusterRinAllLayers = std::make_unique<TCanvas>("mClusterRinAllLayers", "Cluster Radial Position in All MFT Layers");
     getObjectsManager()->startPublishing(mClusterRinAllLayers.get());
+    mFrame = std::make_unique<TH1F>("frame", "Cluster Radial Position in All MFT Layers; r (cm); # entries", 400, 0, 20);
+    mFrame->SetStats(0);
+    mLegend = std::make_unique<TLegend>(0.8, 0.5, 0.9, 0.9);
+    mLegend->SetBorderSize(0);
+    mLegend->SetFillStyle(0);
   }
 }
 
@@ -443,6 +448,8 @@ void QcMFTClusterTask::reset()
       mClusterRinLayer[nMFTLayer]->Reset();
     }
     mClusterRinAllLayers->Clear();
+    mFrame->Reset();
+    mLegend->Clear();
   }
 }
 
@@ -469,6 +476,34 @@ void QcMFTClusterTask::updateCanvas()
 {
   mClusterRinAllLayers->Clear();
   mClusterRinAllLayers->cd();
+  
+  for (auto nMFTLayer = 0; nMFTLayer < 10; nMFTLayer++) {
+    clonedHistos[nMFTLayer] = static_cast<TH1F*>(mClusterRinLayer[nMFTLayer]->getNum()->Clone());
+    clonedHistos[nMFTLayer]->SetDirectory(nullptr);
+    clonedHistos[nMFTLayer]->SetStats(0);
+    clonedHistos[nMFTLayer]->SetLineColor(TColor::GetColor(mColors[nMFTLayer]));
+  }
+  
+  double maxY = 0;
+  for (auto nMFTLayer = 0; nMFTLayer < 10; nMFTLayer++) {
+    double localMax = clonedHistos[nMFTLayer]->GetMaximum();
+    if (localMax > maxY) {
+      maxY = localMax;
+    }
+  }
+  mFrame->SetMaximum(maxY * 1.1);
+  mFrame->Draw();
+  for (auto nMFTLayer = 0; nMFTLayer < 10; nMFTLayer++) {
+    clonedHistos[nMFTLayer]->Draw("hist same");
+  }
+  if (firstRun) {
+    mLegend->Clear();
+    for (auto nMFTLayer = 0; nMFTLayer < 10; nMFTLayer++) {
+      mLegend->AddEntry(clonedHistos[nMFTLayer], Form("D%dF%d", static_cast<int>(std::floor(nMFTLayer / 2.)), nMFTLayer % 2 == 0 ? 0 : 1), "l");
+    }
+    firstRun = false;
+  }
+  mLegend->Draw();
   mClusterRinAllLayers->Update();
 }