[u/mrichter/AliRoot.git] / STEER / AliQAChecker.cxx

/**************************************************************************
 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id: */

///////////////////////////////////////////////////////////////////////////////
//                                                                           //
// class for running the Quality Assurance Checker
//                                                                           //
///////////////////////////////////////////////////////////////////////////////

#include "AliCDBEntry.h"
#include "AliQAManager.h"
#include "AliCDBStorage.h"
#include "AliRunInfo.h" 
#include "AliLog.h"
#include "AliModule.h" 
#include "AliQAv1.h"
#include "AliQAChecker.h"
#include "AliQACheckerBase.h"
#include "AliCorrQAChecker.h"
#include "AliGlobalQAChecker.h"
#include "AliGRPObject.h"

#include <TKey.h>
#include <TObjArray.h>
#include <TObjString.h>
#include <TPluginManager.h> 
#include <TROOT.h>
#include <TStopwatch.h> 
#include <TString.h> 
#include <TSystem.h> 
#include <TList.h>
#include <TNtupleD.h>

ClassImp(AliQAChecker)
  AliQAChecker * AliQAChecker::fgQAChecker = 0x0 ;

//_____________________________________________________________________________
AliQAChecker::AliQAChecker(const char* name, const char* title) :
  TNamed(name, title),
  fDataFile(0x0), 
  fRunInfo(0x0), 
  fRunInfoOwner(kFALSE), 
  fRefFile(0x0), 
  fFoundDetectors("."), 
  fEventSpecie(AliRecoParam::kDefault) 
{
  // ctor: initialise checkers and open the data file   
  for (Int_t det = 0 ; det < AliQAv1::kNDET ; det++) 
    fCheckers[det] = NULL ; 
}

//_____________________________________________________________________________
AliQAChecker::AliQAChecker(const AliQAChecker& qac) :
  TNamed(qac),
  fDataFile(qac.fDataFile), 
  fRunInfo(qac.fRunInfo), 
  fRunInfoOwner(kFALSE),   
  fRefFile(qac.fRefFile), 
  fFoundDetectors(qac.fFoundDetectors),
  fEventSpecie(qac.fEventSpecie)
{
  // copy constructor
  
  for (Int_t det = 0 ; det < AliQAv1::kNDET ; det++) 
    fCheckers[det] = NULL ; 
}

//_____________________________________________________________________________
AliQAChecker& AliQAChecker::operator = (const AliQAChecker& qac)
{
// assignment operator

  this->~AliQAChecker();
  new(this) AliQAChecker(qac);
  return *this;
}

//_____________________________________________________________________________
AliQAChecker::~AliQAChecker()
{
// clean up
  if (fRunInfo)
    delete fRunInfo ; 
  delete [] fCheckers ; 
  AliQAv1::Close() ; 
}

//_____________________________________________________________________________
  AliQACheckerBase * AliQAChecker::GetDetQAChecker(Int_t det)
{
	// Gets the Quality Assurance checker for the detector specified by its name
	
	if (fCheckers[det]) 
    return fCheckers[det];

	AliQACheckerBase * qac = NULL ;

	TString detName(AliQAv1::GetDetName(det)) ; 
	
	if (det == AliQAv1::kGLOBAL) {
		qac = new AliGlobalQAChecker() ; 
	} else if (det == AliQAv1::kCORR) {
		qac = new AliCorrQAChecker() ; 
	} else {
		AliDebug(1, Form("Retrieving QA checker for %s", detName.Data())) ; 
		TPluginManager* pluginManager = gROOT->GetPluginManager() ;
		TString qacName = "Ali" + detName + "QAChecker" ;

		// first check if a plugin is defined for the quality assurance checker
		TPluginHandler* pluginHandler = pluginManager->FindHandler("AliQAChecker", detName.Data());
		// if not, add a plugin for it
		if (!pluginHandler) {
			//AliInfo(Form("defining plugin for %s", qacName.Data()));
			TString libs = gSystem->GetLibraries();
		
			if (libs.Contains("lib" + detName + "base.so") || (gSystem->Load("lib" + detName + "base.so") >= 0))
				pluginManager->AddHandler("AliQAChecker", detName, qacName, detName + "qac", qacName + "()");
			else 
				pluginManager->AddHandler("AliQAChecker", detName, qacName, detName, qacName + "()");

			pluginHandler = pluginManager->FindHandler("AliQAChecker", detName);	

			if (pluginHandler && (pluginHandler->LoadPlugin() == 0)) 
				qac = (AliQACheckerBase *) pluginHandler->ExecPlugin(0);
  
		}
	}
	if (qac) 
		fCheckers[det] = qac ;
	
	return qac ; 
}
 
