Make and print an image of QA user flagged histograms (Yves)
[u/mrichter/AliRoot.git] / ZDC / AliZDCQADataMakerRec.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 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
16 ///////////////////////////////////////////////////////////////////////
17 //                                                                   //
18 //  Produces the data needed to calculate the ZDC quality assurance. //
19 //  QA objects are 1 & 2 Dimensional histograms.                     //
20 //  author: C. Oppedisano                                            //
21 //                                                                   //
22 ///////////////////////////////////////////////////////////////////////
23
24 // --- ROOT system ---
25 #include <TClonesArray.h>
26 #include <TFile.h> 
27 #include <TH1F.h> 
28 #include <TH2F.h>
29 #include <TProfile.h>
30 #include <Riostream.h>
31 // --- Standard library ---
32
33 // --- AliRoot header files ---
34 #include "AliLog.h"
35 #include "AliQAChecker.h"
36 #include "AliRawReader.h"
37 #include "AliZDCQADataMakerRec.h"
38 #include "AliZDCRawStream.h"
39 #include "AliESDZDC.h"
40 #include "AliESDEvent.h"
41
42 ClassImp(AliZDCQADataMakerRec)
43            
44 //____________________________________________________________________________ 
45   AliZDCQADataMakerRec::AliZDCQADataMakerRec() : 
46   AliQADataMakerRec(AliQAv1::GetDetName(AliQAv1::kZDC), "ZDC Quality Assurance Data Maker")
47 {
48   // ctor
49 }
50
51 //____________________________________________________________________________ 
52 AliZDCQADataMakerRec::AliZDCQADataMakerRec(const AliZDCQADataMakerRec& qadm) :
53   AliQADataMakerRec() 
54 {
55   //copy ctor 
56   SetName((const char*)qadm.GetName()); 
57   SetTitle((const char*)qadm.GetTitle()); 
58 }
59
60 //__________________________________________________________________
61 AliZDCQADataMakerRec& AliZDCQADataMakerRec::operator = (const AliZDCQADataMakerRec& qadm )
62 {
63   // Equal operator.
64   this->~AliZDCQADataMakerRec();
65   new(this) AliZDCQADataMakerRec(qadm);
66   return *this;
67 }
68
69 //____________________________________________________________________________
70
71 void AliZDCQADataMakerRec::InitRaws()
72 {
73   // create Digits histograms in Digits subdir
74   //
75   // ------------------- HIGH GAIN CHAIN ---------------------------
76   TH1F * hRawZNCTot = new TH1F("hRawZNCTot", "Raw signal in ZNC", 100, 0., 6000.);
77   TH1F * hRawZNATot = new TH1F("hRawZNATot", "Raw signal in ZNA", 100, 0., 6000.);
78   TH1F * hRawZPCTot = new TH1F("hRawZPCTot", "Raw signal in ZPC", 100, 0., 10000.);
79   TH1F * hRawZPATot = new TH1F("hRawZPATot", "Raw signal in ZPA", 100, 0., 10000.);
80   Add2RawsList(hRawZNCTot, 0);
81   Add2RawsList(hRawZNATot, 1);
82   Add2RawsList(hRawZPCTot, 2);
83   Add2RawsList(hRawZPATot, 3);
84   //
85   TH1F * hRawSumQZNC = new TH1F("hRawSumQZNC", "Raw summed 4 ZNC quadrants",100, 0., 4000.);
86   TH1F * hRawSumQZNA = new TH1F("hRawSumQZNA", "Raw summed 4 ZNA quadrants",100, 0., 4000.);
87   TH1F * hRawSumQZPC = new TH1F("hRawSumQZPC", "Raw summed 4 ZPC quadrants",100, 0., 4000.);
88   TH1F * hRawSumQZPA = new TH1F("hRawSumQZPA", "Raw summed 4 ZPA quadrants",100, 0., 4000.);
89   Add2RawsList(hRawSumQZNC, 4, kTRUE);
90   Add2RawsList(hRawSumQZNA, 5, kTRUE);
91   Add2RawsList(hRawSumQZPC, 6, kTRUE);
92   Add2RawsList(hRawSumQZPA, 7, kTRUE);
93   //
94   TH1F * hRawPMCZNC = new TH1F("hRawPMCZNC", "Raw common ZNC PMT",100, 0., 4000.);
95   TH1F * hRawPMCZNA = new TH1F("hRawPMCZNA", "Raw common ZNA PMT",100, 0., 4000.);
96   TH1F * hRawPMCZPC = new TH1F("hRawPMCZPC", "Raw common ZPC PMT",100, 0., 4000.);
97   TH1F * hRawPMCZPA = new TH1F("hRawPMCZPA", "Raw common ZPA PMT",100, 0., 4000.);
98   Add2RawsList(hRawPMCZNC, 8 , kTRUE);
99   Add2RawsList(hRawPMCZNA, 9 , kTRUE);
100   Add2RawsList(hRawPMCZPC, 10, kTRUE);
101   Add2RawsList(hRawPMCZPA, 11, kTRUE);
102   // 
103 /*  // ------------------- LOW GAIN CHAIN ---------------------------
104   TH1F * hRawZNCTotlg = new TH1F("hRawZNCTotlg", "Rawit lg signal in ZNC", 100, 0., 6000.);
105   TH1F * hRawZNATotlg = new TH1F("hRawZNATotlg", "Rawit lg signal in ZNA", 100, 0., 6000.);
106   TH1F * hRawZPCTotlg = new TH1F("hRawZPCTotlg", "Rawit lg signal in ZPC", 100, 0., 10000.);
107   TH1F * hRawZPATotlg = new TH1F("hRawZPATotlg", "Rawit lg signal in ZPA", 100, 0., 10000.);
108   Add2RawsList(hRawZNCTotlg, 12);
109   Add2RawsList(hRawZNATotlg, 13);
110   Add2RawsList(hRawZPCTotlg, 14);
111   Add2RawsList(hRawZPATotlg, 15);
112   //
113   TH1F * hRawSumQZNClg = new TH1F("hRawSumQZNClg", "Raw summed 4 lg ZNC quadrants",100, 0., 4000.);
114   TH1F * hRawSumQZNAlg = new TH1F("hRawSumQZNAlg", "Raw summed 4 lg ZNA quadrants",100, 0., 4000.);
115   TH1F * hRawSumQZPClg = new TH1F("hRawSumQZPClg", "Raw summed 4 lg ZPC quadrants",100, 0., 4000.);
116   TH1F * hRawSumQZPAlg = new TH1F("hRawSumQZPAlg", "Raw summed 4 lg ZPA quadrants",100, 0., 4000.);
117   Add2RawsList(hRawSumQZNClg, 16, kTRUE);
118   Add2RawsList(hRawSumQZNAlg, 17, kTRUE);
119   Add2RawsList(hRawSumQZPClg, 18, kTRUE);
120   Add2RawsList(hRawSumQZPAlg, 19, kTRUE);
121   //
122   TH1F * hRawPMCZNClg = new TH1F("hRawPMCZNClg", "Raw common lg ZNC PMT",100, 0., 4000.);
123   TH1F * hRawPMCZNAlg = new TH1F("hRawPMCZNAlg", "Raw common lg ZNA PMT",100, 0., 4000.);
124   TH1F * hRawPMCZPClg = new TH1F("hRawPMCZPClg", "Raw common lg ZPC PMT",100, 0., 4000.);
125   TH1F * hRawPMCZPAlg = new TH1F("hRawPMCZPAlg", "Raw common lg ZPA PMT",100, 0., 4000.);
126   Add2RawsList(hRawPMCZNClg, 20, kTRUE);
127   Add2RawsList(hRawPMCZNAlg, 21, kTRUE);
128   Add2RawsList(hRawPMCZPClg, 22, kTRUE);
129   Add2RawsList(hRawPMCZPAlg, 23, kTRUE);*/
130 }
131
132 //____________________________________________________________________________
133 void AliZDCQADataMakerRec::InitESDs()
134 {
135   //Booking ESDs histograms
136   //
137   const Bool_t expert   = kTRUE ; 
138   const Bool_t image    = kTRUE ; 
139   
140   TH2F * hZNC  = new TH2F("hZNC", "Centroid in ZNC", 100, -5.,5.,100,-5.,5.);
141   TH2F * hZNA  = new TH2F("hZNA", "Centroid in ZNA", 100, -5.,5.,100,-5.,5.);
142   Add2ESDsList(hZNC, 0, !expert, image);
143   Add2ESDsList(hZNA, 1, !expert, image);
144   //
145   // ------------------- HIGH GAIN CHAIN ---------------------------
146   TH1F * hESDZNCTot = new TH1F("hESDZNCTot", "Energy in ZNC", 100, 0., 6000.);
147   TH1F * hESDZNATot = new TH1F("hESDZNATot", "Energy in ZNA", 100, 0., 6000.);
148   TH1F * hESDZPCTot = new TH1F("hESDZPCTot", "Energy in ZPC", 100, 0., 10000.);
149   TH1F * hESDZPATot = new TH1F("hESDZPATot", "Energy in ZPA", 100, 0., 10000.);
150   Add2ESDsList(hESDZNCTot, 2, !expert, image);
151   Add2ESDsList(hESDZNATot, 3, !expert, image);
152   Add2ESDsList(hESDZPCTot, 4, !expert, image);
153   Add2ESDsList(hESDZPATot, 5, !expert, image);
154   //
155   TH1F * hESDSumQZNC = new TH1F("hESDSumQZNC", "Sum of 4 ZNC energy",100, 0., 4000.);
156   TH1F * hESDSumQZNA = new TH1F("hESDSumQZNA", "Sum of 4 ZNA energy",100, 0., 4000.);
157   TH1F * hESDSumQZPC = new TH1F("hESDSumQZPC", "Sum of 4 ZPC energy",100, 0., 4000.);
158   TH1F * hESDSumQZPA = new TH1F("hESDSumQZPA", "Sum of 4 ZPA energy",100, 0., 4000.);
159   Add2ESDsList(hESDSumQZNC, 6, expert, !image);
160   Add2ESDsList(hESDSumQZNA, 7, expert, !image);
161   Add2ESDsList(hESDSumQZPC, 8, expert, !image);
162   Add2ESDsList(hESDSumQZPA, 9, expert, !image);
163   //
164   TH1F * hESDPMCZNC = new TH1F("hESDPMCZNC", "Energy in common ZNC PMT",100, 0., 4000.);
165   TH1F * hESDPMCZNA = new TH1F("hESDPMCZNA", "Energy in common ZNA PMT",100, 0., 4000.);
166   TH1F * hESDPMCZPC = new TH1F("hESDPMCZPC", "Energy in common ZPC PMT",100, 0., 4000.);
167   TH1F * hESDPMCZPA = new TH1F("hESDPMCZPA", "Energy in common ZPA PMT",100, 0., 4000.);
168   Add2ESDsList(hESDPMCZNC, 10, expert, !image);
169   Add2ESDsList(hESDPMCZNA, 11, expert, !image);
170   Add2ESDsList(hESDPMCZPC, 12, expert, !image);
171   Add2ESDsList(hESDPMCZPA, 13, expert, !image);
172   // 
173 /*  // ------------------- LOW GAIN CHAIN ---------------------------
174   TH1F * hESDZNCTotlg = new TH1F("hESDZNCTotlg", "ESD lg signal in ZNC", 100, 0., 6000.);
175   TH1F * hESDZNATotlg = new TH1F("hESDZNATotlg", "ESD lg signal in ZNA", 100, 0., 6000.);
176   TH1F * hESDZPCTotlg = new TH1F("hESDZPCTotlg", "ESD lg signal in ZPC", 100, 0., 10000.);
177   TH1F * hESDZPATotlg = new TH1F("hESDZPATotlg", "ESD lg signal in ZPA", 100, 0., 10000.);
178   Add2ESDsList(hESDZNCTotlg, !expert, image);
179   Add2ESDsList(hESDZNATotlg, !expert, image);
180   Add2ESDsList(hESDZPCTotlg, !expert, image);
181   Add2ESDsList(hESDZPATotlg, !expert, image);
182   //
183   TH1F * hESDSumQZNClg = new TH1F("hESDSumQZNClg", "Sum of 4 lg ZNC sectors",100, 0., 4000.);
184   TH1F * hESDSumQZNAlg = new TH1F("hESDSumQZNAlg", "Sum of 4 lg ZNA sectors",100, 0., 4000.);
185   TH1F * hESDSumQZPClg = new TH1F("hESDSumQZPClg", "Sum of 4 lg ZPC sectors",100, 0., 4000.);
186   TH1F * hESDSumQZPAlg = new TH1F("hESDSumQZPAlg", "Sum of 4 lg ZPA sectors",100, 0., 4000.);
187   Add2ESDsList(hESDSumQZNClg, 18, expert, !image);
188   Add2ESDsList(hESDSumQZNAlg, 19, expert, !image);
189   Add2ESDsList(hESDSumQZPClg, 20, expert, !image);
190   Add2ESDsList(hESDSumQZPAlg, 21, expert, !image);
191   //
192   TH1F * hESDPMCZNClg = new TH1F("hESDPMCZNClg", "Signal in common ZNC lg PMT",100, 0., 4000.);
193   TH1F * hESDPMCZNAlg = new TH1F("hESDPMCZNAlg", "Signal in common ZNA lg PMT",100, 0., 4000.);
194   TH1F * hESDPMCZPClg = new TH1F("hESDPMCZPClg", "Signal in common ZPC lg PMT",100, 0., 4000.);
195   TH1F * hESDPMCZPAlg = new TH1F("hESDPMCZPAlg", "Signal in common ZPA lg PMT",100, 0., 4000.);
196   Add2ESDsList(hESDPMCZNClg, 22, expert, !image);
197   Add2ESDsList(hESDPMCZNAlg, 23, expert, !image);
198   Add2ESDsList(hESDPMCZPClg, 24, expert, !image);
199   Add2ESDsList(hESDPMCZPAlg, 25, expert, !image);*/
200 }
201   
202 //____________________________________________________________________________
203
204 void AliZDCQADataMakerRec::MakeRaws(AliRawReader *rawReader)
205 {
206   // Filling Raws QA histos
207   //
208         rawReader->Reset() ; 
209   Float_t sum_ZNC=0., sum_ZNA=0., sum_ZPC=0., sum_ZPA=0.;
210   Float_t sumQ_ZNC=0., sumQ_ZNA=0., sumQ_ZPC=0., sumQ_ZPA=0.;
211   //Float_t sum_ZNC_lg=0., sum_ZNA_lg=0., sum_ZPC_lg=0., sum_ZPA_lg=0.;
212   //Float_t sumQ_ZNC_lg=0., sumQ_ZNA_lg=0., sumQ_ZPC_lg=0., sumQ_ZPA_lg=0.;
213   //
214   AliZDCRawStream stream(rawReader);
215   while(stream.Next()){
216     if(stream.IsADCDataWord() && 
217      (stream.GetADCModule()==0 || stream.GetADCModule()==1)){
218        if(stream.GetSector(0)==1){
219          if(stream.GetADCGain()==0){
220            sum_ZNC += stream.GetADCValue();
221            if(stream.GetSector(1)!=0) sumQ_ZNC += stream.GetADCValue();
222            else GetRawsData(8)->Fill(stream.GetADCValue());
223          }
224          /*else{
225            sum_ZNC_lg += stream.GetADCValue();
226            if(stream.GetSector(1)!=0) sumQ_ZNC_lg += stream.GetADCValue();
227            else GetRawsData(20)->Fill(stream.GetADCValue());
228          }*/
229        }
230        else if(stream.GetSector(0)==2){
231          if(stream.GetADCGain()==0){
232            sum_ZPC += stream.GetADCValue();
233            if(stream.GetSector(1)!=0) sumQ_ZPC += stream.GetADCValue();
234            else GetRawsData(10)->Fill(stream.GetADCValue());
235          }
236          /*else{
237            sum_ZPC_lg += stream.GetADCValue();
238            if(stream.GetSector(1)!=0) sumQ_ZPC_lg += stream.GetADCValue();
239            else GetRawsData(22)->Fill(stream.GetADCValue());
240          }*/
241        }
242        else if(stream.GetSector(0)==4){
243          if(stream.GetADCGain()==0){
244            sum_ZNA += stream.GetADCValue();
245            if(stream.GetSector(1)!=0) sumQ_ZNA += stream.GetADCValue();
246            else GetRawsData(9)->Fill(stream.GetADCValue());
247          }
248          /*else{
249            sum_ZNA_lg += stream.GetADCValue();
250            if(stream.GetSector(1)!=0) sumQ_ZNA_lg += stream.GetADCValue();
251            else GetRawsData(21)->Fill(stream.GetADCValue());
252          }*/
253        }
254        else if(stream.GetSector(0)==5){
255          if(stream.GetADCGain()==0){
256            sum_ZPA += stream.GetADCValue();
257            if(stream.GetSector(1)!=0) sumQ_ZPA += stream.GetADCValue();
258            else GetRawsData(11)->Fill(stream.GetADCValue());
259          }
260          /*else{
261            sum_ZPA_lg += stream.GetADCValue();
262            if(stream.GetSector(1)!=0) sumQ_ZPA_lg += stream.GetADCValue();
263            else GetRawsData(23)->Fill(stream.GetADCValue());
264          }*/
265        }
266     }
267   }
268   //
269   GetRawsData(0)->Fill(sum_ZNC);
270   GetRawsData(1)->Fill(sum_ZNA);
271   GetRawsData(2)->Fill(sum_ZPC);
272   GetRawsData(3)->Fill(sum_ZPA);
273   //
274   GetRawsData(4)->Fill(sumQ_ZNC);
275   GetRawsData(5)->Fill(sumQ_ZNA);
276   GetRawsData(6)->Fill(sumQ_ZPC);
277   GetRawsData(7)->Fill(sumQ_ZPA);
278   //
279   /*GetRawsData(12)->Fill(sum_ZNC_lg);
280   GetRawsData(13)->Fill(sum_ZNA_lg);
281   GetRawsData(14)->Fill(sum_ZPC_lg);
282   GetRawsData(15)->Fill(sum_ZPA_lg);
283   //
284   GetRawsData(16)->Fill(sumQ_ZNC_lg);
285   GetRawsData(17)->Fill(sumQ_ZNA_lg);
286   GetRawsData(18)->Fill(sumQ_ZPC_lg);
287   GetRawsData(19)->Fill(sumQ_ZPA_lg);*/
288   //
289   stream.Delete();
290 }
291
292 //____________________________________________________________________________
293 void AliZDCQADataMakerRec::MakeESDs(AliESDEvent * esd)
294 {
295   // make QA data from ESDs
296   //
297   AliESDZDC * zdcESD =  esd->GetESDZDC();
298   //
299   Double32_t * centr_ZNC = zdcESD->GetZNCCentroid();
300   GetESDsData(0)->Fill(centr_ZNC[0], centr_ZNC[1]);
301
302   Double32_t * centr_ZNA = zdcESD->GetZNACentroid();
303   GetESDsData(1)->Fill(centr_ZNA[0], centr_ZNA[1]);
304
305   //
306   GetESDsData(2)->Fill(esd->GetZDCN1Energy());
307   GetESDsData(3)->Fill(esd->GetZDCN2Energy());
308   GetESDsData(4)->Fill(esd->GetZDCP1Energy());
309   GetESDsData(5)->Fill(esd->GetZDCP2Energy());
310   //
311   Double_t sumQZNC=0., sumQZPC=0., sumQZNA=0., sumQZPA=0.;
312   //Double_t sumQZNC_lg=0., sumQZPC_lg=0., sumQZNA_lg=0., sumQZPA_lg=0.;
313   //
314   const Double_t *towZNC, *towZPC, *towZNA, *towZPA;
315   //const Double_t *towZNC_lg, *towZPC_lg, *towZNA_lg, *towZPA_lg;
316   //
317   towZNC = zdcESD->GetZN1TowerEnergy();
318   towZPC = zdcESD->GetZP1TowerEnergy();
319   towZNA = zdcESD->GetZN2TowerEnergy();
320   towZPA = zdcESD->GetZP2TowerEnergy();
321   //
322   /*towZNC_lg = zdcESD->GetZN1TowerEnergyLR();
323   towZPC_lg = zdcESD->GetZP1TowerEnergyLR();
324   towZNA_lg = zdcESD->GetZN2TowerEnergyLR();
325   towZPA_lg = zdcESD->GetZP2TowerEnergyLR();*/
326   //
327   for(Int_t i=0; i<5; i++){
328      if(i==0){
329        GetESDsData(10)->Fill(towZNC[i]);
330        GetESDsData(11)->Fill(towZNA[i]);
331        GetESDsData(12)->Fill(towZPC[i]);
332        GetESDsData(13)->Fill(towZPA[i]);
333        //
334        /*GetESDsData(22)->Fill(towZNC_lg[i]);
335        GetESDsData(23)->Fill(towZNA_lg[i]);
336        GetESDsData(24)->Fill(towZPC_lg[i]);
337        GetESDsData(25)->Fill(towZPA_lg[i]);*/
338      }
339      else{
340        sumQZNC += towZNC[i];
341        sumQZPC += towZPC[i];
342        sumQZNA += towZNA[i];
343        sumQZPA += towZPA[i];
344        //
345        /*sumQZNC_lg += towZNC_lg[i];
346        sumQZPC_lg += towZPC_lg[i];
347        sumQZNA_lg += towZNA_lg[i];
348        sumQZPA_lg += towZPA_lg[i];*/
349      }
350   }
351   GetESDsData(6)->Fill(sumQZNC);
352   GetESDsData(7)->Fill(sumQZNA);
353   GetESDsData(8)->Fill(sumQZPC);
354   GetESDsData(9)->Fill(sumQZPA);
355   //
356   /*GetESDsData(18)->Fill(sumQZNC_lg);
357   GetESDsData(19)->Fill(sumQZNA_lg);
358   GetESDsData(20)->Fill(sumQZPC_lg);
359   GetESDsData(21)->Fill(sumQZPA_lg);*/
360 }
361
362 //____________________________________________________________________________
363 void AliZDCQADataMakerRec::StartOfDetectorCycle()
364 {
365   //Detector specific actions at start of cycle
366   
367 }
368
369 //____________________________________________________________________________ 
370 void AliZDCQADataMakerRec::EndOfDetectorCycle(AliQAv1::TASKINDEX_t task, TObjArray ** list)
371 {
372   //Detector specific actions at end of cycle
373   // do the QA checking
374   AliQAChecker::Instance()->Run(AliQAv1::kZDC, task, list) ;  
375 }
376