ITS: TrackExtensionStudy more plots

Author: f3sch

Detectors/ITSMFT/ITS/postprocessing/studies/macros/PostTrackExtension.C

@@ -15,10 +15,13 @@
 #include "TError.h"
 #include "TColor.h"
 #include "TCanvas.h"
-#include "TH1F.h"
+#include "TH2D.h"
+#include "TF1.h"
 #include "TEfficiency.h"
+#include "TMarker.h"
 #include "TLegend.h"
 #include "TTree.h"
+#include "TLatex.h"
 
 #include <memory>
 #include <array>
@@ -27,6 +30,7 @@
 
 static constexpr std::array<uint8_t, 9> bitPatternsBefore{15, 30, 31, 60, 62, 63, 120, 124, 126};
 static constexpr std::array<uint8_t, 16> bitPatternsAfter{31, 47, 61, 62, 63, 79, 94, 95, 111, 121, 122, 123, 124, 125, 126, 127};
+inline bool bitsCleared(uint8_t b, uint8_t a) { return (a == b) ? true : (b & ~(a & b)) != 0; }
 static constexpr std::array<int, 16> patternColors = {
   kRed,        // Red
   kBlue,       // Blue
@@ -109,6 +113,10 @@ static const std::array<int, eN> markers{20, 21, 22, 23, 27, 28,
                                          29, 33, 39};
 static const char* const texPtX = "#it{p}_{T} (GeV/#it{c})";
 static const char* const texEff = "Efficiency";
+static const char* const texPtRes = "#sigma(#Delta#it{p}_{T}/#it{p}_{T})";
+static const char* const texDCAxyRes = "#sigma(DCA_{#it{xy}}) (#mum)";
+static const char* const texDCAzRes = "#sigma(DCA_{#it{z}}) (#mum)";
+static const char* const fitOpt{"QWMER"};
 
 void setStyle();
 TEfficiency* makeEff(TFile*, const char* num, const char* den);
@@ -240,10 +248,13 @@ void PostTrackExtension(const char* fileName = "TrackExtensionStudy.root")
       auto h = p->DrawFrame(0.05, 0.0, 10., 1.05);
       h->GetXaxis()->SetTitle(texPtX);
       h->GetYaxis()->SetTitle(texEff);
-      auto leg = new TLegend(0.35, 0.35, 0.7, 0.7);
+      auto leg = new TLegend(0.35, 0.60, 0.7, 0.88);
       leg->SetNColumns(4);
       leg->SetHeader(std::format("BEFORE={:07b} GOOD Pattern AFTER/BEFORE", bitPatternsBefore[i]).c_str());
       for (int j{0}; j < (int)bitPatternsAfter.size(); ++j) {
+        if (bitsCleared(bitPatternsBefore[i], bitPatternsAfter[j])) {
+          continue;
+        }
         auto eff = fIn->Get<TEfficiency>(std::format("eExtensionPatternGood_{:07b}_{:07b}", bitPatternsBefore[i], bitPatternsAfter[j]).c_str());
         stylePattern(eff, j, leg, std::format("{:07b}", bitPatternsAfter[j]).c_str());
         eff->Draw("same");
@@ -263,10 +274,13 @@ void PostTrackExtension(const char* fileName = "TrackExtensionStudy.root")
       auto h = p->DrawFrame(0.05, 0.0, 10., 1.05);
       h->GetXaxis()->SetTitle(texPtX);
       h->GetYaxis()->SetTitle(texEff);
-      auto leg = new TLegend(0.35, 0.35, 0.7, 0.7);
+      auto leg = new TLegend(0.35, 0.60, 0.7, 0.88);
       leg->SetNColumns(4);
       leg->SetHeader(std::format("BEFORE={:07b} FAKE Pattern AFTER/BEFORE", bitPatternsBefore[i]).c_str());
       for (int j{0}; j < (int)bitPatternsAfter.size(); ++j) {
+        if (bitsCleared(bitPatternsBefore[i], bitPatternsAfter[j])) {
+          continue;
+        }
         auto eff = fIn->Get<TEfficiency>(std::format("eExtensionPatternFake_{:07b}_{:07b}", bitPatternsBefore[i], bitPatternsAfter[j]).c_str());
         stylePattern(eff, j, leg, std::format("{:07b}", bitPatternsAfter[j]).c_str());
         eff->Draw("same");
@@ -278,25 +292,292 @@ void PostTrackExtension(const char* fileName = "TrackExtensionStudy.root")
     c->SaveAs("trkExt_fake_pattern_comp.pdf");
   }
 
+  { // DCA
+    auto fGaus = new TF1("fGaus", "gaus", -200., 200.);
+    auto dcaXYVsPtNo = fIn->Get<TH2D>("hDCAxyVsPtResNormal");
+    auto dcaXYVsPtYes = fIn->Get<TH2D>("hDCAxyVsPtResExtended");
+    auto dcazVsPtNo = fIn->Get<TH2D>("hDCAzVsPtResNormal");
+    auto dcazVsPtYes = fIn->Get<TH2D>("hDCAzVsPtResExtended");
+    auto bins = dcazVsPtNo->GetXaxis()->GetXbins();
+    auto dcaXYResNo = new TH1F("hDcaxyResNo", "NORMAL;#it{p}_{T} (GeV/#it{c});#sigma(DCA_{#it{xy}}) (#mum)", bins->GetSize() - 1, bins->GetArray());
+    auto dcaXYResYes = new TH1F("hDcaxyResYes", "EXTENDED;#it{p}_{T} (GeV/#it{c});#sigma(DCA_{#it{xy}}) (#mum)", bins->GetSize() - 1, bins->GetArray());
+    auto dcaZResNo = new TH1F("hDcazResNo", "NORMAL;#it{p}_{T} (GeV/#it{c});#sigma(DCA_{#it{z}}) (#mum)", bins->GetSize() - 1, bins->GetArray());
+    auto dcaZResYes = new TH1F("hDcazResYes", "EXTENDED;#it{p}_{T} (GeV/#it{c});#sigma(DCA_{#it{z}}) (#mum)", bins->GetSize() - 1, bins->GetArray());
+    TH1* proj;
+    for (int iPt{1}; iPt <= bins->GetSize(); ++iPt) {
+      auto ptMin = dcaXYResNo->GetXaxis()->GetBinLowEdge(iPt);
+      if (ptMin < 0.1) {
+        continue;
+      }
+      float minFit = (ptMin < 1.) ? -200. : -75.;
+      float maxFit = (ptMin < 1.) ? 200. : 75.;
+
+      proj = dcaXYVsPtNo->ProjectionY(Form("hProjDCAxy_no_%d", iPt), iPt, iPt);
+      proj->Fit("fGaus", fitOpt, "", minFit, maxFit);
+      dcaXYResNo->SetBinContent(iPt, fGaus->GetParameter(2));
+      dcaXYResNo->SetBinError(iPt, fGaus->GetParError(2));
+
+      proj = dcaXYVsPtYes->ProjectionY(Form("hProjDCAxy_yes_%d", iPt), iPt, iPt);
+      proj->Fit("fGaus", fitOpt, "", minFit, maxFit);
+      dcaXYResYes->SetBinContent(iPt, fGaus->GetParameter(2));
+      dcaXYResYes->SetBinError(iPt, fGaus->GetParError(2));
+
+      proj = dcazVsPtNo->ProjectionY(Form("hProjDCAz_no_%d", iPt), iPt, iPt);
+      proj->Fit("fGaus", fitOpt, "", minFit, maxFit);
+      dcaZResNo->SetBinContent(iPt, fGaus->GetParameter(2));
+      dcaZResNo->SetBinError(iPt, fGaus->GetParError(2));
+
+      proj = dcazVsPtYes->ProjectionY(Form("hProjDCAz_yes_%d", iPt), iPt, iPt);
+      proj->Fit("fGaus", fitOpt, "", minFit, maxFit);
+      dcaZResYes->SetBinContent(iPt, fGaus->GetParameter(2));
+      dcaZResYes->SetBinError(iPt, fGaus->GetParError(2));
+    }
+
+    dcaXYResNo->SetLineColor(kRed);
+    dcaXYResNo->SetMarkerColor(kRed);
+    dcaXYResYes->SetLineColor(kBlue);
+    dcaXYResYes->SetMarkerColor(kBlue);
+    dcaZResNo->SetLineColor(kRed);
+    dcaZResNo->SetMarkerColor(kRed);
+    dcaZResYes->SetLineColor(kBlue);
+    dcaZResYes->SetMarkerColor(kBlue);
+
+    auto c = new TCanvas("cDCA", "", 2 * 800, 600);
+    c->Divide(2, 1);
+    c->cd(1);
+    auto h = gPad->DrawFrame(0.1, 1, 10., 500);
+    h->GetXaxis()->SetTitle(texPtX);
+    h->GetYaxis()->SetTitle(texDCAxyRes);
+    dcaXYResNo->Draw("SAME");
+    dcaXYResYes->Draw("SAME");
+    gPad->SetLogx();
+    gPad->SetLogy();
+    gPad->SetGrid();
+    auto leg = new TLegend(0.20, 0.20, 0.40, 0.40);
+    leg->AddEntry(dcaXYResNo, "Normal");
+    leg->AddEntry(dcaXYResYes, "Extended");
+    leg->Draw();
+
+    c->cd(2);
+    h = gPad->DrawFrame(0.1, 1, 10., 500);
+    h->GetXaxis()->SetTitle(texPtX);
+    h->GetYaxis()->SetTitle(texDCAzRes);
+    dcaZResNo->Draw("SAME");
+    dcaZResYes->Draw("SAME");
+    gPad->SetLogx();
+    gPad->SetLogy();
+    gPad->SetGrid();
+
+    c->SaveAs("trkExt_dca.pdf");
+  }
+
+  return;
   { // Kinematic variables
     auto t = fIn->Get<TTree>("tree");
     auto c = new TCanvas("cKG", "", 800, 600);
     c->Divide(3, 2);
-    auto p = c->cd(1);
-    p->SetGrid();
-    auto h = p->DrawFrame(-.5, 0., .5, 30.);
-    h->GetXaxis()->SetTitle("#it{p}_{T,TRK}-#it{p}_{T,MC}");
-    h->GetYaxis()->SetTitle("n. counts");
-    t->Draw("trk.getPt()-mcTrk.getPt()>>hPtNo(100,-.5,.5)", "isGood&&!isExtended", "HIST;SAME");
-    auto htemp = (TH1F*)p->GetPrimitive("hPtNo");
-    htemp->Scale(1.0 / htemp->Integral("width"));
-    htemp->SetLineColor(kRed);
-    t->Draw("trk.getPt()-mcTrk.getPt()>>hPtYes(100,-.5,.5)", "isGood&&isExtended", "HIST;SAME");
-    htemp = (TH1F*)p->GetPrimitive("hPtYes");
-    htemp->Scale(1.0 / htemp->Integral("width"));
-    htemp->SetLineColor(kBlue);
-    p->Modified();
-    p->Update();
+    {
+      auto p = c->cd(1);
+      p->SetGrid();
+      auto h = p->DrawFrame(-.6, 0., .6, 9.);
+      h->GetXaxis()->SetTitle("#frac{Q^{2}}{p_{T,TRK}}-#frac{Q^{2}}{p_{T,MC}}");
+      h->GetYaxis()->SetTitle("n. counts");
+      t->Draw("trk.getQ2Pt()-mcTrk.getQ2Pt()>>hPtNo(100,-.6,.6)", "isGood&&!isExtended", "HIST;SAME");
+      auto hNo = (TH1F*)p->GetPrimitive("hPtNo");
+      hNo->Scale(1.0 / hNo->Integral("width"));
+      hNo->SetLineColor(kRed);
+      auto fitNo = new TF1("fitNo", "gaus", -0.04, 0.04);
+      hNo->Fit(fitNo, "QR");
+      fitNo->SetLineColor(kRed);
+      fitNo->Draw("SAME");
+      auto textNo = new TLatex(-0.55, 8.2, Form("#mu = %.3f, #sigma = %.3f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textNo->SetTextColor(kRed);
+      textNo->SetNDC(false);
+      textNo->SetTextSize(0.05);
+      textNo->Draw();
+
+      t->Draw("trk.getQ2Pt()-mcTrk.getQ2Pt()>>hPtYes(100,-.6,.6)", "isGood&&isExtended", "HIST;SAME");
+      auto hYes = (TH1F*)p->GetPrimitive("hPtYes");
+      hYes->Scale(1.0 / hYes->Integral("width"));
+      hYes->SetLineColor(kBlue);
+      auto fitYes = new TF1("fitYes", "gaus", -0.04, 0.04);
+      hYes->Fit(fitYes, "QR");
+      fitYes->SetLineColor(kBlue);
+      fitYes->Draw("SAME");
+      auto textYes = new TLatex(-0.55, 7, Form("#mu = %.4f, #sigma = %.4f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textYes->SetTextColor(kBlue);
+      textYes->SetNDC(false);
+      textYes->SetTextSize(0.05);
+      textYes->Draw();
+
+      p->Modified();
+      p->Update();
+    }
+    {
+      auto p = c->cd(2);
+      p->SetGrid();
+      auto h = p->DrawFrame(-3, 0., 3, 2.);
+      h->GetXaxis()->SetTitle("Y_{TRK}-Y_{MC}");
+      h->GetYaxis()->SetTitle("n. counts");
+      t->Draw("trk.getY()-mcTrk.getY()>>hYNo(100,-3,3)", "isGood&&!isExtended", "HIST;SAME");
+      auto hNo = (TH1F*)p->GetPrimitive("hYNo");
+      hNo->Scale(1.0 / hNo->Integral("width"));
+      hNo->SetLineColor(kRed);
+      auto fitNo = new TF1("fitNo", "gaus", -0.5, 0.5);
+      hNo->Fit(fitNo, "QR");
+      fitNo->SetLineColor(kRed);
+      fitNo->Draw("SAME");
+      auto textNo = new TLatex(-2, 1.7, Form("#mu = %.3f, #sigma = %.3f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textNo->SetTextColor(kRed);
+      textNo->SetNDC(false);
+      textNo->SetTextSize(0.05);
+      textNo->Draw();
+
+      t->Draw("trk.getY()-mcTrk.getY()>>hYYes(100,-3,3)", "isGood&&isExtended", "HIST;SAME");
+      auto hYes = (TH1F*)p->GetPrimitive("hYYes");
+      hYes->Scale(1.0 / hYes->Integral("width"));
+      hYes->SetLineColor(kBlue);
+      auto fitYes = new TF1("fitYes", "gaus", -0.5, 0.5);
+      hYes->Fit(fitYes, "QR");
+      fitYes->SetLineColor(kBlue);
+      fitYes->Draw("SAME");
+      auto textYes = new TLatex(-2, 1.5, Form("#mu = %.4f, #sigma = %.4f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textYes->SetTextColor(kBlue);
+      textYes->SetNDC(false);
+      textYes->SetTextSize(0.05);
+      textYes->Draw();
+
+      p->Modified();
+      p->Update();
+    }
+    {
+      auto p = c->cd(3);
+      p->SetGrid();
+      auto h = p->DrawFrame(-2, 0., 2, 4.2);
+      h->GetXaxis()->SetTitle("Z_{TRK}-Z_{MC}");
+      h->GetYaxis()->SetTitle("n. counts");
+      t->Draw("trk.getZ()-mcTrk.getZ()>>hZNo(100,-2,2)", "isGood&&!isExtended", "HIST;SAME");
+      auto hNo = (TH1F*)p->GetPrimitive("hZNo");
+      hNo->Scale(1.0 / hNo->Integral("width"));
+      hNo->SetLineColor(kRed);
+      auto fitNo = new TF1("fitNo", "gaus", -0.2, 0.2);
+      hNo->Fit(fitNo, "QR");
+      fitNo->SetLineColor(kRed);
+      fitNo->Draw("SAME");
+      auto textNo = new TLatex(-1.7, 3.8, Form("#mu = %.3f, #sigma = %.3f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textNo->SetTextColor(kRed);
+      textNo->SetNDC(false);
+      textNo->SetTextSize(0.05);
+      textNo->Draw();
+
+      t->Draw("trk.getZ()-mcTrk.getZ()>>hZYes(100,-2,2)", "isGood&&isExtended", "HIST;SAME");
+      auto hYes = (TH1F*)p->GetPrimitive("hZYes");
+      hYes->Scale(1.0 / hYes->Integral("width"));
+      hYes->SetLineColor(kBlue);
+      auto fitYes = new TF1("fitYes", "gaus", -0.2, 0.2);
+      hYes->Fit(fitYes, "QR");
+      fitYes->SetLineColor(kBlue);
+      fitYes->Draw("SAME");
+      auto textYes = new TLatex(-1.7, 3.5, Form("#mu = %.4f, #sigma = %.4f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textYes->SetTextColor(kBlue);
+      textYes->SetNDC(false);
+      textYes->SetTextSize(0.05);
+      textYes->Draw();
+
+      p->Modified();
+      p->Update();
+    }
+    {
+      auto p = c->cd(4);
+      p->SetGrid();
+      auto h = p->DrawFrame(-0.02, 0., 0.02, 370.);
+      h->GetXaxis()->SetTitle("TGL_{TRK}-TGL_{MC}");
+      h->GetYaxis()->SetTitle("n. counts");
+      t->Draw("trk.getTgl()-mcTrk.getTgl()>>hTglNo(100,-0.02,0.02)", "isGood&&!isExtended", "HIST;SAME");
+      auto hNo = (TH1F*)p->GetPrimitive("hTglNo");
+      hNo->Scale(1.0 / hNo->Integral("width"));
+      hNo->SetLineColor(kRed);
+      auto fitNo = new TF1("fitNo", "gaus", -0.003, 0.003);
+      hNo->Fit(fitNo, "QR");
+      fitNo->SetLineColor(kRed);
+      fitNo->Draw("SAME");
+      auto textNo = new TLatex(-0.018, 330, Form("#mu = %.3f, #sigma = %.3f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textNo->SetTextColor(kRed);
+      textNo->SetNDC(false);
+      textNo->SetTextSize(0.05);
+      textNo->Draw();
+
+      t->Draw("trk.getTgl()-mcTrk.getTgl()>>hTglYes(100,-0.02,0.02)", "isGood&&isExtended", "HIST;SAME");
+      auto hYes = (TH1F*)p->GetPrimitive("hTglYes");
+      hYes->Scale(1.0 / hYes->Integral("width"));
+      hYes->SetLineColor(kBlue);
+      auto fitYes = new TF1("fitYes", "gaus", -0.003, 0.003);
+      hYes->Fit(fitYes, "QR");
+      fitYes->SetLineColor(kBlue);
+      fitYes->Draw("SAME");
+      auto textYes = new TLatex(-0.018, 310, Form("#mu = %.6f, #sigma = %.6f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textYes->SetTextColor(kBlue);
+      textYes->SetNDC(false);
+      textYes->SetTextSize(0.05);
+      textYes->Draw();
+
+      p->Modified();
+      p->Update();
+    }
+    {
+      auto p = c->cd(5);
+      p->SetGrid();
+      auto h = p->DrawFrame(-0.08, 0., 0.08, 80.);
+      h->GetXaxis()->SetTitle("SNP_{TRK}-SNP_{MC}");
+      h->GetYaxis()->SetTitle("n. counts");
+      t->Draw("trk.getSnp()-mcTrk.getSnp()>>hSnpNo(100,-0.08,0.08)", "isGood&&!isExtended", "HIST;SAME");
+      auto hNo = (TH1F*)p->GetPrimitive("hSnpNo");
+      hNo->Scale(1.0 / hNo->Integral("width"));
+      hNo->SetLineColor(kRed);
+      auto fitNo = new TF1("fitNo", "gaus", -0.03, 0.03);
+      hNo->Fit(fitNo, "QR");
+      fitNo->SetLineColor(kRed);
+      fitNo->Draw("SAME");
+      auto textNo = new TLatex(-0.07, 72, Form("#mu = %.3f, #sigma = %.3f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textNo->SetTextColor(kRed);
+      textNo->SetNDC(false);
+      textNo->SetTextSize(0.05);
+      textNo->Draw();
+
+      t->Draw("trk.getSnp()-mcTrk.getSnp()>>hSnpYes(100,-0.08,0.08)", "isGood&&isExtended", "HIST;SAME");
+      auto hYes = (TH1F*)p->GetPrimitive("hSnpYes");
+      hYes->Scale(1.0 / hYes->Integral("width"));
+      hYes->SetLineColor(kBlue);
+      auto fitYes = new TF1("fitYes", "gaus", -0.03, 0.03);
+      hYes->Fit(fitYes, "QR");
+      fitYes->SetLineColor(kBlue);
+      fitYes->Draw("SAME");
+      auto textYes = new TLatex(-0.07, 66, Form("#mu = %.6f, #sigma = %.6f", fitNo->GetParameter(1), fitNo->GetParameter(2)));
+      textYes->SetTextColor(kBlue);
+      textYes->SetNDC(false);
+      textYes->SetTextSize(0.05);
+      textYes->Draw();
+
+      p->Modified();
+      p->Update();
+    }
+    {
+      auto p = c->cd(6);
+      auto legend = new TLegend(0.2, 0.2, 0.8, 0.8);
+      legend->SetTextSize(0.06);
+      legend->SetLineWidth(3);
+      legend->SetHeader("GOOD tracks", "C");
+      auto mBlue = new TMarker();
+      mBlue->SetMarkerColor(kBlue);
+      mBlue->SetMarkerSize(4);
+      legend->AddEntry(mBlue, "extended", "p");
+      auto mRed = new TMarker();
+      mRed->SetMarkerColor(kRed);
+      mRed->SetMarkerSize(4);
+      legend->AddEntry(mRed, "normal", "p");
+      legend->SetLineColor(kRed);
+      legend->Draw();
+    }
     c->SaveAs("trkExt_kinematics.pdf");
   }
 }