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