Processing SPD Mean Vertex only in PHYSICS runs.
[u/mrichter/AliRoot.git] / STEER / AliQADataMakerRec.cxx
index 73d0782..f400185 100644 (file)
@@ -24,7 +24,6 @@
 //
 
 // --- ROOT system ---
-#include <TCanvas.h>
 #include <TFile.h>
 #include <TTree.h>
 #include <TNtupleD.h>
@@ -40,6 +39,7 @@
 
 #include "AliLog.h"
 #include "AliQADataMakerRec.h"
+#include "AliQAManager.h"
 #include "AliESDEvent.h"
 #include "AliRawReader.h"
 
@@ -53,7 +53,8 @@ AliQADataMakerRec::AliQADataMakerRec(const char * name, const char * title) :
   fRawsQAList(NULL), 
   fRecPointsQAList(NULL),
   fCorrNt(NULL), 
-  fRecoParam(NULL) 
+  fRecoParam(NULL),
+  fRecPointsArray(NULL)
 {
   // ctor
        fDetectorDirName = GetName() ; 
@@ -67,7 +68,8 @@ AliQADataMakerRec::AliQADataMakerRec(const AliQADataMakerRec& qadm) :
   fRawsQAList(qadm.fRawsQAList),
   fRecPointsQAList(qadm.fRecPointsQAList),
   fCorrNt(qadm.fCorrNt),  
-  fRecoParam(qadm.fRecoParam) 
+  fRecoParam(qadm.fRecoParam),
+  fRecPointsArray(NULL)
 {
   //copy ctor
        SetName(qadm.GetName()) ; 
@@ -114,7 +116,11 @@ AliQADataMakerRec::~AliQADataMakerRec()
       }
     }
                delete[] fRecPointsQAList ; 
-       }
+  }
+  if (fRecPointsArray) {
+    fRecPointsArray->Clear() ; 
+    delete fRecPointsArray ; 
+  }
 }
 
 //__________________________________________________________________
