]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliQADataMakerSim.cxx
MakeImage is now a method of AliCheckerBase, was AliQADataMaker before. This will...
[u/mrichter/AliRoot.git] / STEER / AliQADataMakerSim.cxx
index 8b0405dceeb272d5535012ea444b16230ec3c93e..ec5e313562faaaae5886b28136bf0495acd1295c 100644 (file)
 
 /* $Id$ */
 
-/*
-  Base Class
-  Produces the data needed to calculate the quality assurance. 
-  All data must be mergeable objects.
-  Y. Schutz CERN July 2007
-*/
+//
+//  Base Class
+//  Produces the data needed to calculate the quality assurance. 
+//  All data must be mergeable objects.
+//  Y. Schutz CERN July 2007
+//
 
 // --- ROOT system ---
-#include <TSystem.h> 
 #include <TFile.h>
-#include <TList.h> 
 #include <TTree.h>
 #include <TClonesArray.h>
 
 // --- AliRoot header files ---
 #include "AliLog.h"
 #include "AliQADataMakerSim.h"
-#include "AliQAChecker.h"
 
 ClassImp(AliQADataMakerSim)
              
 //____________________________________________________________________________ 
 AliQADataMakerSim::AliQADataMakerSim(const char * name, const char * title) : 
   AliQADataMaker(name, title), 
-  fDigitsQAList(0x0), 
-  fHitsQAList(0x0),
-  fSDigitsQAList(0x0)
+  fDigitsQAList(NULL), 
+  fHitsQAList(NULL),
+  fSDigitsQAList(NULL),  
+  fHitsArray(NULL),
+  fSDigitsArray(NULL)
 {
        // ctor
        fDetectorDirName = GetName() ; 
@@ -55,12 +54,49 @@ AliQADataMakerSim::AliQADataMakerSim(const AliQADataMakerSim& qadm) :
   AliQADataMaker(qadm.GetName(), qadm.GetTitle()), 
   fDigitsQAList(qadm.fDigitsQAList),
   fHitsQAList(qadm.fHitsQAList),
-  fSDigitsQAList(qadm.fSDigitsQAList) 
+  fSDigitsQAList(qadm.fSDigitsQAList),  
+  fHitsArray(NULL),
+  fSDigitsArray(NULL)
 {
   //copy ctor
   fDetectorDirName = GetName() ; 
 }
 
+//____________________________________________________________________________ 
+AliQADataMakerSim::~AliQADataMakerSim()
+{
+       //dtor: delete the TObjArray and thei content
+       if ( fDigitsQAList ) { 
+    for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
+      if ( fDigitsQAList[specie]->IsOwner() )
+                       fDigitsQAList[specie]->Delete() ;
+    }
+               delete[] fDigitsQAList ;
+  }
+       if ( fHitsQAList ) {
+    for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
+      if ( fHitsQAList[specie]->IsOwner() ) 
+                       fHitsQAList[specie]->Delete() ;
+    }
+       delete[] fHitsQAList ;
+  }
+       if ( fSDigitsQAList ) { 
+    for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
+      if ( fSDigitsQAList[specie]->IsOwner() ) 
+                       fSDigitsQAList[specie]->Delete() ; 
+    }
+               delete[] fSDigitsQAList ;
+  }
+  if (fHitsArray) {
+    fHitsArray->Clear() ; 
+    delete fHitsArray ;
+  }
+  if (fSDigitsArray) {
+    fSDigitsArray->Clear() ; 
+    delete fSDigitsArray ;
+  }  
+}
+
 //__________________________________________________________________
 AliQADataMakerSim& AliQADataMakerSim::operator = (const AliQADataMakerSim& qadm )
 {
@@ -71,101 +107,150 @@ AliQADataMakerSim& AliQADataMakerSim::operator = (const AliQADataMakerSim& qadm
 }
 
 //____________________________________________________________________________
-void AliQADataMakerSim::EndOfCycle(AliQA::TASKINDEX task) 
+void AliQADataMakerSim::EndOfCycle() 
+{ 
+  // Finishes a cycle of QA for all tasks
+  EndOfCycle(AliQAv1::kHITS) ; 
+  EndOfCycle(AliQAv1::kSDIGITS) ; 
+  EndOfCycle(AliQAv1::kDIGITS) ;
+  ResetCycle() ; 
+}
+
+//____________________________________________________________________________
+void AliQADataMakerSim::EndOfCycle(AliQAv1::TASKINDEX_t task) 
 { 
   // Finishes a cycle of QA data acquistion
-       TObjArray * list = 0x0 ; 
+       TObjArray ** list = NULL ; 
        
-       if ( task == AliQA::kHITS ) 
+       if ( task == AliQAv1::kHITS ) 
                list = fHitsQAList ; 
-       else if ( task == AliQA::kSDIGITS )
+       else if ( task == AliQAv1::kSDIGITS )
                list = fSDigitsQAList ; 
-       else if ( task == AliQA::kDIGITS ) 
+       else if ( task == AliQAv1::kDIGITS ) 
                list = fDigitsQAList ; 
-       
+  
+  if ( ! list ) 
+    return ; 
        EndOfDetectorCycle(task, list) ; 
-       TDirectory * subDir = fDetectorDir->GetDirectory(AliQA::GetTaskName(task)) ; 
-       if (subDir) { 
-               subDir->cd() ; 
-               list->Write() ; 
-       }
+  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++) {
+    if (! AliQAv1::Instance(AliQAv1::GetDetIndex(GetName()))->IsEventSpecieSet(AliRecoParam::ConvertIndex(specie)) ) 
+      continue ;
+    TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ;
+    if (!eventSpecieDir) 
+      eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ; 
+    eventSpecieDir->cd() ; 
+    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()))
+          continue ; 
+        obj->Write() ;
+      }      
+    }
+    fOutput->Save() ; 
+  }
 }
