8bb2306a56efce2249038015ea00c489f79da010
[u/mrichter/AliRoot.git] / STEER / AliQADataMakerRec.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notifce   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16
17 /* $Id$ */
18
19 //
20 //  Base Class
21 //  Produces the data needed to calculate the quality assurance for Reconstruction
22 //  All data must be mergeable objects.
23 //  Y. Schutz CERN July 2007
24 //
25
26 // --- ROOT system ---
27 #include <TCanvas.h>
28 #include <TFile.h>
29 #include <TTree.h>
30 #include <TNtupleD.h>
31 #include <TObjArray.h>
32
33 // --- Standard library ---
34
35 // --- AliRoot header files ---
36 #include "AliCDBPath.h"
37 #include "AliCDBEntry.h"
38 #include "AliDetectorRecoParam.h"
39 #include "AliCDBManager.h"
40
41 #include "AliLog.h"
42 #include "AliQADataMakerRec.h"
43 #include "AliESDEvent.h"
44 #include "AliRawReader.h"
45
46 ClassImp(AliQADataMakerRec)
47              
48 //____________________________________________________________________________ 
49 AliQADataMakerRec::AliQADataMakerRec(const char * name, const char * title) : 
50   AliQADataMaker(name, title), 
51   fESDsQAList(NULL), 
52   fRawsQAList(NULL), 
53   fRecPointsQAList(NULL),
54   fCorrNt(NULL), 
55   fRecoParam(NULL) 
56 {
57   // ctor
58         fDetectorDirName = GetName() ; 
59 }
60
61 //____________________________________________________________________________ 
62 AliQADataMakerRec::AliQADataMakerRec(const AliQADataMakerRec& qadm) :
63   AliQADataMaker(qadm.GetName(), qadm.GetTitle()), 
64   fESDsQAList(qadm.fESDsQAList),
65   fRawsQAList(qadm.fRawsQAList),
66   fRecPointsQAList(qadm.fRecPointsQAList),
67   fCorrNt(qadm.fCorrNt),  
68   fRecoParam(qadm.fRecoParam) 
69 {
70   //copy ctor
71         SetName(qadm.GetName()) ; 
72         SetTitle(qadm.GetTitle()) ; 
73         fDetectorDirName = GetName() ; 
74 }
75
76 //____________________________________________________________________________ 
77 AliQADataMakerRec::~AliQADataMakerRec()
78 {
79         //dtor: delete the TObjArray and thei content
80         if ( fESDsQAList ) {
81     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
82       if ( fESDsQAList[specie] ) {
83         if ( fESDsQAList[specie]->IsOwner() ) 
84           fESDsQAList[specie]->Delete() ;     
85       }
86     }
87     delete[] fESDsQAList ;
88         }
89         if ( fRawsQAList ) {
90     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
91       if ( fRawsQAList[specie] ) {
92         if ( fRawsQAList[specie]->IsOwner() ) 
93           fRawsQAList[specie]->Delete() ;
94       }
95     }
96     delete[] fRawsQAList ;
97   }
98         if ( fRecPointsQAList ) {
99     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
100       if ( fRecPointsQAList[specie] ) {
101         if ( fRecPointsQAList[specie]->IsOwner() ) 
102           fRecPointsQAList[specie]->Delete() ;
103       }
104     }
105                 delete[] fRecPointsQAList ; 
106         }
107 }
108
109 //__________________________________________________________________
110 AliQADataMakerRec& AliQADataMakerRec::operator = (const AliQADataMakerRec& qadm )
111 {
112   // Assignment operator.
113   this->~AliQADataMakerRec();
114   new(this) AliQADataMakerRec(qadm);
115   return *this;
116 }
117
118 //____________________________________________________________________________
119 void AliQADataMakerRec::EndOfCycle() 
120 {
121   // Finishes a cycle of QA for all the tasks
122   EndOfCycle(AliQAv1::kRAWS) ; 
123   EndOfCycle(AliQAv1::kRECPOINTS) ; 
124   EndOfCycle(AliQAv1::kESDS) ; 
125   ResetCycle() ; 
126 }
127
128 //____________________________________________________________________________
129 void AliQADataMakerRec::EndOfCycle(AliQAv1::TASKINDEX_t task) 
130 {
131         // Finishes a cycle of QA 
132         
133         TObjArray ** list = NULL ; 
134         
135         if ( task == AliQAv1::kRAWS )     
136                 list = fRawsQAList ; 
137         else if ( task == AliQAv1::kRECPOINTS ) 
138                 list = fRecPointsQAList ; 
139         else if ( task == AliQAv1::kESDS )
140                 list = fESDsQAList ; 
141
142  
143         if ( ! list && ! fCorrNt ) 
144     return ; 
145   //DefaultEndOfDetectorCycle(task) ;
146         EndOfDetectorCycle(task, list) ;
147         TDirectory * subDir = NULL ;
148         if (fDetectorDir) 
149                 subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ; 
150         if ( subDir ) {
151                 subDir->cd() ; 
152     for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
153       TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ;
154       if (eventSpecieDir) {
155         eventSpecieDir->cd() ;    
156         if (list[specie]) {
157           TIter next(list[specie]) ; 
158           TObject * obj ; 
159           while( (obj = next()) ) {
160             if (!obj->TestBit(AliQAv1::GetExpertBit()))
161               obj->Write() ;
162           }
163           if (WriteExpert()) {
164             TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ; 
165             if ( expertDir ) { // Write only if requested
166               expertDir->cd() ;
167               next.Reset() ; 
168               while( (obj = next()) ) {
169                 if (!obj->TestBit(AliQAv1::GetExpertBit()))
170                   continue ; 
171                 obj->Write() ;
172               }      
173             }
174           }
175         }
176         if ( !fCorrNt )
177           continue ; 
178         if (fCorrNt[specie] && AliQAv1::GetDetIndex(GetName()) == AliQAv1::kCORR) {
179           eventSpecieDir->cd() ; 
180           fCorrNt[specie]->Write() ; 
181         }
182       }
183     }
184     fOutput->Save() ; 
185         }
186   if ( AliDebugLevel()  == AliQAv1::GetQADebugLevel() ) 
187     MakeImage(task) ; 
188 }
189
190 //____________________________________________________________________________ 
191 void AliQADataMakerRec::MakeImage(AliQAv1::TASKINDEX_t task)
192 {
193   // create a drawing of detetor defined histograms
194   TObjArray ** list = NULL ;  
195   switch (task) {
196     case AliQAv1::kRAWS:
197       list = fRawsQAList ; 
198       break;
199     case AliQAv1::kHITS:
200       break;
201     case AliQAv1::kSDIGITS:
202       break;  
203     case AliQAv1::kDIGITS:
204       break;  
205     case AliQAv1::kRECPOINTS:
206       list = fRecPointsQAList ; 
207       break;
208     case AliQAv1::kTRACKSEGMENTS:
209       break;
210     case AliQAv1::kRECPARTICLES:
211       break;
212     case AliQAv1::kESDS:
213       list = fESDsQAList ; 
214       break;
215     case AliQAv1::kNTASKINDEX:
216       break;
217     default:
218       break;
219   }
220   if ( !list) {
221     AliError("data not initialized, call AliQADataMaker::Init"); 
222     return ; 
223   }
224   TIter next(list[0]) ;  
225   TH1 * hdata = NULL ; 
226   Int_t nImages = 0 ;
227   while ( (hdata=dynamic_cast<TH1 *>(next())) ) {
228     if ( hdata->TestBit(AliQAv1::GetImageBit()) )
229       nImages++; 
230   }
231   if ( nImages == 0 ) {
232     AliWarning(Form("No histogram will be plotted for %s %s\n", GetName(), AliQAv1::GetTaskName(task).Data())) ;  
233   } else {
234     AliDebug(AliQAv1::GetQADebugLevel(), Form("%d histograms will be plotted for %s %s\n", nImages, GetName(), AliQAv1::GetTaskName(task).Data())) ;  
235     Double_t w  = 1000 ;
236     Double_t h  = 1000 ;
237     for (Int_t esIndex = 0 ; esIndex < AliRecoParam::kNSpecies ; esIndex++) {
238       TCanvas * canvasQA = new TCanvas(Form("QA_%s_%s_%s", 
239                                             GetName(), 
240                                             AliQAv1::GetTaskName(task).Data(), 
241                                             AliRecoParam::GetEventSpecieName(esIndex)), 
242                                        Form("QA control plots for det=%s task=%s eventspecie=%s", 
243                                             GetName(), 
244                                             AliQAv1::GetTaskName(task).Data(), 
245                                             AliRecoParam::GetEventSpecieName(esIndex)), 
246                                        w, h) ;
247       canvasQA->SetWindowSize(w + (w - canvasQA->GetWw()), h + (h - canvasQA->GetWh())) ;
248       Int_t nx = TMath::Sqrt(nImages) ; 
249       Int_t ny = nx  ; 
250       if ( nx < TMath::Sqrt(nImages)) 
251         ny++ ; 
252       canvasQA->Divide(nx, ny) ; 
253       TIter nexthist(list[esIndex]) ; 
254       TH1* hist = NULL ;
255       Int_t npad = 1 ; 
256       canvasQA->cd(npad) ; 
257       while ( (hist=dynamic_cast<TH1*>(nexthist())) ) {
258         if(hist->TestBit(AliQAv1::GetImageBit())) {
259           hist->Draw() ; 
260           canvasQA->cd(++npad) ; 
261         }
262       }
263       canvasQA->Print() ; 
264     }
265   }
266 }
267
268 //____________________________________________________________________________
269 void AliQADataMakerRec::Exec(AliQAv1::TASKINDEX_t task, TObject * data) 
270
271   // creates the quality assurance data for the various tasks (Hits, SDigits, Digits, ESDs)
272         
273         if ( task == AliQAv1::kRAWS ) {
274                 AliDebug(AliQAv1::GetQADebugLevel(), "Processing Raws QA") ; 
275                 AliRawReader * rawReader = dynamic_cast<AliRawReader *>(data) ; 
276                 if (rawReader) 
277                         MakeRaws(rawReader) ;
278                 else
279       AliDebug(AliQAv1::GetQADebugLevel(), "Raw data are not processed") ;     
280         } else if ( task == AliQAv1::kRECPOINTS ) {
281                 AliDebug(AliQAv1::GetQADebugLevel(), "Processing RecPoints QA") ; 
282                 TTree * tree = dynamic_cast<TTree *>(data) ; 
283                 if (tree) {
284                         MakeRecPoints(tree) ; 
285                 } else {
286                         AliWarning("data are not a TTree") ; 
287                 }
288         } else if ( task == AliQAv1::kESDS ) {
289                 AliDebug(AliQAv1::GetQADebugLevel(), "Processing ESDs QA") ; 
290                 AliESDEvent * esd = dynamic_cast<AliESDEvent *>(data) ; 
291                 if (esd) 
292                         MakeESDs(esd) ;
293                 else 
294                         AliError("Wrong type of esd container") ; 
295         }  
296 }
297
298 //____________________________________________________________________________ 
299 TObjArray **  AliQADataMakerRec::Init(AliQAv1::TASKINDEX_t task, Int_t cycles)
300 {
301   // general intialisation
302   InitRecoParams() ;
303         TObjArray ** rv = NULL ; 
304   
305         if (cycles > 0)
306                 SetCycle(cycles) ;  
307         
308         if ( task == AliQAv1::kRAWS ) {
309                 if (! fRawsQAList ) { 
310       fRawsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
311       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
312         fRawsQAList[specie] = new TObjArray(100) ;       
313         fRawsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
314       }
315                         InitRaws() ;
316                 }
317                 rv = fRawsQAList ;
318         } else if ( task == AliQAv1::kRECPOINTS ) {
319                 if ( ! fRecPointsQAList ) {
320       fRecPointsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
321       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
322         fRecPointsQAList[specie] = new TObjArray(100) ; 
323         fRecPointsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ; 
324       }
325       InitRecPoints() ;
326                 }
327                 rv = fRecPointsQAList ;
328         } else if ( task == AliQAv1::kESDS ) {
329                 if ( ! fESDsQAList ) {
330       fESDsQAList = new TObjArray *[AliRecoParam::kNSpecies] ; 
331       for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
332         fESDsQAList[specie] = new TObjArray(100) ;
333         fESDsQAList[specie]->SetName(Form("%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ; 
334       }
335                         InitESDs() ;
336                 }
337                 rv = fESDsQAList ;
338         }
339         return rv ; 
340 }
341
342 //____________________________________________________________________________ 
343 void AliQADataMakerRec::Init(AliQAv1::TASKINDEX_t task, TObjArray ** list, Int_t run, Int_t cycles)
344 {
345   // Intialisation by passing the list of QA data booked elsewhere
346   
347   InitRecoParams() ;
348   fRun = run ;
349         if (cycles > 0)
350                 SetCycle(cycles) ;  
351         
352         if ( task == AliQAv1::kRAWS ) {
353                 fRawsQAList = list ;     
354         } else if ( task == AliQAv1::kRECPOINTS ) {
355                 fRecPointsQAList = list ; 
356         } else if ( task == AliQAv1::kESDS ) {
357                 fESDsQAList = list ; 
358         }
359 }
360
361 //____________________________________________________________________________
362 void AliQADataMakerRec::InitRecoParams() 
363 {
364   if (!fRecoParam) {
365     AliDebug(AliQAv1::GetQADebugLevel(), Form("Loading reconstruction parameter objects for detector %s", GetName()));
366     AliCDBPath path(GetName(),"Calib","RecoParam");
367     AliCDBEntry *entry=AliCDBManager::Instance()->Get(path.GetPath());
368     if(!entry) {
369       fRecoParam = NULL ; 
370       AliWarning(Form("Couldn't find RecoParam entry in OCDB for detector %s",GetName()));
371     }
372     else {
373       TObject * recoParamObj = entry->GetObject() ; 
374       if (dynamic_cast<TObjArray*>(recoParamObj)) {
375         // The detector has only one set of reco parameters
376         AliDebug(AliQAv1::GetQADebugLevel(), Form("Array of reconstruction parameters found for detector %s",GetName()));
377         TObjArray *recoParamArray = dynamic_cast<TObjArray*>(recoParamObj) ;
378         for (Int_t iRP=0; iRP<recoParamArray->GetEntriesFast(); iRP++) {
379           fRecoParam = dynamic_cast<AliDetectorRecoParam*>(recoParamArray->At(iRP)) ;
380           if (fRecoParam->IsDefault()) break;
381         }
382       }
383       else if (dynamic_cast<AliDetectorRecoParam*>(recoParamObj)) {
384         // The detector has only onse set of reco parameters
385         // Registering it in AliRecoParam
386         AliDebug(AliQAv1::GetQADebugLevel(), Form("Single set of reconstruction parameters found for detector %s",GetName()));
387         dynamic_cast<AliDetectorRecoParam*>(recoParamObj)->SetAsDefault();
388         fRecoParam = dynamic_cast<AliDetectorRecoParam*>(recoParamObj) ;
389       } else { 
390         AliError(Form("No valid RecoParam object found in the OCDB for detector %s",GetName()));
391       }
392     }
393     AliCDBManager::Instance()->UnloadFromCache(path.GetPath());
394   }
395 }
396
397 //____________________________________________________________________________
398 void AliQADataMakerRec::StartOfCycle(Int_t run) 
399 {
400   // Finishes a cycle of QA for all the tasks
401   Bool_t samecycle = kFALSE ; 
402   StartOfCycle(AliQAv1::kRAWS,      run, samecycle) ;
403   samecycle = kTRUE ; 
404   StartOfCycle(AliQAv1::kRECPOINTS, run, samecycle) ; 
405   StartOfCycle(AliQAv1::kESDS,      run, samecycle) ; 
406 }
407
408 //____________________________________________________________________________
409 void AliQADataMakerRec::StartOfCycle(AliQAv1::TASKINDEX_t task, Int_t run, const Bool_t sameCycle) 
410
411   // Finishes a cycle of QA data acquistion
412   if ( run > 0 ) 
413     fRun = run ; 
414         if ( !sameCycle || fCurrentCycle == -1) {
415                 ResetCycle() ;
416                 if (fOutput) 
417                         fOutput->Close() ; 
418                 fOutput = AliQAv1::GetQADataFile(GetName(), fRun) ;     
419         }       
420         AliDebug(AliQAv1::GetQADebugLevel(), Form(" Run %d Cycle %d task %s file %s", 
421                                  fRun, fCurrentCycle, AliQAv1::GetTaskName(task).Data(), fOutput->GetName() )) ;
422
423         fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ; 
424         if (!fDetectorDir)
425                 fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ; 
426
427         TDirectory * subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ; 
428         if (!subDir)
429                 subDir = fDetectorDir->mkdir(AliQAv1::GetTaskName(task)) ;  
430   
431   for ( Int_t specie = AliRecoParam::kDefault ; specie < AliRecoParam::kNSpecies ; specie++ ) {
432     TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ; 
433     if (!eventSpecieDir) 
434       eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ; 
435     TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ; 
436     if (!expertDir)
437       expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ; 
438   } 
439         StartOfDetectorCycle() ; 
440 }