drof macro

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

Author: f3sch

Detectors/ITSMFT/ITS/macros/test/CheckDROF.C

@@ -21,6 +21,7 @@
 #include "TH1F.h"
 #include <TH2F.h>
 #include "TH2D.h"
+#include "TH3D.h"
 #include <TProfile.h>
 #include <TCanvas.h>
 #include <THStack.h>
@@ -323,15 +324,16 @@ void CheckDROF(bool plot = false, bool write = false, const std::string& tracfil
       fOut = TFile::Open("checkDROF.root", "RECREATE");
     }
 
+    const int nb = 100;
+    double xbins[nb + 1], ptcutl = 0.01, ptcuth = 10.;
+    double a = std::log(ptcuth / ptcutl) / nb;
+    for (int i = 0; i <= nb; i++) {
+      xbins[i] = ptcutl * std::exp(i * a);
+    }
+
     //////////////////////
     // Eff Tracks
     {
-      int nb = 100;
-      double xbins[nb + 1], ptcutl = 0.01, ptcuth = 10.;
-      double a = std::log(ptcuth / ptcutl) / nb;
-      for (int i = 0; i <= nb; i++) {
-        xbins[i] = ptcutl * std::exp(i * a);
-      }
       auto num = new TH2D("num", ";#it{p}_{T} (GeV/#it{c});NCls;Efficiency (fake-track rate)", nb, xbins, 4, 4 - 0.5, 8 - 0.5);
       num->Sumw2();
       auto fak = new TH2D("fak", ";#it{p}_{T} (GeV/#it{c});NCls;Fak", nb, xbins, 4, 4 - 0.5, 8 - 0.5);
@@ -848,6 +850,43 @@ void CheckDROF(bool plot = false, bool write = false, const std::string& tracfil
       hVtxMCz->Draw("colz");
       c->Draw();
     }
+    //////////////////////
+    // Fake clusters
+    if (write) {
+      const int nby{4}, nbz{7};
+      double ybins[nby], zbins[nbz];
+      for (int i{0}; i < nby; ++i) {
+        ybins[i] = (4 + i) - 0.5;
+      }
+      for (int i{0}; i < nbz; ++i) {
+        zbins[i] = (0 + i) - 0.5;
+      }
+      auto hFakVal = new TH3D("fakClsVal", "Fake cluster attachment;#it{p}_{T} (GeV/#it{c});NCls;Fake;(fake-cluster rate)", nb, xbins, nby, ybins, nbz, zbins);
+      auto hFakMig = new TH3D("fakClsMig", "Fake cluster attachment;#it{p}_{T} (GeV/#it{c});NCls;Fake;(fake-cluster rate)", nb, xbins, nby, ybins, nbz, zbins);
+
+      for (auto& event : info) {
+        for (auto& part : event) {
+          if (!part.isReconstructable()) {
+            continue;
+          }
+
+          const auto& trk = part.track;
+          for (int iL{0}; iL < 7; ++iL) {
+            if (!trk.hasHitOnLayer(iL) || !trk.isFakeOnLayer(iL) || (part.clusters & (0x1 << iL)) == 0) {
+              continue;
+            }
+            if (trk.hasHitInNextROF()) {
+              hFakMig->Fill(trk.getPt(), trk.getNClusters(), iL);
+            } else {
+              hFakVal->Fill(trk.getPt(), trk.getNClusters(), iL);
+            }
+          }
+        }
+      }
+
+      hFakMig->Write();
+      hFakVal->Write();
+    }
     if (fOut) {
       fOut->Close();
     }