Add resolution class

[u/mrichter/AliRoot.git] / TRD / qaRec / run.C

diff --git a/TRD/qaRec/run.C b/TRD/qaRec/run.C
index a3852317bb9b269fdc44639ab306cb419211cf2a..7f62c432ce3e94b0dc2520c88b09ae55e91a1994 100644 (file)
--- a/TRD/qaRec/run.C
+++ b/TRD/qaRec/run.C
@@ -1,15 +1,68 @@
-void run(const Char_t *list=0x0, Int_t nmax=-1) {
+// Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
+// 
+// Usage:
+//   run.C(tasks, files, entries)
+//   tasks : "ALL" or one/more of the following:
+//     "EFF"  : TRD Tracking Efficiency 
+//     "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations
+//     "RES"  : TRD tracking Resolution
+//     "PID"  : TRD PID - pion efficiency 
+//     "PIDR" : TRD PID - reference data
+// 
+// Authors:
+//   Alex Bercuci (A.Bercuci@gsi.de) 
+//   Markus Fasel (m.Fasel@gsi.de) 
+
+#define BIT(n)       (1 << (n))
+#define SETBIT(n,i)  ((n) |= BIT(i))
+#define TESTBIT(n,i) ((Bool_t)(((n) & BIT(i)) != 0))
+
+const Int_t fknTasks = 3;
+Char_t *fTaskName[fknTasks] = {"Barrel Tracking Effiency", "Combined Tracking Efficiency", "Tracking Resolution"};
+enum AliTRDrecoTasks{
+  kTrackingEfficiency = 0
+  ,kTrackingCombinedEfficiency = 1
+  ,kTrackingResolution = 2
+};
+void run(Char_t *tasks="ALL", const Char_t *files=0x0, Int_t nmax=-1)
+{
+
+
   TStopwatch timer;
   timer.Start();
 
   gSystem->Load("libANALYSIS.so");
   gSystem->Load("libTRDqaRec.so");
-       
+  
+  Int_t fSteerTask = 0; 
+  TObjArray *task = TString(tasks).Tokenize(" ");
+  for(Int_t isel = 0; isel < task->GetEntriesFast(); isel++){
+    TString s = (dynamic_cast<TObjString *>(task->UncheckedAt(isel)))->String();
+    if(s.CompareTo("ALL") == 0){
+      for(Int_t itask = 0; itask < fknTasks; itask++) SETBIT(fSteerTask, itask);
+      continue;
+    } else if(s.CompareTo("EFF") == 0){
+      SETBIT(fSteerTask, kTrackingEfficiency);
+      continue;
+    } else if(s.CompareTo("EFFC") == 0){
+      SETBIT(fSteerTask, kTrackingCombinedEfficiency);
+      continue;
+    } else if(s.CompareTo("RES" ) == 0){
+      SETBIT(fSteerTask, kTrackingResolution);
+      continue;
+    } else{
+      Info("run.C", Form("Task %s not implemented (yet).", s.Data()));
+      continue;
+    }
+  }
+  printf("\n\tRUNNING TRAIN FOR TASKS:\n");
+  for(itask = 0; itask < fknTasks; itask++){
+    if(TESTBIT(fSteerTask, itask)) printf("\t%s\n", fTaskName[itask]);
+  }
 
   //____________________________________________//
-
   gROOT->LoadMacro(Form("%s/TRD/qaRec/CreateESDChain.C", gSystem->ExpandPathName("$ALICE_ROOT")));
-  TChain *chain = CreateESDChain(list, nmax);
+  TChain *chain = CreateESDChain(files, nmax);
   //chain->SetBranchStatus("*", 0);
   chain->SetBranchStatus("*FMD*",0);
   chain->SetBranchStatus("*Calo*",0);
@@ -44,25 +97,39 @@ void run(const Char_t *list=0x0, Int_t nmax=-1) {
 
   //____________________________________________
   // TRD barrel tracking efficiency
-  AliTRDtrackingEfficiency *task2 = new AliTRDtrackingEfficiency();
-  task2->SetDebugLevel(1);
-  mgr->AddTask(task2);
-  //Create containers for input/output
-  AliAnalysisDataContainer *coutput2 = mgr->CreateContainer("TrackingEfficiency", TList::Class(), AliAnalysisManager::kOutputContainer, "TRD.TrackingEfficiency.root");
-  mgr->ConnectInput( task2, 0, coutput1);
-  mgr->ConnectOutput(task2, 0, coutput2);
+  if(TESTBIT(fSteerTask, kTrackingEfficiency)){
+    AliTRDtrackingEfficiency *task2 = new AliTRDtrackingEfficiency();
+    task2->SetDebugLevel(1);
+    mgr->AddTask(task2);
+    //Create containers for input/output
+    AliAnalysisDataContainer *coutput2 = mgr->CreateContainer("TrackingEfficiency", TList::Class(), AliAnalysisManager::kOutputContainer, "TRD.TrackingEfficiency.root");
+    mgr->ConnectInput( task2, 0, coutput1);
+    mgr->ConnectOutput(task2, 0, coutput2);
+  }
 
   //____________________________________________
   // TRD combined tracking efficiency
-  AliTRDtrackingEfficiencyCombined *task3 = new AliTRDtrackingEfficiencyCombined();
-  task3->SetDebugLevel(0);
-  mgr->AddTask(task3);
-  // Create containers for input/output
-  AliAnalysisDataContainer *coutput3 = mgr->CreateContainer("TrackingEfficiencyCombined", TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.TrackingEfficiencyCombined.root");
-  mgr->ConnectInput( task3, 0, coutput1);
-  mgr->ConnectOutput(task3, 0, coutput3);
-
+  if(TESTBIT(fSteerTask, kTrackingCombinedEfficiency)){
+    AliTRDtrackingEfficiencyCombined *task3 = new AliTRDtrackingEfficiencyCombined();
+    task3->SetDebugLevel(0);
+    mgr->AddTask(task3);
+    // Create containers for input/output
+    AliAnalysisDataContainer *coutput3 = mgr->CreateContainer("TrackingEfficiencyCombined", TObjArray::Class(), AliAnalysisManager::kOutputContainer, "TRD.TrackingEfficiencyCombined.root");
+    mgr->ConnectInput( task3, 0, coutput1);
+    mgr->ConnectOutput(task3, 0, coutput3);
+  }
 
+  //____________________________________________
+  // TRD combined tracking efficiency
+  if(TESTBIT(fSteerTask, kTrackingResolution)){
+    AliTRDtrackingResolution *task4 = new AliTRDtrackingResolution();
+    task4->SetDebugLevel(1);
+    mgr->AddTask(task4);
+    // Create containers for input/output
+    AliAnalysisDataContainer *coutput4 = mgr->CreateContainer("Tracking Resolution", TList::Class(), AliAnalysisManager::kOutputContainer, "TRD.TrackingResolution.root");
+    mgr->ConnectInput( task4, 0, coutput1);
+    mgr->ConnectOutput(task4, 0, coutput4);
+  }
 
   if (!mgr->InitAnalysis()) return;
   mgr->PrintStatus();