TRD/qaRec/run.C

   1 // Steer TRD QA train for Reconstruction (Clusterizer, Tracking and PID).
   2 //
   3 // Usage:
   4 //   run.C(tasks, files, entries)
   5 //   tasks : "ALL" or one/more of the following:
   6 //     "EFF"  : TRD Tracking Efficiency
   7 //     "EFFC" : TRD Tracking Efficiency Combined (barrel + stand alone) - only in case of simulations
   8 //     "RES"  : TRD tracking Resolution
   9 //     "CAL"  : TRD calibration
  10 //     "PID"  : TRD PID - pion efficiency
  11 //     "PIDR" : TRD PID - reference data
  12 //     "DET"  : Basic TRD Detector checks
  13 //     "NOFR" : Data set does not have AliESDfriends.root
  14 //     "NOMC" : Data set does not have Monte Carlo Informations (real data), so all tasks which rely
  15 //              on MC information are switched off
  16 //
  17 // In compiled mode :
  18 // Don't forget to load first the libraries
  19 // gSystem->Load("libMemStat.so")
  20 // gSystem->Load("libMemStatGui.so")
  21 // gSystem->Load("libANALYSIS.so")
  22 // gSystem->Load("libTRDqaRec.so")
  23 //
  24 // Authors:
  25 //   Alex Bercuci (A.Bercuci@gsi.de)
  26 //   Markus Fasel (m.Fasel@gsi.de)
  27
  28 #ifndef __CINT__
  29 #include <Riostream.h>
  30
  31 #include "TStopwatch.h"
  32 #include "TMemStat.h"
  33 #include "TMemStatViewerGUI.h"
  34
  35 #include "TROOT.h"
  36 #include "TSystem.h"
  37 #include "TError.h"
  38 #include "TChain.h"
  39
  40 #include "AliMagFMaps.h"
  41 #include "AliTracker.h"
  42 #include "AliLog.h"
  43 #include "AliCDBManager.h"
  44 #include "AliAnalysisManager.h"
  45 #include "AliAnalysisDataContainer.h"
  46 #include "AliMCEventHandler.h"
  47 #include "AliESDInputHandler.h"
  48
  49 #include "TRD/AliTRDtrackerV1.h"
  50 #include "TRD/AliTRDcalibDB.h"
  51 #include "TRD/qaRec/AliTRDtrackInfoGen.h"
  52 #include "TRD/qaRec/AliTRDtrackingEfficiency.h"
  53 #include "TRD/qaRec/AliTRDtrackingEfficiencyCombined.h"
  54 #include "TRD/qaRec/AliTRDtrackingResolution.h"
  55 #include "TRD/qaRec/AliTRDcalibration.h"
  56 #include "TRD/qaRec/AliTRDpidChecker.h"
  57 #include "TRD/qaRec/AliTRDpidRefMaker.h"
  58 #include "TRD/qaRec/AliTRDcheckDetector.h"
  59 #endif
  60
  61 #include "run.h"
  62
  63 Bool_t MEM = kFALSE;
  64
  65 TChain* CreateESDChain(const char* filename = 0x0, Int_t nfiles=-1 );
  66 void run(Char_t *tasks="ALL", const Char_t *files=0x0, Int_t nmax=-1)
  67 {
  68   TMemStat *mem = 0x0;
  69   if(MEM){
  70     gSystem->Load("libMemStat.so");
  71     gSystem->Load("libMemStatGui.so");
  72     mem = new TMemStat("new, gnubuildin");
  73     mem->AddStamp("Start");
  74   }
  75
  76   TStopwatch timer;
  77   timer.Start();
  78
  79   gSystem->Load("libANALYSIS.so");
  80   gSystem->Load("libTRDqaRec.so");
  81
  82   Bool_t fHasMCdata = kTRUE;
  83   Bool_t fHasFriends = kTRUE;
  84   TObjArray *tasksArray = TString(tasks).Tokenize(" ");
  85
  86   Int_t fSteerTask = 0; SETBIT(fSteerTask, kInfoGen);
  87   for(Int_t isel = 0; isel < tasksArray->GetEntriesFast(); isel++){
  88     TString s = (dynamic_cast<TObjString *>(tasksArray->UncheckedAt(isel)))->String();
  89     if(s.CompareTo("ALL") == 0){
  90       for(Int_t itask = 1; itask < fknTasks; itask++) SETBIT(fSteerTask, itask);
  91       continue;
  92     } else if(s.CompareTo("NOFR") == 0){
  93       fHasFriends = kFALSE;
  94     } else if(s.CompareTo("NOMC") == 0){
  95       fHasMCdata = kFALSE;
  96     } else {
  97       Bool_t foundOpt = kFALSE;
  98       for(Int_t itask = 1; itask < fknTasks; itask++){
  99         if(s.CompareTo(fTaskOpt[itask]) != 0) continue;
 100         SETBIT(fSteerTask, itask);
 101         foundOpt = kTRUE;
 102         break;
 103       }
 104       if(!foundOpt) Info("run.C", Form("Task %s not implemented (yet).", s.Data()));
 105     }
 106   }
 107   // define task list pointers;
 108   AliTRDrecoTask *taskPtr[fknTasks], *task = 0x0;
 109   memset(taskPtr, 0, fknTasks*sizeof(AliAnalysisTask*));
 110
 111   //____________________________________________//
 112   //gROOT->LoadMacro(Form("%s/TRD/qaRec/CreateESDChain.C", gSystem->ExpandPathName("$ALICE_ROOT")));
 113   TChain *chain = CreateESDChain(files, nmax);
 114   //chain->SetBranchStatus("*", 0);
 115   chain->SetBranchStatus("*FMD*",0);
 116   chain->SetBranchStatus("*Calo*",0);
 117   chain->SetBranchStatus("Tracks", 1);
 118   chain->SetBranchStatus("ESDfriend*",1);
 119   chain->Lookup();
 120   chain->GetListOfFiles()->Print();
 121   printf("\n ----> CHAIN HAS %d ENTRIES <----\n\n", (Int_t)chain->GetEntries());
 122
 123   AliLog::SetGlobalLogLevel(AliLog::kError);
 124
 125   //____________________________________________
 126   // Make the analysis manager
 127   AliAnalysisManager *mgr = new AliAnalysisManager("TRD QA Reconstruction Manager");
 128   //mgr->SetSpecialOutputLocation(source); // To Be Changed
 129   AliVEventHandler *esdH = 0x0, *mcH = 0x0;
 130   mgr->SetInputEventHandler(esdH = new AliESDInputHandler);
 131   if(fHasMCdata) mgr->SetMCtruthEventHandler(mcH = new AliMCEventHandler());
 132   //mgr->SetDebugLevel(10);
 133
 134   //____________________________________________
 135   // TRD track summary generator
 136   mgr->AddTask(task = new AliTRDtrackInfoGen());
 137   taskPtr[(Int_t)kInfoGen] = task;
 138   task->SetDebugLevel(1);
 139   task->SetMCdata(fHasMCdata);
 140   // Create containers for input/output
 141   AliAnalysisDataContainer *cinput1 = mgr->CreateContainer("data", TChain::Class(), AliAnalysisManager::kInputContainer);
 142   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("trackInfo", TObjArray::Class(), AliAnalysisManager::kExchangeContainer);
 143   AliAnalysisDataContainer *coutput1a = mgr->CreateContainer("eventInfo", AliTRDeventInfo::Class(), AliAnalysisManager::kExchangeContainer);
 144   mgr->ConnectInput( task, 0, cinput1);
 145   mgr->ConnectOutput(task, 0, coutput1);
 146   mgr->ConnectOutput(task, 1, coutput1a);
 147
 148   //____________________________________________
 149   // TRD detector checker
 150         if(TESTBIT(fSteerTask, kCheckDetector)){
 151     mgr->AddTask(task = new AliTRDcheckDetector());
 152     taskPtr[(Int_t)kCheckDetector] = task;
 153     task->SetDebugLevel(4);
 154     task->SetMCdata(fHasMCdata);
 155
 156     // Create containers for input/output
 157     mgr->ConnectInput( task, 0, coutput1);
 158     mgr->ConnectInput( task, 1, coutput1a);
 159     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
 160   }
 161
 162   //____________________________________________
 163   // TRD barrel tracking efficiency
 164   if(fHasMCdata && TESTBIT(fSteerTask, kTrackingEfficiency)){
 165     mgr->AddTask(task = new AliTRDtrackingEfficiency());
 166     taskPtr[(Int_t)kTrackingEfficiency] = task;
 167     task->SetDebugLevel(0);
 168
 169     //Create containers for input/output
 170     mgr->ConnectInput( task, 0, coutput1);
 171     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
 172   }
 173
 174   //____________________________________________
 175   // TRD combined tracking efficiency
 176   if(fHasMCdata && TESTBIT(fSteerTask, kTrackingCombinedEfficiency)){
 177     mgr->AddTask(task = new AliTRDtrackingEfficiencyCombined());
 178     taskPtr[(Int_t)kTrackingCombinedEfficiency] = task;
 179     task->SetDebugLevel(0);
 180
 181     // Create containers for input/output
 182     mgr->ConnectInput( task, 0, coutput1);
 183     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
 184   }
 185
 186   //____________________________________________
 187   // TRD tracking resolution
 188   if(TESTBIT(fSteerTask, kTrackingResolution)){
 189     mgr->AddTask(task = new AliTRDtrackingResolution());
 190     taskPtr[(Int_t)kTrackingResolution] = task;
 191     task->SetMCdata(fHasMCdata);
 192     task->SetPostProcess(kFALSE);
 193     task->SetDebugLevel(1);
 194
 195     // Create containers for input/output
 196     mgr->ConnectInput( task, 0, coutput1);
 197     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
 198   }
 199
 200   //____________________________________________
 201   // TRD calibration
 202   if(TESTBIT(fSteerTask, kCalibration)){
 203     mgr->AddTask(task = new AliTRDcalibration());
 204     taskPtr[(Int_t)kCalibration] = task;
 205     ((AliTRDcalibration*)task)->SetLow(0);
 206     ((AliTRDcalibration*)task)->SetHigh(30);
 207     ((AliTRDcalibration*)task)->SetFillZero(kFALSE);
 208     task->SetDebugLevel(0);
 209
 210     // Create containers for input/output
 211     mgr->ConnectInput(task, 0, coutput1);
 212     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
 213   }
 214
 215   //____________________________________________
 216   // TRD pid checker
 217   if(TESTBIT(fSteerTask, kPIDChecker)){
 218     mgr->AddTask(task = new AliTRDpidChecker());
 219     taskPtr[(Int_t)kPIDChecker] = task;
 220     task->SetDebugLevel(0);
 221     task->SetMCdata(fHasMCdata);
 222
 223     // Create containers for input/output
 224     mgr->ConnectInput( task, 0, coutput1);
 225     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
 226   }
 227
 228
 229   //____________________________________________
 230   // TRD pid reference
 231   if(TESTBIT(fSteerTask, kPIDRefMaker)){
 232     mgr->AddTask(task = new AliTRDpidRefMaker());
 233     taskPtr[(Int_t)kPIDRefMaker] = task;
 234     task->SetDebugLevel(0);
 235     task->SetMCdata(fHasMCdata);
 236
 237     // Create containers for input/output
 238     mgr->ConnectInput( task, 0, coutput1);
 239     mgr->ConnectOutput(task, 0, mgr->CreateContainer(task->GetName(), TObjArray::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%s.root", task->GetName())));
 240     mgr->ConnectOutput(task, 1, mgr->CreateContainer(Form("%sNN", task->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sNN.root", task->GetName())));
 241     mgr->ConnectOutput(task, 2, mgr->CreateContainer(Form("%sLQ", task->GetName()), TTree::Class(), AliAnalysisManager::kOutputContainer, Form("TRD.Task%sLQ.root", task->GetName())));
 242   }
 243
 244
 245   if (!mgr->InitAnalysis()) return;
 246   printf("\n\tRUNNING TRAIN FOR TASKS:\n");
 247   for(Int_t itask = 1; itask < fknTasks; itask++){
 248     if(TESTBIT(fSteerTask, itask)) printf("\t   %s [%s]\n", taskPtr[itask]->GetTitle(), taskPtr[itask]->GetName());
 249   }
 250   printf("\n\n");
 251   //mgr->PrintStatus();
 252
 253
 254   AliCDBManager *cdbManager = AliCDBManager::Instance();
 255   cdbManager->SetDefaultStorage("local://$ALICE_ROOT");
 256   //cdbManager->SetSpecificStorage("TRD/Calib/FEE","local:///u/bailhach/aliroot/database30head/database");
 257   cdbManager->SetRun(0);
 258   cdbManager->SetCacheFlag(kFALSE);
 259
 260   // initialize TRD settings
 261   AliMagFMaps *field = new AliMagFMaps("Maps","Maps", 2, 1., 10., AliMagFMaps::k5kG);
 262   AliTracker::SetFieldMap(field, kTRUE);
 263   AliTRDcalibDB *cal = AliTRDcalibDB::Instance();
 264   AliTRDtrackerV1::SetNTimeBins(cal->GetNumberOfTimeBins());
 265
 266
 267   mgr->StartAnalysis("local",chain);
 268
 269   timer.Stop();
 270   timer.Print();
 271
 272   cal->Terminate();
 273   delete field;
 274   delete cdbManager;
 275   for(Int_t it=fknTasks-1; it>=0; it--) if(taskPtr[it]) delete taskPtr[it];
 276   if(mcH) delete mcH;
 277   delete esdH;
 278   delete mgr;
 279   delete chain;
 280   if(MEM) delete mem;
 281   if(MEM) TMemStatViewerGUI::ShowGUI();
 282 }
 283
 284
 285 TChain* CreateESDChain(const char* filename, Int_t nfiles)
 286 {
 287   // Create the chain
 288   TChain* chain = new TChain("esdTree");
 289
 290   if(!filename){
 291     chain->Add(Form("%s/AliESDs.root", gSystem->pwd()));
 292     return chain;
 293   }
 294
 295
 296   // read ESD files from the input list.
 297   ifstream in;
 298   in.open(filename);
 299   TString esdfile;
 300   while(in.good() && (nfiles--) ) {
 301     in >> esdfile;
 302     if (!esdfile.Contains("root")) continue; // protection
 303     chain->Add(esdfile.Data());
 304   }
 305
 306   in.close();
 307
 308   return chain;
 309 }