PWG1/TPC/AliPerformanceTask.cxx

   1 /**************************************************************************\r
   2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
   3 *                                                                        *\r
   4 * Author: The ALICE Off-line Project.                                    *\r
   5 * Contributors are mentioned in the code where appropriate.              *\r
   6 *                                                                        *\r
   7 * Permission to use, copy, modify and distribute this software and its   *\r
   8 * documentation strictly for non-commercial purposes is hereby granted   *\r
   9 * without fee, provided that the above copyright notice appears in all   *\r
  10 * copies and that both the copyright notice and this permission notice   *\r
  11 * appear in the supporting documentation. The authors make no claims     *\r
  12 * about the suitability of this software for any purpose. It is          *\r
  13 * provided "as is" without express or implied warranty.                  *\r
  14 **************************************************************************/\r
  15 \r
  16 //------------------------------------------------------------------------------\r
  17 // Implementation of the AliPerformanceTask class. It checks reconstruction performance \r
  18 // for the reconstructed vs MC particle tracks under several conditions. For real data \r
  19 // the control QA histograms are filled.\r
  20 //\r
  21 // The comparison output objects deriving from AliPerformanceObject \r
  22 // (e.g. AliPerformanceRes, AliPerformanceEff, AliPerformanceDEdx, AliPerformanceDCA ...) \r
  23 // are stored in the output file (details in description of these classes).\r
  24 // \r
  25 // Author: J.Otwinowski 01/04/2009 \r
  26 // Changes by M.Knichel 15/10/2010\r
  27 //------------------------------------------------------------------------------\r
  28 \r
  29 #include "iostream"\r
  30 \r
  31 #include "TChain.h"\r
  32 #include "TTree.h"\r
  33 #include "TH1F.h"\r
  34 #include "TCanvas.h"\r
  35 #include "TList.h"\r
  36 #include "TFile.h"\r
  37 #include "TSystem.h"\r
  38 \r
  39 #include "AliAnalysisTask.h"\r
  40 #include "AliAnalysisManager.h"\r
  41 #include "AliESDEvent.h"\r
  42 #include "AliESDfriend.h"\r
  43 #include "AliMCEvent.h"\r
  44 #include "AliESDInputHandler.h"\r
  45 #include "AliMCEventHandler.h"\r
  46 #include "AliESDVertex.h"\r
  47 #include "AliMagF.h"\r
  48 #include "AliTracker.h"\r
  49 #include "AliGeomManager.h"\r
  50 #include "AliCDBManager.h"\r
  51 \r
  52 #include "AliCentrality.h"\r
  53 #include "AliESDVZERO.h"\r
  54 #include "AliMultiplicity.h"\r
  55 \r
  56 #include "AliMCInfo.h"\r
  57 #include "AliESDRecInfo.h"\r
  58 #include "AliMCInfoCuts.h"\r
  59 #include "AliRecInfoCuts.h"\r
  60 #include "AliPerformanceObject.h"\r
  61 #include "AliTPCPerformanceSummary.h"\r
  62 #include "AliPerformanceTPC.h"\r
  63 #include "AliPerformanceDEdx.h"\r
  64 #include "AliPerformanceMatch.h"\r
  65 #include "AliPerformanceTask.h"\r
  66 \r
  67 \r
  68 using namespace std;\r
  69 \r
  70 ClassImp(AliPerformanceTask)\r
  71 \r
  72 //_____________________________________________________________________________\r
  73 AliPerformanceTask::AliPerformanceTask() \r
  74   : AliAnalysisTaskSE("Performance")\r
  75   , fESD(0)\r
  76   , fESDfriend(0)\r
  77   , fMC(0)\r
  78   , fOutput(0)\r
  79   , fOutputSummary(0)\r
  80   , fPitList(0)\r
  81   , fCompList(0)\r
  82   , fUseMCInfo(kFALSE)\r
  83   , fUseESDfriend(kFALSE)\r
  84   , fUseHLT(kFALSE)\r
  85   , fUseTerminate(kTRUE)\r
  86   , fUseCentrality(0)\r
  87   , fUseOCDB(kTRUE)\r
  88   , fUseCentralityBin(0)\r
  89 {\r
  90   // Dummy Constructor\r
  91   // should not be used\r
  92 }\r
  93 \r
  94 //_____________________________________________________________________________\r
  95 AliPerformanceTask::AliPerformanceTask(const char *name, const char */*title*/) \r
  96   : AliAnalysisTaskSE(name)\r
  97   , fESD(0)\r
  98   , fESDfriend(0)\r
  99   , fMC(0)\r
 100   , fOutput(0)\r
 101   , fOutputSummary(0)\r
 102   , fPitList(0)\r
 103   , fCompList(0)\r
 104   , fUseMCInfo(kFALSE)\r
 105   , fUseESDfriend(kFALSE)\r
 106   , fUseHLT(kFALSE)\r
 107   , fUseTerminate(kTRUE)\r
 108   , fUseCentrality(0)\r
 109   , fUseOCDB(kTRUE)\r
 110   , fUseCentralityBin(0)\r
 111 {\r
 112   // Constructor\r
 113 \r
 114   // Define input and output slots here\r
 115   DefineOutput(0, TTree::Class());\r
 116   DefineOutput(1, TList::Class());\r
 117 \r
 118   // create the list for comparison objects\r
 119   fCompList = new TList;\r
 120 }\r
 121 \r
 122 //_____________________________________________________________________________\r
 123 AliPerformanceTask::~AliPerformanceTask()\r
 124 {\r
 125   if (fOutput)     delete fOutput;    fOutput   = 0; \r
 126   if (fOutputSummary) delete fOutputSummary; fOutputSummary = 0;\r
 127   if (fCompList)   delete fCompList;  fCompList = 0; \r
 128 }\r
 129 \r
 130 //_____________________________________________________________________________\r
 131 Bool_t AliPerformanceTask::AddPerformanceObject(AliPerformanceObject *pObj) \r
 132 {\r
 133   // add comparison object to the list\r
 134   if(pObj == 0) {\r
 135     Printf("ERROR: Could not add comparison object");\r
 136     return kFALSE;\r
 137   }\r
 138 \r
 139   // add object to the list\r
 140   fCompList->AddLast(pObj);\r
 141        \r
 142 return kTRUE;\r
 143 }\r
 144 \r
 145 //_____________________________________________________________________________\r
 146 void AliPerformanceTask::UserCreateOutputObjects()\r
 147 {\r
 148   // Create histograms\r
 149   // Called once\r
 150 \r
 151   // create output list\r
 152   fOutput = new TList;\r
 153   fOutput->SetOwner();\r
 154   fPitList = fOutput->MakeIterator();\r
 155   \r
 156   // create output list\r
 157   //fOutputSummary = new TTree;\r
 158   \r
 159   // add comparison objects to the output\r
 160   AliPerformanceObject *pObj=0;\r
 161   Int_t count=0;\r
 162   TIterator *pitCompList = fCompList->MakeIterator();\r
 163   pitCompList->Reset();\r
 164   while(( pObj = (AliPerformanceObject *)pitCompList->Next()) != NULL) {\r
 165     fOutput->Add(pObj);\r
 166     count++;\r
 167   }\r
 168   Printf("UserCreateOutputObjects(): Number of output comparison objects: %d \n", count);\r
 169   \r
 170   PostData(1, fOutput);  \r
 171   PostData(0, fOutputSummary);  \r
 172 }\r
 173 \r
 174 //_____________________________________________________________________________\r
 175 void AliPerformanceTask::UserExec(Option_t *) \r
 176 {\r
 177   // Main loop\r
 178   // Called for each event\r
 179 \r
 180 \r
 181   // Decide whether to use HLT or Offline ESD\r
 182   if(fUseHLT){\r
 183 \r
 184     AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> \r
 185       (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
 186     \r
 187     if (!esdH) {\r
 188       printf("ERROR: Could not get ESDInputHandler");\r
 189       return;\r
 190     } \r
 191     fESD = esdH->GetHLTEvent();\r
 192   }// end if fUseHLT\r
 193   else  \r
 194     fESD = (AliESDEvent*) (InputEvent());\r
 195 \r
 196   if(fUseESDfriend)\r
 197     {\r
 198       fESDfriend = static_cast<AliESDfriend*>(fESD->FindListObject("AliESDfriend"));\r
 199       if(!fESDfriend) {\r
 200         Printf("ERROR: ESD friends not available");\r
 201       }\r
 202     }\r
 203   \r
 204   if(fUseMCInfo) {\r
 205       fMC = MCEvent();\r
 206   }  \r
 207 \r
 208 \r
 209   if (!fESD) {\r
 210     Printf("ERROR: ESD event not available");\r
 211     return;\r
 212   }\r
 213   \r
 214   if (fUseMCInfo && !fMC) {\r
 215     Printf("ERROR: MC event not available");\r
 216     return;\r
 217   }\r
 218 \r
 219   if(fUseESDfriend)\r
 220   {\r
 221     if(!fESDfriend) {\r
 222     Printf("ERROR: ESD friends not available");\r
 223     }\r
 224   }\r
 225 \r
 226   // Process analysis\r
 227   Bool_t process = kTRUE;\r
 228 \r
 229   // Check for centrality\r
 230   if (fUseCentrality) {\r
 231     if ( CalculateCentralityBin() != fUseCentralityBin ) {\r
 232       process = kFALSE;\r
 233     }\r
 234   }\r
 235 \r
 236   // Process comparison\r
 237   if (process) {\r
 238     AliPerformanceObject *pObj=0;\r
 239     fPitList->Reset();\r
 240     while(( pObj = (AliPerformanceObject *)fPitList->Next()) != NULL) {\r
 241           pObj->Exec(fMC,fESD,fESDfriend,fUseMCInfo,fUseESDfriend);\r
 242     }\r
 243   }\r
 244 \r
 245   // Post output data.\r
 246   PostData(1, fOutput);\r
 247 }\r
 248 \r
 249 //_____________________________________________________________________________\r
 250 void AliPerformanceTask::Terminate(Option_t *) \r
 251 {\r
 252   // Called once at the end \r
 253 \r
 254   if ( !fUseTerminate )\r
 255     return;\r
 256 \r
 257   // check output data\r
 258     fOutputSummary = dynamic_cast<TTree*> (GetOutputData(0));\r
 259     fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
 260     if (!fOutput) {\r
 261         Printf("ERROR: AliPerformanceTask::Terminate(): fOutput data not available  ..." );\r
 262         return;\r
 263    }\r
 264     if (fOutputSummary) { delete fOutputSummary; fOutputSummary=0; }      \r
 265     AliPerformanceObject* pObj=0;\r
 266     AliPerformanceTPC*  pTPC = 0;\r
 267     AliPerformanceDEdx* pDEdx = 0;\r
 268     AliPerformanceMatch* pMatch = 0;\r
 269     TIterator* itOut = fOutput->MakeIterator();\r
 270     itOut->Reset();\r
 271     while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) { \r
 272         pObj->AnalyseFinal();\r
 273         if (!  pTPC)  {    pTPC = dynamic_cast<AliPerformanceTPC*>(pObj); }\r
 274         if (! pDEdx)  {   pDEdx = dynamic_cast<AliPerformanceDEdx*>(pObj); }\r
 275         if (! pMatch) {  pMatch = dynamic_cast<AliPerformanceMatch*>(pObj); }\r
 276     }\r
 277    \r
 278     if(!fUseOCDB)  { \r
 279       printf("DO NOT USE OCDB \n");\r
 280       return;\r
 281     }\r
 282   \r
 283     if (! AliCDBManager::Instance()->GetDefaultStorage()) { AliCDBManager::Instance()->SetDefaultStorage("raw://"); }\r
 284     TUUID uuid;\r
 285     TString tmpFile = gSystem->TempDirectory() + TString("/TPCQASummary.") + uuid.AsString() + TString(".root");\r
 286     AliTPCPerformanceSummary::WriteToFile(pTPC, pDEdx, pMatch, tmpFile.Data());\r
 287     TChain* chain = new TChain("tpcQA");\r
 288     chain->Add(tmpFile.Data());\r
 289     TTree *tree = chain->CopyTree("1");\r
 290     if (chain) { delete chain; chain=0; }\r
 291     fOutputSummary = tree;\r
 292       \r
 293      // Post output data.\r
 294      PostData(0, fOutputSummary);\r
 295 \r
 296 }\r
 297 \r
 298 //_____________________________________________________________________________\r
 299 void AliPerformanceTask::FinishTaskOutput()\r
 300 {\r
 301     // called once at the end of each job (on the workernode)\r
 302     //\r
 303     // projects THnSparse to TH1,2,3\r
 304     \r
 305     fOutput = dynamic_cast<TList*> (GetOutputData(1));\r
 306     if (!fOutput) {\r
 307         Printf("ERROR: AliPerformanceTask::FinishTaskOutput(): fOutput data not available  ..." );\r
 308         return;\r
 309    }\r
 310 \r
 311       AliPerformanceObject* pObj=0;\r
 312       TIterator* itOut = fOutput->MakeIterator();  \r
 313       itOut->Reset();\r
 314       while(( pObj = dynamic_cast<AliPerformanceObject*>(itOut->Next())) != NULL) {\r
 315           pObj->SetRunNumber(fCurrentRunNumber);\r
 316           pObj->Analyse();\r
 317       }\r
 318       \r
 319      // Post output data.\r
 320      PostData(1, fOutput);\r
 321 }\r
 322 \r
 323 //_____________________________________________________________________________\r
 324 Bool_t AliPerformanceTask::Notify()\r
 325 {\r
 326   static Int_t count = 0;\r
 327   count++;\r
 328   Printf("Processing %d. file", count);\r
 329 \r
 330   return kTRUE;\r
 331 }\r
 332 \r
 333 //________________________________________________________________________\r
 334 Int_t AliPerformanceTask::CalculateCentralityBin(){\r
 335   // Get Centrality bin\r
 336 \r
 337   Int_t centrality = -1;\r
 338   Float_t centralityF = -1;\r
 339 \r
 340   if (fUseCentrality == 0)\r
 341     return centrality;\r
 342 \r
 343   AliCentrality *esdCentrality = fESD->GetCentrality();\r
 344     \r
 345   // New : 2010-11-18 JMT \r
 346   if ( fUseCentrality == 1 )\r
 347     centralityF = esdCentrality->GetCentralityPercentile("V0M");\r
 348   else if ( fUseCentrality == 2 )\r
 349     centralityF = esdCentrality->GetCentralityPercentile("CL1");\r
 350   else if ( fUseCentrality == 3 )\r
 351     centralityF = esdCentrality->GetCentralityPercentile("TRK"); \r
 352   if (centralityF == 0.)\r
 353     centralityF = 100.;\r
 354 \r
 355   if      ( centralityF >=  0. && centralityF <   5.) centrality =  0;\r
 356   else if ( centralityF >=  5. && centralityF <  10.) centrality =  5;\r
 357   else if ( centralityF >= 10. && centralityF <  20.) centrality = 10;\r
 358   else if ( centralityF >= 20. && centralityF <  30.) centrality = 20;\r
 359   else if ( centralityF >= 30. && centralityF <  40.) centrality = 30;\r
 360   else if ( centralityF >= 40. && centralityF <  50.) centrality = 40;\r
 361   else if ( centralityF >= 50. && centralityF <  60.) centrality = 50;\r
 362   else if ( centralityF >= 60. && centralityF <  70.) centrality = 60;\r
 363   else if ( centralityF >= 70. && centralityF <  80.) centrality = 70;\r
 364   else if ( centralityF >= 80. && centralityF <  90.) centrality = 80;\r
 365   else if ( centralityF >= 90. && centralityF <=100.) centrality = 90;\r
 366   \r
 367   return centrality;\r
 368 }\r