STEER/AliQADataMakerRec.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 notifce   *
  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 /* $Id$ */
  18
  19 //
  20 //  Base Class
  21 //  Produces the data needed to calculate the quality assurance for Reconstruction
  22 //  All data must be mergeable objects.
  23 //  Y. Schutz CERN July 2007
  24 //
  25
  26 // --- ROOT system ---
  27 #include <TFile.h>
  28 #include <TTree.h>
  29 #include <TNtupleD.h>
  30 #include <TObjArray.h>
  31
  32 // --- Standard library ---
  33
  34 // --- AliRoot header files ---
  35 #include "AliCDBPath.h"
  36 #include "AliCDBEntry.h"
  37 #include "AliDetectorRecoParam.h"
  38 #include "AliCDBManager.h"
  39
  40 #include "AliLog.h"
  41 #include "AliQADataMakerRec.h"
  42 #include "AliQAManager.h"
  43 #include "AliESDEvent.h"
  44 #include "AliRawReader.h"
  45
  46 ClassImp(AliQADataMakerRec)
  47
  48 //____________________________________________________________________________
  49 AliQADataMakerRec::AliQADataMakerRec(const char * name, const char * title) :
  50   AliQADataMaker(name, title),
  51   fDigitsQAList(NULL),
  52   fESDsQAList(NULL),
  53   fRawsQAList(NULL),
  54   fRecPointsQAList(NULL),
  55   fCorrNt(NULL),
  56   fRecoParam(NULL),
  57   fRecPointsArray(NULL)
  58 {
  59   // ctor
  60         fDetectorDirName = GetName() ;
  61 }
  62
  63 //____________________________________________________________________________
  64 AliQADataMakerRec::AliQADataMakerRec(const AliQADataMakerRec& qadm) :
  65   AliQADataMaker(qadm.GetName(), qadm.GetTitle()),
  66   fDigitsQAList(qadm.fDigitsQAList),
  67   fESDsQAList(qadm.fESDsQAList),
  68   fRawsQAList(qadm.fRawsQAList),
  69   fRecPointsQAList(qadm.fRecPointsQAList),
  70   fCorrNt(qadm.fCorrNt),
  71   fRecoParam(qadm.fRecoParam),
  72   fRecPointsArray(NULL)
  73 {
  74   //copy ctor
  75         SetName(qadm.GetName()) ;
  76         SetTitle(qadm.GetTitle()) ;
  77         fDetectorDirName = GetName() ;
  78 }
  79
  80 //____________________________________________________________________________
  81 AliQADataMakerRec::~AliQADataMakerRec()
  82 {
  83         //dtor: delete the TObjArray and thei content
  84         if ( fESDsQAList ) {
  85     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
  86       if ( fESDsQAList[specie] ) {
  87           fESDsQAList[specie]->Delete() ;
  88       }
  89     }
  90     delete[] fESDsQAList ;
  91         }
  92         if ( fRawsQAList ) {
  93     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
  94       if ( fRawsQAList[specie] ) {
  95         fRawsQAList[specie]->Delete() ;
  96       }
  97     }
  98     delete[] fRawsQAList ;
  99   }
 100         if ( fDigitsQAList ) {
 101     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
 102       if ( fDigitsQAList[specie] ) {
 103         fDigitsQAList[specie]->Delete() ;
 104       }
 105     }
 106                 delete[] fDigitsQAList ;
 107   }
 108         if ( fRecPointsQAList ) {
 109     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
 110       if ( fRecPointsQAList[specie] ) {
 111         fRecPointsQAList[specie]->Delete() ;
 112       }
 113     }
 114                 delete[] fRecPointsQAList ;
 115   }
 116   if (fRecPointsArray) {
 117     fRecPointsArray->Clear() ;
 118     delete fRecPointsArray ;
 119   }
 120 }
 121
 122 //__________________________________________________________________
 123 AliQADataMakerRec& AliQADataMakerRec::operator = (const AliQADataMakerRec& qadm )
 124 {
 125   // Assignment operator.
 126   this->~AliQADataMakerRec();
 127   new(this) AliQADataMakerRec(qadm);
 128   return *this;
 129 }
 130
 131 //____________________________________________________________________________
 132 void AliQADataMakerRec::EndOfCycle()
 133 {
 134   // Finishes a cycle of QA for all the tasks
 135   EndOfCycle(AliQAv1::kRAWS) ;
 136   EndOfCycle(AliQAv1::kDIGITSR) ;
 137   EndOfCycle(AliQAv1::kRECPOINTS) ;
 138   EndOfCycle(AliQAv1::kESDS) ;
 139   ResetCycle() ;
 140 }
 141
 142 //____________________________________________________________________________
 143 void AliQADataMakerRec::EndOfCycle(AliQAv1::TASKINDEX_t task)
 144 {
 145         // Finishes a cycle of QA
 146
 147
 148   TObjArray ** list = NULL ;
 149
 150         if ( task == AliQAv1::kRAWS )
 151                 list = fRawsQAList ;
 152         else if ( task == AliQAv1::kDIGITSR )
 153                 list = fDigitsQAList ;
 154         else if ( task == AliQAv1::kRECPOINTS )
 155                 list = fRecPointsQAList ;
 156         else if ( task == AliQAv1::kESDS )
 157                 list = fESDsQAList ;
 158
 159
 160         if ( ! list && ! fCorrNt )
 161     return ;
 162   //DefaultEndOfDetectorCycle(task) ;
 163         EndOfDetectorCycle(task, list) ;
 164
 165   if (! AliQAManager::QAManager(AliQAv1::kRECMODE)->IsSaveData())
 166     return ;
 167
 168   fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ;
 169   if (!fDetectorDir)
 170     fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ;
 171   TDirectory * subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ;
 172   if (!subDir)
 173     subDir = fDetectorDir->mkdir(AliQAv1::GetTaskName(task)) ;
 174   subDir->cd() ;
 175   for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) { // skip Default
 176     if (! AliQAv1::Instance(AliQAv1::GetDetIndex(GetName()))->IsEventSpecieSet(AliRecoParam::ConvertIndex(specie)) || AliRecoParam::ConvertIndex(specie) == AliRecoParam::kDefault)
 177       continue ;
 178     TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ;
 179     if (!eventSpecieDir)
 180       eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ;
 181     eventSpecieDir->cd() ;
 182     if (list) {
 183       if (list[specie]) {
 184         TIter next(list[specie]) ;
 185         TObject * obj ;
 186         while( (obj = next()) ) {
 187           if (!obj->TestBit(AliQAv1::GetExpertBit()))
 188             obj->Write() ;
 189         }
 190         if (WriteExpert()) {
 191           TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ;
 192           if (!expertDir)
 193             expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ;
 194           expertDir->cd() ;
 195           next.Reset() ;
 196           while( (obj = next()) ) {
 197             if (!obj->TestBit(AliQAv1::GetExpertBit()))
 198               continue ;
 199             obj->Write() ;
 200           }
 201         }
 202       }
 203     }
 204     else if ( fCorrNt ) {
 205       if (fCorrNt[specie] && AliQAv1::GetDetIndex(GetName()) == AliQAv1::kCORR) {
 206         if (fCorrNt[specie]->GetNvar() != 0) {
 207           eventSpecieDir->cd() ;
 208           fCorrNt[specie]->Write() ;
 209         }
 210       }
 211       fOutput->Save() ;
 212     }
 213   }
 214 }
 215
 216 //____________________________________________________________________________
 217 void AliQADataMakerRec::Exec(AliQAv1::TASKINDEX_t task, TObject * data)
 218 {
 219   // creates the quality assurance data for the various tasks (Hits, SDigits, Digits, ESDs)
 220
 221         if ( task == AliQAv1::kRAWS ) {
 222                 AliDebug(AliQAv1::GetQADebugLevel(), "Processing Raws QA") ;
 223                 AliRawReader * rawReader = static_cast<AliRawReader *>(data) ;
 224                 if (rawReader)
 225                         MakeRaws(rawReader) ;
 226                 else
 227       AliDebug(AliQAv1::GetQADebugLevel(), "Raw data are not processed") ;
 228         } else if ( task == AliQAv1::kDIGITSR ) {
 229                 AliDebug(AliQAv1::GetQADebugLevel(), "Processing Digits QA") ;
 230                 TTree * tree = static_cast<TTree *>(data) ;
 231                 if (strcmp(tree->ClassName(), "TTree") == 0) {
 232                         MakeDigits(tree) ;
 233                 } else {
 234                         AliWarning("data are not a TTree") ;
 235                 }
 236         } else if ( task == AliQAv1::kRECPOINTS ) {
 237                 AliDebug(AliQAv1::GetQADebugLevel(), "Processing RecPoints QA") ;
 238                 TTree * tree = static_cast<TTree *>(data) ;
 239                 if (strcmp(tree->ClassName(), "TTree") == 0) {
 240                         MakeRecPoints(tree) ;
 241                 } else {
 242                         AliWarning("data are not a TTree") ;
 243                 }
 244         } else if ( task == AliQAv1::kESDS ) {
 245                 AliDebug(AliQAv1::GetQADebugLevel(), "Processing ESDs QA") ;
 246                 AliESDEvent * esd = static_cast<AliESDEvent *>(data) ;
 247                 if (strcmp(esd->ClassName(), "AliESDEvent") == 0)
 248                         MakeESDs(esd) ;
 249                 else
 250                         AliError("Wrong type of esd container") ;
 251         }
 252 }
 253
 254 //____________________________________________________________________________
 255 TObjArray **  AliQADataMakerRec::Init(AliQAv1::TASKINDEX_t task, Int_t cycles)
 256 {
 257   // general intialisation
 258   InitRecoParams() ;
 259         TObjArray ** rv = NULL ;
 260
 261         if (cycles > 0)
 262                 SetCycle(cycles) ;
 263
 264         if ( task == AliQAv1::kRAWS ) {
 265                 if (! fRawsQAList ) {
 266       fRawsQAList = new TObjArray *[AliRecoParam::kNSpecies] ;
 267       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
 268         fRawsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
 269         fRawsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
 270       }
 271                 }
 272                 rv = fRawsQAList ;
 273         } else if ( task == AliQAv1::kDIGITSR ) {
 274                 if ( ! fDigitsQAList ) {
 275       fDigitsQAList = new TObjArray *[AliRecoParam::kNSpecies] ;
 276       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
 277         fDigitsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
 278         fDigitsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
 279       }
 280                 }
 281                 rv = fDigitsQAList ;
 282         } else if ( task == AliQAv1::kRECPOINTS ) {
 283                 if ( ! fRecPointsQAList ) {
 284       fRecPointsQAList = new TObjArray *[AliRecoParam::kNSpecies] ;
 285       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
 286         fRecPointsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
 287         fRecPointsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
 288       }
 289     }
 290                 rv = fRecPointsQAList ;
 291         } else if ( task == AliQAv1::kESDS ) {
 292                 if ( ! fESDsQAList ) {
 293       fESDsQAList = new TObjArray *[AliRecoParam::kNSpecies] ;
 294       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
 295         fESDsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
 296         fESDsQAList[specie]->SetName(Form("%s_%s", GetName(), AliQAv1::GetTaskName(task).Data())); //, AliRecoParam::GetEventSpecieName(specie))) ;
 297       }
 298                 }
 299                 rv = fESDsQAList ;
 300         }
 301         return rv ;
 302 }
 303
 304 //____________________________________________________________________________
 305 void AliQADataMakerRec::Init(AliQAv1::TASKINDEX_t task, TObjArray ** list, Int_t run, Int_t cycles)
 306 {
 307   // Intialisation by passing the list of QA data booked elsewhere
 308
 309   InitRecoParams() ;
 310   fRun = run ;
 311         if (cycles > 0)
 312                 SetCycle(cycles) ;
 313
 314         if ( task == AliQAv1::kRAWS ) {
 315                 fRawsQAList = list ;
 316         } else if ( task == AliQAv1::kDIGITSR ) {
 317                 fDigitsQAList = list ;
 318         } else if ( task == AliQAv1::kRECPOINTS ) {
 319                 fRecPointsQAList = list ;
 320         } else if ( task == AliQAv1::kESDS ) {
 321                 fESDsQAList = list ;
 322         }
 323 }
 324
 325 //____________________________________________________________________________
 326 void AliQADataMakerRec::InitRecoParams()
 327 {
 328   // Get the recoparam form the OCDB
 329   if (!fRecoParam) {
 330     AliDebug(AliQAv1::GetQADebugLevel(), Form("Loading reconstruction parameter objects for detector %s", GetName()));
 331     AliCDBPath path(GetName(),"Calib","RecoParam");
 332     AliCDBEntry *entry=AliCDBManager::Instance()->Get(path.GetPath());
 333     if(!entry) {
 334       fRecoParam = NULL ;
 335       AliDebug(AliQAv1::GetQADebugLevel(), Form("Couldn't find RecoParam entry in OCDB for detector %s",GetName()));
 336     }
 337     else {
 338       TObject * recoParamObj = entry->GetObject() ;
 339       if ( strcmp(recoParamObj->ClassName(), "TObjArray") == 0 ) {
 340         // The detector has only one set of reco parameters
 341         AliDebug(AliQAv1::GetQADebugLevel(), Form("Array of reconstruction parameters found for detector %s",GetName()));
 342         TObjArray *recoParamArray = static_cast<TObjArray*>(recoParamObj) ;
 343         for (Int_t iRP=0; iRP<recoParamArray->GetEntriesFast(); iRP++) {
 344           fRecoParam = static_cast<AliDetectorRecoParam*>(recoParamArray->At(iRP)) ;
 345           if (!fRecoParam)
 346             break ;
 347           else if (fRecoParam->IsDefault())
 348             break ;
 349         }
 350       }
 351       else if (recoParamObj->InheritsFrom("AliDetectorRecoParam")) {
 352         // The detector has only one set of reco parameters
 353         // Registering it in AliRecoParam
 354         AliDebug(AliQAv1::GetQADebugLevel(), Form("Single set of reconstruction parameters found for detector %s",GetName()));
 355         fRecoParam = static_cast<AliDetectorRecoParam*>(recoParamObj) ;
 356         static_cast<AliDetectorRecoParam*>(recoParamObj)->SetAsDefault();
 357       } else {
 358         AliError(Form("No valid RecoParam object found in the OCDB for detector %s",GetName()));
 359       }
 360     }
 361   }
 362 }
 363
 364 //____________________________________________________________________________
 365 void AliQADataMakerRec::ResetDetector(AliQAv1::TASKINDEX_t task)
 366 {
 367     // default reset that resets all the QA objects.
 368     // to be overloaded by detectors, if necessary
 369
 370   TObjArray ** list = NULL ;
 371   if ( task == AliQAv1::kRAWS ) {
 372                 list = fRawsQAList ;
 373         } else if ( task == AliQAv1::kDIGITSR ) {
 374                 list = fDigitsQAList ;
 375         } else if ( task == AliQAv1::kRECPOINTS ) {
 376                 list = fRecPointsQAList ;
 377         } else if ( task == AliQAv1::kESDS ) {
 378                 list = fESDsQAList ;
 379         }
 380     //list was not initialized, skip
 381   if (!list)
 382     return ;
 383
 384   for (int spec = 0; spec < AliRecoParam::kNSpecies; spec++) {
 385     if (!AliQAv1::Instance()->IsEventSpecieSet(AliRecoParam::ConvertIndex(spec)))
 386       continue;
 387     TIter next(list[spec]) ;
 388     TH1 * histo = NULL ;
 389     while ( (histo = dynamic_cast<TH1*> (next())) ) {
 390       histo->Reset("ICE") ;
 391       histo->ResetStats() ;
 392     }
 393   }
 394 }
 395
 396 //____________________________________________________________________________
 397 void AliQADataMakerRec::StartOfCycle(Int_t run)
 398 {
 399   // Finishes a cycle of QA for all the tasks
 400   Bool_t samecycle = kFALSE ;
 401   StartOfCycle(AliQAv1::kRAWS,      run, samecycle) ;
 402   samecycle = kTRUE ;
 403   StartOfCycle(AliQAv1::kDIGITSR,   run, samecycle) ;
 404   StartOfCycle(AliQAv1::kRECPOINTS, run, samecycle) ;
 405   StartOfCycle(AliQAv1::kESDS,      run, samecycle) ;
 406 }
 407
 408 //____________________________________________________________________________
 409 void AliQADataMakerRec::StartOfCycle(AliQAv1::TASKINDEX_t task, Int_t run, const Bool_t sameCycle)
 410 {
 411   // Finishes a cycle of QA data acquistion
 412   if ( run > 0 )
 413     fRun = run ;
 414         if ( !sameCycle || fCurrentCycle == -1) {
 415                 ResetCycle() ;
 416                 if (fOutput)
 417                         fOutput->Close() ;
 418                 if (AliQAManager::QAManager(AliQAv1::kRECMODE)->IsSaveData())
 419       fOutput = AliQAv1::GetQADataFile(GetName(), fRun) ;
 420         }
 421         AliDebug(AliQAv1::GetQADebugLevel(), Form(" Run %d Cycle %d task %s file %s",
 422                                  fRun, fCurrentCycle, AliQAv1::GetTaskName(task).Data(), fOutput->GetName() )) ;
 423
 424 //      fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ;
 425 //      if (!fDetectorDir)
 426 //              fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ;
 427 //
 428 //      TDirectory * subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ;
 429 //      if (!subDir)
 430 //              subDir = fDetectorDir->mkdir(AliQAv1::GetTaskName(task)) ;
 431 //
 432 //  for ( Int_t specie = AliRecoParam::kDefault ; specie < AliRecoParam::kNSpecies ; specie++ ) {
 433 //    TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ;
 434 //    if (!eventSpecieDir)
 435 //      eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ;
 436 //    TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ;
 437 //    if (!expertDir)
 438 //      expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ;
 439 //  }
 440         StartOfDetectorCycle() ;
 441 }