+
 //____________________________________________________________________________
-void AliQADataMakerSim::Exec(AliQA::TASKINDEX task, TObject * data) 
+void AliQADataMakerSim::Exec(AliQAv1::TASKINDEX_t task, TObject * data) 
 { 
   // creates the quality assurance data for the various tasks (Hits, SDigits, Digits, ESDs)
-    
-       if ( task == AliQA::kHITS ) {  
-               AliDebug(1, "Processing Hits QA") ; 
-               TClonesArray * arr = dynamic_cast<TClonesArray *>(data) ; 
-               if (arr) { 
-                       MakeHits(arr) ;
-               } else {
-                       TTree * tree = dynamic_cast<TTree *>(data) ; 
-                       if (tree) {
-                               MakeHits(tree) ; 
-                       } else {
-                               AliWarning("data are neither a TClonesArray nor a TTree") ; 
-                       }
-               }
-       } else if ( task == AliQA::kSDIGITS ) {
-               AliDebug(1, "Processing SDigits QA") ; 
-               TClonesArray * arr = dynamic_cast<TClonesArray *>(data) ; 
-               if (arr) { 
-                       MakeSDigits(arr) ;
-               } else {
-                       TTree * tree = dynamic_cast<TTree *>(data) ; 
-                       if (tree) {
-                               MakeSDigits(tree) ; 
-                       } else {
-                               AliWarning("data are neither a TClonesArray nor a TTree") ; 
-                       }
-               }
-       } else if ( task == AliQA::kDIGITS ) {
-               AliDebug(1, "Processing Digits QA") ; 
-               TClonesArray * arr = dynamic_cast<TClonesArray *>(data) ; 
-               if (arr) { 
-                       MakeDigits(arr) ;
-               } else {
-                       TTree * tree = dynamic_cast<TTree *>(data) ; 
-                       if (tree) {
-                               MakeDigits(tree) ; 
-                       } else {
-                               AliWarning("data are neither a TClonesArray nor a TTree") ; 
-                       }
-               }
-       }
+  
+       if ( task == AliQAv1::kHITS ) {  
+               AliDebug(AliQAv1::GetQADebugLevel(), "Processing Hits QA") ; 
+               if (strcmp(data->ClassName(), "TClonesArray") == 0) { 
+      fHitsArray = static_cast<TClonesArray *>(data) ; 
+                       MakeHits() ;
+               } else if (strcmp(data->ClassName(), "TTree") == 0) {
+                       TTree * tree = static_cast<TTree *>(data) ; 
+      MakeHits(tree) ; 
+    } else {
+      AliWarning("data are neither a TClonesArray nor a TTree") ; 
+    }
+       } else if ( task == AliQAv1::kSDIGITS ) {
+               AliDebug(AliQAv1::GetQADebugLevel(), "Processing SDigits QA") ; 
+               if (strcmp(data->ClassName(), "TClonesArray") == 0) { 
+      fSDigitsArray = static_cast<TClonesArray *>(data) ; 
+                       MakeSDigits() ;
+               } else if (strcmp(data->ClassName(), "TTree") == 0) {
+                       TTree * tree = static_cast<TTree *>(data) ; 
+      MakeSDigits(tree) ; 
+    } else {
+      AliWarning("data are neither a TClonesArray nor a TTree") ; 
+    }
+       } else if ( task == AliQAv1::kDIGITS ) {
+               AliDebug(AliQAv1::GetQADebugLevel(), "Processing Digits QA") ; 
+               if (strcmp(data->ClassName(), "TClonesArray") == 0) { 
+      fDigitsArray = static_cast<TClonesArray *>(data) ; 
+                       MakeDigits() ;
+               } else if (strcmp(data->ClassName(), "TTree") == 0)  {
+                       TTree * tree = static_cast<TTree *>(data) ; 
+      MakeDigits(tree) ; 
+    } else {
+      AliWarning("data are neither a TClonesArray nor a TTree") ; 
+    }
+  }
 }
 
 //____________________________________________________________________________ 
