PWGPP/TPC/AliPerformanceTask.cxx

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