Corrected protection.
[u/mrichter/AliRoot.git] / STEER / AliQADataMakerRec.cxx
CommitLineData
04236e67 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 *
940d8e5f 10 * copies and that both the copyright notice and this permission notifce *
04236e67 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
202374b1 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//
04236e67 25
26// --- ROOT system ---
04236e67 27#include <TFile.h>
04236e67 28#include <TTree.h>
57acd2d2 29#include <TNtupleD.h>
fb6e511e 30#include <TObjArray.h>
04236e67 31
32// --- Standard library ---
33
34// --- AliRoot header files ---
fb6e511e 35#include "AliCDBPath.h"
36#include "AliCDBEntry.h"
37#include "AliDetectorRecoParam.h"
38#include "AliCDBManager.h"
39
04236e67 40#include "AliLog.h"
41#include "AliQADataMakerRec.h"
50dee02c 42#include "AliQAManager.h"
04236e67 43#include "AliESDEvent.h"
44#include "AliRawReader.h"
45
46ClassImp(AliQADataMakerRec)
47
48//____________________________________________________________________________
49AliQADataMakerRec::AliQADataMakerRec(const char * name, const char * title) :
50 AliQADataMaker(name, title),
71f27f1f 51 fDigitsQAList(NULL),
a2b64fbd 52 fESDsQAList(NULL),
53 fRawsQAList(NULL),
54 fRecPointsQAList(NULL),
57acd2d2 55 fCorrNt(NULL),
6252ceeb 56 fRecoParam(NULL),
57 fRecPointsArray(NULL)
04236e67 58{
59 // ctor
60 fDetectorDirName = GetName() ;
61}
62
63//____________________________________________________________________________
64AliQADataMakerRec::AliQADataMakerRec(const AliQADataMakerRec& qadm) :
b8bd1ab8 65 AliQADataMaker(qadm.GetName(), qadm.GetTitle()),
71f27f1f 66 fDigitsQAList(qadm.fDigitsQAList),
b8bd1ab8 67 fESDsQAList(qadm.fESDsQAList),
68 fRawsQAList(qadm.fRawsQAList),
69 fRecPointsQAList(qadm.fRecPointsQAList),
57acd2d2 70 fCorrNt(qadm.fCorrNt),
6252ceeb 71 fRecoParam(qadm.fRecoParam),
72 fRecPointsArray(NULL)
04236e67 73{
74 //copy ctor
75 SetName(qadm.GetName()) ;
76 SetTitle(qadm.GetTitle()) ;
77 fDetectorDirName = GetName() ;
78}
79
63c6f8ae 80//____________________________________________________________________________
81AliQADataMakerRec::~AliQADataMakerRec()
82{
83 //dtor: delete the TObjArray and thei content
7ff8385d 84 if ( fESDsQAList ) {
57acd2d2 85 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
86 if ( fESDsQAList[specie] ) {
57acd2d2 87 fESDsQAList[specie]->Delete() ;
88 }
89 }
90 delete[] fESDsQAList ;
7ff8385d 91 }
92 if ( fRawsQAList ) {
57acd2d2 93 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
94 if ( fRawsQAList[specie] ) {
2f9129dd 95 fRawsQAList[specie]->Delete() ;
57acd2d2 96 }
97 }
98 delete[] fRawsQAList ;
99 }
44ed7a66 100 if ( fDigitsQAList ) {
101 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
102 if ( fDigitsQAList[specie] ) {
2f9129dd 103 fDigitsQAList[specie]->Delete() ;
44ed7a66 104 }
105 }
106 delete[] fDigitsQAList ;
107 }
7ff8385d 108 if ( fRecPointsQAList ) {
57acd2d2 109 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
110 if ( fRecPointsQAList[specie] ) {
2f9129dd 111 fRecPointsQAList[specie]->Delete() ;
57acd2d2 112 }
113 }
114 delete[] fRecPointsQAList ;
eca4fa66 115 }
6252ceeb 116 if (fRecPointsArray) {
117 fRecPointsArray->Clear() ;
118 delete fRecPointsArray ;
119 }
63c6f8ae 120}
121
04236e67 122//__________________________________________________________________
123AliQADataMakerRec& AliQADataMakerRec::operator = (const AliQADataMakerRec& qadm )
124{
125 // Assignment operator.
126 this->~AliQADataMakerRec();
127 new(this) AliQADataMakerRec(qadm);
128 return *this;
129}
130
131//____________________________________________________________________________
930e6e3e 132void AliQADataMakerRec::EndOfCycle()
133{
134 // Finishes a cycle of QA for all the tasks
4e25ac79 135 EndOfCycle(AliQAv1::kRAWS) ;
44ed7a66 136 EndOfCycle(AliQAv1::kDIGITSR) ;
4e25ac79 137 EndOfCycle(AliQAv1::kRECPOINTS) ;
138 EndOfCycle(AliQAv1::kESDS) ;
930e6e3e 139 ResetCycle() ;
140}
141
142//____________________________________________________________________________
4e25ac79 143void AliQADataMakerRec::EndOfCycle(AliQAv1::TASKINDEX_t task)
04236e67 144{
930e6e3e 145 // Finishes a cycle of QA
04236e67 146
ea5df941 147
148 TObjArray ** list = NULL ;
04236e67 149
4e25ac79 150 if ( task == AliQAv1::kRAWS )
04236e67 151 list = fRawsQAList ;
44ed7a66 152 else if ( task == AliQAv1::kDIGITSR )
153 list = fDigitsQAList ;
4e25ac79 154 else if ( task == AliQAv1::kRECPOINTS )
04236e67 155 list = fRecPointsQAList ;
4e25ac79 156 else if ( task == AliQAv1::kESDS )
04236e67 157 list = fESDsQAList ;
7c002d48 158
930e6e3e 159
57acd2d2 160 if ( ! list && ! fCorrNt )
930e6e3e 161 return ;
162 //DefaultEndOfDetectorCycle(task) ;
0774e4a4 163 EndOfDetectorCycle(task, list) ;
50dee02c 164
ea5df941 165 if (! AliQAManager::QAManager(AliQAv1::kRECMODE)->IsSaveData())
166 return ;
167
eca4fa66 168 fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ;
169 if (!fDetectorDir)
170 fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ;
171 TDirectory * subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ;
172 if (!subDir)
173 subDir = fDetectorDir->mkdir(AliQAv1::GetTaskName(task)) ;
174 subDir->cd() ;
6252ceeb 175 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) { // skip Default
176 if (! AliQAv1::Instance(AliQAv1::GetDetIndex(GetName()))->IsEventSpecieSet(AliRecoParam::ConvertIndex(specie)) || AliRecoParam::ConvertIndex(specie) == AliRecoParam::kDefault)
eca4fa66 177 continue ;
7822bb76 178 TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ;
179 if (!eventSpecieDir)
180 eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ;
181 eventSpecieDir->cd() ;
af65f4c8 182 if (list) {
183 if (list[specie]) {
184 TIter next(list[specie]) ;
185 TObject * obj ;
eca4fa66 186 while( (obj = next()) ) {
187 if (!obj->TestBit(AliQAv1::GetExpertBit()))
af65f4c8 188 obj->Write() ;
189 }
190 if (WriteExpert()) {
191 TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ;
192 if (!expertDir)
193 expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ;
194 expertDir->cd() ;
195 next.Reset() ;
196 while( (obj = next()) ) {
197 if (!obj->TestBit(AliQAv1::GetExpertBit()))
198 continue ;
199 obj->Write() ;
200 }
201 }
eca4fa66 202 }
af65f4c8 203 }
204 else if ( fCorrNt ) {
205 if (fCorrNt[specie] && AliQAv1::GetDetIndex(GetName()) == AliQAv1::kCORR) {
206 eventSpecieDir->cd() ;
207 fCorrNt[specie]->Write() ;
208 }
209 fOutput->Save() ;
7822bb76 210 }
eca4fa66 211 }
7d297381 212}
213
04236e67 214//____________________________________________________________________________
4e25ac79 215void AliQADataMakerRec::Exec(AliQAv1::TASKINDEX_t task, TObject * data)
04236e67 216{
217 // creates the quality assurance data for the various tasks (Hits, SDigits, Digits, ESDs)
218
4e25ac79 219 if ( task == AliQAv1::kRAWS ) {
5379c4a3 220 AliDebug(AliQAv1::GetQADebugLevel(), "Processing Raws QA") ;
eca4fa66 221 AliRawReader * rawReader = static_cast<AliRawReader *>(data) ;
04236e67 222 if (rawReader)
223 MakeRaws(rawReader) ;
224 else
5379c4a3 225 AliDebug(AliQAv1::GetQADebugLevel(), "Raw data are not processed") ;
44ed7a66 226 } else if ( task == AliQAv1::kDIGITSR ) {
227 AliDebug(AliQAv1::GetQADebugLevel(), "Processing Digits QA") ;
eca4fa66 228 TTree * tree = static_cast<TTree *>(data) ;
229 if (strcmp(tree->ClassName(), "TTree") == 0) {
44ed7a66 230 MakeDigits(tree) ;
231 } else {
232 AliWarning("data are not a TTree") ;
233 }
4e25ac79 234 } else if ( task == AliQAv1::kRECPOINTS ) {
5379c4a3 235 AliDebug(AliQAv1::GetQADebugLevel(), "Processing RecPoints QA") ;
eca4fa66 236 TTree * tree = static_cast<TTree *>(data) ;
237 if (strcmp(tree->ClassName(), "TTree") == 0) {
04236e67 238 MakeRecPoints(tree) ;
239 } else {
240 AliWarning("data are not a TTree") ;
241 }
4e25ac79 242 } else if ( task == AliQAv1::kESDS ) {
5379c4a3 243 AliDebug(AliQAv1::GetQADebugLevel(), "Processing ESDs QA") ;
eca4fa66 244 AliESDEvent * esd = static_cast<AliESDEvent *>(data) ;
245 if (strcmp(esd->ClassName(), "AliESDEvent") == 0)
04236e67 246 MakeESDs(esd) ;
247 else
248 AliError("Wrong type of esd container") ;
249 }
250}
251
252//____________________________________________________________________________
4e25ac79 253TObjArray ** AliQADataMakerRec::Init(AliQAv1::TASKINDEX_t task, Int_t cycles)
04236e67 254{
255 // general intialisation
fb6e511e 256 InitRecoParams() ;
57acd2d2 257 TObjArray ** rv = NULL ;
04236e67 258
04236e67 259 if (cycles > 0)
260 SetCycle(cycles) ;
261
4e25ac79 262 if ( task == AliQAv1::kRAWS ) {
63c6f8ae 263 if (! fRawsQAList ) {
57acd2d2 264 fRawsQAList = new TObjArray *[AliRecoParam::kNSpecies] ;
265 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
49466ea2 266 fRawsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
4e25ac79 267 fRawsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
57acd2d2 268 }
63c6f8ae 269 }
04236e67 270 rv = fRawsQAList ;
44ed7a66 271 } else if ( task == AliQAv1::kDIGITSR ) {
272 if ( ! fDigitsQAList ) {
273 fDigitsQAList = new TObjArray *[AliRecoParam::kNSpecies] ;
274 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
49466ea2 275 fDigitsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
44ed7a66 276 fDigitsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
277 }
44ed7a66 278 }
279 rv = fDigitsQAList ;
4e25ac79 280 } else if ( task == AliQAv1::kRECPOINTS ) {
63c6f8ae 281 if ( ! fRecPointsQAList ) {
57acd2d2 282 fRecPointsQAList = new TObjArray *[AliRecoParam::kNSpecies] ;
283 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
49466ea2 284 fRecPointsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
4e25ac79 285 fRecPointsQAList[specie]->SetName(Form("%s_%s_%s", GetName(), AliQAv1::GetTaskName(task).Data(), AliRecoParam::GetEventSpecieName(specie))) ;
57acd2d2 286 }
eca4fa66 287 }
04236e67 288 rv = fRecPointsQAList ;
4e25ac79 289 } else if ( task == AliQAv1::kESDS ) {
63c6f8ae 290 if ( ! fESDsQAList ) {
57acd2d2 291 fESDsQAList = new TObjArray *[AliRecoParam::kNSpecies] ;
292 for (Int_t specie = 0 ; specie < AliRecoParam::kNSpecies ; specie++) {
49466ea2 293 fESDsQAList[specie] = new TObjArray(AliQAv1::GetMaxQAObj()) ;
65b25288 294 fESDsQAList[specie]->SetName(Form("%s_%s", GetName(), AliQAv1::GetTaskName(task).Data())); //, AliRecoParam::GetEventSpecieName(specie))) ;
57acd2d2 295 }
63c6f8ae 296 }
04236e67 297 rv = fESDsQAList ;
298 }
04236e67 299 return rv ;
300}
301
302//____________________________________________________________________________
4e25ac79 303void AliQADataMakerRec::Init(AliQAv1::TASKINDEX_t task, TObjArray ** list, Int_t run, Int_t cycles)
04236e67 304{
305 // Intialisation by passing the list of QA data booked elsewhere
306
fb6e511e 307 InitRecoParams() ;
308 fRun = run ;
04236e67 309 if (cycles > 0)
310 SetCycle(cycles) ;
311
4e25ac79 312 if ( task == AliQAv1::kRAWS ) {
04236e67 313 fRawsQAList = list ;
44ed7a66 314 } else if ( task == AliQAv1::kDIGITSR ) {
315 fDigitsQAList = list ;
4e25ac79 316 } else if ( task == AliQAv1::kRECPOINTS ) {
04236e67 317 fRecPointsQAList = list ;
4e25ac79 318 } else if ( task == AliQAv1::kESDS ) {
04236e67 319 fESDsQAList = list ;
320 }
321}
322
323//____________________________________________________________________________
fb6e511e 324void AliQADataMakerRec::InitRecoParams()
325{
f14c8c46 326 // Get the recoparam form the OCDB
fb6e511e 327 if (!fRecoParam) {
5379c4a3 328 AliDebug(AliQAv1::GetQADebugLevel(), Form("Loading reconstruction parameter objects for detector %s", GetName()));
fb6e511e 329 AliCDBPath path(GetName(),"Calib","RecoParam");
330 AliCDBEntry *entry=AliCDBManager::Instance()->Get(path.GetPath());
331 if(!entry) {
332 fRecoParam = NULL ;
55e356ec 333 AliDebug(AliQAv1::GetQADebugLevel(), Form("Couldn't find RecoParam entry in OCDB for detector %s",GetName()));
fb6e511e 334 }
335 else {
336 TObject * recoParamObj = entry->GetObject() ;
eca4fa66 337 if ( strcmp(recoParamObj->ClassName(), "TObjArray") == 0 ) {
42383dda 338 // The detector has only one set of reco parameters
5379c4a3 339 AliDebug(AliQAv1::GetQADebugLevel(), Form("Array of reconstruction parameters found for detector %s",GetName()));
eca4fa66 340 TObjArray *recoParamArray = static_cast<TObjArray*>(recoParamObj) ;
42383dda 341 for (Int_t iRP=0; iRP<recoParamArray->GetEntriesFast(); iRP++) {
eca4fa66 342 fRecoParam = static_cast<AliDetectorRecoParam*>(recoParamArray->At(iRP)) ;
343 if (!fRecoParam)
344 break ;
345 else if (fRecoParam->IsDefault())
346 break ;
42383dda 347 }
348 }
eca4fa66 349 else if (recoParamObj->InheritsFrom("AliDetectorRecoParam")) {
350 // The detector has only one set of reco parameters
fb6e511e 351 // Registering it in AliRecoParam
5379c4a3 352 AliDebug(AliQAv1::GetQADebugLevel(), Form("Single set of reconstruction parameters found for detector %s",GetName()));
eca4fa66 353 fRecoParam = static_cast<AliDetectorRecoParam*>(recoParamObj) ;
354 static_cast<AliDetectorRecoParam*>(recoParamObj)->SetAsDefault();
fb6e511e 355 } else {
356 AliError(Form("No valid RecoParam object found in the OCDB for detector %s",GetName()));
357 }
358 }
359 }
360}
361
9ac91920 362//____________________________________________________________________________
363void AliQADataMakerRec::ResetDetector(AliQAv1::TASKINDEX_t task)
364{
365 // default reset that resets all the QA objects.
366 // to be overloaded by detectors, if necessary
367
368 TObjArray ** list = NULL ;
369 if ( task == AliQAv1::kRAWS ) {
370 list = fRawsQAList ;
371 } else if ( task == AliQAv1::kDIGITSR ) {
372 list = fDigitsQAList ;
373 } else if ( task == AliQAv1::kRECPOINTS ) {
374 list = fRecPointsQAList ;
375 } else if ( task == AliQAv1::kESDS ) {
376 list = fESDsQAList ;
377 }
378 //list was not initialized, skip
379 if (!list)
380 return ;
381
382 for (int spec = 0; spec < AliRecoParam::kNSpecies; spec++) {
383 if (!AliQAv1::Instance()->IsEventSpecieSet(AliRecoParam::ConvertIndex(spec)))
384 continue;
385 TIter next(list[spec]) ;
386 TH1 * histo = NULL ;
387 while ( (histo = dynamic_cast<TH1*> (next())) ) {
3d08fd77 388 histo->Reset("ICE") ;
fc6a0539 389 histo->ResetStats() ;
9ac91920 390 }
391 }
392}
393
fb6e511e 394//____________________________________________________________________________
930e6e3e 395void AliQADataMakerRec::StartOfCycle(Int_t run)
396{
397 // Finishes a cycle of QA for all the tasks
398 Bool_t samecycle = kFALSE ;
4e25ac79 399 StartOfCycle(AliQAv1::kRAWS, run, samecycle) ;
930e6e3e 400 samecycle = kTRUE ;
44ed7a66 401 StartOfCycle(AliQAv1::kDIGITSR, run, samecycle) ;
4e25ac79 402 StartOfCycle(AliQAv1::kRECPOINTS, run, samecycle) ;
403 StartOfCycle(AliQAv1::kESDS, run, samecycle) ;
930e6e3e 404}
405
406//____________________________________________________________________________
4e25ac79 407void AliQADataMakerRec::StartOfCycle(AliQAv1::TASKINDEX_t task, Int_t run, const Bool_t sameCycle)
04236e67 408{
409 // Finishes a cycle of QA data acquistion
930e6e3e 410 if ( run > 0 )
411 fRun = run ;
04236e67 412 if ( !sameCycle || fCurrentCycle == -1) {
413 ResetCycle() ;
414 if (fOutput)
415 fOutput->Close() ;
50dee02c 416 if (AliQAManager::QAManager(AliQAv1::kRECMODE)->IsSaveData())
417 fOutput = AliQAv1::GetQADataFile(GetName(), fRun) ;
04236e67 418 }
5379c4a3 419 AliDebug(AliQAv1::GetQADebugLevel(), Form(" Run %d Cycle %d task %s file %s",
4e25ac79 420 fRun, fCurrentCycle, AliQAv1::GetTaskName(task).Data(), fOutput->GetName() )) ;
04236e67 421
eca4fa66 422// fDetectorDir = fOutput->GetDirectory(GetDetectorDirName()) ;
423// if (!fDetectorDir)
424// fDetectorDir = fOutput->mkdir(GetDetectorDirName()) ;
425//
426// TDirectory * subDir = fDetectorDir->GetDirectory(AliQAv1::GetTaskName(task)) ;
427// if (!subDir)
428// subDir = fDetectorDir->mkdir(AliQAv1::GetTaskName(task)) ;
429//
430// for ( Int_t specie = AliRecoParam::kDefault ; specie < AliRecoParam::kNSpecies ; specie++ ) {
431// TDirectory * eventSpecieDir = subDir->GetDirectory(AliRecoParam::GetEventSpecieName(specie)) ;
432// if (!eventSpecieDir)
433// eventSpecieDir = subDir->mkdir(AliRecoParam::GetEventSpecieName(specie)) ;
434// TDirectory * expertDir = eventSpecieDir->GetDirectory(AliQAv1::GetExpert()) ;
435// if (!expertDir)
436// expertDir = eventSpecieDir->mkdir(AliQAv1::GetExpert()) ;
437// }
04236e67 438 StartOfDetectorCycle() ;
439}