-TObjArray *  AliQADataMakerSim::Init(AliQA::TASKINDEX task, Int_t run, Int_t cycles)
+TObjArray **  AliQADataMakerSim::Init(AliQAv1::TASKINDEX_t task, Int_t cycles)
 {
   // general intialisation
        
-       fRun = run ;
        if (cycles > 0)
                SetCycle(cycles) ;  
-       TObjArray * rv = NULL ; 
-       if ( task == AliQA::kHITS ) {
-               fHitsQAList = new TObjArray(100) ;       
-               InitHits() ;
+       TObjArray ** rv = NULL ; 
+       if ( task == AliQAv1::kHITS ) {
+               if ( ! fHitsQAList ) {
+      fHitsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
+      for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
+        fHitsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;   
+        fHitsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
+      }
+               }
                rv = fHitsQAList ;
-       } else if ( task == AliQA::kSDIGITS ) {
-               fSDigitsQAList = new TObjArray(100) ; 
-               InitSDigits() ;
+       } else if ( task == AliQAv1::kSDIGITS ) {
+               if ( ! fSDigitsQAList ) {
+      fSDigitsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
+      for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
+        fSDigitsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ; 
+        fSDigitsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ; 
+      }
+               }
                rv = fSDigitsQAList ;
-   } else if ( task == AliQA::kDIGITS ) {
-          fDigitsQAList = new TObjArray(100); 
-          InitDigits() ;
+   } else if ( task == AliQAv1::kDIGITS ) {
+          if ( ! fDigitsQAList ) {
+       fDigitsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
+       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {    
+         fDigitsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
+         fDigitsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
+       }
+          }
           rv =  fDigitsQAList ;
    }
   
        return rv ; 
-}
+} 
 
 //____________________________________________________________________________ 
-void AliQADataMakerSim::Init(AliQA::TASKINDEX task, TObjArray * list, Int_t run, Int_t cycles)
+void AliQADataMakerSim::Init(AliQAv1::TASKINDEX_t task, TObjArray ** list, Int_t run, Int_t cycles)
 {
   // Intialisation by passing the list of QA data booked elsewhere
   
@@ -173,51 +258,57 @@ void AliQADataMakerSim::Init(AliQA::TASKINDEX task, TObjArray * list, Int_t run,
        if (cycles > 0)
                SetCycle(cycles) ;  
        
-       if ( task == AliQA::kHITS ) {
+       if ( task == AliQAv1::kHITS ) {
                fHitsQAList = list ;     
-       } else if ( task == AliQA::kSDIGITS) {
+       } else if ( task == AliQAv1::kSDIGITS) {
                fSDigitsQAList = list ; 
-       } else if ( task == AliQA::kDIGITS ) {
+       } else if ( task == AliQAv1::kDIGITS ) {
                fDigitsQAList = list ; 
        } 
 }
 
 //____________________________________________________________________________
-void AliQADataMakerSim::StartOfCycle(AliQA::TASKINDEX task, const Bool_t sameCycle) 
+void AliQADataMakerSim::StartOfCycle(Int_t run) 
+{ 
+  // Finishes a cycle of QA for all tasks
+  Bool_t samecycle = kFALSE ; 
+  StartOfCycle(AliQAv1::kHITS,    run, samecycle) ;
+  samecycle = kTRUE ; 
+  StartOfCycle(AliQAv1::kSDIGITS, run, samecycle) ;
+  StartOfCycle(AliQAv1::kDIGITS,  run, samecycle) ;
+}
+
+//____________________________________________________________________________
+void AliQADataMakerSim::StartOfCycle(AliQAv1::TASKINDEX_t task, Int_t run, const Bool_t sameCycle) 
 { 
   // Finishes a cycle of QA data acquistion
+  if ( run > 0 ) 
+    fRun = run ; 
        if ( !sameCycle || fCurrentCycle == -1) {
                ResetCycle() ;
        if (fOutput) 
                fOutput->Close() ; 
-       fOutput = AliQA::GetQADataFile(GetName(), fRun, fCurrentCycle) ;        
+       fOutput = AliQAv1::GetQADataFile(GetName(), fRun) ;     
        }       
 
-       AliInfo(Form(" Run %d Cycle %d task %s file %s", 
-                                fRun, fCurrentCycle, AliQA::GetTaskName(task).Data(), fOutput->GetName() )) ;
-
-       fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ; 
-       if (!fDetectorDir)
-               fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ; 
-
-       TDirectory * subDir = fDetectorDir->GetDirectory(AliQA::GetTaskName(task)) ; 
-       if (!subDir)
-               subDir = fDetectorDir->mkdir(AliQA::GetTaskName(task)) ;  
-       subDir->cd() ; 
-       
-       TObjArray * list = 0x0 ; 
-  
-       if ( task == AliQA::kHITS )  
-               list = fHitsQAList ;
-       else if ( task == AliQA::kSDIGITS )  
-               list = fSDigitsQAList ;
-       else  if ( task == AliQA::kDIGITS ) 
-               list = fDigitsQAList ;
-
-       TIter next(list) ;
-       TH1 * h ; 
-       while ( (h = dynamic_cast<TH1 *>(next())) )
-               h->Reset() ;  
+       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 index = AliRecoParam::kDefault ; index < AliRecoParam::kNSpecies ; index++ ) {
+//    TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(index)) ; 
+//    if (!eventSpecieDir) 
+//      eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(index)) ; 
+//    TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ; 
+//    if (!expertDir) 
+//      expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ; 
+//   }   
        StartOfDetectorCycle() ; 
 }