correct error when storing CORR TNtuple
[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 <TFile.h>
28 #include <TTree.h>
29
30 // --- Standard library ---
31
32 // --- AliRoot header files ---
33 #include "AliLog.h"
34 #include "AliQADataMakerRec.h"
35 #include "AliESDEvent.h"
36 #include "AliRawReader.h"
37
38 ClassImp(AliQADataMakerRec)
39              
40 //____________________________________________________________________________ 
41 AliQADataMakerRec::AliQADataMakerRec(const char * name, const char * title) : 
42   AliQADataMaker(name, title), 
43   fESDsQAList(NULL), 
44   fRawsQAList(NULL), 
45   fRecPointsQAList(NULL),
46   fObject(NULL), 
47   fRecoParam(NULL) 
48 {
49   // ctor
50         fDetectorDirName = GetName() ; 
51 }
52
53 //____________________________________________________________________________ 
54 AliQADataMakerRec::AliQADataMakerRec(const AliQADataMakerRec& qadm) :
55   AliQADataMaker(qadm.GetName(), qadm.GetTitle()), 
56   fESDsQAList(qadm.fESDsQAList),
57   fRawsQAList(qadm.fRawsQAList),
58   fRecPointsQAList(qadm.fRecPointsQAList),
59   fObject(qadm.fObject),  
60   fRecoParam(qadm.fRecoParam) 
61 {
62   //copy ctor
63         SetName(qadm.GetName()) ; 
64         SetTitle(qadm.GetTitle()) ; 
65         fDetectorDirName = GetName() ; 
66 }
67
68 //____________________________________________________________________________ 
69 AliQADataMakerRec::~AliQADataMakerRec()
70 {
71         //dtor: delete the TObjArray and thei content
72         if ( fESDsQAList ) {
73                 if ( fESDsQAList->IsOwner() ) 
74                         fESDsQAList->Delete() ;     
75                 delete fESDsQAList ;     
76         }
77         if ( fRawsQAList ) {
78                 if ( fRawsQAList->IsOwner() ) 
79                         fRawsQAList->Delete() ;
80                 delete fRawsQAList ;
81         }
82         if ( fRecPointsQAList ) {
83                 if ( fRecPointsQAList->IsOwner() ) 
84                         fRecPointsQAList->Delete() ; 
85                 delete fRecPointsQAList ; 
86         }
87 }
88
89 //__________________________________________________________________
90 AliQADataMakerRec& AliQADataMakerRec::operator = (const AliQADataMakerRec& qadm )
91 {
92   // Assignment operator.
93   this->~AliQADataMakerRec();
94   new(this) AliQADataMakerRec(qadm);
95   return *this;
96 }
97
98 //____________________________________________________________________________
99 void AliQADataMakerRec::EndOfCycle() 
100 {
101   // Finishes a cycle of QA for all the tasks
102   EndOfCycle(AliQA::kRAWS) ; 
103   EndOfCycle(AliQA::kRECPOINTS) ; 
104   EndOfCycle(AliQA::kESDS) ; 
105   ResetCycle() ; 
106 }
107
108 //____________________________________________________________________________
109 void AliQADataMakerRec::EndOfCycle(AliQA::TASKINDEX_t task) 
110 {
111         // Finishes a cycle of QA 
112         
113         TObjArray * list = NULL ; 
114         
115         if ( task == AliQA::kRAWS )     
116                 list = fRawsQAList ; 
117         else if ( task == AliQA::kRECPOINTS ) 
118                 list = fRecPointsQAList ; 
119         else if ( task == AliQA::kESDS )
120                 list = fESDsQAList ; 
121
122  
123         if ( ! list && ! fObject ) 
124     return ; 
125   //DefaultEndOfDetectorCycle(task) ;
126         EndOfDetectorCycle(task, list) ;
127         TDirectory * subDir = NULL ;
128         if (fDetectorDir) 
129                 subDir = fDetectorDir->GetDirectory(AliQA::GetTaskName(task)) ; 
130         if ( subDir ) {
131                 subDir->cd() ; 
132                 if (list) {
133       TIter next(list) ; 
134       TObject * obj ; 
135       while( (obj = next()) ) {
136         if (!obj->TestBit(AliQA::GetExpertBit()))
137           obj->Write() ;
138       }
139       if (WriteExpert()) {
140         TDirectory * expertDir = subDir->GetDirectory(AliQA::GetExpert()) ; 
141         if ( expertDir ) { // Write only if requested
142           expertDir->cd() ;
143           next.Reset() ; 
144           while( (obj = next()) ) {
145             if (!obj->TestBit(AliQA::GetExpertBit()))
146               continue ; 
147             obj->Write() ;
148           }      
149         }
150       }
151     }
152     if (fObject && AliQA::GetDetIndex(GetName()) == AliQA::kCORR) {
153       subDir->cd() ; 
154       fObject->Write() ; 
155     }
156         }
157 }
158  
159 //____________________________________________________________________________
160 void AliQADataMakerRec::Exec(AliQA::TASKINDEX_t task, TObject * data) 
161
162   // creates the quality assurance data for the various tasks (Hits, SDigits, Digits, ESDs)
163         
164         if ( task == AliQA::kRAWS ) {
165                 AliDebug(1, "Processing Raws QA") ; 
166                 AliRawReader * rawReader = dynamic_cast<AliRawReader *>(data) ; 
167                 if (rawReader) 
168                         MakeRaws(rawReader) ;
169                 else
170                 AliInfo("Raw data are not processed") ;     
171         } else if ( task == AliQA::kRECPOINTS ) {
172                 AliDebug(1, "Processing RecPoints QA") ; 
173                 TTree * tree = dynamic_cast<TTree *>(data) ; 
174                 if (tree) {
175                         MakeRecPoints(tree) ; 
176                 } else {
177                         AliWarning("data are not a TTree") ; 
178                 }
179         } else if ( task == AliQA::kESDS ) {
180                 AliDebug(1, "Processing ESDs QA") ; 
181                 AliESDEvent * esd = dynamic_cast<AliESDEvent *>(data) ; 
182                 if (esd) 
183                         MakeESDs(esd) ;
184                 else 
185                         AliError("Wrong type of esd container") ; 
186         }  
187 }
188
189 //____________________________________________________________________________ 
190 TObjArray *  AliQADataMakerRec::Init(AliQA::TASKINDEX_t task, Int_t cycles)
191 {
192   // general intialisation
193         
194         TObjArray * rv = NULL ; 
195   
196         if (cycles > 0)
197                 SetCycle(cycles) ;  
198         
199         if ( task == AliQA::kRAWS ) {
200                 if (! fRawsQAList ) { 
201                         fRawsQAList = new TObjArray(100) ;       
202       fRawsQAList->SetName(Form("%s_%s", GetName(), AliQA::GetTaskName(task).Data())) ; 
203                         InitRaws() ;
204                 }
205                 rv = fRawsQAList ;
206         } else if ( task == AliQA::kRECPOINTS ) {
207                 if ( ! fRecPointsQAList ) {
208                         fRecPointsQAList = new TObjArray(100) ; 
209       fRecPointsQAList->SetName(Form("%s_%s", GetName(), AliQA::GetTaskName(task).Data())) ; 
210                         InitRecPoints() ;
211                 }
212                 rv = fRecPointsQAList ;
213         } else if ( task == AliQA::kESDS ) {
214                 if ( ! fESDsQAList ) {
215                         fESDsQAList = new TObjArray(100) ;
216       fESDsQAList->SetName(Form("%s_%s", GetName(), AliQA::GetTaskName(task).Data())) ; 
217                         InitESDs() ;
218                 }
219                 rv = fESDsQAList ;
220         }
221         
222         return rv ; 
223 }
224
225 //____________________________________________________________________________ 
226 void AliQADataMakerRec::Init(AliQA::TASKINDEX_t task, TObjArray * list, Int_t run, Int_t cycles)
227 {
228   // Intialisation by passing the list of QA data booked elsewhere
229   
230         fRun = run ;
231         if (cycles > 0)
232                 SetCycle(cycles) ;  
233         
234         if ( task == AliQA::kRAWS ) {
235                 fRawsQAList = list ;     
236         } else if ( task == AliQA::kRECPOINTS ) {
237                 fRecPointsQAList = list ; 
238         } else if ( task == AliQA::kESDS ) {
239                 fESDsQAList = list ; 
240         }
241 }
242
243 //____________________________________________________________________________
244 void AliQADataMakerRec::StartOfCycle(Int_t run) 
245 {
246   // Finishes a cycle of QA for all the tasks
247   Bool_t samecycle = kFALSE ; 
248   StartOfCycle(AliQA::kRAWS,      run, samecycle) ;
249   samecycle = kTRUE ; 
250   StartOfCycle(AliQA::kRECPOINTS, run, samecycle) ; 
251   StartOfCycle(AliQA::kESDS,      run, samecycle) ; 
252 }
253
254 //____________________________________________________________________________
255 void AliQADataMakerRec::StartOfCycle(AliQA::TASKINDEX_t task, Int_t run, const Bool_t sameCycle) 
256
257   // Finishes a cycle of QA data acquistion
258   if ( run > 0 ) 
259     fRun = run ; 
260         if ( !sameCycle || fCurrentCycle == -1) {
261                 ResetCycle() ;
262                 if (fOutput) 
263                         fOutput->Close() ; 
264                 fOutput = AliQA::GetQADataFile(GetName(), fRun) ;       
265         }       
266         AliInfo(Form(" Run %d Cycle %d task %s file %s", 
267                                  fRun, fCurrentCycle, AliQA::GetTaskName(task).Data(), fOutput->GetName() )) ;
268
269         fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ; 
270         if (!fDetectorDir)
271                 fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ; 
272
273         TDirectory * subDir = fDetectorDir->GetDirectory(AliQA::GetTaskName(task)) ; 
274         if (!subDir)
275                 subDir = fDetectorDir->mkdir(AliQA::GetTaskName(task)) ;  
276  
277   TDirectory * expertDir = subDir->GetDirectory(AliQA::GetExpert()) ; 
278   if (!expertDir)
279     expertDir = subDir->mkdir(AliQA::GetExpert()) ; 
280   
281         subDir->cd() ; 
282   
283
284         StartOfDetectorCycle() ; 
285 }