Adding the proper nested loop
[u/mrichter/AliRoot.git] / FMD / AliFMDQADataMakerRec.cxx
CommitLineData
c9dd1c4d 1/**************************************************************************
2 * Copyright(c) 2004, 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 notice *
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// --- ROOT system ---
16#include <iostream>
17#include <TClonesArray.h>
18#include <TFile.h>
19#include <TH1F.h>
20#include <TH1I.h>
05e5e0c1 21#include <TH2I.h>
1306ba55 22#include <TGeoManager.h>
c9dd1c4d 23
24// --- AliRoot header files ---
25#include "AliESDEvent.h"
26#include "AliLog.h"
1306ba55 27#include "AliGeomManager.h"
c9dd1c4d 28#include "AliFMDQADataMakerRec.h"
29#include "AliFMDDigit.h"
30#include "AliFMDRecPoint.h"
31#include "AliQAChecker.h"
32#include "AliESDFMD.h"
33#include "AliFMDParameters.h"
86c6cec8 34#include "AliFMDRawReader.h"
1306ba55 35#include "AliFMDReconstructor.h"
3ceaa9ad 36#include "AliRawReader.h"
b995fc28 37#include "AliFMDAltroMapping.h"
f4b5062e 38#include "AliFMDDebug.h"
c9dd1c4d 39
05e5e0c1 40namespace {
41 Int_t colors[3] = {kRed,kGreen,kBlue};
42}
c9dd1c4d 43//_____________________________________________________________________
44// This is the class that collects the QA data for the FMD during
45// reconstruction.
46//
47// The following data types are picked up:
c9dd1c4d 48// - rec points
49// - esd data
c9dd1c4d 50// - raws
51// Author : Hans Hjersing Dalsgaard, hans.dalsgaard@cern.ch
52//_____________________________________________________________________
53
54ClassImp(AliFMDQADataMakerRec)
55#if 0
56; // For Emacs - do not delete!
57#endif
58
59//_____________________________________________________________________
1306ba55 60AliFMDQADataMakerRec::AliFMDQADataMakerRec()
61 : AliQADataMakerRec(AliQAv1::GetDetName(AliQAv1::kFMD),
62 "FMD Quality Assurance Data Maker"),
63 fRecPointsArray("AliFMDRecPoint", 1000),
64 fReconstructor(0),
65 fUseReconstructor(true)
c9dd1c4d 66{
67 // ctor
56236ce9 68
c9dd1c4d 69}
70
71//_____________________________________________________________________
a7e41e8d 72AliFMDQADataMakerRec::AliFMDQADataMakerRec(const AliFMDQADataMakerRec& qadm)
4e25ac79 73 : AliQADataMakerRec(AliQAv1::GetDetName(AliQAv1::kFMD),
a7e41e8d 74 "FMD Quality Assurance Data Maker"),
1306ba55 75 fRecPointsArray(qadm.fRecPointsArray),
76 fReconstructor(qadm.fReconstructor),
77 fUseReconstructor(qadm.fUseReconstructor)
c9dd1c4d 78{
ffa78f64 79 // copy ctor
c9dd1c4d 80 // Parameters:
81 // qadm Object to copy from
82
83}
9bd2ccc2 84//_____________________________________________________________________
63720979 85AliFMDQADataMakerRec&
86AliFMDQADataMakerRec::operator = (const AliFMDQADataMakerRec& qadm )
a7e41e8d 87{
09b6c804 88 //
89 // Assignment operator
90 //
91 // Parameters:
92 // qadm What to assign from
93 //
94 // Return:
95 // Reference to this
96 //
1306ba55 97 fRecPointsArray = qadm.fRecPointsArray;
98 fReconstructor = qadm.fReconstructor;
99 fUseReconstructor = qadm.fUseReconstructor;
a7e41e8d 100 return *this;
101}
102//_____________________________________________________________________
9bd2ccc2 103AliFMDQADataMakerRec::~AliFMDQADataMakerRec()
104{
09b6c804 105 //
106 // Destrcutor
107 //
9bd2ccc2 108}
c9dd1c4d 109
110
111//_____________________________________________________________________
112
113void
4e25ac79 114AliFMDQADataMakerRec::EndOfDetectorCycle(AliQAv1::TASKINDEX_t task,
57acd2d2 115 TObjArray ** list)
c9dd1c4d 116{
117 // Detector specific actions at end of cycle
118 // do the QA checking
92664bc8 119 ResetEventTrigClasses(); // reset triggers list to select all histos
ffa78f64 120 AliLog::Message(5,"FMD: end of detector cycle",
121 "AliFMDQADataMakerRec","AliFMDQADataMakerRec",
122 "AliFMDQADataMakerRec::EndOfDetectorCycle",
123 "AliFMDQADataMakerRec.cxx",95);
4e25ac79 124 AliQAChecker::Instance()->Run(AliQAv1::kFMD, task, list);
c9dd1c4d 125}
126
127//_____________________________________________________________________
05e5e0c1 128TH1* AliFMDQADataMakerRec::MakeADCHist(UShort_t d, Char_t r, Short_t b)
129{
130 TString name("adc");
131 TString title("ADC counts");
132 Int_t color = kRed+1;
133 if (d > 0) {
134 name.Append(Form("FMD%d%c", d, r));
135 title.Append(Form(" in FMD%d%c", d, r));
136 color = colors[d-1]+3+(r == 'I' || r == 'i' ? 0 : 1);
137 if (b >= 0) {
138 name.Append(Form("_0x%02x", b));
139 title.Append(Form("[0x%02x]", b));
140 }
141 }
142 TH1* hist = new TH1F(name, title,1024,0,1024);
143 hist->SetXTitle("Amplitude [ADC counts]");
144 hist->SetYTitle("Events [log]");
145 hist->SetFillStyle(3001);
146 hist->SetFillColor(color);
147 hist->SetLineColor(color);
148 hist->SetMarkerColor(color);
149 hist->GetXaxis()->SetNdivisions(408,false);
dacd22c9 150 hist->SetDirectory(0);
05e5e0c1 151 // hist->SetStats(0);
152
153 return hist;
154}
155//_____________________________________________________________________
156TH1* AliFMDQADataMakerRec::MakeELossHist(UShort_t d, Char_t r, Short_t b)
157{
158 TString name("eloss");
159 TString title("Energy loss");
160 Int_t color = kBlue+1;
161 if (d > 0) {
162 name.Append(Form("FMD%d%c", d, r));
163 title.Append(Form(" in FMD%d%c", d, r));
164 color = colors[d-1]+3+(r == 'I' || r == 'i' ? 0 : 1);
165 if (b >= 0) {
166 name.Append(Form("_0x%02x", b));
167 title.Append(Form("[0x%02x]", b));
168 }
169 }
1306ba55 170 TH1* hist = new TH1F(name, title,600,0, 15);
171 hist->SetXTitle("#Delta/#Delta_{mip}");
05e5e0c1 172 hist->SetYTitle("Events [log]");
173 hist->SetFillStyle(3001);
174 hist->SetFillColor(color);
175 hist->SetLineColor(color);
176 hist->SetMarkerColor(color);
1306ba55 177 hist->Sumw2();
dacd22c9 178 hist->SetDirectory(0);
05e5e0c1 179 // hist->SetStats(0);
180
181 return hist;
182}
183
184
185//_____________________________________________________________________
c9dd1c4d 186void AliFMDQADataMakerRec::InitESDs()
187{
188 // create Digits histograms in Digits subdir
1306ba55 189 Info("InitESDs", "Initializing ESDs");
7d297381 190 const Bool_t expert = kTRUE ;
191 const Bool_t image = kTRUE ;
192
05e5e0c1 193 TH1* hist = MakeELossHist();
194 Add2ESDsList(hist, 0, !expert, image);
92664bc8 195 ClonePerTrigClass(AliQAv1::kESDS); // this should be the last line
c9dd1c4d 196}
197
44ed7a66 198//_____________________________________________________________________
199void AliFMDQADataMakerRec::InitDigits()
200{
201 // create Digits histograms in Digits subdir
1306ba55 202 Info("InitDigits", "Initializing Digits");
44ed7a66 203 const Bool_t expert = kTRUE ;
204 const Bool_t image = kTRUE ;
205
05e5e0c1 206 TH1* hist = MakeADCHist();
207 Add2DigitsList(hist, 0, !expert, image);
92664bc8 208 ClonePerTrigClass(AliQAv1::kDIGITS); // this should be the last line
44ed7a66 209}
210
c9dd1c4d 211//_____________________________________________________________________
c9dd1c4d 212void AliFMDQADataMakerRec::InitRecPoints()
213{
7d297381 214 // create Reconstructed Points histograms in RecPoints subdir
1306ba55 215 Info("InitRecPoints", "Initializing RecPoints");
7d297381 216 const Bool_t expert = kTRUE ;
217 const Bool_t image = kTRUE ;
218
05e5e0c1 219 TH1* hist = MakeELossHist();
220 Add2RecPointsList(hist,0, !expert, image);
92664bc8 221 ClonePerTrigClass(AliQAv1::kRECPOINTS); // this should be the last linea
c9dd1c4d 222}
223
224//_____________________________________________________________________
225void AliFMDQADataMakerRec::InitRaws()
226{
7d297381 227 // create Raws histograms in Raws subdir
1306ba55 228 Info("InitRaws", "Initializing Raws");
7d297381 229 const Bool_t expert = kTRUE ;
230 const Bool_t saveCorr = kTRUE ;
231 const Bool_t image = kTRUE ;
05e5e0c1 232 TH2I* hErrors = new TH2I("readoutErrors", "Read out errors", 3, .5, 3.5,
0f666f27 233 160, -.5, 159.5);
05e5e0c1 234 hErrors->GetXaxis()->SetBinLabel(1, "FMD1");
235 hErrors->GetXaxis()->SetBinLabel(2, "FMD2");
236 hErrors->GetXaxis()->SetBinLabel(3, "FMD3");
237 hErrors->SetYTitle("# errors");
238 hErrors->SetZTitle("Events [log]");
dacd22c9 239 hErrors->SetDirectory(0);
05e5e0c1 240 Add2RawsList(hErrors, 1, !expert, image, !saveCorr);
241 //AliInfo(Form("Adding %30s to raw list @ %2d", hErrors->GetName(), 1));
242
1306ba55 243 if (fUseReconstructor && !fReconstructor) {
244 // Int_t oldDbg = AliLog::GetDebugLevel("FMD","");
245 // AliLog::SetModuleDebugLevel("FMD", 5);
246
247 if (!gGeoManager) {
248 Info("InitRaws", "Loading the geometry");
249 AliGeomManager::LoadGeometry();
250 }
251
252 fReconstructor = new AliFMDReconstructor();
253 fReconstructor->SetDiagnose(false);
254 fReconstructor->Init();
255 // AliLog::SetModuleDebugLevel("FMD", oldDbg);
256 }
257
05e5e0c1 258 TH1* hist;
259 Int_t idx = 0;
260 for(UShort_t d = 1; d<=3; d++) {
261 UShort_t nR = (d == 1 ? 1 : 2);
262 for(UShort_t q = 0; q < nR; q++) {
263 Char_t r = (q == 1 ? 'O' : 'I');
1306ba55 264 hist = (fUseReconstructor ?
265 MakeELossHist(d, r, -1) :
266 MakeADCHist(d, r, -1));
05e5e0c1 267
268 Int_t index1 = GetHalfringIndex(d, r, 0, 1);
269 idx = TMath::Max(index1, idx);
270 Add2RawsList(hist, index1, !expert, image, !saveCorr);
271 //AliInfo(Form("Adding %30s to raw list @ %2d", hist->GetName(), index1));
c18fe410 272
1306ba55 273 // If we're using the reconstructor, do not make expert histograms
274 if (fUseReconstructor) continue;
275
05e5e0c1 276 for(UShort_t b = 0; b <= 1; b++) {
c18fe410 277 //Hexadecimal board numbers 0x0, 0x1, 0x10, 0x11;
05e5e0c1 278 UShort_t board = (q == 1 ? 0 : 1) + b*16;
c18fe410 279
05e5e0c1 280 hist = MakeADCHist(d, r, board);
281 Int_t index2 = GetHalfringIndex(d, r, board/16,0);
282 idx = TMath::Max(index2, idx);
283 Add2RawsList(hist, index2, expert, !image, !saveCorr);
284 //AliInfo(Form("Adding %30s to raw list @ %2d",hist->GetName(),index2));
b4da452d 285 }
286 }
287 }
92664bc8 288 //
289 ClonePerTrigClass(AliQAv1::kRAWS); // this should be the last line
c9dd1c4d 290}
291
408bf2b4 292#if 0
293struct FillESDHist : public AliESDFMD::ForOne
294{
295 FillESDHist(AliFMDQADataMakerRec* m) : fM(m) {}
296 FillESDHist(const FillESDHist& o) : fM(o.fM) {}
297 FillESDHist& operator=(const FillESDHist& o) { fM = o.fM; return *this; }
298 Bool_t operator()(UShort_t, Char_t, UShort_t, UShort_t, Float_t m, Float_t)
299 {
300 // Float_t mult = fmd->Multiplicity(det,ring,sec,strip);
301 if(m == AliESDFMD::kInvalidMult) return true;
302
92664bc8 303 fM->FillESDsData(0,m);
408bf2b4 304 return true;
305 }
306 AliFMDQADataMakerRec* fM;
307};
308#endif
309
c9dd1c4d 310//_____________________________________________________________________
311void AliFMDQADataMakerRec::MakeESDs(AliESDEvent * esd)
312{
09b6c804 313 //
314 // Analyse ESD event
315 //
316 // Parameters:
317 // esd ESD event
318 //
c9dd1c4d 319 if(!esd) {
320 AliError("FMD ESD object not found!!") ;
321 return;
322 }
f4b5062e 323 AliFMDDebug(2, ("Will loop over ESD data and fill histogram"));
324
c9dd1c4d 325 AliESDFMD* fmd = esd->GetFMDData();
326 if (!fmd) return;
f4b5062e 327
408bf2b4 328#if 0
329 FillESDHist f(this);
330 fmd->ForEach(f);
331#else
332
333
334
f4b5062e 335 // FIXME - we should use AliESDFMD::ForOne subclass to do this!
c9dd1c4d 336 for(UShort_t det=1;det<=3;det++) {
f4b5062e 337 UShort_t nrng = (det == 1 ? 1 : 2);
338 for (UShort_t ir = 0; ir < nrng; ir++) {
c9dd1c4d 339 Char_t ring = (ir == 0 ? 'I' : 'O');
340 UShort_t nsec = (ir == 0 ? 20 : 40);
341 UShort_t nstr = (ir == 0 ? 512 : 256);
342 for(UShort_t sec =0; sec < nsec; sec++) {
343 for(UShort_t strip = 0; strip < nstr; strip++) {
344 Float_t mult = fmd->Multiplicity(det,ring,sec,strip);
345 if(mult == AliESDFMD::kInvalidMult) continue;
346
92664bc8 347 FillESDsData(0,mult);
c9dd1c4d 348 }
349 }
350 }
351 }
408bf2b4 352#endif
92664bc8 353 IncEvCountCycleESDs();
354 IncEvCountTotalESDs();
c9dd1c4d 355}
356
44ed7a66 357
c9dd1c4d 358//_____________________________________________________________________
6252ceeb 359void AliFMDQADataMakerRec::MakeDigits()
c9dd1c4d 360{
361 // makes data from Digits
6252ceeb 362 if(!fDigitsArray) {
ffa78f64 363 AliError("FMD Digit object not found!!") ;
364 return;
c9dd1c4d 365 }
eca4fa66 366
6252ceeb 367 for(Int_t i=0;i<fDigitsArray->GetEntriesFast();i++) {
c9dd1c4d 368 //Raw ADC counts
6252ceeb 369 AliFMDDigit* digit = static_cast<AliFMDDigit*>(fDigitsArray->At(i));
92664bc8 370 FillDigitsData(0,digit->Counts());
c9dd1c4d 371 }
372}
373
374//_____________________________________________________________________
375void AliFMDQADataMakerRec::MakeDigits(TTree * digitTree)
376{
09b6c804 377 //
378 // Analyse digits
379 //
380 // Parameters:
381 // digitTree Tree of digits
382 //
6252ceeb 383 if (fDigitsArray)
384 fDigitsArray->Clear();
385 else
386 fDigitsArray = new TClonesArray("AliFMDDigit", 1000);
387
ffa78f64 388 TBranch* branch = digitTree->GetBranch("FMD");
389 if (!branch) {
c9dd1c4d 390 AliWarning("FMD branch in Digit Tree not found") ;
391 return;
392 }
6252ceeb 393 branch->SetAddress(&fDigitsArray);
ffa78f64 394 branch->GetEntry(0);
6252ceeb 395 MakeDigits();
92664bc8 396 //
397 IncEvCountCycleDigits();
398 IncEvCountTotalDigits();
c9dd1c4d 399}
44ed7a66 400
c9dd1c4d 401//_____________________________________________________________________
86c6cec8 402void AliFMDQADataMakerRec::MakeRaws(AliRawReader* rawReader)
c9dd1c4d 403{
09b6c804 404 //
405 // Analyse raw
406 //
407 // Parameters:
408 // rawReader Raw reader
409 //
09b6c804 410 AliFMDRawReader fmdReader(rawReader,0);
1306ba55 411 if (fDigitsArray) fDigitsArray->Clear();
412 else fDigitsArray = new TClonesArray("AliFMDDigit", 1000);
413
6252ceeb 414 TClonesArray* digitsAddress = fDigitsArray;
1306ba55 415
3ceaa9ad 416 rawReader->Reset();
1306ba55 417
c18fe410 418 digitsAddress->Clear();
419 fmdReader.ReadAdcs(digitsAddress);
92664bc8 420 //
421 FillRawsData(1,1, fmdReader.GetNErrors(0));
422 FillRawsData(1,2, fmdReader.GetNErrors(1));
423 FillRawsData(1,3, fmdReader.GetNErrors(2));
1306ba55 424
425 if (fUseReconstructor) {
426 AliESDFMD* fmd = fReconstructor->GetESDObject();
427 fmd->Clear();
428
429 // AliLog::SetModuleDebugLevel("FMD", 15);
430 fReconstructor->ProcessDigits(digitsAddress, fmdReader);
431
432 if (!fmd) AliFatal("No ESD object from reconstructor");
433
434 for(UShort_t det=1;det<=3;det++) {
435 UShort_t nrng = (det == 1 ? 1 : 2);
436 for (UShort_t ir = 0; ir < nrng; ir++) {
437 Char_t ring = (ir == 0 ? 'I' : 'O');
438 UShort_t nsec = (ir == 0 ? 20 : 40);
439 UShort_t nstr = (ir == 0 ? 512 : 256);
440 for(UShort_t sec =0; sec < nsec; sec++) {
441 for(UShort_t strip = 0; strip < nstr; strip++) {
442 Float_t mult = fmd->Multiplicity(det,ring,sec,strip);
443 if(mult == AliESDFMD::kInvalidMult) continue;
444
445 Int_t index1 = GetHalfringIndex(det, ring, 0, 1);
446 FillRawsData(index1,mult);
447 }
448 }
449 }
450 }
451 }
452 else {
453 for(Int_t i=0;i<digitsAddress->GetEntriesFast();i++) {
454 //Raw ADC counts
455 AliFMDDigit* digit = static_cast<AliFMDDigit*>(digitsAddress->At(i));
456 UShort_t det = digit->Detector();
457 Char_t ring = digit->Ring();
458 UShort_t sec = digit->Sector();
459 // UShort_t strip = digit->Strip();
460 AliFMDParameters* pars = AliFMDParameters::Instance();
461 Short_t board = pars->GetAltroMap()->Sector2Board(ring, sec);
462
463 Int_t index1 = GetHalfringIndex(det, ring, 0, 1);
464 FillRawsData(index1,digit->Counts());
465 Int_t index2 = GetHalfringIndex(det, ring, board/16,0);
466 FillRawsData(index2,digit->Counts());
467 }
468 }
92664bc8 469 //
470 IncEvCountCycleRaws();
471 IncEvCountTotalRaws();
c9dd1c4d 472}
473
474//_____________________________________________________________________
475void AliFMDQADataMakerRec::MakeRecPoints(TTree* clustersTree)
476{
477 // makes data from RecPoints
eca4fa66 478
6252ceeb 479 AliFMDParameters* pars = AliFMDParameters::Instance();
56236ce9 480 fRecPointsArray.Clear();
c9dd1c4d 481 TBranch *fmdbranch = clustersTree->GetBranch("FMD");
482 if (!fmdbranch) {
483 AliError("can't get the branch with the FMD recpoints !");
484 return;
485 }
486
56236ce9 487 TClonesArray* RecPointsAddress = &fRecPointsArray;
9bd2ccc2 488
56236ce9 489 fmdbranch->SetAddress(&RecPointsAddress);
c9dd1c4d 490 fmdbranch->GetEntry(0);
56236ce9 491 TIter next(RecPointsAddress) ;
c9dd1c4d 492 AliFMDRecPoint * rp ;
493 while ((rp = static_cast<AliFMDRecPoint*>(next()))) {
92664bc8 494 FillRecPointsData(0,rp->Edep()/pars->GetEdepMip());
c9dd1c4d 495 }
92664bc8 496 IncEvCountCycleRecPoints();
497 IncEvCountTotalRecPoints();
498 //
c9dd1c4d 499}
500
501//_____________________________________________________________________
502void AliFMDQADataMakerRec::StartOfDetectorCycle()
503{
504 // What
505 // to
506 // do?
507}
508//_____________________________________________________________________
b4da452d 509Int_t AliFMDQADataMakerRec::GetHalfringIndex(UShort_t det,
510 Char_t ring,
c18fe410 511 UShort_t board,
09b6c804 512 UShort_t monitor) const
513{
514 //
515 // Get the half-ring index
516 //
517 // Parameters:
518 // det Detector
519 // ring Ring
520 // board Board number
521 // monitor Monitor
522 //
523 // Return:
524 // Half ring index
525 //
05e5e0c1 526 UShort_t iring = (ring == 'I' || ring == 'i' ? 1 : 0);
527 Int_t index = ((((det-1) & 0x3) << 3) |
528 ((iring & 0x1) << 2) |
529 ((board & 0x1) << 1) |
530 ((monitor & 0x1) << 0));
531#if 0
532 AliInfo(Form("d=%d, r=%c, b=%d, m=%d -> (%d<<3)|(%d<<2)|(%d<<1)|(%d<<0)=%2d",
533 det, ring, board, monitor, (det-1) & 0x3, iring & 0x1,
534 board & 0x1, monitor & 0x1, index));
535#endif
b4da452d 536 return index-2;
b4da452d 537}
538
539//_____________________________________________________________________
540
541
c9dd1c4d 542//
543// EOF
544//