]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliQADataMakerSteer.cxx
Bug fix
[u/mrichter/AliRoot.git] / STEER / AliQADataMakerSteer.cxx
index 9950708086d8c63707d93dc3c88f490a01d27ea5..d2eeb5fb078cf2c375f89e104c5d991f10ec8485 100644 (file)
  **************************************************************************/
 
 /* $Id$ */
+///////////////////////////////////////////////////////////////////////////////
+//                                                                           //
+// class for running the QA makers                                           //
+//                                                                           //
+//   AliQADataMakerSteer qas;                                                //
+//   qas.Run(AliQA::kRAWS, rawROOTFileName);                                 //
+//   qas.Run(AliQA::kHITS);                                                  //
+//   qas.Run(AliQA::kSDIGITS);                                               //
+//   qas.Run(AliQA::kDIGITS);                                                //
+//   qas.Run(AliQA::kRECPOINTS);                                             //
+//   qas.Run(AliQA::kESDS);                                                  //
+//                                                                           //
+///////////////////////////////////////////////////////////////////////////////
 
 #include <TKey.h>
 #include <TFile.h>
@@ -46,19 +59,23 @@ ClassImp(AliQADataMakerSteer)
 //_____________________________________________________________________________
 AliQADataMakerSteer::AliQADataMakerSteer(const char* gAliceFilename, const char * name, const char * title) :
        TNamed(name, title), 
+       fCurrentEvent(0),  
        fCycleSame(kFALSE),
-    fDetectors("ALL"), 
+       fDetectors("ALL"), 
+       fDetectorsW("ALL"), 
        fESD(NULL), 
        fESDTree(NULL),
        fFirst(kTRUE),  
        fGAliceFileName(gAliceFilename), 
-       fRunNumber(0), 
+    fMaxEvents(0),        
        fNumberOfEvents(999999), 