//_____________________________________________________________________________
void AliQAChecker::GetRefSubDir(const char * det, const char * task, TDirectory *& dirFile, TObjArray **& dirOCDB)     
{ 
  // Opens and returns the file with the reference data 
  dirFile = NULL ; 
  TString refStorage(AliQAv1::GetQARefStorage()) ;
//  if (refStorage.Contains(AliQAv1::GetLabLocalFile())) {	
//    refStorage.ReplaceAll(AliQAv1::GetLabLocalFile(), "") ; 
//    refStorage += AliQAv1::GetQARefFileName() ;
//    if ( fRefFile ) 
//      if ( fRefFile->IsOpen() ) 
//					fRefFile->Close() ; 
//    fRefFile = TFile::Open(refStorage.Data()) ; 
//    if (!fRefFile) { 
//      AliError(Form("Cannot find reference file %s", refStorage.Data())) ; 
//      dirFile = NULL ; 
//    }
//    dirFile = fRefFile->GetDirectory(det) ; 
//    if (!dirFile) {
//      AliWarning(Form("Directory %s not found in %d", det, refStorage.Data())) ; 
//    } else {
//			dirFile = dirFile->GetDirectory(task) ; 
//      if (!dirFile) 
//				AliWarning(Form("Directory %s/%s not found in %s", det, task, refStorage.Data())) ; 
//    }  
//  } else 
  if (!refStorage.Contains(AliQAv1::GetLabLocalOCDB()) && !refStorage.Contains(AliQAv1::GetLabAliEnOCDB())) {
    AliError(Form("%s is not a valid location for reference data", refStorage.Data())) ; 
    return ; 
  } else {
    AliQAManager* manQA = AliQAManager::QAManager() ;
    for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
      if ( !AliQAv1::Instance()->IsEventSpecieSet(specie) ) 
        continue ; 
      //if ( strcmp(AliQAv1::GetRefDataDirName(), "") == 0 ) { // the name of the last level of the directory is not set (EventSpecie)
        // Get it from RunInfo
        //if (!fRunInfo)  // not yet set, get the info from GRP
        //  LoadRunInfoFromGRP() ; 
      AliQAv1::SetQARefDataDirName(specie) ;
      //}
      if ( ! manQA->GetLock() ) { 
        manQA->SetDefaultStorage(AliQAv1::GetQARefStorage()) ; 
        manQA->SetSpecificStorage("*", AliQAv1::GetQARefStorage()) ;
        manQA->SetRun(AliCDBManager::Instance()->GetRun()) ; 
        manQA->SetLock() ; 
      }
      char * detOCDBDir = Form("%s/%s/%s", det, AliQAv1::GetRefOCDBDirName(), AliQAv1::GetRefDataDirName()) ; 
      AliCDBEntry * entry = manQA->Get(detOCDBDir, manQA->GetRun()) ;
      if (entry) {
        dirOCDB = new TObjArray*[AliRecoParam::kNSpecies] ;	
        TList * listDetQAD = dynamic_cast<TList *>(entry->GetObject()) ;
        TIter next(listDetQAD) ;
        TObjArray * ar ; 
        while ( (ar = (TObjArray*)next()) )
          if ( listDetQAD ) 
          dirOCDB[specie] = dynamic_cast<TObjArray *>(listDetQAD->FindObject(Form("%s/%s", task, AliRecoParam::GetEventSpecieName(specie)))) ; 
      }
    }
  }
}

//_____________________________________________________________________________
AliQAChecker * AliQAChecker::Instance()
{
	// returns unique instance of the checker
  if ( ! fgQAChecker ) 
   fgQAChecker = new AliQAChecker() ; 
  return fgQAChecker ;  
}

