Adapt macros to use vector instead of MCTruthContainer for ITS,TPC track labels

Author: shahor02

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

@@ -103,7 +103,7 @@ void CheckTracks(std::string tracfile = "o2trac_its.root", std::string clusfile
   std::vector<TrackITS>* recArr = nullptr;
   recTree->SetBranchAddress("ITSTrack", &recArr);
   // Track MC labels
-  o2::dataformats::MCTruthContainer<o2::MCCompLabel>* trkLabArr = nullptr;
+  std::vector<o2::MCCompLabel>* trkLabArr = nullptr;
   recTree->SetBranchAddress("ITSTrackMCTruth", &trkLabArr);
 
   Int_t lastEventIDcl = -1, cf = 0;
@@ -154,7 +154,7 @@ void CheckTracks(std::string tracfile = "o2trac_its.root", std::string clusfile
       continue;
     int loadedEventTracks = frame;
     for (unsigned int i = 0; i < recArr->size(); i++) { // Find the last MC event within this reconstructed entry
-      auto lab = (trkLabArr->getLabels(i))[0];
+      auto lab = (*trkLabArr)[i];
       if (!lab.isValid()) {
         const TrackITS& recTrack = (*recArr)[i];
         fak->Fill(recTrack.getPt());
@@ -210,7 +210,7 @@ void CheckTracks(std::string tracfile = "o2trac_its.root", std::string clusfile
       long lastIndex = (frame == f.lastFrame) ? f.lastIndex : nentr - 1;
 
       for (long i = firstIndex; i <= lastIndex; i++) {
-        auto lab = (trkLabArr->getLabels(i))[0];
+        auto lab = (*trkLabArr)[i];
         if (!lab.isValid() || lab.getSourceID() != 0 || lab.getEventID() != n)
           continue;
         int mcid = lab.getTrackID();

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

@@ -205,7 +205,7 @@ found:
   tree->SetBranchAddress("ITSTrack", &trkArr);
   tree->SetBranchAddress("ITSTrackClusIdx", &clIdx);
   // Track MC labels
-  o2::dataformats::MCTruthContainer<o2::MCCompLabel>* trkLabArr = nullptr;
+  std::vector<o2::MCCompLabel>* trkLabArr = nullptr;
   tree->SetBranchAddress("ITSTrackMCTruth", &trkLabArr);
 
   tree->GetEvent(0);
@@ -214,7 +214,7 @@ found:
   n = 0;
   while (nt--) {
     const TrackITS& t = (*trkArr)[nt];
-    auto lab = (trkLabArr->getLabels(nt))[0];
+    auto lab = (*trkLabArr)[nt];
     if (TMath::Abs(lab.getEventID()) != event)
       continue;
     if (TMath::Abs(lab.getTrackID()) != track)

macro/checkTOFMatching.C

@@ -40,15 +40,15 @@ void checkTOFMatching(bool batchMode = true)
   TTree* tpcTree = (TTree*)ftracksTPC->Get("events");
   std::vector<o2::tpc::TrackTPC>* mTPCTracksArrayInp = new std::vector<o2::tpc::TrackTPC>;
   tpcTree->SetBranchAddress("Tracks", &mTPCTracksArrayInp);
-  o2::dataformats::MCTruthContainer<o2::MCCompLabel>* mcTPC = new o2::dataformats::MCTruthContainer<o2::MCCompLabel>();
+  std::vector<o2::MCCompLabel>* mcTPC = new std::vector<o2::MCCompLabel>();
   tpcTree->SetBranchAddress("TracksMCTruth", &mcTPC);
 
   // getting the ITS tracks
   TFile* ftracksITS = new TFile("o2trac_its.root");
   TTree* itsTree = (TTree*)ftracksITS->Get("o2sim");
   std::vector<o2::its::TrackITS>* mITSTracksArrayInp = new std::vector<o2::its::TrackITS>;
   itsTree->SetBranchAddress("ITSTrack", &mITSTracksArrayInp);
-  o2::dataformats::MCTruthContainer<o2::MCCompLabel>* mcITS = new o2::dataformats::MCTruthContainer<o2::MCCompLabel>();
+  std::vector<o2::MCCompLabel>* mcITS = new std::vector<o2::MCCompLabel>();
   itsTree->SetBranchAddress("ITSTrackMCTruth", &mcITS);
 
   // getting the TOF clusters
@@ -274,23 +274,19 @@ void checkTOFMatching(bool batchMode = true)
       Printf("matched ITStrack index = %d", evIdxITS);
 #endif
       // getting the TPC labels
-      const auto& labelsTPC = mcTPC->getLabels(evIdxTPC);
+      const auto& labelsTPC = (*mcTPC)[evIdxTPC];
 #ifdef DEBUG
-      for (uint ilabel = 0; ilabel < labelsTPC.size(); ilabel++) {
-        Printf("TPC label %d: trackID = %d, eventID = %d, sourceID = %d", ilabel, labelsTPC[ilabel].getTrackID(), labelsTPC[ilabel].getEventID(), labelsTPC[ilabel].getSourceID());
-      }
+      Printf("TPC label: trackID = %d, eventID = %d, sourceID = %d", labelsTPC.getTrackID(), labelsTPC.getEventID(), labelsTPC.getSourceID());
 #endif
 
       // getting the ITS labels
-      const auto& labelsITS = mcITS->getLabels(evIdxITS);
+      const auto& labelsITS = (*mcITS)[evIdxITS];
 #ifdef DEBUG
-      for (uint ilabel = 0; ilabel < labelsITS.size(); ilabel++) {
-        Printf("ITS label %d: trackID = %d, eventID = %d, sourceID = %d", ilabel, labelsITS[ilabel].getTrackID(), labelsITS[ilabel].getEventID(), labelsITS[ilabel].getSourceID());
-      }
+      Printf("ITS label: trackID = %d, eventID = %d, sourceID = %d", ilabel, labelsITS.getTrackID(), labelsITS.getEventID(), labelsITS.getSourceID());
 #endif
       bool bMatched = kFALSE;
       for (uint ilabel = 0; ilabel < labelsTOF.size(); ilabel++) {
-        if ((abs(labelsTPC[0].getTrackID()) == labelsTOF[ilabel].getTrackID() && labelsTPC[0].getEventID() == labelsTOF[ilabel].getEventID() && labelsTPC[0].getSourceID() == labelsTOF[ilabel].getSourceID()) || (labelsITS[0].getTrackID() == labelsTOF[ilabel].getTrackID() && labelsITS[0].getEventID() == labelsTOF[ilabel].getEventID() && labelsITS[0].getSourceID() == labelsTOF[ilabel].getSourceID())) {
+        if ((abs(labelsTPC.getTrackID()) == labelsTOF[ilabel].getTrackID() && labelsTPC.getEventID() == labelsTOF[ilabel].getEventID() && labelsTPC.getSourceID() == labelsTOF[ilabel].getSourceID()) || (labelsITS.getTrackID() == labelsTOF[ilabel].getTrackID() && labelsITS.getEventID() == labelsTOF[ilabel].getEventID() && labelsITS.getSourceID() == labelsTOF[ilabel].getSourceID())) {
           nGoodMatches++;
           bMatched = kTRUE;
           break;
@@ -352,23 +348,19 @@ void checkTOFMatching(bool batchMode = true)
         o2::dataformats::TrackTPCITS trackITSTPC = mTracksArrayInp->at(i);
         const auto evIdxTPCcheck = trackITSTPC.getRefTPC();
         const auto evIdxITScheck = trackITSTPC.getRefITS();
-        const auto& labelsTPCcheck = mcTPC->getLabels(evIdxTPCcheck);
-        for (uint ilabel = 0; ilabel < labelsTPCcheck.size(); ilabel++) {
-          if (abs(labelsTPCcheck[ilabel].getTrackID()) == trackIdTOF && labelsTPCcheck[ilabel].getEventID() == eventIdTOF && labelsTPCcheck[ilabel].getSourceID() == sourceIdTOF) {
+        const auto& labelsTPCcheck = (*mcTPC)[evIdxTPCcheck];
+        if (abs(labelsTPCcheck.getTrackID()) == trackIdTOF && labelsTPCcheck.getEventID() == eventIdTOF && labelsTPCcheck.getSourceID() == sourceIdTOF) {
 #ifdef DEBUG
-            Printf("The TPC track that should have been matched to TOF is number %d", i);
+          Printf("The TPC track that should have been matched to TOF is number %d", i);
 #endif
-            TPCfound = true;
-          }
+          TPCfound = true;
         }
-        const auto& labelsITScheck = mcITS->getLabels(evIdxITScheck);
-        for (uint ilabel = 0; ilabel < labelsITScheck.size(); ilabel++) {
-          if (labelsITScheck[ilabel].getTrackID() == trackIdTOF && labelsITScheck[ilabel].getEventID() == eventIdTOF && labelsITScheck[ilabel].getSourceID() == sourceIdTOF) {
+        const auto& labelsITScheck = (*mcITS)[evIdxITScheck];
+        if (labelsITScheck.getTrackID() == trackIdTOF && labelsITScheck.getEventID() == eventIdTOF && labelsITScheck.getSourceID() == sourceIdTOF) {
 #ifdef DEBUG
-            Printf("The ITS track that should have been matched to TOF is number %d", i);
+          Printf("The ITS track that should have been matched to TOF is number %d", i);
 #endif
-            ITSfound = true;
-          }
+          ITSfound = true;
         }
       }
 #ifdef DEBUG

macro/runCATrackingClusterNative.C

@@ -100,7 +100,7 @@ int runCATrackingClusterNative(TString inputFile, TString outputFile)
 
   vector<TrackTPC> tracks;
   vector<TPCClRefElem> trackClusRefs;
-  MCLabelContainer tracksMC;
+  std::vector<o2::MCCompLabel> tracksMC;
 
   TFile fout(outputFile, "recreate");
   TTree tout("events", "events");

macro/run_trac_ca_its.C

@@ -165,7 +165,7 @@ void run_trac_ca_its(std::string path = "./",
   std::vector<o2::itsmft::ROFRecord> vertROFvec, *vertROFvecPtr = &vertROFvec;
   std::vector<Vertex> vertices, *verticesPtr = &vertices;
 
-  MCLabCont trackLabels, *trackLabelsPtr = &trackLabels;
+  std::vector<o2::MCCompLabel> trackLabels, *trackLabelsPtr = &trackLabels;
   outTree.Branch("ITSTrack", &tracksITSPtr);
   outTree.Branch("ITSTrackClusIdx", &trackClIdxPtr);
   outTree.Branch("ITSTrackMCTruth", &trackLabelsPtr);

macro/run_trac_its.C

@@ -38,6 +38,7 @@
 #include "ITStracking/VertexerTraits.h"
 
 using MCLabCont = o2::dataformats::MCTruthContainer<o2::MCCompLabel>;
+using MCLabContTr = std::vector<o2::MCCompLabel>;
 using Vertex = o2::dataformats::Vertex<o2::dataformats::TimeStamp<int>>;
 
 void run_trac_its(std::string path = "./", std::string outputfile = "o2trac_its.root",
@@ -141,7 +142,7 @@ void run_trac_its(std::string path = "./", std::string outputfile = "o2trac_its.
   std::vector<o2::itsmft::ROFRecord> vertROFvec, *vertROFvecPtr = &vertROFvec;
   std::vector<Vertex> vertices, *verticesPtr = &vertices;
 
-  MCLabCont trackLabels, *trackLabelsPtr = &trackLabels;
+  MCLabContTr trackLabels, *trackLabelsPtr = &trackLabels;
   outTree.Branch("ITSTrack", &tracksITSPtr);
   outTree.Branch("ITSTrackClusIdx", &trackClIdxPtr);
   outTree.Branch("ITSTrackMCTruth", &trackLabelsPtr);