+    fRunNumber(0), 
        fRawReader(NULL), 
        fRawReaderDelete(kTRUE), 
        fRunLoader(NULL)  
 {
        // default ctor
+       fMaxEvents = fNumberOfEvents ; 
        for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
                if (IsSelected(AliQA::GetDetName(iDet))) {
                        fLoader[iDet]      = NULL ;
@@ -71,14 +88,17 @@ AliQADataMakerSteer::AliQADataMakerSteer(const char* gAliceFilename, const char
 //_____________________________________________________________________________
 AliQADataMakerSteer::AliQADataMakerSteer(const AliQADataMakerSteer & qas) : 
        TNamed(qas), 
+       fCurrentEvent(qas.fCurrentEvent),  
        fCycleSame(kFALSE),
-    fDetectors(qas.fDetectors), 
+       fDetectors(qas.fDetectors), 
+       fDetectorsW(qas.fDetectorsW), 
        fESD(NULL), 
        fESDTree(NULL), 
        fFirst(qas.fFirst),  
        fGAliceFileName(qas.fGAliceFileName), 
-       fRunNumber(qas.fRunNumber), 
+    fMaxEvents(qas.fMaxEvents),        
        fNumberOfEvents(qas.fNumberOfEvents), 
+    fRunNumber(qas.fRunNumber), 
        fRawReader(NULL), 
        fRawReaderDelete(kTRUE), 
        fRunLoader(NULL)  
@@ -126,12 +146,14 @@ AliQADataMakerSteer::~AliQADataMakerSteer()
 Bool_t AliQADataMakerSteer::DoIt(const AliQA::TASKINDEX taskIndex, const char * mode)
 {
        // Runs all the QA data Maker for every detector
+
        Bool_t rv = kFALSE ;
-       
     // Fill QA data in event loop 
-       for (UInt_t iEvent = 0 ; iEvent < fNumberOfEvents ; iEvent++) {
+       for (UInt_t iEvent = 0 ; iEvent < (UInt_t)fMaxEvents ; iEvent++) {
+               fCurrentEvent++ ; 
                // Get the event
-               AliDebug(1, Form("processing event %d", iEvent));
+               if ( iEvent%10 == 0  ) 
+                       AliInfo(Form("processing event %d", iEvent));
                if ( taskIndex == AliQA::kRAWS ) {
                        if ( !fRawReader->NextEvent() )
                                break ;
@@ -214,12 +236,13 @@ Bool_t AliQADataMakerSteer::DoIt(const AliQA::TASKINDEX taskIndex, const char *
                                                case AliQA::kNTASKINDEX :
                                                break; 
                                } //task switch
-                               qadm->Increment() ; 
+                               //qadm->Increment() ; 
                        } //data maker exist
                } // detector loop
        } // event loop 
-       // Save QA data for all detectors
+//     // Save QA data for all detectors
        rv = Finish(taskIndex, mode) ;
+
        return rv ; 
 }
 
@@ -239,7 +262,7 @@ Bool_t AliQADataMakerSteer::Finish(const AliQA::TASKINDEX taskIndex, const char
 }
 
 //_____________________________________________________________________________
-TObjArray * AliQADataMakerSteer::GetFromOCDB(AliQA::DETECTORINDEX det, AliQA::TASKINDEX task) const 
+TObjArray * AliQADataMakerSteer::GetFromOCDB(AliQA::DETECTORINDEX det, AliQA::TASKINDEX task, const char * year) const 
 {
        // Retrieve the list of QA data for a given detector and a given task 
        TObjArray * rv = NULL ;
@@ -250,7 +273,10 @@ TObjArray * AliQADataMakerSteer::GetFromOCDB(AliQA::DETECTORINDEX det, AliQA::TA
        }       
        AliCDBManager* man = AliCDBManager::Instance() ; 
        if ( ! man->IsDefaultStorageSet() ) {
-               man->SetDefaultStorage(AliQA::GetQARefDefaultStorage()) ; 
+               TString tmp(AliQA::GetQARefDefaultStorage()) ; 
+               tmp.Append(year) ; 
+               tmp.Append("/") ; 
+               man->SetDefaultStorage(tmp.Data()) ;            
                man->SetSpecificStorage(Form("%s/*", AliQA::GetQAOCDBDirName()), AliQA::GetQARefStorage()) ;
        }
        char detOCDBDir[10] ; 
@@ -361,13 +387,15 @@ Bool_t AliQADataMakerSteer::Init(const AliQA::TASKINDEX taskIndex, const char *
                }
            if ( ! fRawReader ) 
                        return kFALSE ; 
-               fRawReaderDelete = kFALSE ; 
+               fRawReaderDelete = kTRUE ; 
                fRawReader->NextEvent() ; 
                fRunNumber = fRawReader->GetRunNumber() ; 
                AliCDBManager::Instance()->SetRun(fRunNumber) ; 
                fRawReader->RewindEvents();
                fNumberOfEvents = 999999 ;
-       } else if (taskIndex == AliQA::kESDS) {
+               if ( fMaxEvents < 0 ) 
+                       fMaxEvents = fNumberOfEvents ; 
+               } else if (taskIndex == AliQA::kESDS) {
                if (!gSystem->AccessPathName("AliESDs.root")) { // AliESDs.root exists
                        TFile * esdFile = TFile::Open("AliESDs.root") ;
                        fESDTree = dynamic_cast<TTree *> (esdFile->Get("esdTree")) ; 
@@ -380,6 +408,8 @@ Bool_t AliQADataMakerSteer::Init(const AliQA::TASKINDEX taskIndex, const char *
                                fESDTree->GetEntry(0) ; 
                                fRunNumber = fESD->GetRunNumber() ; 
                                fNumberOfEvents = fESDTree->GetEntries() ;
+                               if ( fMaxEvents < 0 ) 
+                                       fMaxEvents = fNumberOfEvents ; 
                        }
                } else {
                        AliError("AliESDs.root not found") ; 
@@ -390,16 +420,21 @@ Bool_t AliQADataMakerSteer::Init(const AliQA::TASKINDEX taskIndex, const char *
                        AliWarning("No Run Loader not found") ; 
                } else {
                        fNumberOfEvents = fRunLoader->GetNumberOfEvents() ;
+                       if ( fMaxEvents < 0 ) 
+                               fMaxEvents = fNumberOfEvents ; 
+
                }
        }
                // Initialize all QA data makers for all detectors
        for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
+
                if (IsSelected(AliQA::GetDetName(iDet))) {
                        AliQADataMaker * qadm = GetQADataMaker(iDet, mode) ;
                        if (!qadm) {
-                               AliWarning(Form("AliQADataMaker not found for %s", AliQA::GetDetName(iDet))) ; 
+                               AliError(Form("AliQADataMaker not found for %s", AliQA::GetDetName(iDet))) ; 
+                               fDetectorsW.ReplaceAll(AliQA::GetDetName(iDet), "") ; 
                        } else {
-                               AliInfo(Form("Data Maker found for %s", qadm->GetName())) ; 
+                               AliDebug(1, Form("Data Maker found for %s", qadm->GetName())) ; 
                                qadm->Init(taskIndex, fRunNumber, GetQACycles(iDet)) ;
                                qadm->StartOfCycle(taskIndex, fCycleSame) ;
                        }
@@ -472,6 +507,7 @@ Bool_t AliQADataMakerSteer::IsSelected(const char * det)
        
        // search for the given detector
        Bool_t rv = kFALSE;
+       //AliInfo(Form("SSSSSSSSSSSSS fd = %s det = %s ", fDetectors.Data(), det)) ; 
        if ((fDetectors.CompareTo(detName) == 0) ||
                fDetectors.BeginsWith(detName+" ") ||
                fDetectors.EndsWith(" "+detName) ||
@@ -511,6 +547,7 @@ Bool_t AliQADataMakerSteer::Merge(const Int_t runNumber) const
                in >> file[index] ; 
                if ( !in.good() ) 
                        break ; 
+               AliInfo(Form("index = %d file = %s", index, (file[index]).Data())) ; 
                index++ ;
        }
        
@@ -519,27 +556,29 @@ Bool_t AliQADataMakerSteer::Merge(const Int_t runNumber) const
                return kFALSE ; 
        }
        
-       Int_t runIndex = 1 ;  
+       Int_t runIndex    = 0 ;  
+       Int_t runIndexMax = 0 ; 
        char stmp[10] ; 
        sprintf(stmp, ".%s.", AliQA::GetQADataFileName()) ; 
-       for (Int_t ifile = 0 ; ifile < index-1 ; ifile++) {
+       for (Int_t ifile = 0 ; ifile < index ; ifile++) {
                TString tmp(file[ifile]) ; 
                tmp.ReplaceAll(".root", "") ; 
                TString det = tmp(0, tmp.Index(".")) ; 
                tmp.Remove(0, tmp.Index(stmp)+4) ; 
                TString ttmp = tmp(0, tmp.Index(".")) ; 
                Int_t newRun = ttmp.Atoi() ;
-               for (Int_t irun = 0; irun < runIndex; irun++) {
+               for (Int_t irun = 0; irun <= runIndexMax; irun++) {
                        if (newRun == run[irun]) 
                                break ; 
                        run[runIndex] = newRun ; 
                        runIndex++ ; 
                }
-               ttmp = tmp(tmp.Index("."), tmp.Length()) ; 
+               runIndexMax = runIndex ; 
+               ttmp = tmp(tmp.Index(".")+1, tmp.Length()) ; 
                Int_t cycle = ttmp.Atoi() ;  
-               AliInfo(Form("%s : det = %s run = %d cycle = %d \n", file[ifile].Data(), det.Data(), newRun, cycle)) ; 
+               AliDebug(1, Form("%s : det = %s run = %d cycle = %d \n", file[ifile].Data(), det.Data(), newRun, cycle)) ; 
        }
-       for (Int_t irun = 0 ; irun < runIndex ; irun++) {
+       for (Int_t irun = 0 ; irun < runIndexMax ; irun++) {
                TFileMerger merger ; 
                char outFileName[20] ; 
                sprintf(outFileName, "Merged.%s.%d.root", AliQA::GetQADataFileName(), run[irun]) ; 
@@ -548,8 +587,9 @@ Bool_t AliQADataMakerSteer::Merge(const Int_t runNumber) const
                        char pattern[100] ; 
                        sprintf(pattern, "%s.%d.", AliQA::GetQADataFileName(), run[irun]) ; 
                        TString tmp(file[ifile]) ; 
-                       if (tmp.Contains(pattern))
+                       if (tmp.Contains(pattern)) {
                                merger.AddFile(tmp) ; 
+                       }
                }
                merger.Merge() ; 
        }
@@ -558,14 +598,14 @@ Bool_t AliQADataMakerSteer::Merge(const Int_t runNumber) const
 }
 
 //_____________________________________________________________________________
-void AliQADataMakerSteer::Reset()
+void AliQADataMakerSteer::Reset(const Bool_t sameCycle)
 {
        // Reset the default data members
        for (UInt_t iDet = 0; iDet < fgkNDetectors; iDet++) {
                if (IsSelected(AliQA::GetDetName(iDet))) {
                        fLoader[iDet] = NULL;
                        if (fQADataMaker[iDet]) {
-                               (fQADataMaker[iDet])->Reset() ; 
+                               (fQADataMaker[iDet])->Reset(sameCycle) ; 
                                //delete fQADataMaker[iDet] ;
                                //fQADataMaker[iDet] = NULL ;
                        }
@@ -577,7 +617,7 @@ void AliQADataMakerSteer::Reset()
                fRawReader      = NULL ;
        }
 
-       fCycleSame      = kFALSE ; 
+       fCycleSame      = sameCycle ; 
        fESD            = NULL ; 
        fESDTree        = NULL ; 
        fFirst          = kTRUE ;   
@@ -585,89 +625,66 @@ void AliQADataMakerSteer::Reset()
 }
 
 //_____________________________________________________________________________
-Bool_t AliQADataMakerSteer::Run(const char * detectors, AliRawReader * rawReader) 
+TString AliQADataMakerSteer::Run(const char * detectors, AliRawReader * rawReader) 
 {
        //Runs all the QA data Maker for Raws only
+
        fRawReader       = rawReader ;          
-       fRawReaderDelete = kFALSE ; 
-       fCycleSame       = kTRUE ; 
        fDetectors       = detectors ; 
+       fDetectorsW      = detectors ; 
 
-       // Initialize all QA data makers for all detectors
-       for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
-               if (IsSelected(AliQA::GetDetName(iDet))) {
-                       AliQADataMaker * qadm = GetQADataMaker(iDet, "rec") ;
-                       if (!qadm) {
-                               AliWarning(Form("AliQADataMaker not found for %s", AliQA::GetDetName(iDet))) ; 
-                       } else {
-                               AliInfo(Form("Data Maker found for %s", qadm->GetName())) ; 
-                               qadm->Init(AliQA::kRAWS, fRunNumber, GetQACycles(iDet)) ;
-                               qadm->StartOfCycle(AliQA::kRAWS, fCycleSame) ;
-                       }
-               }
-       } 
-       fFirst = kFALSE ;
-               
-       return DoIt(AliQA::kRAWS, "rec") ; 
+       if ( !Init(AliQA::kRAWS, "rec") ) 
+               return kFALSE ; 
+       fRawReaderDelete = kFALSE ; 
+
+       DoIt(AliQA::kRAWS, "rec") ; 
+       return  fDetectorsW ;
 }
 
 //_____________________________________________________________________________
-Bool_t AliQADataMakerSteer::Run(const char * detectors, const char * fileName) 
+TString AliQADataMakerSteer::Run(const char * detectors, const char * fileName) 
 {
        //Runs all the QA data Maker for Raws only
-       fCycleSame       = kTRUE ; 
        fDetectors       = detectors ; 
+       fDetectorsW      = detectors ;  
        
        if ( !Init(AliQA::kRAWS, "rec", fileName) ) 
                return kFALSE ; 
 
-       // Initialize all QA data makers for all detectors
-       for (UInt_t iDet = 0; iDet < fgkNDetectors ; iDet++) {
-               if (IsSelected(AliQA::GetDetName(iDet))) {
-                       AliQADataMaker * qadm = GetQADataMaker(iDet, "rec") ;
-                       if (!qadm) {
-                               AliWarning(Form("AliQADataMaker not found for %s", AliQA::GetDetName(iDet))) ; 
-                       } else {
-                               AliInfo(Form("Data Maker found for %s", qadm->GetName())) ; 
-                               qadm->Init(AliQA::kRAWS, fRunNumber, GetQACycles(iDet)) ;
-                               qadm->StartOfCycle(AliQA::kRAWS, fCycleSame) ;
-                       }
-               }
-       } 
-       fFirst = kFALSE ;
-       
-       return DoIt(AliQA::kRAWS, "rec") ; 
+       DoIt(AliQA::kRAWS, "rec") ; 
+       return  fDetectorsW ;
 }
 
 //_____________________________________________________________________________
-Bool_t AliQADataMakerSteer::Run(const char * detectors, const AliQA::TASKINDEX taskIndex, const  char * fileName )
+TString AliQADataMakerSteer::Run(const char * detectors, const AliQA::TASKINDEX taskIndex, const  char * fileName )
 {
        // Runs all the QA data Maker for every detector
 
-       Bool_t rv  = kFALSE ;
-       fDetectors = detectors ; 
+       fDetectors  = detectors ; 
+       fDetectorsW = detectors ; 
 
-       char * mode ; 
+       TString mode ; 
        if ( (taskIndex == AliQA::kHITS) || (taskIndex == AliQA::kSDIGITS) || (taskIndex == AliQA::kDIGITS) ) 
                mode = "sim" ; 
        else if ( (taskIndex == AliQA::kRAWS) || (taskIndex == AliQA::kRECPOINTS) || (taskIndex == AliQA::kESDS) )
                mode = "rec" ; 
        else {
                AliError(Form("%s not implemented", AliQA::GetTaskName(taskIndex).Data())) ; 
-               return rv ;
+               return "" ;
        }
-       
-       if ( !Init(taskIndex, mode, fileName) ) 
+
+       if ( !Init(taskIndex, mode.Data(), fileName) ) 
                return kFALSE ; 
+
+       DoIt(taskIndex, mode.Data()) ;
        
-       rv = DoIt(taskIndex, mode) ;
-       
-       return rv ;   
+       return fDetectorsW ;
 
 }
 
 //_____________________________________________________________________________
-Bool_t AliQADataMakerSteer::Save2OCDB(const Int_t runNumber, const Int_t cycleNumber, const char * detectors) const
+Bool_t AliQADataMakerSteer::Save2OCDB(const Int_t runNumber, const char * year, const Int_t cycleNumber, const char * detectors) const
 {
        // take the locasl QA data merge into a single file and save in OCDB 
        Bool_t rv = kTRUE ; 
@@ -690,14 +707,14 @@ Bool_t AliQADataMakerSteer::Save2OCDB(const Int_t runNumber, const Int_t cycleNu
                char inputFileName[20] ; 
                sprintf(inputFileName, "Merged.%s.%d.root", AliQA::GetQADataFileName(), runNumber) ; 
                inputFile = TFile::Open(inputFileName) ; 
-               rv = SaveIt2OCDB(runNumber, inputFile) ; 
+               rv = SaveIt2OCDB(runNumber, inputFile, year) ; 
        } else {
                for (Int_t index = 0; index < AliQA::kNDET; index++) {
                        if (sdet.Contains(AliQA::GetDetName(index))) {
                                char inputFileName[20] ; 
                                sprintf(inputFileName, "%s.%s.%d.%d.root", AliQA::GetDetName(index), AliQA::GetQADataFileName(), runNumber, cycleNumber) ; 
                                inputFile = TFile::Open(inputFileName) ;                        
-                               rv *= SaveIt2OCDB(runNumber, inputFile) ; 
+                               rv *= SaveIt2OCDB(runNumber, inputFile, year) ; 
                        }
                }
        }
@@ -705,29 +722,36 @@ Bool_t AliQADataMakerSteer::Save2OCDB(const Int_t runNumber, const Int_t cycleNu
 }
 
 //_____________________________________________________________________________
-Bool_t AliQADataMakerSteer::SaveIt2OCDB(const Int_t runNumber, TFile * inputFile) const
+Bool_t AliQADataMakerSteer::SaveIt2OCDB(const Int_t runNumber, TFile * inputFile, const char * year) const
 {
        // reads the TH1 from file and adds it to appropriate list before saving to OCDB
        Bool_t rv = kTRUE ;
        AliInfo(Form("Saving TH1s in %s to %s", inputFile->GetName(), AliQA::GetQARefStorage())) ; 
        AliCDBManager* man = AliCDBManager::Instance() ; 
        if ( ! man->IsDefaultStorageSet() ) {
-               man->SetDefaultStorage(AliQA::GetQARefDefaultStorage()) ; 
-               man->SetSpecificStorage(Form("%s/*", AliQA::GetQAOCDBDirName()), AliQA::GetQARefStorage()) ; 
+               TString tmp(AliQA::GetQARefDefaultStorage()) ; 
+               tmp.Append(year) ; 
+               tmp.Append("?user=alidaq") ; 
+               man->SetDefaultStorage(tmp.Data()) ; 
+               man->SetSpecificStorage("*", AliQA::GetQARefStorage()) ; 
        }
        if(man->GetRun() < 0) 
                man->SetRun(runNumber);
 
+       AliCDBMetaData mdr ;
+       mdr.SetResponsible("yves schutz");
+
        for ( Int_t detIndex = 0 ; detIndex < AliQA::kNDET ; detIndex++) {
                TDirectory * detDir = inputFile->GetDirectory(AliQA::GetDetName(detIndex)) ; 
                if ( detDir ) {
                        AliInfo(Form("Entering %s", detDir->GetName())) ;
                        char detOCDBDir[20] ;
-                       sprintf(detOCDBDir, "%s/%s/%s", AliQA::GetQAOCDBDirName(), AliQA::GetDetName(detIndex), AliQA::GetRefOCDBDirName()) ; 
-                       AliCDBId idr(detOCDBDir, runNumber, 999999999)  ;
+                       sprintf(detOCDBDir, "%s/%s/%s", AliQA::GetDetName(detIndex), AliQA::GetRefOCDBDirName(), AliQA::GetRefDataDirName()) ; 
+                       AliCDBId idr(detOCDBDir, runNumber, AliCDBRunRange::Infinity())  ;
                        TList * listDetQAD = new TList() ;
                        char listName[20] ; 
                        sprintf(listName, "%s QA data Reference", AliQA::GetDetName(detIndex)) ; 
+                       mdr.SetComment("HMPID QA stuff");
                        listDetQAD->SetName(listName) ; 
                        TList * taskList = detDir->GetListOfKeys() ; 
                        TIter nextTask(taskList) ; 
@@ -750,9 +774,9 @@ Bool_t AliQADataMakerSteer::SaveIt2OCDB(const Int_t runNumber, TFile * inputFile
                                        }
                                }
                        }
-                       AliCDBMetaData mdr ;
                        man->Put(listDetQAD, idr, &mdr) ;
                }
        }
        return rv ; 
 }      
+