//_____________________________________________________________________________
void AliQAChecker::LoadRunInfoFromGRP() 
{
  AliCDBManager* man = AliCDBManager::Instance() ;
  AliCDBEntry* entry = man->Get(AliQAv1::GetGRPPath().Data());
  AliGRPObject* grpObject = 0x0;
  if (entry) {

	  TMap* m = dynamic_cast<TMap*>(entry->GetObject());  // old GRP entry

	  if (m) {
	    AliInfo("It is a map");
	    //m->Print();
	    grpObject = new AliGRPObject();
	         grpObject->ReadValuesFromMap(m);
    }

    else {
	    AliInfo("It is a new GRP object");
        grpObject = dynamic_cast<AliGRPObject*>(entry->GetObject());  // new GRP entry
    }

    entry->SetOwner(0);
    AliCDBManager::Instance()->UnloadFromCache("GRP/GRP/Data");
  }

  if (!grpObject) {
     AliFatal("No GRP entry found in OCDB!");
  }

  TString lhcState = grpObject->GetLHCState();
  if (lhcState==AliGRPObject::GetInvalidString()) {
    AliError("GRP/GRP/Data entry:  missing value for the LHC state ! Using UNKNOWN");
    lhcState = "UNKNOWN";
  }

  TString beamType = grpObject->GetBeamType();
  if (beamType==AliGRPObject::GetInvalidString()) {
    AliError("GRP/GRP/Data entry:  missing value for the beam type ! Using UNKNOWN");
    beamType = "UNKNOWN";
  }

  Float_t beamEnergy = grpObject->GetBeamEnergy();
  if (beamEnergy==AliGRPObject::GetInvalidFloat()) {
    AliError("GRP/GRP/Data entry:  missing value for the beam energy ! Using 0");
    beamEnergy = 0;
  }

  TString runType = grpObject->GetRunType();
  if (runType==AliGRPObject::GetInvalidString()) {
    AliError("GRP/GRP/Data entry:  missing value for the run type ! Using UNKNOWN");
    runType = "UNKNOWN";
  }

  Int_t activeDetectors = grpObject->GetDetectorMask();
  if (activeDetectors==AliGRPObject::GetInvalidInt()) {
    AliError("GRP/GRP/Data entry:  missing value for the detector mask ! Using 1074790399");
    activeDetectors = 1074790399;
  }

  fRunInfo = new AliRunInfo(lhcState, beamType, beamEnergy, runType, activeDetectors);

  fRunInfoOwner = kTRUE ; 

  // set the event specie
  fEventSpecie = AliRecoParam::kDefault ;
  if (strcmp(runType,"PHYSICS")) {
    // Not a physics run, the event specie is set to kCalib
    fEventSpecie = AliRecoParam::kCalib ;
    return;
  }
  if (strcmp(lhcState,"STABLE_BEAMS") == 0) {
    // Heavy ion run (any beam tha is not pp, the event specie is set to kHighMult
    fEventSpecie = AliRecoParam::kHighMult ;
    if ((strcmp(beamType,"p-p") == 0) ||
        (strcmp(beamType,"p-")  == 0) ||
        (strcmp(beamType,"-p")  == 0) ||
        (strcmp(beamType,"P-P") == 0) ||
        (strcmp(beamType,"P-")  == 0) ||
        (strcmp(beamType,"-P")  == 0)) {
      // Proton run, the event specie is set to kLowMult
      fEventSpecie = AliRecoParam::kLowMult ;
    }
    else if (strcmp(beamType,"-") == 0) {
      // No beams, we assume cosmic data
      fEventSpecie = AliRecoParam::kCosmic ;
    }
    else if (strcmp(beamType,"UNKNOWN") == 0) {
      // No LHC beam information is available, we use the default event specie
      fEventSpecie = AliRecoParam::kDefault ;
    }
  }
}

