Load pythia libraries.
[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
31 //_____________________________________________________________________
32 // This is the class that collects the QA data for the FMD during simulation.
33 // The following data types are picked up:
34 // - hits
35 // - digits
36 // The following data types are not supported (yet):
37 // - raws
38 // - sdigits
39 // Author : Hans Hjersing Dalsgaard, Niels Bohr Institute, hans.dalsgaard@cern.ch
40 //_____________________________________________________________________
41
42 ClassImp(AliFMDQADataMakerSim)
43 #if 0
44 ; // This line is for Emacs - do not delete!
45 #endif
46 //_____________________________________________________________________
47 AliFMDQADataMakerSim::AliFMDQADataMakerSim() 
48   :  AliQADataMakerSim(AliQA::GetDetName(AliQA::kFMD),
49                        "FMD Quality Assurance Data Maker")
50 {
51   // ctor
52   
53 }
54
55 //_____________________________________________________________________
56 AliFMDQADataMakerSim::AliFMDQADataMakerSim(const AliFMDQADataMakerSim& /*qadm*/) 
57   : AliQADataMakerSim()
58 {
59   //copy ctor 
60   // Parameters: 
61   //    qadm    Object to copy from
62   
63 }
64
65 //_____________________________________________________________________
66 void AliFMDQADataMakerSim::EndOfDetectorCycle(AliQA::TASKINDEX task, 
67                                               TObjArray * list)
68 {
69   //Detector specific actions at end of cycle
70   // do the QA checking
71   AliLog::Message(5,"FMD: end of detector cycle",
72                   "AliFMDQADataMakerSim","AliFMDQADataMakerSim",
73                   "AliFMDQADataMakerSim::EndOfDetectorCycle",
74                   "AliFMDQADataMakerSim.cxx",83);
75   AliQAChecker::Instance()->Run(AliQA::kFMD, task, list) ;  
76   
77 }
78
79 //____________________________________________________________________ 
80 void AliFMDQADataMakerSim::InitHits()
81 {
82   // create Digits histograms in Digits subdir
83   TH1F* hEnergyOfHits = new TH1F("hEnergyOfHits","Energy distribution",100,0,3);
84   hEnergyOfHits->SetXTitle("Edep");
85   hEnergyOfHits->SetYTitle("Counts");
86   Add2HitsList(hEnergyOfHits, 0);
87 }
88
89 //_____________________________________________________________________
90 void AliFMDQADataMakerSim::InitDigits()
91 {
92   // create Digits histograms in Digits subdir
93   TH1I* hADCCounts = new TH1I("hADCCounts","Dist of ADC counts",1024,0,1024);
94   hADCCounts->SetXTitle("ADC counts");
95   Add2DigitsList(hADCCounts, 0);
96 }
97
98 //_____________________________________________________________________
99 void AliFMDQADataMakerSim::MakeHits(TClonesArray * hits)
100 {
101   TIter next(hits);
102   AliFMDHit * hit;
103   while ((hit = static_cast<AliFMDHit *>(next()))) 
104     GetHitsData(0)->Fill(hit->Edep()/hit->Length()*0.032);
105 }
106
107 //_____________________________________________________________________
108 void AliFMDQADataMakerSim::MakeHits(TTree * hitTree)
109 {
110   // make QA data from Hit Tree
111   
112   TBranch * branch = hitTree->GetBranch("FMD") ;
113   if (!branch) {
114     AliWarning("FMD branch in Hit Tree not found") ; 
115     return;
116   }
117
118   TClonesArray * tmp =  new TClonesArray("AliFMDHit", 10) ;
119   branch->SetAddress(&tmp) ;
120   
121   for (Int_t ientry = 0 ; ientry < branch->GetEntries() ; ientry++) {
122     branch->GetEntry(ientry);
123     MakeHits(tmp); 
124   }     
125   tmp->Delete() ; 
126   delete tmp ; 
127 }
128
129 //_____________________________________________________________________
130 void AliFMDQADataMakerSim::MakeDigits(TClonesArray * digits)
131 {
132   // makes data from Digits
133   if(!digits) return;
134
135   for(Int_t i = 0 ; i < digits->GetEntries() ; i++) {
136     //Raw ADC counts
137     AliFMDDigit* digit = static_cast<AliFMDDigit*>(digits->At(i));
138     GetDigitsData(0)->Fill(digit->Counts());
139   }
140 }
141
142 //_____________________________________________________________________
143 void AliFMDQADataMakerSim::MakeDigits(TTree * digitTree)
144 {
145   
146   TClonesArray * digits = new TClonesArray("AliFMDDigit", 1000) ; 
147   
148   TBranch * branch = digitTree->GetBranch("FMD") ;
149   if (!branch)    {
150       AliWarning("FMD branch in Digit Tree not found") ; 
151       return;
152   } 
153   branch->SetAddress(&digits) ;
154   branch->GetEntry(0) ; 
155   MakeDigits(digits) ; 
156 }
157
158 //_____________________________________________________________________ 
159 void AliFMDQADataMakerSim::StartOfDetectorCycle()
160 {
161    
162 }
163 //_____________________________________________________________________ 
164 //
165 // EOF
166 //