STEER/AliQAChecker.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 /* $Id: */
  17
  18 ///////////////////////////////////////////////////////////////////////////////
  19 //                                                                           //
  20 // class for running the Quality Assurance Checker
  21 //                                                                           //
  22 ///////////////////////////////////////////////////////////////////////////////
  23
  24 #include "AliCDBEntry.h"
  25 #include "AliQAManager.h"
  26 #include "AliCDBStorage.h"
  27 #include "AliRunInfo.h"
  28 #include "AliLog.h"
  29 #include "AliModule.h"
  30 #include "AliQAv1.h"
  31 #include "AliQAChecker.h"
  32 #include "AliQACheckerBase.h"
  33 #include "AliCorrQAChecker.h"
  34 #include "AliGlobalQAChecker.h"
  35 #include "AliGRPObject.h"
  36
  37 #include <TKey.h>
  38 #include <TObjArray.h>
  39 #include <TObjString.h>
  40 #include <TPluginManager.h>
  41 #include <TROOT.h>
  42 #include <TStopwatch.h>
  43 #include <TString.h>
  44 #include <TSystem.h>
  45 #include <TList.h>
  46 #include <TNtupleD.h>
  47
  48 ClassImp(AliQAChecker)
  49   AliQAChecker * AliQAChecker::fgQAChecker = 0x0 ;
  50
  51 //_____________________________________________________________________________
  52 AliQAChecker::AliQAChecker(const char* name, const char* title) :
  53   TNamed(name, title),
  54   fDataFile(0x0),
  55   fRunInfo(0x0),
  56   fRunInfoOwner(kFALSE),
  57   fRefFile(0x0),
  58   fFoundDetectors("."),
  59   fEventSpecie(AliRecoParam::kDefault),
  60   fRun(0)
  61 {
  62   // ctor: initialise checkers and open the data file
  63   for (Int_t det = 0 ; det < AliQAv1::kNDET ; det++)
  64     fCheckers[det] = NULL ;
  65 }
  66
  67 //_____________________________________________________________________________
  68 AliQAChecker::AliQAChecker(const AliQAChecker& qac) :
  69   TNamed(qac),
  70   fDataFile(qac.fDataFile),
  71   fRunInfo(qac.fRunInfo),
  72   fRunInfoOwner(kFALSE),
  73   fRefFile(qac.fRefFile),
  74   fFoundDetectors(qac.fFoundDetectors),
  75   fEventSpecie(qac.fEventSpecie),
  76   fRun(qac.fRun)
  77 {
  78   // copy constructor
  79
  80   for (Int_t det = 0 ; det < AliQAv1::kNDET ; det++)
  81     fCheckers[det] = NULL ;
  82 }
  83
  84 //_____________________________________________________________________________
  85 AliQAChecker& AliQAChecker::operator = (const AliQAChecker& qac)
  86 {
  87 // assignment operator
  88
  89   this->~AliQAChecker();
  90   new(this) AliQAChecker(qac);
  91   return *this;
  92 }
  93
  94 //_____________________________________________________________________________
  95 AliQAChecker::~AliQAChecker()
  96 {
  97 // clean up
  98   if (fRunInfo)
  99     delete fRunInfo ;
 100   delete [] fCheckers ;
 101   AliQAv1::Close() ;
 102 }
 103
 104 //_____________________________________________________________________________
 105   AliQACheckerBase * AliQAChecker::GetDetQAChecker(Int_t det)
 106 {
 107         // Gets the Quality Assurance checker for the detector specified by its name
 108
 109         if (fCheckers[det])
 110     return fCheckers[det];
 111
 112         AliQACheckerBase * qac = NULL ;
 113
 114         TString detName(AliQAv1::GetDetName(det)) ;
 115
 116         if (det == AliQAv1::kGLOBAL) {
 117                 qac = new AliGlobalQAChecker() ;
 118         } else if (det == AliQAv1::kCORR) {
 119                 qac = new AliCorrQAChecker() ;
 120         } else {
 121                 AliDebugClass(AliQAv1::GetQADebugLevel(), Form("Retrieving QA checker for %s", detName.Data())) ;
 122                 TPluginManager* pluginManager = gROOT->GetPluginManager() ;
 123                 TString qacName = "Ali" + detName + "QAChecker" ;
 124
 125                 // first check if a plugin is defined for the quality assurance checker
 126                 TPluginHandler* pluginHandler = pluginManager->FindHandler("AliQAChecker", detName.Data());
 127                 // if not, add a plugin for it
 128                 if (!pluginHandler) {
 129                         //AliDebug(AliQAv1::GetQADebugLevel(), Form("defining plugin for %s", qacName.Data()));
 130                         TString libs = gSystem->GetLibraries();
 131
 132                         if (libs.Contains("lib" + detName + "base.so") || (gSystem->Load("lib" + detName + "base.so") >= 0))
 133                                 pluginManager->AddHandler("AliQAChecker", detName, qacName, detName + "qac", qacName + "()");
 134                         else
 135                                 pluginManager->AddHandler("AliQAChecker", detName, qacName, detName, qacName + "()");
 136
 137                         pluginHandler = pluginManager->FindHandler("AliQAChecker", detName);
 138
 139                         if (pluginHandler && (pluginHandler->LoadPlugin() == 0))
 140                                 qac = (AliQACheckerBase *) pluginHandler->ExecPlugin(0);
 141
 142                 }
 143         }
 144         if (qac)
 145                 fCheckers[det] = qac ;
 146
 147         return qac ;
 148 }
 149
 150 //_____________________________________________________________________________
 151 void AliQAChecker::GetRefSubDir(const char * det, const char * task, TDirectory *& dirFile, TObjArray **& dirOCDB)
 152 {
 153   // Opens and returns the file with the reference data
 154   dirFile = NULL ;
 155   TString refStorage(AliQAv1::GetQARefStorage()) ;
 156 //  if (refStorage.Contains(AliQAv1::GetLabLocalFile())) {
 157 //    refStorage.ReplaceAll(AliQAv1::GetLabLocalFile(), "") ;
 158 //    refStorage += AliQAv1::GetQARefFileName() ;
 159 //    if ( fRefFile )
 160 //      if ( fRefFile->IsOpen() )
 161 //                                      fRefFile->Close() ;
 162 //    fRefFile = TFile::Open(refStorage.Data()) ;
 163 //    if (!fRefFile) {
 164 //      AliError(Form("Cannot find reference file %s", refStorage.Data())) ;
 165 //      dirFile = NULL ;
 166 //    }
 167 //    dirFile = fRefFile->GetDirectory(det) ;
 168 //    if (!dirFile) {
 169 //      AliWarning(Form("Directory %s not found in %d", det, refStorage.Data())) ;
 170 //    } else {
 171 //                      dirFile = dirFile->GetDirectory(task) ;
 172 //      if (!dirFile)
 173 //                              AliWarning(Form("Directory %s/%s not found in %s", det, task, refStorage.Data())) ;
 174 //    }
 175 //  } else
 176   if (!refStorage.Contains(AliQAv1::GetLabLocalOCDB()) && !refStorage.Contains(AliQAv1::GetLabAliEnOCDB())) {
 177     AliError(Form("%s is not a valid location for reference data", refStorage.Data())) ;
 178     return ;
 179   } else {
 180     AliQAManager* manQA = AliQAManager::QAManager(AliQAv1::GetTaskIndex(task)) ;
 181     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
 182       if ( !AliQAv1::Instance()->IsEventSpecieSet(specie) )
 183         continue ;
 184       //if ( strcmp(AliQAv1::GetRefDataDirName(), "") == 0 ) { // the name of the last level of the directory is not set (EventSpecie)
 185         // Get it from RunInfo
 186         //if (!fRunInfo)  // not yet set, get the info from GRP
 187         //  LoadRunInfoFromGRP() ;
 188       AliQAv1::SetQARefDataDirName(specie) ;
 189       //}
 190       if ( ! manQA->GetLock() ) {
 191         manQA->SetDefaultStorage(AliQAv1::GetQARefStorage()) ;
 192         manQA->SetSpecificStorage("*", AliQAv1::GetQARefStorage()) ;
 193         manQA->SetRun(AliCDBManager::Instance()->GetRun()) ;
 194         manQA->SetLock() ;
 195       }
 196       char * detOCDBDir = Form("%s/%s/%s", det, AliQAv1::GetRefOCDBDirName(), AliQAv1::GetRefDataDirName()) ;
 197       AliCDBEntry * entry = manQA->Get(detOCDBDir, manQA->GetRun()) ;
 198       if (entry) {
 199         dirOCDB = new TObjArray*[AliRecoParam::kNSpecies] ;
 200         TList * listDetQAD =static_cast<TList *>(entry->GetObject()) ;
 201         if ( strcmp(listDetQAD->ClassName(), "TList") != 0 ) {
 202           AliError(Form("Expected a Tlist and found a %s for detector %s", listDetQAD->ClassName(), det)) ;
 203           continue ;
 204         }
 205         TIter next(listDetQAD) ;
 206         TObjArray * ar ;
 207         while ( (ar = (TObjArray*)next()) )
 208           if ( listDetQAD )
 209           dirOCDB[specie] = static_cast<TObjArray *>(listDetQAD->FindObject(Form("%s/%s", task, AliRecoParam::GetEventSpecieName(specie)))) ;
 210       }
 211     }
 212   }
 213 }
 214
 215 //_____________________________________________________________________________
 216 AliQAChecker * AliQAChecker::Instance()
 217 {
 218         // returns unique instance of the checker
 219   if ( ! fgQAChecker )
 220    fgQAChecker = new AliQAChecker() ;
 221   return fgQAChecker ;
 222 }
 223
 224 //_____________________________________________________________________________
 225 void AliQAChecker::LoadRunInfoFromGRP()
 226 {
 227   AliCDBManager* man = AliCDBManager::Instance() ;
 228   AliCDBEntry* entry = man->Get(AliQAv1::GetGRPPath().Data());
 229   AliGRPObject* grpObject = 0x0;
 230   if (entry) {
 231
 232           TMap* m = static_cast<TMap*>(entry->GetObject());  // old GRP entry
 233
 234           if (m) {
 235             AliDebug(AliQAv1::GetQADebugLevel(), "It is a map");
 236             //m->Print();
 237             grpObject = new AliGRPObject();
 238                  grpObject->ReadValuesFromMap(m);
 239     }
 240
 241     else {
 242             AliDebug(AliQAv1::GetQADebugLevel(), "It is a new GRP object");
 243         grpObject = static_cast<AliGRPObject*>(entry->GetObject());  // new GRP entry
 244     }
 245
 246     entry->SetOwner(0);
 247     AliCDBManager::Instance()->UnloadFromCache("GRP/GRP/Data");
 248   }
 249
 250   if (!grpObject) {
 251      AliFatal("No GRP entry found in OCDB!");
 252   }
 253
 254   TString lhcState = grpObject->GetLHCState();
 255   if (lhcState==AliGRPObject::GetInvalidString()) {
 256     AliError("GRP/GRP/Data entry:  missing value for the LHC state ! Using UNKNOWN");
 257     lhcState = "UNKNOWN";
 258   }
 259
 260   TString beamType = grpObject->GetBeamType();
 261   if (beamType==AliGRPObject::GetInvalidString()) {
 262     AliError("GRP/GRP/Data entry:  missing value for the beam type ! Using UNKNOWN");
 263     beamType = "UNKNOWN";
 264   }
 265
 266   Float_t beamEnergy = grpObject->GetBeamEnergy();
 267   if (beamEnergy==AliGRPObject::GetInvalidFloat()) {
 268     AliError("GRP/GRP/Data entry:  missing value for the beam energy ! Using 0");
 269     beamEnergy = 0;
 270   }
 271
 272   TString runType = grpObject->GetRunType();
 273   if (runType==AliGRPObject::GetInvalidString()) {
 274     AliError("GRP/GRP/Data entry:  missing value for the run type ! Using UNKNOWN");
 275     runType = "UNKNOWN";
 276   }
 277
 278   Int_t activeDetectors = grpObject->GetDetectorMask();
 279   if (activeDetectors==AliGRPObject::GetInvalidInt()) {
 280     AliError("GRP/GRP/Data entry:  missing value for the detector mask ! Using 1074790399");
 281     activeDetectors = 1074790399;
 282   }
 283
 284   fRunInfo = new AliRunInfo(lhcState, beamType, beamEnergy, runType, activeDetectors);
 285
 286   fRunInfoOwner = kTRUE ;
 287
 288   // set the event specie
 289   fEventSpecie = AliRecoParam::kDefault ;
 290   if (strcmp(runType,"PHYSICS")) {
 291     // Not a physics run, the event specie is set to kCalib
 292     fEventSpecie = AliRecoParam::kCalib ;
 293     return;
 294   }
 295   if (strcmp(lhcState,"STABLE_BEAMS") == 0) {
 296     // Heavy ion run (any beam tha is not pp, the event specie is set to kHighMult
 297     fEventSpecie = AliRecoParam::kHighMult ;
 298     if ((strcmp(beamType,"p-p") == 0) ||
 299         (strcmp(beamType,"p-")  == 0) ||
 300         (strcmp(beamType,"-p")  == 0) ||
 301         (strcmp(beamType,"P-P") == 0) ||
 302         (strcmp(beamType,"P-")  == 0) ||
 303         (strcmp(beamType,"-P")  == 0)) {
 304       // Proton run, the event specie is set to kLowMult
 305       fEventSpecie = AliRecoParam::kLowMult ;
 306     }
 307     else if (strcmp(beamType,"-") == 0) {
 308       // No beams, we assume cosmic data
 309       fEventSpecie = AliRecoParam::kCosmic ;
 310     }
 311     else if (strcmp(beamType,"UNKNOWN") == 0) {
 312       // No LHC beam information is available, we use the default event specie
 313       fEventSpecie = AliRecoParam::kDefault ;
 314     }
 315   }
 316 }
 317
 318 //_____________________________________________________________________________
 319 Bool_t AliQAChecker::Run(const char * fileName)
 320 {
 321   // run the Quality Assurance Checker for all tasks Hits, SDigits, Digits, DigitsR, RecPoints, TrackSegments, RecParticles and ESDs
 322   // starting from data in file
 323   TStopwatch stopwatch;
 324   stopwatch.Start();
 325
 326   //search for all detectors QA directories
 327   TList * detKeyList = AliQAv1::GetQADataFile(fileName)->GetListOfKeys() ;
 328   TIter nextd(detKeyList) ;
 329   TKey * detKey ;
 330   while ( (detKey = static_cast<TKey *>(nextd()) ) ) {
 331     AliDebug(AliQAv1::GetQADebugLevel(), Form("Found %s", detKey->GetName())) ;
 332     //Check which detector
 333     TString detName ;
 334     TString detNameQA(detKey->GetName()) ;
 335     Int_t det ;
 336     for ( det = 0; det < AliQAv1::kNDET ; det++) {
 337       detName = AliQAv1::GetDetName(det) ;
 338       if (detNameQA.Contains(detName)) {
 339         fFoundDetectors+=detName ;
 340         fFoundDetectors+="." ;
 341         break ;
 342       }
 343     }
 344     TDirectory * detDir = AliQAv1::GetQADataFile(fileName)->GetDirectory(detKey->GetName()) ;
 345     TList * taskKeyList = detDir->GetListOfKeys() ;
 346     TIter nextt(taskKeyList) ;
 347     TKey * taskKey ;
 348     // now search for the tasks dir
 349     while ( (taskKey = static_cast<TKey *>(nextt()) ) ) {
 350       TString taskName( taskKey->GetName() ) ;
 351       AliDebug(AliQAv1::GetQADebugLevel(), Form("Found %s", taskName.Data())) ;
 352       TDirectory * taskDir = detDir->GetDirectory(taskName.Data()) ;
 353       taskDir->cd() ;
 354       AliQACheckerBase * qac = GetDetQAChecker(det) ;
 355       if (qac)
 356         AliDebug(AliQAv1::GetQADebugLevel(), Form("QA checker found for %s", detName.Data())) ;
 357       if (!qac)
 358         AliFatal(Form("QA checker not found for %s", detName.Data())) ;
 359       AliQAv1::ALITASK_t index = AliQAv1::kNULLTASK ;
 360       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kHITS) )
 361         index = AliQAv1::kSIM ;
 362       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kSDIGITS) )
 363         index = AliQAv1::kSIM ;
 364       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kDIGITS) )
 365         index = AliQAv1::kSIM ;
 366       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kDIGITSR) )
 367         index = AliQAv1::kREC ;
 368       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kRECPOINTS) )
 369         index = AliQAv1::kREC ;
 370       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kTRACKSEGMENTS) )
 371         index = AliQAv1::kREC ;
 372       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kRECPARTICLES) )
 373         index = AliQAv1::kREC ;
 374       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kESDS) )
 375         index = AliQAv1::kESD ;
 376       qac->Init(AliQAv1::DETECTORINDEX_t(det)) ;
 377
 378       TDirectory * refDir     = NULL ;
 379       TObjArray ** refOCDBDir = NULL ;
 380       GetRefSubDir(detNameQA.Data(), taskName.Data(), refDir, refOCDBDir) ;
 381                   qac->SetRefandData(refDir, refOCDBDir, taskDir) ;
 382                   qac->Run(index) ;
 383     }
 384   }
 385   AliInfo("QA performed for following detectors:") ;
 386   for ( Int_t det = 0; det < AliQAv1::kNDET; det++) {
 387     if (fFoundDetectors.Contains(AliQAv1::GetDetName(det))) {
 388       AliInfoClass(Form("%s, ",AliQAv1::GetDetName(det))) ;
 389       fFoundDetectors.ReplaceAll(AliQAv1::GetDetName(det), "") ;
 390     }
 391   }
 392   printf("\n") ;
 393   return kTRUE ;
 394 }
 395
 396 //_____________________________________________________________________________
 397 Bool_t AliQAChecker::Run(AliQAv1::DETECTORINDEX_t det, AliQAv1::TASKINDEX_t task, TObjArray ** list)
 398 {
 399         // run the Quality Assurance Checker for detector det, for task task starting from data in list
 400
 401         AliQACheckerBase * qac = GetDetQAChecker(det) ;
 402         if (qac)
 403                 AliDebug(AliQAv1::GetQADebugLevel(), Form("QA checker found for %s", AliQAv1::GetDetName(det).Data())) ;
 404         if (!qac)
 405                 AliError(Form("QA checker not found for %s", AliQAv1::GetDetName(det).Data())) ;
 406
 407         AliQAv1::ALITASK_t index = AliQAv1::kNULLTASK ;
 408         if ( task == AliQAv1::kRAWS )
 409                 index = AliQAv1::kRAW ;
 410         else if ( task == AliQAv1::kHITS )
 411                 index = AliQAv1::kSIM ;
 412         else if ( task == AliQAv1::kSDIGITS )
 413                 index = AliQAv1::kSIM ;
 414         else if ( task == AliQAv1::kDIGITS )
 415                 index = AliQAv1::kSIM ;
 416         else if ( task == AliQAv1::kDIGITSR )
 417                 index = AliQAv1::kREC ;
 418         else if ( task == AliQAv1::kRECPOINTS )
 419                 index = AliQAv1::kREC ;
 420         else if ( task == AliQAv1::kTRACKSEGMENTS )
 421                 index = AliQAv1::kREC ;
 422         else if ( task == AliQAv1::kRECPARTICLES )
 423                 index = AliQAv1::kREC ;
 424         else if ( task == AliQAv1::kESDS )
 425                 index = AliQAv1::kESD ;
 426
 427         TDirectory * refDir     = NULL ;
 428         TObjArray ** refOCDBDir = NULL  ;
 429   qac->Init(det) ;
 430   GetRefSubDir(AliQAv1::GetDetName(det), AliQAv1::GetTaskName(task), refDir, refOCDBDir) ;
 431   qac->SetRefandData(refDir, refOCDBDir) ;
 432   qac->Run(index, list) ;
 433
 434   // make the image
 435   qac->MakeImage(list, task, AliQAv1::Mode(task)) ;
 436
 437         return kTRUE ;
 438 }
 439
 440 //_____________________________________________________________________________
 441 Bool_t AliQAChecker::Run(AliQAv1::DETECTORINDEX_t det, AliQAv1::TASKINDEX_t task, TNtupleD ** list)
 442 {
 443         // run the Quality Assurance Checker for detector det, for task task starting from data in list
 444
 445         AliQACheckerBase * qac = GetDetQAChecker(det) ;
 446         if (qac)
 447                 AliDebug(AliQAv1::GetQADebugLevel(), Form("QA checker found for %s", AliQAv1::GetDetName(det).Data())) ;
 448         if (!qac)
 449                 AliError(Form("QA checker not found for %s", AliQAv1::GetDetName(det).Data())) ;
 450
 451         AliQAv1::ALITASK_t index = AliQAv1::kNULLTASK ;
 452         if ( task == AliQAv1::kRAWS )
 453                 index = AliQAv1::kRAW ;
 454         else if ( task == AliQAv1::kHITS )
 455                 index = AliQAv1::kSIM ;
 456         else if ( task == AliQAv1::kSDIGITS )
 457                 index = AliQAv1::kSIM ;
 458         else if ( task == AliQAv1::kDIGITS )
 459                 index = AliQAv1::kSIM ;
 460         else if ( task == AliQAv1::kDIGITSR )
 461                 index = AliQAv1::kREC ;
 462         else if ( task == AliQAv1::kRECPOINTS )
 463                 index = AliQAv1::kREC ;
 464         else if ( task == AliQAv1::kTRACKSEGMENTS )
 465                 index = AliQAv1::kREC ;
 466         else if ( task == AliQAv1::kRECPARTICLES )
 467                 index = AliQAv1::kREC ;
 468         else if ( task == AliQAv1::kESDS )
 469                 index = AliQAv1::kESD ;
 470
 471         TDirectory * refDir     = NULL ;
 472         TObjArray ** refOCDBDir = NULL ;
 473   qac->Init(det) ;
 474   GetRefSubDir(AliQAv1::GetDetName(det), AliQAv1::GetTaskName(task), refDir, refOCDBDir) ;
 475   qac->SetRefandData(refDir, refOCDBDir) ;
 476   qac->Run(index, list) ;
 477
 478   return kTRUE ;
 479 }