#include <TCanvas.h>
#include <TKey.h>
#include <TFile.h>
#include <TFileMerger.h>
#include <TGrid.h>
#include <TGridCollection.h>
#include <TGridResult.h>
#include <TPluginManager.h>
#include <TROOT.h>
#include <TString.h>
#include <TSystem.h>
#include <TStopwatch.h>
#include "AliCDBManager.h"
#include "AliCDBEntry.h"
#include "AliCDBId.h"
#include "AliCDBMetaData.h"
#include "AliCodeTimer.h"
#include "AliCorrQADataMakerRec.h"
#include "AliDetectorRecoParam.h"
#include "AliESDEvent.h"
#include "AliGeomManager.h"
#include "AliGlobalQADataMaker.h"
#include "AliHeader.h"
#include "AliLog.h"
#include "AliModule.h"
#include "AliQAv1.h"
#include "AliQAChecker.h"
#include "AliQACheckerBase.h"
#include "AliQADataMakerRec.h"
#include "AliQADataMakerSim.h"
#include "AliQAManager.h"
#include "AliRawReaderDate.h"
#include "AliRawReaderFile.h"
#include "AliRawReaderRoot.h"
#include "AliRun.h"
#include "AliRunLoader.h"
#include "AliRunTag.h"
using std::ifstream;
ClassImp(AliQAManager)
AliQAManager* AliQAManager::fgQAInstance = 0x0;
AliQAManager::AliQAManager() :
AliCDBManager(),
fCurrentEvent(0),
fCycleSame(kFALSE),
fDetectors("ALL"),
fDetectorsW("ALL"),
fESD(NULL),
fESDTree(NULL),
fEventInfo(NULL),
fGAliceFileName(""),
fFirstEvent(0),
fMaxEvents(0),
fMode(""),
fNumberOfEvents(999999),
fRecoParam(),
fRunNumber(0),
fRawReader(NULL),
fRawReaderDelete(kTRUE),
fRunLoader(NULL),
fTasks(""),
fEventSpecie(AliRecoParam::kDefault),
fPrintImage(kTRUE),
fSaveData(kTRUE)
{
fMaxEvents = fNumberOfEvents ;
for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
fLoader[iDet] = NULL ;
fQADataMaker[iDet] = NULL ;
fQACycles[iDet] = 999999 ;
}
}
SetWriteExpert() ;
}
AliQAManager::AliQAManager(AliQAv1::MODE_t mode, const Char_t* gAliceFilename) :
AliCDBManager(),
fCurrentEvent(0),
fCycleSame(kFALSE),
fDetectors("ALL"),
fDetectorsW("ALL"),
fESD(NULL),
fESDTree(NULL),
fEventInfo(NULL),
fGAliceFileName(gAliceFilename),
fFirstEvent(0),
fMaxEvents(0),
fMode(AliQAv1::GetModeName(mode)),
fNumberOfEvents(999999),
fRecoParam(),
fRunNumber(0),
fRawReader(NULL),
fRawReaderDelete(kTRUE),
fRunLoader(NULL),
fTasks(""),
fEventSpecie(AliRecoParam::kDefault),
fPrintImage(kTRUE),
fSaveData(kTRUE)
{
fMaxEvents = fNumberOfEvents ;
for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
fLoader[iDet] = NULL ;
fQADataMaker[iDet] = NULL ;
fQACycles[iDet] = 999999 ;
}
}
SetWriteExpert() ;
}
AliQAManager::AliQAManager(const AliQAManager & qas) :
AliCDBManager(),
fCurrentEvent(qas.fCurrentEvent),
fCycleSame(kFALSE),
fDetectors(qas.fDetectors),
fDetectorsW(qas.fDetectorsW),
fESD(NULL),
fESDTree(NULL),
fEventInfo(NULL),
fGAliceFileName(qas.fGAliceFileName),
fFirstEvent(qas.fFirstEvent),
fMaxEvents(qas.fMaxEvents),
fMode(qas.fMode),
fNumberOfEvents(qas.fNumberOfEvents),
fRecoParam(),
fRunNumber(qas.fRunNumber),
fRawReader(NULL),
fRawReaderDelete(kTRUE),
fRunLoader(NULL),
fTasks(qas.fTasks),
fEventSpecie(qas.fEventSpecie),
fPrintImage(qas.fPrintImage),
fSaveData(qas.fSaveData)
{
for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
fLoader[iDet] = qas.fLoader[iDet] ;
fQADataMaker[iDet] = qas.fQADataMaker[iDet] ;
fQACycles[iDet] = qas.fQACycles[iDet] ;
fQAWriteExpert[iDet] = qas.fQAWriteExpert[iDet] ;
}
}
AliQAManager & AliQAManager::operator = (const AliQAManager & qas)
{
this->~AliQAManager() ;
new(this) AliQAManager(qas) ;
return *this ;
}
AliQAManager::~AliQAManager()
{
for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
fLoader[iDet] = NULL;
if (fQADataMaker[iDet]) {
(fQADataMaker[iDet])->Finish() ;
delete fQADataMaker[iDet] ;
}
}
}
if (fRawReaderDelete) {
fRunLoader = NULL ;
delete fRawReader ;
fRawReader = NULL ;
}
}
Bool_t AliQAManager::DoIt(const AliQAv1::TASKINDEX_t taskIndex)
{
Bool_t rv = kFALSE ;
for (UInt_t iEvent = fFirstEvent ; iEvent < (UInt_t)fMaxEvents ; iEvent++) {
fCurrentEvent++ ;
if ( iEvent%10 == 0 )
AliDebug(AliQAv1::GetQADebugLevel(), Form("processing event %d", iEvent));
if ( taskIndex == AliQAv1::kRAWS ) {
if ( !fRawReader->NextEvent() )
break ;
} else if ( taskIndex == AliQAv1::kESDS ) {
if ( fESDTree->GetEntry(iEvent) == 0 )
break ;
} else {
if ( fRunLoader->GetEvent(iEvent) != 0 )
break ;
}
TString detList ;
if ( GetEventInfo())
detList = GetEventInfo()->GetTriggerCluster() ;
for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
if (!detList.IsNull() && !detList.Contains(AliQAv1::GetDetName(iDet)))
continue ;
if (IsSelected(AliQAv1::GetDetName(iDet)) ){
AliQADataMaker * qadm = GetQADataMaker(iDet) ;
if (!qadm) continue;
if ( qadm->IsCycleDone() ) {
qadm->EndOfCycle(taskIndex) ;
}
TTree * data = NULL ;
AliLoader* loader = GetLoader(qadm->GetUniqueID());
switch (taskIndex) {
case AliQAv1::kNULLTASKINDEX :
break ;
case AliQAv1::kRAWS :
qadm->Exec(taskIndex, fRawReader) ;
break ;
case AliQAv1::kHITS :
if( loader ) {
loader->LoadHits() ;
data = loader->TreeH() ;
if ( ! data ) {
AliWarning(Form(" Hit Tree not found for %s", AliQAv1::GetDetName(iDet))) ;
break ;
}
qadm->Exec(taskIndex, data) ;
}
break ;
case AliQAv1::kSDIGITS :
{
TString fileName(Form("%s.SDigits.root", AliQAv1::GetDetName(iDet))) ;
if (gSystem->FindFile("./", fileName)) {
if( loader ) {
loader->LoadSDigits() ;
data = loader->TreeS() ;
if ( ! data ) {
AliWarning(Form(" SDigit Tree not found for %s", AliQAv1::GetDetName(iDet))) ;
break ;
}
qadm->Exec(taskIndex, data) ;
}
}
}
break;
case AliQAv1::kDIGITS :
if( loader ) {
loader->LoadDigits() ;
data = loader->TreeD() ;
if ( ! data ) {
AliWarning(Form(" Digit Tree not found for %s", AliQAv1::GetDetName(iDet))) ;
break ;
}
qadm->Exec(taskIndex, data) ;
}
break;
case AliQAv1::kDIGITSR :
if( loader ) {
loader->LoadDigits() ;
data = loader->TreeD() ;
if ( ! data ) {
AliWarning(Form(" Digit Tree not found for %s", AliQAv1::GetDetName(iDet))) ;
break ;
}
qadm->Exec(taskIndex, data) ;
}
break;
case AliQAv1::kRECPOINTS :
if( loader ) {
loader->LoadRecPoints() ;
data = loader->TreeR() ;
if (!data) {
AliWarning(Form("RecPoints not found for %s", AliQAv1::GetDetName(iDet))) ;
break ;
}
qadm->Exec(taskIndex, data) ;
}
break;
case AliQAv1::kTRACKSEGMENTS :
break;
case AliQAv1::kRECPARTICLES :
break;
case AliQAv1::kESDS :
qadm->Exec(taskIndex, fESD) ;
break;
case AliQAv1::kNTASKINDEX :
break;
}
}
}
Increment(taskIndex) ;
}
EndOfCycle() ;
if ( taskIndex == AliQAv1::kRAWS )
fRawReader->RewindEvents() ;
return rv ;
}
Bool_t AliQAManager::Finish(const AliQAv1::TASKINDEX_t taskIndex)
{
AliQAChecker::Instance()->SetRunNumber(fRunNumber) ;
for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
AliQADataMaker * qadm = GetQADataMaker(iDet) ;
if (qadm)
qadm->EndOfCycle(taskIndex) ;
}
}
return kTRUE ;
}
TObjArray * AliQAManager::GetFromOCDB(AliQAv1::DETECTORINDEX_t det, AliQAv1::TASKINDEX_t task, const Char_t * year) const
{
TObjArray * rv = NULL ;
if ( !strlen(AliQAv1::GetQARefStorage()) ) {
AliError("No storage defined, use AliQAv1::SetQARefStorage") ;
return NULL ;
}
if ( ! IsDefaultStorageSet() ) {
TString tmp(AliQAv1::GetQARefDefaultStorage()) ;
tmp.Append(year) ;
tmp.Append("/") ;
Instance()->SetDefaultStorage(tmp.Data()) ;
Instance()->SetSpecificStorage(Form("%s/*", AliQAv1::GetQAName()), AliQAv1::GetQARefStorage()) ;
}
TString detOCDBDir(Form("%s/%s/%s", AliQAv1::GetQAName(), AliQAv1::GetDetName((Int_t)det), AliQAv1::GetRefOCDBDirName())) ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("Retrieving reference data from %s/%s for %s", AliQAv1::GetQARefStorage(), detOCDBDir.Data(), AliQAv1::GetTaskName(task).Data())) ;
AliCDBEntry* entry = QAManager()->Get(detOCDBDir.Data(), 0) ;
TList * listDetQAD = static_cast<TList *>(entry->GetObject()) ;
if ( listDetQAD )
rv = static_cast<TObjArray *>(listDetQAD->FindObject(AliQAv1::GetTaskName(task))) ;
return rv ;
}
TCanvas ** AliQAManager::GetImage(Char_t * detName)
{
TCanvas ** rv = NULL ;
Int_t detIndex = AliQAv1::GetDetIndex(detName) ;
if ( detIndex != AliQAv1::kNULLDET) {
AliQACheckerBase * qac = AliQAChecker::Instance()->GetDetQAChecker(detIndex) ;
rv = qac->GetImage() ;
}
return rv ;
}
AliLoader * AliQAManager::GetLoader(Int_t iDet)
{
if ( !fRunLoader || iDet == AliQAv1::kCORR || iDet == AliQAv1::kGLOBAL )
return NULL ;
TString detName = AliQAv1::GetDetName(iDet) ;
fLoader[iDet] = fRunLoader->GetLoader(detName + "Loader");
if (fLoader[iDet])
return fLoader[iDet] ;
TPluginManager* pluginManager = gROOT->GetPluginManager() ;
TString loaderName = "Ali" + detName + "Loader" ;
AliLoader * loader = NULL ;
TPluginHandler* pluginHandler = pluginManager->FindHandler("AliLoader", detName) ;
if (!pluginHandler) {
AliDebug(AliQAv1::GetQADebugLevel(), Form("defining plugin for %s", loaderName.Data())) ;
TString libs = gSystem->GetLibraries() ;
if (libs.Contains("lib" + detName + "base.so") || (gSystem->Load("lib" + detName + "base.so") >= 0)) {
pluginManager->AddHandler("AliQADataMaker", detName, loaderName, detName + "loader", loaderName + "()") ;
} else {
pluginManager->AddHandler("AliLoader", detName, loaderName, detName, loaderName + "()") ;
}
pluginHandler = pluginManager->FindHandler("AliLoader", detName) ;
}
if (pluginHandler && (pluginHandler->LoadPlugin() == 0)) {
loader = (AliLoader *) pluginHandler->ExecPlugin(0) ;
}
if (loader)
fLoader[iDet] = loader ;
return loader ;
}
AliQAv1 * AliQAManager::GetQA(UInt_t run, UInt_t evt)
{
Char_t * fileName = Form("Run%d.Event%d_1.ESD.tag.root", run, evt) ;
TFile * tagFile = TFile::Open(fileName) ;
if ( !tagFile ) {
AliError(Form("File %s not found", fileName)) ;
return NULL ;
}
TTree * tagTree = static_cast<TTree *>(tagFile->Get("T")) ;
if ( !tagTree ) {
AliError(Form("Tree T not found in %s", fileName)) ;
tagFile->Close() ;
return NULL ;
}
AliRunTag * tag = new AliRunTag ;
tagTree->SetBranchAddress("AliTAG", &tag) ;
tagTree->GetEntry(evt) ;
AliQAv1 * qa = AliQAv1::Instance(tag->GetQALength(), tag->GetQAArray(), tag->GetESLength(), tag->GetEventSpecies()) ;
tagFile->Close() ;
return qa ;
}
AliQADataMaker * AliQAManager::GetQADataMaker(const Int_t iDet)
{
AliQADataMaker * qadm = fQADataMaker[iDet] ;
if (qadm) {
qadm->SetEventSpecie(fEventSpecie) ;
if ( qadm->GetRecoParam() )
if ( AliRecoParam::Convert(qadm->GetRecoParam()->GetEventSpecie()) != AliRecoParam::kDefault)
qadm->SetEventSpecie(qadm->GetRecoParam()->GetEventSpecie()) ;
} else if (iDet == AliQAv1::kGLOBAL && strcmp(GetMode(), AliQAv1::GetModeName(AliQAv1::kRECMODE)) == 0) {
qadm = new AliGlobalQADataMaker();
qadm->SetName(AliQAv1::GetDetName(iDet));
qadm->SetUniqueID(iDet);
fQADataMaker[iDet] = qadm;
qadm->SetEventSpecie(fEventSpecie) ;
if ( qadm->GetRecoParam() )
if ( AliRecoParam::Convert(qadm->GetRecoParam()->GetEventSpecie()) != AliRecoParam::kDefault)
qadm->SetEventSpecie(qadm->GetRecoParam()->GetEventSpecie()) ;
} else if (iDet == AliQAv1::kCORR && strcmp(GetMode(), AliQAv1::GetModeName(AliQAv1::kRECMODE)) == 0 ) {
qadm = new AliCorrQADataMakerRec(fQADataMaker) ;
qadm->SetName(AliQAv1::GetDetName(iDet));
qadm->SetUniqueID(iDet);
fQADataMaker[iDet] = qadm;
qadm->SetEventSpecie(fEventSpecie) ;
if ( qadm->GetRecoParam() )
if ( AliRecoParam::Convert(qadm->GetRecoParam()->GetEventSpecie()) != AliRecoParam::kDefault)
qadm->SetEventSpecie(qadm->GetRecoParam()->GetEventSpecie()) ;
} else if ( iDet < AliQAv1::kGLOBAL ) {
TString smode(GetMode()) ;
if (smode.Contains(AliQAv1::GetModeName(AliQAv1::kQAMODE)))
smode = AliQAv1::GetModeName(AliQAv1::kRECMODE) ;
TPluginManager* pluginManager = gROOT->GetPluginManager() ;
TString detName = AliQAv1::GetDetName(iDet) ;
TString qadmName = "Ali" + detName + "QADataMaker" + smode ;
TPluginHandler* pluginHandler = pluginManager->FindHandler("AliQADataMaker", detName) ;
if (!pluginHandler) {
AliDebug(AliQAv1::GetQADebugLevel(), Form("defining plugin for %s", qadmName.Data())) ;
TString libs = gSystem->GetLibraries() ;
TString temp(smode) ;
temp.ToLower() ;
if (libs.Contains("lib" + detName + smode + ".so") || (gSystem->Load("lib" + detName + temp.Data() + ".so") >= 0)) {
if ( iDet == AliQAv1::kMUON ) {
pluginManager->AddHandler("AliQADataMaker", detName, qadmName, detName + "qadm", qadmName + "(Bool_t,Bool_t)");
} else {
pluginManager->AddHandler("AliQADataMaker", detName, qadmName, detName + "qadm", qadmName + "()");
}
} else {
pluginManager->AddHandler("AliQADataMaker", detName, qadmName, detName, qadmName + "()");
}
pluginHandler = pluginManager->FindHandler("AliQADataMaker", detName) ;
}
if (pluginHandler && (pluginHandler->LoadPlugin() == 0)) {
if ( iDet == AliQAv1::kMUON ) {
Bool_t mch = fActiveOnlineDetectors.Contains("MUONTRK");
Bool_t mtr = fActiveOnlineDetectors.Contains("MUONTRG");
qadm = (AliQADataMaker *) pluginHandler->ExecPlugin(2,mch,mtr);
}
else {
qadm = (AliQADataMaker *) pluginHandler->ExecPlugin(0);
}
}
if (qadm) {
qadm->SetName(AliQAv1::GetDetName(iDet));
qadm->SetUniqueID(iDet);
fQADataMaker[iDet] = qadm ;
qadm->SetEventSpecie(fEventSpecie) ;
if ( qadm->GetRecoParam() )
if ( AliRecoParam::Convert(qadm->GetRecoParam()->GetEventSpecie()) != AliRecoParam::kDefault)
qadm->SetEventSpecie(qadm->GetRecoParam()->GetEventSpecie()) ;
}
}
return qadm ;
}
void AliQAManager::EndOfCycle(TObjArray * detArray)
{
AliQAChecker::Instance()->SetRunNumber(fRunNumber) ;
TCanvas fakeCanvas ;
fakeCanvas.Print(Form("%s%s%d.%s[", AliQAv1::GetImageFileName(), GetMode(), fRunNumber, AliQAv1::GetImageFileFormat()), "ps") ;
for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
AliQADataMaker * qadm = GetQADataMaker(iDet) ;
if (!qadm)
continue ;
if (detArray) {
AliModule* det = static_cast<AliModule*>(detArray->FindObject(AliQAv1::GetDetName(iDet))) ;
if (!det || !det->IsActive())
continue ;
}
AliQACheckerBase * qac = AliQAChecker::Instance()->GetDetQAChecker(iDet) ;
if (qac)
qac->SetPrintImage(fPrintImage) ;
for (UInt_t taskIndex = 0; taskIndex < AliQAv1::kNTASKINDEX; taskIndex++) {
if ( fTasks.Contains(Form("%d", taskIndex)) )
qadm->EndOfCycle(AliQAv1::GetTaskIndex(AliQAv1::GetTaskName(taskIndex))) ;
}
qadm->Finish();
}
}
if (fPrintImage)
fakeCanvas.Print(Form("%s%s%d.%s]", AliQAv1::GetImageFileName(), GetMode(), fRunNumber, AliQAv1::GetImageFileFormat()), "ps");
}
void AliQAManager::EndOfCycle(TString detectors)
{
AliQAChecker::Instance()->SetRunNumber(fRunNumber) ;
TCanvas fakeCanvas ;
if (fPrintImage)
fakeCanvas.Print(Form("%s%s%d.%s[", AliQAv1::GetImageFileName(), GetMode(), fRunNumber, AliQAv1::GetImageFileFormat()), "ps") ;
for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
AliQADataMaker * qadm = GetQADataMaker(iDet) ;
if (!qadm)
continue ;
if (!detectors.Contains(AliQAv1::GetDetName(iDet)))
continue ;
AliQACheckerBase * qac = AliQAChecker::Instance()->GetDetQAChecker(iDet) ;
if (qac)
qac->SetPrintImage(fPrintImage) ;
for (UInt_t taskIndex = 0; taskIndex < AliQAv1::kNTASKINDEX; taskIndex++) {
if ( fTasks.Contains(Form("%d", taskIndex)) )
qadm->EndOfCycle(AliQAv1::GetTaskIndex(AliQAv1::GetTaskName(taskIndex))) ;
}
qadm->Finish();
}
}
if (fPrintImage)
fakeCanvas.Print(Form("%s%s%d.%s]", AliQAv1::GetImageFileName(), GetMode(), fRunNumber, AliQAv1::GetImageFileFormat()), "ps");
}
AliRecoParam::EventSpecie_t AliQAManager::GetEventSpecieFromESD()
{
AliRecoParam::EventSpecie_t runtype = AliRecoParam::kDefault ;
if (!gSystem->AccessPathName("AliESDs.root")) {
TFile * esdFile = TFile::Open("AliESDs.root") ;
TTree * esdTree = static_cast<TTree *> (esdFile->Get("esdTree")) ;
if ( !esdTree ) {
AliError("esdTree not found") ;
} else {
AliESDEvent * esd = new AliESDEvent() ;
esd->ReadFromTree(esdTree) ;
esdTree->GetEntry(0) ;
runtype = AliRecoParam::Convert(esd->GetEventType()) ;
}
} else {
AliError("AliESDs.root not found") ;
}
return runtype ;
}
void AliQAManager::Increment(const AliQAv1::TASKINDEX_t taskIndex)
{
static AliQAv1::TASKINDEX_t currentTask = AliQAv1::kNTASKINDEX ;
if ( (currentTask == taskIndex) && taskIndex != AliQAv1::kNULLTASKINDEX )
return ;
else
currentTask = taskIndex ;
for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
AliQADataMaker * qadm = GetQADataMaker(iDet) ;
if (qadm)
qadm->Increment() ;
}
}
}
Bool_t AliQAManager::InitQA(const AliQAv1::TASKINDEX_t taskIndex, const Char_t * input )
{
fTasks += Form("%d", taskIndex) ;
if (taskIndex == AliQAv1::kRAWS) {
if (!fRawReader) {
fRawReader = AliRawReader::Create(input);
}
if ( ! fRawReader )
return kFALSE ;
fRawReaderDelete = kTRUE ;
fRawReader->NextEvent() ;
fRunNumber = fRawReader->GetRunNumber() ;
SetRun(fRunNumber) ;
fRawReader->RewindEvents();
fNumberOfEvents = 999999 ;
if ( fMaxEvents < 0 )
fMaxEvents = fNumberOfEvents ;
} else if (taskIndex == AliQAv1::kESDS) {
fTasks = AliQAv1::GetTaskName(AliQAv1::kESDS) ;
if (!gSystem->AccessPathName("AliESDs.root")) {
TFile * esdFile = TFile::Open("AliESDs.root") ;
fESDTree = static_cast<TTree *> (esdFile->Get("esdTree")) ;
if ( !fESDTree ) {
AliError("esdTree not found") ;
return kFALSE ;
} else {
fESD = new AliESDEvent() ;
fESD->ReadFromTree(fESDTree) ;
fESDTree->GetEntry(0) ;
fRunNumber = fESD->GetRunNumber() ;
fNumberOfEvents = fESDTree->GetEntries() ;
if ( fMaxEvents < 0 )
fMaxEvents = fNumberOfEvents ;
}
} else {
AliError("AliESDs.root not found") ;
return kFALSE ;
}
} else {
if ( !InitRunLoader() ) {
AliWarning("No Run Loader not found") ;
} else {
fNumberOfEvents = fRunLoader->GetNumberOfEvents() ;
if ( fMaxEvents < 0 )
fMaxEvents = fNumberOfEvents ;
}
}
TObjArray* detArray = NULL ;
if (fRunLoader)
if ( fRunLoader->GetAliRun() ) {
detArray = fRunLoader->GetAliRun()->Detectors() ;
fRunNumber = fRunLoader->GetHeader()->GetRun() ;
}
fRunNumber = AliCDBManager::Instance()->GetRun() ;
if ( ! AliGeomManager::GetGeometry() )
AliGeomManager::LoadGeometry() ;
InitQADataMaker(fRunNumber, detArray) ;
if (fPrintImage) {
TCanvas fakeCanvas ;
TStopwatch timer ;
timer.Start() ;
while (timer.CpuTime()<5) {
timer.Continue();
gSystem->ProcessEvents();
}
fakeCanvas.Print(Form("%s%s%d.%s[", AliQAv1::GetImageFileName(), GetMode(), fRunNumber, AliQAv1::GetImageFileFormat()), "ps") ;
}
return kTRUE ;
}
void AliQAManager::InitQADataMaker(UInt_t run, TObjArray * detArray)
{
fRunNumber = run ;
for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
AliQADataMaker * qadm = GetQADataMaker(iDet) ;
if (!qadm) {
AliError(Form("AliQADataMaker not found for %s", AliQAv1::GetDetName(iDet))) ;
fDetectorsW.ReplaceAll(AliQAv1::GetDetName(iDet), "") ;
} else {
if (fQAWriteExpert[iDet])
qadm->SetWriteExpert() ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("Data Maker found for %s %d", qadm->GetName(), qadm->WriteExpert())) ;
if (detArray) {
AliModule* det = static_cast<AliModule*>(detArray->FindObject(AliQAv1::GetDetName(iDet))) ;
if (!det || !det->IsActive())
continue ;
}
Bool_t sameCycle = kFALSE ;
for (UInt_t taskIndex = 0; taskIndex < AliQAv1::kNTASKINDEX; taskIndex++) {
if ( fTasks.Contains(Form("%d", taskIndex)) ) {
qadm->Init(AliQAv1::GetTaskIndex(AliQAv1::GetTaskName(taskIndex)), GetQACycles(qadm->GetUniqueID())) ;
qadm->StartOfCycle(AliQAv1::GetTaskIndex(AliQAv1::GetTaskName(taskIndex)), run, sameCycle) ;
sameCycle = kTRUE ;
}
}
}
}
}
}
Bool_t AliQAManager::InitRunLoader()
{
if (fRunLoader) {
fCycleSame = kTRUE ;
} else {
if (!gSystem->AccessPathName(fGAliceFileName.Data())) {
TString libs = gSystem->GetLibraries() ;
for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
if (!IsSelected(AliQAv1::GetDetName(iDet)))
continue ;
TString detName = AliQAv1::GetDetName(iDet) ;
if (detName == "HLT")
continue;
if (libs.Contains("lib" + detName + "base.so"))
continue;
gSystem->Load("lib" + detName + "base.so");
}
fRunLoader = AliRunLoader::Open(fGAliceFileName.Data());
if (!fRunLoader) {
AliError(Form("no run loader found in file %s", fGAliceFileName.Data()));
return kFALSE;
}
fRunLoader->CdGAFile();
if (fRunLoader->LoadgAlice() == 0) {
gAlice = fRunLoader->GetAliRun();
}
if (!gAlice) {
AliError(Form("no gAlice object found in file %s", fGAliceFileName.Data()));
return kFALSE;
}
} else {
AliError(Form("the file %s does not exist", fGAliceFileName.Data()));
return kFALSE;
}
}
if (!fRunNumber) {
fRunLoader->LoadHeader();
fRunNumber = fRunLoader->GetHeader()->GetRun() ;
}
return kTRUE;
}
Bool_t AliQAManager::IsSelected(const Char_t * det)
{
Bool_t rv = kFALSE;
const TString detName(det) ;
if (fDetectors.Contains("ALL")) {
fDetectors = "ALL";
rv = kTRUE;
} else if ((fDetectors.CompareTo(detName) == 0) ||
fDetectors.BeginsWith(detName+" ") ||
fDetectors.EndsWith(" "+detName) ||
fDetectors.Contains(" "+detName+" ")) {
rv = kTRUE;
}
return rv ;
}
Bool_t AliQAManager::Merge(Int_t runNumber, const char *fileName) const
{
if ( runNumber == -1)
runNumber = fRunNumber ;
Bool_t rv = MergeData(runNumber,fileName) ;
return rv ;
}
Bool_t AliQAManager::MergeXML(const Char_t * collectionFile, const Char_t * subFile, const Char_t * outFile)
{
Bool_t rv = kFALSE ;
if ( strstr(collectionFile, ".xml") == 0 ) {
AliError("Input collection file must be an \".xml\" file\n") ;
return kFALSE ;
}
if ( !gGrid )
TGrid::Connect("alien://");
if ( !gGrid )
return kFALSE ;
AliInfoClass(Form("*** Create Collection ***\n*** Wk-Dir = |%s| \n*** Coll = |%s| \n",gSystem->WorkingDirectory(), collectionFile));
TGridCollection * collection = (TGridCollection*)gROOT->ProcessLine(Form("TAlienCollection::Open(\"%s\")",collectionFile));
TGridResult* result = collection->GetGridResult("", 0, 0);
Int_t index = 0 ;
const Char_t * turl ;
TFileMerger merger(kFALSE) ;
if (!outFile) {
TString tempo(collectionFile) ;
if ( subFile)
tempo.ReplaceAll(".xml", subFile) ;
else
tempo.ReplaceAll(".xml", "_Merged.root") ;
outFile = tempo.Data() ;
}
merger.OutputFile(outFile) ;
while ( (turl = result->GetKey(index, "turl")) ) {
Char_t * file ;
if ( subFile )
file = Form("%s#%s", turl, subFile) ;
else
file = Form("%s", turl) ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("%s\n", file)) ;
merger.AddFile(file) ;
index++ ;
}
if (index)
merger.Merge() ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("Files merged into %s\n", outFile)) ;
rv = kFALSE;
return rv ;
}
void AliQAManager::MergeCustom() const
{
gROOT->ProcessLine(".! ls *QA*.root > QAtempo.txt") ;
TString theQAfile ;
FILE * theQAfiles = fopen("QAtempo.txt", "r") ;
Int_t index = 0 ;
TList srunList ;
TIter nextRun(&srunList) ;
TObjString * srun = NULL ;
Int_t loRun = 999999999 ;
Int_t hiRun = 0 ;
while ( theQAfile.Gets(theQAfiles) ) {
Bool_t runExist = kFALSE ;
TString srunNew(theQAfile(theQAfile.Index("QA.")+3, theQAfile.Index(".root")-(theQAfile.Index("QA.")+3))) ;
Int_t cuRun = srunNew.Atoi() ;
if (cuRun < loRun)
loRun = cuRun ;
if (cuRun > hiRun)
hiRun = cuRun ;
while ( (srun = static_cast<TObjString *> (nextRun())) ) {
if ( cuRun == (srun->String()).Atoi() ) {
runExist = kTRUE ;
break ;
}
}
nextRun.Reset() ;
if ( ! runExist )
srunList.Add(new TObjString(srunNew.Data()));
}
nextRun.Reset() ;
Int_t runNumber = 0 ;
TFile mergedFile(Form("Merged.%s.Data.root", AliQAv1::GetQADataFileName()), "RECREATE") ;
TH1I * hisRun = new TH1I("hLMR", "List of merged runs", hiRun-loRun+10, loRun, hiRun+10) ;
for (Int_t iDet = 0; iDet < AliQAv1::kNDET ; iDet++) {
TDirectory * detDir = mergedFile.mkdir(AliQAv1::GetDetName(iDet)) ;
for (Int_t taskIndex = 0; taskIndex < AliQAv1::kNTASKINDEX; taskIndex++) {
detDir->cd() ;
TDirectory * taskDir = gDirectory->mkdir(AliQAv1::GetTaskName(taskIndex)) ;
for (Int_t es = 0 ; es < AliRecoParam::kNSpecies ; es++) {
taskDir->cd() ;
TDirectory * esDir = gDirectory->mkdir(AliRecoParam::GetEventSpecieName(es)) ;
esDir->cd() ;
gDirectory->mkdir(AliQAv1::GetExpert()) ;
}
}
}
while ( (srun = static_cast<TObjString *> (nextRun())) ) {
runNumber = (srun->String()).Atoi() ;
hisRun->Fill(runNumber) ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("Merging run number %d", runNumber)) ;
Char_t * fileList[AliQAv1::kNDET] ;
index = 0 ;
for (Int_t iDet = 0; iDet < AliQAv1::kNDET ; iDet++) {
Char_t * file = gSystem->Which(gSystem->WorkingDirectory(), Form("%s.%s.%d.root", AliQAv1::GetDetName(iDet), AliQAv1::GetQADataFileName(), runNumber));
if (file)
fileList[index++] = file ;
}
if ( index == 0 ) {
AliError("No QA data file found\n") ;
return ;
}
for ( Int_t i = 0 ; i < index ; i++) {
TFile * inFile = TFile::Open(fileList[i]) ;
TList * listOfKeys =inFile->GetListOfKeys() ;
TIter nextkey(listOfKeys) ;
TObject * obj1 ;
TString dirName("") ;
while ( (obj1 = nextkey()) ) {
TDirectory * directoryDet = inFile->GetDirectory(obj1->GetName()) ;
if ( directoryDet ) {
AliDebug(AliQAv1::GetQADebugLevel(), Form("%s dir = %s", inFile->GetName(), directoryDet->GetName())) ;
dirName += Form("%s/", directoryDet->GetName() ) ;
directoryDet->cd() ;
TList * listOfTasks = directoryDet->GetListOfKeys() ;
TIter nextTask(listOfTasks) ;
TObject * obj2 ;
while ( (obj2 = nextTask()) ) {
TDirectory * directoryTask = directoryDet->GetDirectory(obj2->GetName()) ;
if ( directoryTask ) {
dirName += Form("%s", obj2->GetName()) ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("%s", dirName.Data())) ;
directoryTask->cd() ;
TList * listOfEventSpecie = directoryTask->GetListOfKeys() ;
TIter nextEventSpecie(listOfEventSpecie) ;
TObject * obj3 ;
while ( (obj3 = nextEventSpecie()) ) {
TDirectory * directoryEventSpecie = directoryTask->GetDirectory(obj3->GetName()) ;
if ( directoryEventSpecie ) {
dirName += Form("/%s/", obj3->GetName()) ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("%s\n", dirName.Data())) ;
directoryEventSpecie->cd() ;
TDirectory * mergedDirectory = mergedFile.GetDirectory(dirName.Data()) ;
TList * listOfData = directoryEventSpecie->GetListOfKeys() ;
TIter nextData(listOfData) ;
TKey * key ;
while ( (key = static_cast<TKey *>(nextData())) ) {
TString className(key->GetClassName()) ;
if ( className.Contains("TH") || className.Contains("TProfile") ) {
TH1 * histIn = static_cast<TH1*> (key->ReadObj()) ;
TH1 * histOu = static_cast<TH1*> (mergedDirectory->FindObjectAny(histIn->GetName())) ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("%s %p %p\n", key->GetName(), histIn, histOu)) ;
mergedDirectory->cd() ;
if ( ! histOu ) {
histIn->Write() ;
} else {
histOu->Add(histIn) ;
histOu->Write(histOu->GetName(), kOverwrite) ;
}
}
else if ( className.Contains("TDirectoryFile") ) {
TDirectory * dirExpert = directoryEventSpecie->GetDirectory(key->GetName()) ;
dirExpert->cd() ;
TDirectory * mergedDirectoryExpert = mergedDirectory->GetDirectory(dirExpert->GetName()) ;
TList * listOfExpertData = dirExpert->GetListOfKeys() ;
TIter nextExpertData(listOfExpertData) ;
TKey * keykey ;
while ( (keykey = static_cast<TKey *>(nextExpertData())) ) {
TString classNameExpert(keykey->GetClassName()) ;
if (classNameExpert.Contains("TH")) {
TH1 * histInExpert = static_cast<TH1*> (keykey->ReadObj()) ;
TH1 * histOuExpert = static_cast<TH1*> (mergedDirectory->FindObjectAny(histInExpert->GetName())) ;
mergedDirectoryExpert->cd() ;
if ( ! histOuExpert ) {
histInExpert->Write() ;
} else {
histOuExpert->Add(histInExpert) ;
histOuExpert->Write(histOuExpert->GetName(), kOverwrite) ;
}
}
}
} else {
AliError(Form("No merge done for this object %s in %s", key->GetName(), dirName.Data())) ;
}
}
dirName.ReplaceAll(Form("/%s/",obj3->GetName()), "") ;
}
}
dirName.ReplaceAll(obj2->GetName(), "") ;
}
}
}
}
inFile->Close() ;
}
}
mergedFile.cd() ;
hisRun->Write() ;
mergedFile.Close() ;
srunList.Delete() ;
}
Bool_t AliQAManager::MergeData(const Int_t runNumber, const char *fileName) const
{
TFileMerger merger(kFALSE) ;
TString outFileName = fileName;
if (outFileName.IsNull()) outFileName.Form("Merged.%s.Data.root",AliQAv1::GetQADataFileName());
merger.OutputFile(outFileName.Data()) ;
for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
Char_t * file = gSystem->Which(gSystem->WorkingDirectory(), Form("%s.%s.%d.root", AliQAv1::GetDetName(iDet), AliQAv1::GetQADataFileName(), runNumber));
if (file)
merger.AddFile(file);
delete[] file;
}
merger.Merge() ;
return kTRUE ;
}
Bool_t AliQAManager::MergeResults(const Int_t runNumber) const
{
TString cmd ;
cmd = Form(".! ls %s*.root > tempo.txt", AliQAv1::GetQADataFileName()) ;
gROOT->ProcessLine(cmd.Data()) ;
ifstream in("tempo.txt") ;
const Int_t chunkMax = 100 ;
TString fileList[chunkMax] ;
Int_t index = 0 ;
while ( 1 ) {
TString file ;
in >> fileList[index] ;
if ( !in.good() )
break ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("index = %d file = %s", index, (fileList[index].Data()))) ;
index++ ;
}
if ( index == 0 ) {
AliError("No QA Result File found") ;
return kFALSE ;
}
TFileMerger merger ;
TString outFileName ;
if (runNumber != -1)
outFileName = Form("Merged.%s.Result.%d.root",AliQAv1::GetQADataFileName(),runNumber);
else
outFileName = Form("Merged.%s.Result.root",AliQAv1::GetQADataFileName());
merger.OutputFile(outFileName.Data()) ;
for (Int_t ifile = 0 ; ifile < index ; ifile++) {
TString file = fileList[ifile] ;
merger.AddFile(file) ;
}
merger.Merge() ;
return kTRUE ;
}
void AliQAManager::Reset(const Bool_t sameCycle)
{
for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
AliQADataMaker * qadm = GetQADataMaker(iDet);
if (qadm)
qadm->Reset();
}
}
if (fRawReaderDelete) {
delete fRawReader ;
fRawReader = NULL ;
}
fCycleSame = sameCycle ;
fESD = NULL ;
fESDTree = NULL ;
fNumberOfEvents = 999999 ;
}
void AliQAManager::ResetDetectors(AliQAv1::TASKINDEX_t task, AliQAv1::DETECTORINDEX_t det)
{
UInt_t iDet = 0 ;
UInt_t iDetMax = fgkNDetectors ;
if ( det != AliQAv1::kNULLDET ) {
iDet = det ;
iDetMax = det+1 ;
}
for (iDet = 0; iDet < iDetMax ; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet))) {
AliQADataMaker * qadm = GetQADataMaker(iDet);
qadm->ResetDetector(task);
}
}
}
AliQAManager * AliQAManager::QAManager(AliQAv1::MODE_t mode, TMap *entryCache, Int_t run)
{
if (!fgQAInstance) {
if ( (mode != AliQAv1::kSIMMODE) && (mode != AliQAv1::kRECMODE) && (mode != AliQAv1::kQAMODE) ) {
AliWarningClass("You must specify kSIMMODE or kRECMODE or kQAMODE") ;
return NULL ;
}
fgQAInstance = new AliQAManager(mode) ;
if (!entryCache)
fgQAInstance->Init();
else
fgQAInstance->InitFromCache(entryCache,run);
}
return fgQAInstance;
}
AliQAManager * AliQAManager::QAManager(AliQAv1::TASKINDEX_t task)
{
return QAManager(AliQAv1::Mode(task)) ;
}
TString AliQAManager::Run(const Char_t * detectors, AliRawReader * rawReader, const Bool_t sameCycle)
{
fCycleSame = sameCycle ;
fRawReader = rawReader ;
fDetectors = detectors ;
fDetectorsW = detectors ;
AliCDBManager* man = AliCDBManager::Instance() ;
if ( man->GetRun() == -1 ) {
rawReader->NextEvent() ;
man->SetRun(fRawReader->GetRunNumber()) ;
rawReader->RewindEvents() ;
}
if (!fCycleSame)
if ( !InitQA(AliQAv1::kRAWS) )
return "" ;
fRawReaderDelete = kFALSE ;
DoIt(AliQAv1::kRAWS) ;
return fDetectorsW ;
}
TString AliQAManager::Run(const Char_t * detectors, const Char_t * fileName, const Bool_t sameCycle)
{
fCycleSame = sameCycle ;
fDetectors = detectors ;
fDetectorsW = detectors ;
AliCDBManager* man = AliCDBManager::Instance() ;
if ( man->GetRun() == -1 ) {
AliRunLoader * rl = AliRunLoader::Open("galice.root") ;
if ( ! rl ) {
AliFatal("galice.root file not found in current directory") ;
} else {
rl->CdGAFile() ;
rl->LoadgAlice() ;
if ( ! rl->GetAliRun() ) {
AliFatal("AliRun not found in galice.root") ;
} else {
rl->LoadHeader() ;
man->SetRun(rl->GetHeader()->GetRun());
}
}
}
if (!fCycleSame)
if ( !InitQA(AliQAv1::kRAWS, fileName) )
return "" ;
DoIt(AliQAv1::kRAWS) ;
return fDetectorsW ;
}
TString AliQAManager::Run(const Char_t * detectors, const AliQAv1::TASKINDEX_t taskIndex, Bool_t const sameCycle, const Char_t * fileName )
{
fCycleSame = sameCycle ;
fDetectors = detectors ;
fDetectorsW = detectors ;
AliCDBManager* man = AliCDBManager::Instance() ;
if ( man->GetRun() == -1 ) {
AliRunLoader * rl = AliRunLoader::Open("galice.root") ;
if ( ! rl ) {
AliFatal("galice.root file not found in current directory") ;
} else {
rl->CdGAFile() ;
rl->LoadgAlice() ;
if ( ! rl->GetAliRun() ) {
AliDebug(AliQAv1::GetQADebugLevel(), "AliRun not found in galice.root") ;
} else {
rl->LoadHeader() ;
man->SetRun(rl->GetHeader()->GetRun()) ;
}
}
}
if ( taskIndex == AliQAv1::kNULLTASKINDEX) {
for (UInt_t task = 0; task < AliQAv1::kNTASKINDEX; task++) {
if ( fTasks.Contains(Form("%d", task)) ) {
if (!fCycleSame)
if ( !InitQA(AliQAv1::GetTaskIndex(AliQAv1::GetTaskName(task)), fileName) )
return "" ;
DoIt(AliQAv1::GetTaskIndex(AliQAv1::GetTaskName(task))) ;
}
}
} else {
if (! fCycleSame )
if ( !InitQA(taskIndex, fileName) )
return "" ;
DoIt(taskIndex) ;
}
return fDetectorsW ;
}
void AliQAManager::RunOneEvent(AliRawReader * rawReader)
{
if ( ! rawReader )
return ;
if (fTasks.Contains(Form("%d", AliQAv1::kRAWS))){
TString detList ;
if ( GetEventInfo())
detList = GetEventInfo()->GetTriggerCluster() ;
for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
if (!IsSelected(AliQAv1::GetDetName(iDet)) || (!detList.IsNull() && !detList.Contains(AliQAv1::GetDetName(iDet))))
continue;
AliQADataMaker *qadm = GetQADataMaker(iDet);
if (!qadm)
continue;
if ( qadm->IsCycleDone() ) {
qadm->EndOfCycle() ;
}
qadm->SetEventSpecie(fEventSpecie) ;
if ( qadm->GetRecoParam() )
if ( AliRecoParam::Convert(qadm->GetRecoParam()->GetEventSpecie()) != AliRecoParam::kDefault)
qadm->SetEventSpecie(qadm->GetRecoParam()->GetEventSpecie()) ;
qadm->Exec(AliQAv1::kRAWS, rawReader) ;
}
}
}
void AliQAManager::RunOneEvent(AliESDEvent *& esd, AliESDEvent *& hltesd)
{
if (fTasks.Contains(Form("%d", AliQAv1::kESDS))) {
TString detList ;
if ( GetEventInfo())
detList = GetEventInfo()->GetTriggerCluster() ;
for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
if (!IsSelected(AliQAv1::GetDetName(iDet)) || (!detList.IsNull() && !detList.Contains(AliQAv1::GetDetName(iDet))))
continue;
AliQADataMaker *qadm = GetQADataMaker(iDet);
if (!qadm)
continue;
qadm->SetEventSpecie(fEventSpecie) ;
if ( qadm->GetRecoParam() )
if ( AliRecoParam::Convert(qadm->GetRecoParam()->GetEventSpecie()) != AliRecoParam::kDefault)
qadm->SetEventSpecie(qadm->GetRecoParam()->GetEventSpecie()) ;
if ( qadm->IsCycleDone() ) {
qadm->EndOfCycle() ;
}
if (iDet == AliQAv1::kHLT) {
TObjArray esdarray;
esdarray.Add(esd);
esdarray.Add(hltesd);
qadm->Exec(AliQAv1::kESDS, &esdarray);
} else {
qadm->Exec(AliQAv1::kESDS, esd) ;
}
}
}
}
void AliQAManager::RunOneEventInOneDetector(Int_t det, TTree * tree)
{
TString detList ;
if ( GetEventInfo())
detList = GetEventInfo()->GetTriggerCluster() ;
if (!detList.IsNull() && !detList.Contains(AliQAv1::GetDetName(det)))
return ;
TString test(tree->GetName()) ;
if (fTasks.Contains(Form("%d", AliQAv1::kRECPOINTS))) {
if (IsSelected(AliQAv1::GetDetName(det))) {
AliQADataMaker *qadm = GetQADataMaker(det);
if (qadm) {
qadm->SetEventSpecie(fEventSpecie) ;
if ( qadm->GetRecoParam() ) {
if ( AliRecoParam::Convert(qadm->GetRecoParam()->GetEventSpecie()) != AliRecoParam::kDefault)
qadm->SetEventSpecie(qadm->GetRecoParam()->GetEventSpecie()) ;
else
AliError(Form("%d defined by %s is not an event specie", qadm->GetRecoParam()->GetEventSpecie(), qadm->GetName())) ;
}
if ( qadm->IsCycleDone() ) {
qadm->EndOfCycle() ;
}
if (test.Contains("TreeD")) {
qadm->Exec(AliQAv1::kDIGITSR, tree) ;
} else if (test.Contains("TreeR")) {
qadm->Exec(AliQAv1::kRECPOINTS, tree) ;
}
}
}
}
}
Bool_t AliQAManager::Save2OCDB(const Int_t runNumber, AliRecoParam::EventSpecie_t es, const Char_t * year, const Char_t * detectors) const
{
Bool_t rv = kTRUE ;
TString tmp(AliQAv1::GetQARefStorage()) ;
if ( tmp.IsNull() ) {
AliError("No storage defined, use AliQAv1::SetQARefStorage") ;
return kFALSE ;
}
if ( !(tmp.Contains(AliQAv1::GetLabLocalOCDB()) || tmp.Contains(AliQAv1::GetLabAliEnOCDB())) ) {
AliError(Form("%s is a wrong storage, use %s or %s", AliQAv1::GetQARefStorage(), AliQAv1::GetLabLocalOCDB().Data(), AliQAv1::GetLabAliEnOCDB().Data())) ;
return kFALSE ;
}
TString sdet(detectors) ;
sdet.ToUpper() ;
TFile * inputFile ;
if ( sdet.Contains("ALL") ) {
rv = Merge(runNumber) ;
if ( ! rv )
return kFALSE ;
TString inputFileName(Form("Merged.%s.Data.%d.root", AliQAv1::GetQADataFileName(), runNumber)) ;
inputFile = TFile::Open(inputFileName.Data()) ;
rv = SaveIt2OCDB(runNumber, inputFile, year, es) ;
} else {
for (Int_t index = 0; index < AliQAv1::kNDET; index++) {
if (sdet.Contains(AliQAv1::GetDetName(index))) {
TString inputFileName(Form("%s.%s.%d.root", AliQAv1::GetDetName(index), AliQAv1::GetQADataFileName(), runNumber)) ;
inputFile = TFile::Open(inputFileName.Data()) ;
rv *= SaveIt2OCDB(runNumber, inputFile, year, es) ;
}
}
}
return rv ;
}
Bool_t AliQAManager::SaveIt2OCDB(const Int_t runNumber, TFile * inputFile, const Char_t * year, AliRecoParam::EventSpecie_t es) const
{
Bool_t rv = kTRUE ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("Saving TH1s in %s to %s", inputFile->GetName(), AliQAv1::GetQARefStorage())) ;
if ( ! IsDefaultStorageSet() ) {
TString tmp( AliQAv1::GetQARefStorage() ) ;
if ( tmp.Contains(AliQAv1::GetLabLocalOCDB()) )
Instance()->SetDefaultStorage(AliQAv1::GetQARefStorage()) ;
else {
TString tmp1(AliQAv1::GetQARefDefaultStorage()) ;
tmp1.Append(year) ;
tmp1.Append("?user=alidaq") ;
Instance()->SetDefaultStorage(tmp1.Data()) ;
}
}
Instance()->SetSpecificStorage("*", AliQAv1::GetQARefStorage()) ;
if(GetRun() < 0)
Instance()->SetRun(runNumber);
AliCDBMetaData mdr ;
mdr.SetResponsible("yves schutz");
for ( Int_t detIndex = 0 ; detIndex < AliQAv1::kNDET ; detIndex++) {
TDirectory * detDir = inputFile->GetDirectory(AliQAv1::GetDetName(detIndex)) ;
if ( detDir ) {
AliDebug(AliQAv1::GetQADebugLevel(), Form("Entering %s", detDir->GetName())) ;
AliQAv1::SetQARefDataDirName(es) ;
TString detOCDBDir(Form("%s/%s/%s", AliQAv1::GetDetName(detIndex), AliQAv1::GetRefOCDBDirName(), AliQAv1::GetRefDataDirName())) ;
AliCDBId idr(detOCDBDir.Data(), runNumber, AliCDBRunRange::Infinity()) ;
TList * listDetQAD = new TList() ;
TString listName(Form("%s QA data Reference", AliQAv1::GetDetName(detIndex))) ;
mdr.SetComment(Form("%s QA stuff", AliQAv1::GetDetName(detIndex)));
listDetQAD->SetName(listName) ;
TList * taskList = detDir->GetListOfKeys() ;
TIter nextTask(taskList) ;
TKey * taskKey ;
while ( (taskKey = static_cast<TKey*>(nextTask())) ) {
TDirectory * taskDir = detDir->GetDirectory(taskKey->GetName()) ;
TDirectory * esDir = taskDir->GetDirectory(AliRecoParam::GetEventSpecieName(es)) ;
AliDebug(AliQAv1::GetQADebugLevel(), Form("Saving %s", esDir->GetName())) ;
TObjArray * listTaskQAD = new TObjArray(100) ;
listTaskQAD->SetName(Form("%s/%s", taskKey->GetName(), AliRecoParam::GetEventSpecieName(es))) ;
listDetQAD->Add(listTaskQAD) ;
TList * histList = esDir->GetListOfKeys() ;
TIter nextHist(histList) ;
TKey * histKey ;
while ( (histKey = static_cast<TKey*>(nextHist())) ) {
TObject * odata = esDir->Get(histKey->GetName()) ;
if ( !odata ) {
AliError(Form("%s in %s/%s returns a NULL pointer !!", histKey->GetName(), detDir->GetName(), taskDir->GetName())) ;
} else {
if ( AliQAv1::GetExpert() == histKey->GetName() ) {
TDirectory * expertDir = esDir->GetDirectory(histKey->GetName()) ;
TList * expertHistList = expertDir->GetListOfKeys() ;
TIter nextExpertHist(expertHistList) ;
TKey * expertHistKey ;
while ( (expertHistKey = static_cast<TKey*>(nextExpertHist())) ) {
TObject * expertOdata = expertDir->Get(expertHistKey->GetName()) ;
if ( !expertOdata ) {
AliError(Form("%s in %s/%s/Expert returns a NULL pointer !!", expertHistKey->GetName(), detDir->GetName(), taskDir->GetName())) ;
} else {
AliDebug(AliQAv1::GetQADebugLevel(), Form("Adding %s", expertHistKey->GetName())) ;
if ( expertOdata->IsA()->InheritsFrom("TH1") ) {
AliDebug(AliQAv1::GetQADebugLevel(), Form("Adding %s", expertHistKey->GetName())) ;
TH1 * hExpertdata = static_cast<TH1*>(expertOdata) ;
listTaskQAD->Add(hExpertdata) ;
}
}
}
}
AliDebug(AliQAv1::GetQADebugLevel(), Form("Adding %s", histKey->GetName())) ;
if ( odata->IsA()->InheritsFrom("TH1") ) {
AliDebug(AliQAv1::GetQADebugLevel(), Form("Adding %s", histKey->GetName())) ;
TH1 * hdata = static_cast<TH1*>(odata) ;
listTaskQAD->Add(hdata) ;
}
}
}
}
Instance()->Put(listDetQAD, idr, &mdr) ;
}
}
return rv ;
}
void AliQAManager::SetCheckerExternParam(AliQAv1::DETECTORINDEX_t detIndex, TList * parameterList)
{
AliQACheckerBase * qac = AliQAChecker::Instance()->GetDetQAChecker(detIndex) ;
qac->SetExternParamlist(parameterList) ;
qac->PrintExternParam() ;
}
void AliQAManager::SetEventSpecie(AliRecoParam::EventSpecie_t es)
{
fEventSpecie = es ;
AliQAv1::Instance()->SetEventSpecie(es) ;
}
void AliQAManager::SetRecoParam(const Int_t det, const AliDetectorRecoParam *par)
{
GetQADataMaker(det)->SetRecoParam(par) ;
}
void AliQAManager::SetWriteExpert()
{
for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
if (IsSelected(AliQAv1::GetDetName(iDet)))
fQAWriteExpert[iDet] = kTRUE ;
}
}
void AliQAManager::Destroy() {
if (fgQAInstance) {
delete fgQAInstance ;
fgQAInstance = NULL ;
}
}
void AliQAManager::ShowQA() {
for ( Int_t detIndex = 0 ; detIndex < AliQAv1::kNDET ; detIndex++)
if ( IsSelected(AliQAv1::GetDetName(detIndex)) )
AliQAv1::Instance(AliQAv1::GetDetIndex(AliQAv1::GetDetName(detIndex)))->Show() ;
}