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::GetLabLocalOCDB()) && !refStorage.Contains(AliQAv1::GetLabAliEnOCDB())) {
 157     AliError(Form("%s is not a valid location for reference data", refStorage.Data())) ;
 158     return ;
 159   } else {
 160     AliQAManager* manQA = AliQAManager::QAManager(AliQAv1::GetTaskIndex(task)) ;
 161     dirOCDB = new TObjArray*[AliRecoParam::kNSpecies] ;
 162     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
 163       dirOCDB[specie] = NULL ;
 164       if ( !AliQAv1::Instance()->IsEventSpecieSet(specie) )
 165         continue ;
 166       AliQAv1::SetQARefDataDirName(specie) ;
 167       if ( ! manQA->GetLock() ) {
 168         manQA->SetDefaultStorage(AliQAv1::GetQARefStorage()) ;
 169         manQA->SetSpecificStorage("*", AliQAv1::GetQARefStorage()) ;
 170         manQA->SetRun(AliCDBManager::Instance()->GetRun()) ;
 171         manQA->SetLock() ;
 172       }
 173       char * detOCDBDir = Form("%s/%s/%s", det, AliQAv1::GetRefOCDBDirName(), AliQAv1::GetRefDataDirName()) ;
 174       AliCDBEntry * entry = manQA->Get(detOCDBDir, manQA->GetRun()) ;
 175       if (entry) {
 176         TList * listDetQAD =static_cast<TList *>(entry->GetObject()) ;
 177         if ( strcmp(listDetQAD->ClassName(), "TList") != 0 ) {
 178           AliError(Form("Expected a Tlist and found a %s for detector %s", listDetQAD->ClassName(), det)) ;
 179           listDetQAD = NULL ;
 180           continue ;
 181         }
 182         if ( listDetQAD ) {
 183           TIter next(listDetQAD) ;
 184           TObjArray * ar ;
 185           while ( (ar = (TObjArray*)next()) )
 186             dirOCDB[specie] = static_cast<TObjArray *>(listDetQAD->FindObject(Form("%s/%s", task, AliRecoParam::GetEventSpecieName(specie)))) ;
 187         }
 188       }
 189     }
 190   }
 191 }
 192
 193 //_____________________________________________________________________________
 194 AliQAChecker * AliQAChecker::Instance()
 195 {
 196         // returns unique instance of the checker
 197   if ( ! fgQAChecker )
 198    fgQAChecker = new AliQAChecker() ;
 199   return fgQAChecker ;
 200 }
 201
 202 //_____________________________________________________________________________
 203 void AliQAChecker::LoadRunInfoFromGRP()
 204 {
 205   AliCDBManager* man = AliCDBManager::Instance() ;
 206   AliCDBEntry* entry = man->Get(AliQAv1::GetGRPPath().Data());
 207   AliGRPObject* grpObject = 0x0;
 208   if (entry) {
 209
 210           TMap* m = static_cast<TMap*>(entry->GetObject());  // old GRP entry
 211
 212           if (m) {
 213             AliDebug(AliQAv1::GetQADebugLevel(), "It is a map");
 214             //m->Print();
 215             grpObject = new AliGRPObject();
 216                  grpObject->ReadValuesFromMap(m);
 217     }
 218
 219     else {
 220             AliDebug(AliQAv1::GetQADebugLevel(), "It is a new GRP object");
 221         grpObject = static_cast<AliGRPObject*>(entry->GetObject());  // new GRP entry
 222     }
 223
 224     entry->SetOwner(0);
 225     AliCDBManager::Instance()->UnloadFromCache("GRP/GRP/Data");
 226   }
 227
 228   if (!grpObject) {
 229      AliFatal("No GRP entry found in OCDB!");
 230   }
 231
 232   TString lhcState = grpObject->GetLHCState();
 233   if (lhcState==AliGRPObject::GetInvalidString()) {
 234     AliError("GRP/GRP/Data entry:  missing value for the LHC state ! Using UNKNOWN");
 235     lhcState = "UNKNOWN";
 236   }
 237
 238   TString beamType = grpObject->GetBeamType();
 239   if (beamType==AliGRPObject::GetInvalidString()) {
 240     AliError("GRP/GRP/Data entry:  missing value for the beam type ! Using UNKNOWN");
 241     beamType = "UNKNOWN";
 242   }
 243
 244   Float_t beamEnergy = grpObject->GetBeamEnergy();
 245   if (beamEnergy==AliGRPObject::GetInvalidFloat()) {
 246     AliError("GRP/GRP/Data entry:  missing value for the beam energy ! Using 0");
 247     beamEnergy = 0;
 248   }
 249
 250   TString runType = grpObject->GetRunType();
 251   if (runType==AliGRPObject::GetInvalidString()) {
 252     AliError("GRP/GRP/Data entry:  missing value for the run type ! Using UNKNOWN");
 253     runType = "UNKNOWN";
 254   }
 255
 256   Int_t activeDetectors = grpObject->GetDetectorMask();
 257   if (activeDetectors==AliGRPObject::GetInvalidInt()) {
 258     AliError("GRP/GRP/Data entry:  missing value for the detector mask ! Using 1074790399");
 259     activeDetectors = 1074790399;
 260   }
 261
 262   fRunInfo = new AliRunInfo(lhcState, beamType, beamEnergy, runType, activeDetectors);
 263
 264   fRunInfoOwner = kTRUE ;
 265
 266   // set the event specie
 267   fEventSpecie = AliRecoParam::kDefault ;
 268   if (strcmp(runType,"PHYSICS")) {
 269     // Not a physics run, the event specie is set to kCalib
 270     fEventSpecie = AliRecoParam::kCalib ;
 271     return;
 272   }
 273   if (strcmp(lhcState,"STABLE_BEAMS") == 0) {
 274     // Heavy ion run (any beam tha is not pp, the event specie is set to kHighMult
 275     fEventSpecie = AliRecoParam::kHighMult ;
 276     if ((strcmp(beamType,"p-p") == 0) ||
 277         (strcmp(beamType,"p-")  == 0) ||
 278         (strcmp(beamType,"-p")  == 0) ||
 279         (strcmp(beamType,"P-P") == 0) ||
 280         (strcmp(beamType,"P-")  == 0) ||
 281         (strcmp(beamType,"-P")  == 0)) {
 282       // Proton run, the event specie is set to kLowMult
 283       fEventSpecie = AliRecoParam::kLowMult ;
 284     }
 285     else if (strcmp(beamType,"-") == 0) {
 286       // No beams, we assume cosmic data
 287       fEventSpecie = AliRecoParam::kCosmic ;
 288     }
 289     else if (strcmp(beamType,"UNKNOWN") == 0) {
 290       // No LHC beam information is available, we use the default event specie
 291       fEventSpecie = AliRecoParam::kDefault ;
 292     }
 293   }
 294 }
 295
 296 //_____________________________________________________________________________
 297 Bool_t AliQAChecker::Run(const char * fileName, AliDetectorRecoParam * recoParam)
 298 {
 299   // run the Quality Assurance Checker for all tasks Hits, SDigits, Digits, DigitsR, RecPoints, TrackSegments, RecParticles and ESDs
 300   // starting from data in file
 301   TStopwatch stopwatch;
 302   stopwatch.Start();
 303
 304   //search for all detectors QA directories
 305   TList * detKeyList = AliQAv1::GetQADataFile(fileName)->GetListOfKeys() ;
 306   TIter nextd(detKeyList) ;
 307   TKey * detKey ;
 308   while ( (detKey = static_cast<TKey *>(nextd()) ) ) {
 309     AliDebug(AliQAv1::GetQADebugLevel(), Form("Found %s", detKey->GetName())) ;
 310     //Check which detector
 311     TString detName ;
 312     TString detNameQA(detKey->GetName()) ;
 313     Int_t det ;
 314     for ( det = 0; det < AliQAv1::kNDET ; det++) {
 315       detName = AliQAv1::GetDetName(det) ;
 316       if (detNameQA.Contains(detName)) {
 317         fFoundDetectors+=detName ;
 318         fFoundDetectors+="." ;
 319         break ;
 320       }
 321     }
 322     TDirectory * detDir = AliQAv1::GetQADataFile(fileName)->GetDirectory(detKey->GetName()) ;
 323     TList * taskKeyList = detDir->GetListOfKeys() ;
 324     TIter nextt(taskKeyList) ;
 325     TKey * taskKey ;
 326     // now search for the tasks dir
 327     while ( (taskKey = static_cast<TKey *>(nextt()) ) ) {
 328       TString taskName( taskKey->GetName() ) ;
 329       AliDebug(AliQAv1::GetQADebugLevel(), Form("Found %s", taskName.Data())) ;
 330       TDirectory * taskDir = detDir->GetDirectory(taskName.Data()) ;
 331       taskDir->cd() ;
 332       AliQACheckerBase * qac = GetDetQAChecker(det) ;
 333       if (qac)
 334         AliDebug(AliQAv1::GetQADebugLevel(), Form("QA checker found for %s", detName.Data())) ;
 335       if (!qac)
 336         AliFatal(Form("QA checker not found for %s", detName.Data())) ;
 337       AliQAv1::ALITASK_t index = AliQAv1::kNULLTASK ;
 338       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kHITS) )
 339         index = AliQAv1::kSIM ;
 340       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kSDIGITS) )
 341         index = AliQAv1::kSIM ;
 342       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kDIGITS) )
 343         index = AliQAv1::kSIM ;
 344       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kRAWS) )
 345         index = AliQAv1::kRAW ;
 346       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kDIGITSR) )
 347         index = AliQAv1::kREC ;
 348       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kRECPOINTS) )
 349         index = AliQAv1::kREC ;
 350       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kTRACKSEGMENTS) )
 351         index = AliQAv1::kREC ;
 352       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kRECPARTICLES) )
 353         index = AliQAv1::kREC ;
 354       if ( taskName == AliQAv1::GetTaskName(AliQAv1::kESDS) )
 355         index = AliQAv1::kESD ;
 356       qac->Init(AliQAv1::DETECTORINDEX_t(det)) ;
 357
 358       TDirectory * refDir     = NULL ;
 359       TObjArray ** refOCDBDir = NULL ;
 360       GetRefSubDir(detNameQA.Data(), taskName.Data(), refDir, refOCDBDir) ;
 361                   qac->SetRefandData(refDir, refOCDBDir, taskDir) ;
 362                   qac->Run(index, recoParam) ;
 363     }
 364   }
 365   TString detList ;
 366   for ( Int_t det = 0; det < AliQAv1::kNDET; det++) {
 367     if (fFoundDetectors.Contains(AliQAv1::GetDetName(det))) {
 368       detList += AliQAv1::GetDetName(det) ;
 369       detList += " " ;
 370       fFoundDetectors.ReplaceAll(AliQAv1::GetDetName(det), "") ;
 371       AliQAv1::Instance()->Show(AliQAv1::GetDetIndex(AliQAv1::GetDetName(det))) ;
 372     }
 373   }
 374   AliInfo(Form("QA performed for following detectors: %s", detList.Data())) ;
 375   return kTRUE ;
 376 }
 377
 378 //_____________________________________________________________________________
 379 Bool_t AliQAChecker::Run(AliQAv1::DETECTORINDEX_t det, AliQAv1::TASKINDEX_t task, TObjArray ** list, AliDetectorRecoParam * recoParam)
 380 {
 381         // run the Quality Assurance Checker for detector det, for task task starting from data in list
 382
 383         AliQACheckerBase * qac = GetDetQAChecker(det) ;
 384         if (qac)
 385                 AliDebug(AliQAv1::GetQADebugLevel(), Form("QA checker found for %s", AliQAv1::GetDetName(det).Data())) ;
 386         if (!qac)
 387                 AliError(Form("QA checker not found for %s", AliQAv1::GetDetName(det).Data())) ;
 388
 389         AliQAv1::ALITASK_t index = AliQAv1::kNULLTASK ;
 390         if ( task == AliQAv1::kRAWS )
 391                 index = AliQAv1::kRAW ;
 392         else if ( task == AliQAv1::kHITS )
 393                 index = AliQAv1::kSIM ;
 394         else if ( task == AliQAv1::kSDIGITS )
 395                 index = AliQAv1::kSIM ;
 396         else if ( task == AliQAv1::kDIGITS )
 397                 index = AliQAv1::kSIM ;
 398         else if ( task == AliQAv1::kDIGITSR )
 399                 index = AliQAv1::kREC ;
 400         else if ( task == AliQAv1::kRECPOINTS )
 401                 index = AliQAv1::kREC ;
 402         else if ( task == AliQAv1::kTRACKSEGMENTS )
 403                 index = AliQAv1::kREC ;
 404         else if ( task == AliQAv1::kRECPARTICLES )
 405                 index = AliQAv1::kREC ;
 406         else if ( task == AliQAv1::kESDS )
 407                 index = AliQAv1::kESD ;
 408
 409         TDirectory * refDir     = NULL ;
 410         TObjArray ** refOCDBDir = NULL  ;
 411   qac->Init(det) ;
 412   GetRefSubDir(AliQAv1::GetDetName(det), AliQAv1::GetTaskName(task), refDir, refOCDBDir) ;
 413   qac->SetRefandData(refDir, refOCDBDir) ;
 414   qac->Run(index, list, recoParam) ;
 415
 416   // make the image
 417   qac->MakeImage(list, task, AliQAv1::Mode(task)) ;
 418
 419         return kTRUE ;
 420 }
 421
 422 //_____________________________________________________________________________
 423 Bool_t AliQAChecker::Run(AliQAv1::DETECTORINDEX_t det, AliQAv1::TASKINDEX_t task, TNtupleD ** list, AliDetectorRecoParam * recoParam)
 424 {
 425         // run the Quality Assurance Checker for detector det, for task task starting from data in list
 426
 427         AliQACheckerBase * qac = GetDetQAChecker(det) ;
 428         if (qac)
 429                 AliDebug(AliQAv1::GetQADebugLevel(), Form("QA checker found for %s", AliQAv1::GetDetName(det).Data())) ;
 430         if (!qac)
 431                 AliError(Form("QA checker not found for %s", AliQAv1::GetDetName(det).Data())) ;
 432
 433         AliQAv1::ALITASK_t index = AliQAv1::kNULLTASK ;
 434         if ( task == AliQAv1::kRAWS )
 435                 index = AliQAv1::kRAW ;
 436         else if ( task == AliQAv1::kHITS )
 437                 index = AliQAv1::kSIM ;
 438         else if ( task == AliQAv1::kSDIGITS )
 439                 index = AliQAv1::kSIM ;
 440         else if ( task == AliQAv1::kDIGITS )
 441                 index = AliQAv1::kSIM ;
 442         else if ( task == AliQAv1::kDIGITSR )
 443                 index = AliQAv1::kREC ;
 444         else if ( task == AliQAv1::kRECPOINTS )
 445                 index = AliQAv1::kREC ;
 446         else if ( task == AliQAv1::kTRACKSEGMENTS )
 447                 index = AliQAv1::kREC ;
 448         else if ( task == AliQAv1::kRECPARTICLES )
 449                 index = AliQAv1::kREC ;
 450         else if ( task == AliQAv1::kESDS )
 451                 index = AliQAv1::kESD ;
 452
 453         TDirectory * refDir     = NULL ;
 454         TObjArray ** refOCDBDir = NULL ;
 455   qac->Init(det) ;
 456   GetRefSubDir(AliQAv1::GetDetName(det), AliQAv1::GetTaskName(task), refDir, refOCDBDir) ;
 457   qac->SetRefandData(refDir, refOCDBDir) ;
 458   qac->Run(index, list, recoParam) ;
 459
 460   return kTRUE ;
 461 }