ITS3: fix and extend some QA macros

Signed-off-by: Felix Schlepper <[email protected]>

Author: f3sch

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

@@ -335,6 +335,8 @@ void CheckTracksCA(bool doEffStud = true,
       multiFak->Write("multiFake");
       clone->Write("clones");
     }
+    c1->SaveAs("efficiency.png");
+    c2->SaveAs("vertex.png");
     std::cout << " done." << std::endl;
   }
 
@@ -639,6 +641,7 @@ void CheckTracksCA(bool doEffStud = true,
     c->cd(25);
     doPullCalc(hQoPtPull);
     c->Draw();
+    c->SaveAs("pulls.png");
     std::cout << " done\n";
   }
 

Detectors/Upgrades/ITS3/macros/test/CheckClustersITS3.C

@@ -283,7 +283,7 @@ void CheckClustersITS3(const std::string& clusfile = "o2clus_its.root",
   nt.Draw("cgy:cgx>>h_cgy_vs_cgx_OB(1000, -50, 50, 1000, -50, 50)", "id >= 3456", "colz");
   canvCgXCgY->cd(4);
   nt.Draw("cgy:cgz>>h_cgy_vs_cgz_OB(1000, -100, 100, 1000, -50, 50)", "id >= 3456", "colz");
-  canvCgXCgY->SaveAs("it3clusters_y_vs_x_vs_z.pdf");
+  canvCgXCgY->SaveAs("it3clusters_y_vs_x_vs_z.png");
 
   auto canvdXdZ = new TCanvas("canvdXdZ", "", 1600, 800);
   canvdXdZ->Divide(2, 2);
@@ -295,7 +295,7 @@ void CheckClustersITS3(const std::string& clusfile = "o2clus_its.root",
   nt.Draw("dx:dz>>h_dx_vs_dz_IB_z(1000, -0.01, 0.01, 1000, -0.01, 0.01)", "id < 3456 && abs(cgz) < 2", "colz");
   canvdXdZ->cd(4)->SetLogz();
   nt.Draw("dx:dz>>h_dx_vs_dz_OB_z(1000, -0.01, 0.01, 1000, -0.01, 0.01)", "id >= 3456 && abs(cgz) < 2", "colz");
-  canvdXdZ->SaveAs("it3clusters_dx_vs_dz.pdf");
+  canvdXdZ->SaveAs("it3clusters_dx_vs_dz.png");
 
   auto canvCHXZ = new TCanvas("canvCHXZ", "", 1600, 1600);
   canvCHXZ->Divide(2, 2);
@@ -307,7 +307,7 @@ void CheckClustersITS3(const std::string& clusfile = "o2clus_its.root",
   nt.Draw("(cgz-hgz)*10000:eta>>h_chz_IB(101,-1.4,1.4,101,-50,50)", "id<3456", "prof");
   canvCHXZ->cd(4);
   nt.Draw("(cgz-hgz)*10000:eta>>h_chz_OB(101,-1.4,1.4,101,-50,50)", "id>=3456", "prof");
-  canvCgXCgY->SaveAs("it3clusters_xz_eta.pdf");
+  canvCgXCgY->SaveAs("it3clusters_xz_eta.png");
 
   auto c1 = new TCanvas("p1", "pullX");
   c1->cd();

Detectors/Upgrades/ITS3/macros/test/CheckDCA.C

@@ -54,7 +54,7 @@
 
 namespace fs = std::filesystem;
 
-constexpr auto mMatCorr{o2::base::Propagator::MatCorrType::USEMatCorrNONE};
+constexpr auto mMatCorr{o2::base::Propagator::MatCorrType::USEMatCorrTGeo};
 constexpr float mMaxStep{2};
 
 constexpr float rapMax{0.9};
@@ -79,6 +79,7 @@ void CheckDCA(const std::string& collisioncontextFileName = "collisioncontext.ro
               const std::string& tpcTracksFileName = "tpctracks.root",
               const std::string& itsTracksFileName = "o2trac_its.root",
               const std::string& itstpcTracksFileName = "o2match_itstpc.root",
+              const std::string& geomFileName = "o2sim_geometry.root",
               const std::string& magFileName = "o2sim_grp.root")
 {
   gROOT->SetBatch();
@@ -88,28 +89,12 @@ void CheckDCA(const std::string& collisioncontextFileName = "collisioncontext.ro
   gStyle->SetPadTickX(1);
   gStyle->SetPadTickY(1);
   gErrorIgnoreLevel = 2001; // suppress warnings
-  ProcInfo_t procInfo;
 
   const int nPtBins = 35;
   const int nPtBinsEff = 39;
   double ptLimits[nPtBins] = {0.05, 0.1, 0.15, 0.2, 0.25, 0.3, 0.35, 0.4, 0.45, 0.5, 0.6, 0.7, 0.8, 0.9, 1., 1.1, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2., 2.2, 2.5, 3., 4., 5., 6., 8., 10., 15., 20.};
   double ptLimitsEff[nPtBinsEff] = {0.05, 0.06, 0.07, 0.08, 0.09, 0.1, 0.15, 0.2, 0.25, 0.3, 0.35, 0.4, 0.45, 0.5, 0.6, 0.7, 0.8, 0.9, 1., 1.1, 1.2, 1.3, 1.4, 1.5, 1.6, 1.7, 1.8, 1.9, 2., 2.2, 2.5, 3., 4., 5., 6., 8., 10., 15., 20.};
 
-  const std::regex tf_pattern(R"(tf\d+)");
-  auto tf_matcher = [&tf_pattern](fs::path const& p) -> bool {
-    return std::regex_search(p.string(), tf_pattern);
-  };
-  auto tf_sorter = [&tf_pattern](fs::path const& a, fs::path const& b) -> bool {
-    const auto &as = a.string(), &bs = b.string();
-    std::smatch am, bm;
-    if (std::regex_search(as, am, tf_pattern) && std::regex_search(bs, bm, tf_pattern)) {
-      return std::stoi(am.str().substr(2)) < std::stoi(bm.str().substr(2));
-    } else {
-      LOGP(fatal, "TF Regex matching failed");
-      return false;
-    }
-  };
-
   const int nSpecies = 4;
   std::array<int, nSpecies> pdgCodes{11, 211, 321, 2212};
   auto fGaus = new TF1("fGaus", "gaus", -200., 200.);
@@ -119,6 +104,7 @@ void CheckDCA(const std::string& collisioncontextFileName = "collisioncontext.ro
     {321, "Kaons"},
     {2212, "Protons"}};
   std::map<int, int> colors{{11, kOrange + 7}, {211, kRed + 1}, {321, kAzure + 4}, {2212, kGreen + 2}};
+
   /// ITS
   std::map<int, TH1F*> hDcaxyResAllLayersITS = {
     {11, new TH1F("hDcaxyResElectronsAllLayersITS", "ITS;#it{p}_{T} (GeV/#it{c});#sigma(DCA_{#it{xy}}) (#mum)", nPtBins - 1, ptLimits)},
@@ -287,6 +273,21 @@ void CheckDCA(const std::string& collisioncontextFileName = "collisioncontext.ro
   o2::dataformats::VertexBase collision;
   o2::dataformats::DCA impactParameter;
 
+  const std::regex tf_pattern(R"(tf\d+)");
+  auto tf_matcher = [&tf_pattern](fs::path const& p) -> bool {
+    return std::regex_search(p.string(), tf_pattern);
+  };
+  auto tf_sorter = [&tf_pattern](fs::path const& a, fs::path const& b) -> bool {
+    const auto &as = a.string(), &bs = b.string();
+    std::smatch am, bm;
+    if (std::regex_search(as, am, tf_pattern) && std::regex_search(bs, bm, tf_pattern)) {
+      return std::stoi(am.str().substr(2)) < std::stoi(bm.str().substr(2));
+    } else {
+      LOGP(fatal, "TF Regex matching failed");
+      return false;
+    }
+  };
+
   const auto origWD{fs::current_path()};
   const auto tfDirs = find_dirs(fs::current_path(), tf_matcher, tf_sorter);
   for (const auto& tfDir : tfDirs) {
@@ -305,16 +306,17 @@ void CheckDCA(const std::string& collisioncontextFileName = "collisioncontext.ro
     static bool initOnce{false};
     if (!initOnce) {
       initOnce = true;
+      o2::base::GeometryManager::loadGeometry(geomFileName);
       o2::base::Propagator::initFieldFromGRP(magFileName);
       bz = o2::base::Propagator::Instance()->getNominalBz();
     }
 
     LOGP(info, "Loading ITS Tracks");
-    auto fITSTracks = TFile::Open(itsTracksFileName.c_str(), "READ");
+    std::unique_ptr<TFile> fITSTracks(TFile::Open(itsTracksFileName.c_str(), "READ"));
     auto tITSTracks = fITSTracks->Get<TTree>("o2sim");
-    std::vector<o2::its::TrackITS>* itsTracks{nullptr};
+    auto itsTracks = new std::vector<o2::its::TrackITS>();
     tITSTracks->SetBranchAddress("ITSTrack", &itsTracks);
-    std::vector<o2::MCCompLabel>* itsTrkLab{nullptr};
+    auto itsTrkLab = new std::vector<o2::MCCompLabel>();
     tITSTracks->SetBranchAddress("ITSTrackMCTruth", &itsTrkLab);
 
     for (Long64_t iEntry{0}; tITSTracks->LoadTree(iEntry) >= 0; ++iEntry) {
@@ -367,18 +369,19 @@ void CheckDCA(const std::string& collisioncontextFileName = "collisioncontext.ro
     }
 
     LOGP(info, "Loading ITS-TPC Tracks");
-    auto fITSTPCTracks = TFile::Open(itstpcTracksFileName.c_str(), "READ");
+    std::unique_ptr<TFile> fITSTPCTracks(TFile::Open(itstpcTracksFileName.c_str(), "READ"));
     auto tITSTPCTracks = fITSTPCTracks->Get<TTree>("matchTPCITS");
-    std::vector<o2::dataformats::TrackTPCITS>* itstpcTracks{nullptr};
+    auto itstpcTracks = new std::vector<o2::dataformats::TrackTPCITS>();
     tITSTPCTracks->SetBranchAddress("TPCITS", &itstpcTracks);
-    std::vector<o2::MCCompLabel>* itstpcTrkLab{nullptr};
+    auto itstpcTrkLab = new std::vector<o2::MCCompLabel>();
     tITSTPCTracks->SetBranchAddress("MatchMCTruth", &itstpcTrkLab);
+
     // TPC Tracks
-    auto fTPCTracks = TFile::Open(tpcTracksFileName.c_str(), "READ");
+    std::unique_ptr<TFile> fTPCTracks(TFile::Open(tpcTracksFileName.c_str(), "READ"));
     auto tTPCTracks = fTPCTracks->Get<TTree>("tpcrec");
-    std::vector<o2::tpc::TrackTPC>* tpcTracks{nullptr};
+    auto tpcTracks = new std::vector<o2::tpc::TrackTPC>();
     tTPCTracks->SetBranchAddress("TPCTracks", &tpcTracks);
-    std::vector<o2::MCCompLabel>* tpcTrkLab{nullptr};
+    auto tpcTrkLab = new std::vector<o2::MCCompLabel>();
     tTPCTracks->SetBranchAddress("TPCTracksMCTruth", &tpcTrkLab);
     for (Long64_t iEntry{0}; tITSTPCTracks->LoadTree(iEntry) >= 0; ++iEntry) {
       tITSTPCTracks->GetEntry(iEntry);
@@ -436,26 +439,10 @@ void CheckDCA(const std::string& collisioncontextFileName = "collisioncontext.ro
         }
       }
     }
-
-    delete itsTracks;
-    delete itsTrkLab;
-    delete tpcTracks;
-    delete tpcTrkLab;
-    delete itstpcTracks;
-    delete itstpcTrkLab;
-    delete tITSTracks;
-    delete tTPCTracks;
-    delete tITSTPCTracks;
-    delete fITSTracks;
-    delete fTPCTracks;
-    delete fITSTPCTracks;
-
-    gSystem->GetProcInfo(&procInfo);
-    LOGF(info, "MemVirtual (%ld), MemResident (%ld)", procInfo.fMemVirtual, procInfo.fMemResident);
-    LOGP(info, "Done with {:?}", tfDir.c_str());
-    if (procInfo.fMemResident > 200'000'000) {
-      LOGP(error, "Exceeding 200GBs stopping!");
-      break;
+    for (size_t src{0}; src < mcReader.getNSources(); ++src) {
+      for (size_t eve{0}; eve < mcReader.getNEvents(src); ++eve) {
+        mcReader.releaseTracksForSourceAndEvent(src, eve);
+      }
     }
   }
   LOGP(info, "Restoring original CWD to {:?}", origWD.c_str());