Several Changes:
[u/mrichter/AliRoot.git] / T0 / AliT0.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 /* $Id$ */
17
18 ///////////////////////////////////////////////////////////////////////////////
19 //                                                                           //
20 //  T0 (T-Zero) Detector                                            //
21 //  This class contains the base procedures for the T0     //
22 //  detector                                                                 //
23 //                                                                           //
24 //Begin_Html
25 /*
26 <img src="gif/AliT0Class.gif">
27 </pre>
28 <br clear=left>
29 <font size=+2 color=red>
30 <p>The responsible person for this module is
31 <a href="mailto:Alla.Maevskaia@cern.ch">Alla Maevskaia</a>.
32 </font>
33 <pre>
34 */
35 //End_Html
36 //                                                                           //
37 //                                                                           //
38 ///////////////////////////////////////////////////////////////////////////////
39
40 #include "TClonesArray.h"
41
42 #include "AliLoader.h"
43 #include "AliLog.h"
44 #include "AliLog.h"
45 #include "AliMC.h"
46 #include "AliRun.h"
47 #include "AliT0.h"
48 #include "AliT0Digitizer.h"
49 #include "AliT0RawData.h"
50 #include "AliT0RecPoint.h"
51 #include "AliT0digit.h"
52 #include "AliT0hit.h"
53
54 ClassImp(AliT0)
55
56   //static  AliT0digit *digits; 
57
58 //_____________________________________________________________________________
59 AliT0::AliT0()
60   : AliDetector(), fIdSens(0), fDigits(NULL), fRecPoints(NULL)
61 {
62   //
63   // Default constructor for class AliT0
64   //
65   fIshunt   = 1;
66   fHits     = 0;
67   fDigits   = 0;
68   fRecPoints = 0;
69 }
70  
71 //_____________________________________________________________________________
72 AliT0::AliT0(const char *name, const char *title)
73   : AliDetector(name,title), fIdSens(0), fDigits(new AliT0digit()), fRecPoints(new AliT0RecPoint())
74 {
75   //
76   // Standard constructor for T0 Detector
77   //
78
79   
80   //
81   // Initialise Hit array
82   fHits       = new TClonesArray("AliT0hit",  405);
83   gAlice->GetMCApp()->AddHitList(fHits);
84   //  fDigits    = new AliT0digit();
85   //  fRecPoints = new AliT0RecPoint();
86   fIshunt     =  1;
87   //  fIdSens   =  0;
88   //PH  SetMarkerColor(kRed);
89 }
90
91 //_____________________________________________________________________________
92 AliT0::~AliT0() {
93   
94   //destructor
95   if (fHits) {
96     fHits->Delete();
97     delete fHits;
98   }
99   /*
100   if (fDigits) {
101     fDigits->Delete();
102     delete fDigits;
103     cout<<" delete fDigits; "<<endl;
104   }
105   if (fRecPoints) {
106    fRecPoints ->Delete();
107     delete fRecPoints;
108     cout<<" delete fRecPoints; "<<endl;
109   }
110   */ 
111 }
112
113 //_____________________________________________________________________________
114 void AliT0::AddHit(Int_t track, Int_t *vol, Float_t *hits)
115 {
116   //
117   // Add a T0 hit
118   //
119   TClonesArray &lhits = *fHits;
120   new(lhits[fNhits++]) AliT0hit(fIshunt,track,vol,hits);
121 }
122
123
124 //_____________________________________________________________________________
125
126 void AliT0::AddDigit(Int_t besttimeright, Int_t besttimeleft, Int_t meantime, 
127                      Int_t timediff, Int_t sumMult, Int_t refpoint,
128                         TArrayI *timeCFD, TArrayI *qt0, TArrayI *timeLED, TArrayI *qt1)
129 {
130   
131   //  Add a T0 digit to the list.
132  //
133   
134   if (!fDigits) {
135     fDigits = new AliT0digit();
136   }
137   fDigits-> SetTimeBestA(besttimeright);
138   fDigits->SetTimeBestC(besttimeleft);
139   fDigits-> SetMeanTime(meantime);
140   fDigits-> SetDiffTime(timediff);
141   fDigits-> SetSumMult(sumMult);
142   fDigits->SetTimeCFD(*timeCFD);
143   fDigits->SetTimeLED(*timeLED);
144   fDigits->SetQT0(*qt0);
145   fDigits->SetQT1(*qt1);
146   fDigits->SetRefPoint(refpoint);
147 }
148
149 //-------------------------------------------------------------------------
150 void AliT0::Init()
151 {
152   //
153   // Initialis the T0 after it has been built
154   Int_t i;
155   //
156   if(AliLog::GetGlobalDebugLevel()>0) {
157     printf("\n%s: ",ClassName());
158     for(i=0;i<35;i++) printf("*");
159     printf(" T0_INIT ");
160     for(i=0;i<35;i++) printf("*");
161     printf("\n%s: ",ClassName());
162     //
163     // Here the T0 initialisation code (if any!)
164     for(i=0;i<80;i++) printf("*");
165     printf("\n");
166   }
167 }
168
169 //---------------------------------------------------------------------------
170 void AliT0::MakeBranch(Option_t* option)
171 {
172   //
173 // Create Tree branches for the T0.
174
175  // Options:
176   //
177   //    H          Make a branch of TClonesArray of AliT0Hit's
178   //    D          Make a branch of TClonesArray of AliT0Digit's
179   //
180   //    R         Make a branch of  AliT0RecPoints
181   //
182   char branchname[20];
183   sprintf(branchname,"%s",GetName());
184
185   const char *cH = strstr(option,"H");
186   const char *cD = strstr(option,"D");
187   const char *cR = strstr(option,"R");
188
189     if (cH && fLoader->TreeH())
190   {
191      if (fHits == 0x0) fHits  = new TClonesArray("AliT0hit",  405);
192      AliDetector::MakeBranch(option);
193   } 
194     
195     
196   if (cD && fLoader->TreeD())
197     {
198       if (fDigits == 0x0) fDigits  = new AliT0digit();
199       //     MakeBranchInTree(fLoader->TreeD(), branchname,
200       //                       &fDigits, 405, 0);
201       fLoader->TreeD()->Branch(branchname,"AliT0digit",&fDigits,405,1);
202       //   fLoader->TreeD()->Print();
203     } 
204   if (cR && fLoader->TreeR())
205     {
206       if (fRecPoints == 0x0) fRecPoints  = new AliT0RecPoint();
207       MakeBranchInTree(fLoader->TreeR(), branchname,
208                        &fRecPoints, 405, 0);
209     } 
210   
211 }    
212
213 //_____________________________________________________________________________
214 void AliT0::ResetHits()
215 {
216   //
217   //reset hits
218   //
219   AliDetector::ResetHits();
220   
221 }
222 //____________________________________________________________________
223 void AliT0::ResetDigits()
224 {
225   //
226   // Reset number of digits and the digits array for this detector
227   //
228   if (fDigits) fDigits->Clear();
229 }
230
231 //_____________________________________________________________________________
232 void AliT0::SetTreeAddress()
233 {
234
235   TTree    *treeH = fLoader->TreeH();
236   
237   if (treeH)
238     {
239       if (fHits == 0x0) fHits  = new TClonesArray("AliT0hit",  405);
240     }
241     
242   AliDetector::SetTreeAddress();
243   TTree *treeD = fLoader->TreeD();
244   if (treeD) {
245     if (fDigits == 0x0)  fDigits  = new AliT0digit();
246     TBranch* branch = treeD->GetBranch ("T0");
247     if (branch) branch->SetAddress(&fDigits);
248   }
249
250   TTree *treeR = fLoader->TreeR();
251   if (treeR) {
252     if (fRecPoints == 0x0) fRecPoints  = new  AliT0RecPoint()  ;
253     TBranch* branch = treeR->GetBranch ("T0");
254     if (branch) branch->SetAddress(&fRecPoints);
255   }
256  
257 }
258
259
260 //_____________________________________________________________________________
261 void AliT0::MakeBranchInTreeD(TTree *treeD, const char *file)
262 {
263     //
264     // Create TreeD branches for the FMD
265     //
266     const Int_t kBufferSize = 4000;
267     char branchname[20];
268     sprintf(branchname,"%s",GetName());
269     if(treeD)
270      {
271        MakeBranchInTree(treeD,  branchname,&fDigits, kBufferSize, file);
272      }
273 }
274
275 //_____________________________________________________________________________
276 AliDigitizer* AliT0::CreateDigitizer(AliRunDigitizer* manager) const
277 {
278   return new AliT0Digitizer(manager);
279 }
280 //____________________________________________________________________________
281 void AliT0::Digits2Raw()
282 {
283 //
284 // Starting from the T0 digits, writes the Raw Data objects
285 //
286 //  AliT0Loader* pStartLoader = (AliT0Loader*)fLoader;
287   fLoader ->LoadDigits("read");
288   TTree* treeD = fLoader->TreeD();
289   if (!treeD) {
290     AliError("no digits tree");
291     return;
292   }
293   if (fDigits == 0x0)  fDigits  = new AliT0digit();
294   
295   TBranch *branch = treeD->GetBranch("T0");
296   if (branch) {
297     branch->SetAddress(&fDigits);
298   }else{
299     AliError("Branch T0 DIGIT not found");
300     exit(111);
301   } 
302   AliT0RawData rawWriter;
303   rawWriter.SetVerbose(0);
304   
305   AliDebug(2,Form(" Formatting raw data for T0 "));
306   branch->GetEntry(0);
307   //  rawWriter.RawDataT0(treeD->GetBranch("T0"));
308   rawWriter.RawDataT0(fDigits);
309   
310   
311   fLoader->UnloadDigits();
312   
313 }
314
315 //____________________________________________________________________________
316 void AliT0::Raw2Digits(AliRawReader *rawReader,TTree* digitsTree)
317 {
318
319  //T0 raw data-> digits conversion
320  // reconstruct time information from raw data
321  // cout<<"  AliT0::Raw2Digits(AliRawReader *rawReader,TTree* digitsTree) "<<
322   // rawReader<<" "<<digitsTree<<endl;
323
324  
325   //  AliT0RawReader myrawreader(rawReader,digitsTree);
326    AliT0RawReader myrawreader(rawReader);
327    if (!myrawreader.Next())
328      AliDebug(1,Form(" no raw data found!! %i", myrawreader.Next()));
329    Int_t allData[110][5];
330    for (Int_t i=0; i<110; i++) {
331      allData[i][0]=myrawreader.GetData(i,0);
332    }
333
334
335    fDigits = new AliT0digit();
336    digitsTree->Branch("T0","AliT0digit",&fDigits,405,1);
337    
338    
339    TArrayI *timeLED = new TArrayI(24);
340    TArrayI * timeCFD = new TArrayI(24);
341    TArrayI *chargeQT0 = new TArrayI(24);
342    TArrayI *chargeQT1 = new TArrayI(24);
343    
344    for (Int_t in=0; in<24; in++)
345      {
346        timeLED->AddAt(allData[in+1][0],in);
347        timeCFD->AddAt(allData[in+25][0],in);
348        chargeQT0->AddAt(allData[in+55][0],in);
349        chargeQT1->AddAt(allData[in+79][0],in);
350        AliDebug(2, Form(" readed Raw %i %i %i %i %i", in, timeLED->At(in),timeCFD->At(in),chargeQT0->At(in),chargeQT1->At(in)));
351      }
352   
353    fDigits->SetTimeCFD(*timeCFD);
354    fDigits->SetQT0(*chargeQT1);
355
356    fDigits->SetTimeLED(*timeLED);
357    fDigits->SetQT1(*chargeQT1);
358
359    fDigits->SetMeanTime(allData[49][0]);
360    fDigits->SetDiffTime(allData[50][0]);
361    fDigits->SetTimeBestA(allData[51][0]);
362    fDigits->SetTimeBestC(allData[52][0]);
363    digitsTree->Fill();
364    fDigits->Write();
365  
366    delete timeCFD ;
367    delete chargeQT0;
368    delete timeLED ;
369    delete chargeQT1;
370
371
372 }