//_____________________________________________________________________________
Bool_t AliQAChecker::Run(const char * fileName)
{
  // run the Quality Assurance Checker for all tasks Hits, SDigits, Digits, recpoints, tracksegments, recparticles and ESDs
  // starting from data in file  
  TStopwatch stopwatch;
  stopwatch.Start();

  //search for all detectors QA directories
  TList * detKeyList = AliQAv1::GetQADataFile(fileName)->GetListOfKeys() ; 
  TIter nextd(detKeyList) ; 
  TKey * detKey ; 
  while ( (detKey = dynamic_cast<TKey *>(nextd()) ) ) {
    AliDebug(1, Form("Found %s", detKey->GetName())) ;
    //Check which detector
    TString detName ; 
    TString detNameQA(detKey->GetName()) ; 
    Int_t det ; 
    for ( det = 0; det < AliQAv1::kNDET ; det++) {
      detName = AliQAv1::GetDetName(det) ; 
      if (detNameQA.Contains(detName)) {
        fFoundDetectors+=detName ; 
        fFoundDetectors+="." ;		
        break ; 
      }
    } 
    TDirectory * detDir = AliQAv1::GetQADataFile(fileName)->GetDirectory(detKey->GetName()) ; 
    TList * taskKeyList = detDir->GetListOfKeys() ;
    TIter nextt(taskKeyList) ; 
    TKey * taskKey ; 
    // now search for the tasks dir
    while ( (taskKey = static_cast<TKey *>(nextt()) ) ) {
      TString taskName( taskKey->GetName() ) ; 
      AliInfo(Form("Found %s", taskName.Data())) ;
      TDirectory * taskDir = detDir->GetDirectory(taskName.Data()) ; 
      taskDir->cd() ; 
      AliQACheckerBase * qac = GetDetQAChecker(det) ; 
      if (qac)
        AliInfo(Form("QA checker found for %s", detName.Data())) ; 
      if (!qac)
        AliFatal(Form("QA checker not found for %s", detName.Data())) ; 
      AliQAv1::ALITASK_t index = AliQAv1::kNULLTASK ; 
      if ( taskName == AliQAv1::GetTaskName(AliQAv1::kHITS) ) 
        index = AliQAv1::kSIM ; 
      if ( taskName == AliQAv1::GetTaskName(AliQAv1::kSDIGITS) ) 
        index = AliQAv1::kSIM ; 
      if ( taskName == AliQAv1::GetTaskName(AliQAv1::kDIGITS) ) 
        index = AliQAv1::kSIM ; 
      if ( taskName == AliQAv1::GetTaskName(AliQAv1::kRECPOINTS) ) 
        index = AliQAv1::kREC ; 
      if ( taskName == AliQAv1::GetTaskName(AliQAv1::kTRACKSEGMENTS) ) 
        index = AliQAv1::kREC ; 
      if ( taskName == AliQAv1::GetTaskName(AliQAv1::kRECPARTICLES) ) 
        index = AliQAv1::kREC ; 
      if ( taskName == AliQAv1::GetTaskName(AliQAv1::kESDS) ) 
        index = AliQAv1::kESD ; 
      qac->Init(AliQAv1::DETECTORINDEX_t(det)) ; 
      
      TDirectory * refDir     = NULL ; 
      TObjArray ** refOCDBDir = NULL ;	
      GetRefSubDir(detNameQA.Data(), taskName.Data(), refDir, refOCDBDir) ;
		  qac->SetRefandData(refDir, refOCDBDir, taskDir) ;
		  qac->Run(index) ; 
    }
  }
  AliInfo("QA performed for following detectors:") ; 
  for ( Int_t det = 0; det < AliQAv1::kNDET; det++) {
    if (fFoundDetectors.Contains(AliQAv1::GetDetName(det))) {
      printf("%s, ",AliQAv1::GetDetName(det)) ; 
      fFoundDetectors.ReplaceAll(AliQAv1::GetDetName(det), "") ; 
    }	
  }
  printf("\n") ; 
  return kTRUE ; 
}

//_____________________________________________________________________________
Bool_t AliQAChecker::Run(AliQAv1::DETECTORINDEX_t det, AliQAv1::TASKINDEX_t task, TObjArray ** list)
{
	// run the Quality Assurance Checker for detector det, for task task starting from data in list

	AliQACheckerBase * qac = GetDetQAChecker(det) ; 
	if (qac)
		AliDebug(1, Form("QA checker found for %s", AliQAv1::GetDetName(det).Data())) ;
	if (!qac)
		AliError(Form("QA checker not found for %s", AliQAv1::GetDetName(det).Data())) ; 
  
	AliQAv1::ALITASK_t index = AliQAv1::kNULLTASK ; 
	if ( task == AliQAv1::kRAWS ) 
		index = AliQAv1::kRAW ; 
	else if ( task == AliQAv1::kHITS ) 
		index = AliQAv1::kSIM ; 
	else if ( task == AliQAv1::kSDIGITS ) 
		index = AliQAv1::kSIM ; 
	else if ( task == AliQAv1::kDIGITS ) 
		index = AliQAv1::kSIM ; 
	else if ( task == AliQAv1::kRECPOINTS ) 
		index = AliQAv1::kREC ; 
	else if ( task == AliQAv1::kTRACKSEGMENTS ) 
		index = AliQAv1::kREC ; 
	else if ( task == AliQAv1::kRECPARTICLES ) 
		index = AliQAv1::kREC ; 
	else if ( task == AliQAv1::kESDS ) 
		index = AliQAv1::kESD ; 

	TDirectory * refDir     = NULL ; 
	TObjArray ** refOCDBDir = NULL  ;	
  qac->Init(det) ; 
  GetRefSubDir(AliQAv1::GetDetName(det), AliQAv1::GetTaskName(task), refDir, refOCDBDir) ;
  qac->SetRefandData(refDir, refOCDBDir) ; 
  qac->Run(index, list) ; 
	return kTRUE ; 
}

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