@@ -142,7 +148,8 @@ void AliQADataMakerRec::EndOfCycle(AliQAv1::TASKINDEX_t task)
 {
        // Finishes a cycle of QA 
        
-       TObjArray ** list = NULL ; 
+    
+  TObjArray ** list = NULL ; 
        
        if ( task == AliQAv1::kRAWS )     
                list = fRawsQAList ; 
@@ -158,129 +165,52 @@ void AliQADataMakerRec::EndOfCycle(AliQAv1::TASKINDEX_t task)
     return ; 
   //DefaultEndOfDetectorCycle(task) ;
        EndOfDetectorCycle(task, list) ;
-       TDirectory * subDir = NULL ;
-       if (fDetectorDir) 
-               subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ; 
-       if ( subDir ) {
-               subDir->cd() ; 
-    for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
-      TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ;
-      if (eventSpecieDir) {
-        eventSpecieDir->cd() ;    
-        if (list[specie]) {
-          TIter next(list[specie]) ; 
-          TObject * obj ; 
+  
+  if (! AliQAManager::QAManager(AliQAv1::kRECMODE)->IsSaveData())
+    return ; 
+
+  fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ; 
+  if (!fDetectorDir)
+    fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ; 
+  TDirectory * subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ; 
+  if (!subDir)
+    subDir = fDetectorDir->mkdir(AliQAv1::GetTaskName(task)) ;  
+  subDir->cd() ; 
+  for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) { // skip Default
+    if (! AliQAv1::Instance(AliQAv1::GetDetIndex(GetName()))->IsEventSpecieSet(AliRecoParam::ConvertIndex(specie)) || AliRecoParam::ConvertIndex(specie) == AliRecoParam::kDefault) 
+      continue ; 
+    TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ;
+    if (!eventSpecieDir) 
+      eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ; 
+    eventSpecieDir->cd() ;    
+    if (list) {
+      if (list[specie]) {
+        TIter next(list[specie]) ; 
+        TObject * obj ; 
+        while( (obj = next()) ) {
+          if (!obj->TestBit(AliQAv1::GetExpertBit()))
+            obj->Write() ;
+        }
+        if (WriteExpert()) {
+          TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ; 
+          if (!expertDir)
+            expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ; 
+          expertDir->cd() ;
+          next.Reset() ; 
           while( (obj = next()) ) {
             if (!obj->TestBit(AliQAv1::GetExpertBit()))
-              obj->Write() ;
-          }
-          if (WriteExpert()) {
-            TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ; 
-            if ( expertDir ) { // Write only if requested
-              expertDir->cd() ;
-              next.Reset() ; 
-              while( (obj = next()) ) {
-                if (!obj->TestBit(AliQAv1::GetExpertBit()))
-                  continue ; 
-                obj->Write() ;
-              }      
-            }
-          }
-        }
-        if ( !fCorrNt )
-          continue ; 
-        if (fCorrNt[specie] && AliQAv1::GetDetIndex(GetName()) == AliQAv1::kCORR) {
-          eventSpecieDir->cd() ; 
-          fCorrNt[specie]->Write() ; 
+              continue ; 
+            obj->Write() ;
+          }      
         }
       }
-    }
-    fOutput->Save() ; 
-       }
-    MakeImage(task) ; 
-}
-
-//____________________________________________________________________________ 
-void AliQADataMakerRec::MakeImage(AliQAv1::TASKINDEX_t task)
-{
-  // create a drawing of detetor defined histograms
-  TObjArray ** list = NULL ;  
-  switch (task) {
-    case AliQAv1::kRAWS:
-      list = fRawsQAList ; 
-      break;
-    case AliQAv1::kHITS:
-      break;
-    case AliQAv1::kSDIGITS:
-      break;  
-    case AliQAv1::kDIGITS:
-      break;  
-    case AliQAv1::kDIGITSR:
-      list = fDigitsQAList ; 
-      break;  
-    case AliQAv1::kRECPOINTS:
-      list = fRecPointsQAList ; 
-      break;
-    case AliQAv1::kTRACKSEGMENTS:
-      break;
-    case AliQAv1::kRECPARTICLES:
-      break;
-    case AliQAv1::kESDS:
-      list = fESDsQAList ; 
-      break;
-    case AliQAv1::kNTASKINDEX:
-      break;
-    default:
-      break;
-  }
-  if ( !list) {
-    AliError("data not initialized, call AliQADataMaker::Init"); 
-    return ; 
-  }
-  TIter next(list[0]) ;  
-  TH1 * hdata = NULL ; 
-  Int_t nImages = 0 ;
-  while ( (hdata=dynamic_cast<TH1 *>(next())) ) {
-    if ( hdata->TestBit(AliQAv1::GetImageBit()) )
-      nImages++; 
-  }
-  if ( nImages == 0 ) {
-    AliWarning(Form("No histogram will be plotted for %s %s\n", GetName(), AliQAv1::GetTaskName(task).Data())) ;  
-  } else {
-    AliDebug(AliQAv1::GetQADebugLevel(), Form("%d histograms will be plotted for %s %s\n", nImages, GetName(), AliQAv1::GetTaskName(task).Data())) ;  
-    Double_t w  = 1000 ;
-    Double_t h  = 1000 ;
-    for (Int_t esIndex = 0 ; esIndex < AliRecoParam::kNSpecies ; esIndex++) {
-      if ( !fImage[esIndex] ) {
-        fImage[esIndex] = new TCanvas(Form("QA_%s_%s_%s", 
-                                           GetName(), 
-                                           AliQAv1::GetTaskName(task).Data(), 
-                                           AliRecoParam::GetEventSpecieName(esIndex)), 
-                                      Form("QA control plots for det=%s task=%s eventspecie=%s", 
-                                           GetName(), 
-                                           AliQAv1::GetTaskName(task).Data(), 
-                                           AliRecoParam::GetEventSpecieName(esIndex)), 
-                                      w, h) ;
-      }
-      fImage[esIndex]->Clear() ; 
-      fImage[esIndex]->SetWindowSize(w + (w - fImage[esIndex]->GetWw()), h + (h - fImage[esIndex]->GetWh())) ;
-      Int_t nx = TMath::Sqrt(nImages) ; 
-      Int_t ny = nx  ; 
-      if ( nx < TMath::Sqrt(nImages)) 
-        ny++ ; 
-      fImage[esIndex]->Divide(nx, ny) ; 
-      TIter nexthist(list[esIndex]) ; 
-      TH1* hist = NULL ;
-      Int_t npad = 1 ; 
-      fImage[esIndex]->cd(npad) ; 
-      while ( (hist=dynamic_cast<TH1*>(nexthist())) ) {
-        if(hist->TestBit(AliQAv1::GetImageBit())) {
-          hist->Draw() ; 
-          fImage[esIndex]->cd(++npad) ; 
-        }
+    } 
+    else if ( fCorrNt ) {
+      if (fCorrNt[specie] && AliQAv1::GetDetIndex(GetName()) == AliQAv1::kCORR) {
+        eventSpecieDir->cd() ; 
+        fCorrNt[specie]->Write() ; 
       }
-      if (fPrintImage) 
-        fImage[esIndex]->Print() ; 
+      fOutput->Save() ; 
     }
   }
 }
@@ -292,31 +222,31 @@ void AliQADataMakerRec::Exec(AliQAv1::TASKINDEX_t task, TObject * data)
        
        if ( task == AliQAv1::kRAWS ) {
                AliDebug(AliQAv1::GetQADebugLevel(), "Processing Raws QA") ; 
-               AliRawReader * rawReader = dynamic_cast<AliRawReader *>(data) ; 
+               AliRawReader * rawReader = static_cast<AliRawReader *>(data) ; 
                if (rawReader) 
                        MakeRaws(rawReader) ;
                else
       AliDebug(AliQAv1::GetQADebugLevel(), "Raw data are not processed") ;     
        } else if ( task == AliQAv1::kDIGITSR ) {
                AliDebug(AliQAv1::GetQADebugLevel(), "Processing Digits QA") ; 
-               TTree * tree = dynamic_cast<TTree *>(data) ; 
-               if (tree) {
+               TTree * tree = static_cast<TTree *>(data) ; 
+               if (strcmp(tree->ClassName(), "TTree") == 0) {
                        MakeDigits(tree) ; 
                } else {
                        AliWarning("data are not a TTree") ; 
                }
        } else if ( task == AliQAv1::kRECPOINTS ) {
                AliDebug(AliQAv1::GetQADebugLevel(), "Processing RecPoints QA") ; 
-               TTree * tree = dynamic_cast<TTree *>(data) ; 
-               if (tree) {
+               TTree * tree = static_cast<TTree *>(data) ; 
+               if (strcmp(tree->ClassName(), "TTree") == 0) {
                        MakeRecPoints(tree) ; 
                } else {
                        AliWarning("data are not a TTree") ; 
                }
        } else if ( task == AliQAv1::kESDS ) {
                AliDebug(AliQAv1::GetQADebugLevel(), "Processing ESDs QA") ; 
-               AliESDEvent * esd = dynamic_cast<AliESDEvent *>(data) ; 
-               if (esd) 
+               AliESDEvent * esd = static_cast<AliESDEvent *>(data) ; 
+               if (strcmp(esd->ClassName(), "AliESDEvent") == 0) 
                        MakeESDs(esd) ;
                else 
                        AliError("Wrong type of esd container") ; 
@@ -337,40 +267,36 @@ TObjArray **  AliQADataMakerRec::Init(AliQAv1::TASKINDEX_t task, Int_t cycles)
                if (! fRawsQAList ) { 
       fRawsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
-        fRawsQAList[specie] = new TObjArray(100) ;      
+        fRawsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;   
         fRawsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
       }
-                       InitRaws() ;
                }
                rv = fRawsQAList ;
        } else if ( task == AliQAv1::kDIGITSR ) {
                if ( ! fDigitsQAList ) {
       fDigitsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
-        fDigitsQAList[specie] = new TObjArray(100) ; 
+        fDigitsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ; 
         fDigitsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ; 
       }
-      InitDigits() ;
                }
                rv = fDigitsQAList ;
        } else if ( task == AliQAv1::kRECPOINTS ) {
                if ( ! fRecPointsQAList ) {
       fRecPointsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
-        fRecPointsQAList[specie] = new TObjArray(100) ; 
+        fRecPointsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ; 
         fRecPointsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ; 
       }
-      InitRecPoints() ;
-               }
+    }
                rv = fRecPointsQAList ;
        } else if ( task == AliQAv1::kESDS ) {
                if ( ! fESDsQAList ) {
       fESDsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
-        fESDsQAList[specie] = new TObjArray(100) ;
+        fESDsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
         fESDsQAList[specie]->SetName(Form("%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ; 
       }
-                       InitESDs() ;
                }
                rv = fESDsQAList ;
        }
@@ -407,30 +333,63 @@ void AliQADataMakerRec::InitRecoParams()
     AliCDBEntry *entry=AliCDBManager::Instance()->Get(path.GetPath());
     if(!entry) {
       fRecoParam = NULL ; 
-      AliWarning(Form("Couldn't find RecoParam entry in OCDB for detector %s",GetName()));
+      AliDebug(AliQAv1::GetQADebugLevel(), Form("Couldn't find RecoParam entry in OCDB for detector %s",GetName()));
     }
     else {
       TObject * recoParamObj = entry->GetObject() ; 
-      if (dynamic_cast<TObjArray*>(recoParamObj)) {
+      if ( strcmp(recoParamObj->ClassName(), "TObjArray") == 0 ) {
         // The detector has only one set of reco parameters
         AliDebug(AliQAv1::GetQADebugLevel(), Form("Array of reconstruction parameters found for detector %s",GetName()));
-        TObjArray *recoParamArray = dynamic_cast<TObjArray*>(recoParamObj) ;
+        TObjArray *recoParamArray = static_cast<TObjArray*>(recoParamObj) ;
         for (Int_t iRP=0; iRP<recoParamArray->GetEntriesFast(); iRP++) {
-          fRecoParam = dynamic_cast<AliDetectorRecoParam*>(recoParamArray->At(iRP)) ;
-          if (fRecoParam->IsDefault()) break;
+          fRecoParam = static_cast<AliDetectorRecoParam*>(recoParamArray->At(iRP)) ;
+          if (!fRecoParam) 
+            break ; 
+          else if (fRecoParam->IsDefault()) 
+            break ; 
         }
       }
-      else if (dynamic_cast<AliDetectorRecoParam*>(recoParamObj)) {
-        // The detector has only onse set of reco parameters
+      else if (recoParamObj->InheritsFrom("AliDetectorRecoParam")) {
+        // The detector has only one set of reco parameters
         // Registering it in AliRecoParam
         AliDebug(AliQAv1::GetQADebugLevel(), Form("Single set of reconstruction parameters found for detector %s",GetName()));
-        dynamic_cast<AliDetectorRecoParam*>(recoParamObj)->SetAsDefault();
-        fRecoParam = dynamic_cast<AliDetectorRecoParam*>(recoParamObj) ;
+        fRecoParam = static_cast<AliDetectorRecoParam*>(recoParamObj) ;
+        static_cast<AliDetectorRecoParam*>(recoParamObj)->SetAsDefault();
       } else { 
         AliError(Form("No valid RecoParam object found in the OCDB for detector %s",GetName()));
       }
     }
-    AliCDBManager::Instance()->UnloadFromCache(path.GetPath());
+  }
+}
+
+//____________________________________________________________________________ 
+void AliQADataMakerRec::ResetDetector(AliQAv1::TASKINDEX_t task)
+{
+    // default reset that resets all the QA objects.
+    // to be overloaded by detectors, if necessary
+
+  TObjArray ** list = NULL ; 
+  if ( task == AliQAv1::kRAWS ) {
+               list = fRawsQAList ;     
+       } else if ( task == AliQAv1::kDIGITSR ) {
+               list = fDigitsQAList ; 
+       } else if ( task == AliQAv1::kRECPOINTS ) {
+               list = fRecPointsQAList ; 
+       } else if ( task == AliQAv1::kESDS ) {
+               list = fESDsQAList ; 
+       }
+    //list was not initialized, skip
+  if (!list) 
+    return ; 
+  
+  for (int spec = 0; spec < AliRecoParam::kNSpecies; spec++) {
+    if (!AliQAv1::Instance()->IsEventSpecieSet(AliRecoParam::ConvertIndex(spec)))
+      continue;
+    TIter next(list[spec]) ; 
+    TH1 * histo = NULL ; 
+    while ( (histo = dynamic_cast<TH1*> (next())) ) {
+      histo->Reset() ;
+    }
   }
 }
 
@@ -456,26 +415,27 @@ void AliQADataMakerRec::StartOfCycle(AliQAv1::TASKINDEX_t task, Int_t run, const
                ResetCycle() ;
                if (fOutput) 
                        fOutput->Close() ; 
-               fOutput = AliQAv1::GetQADataFile(GetName(), fRun) ;     
+               if (AliQAManager::QAManager(AliQAv1::kRECMODE)->IsSaveData())
+      fOutput = AliQAv1::GetQADataFile(GetName(), fRun) ;      
        }       
        AliDebug(AliQAv1::GetQADebugLevel(), Form(" Run %d Cycle %d task %s file %s", 
                                 fRun, fCurrentCycle, AliQAv1::GetTaskName(task).Data(), fOutput->GetName() )) ;
 
-       fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ; 
-       if (!fDetectorDir)
-               fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ; 
-
-       TDirectory * subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ; 
-       if (!subDir)
-               subDir = fDetectorDir->mkdir(AliQAv1::GetTaskName(task)) ;  
-  
-  for ( Int_t specie = AliRecoParam::kDefault ; specie < AliRecoParam::kNSpecies ; specie++ ) {
-    TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ; 
-    if (!eventSpecieDir) 
-      eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ; 
-    TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ; 
-    if (!expertDir)
-      expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ; 
-  } 
+//     fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ; 
+//     if (!fDetectorDir)
+//             fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ; 
+//  
+//     TDirectory * subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ; 
+//     if (!subDir)
+//             subDir = fDetectorDir->mkdir(AliQAv1::GetTaskName(task)) ;  
+//  
+//  for ( Int_t specie = AliRecoParam::kDefault ; specie < AliRecoParam::kNSpecies ; specie++ ) {
+//    TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ; 
+//    if (!eventSpecieDir) 
+//      eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ; 
+//    TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ; 
+//    if (!expertDir)
+//      expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ; 
+//  } 
        StartOfDetectorCycle() ; 
 }