Make and print an image of QA user flagged histograms (Yves)
[u/mrichter/AliRoot.git] / FMD / AliFMDQADataMakerSim.cxx
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> 
21
22 // --- AliRoot header files ---
23 #include "AliESDEvent.h"
24 #include "AliLog.h"
25 #include "AliFMDQADataMakerSim.h"
26 #include "AliFMDDigit.h"
27 #include "AliFMDHit.h"
28 #include "AliQAChecker.h"
29 #include "AliFMDParameters.h"
30 #include "AliFMDSDigit.h"
31
32 //_____________________________________________________________________
33 // This is the class that collects the QA data for the FMD during simulation.
34 // The following data types are picked up:
35 // - hits
36 // - digits
37 // The following data types are not supported (yet):
38 // - raws
39 // - sdigits
40 // Author : Hans Hjersing Dalsgaard, Niels Bohr Institute, hans.dalsgaard@cern.ch
41 //_____________________________________________________________________
42
43 ClassImp(AliFMDQADataMakerSim)
44 #if 0
45 ; // This line is for Emacs - do not delete!
46 #endif
47 //_____________________________________________________________________
48 AliFMDQADataMakerSim::AliFMDQADataMakerSim() 
49   :  AliQADataMakerSim(AliQAv1::GetDetName(AliQAv1::kFMD),
50                        "FMD Quality Assurance Data Maker"),
51      fSDigitsArray("AliFMDSDigit", 1000),
52      fDigitsArray("AliFMDDigit", 1000),
53      fHitsArray("AliFMDHit", 10)
54 {
55   // ctor
56
57 }
58
59 //_____________________________________________________________________
60 AliFMDQADataMakerSim::AliFMDQADataMakerSim(const AliFMDQADataMakerSim& qadm) 
61   : AliQADataMakerSim(),
62     fSDigitsArray(qadm.fSDigitsArray),
63     fDigitsArray(qadm.fDigitsArray),
64     fHitsArray(qadm.fHitsArray)
65 {
66   //copy ctor 
67   
68   // Parameters: 
69   //    qadm    Object to copy from
70   
71 }
72 //_____________________________________________________________________
73 AliFMDQADataMakerSim& AliFMDQADataMakerSim::operator = (const AliFMDQADataMakerSim& qadm ) 
74 {
75   fSDigitsArray = qadm.fSDigitsArray;
76   fDigitsArray = qadm.fDigitsArray;
77   fHitsArray = qadm.fHitsArray;
78   
79   return *this;
80 }
81 //_____________________________________________________________________
82 AliFMDQADataMakerSim::~AliFMDQADataMakerSim()
83 {
84
85 }
86
87 //_____________________________________________________________________
88 void AliFMDQADataMakerSim::EndOfDetectorCycle(AliQAv1::TASKINDEX_t task, 
89                                               TObjArray ** list)
90 {
91   //Detector specific actions at end of cycle
92   // do the QA checking
93   AliLog::Message(5,"FMD: end of detector cycle",
94                   "AliFMDQADataMakerSim","AliFMDQADataMakerSim",
95                   "AliFMDQADataMakerSim::EndOfDetectorCycle",
96                   "AliFMDQADataMakerSim.cxx",83);
97   AliQAChecker::Instance()->Run(AliQAv1::kFMD, task, list) ;  
98   
99 }
100 //_____________________________________________________________________
101 void AliFMDQADataMakerSim::InitSDigits()
102 {
103   // create SDigits histograms in SDigits subdir
104   const Bool_t expert   = kTRUE ; 
105   const Bool_t image    = kTRUE ; 
106   
107   TH1I* hADCCounts = new TH1I("hADCCounts","Dist of ADC counts",1024,0,1024);
108   hADCCounts->SetXTitle("ADC counts");
109   Add2SDigitsList(hADCCounts, 0, !expert, image);
110 }
111
112 //____________________________________________________________________ 
113 void AliFMDQADataMakerSim::InitHits()
114 {
115   // create Digits histograms in Digits subdir
116   const Bool_t expert   = kTRUE ; 
117   const Bool_t image    = kTRUE ; 
118   
119   TH1F* hEnergyOfHits = new TH1F("hEnergyOfHits","Energy distribution",100,0,3);
120   hEnergyOfHits->SetXTitle("Edep");
121   hEnergyOfHits->SetYTitle("Counts");
122   Add2HitsList(hEnergyOfHits, 0, !expert, image);
123 }
124
125 //_____________________________________________________________________
126 void AliFMDQADataMakerSim::InitDigits()
127 {
128   // create Digits histograms in Digits subdir
129   const Bool_t expert   = kTRUE ; 
130   const Bool_t image    = kTRUE ; 
131   
132   TH1I* hADCCounts = new TH1I("hADCCounts","Dist of ADC counts",1024,0,1024);
133   hADCCounts->SetXTitle("ADC counts");
134   Add2DigitsList(hADCCounts, 0, !expert, image);
135 }
136
137 //_____________________________________________________________________
138 void AliFMDQADataMakerSim::MakeHits(TClonesArray * hits)
139 {
140   TIter next(hits);
141   AliFMDHit * hit;
142   while ((hit = static_cast<AliFMDHit *>(next()))) 
143     GetHitsData(0)->Fill(hit->Edep()/hit->Length()*0.032);
144 }
145
146 //_____________________________________________________________________
147 void AliFMDQADataMakerSim::MakeHits(TTree * hitTree)
148 {
149   // make QA data from Hit Tree
150   
151   fHitsArray.Clear();
152   
153   TBranch * branch = hitTree->GetBranch("FMD") ;
154   if (!branch) {
155     AliWarning("FMD branch in Hit Tree not found") ; 
156     return;
157   }
158   
159   TClonesArray* hitsAddress = &fHitsArray;
160   
161   branch->SetAddress(&hitsAddress) ;
162   
163   for (Int_t ientry = 0 ; ientry < branch->GetEntries() ; ientry++) {
164     branch->GetEntry(ientry);
165     MakeHits(hitsAddress);   //tmp); 
166   }     
167   
168 }
169
170 //_____________________________________________________________________
171 void AliFMDQADataMakerSim::MakeDigits(TClonesArray * digits)
172 {
173   // makes data from Digits
174   if(!digits) return;
175
176   for(Int_t i = 0 ; i < digits->GetEntriesFast() ; i++) {
177     //Raw ADC counts
178     AliFMDDigit* digit = static_cast<AliFMDDigit*>(digits->At(i));
179     GetDigitsData(0)->Fill(digit->Counts());
180   }
181 }
182
183 //_____________________________________________________________________
184 void AliFMDQADataMakerSim::MakeDigits(TTree * digitTree)
185 {
186   
187   fDigitsArray.Clear();
188   TBranch * branch = digitTree->GetBranch("FMD") ;
189   if (!branch)    {
190       AliWarning("FMD branch in Digit Tree not found") ; 
191       return;
192   } 
193   TClonesArray* digitAddress = &fDigitsArray;
194   branch->SetAddress(&digitAddress) ;
195   branch->GetEntry(0) ; 
196   MakeDigits(digitAddress) ; 
197 }
198
199 //_____________________________________________________________________
200 void AliFMDQADataMakerSim::MakeSDigits(TClonesArray * sdigits)
201 {
202   // makes data from Digits
203   if(!sdigits) return;
204   
205   for(Int_t i = 0 ; i < sdigits->GetEntriesFast() ; i++) {
206     //Raw ADC counts
207     AliFMDSDigit* sdigit = static_cast<AliFMDSDigit*>(sdigits->At(i));
208     GetSDigitsData(0)->Fill(sdigit->Counts());
209   }
210 }
211
212 //_____________________________________________________________________
213 void AliFMDQADataMakerSim::MakeSDigits(TTree * sdigitTree)
214 {
215   
216   fSDigitsArray.Clear();
217   TBranch * branch = sdigitTree->GetBranch("FMD") ;
218   if (!branch)    {
219     AliWarning("FMD branch in SDigit Tree not found") ; 
220     return;
221   } 
222   TClonesArray* sdigitAddress = &fSDigitsArray;
223   branch->SetAddress(&sdigitAddress) ;
224   branch->GetEntry(0) ; 
225   MakeSDigits(sdigitAddress) ; 
226 }
227
228 //_____________________________________________________________________ 
229 void AliFMDQADataMakerSim::StartOfDetectorCycle()
230 {
231    
232 }
233 //_____________________________________________________________________ 
234 //
235 // EOF
236 //