bug in reconstruction cosmic runs
[u/mrichter/AliRoot.git] / T0 / AliT0Reconstructor.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  *  T0 reconstruction and filling ESD
19  *  - reconstruct mean time (interation time) 
20  *  - vertex position
21  *  -  multiplicity
22  ********************************************************************/
23
24 #include <AliESDEvent.h>
25 #include "AliLog.h"
26 #include "AliT0RecPoint.h"
27 #include "AliRawReader.h"
28 #include "AliT0RawReader.h"
29 #include "AliT0digit.h"
30 #include "AliT0Reconstructor.h"
31 #include "AliT0Parameters.h"
32 #include "AliT0Calibrator.h"
33
34 #include <TArrayI.h>
35 #include <TGraph.h>
36 #include <TMath.h>
37 //#include <iostream.h>
38
39 ClassImp(AliT0Reconstructor)
40
41   AliT0Reconstructor:: AliT0Reconstructor(): AliReconstructor(),
42                                              fdZonA(0),
43                                              fdZonC(0),
44                                              fZposition(0),
45                                              fParam(NULL),
46                                              fAmpLEDrec()
47 {
48   //constructor
49
50  AliDebug(1,"Start reconstructor ");
51   
52   fParam = AliT0Parameters::Instance();
53   fParam->Init();
54   for (Int_t i=0; i<24; i++){
55     TGraph* gr = fParam ->GetAmpLEDRec(i);
56     fAmpLEDrec.AddAtAndExpand(gr,i) ;  
57   }
58   fdZonC = TMath::Abs(fParam->GetZPositionShift("T0/C/PMT1"));
59   fdZonA = TMath::Abs(fParam->GetZPositionShift("T0/A/PMT15"));
60
61 }
62 //____________________________________________________________________
63
64 AliT0Reconstructor::AliT0Reconstructor(const AliT0Reconstructor &r):
65   AliReconstructor(r),
66                                              fdZonA(0),
67                                              fdZonC(0),
68                                              fZposition(0),
69                                              fParam(NULL),
70   fAmpLEDrec()
71  {
72   //
73   // AliT0Reconstructor copy constructor
74   //
75
76   ((AliT0Reconstructor &) r).Copy(*this);
77
78 }
79
80 //_____________________________________________________________________________
81 AliT0Reconstructor &AliT0Reconstructor::operator=(const AliT0Reconstructor &r)
82 {
83   //
84   // Assignment operator
85   //
86
87   if (this != &r) ((AliT0Reconstructor &) r).Copy(*this);
88   return *this;
89
90 }
91
92 //_____________________________________________________________________________
93
94 void AliT0Reconstructor::Reconstruct(TTree*digitsTree, TTree*clustersTree) const
95   
96 {
97   // T0 digits reconstruction
98   // T0RecPoint writing 
99   
100    
101   TArrayI * timeCFD = new TArrayI(24); 
102   TArrayI * timeLED = new TArrayI(24); 
103   TArrayI * chargeQT0 = new TArrayI(24); 
104   TArrayI * chargeQT1 = new TArrayI(24); 
105
106   AliT0Calibrator *calib=new AliT0Calibrator(); 
107
108   //  Int_t mV2Mip = param->GetmV2Mip();     
109   //mV2Mip = param->GetmV2Mip();     
110   Float_t channelWidth = fParam->GetChannelWidth() ;  
111   Int_t meanT0 = fParam->GetMeanT0();
112   
113   AliDebug(1,Form("Start DIGITS reconstruction "));
114   
115   TBranch *brDigits=digitsTree->GetBranch("T0");
116   AliT0digit *fDigits = new AliT0digit() ;
117   if (brDigits) {
118     brDigits->SetAddress(&fDigits);
119   }else{
120     AliError(Form("EXEC Branch T0 digits not found"));
121      return;
122   }
123   
124   digitsTree->GetEvent(0);
125   digitsTree->GetEntry(0);
126   brDigits->GetEntry(0);
127   fDigits->GetTimeCFD(*timeCFD);
128   fDigits->GetTimeLED(*timeLED);
129   fDigits->GetQT0(*chargeQT0);
130   fDigits->GetQT1(*chargeQT1);
131
132   
133   Float_t besttimeA=999999;
134   Float_t besttimeC=999999;
135   Int_t pmtBestA=99999;
136   Int_t pmtBestC=99999;
137   Float_t timeDiff=999999, meanTime=0;
138   
139
140
141   AliT0RecPoint* frecpoints= new AliT0RecPoint ();
142   clustersTree->Branch( "T0", "AliT0RecPoint" ,&frecpoints, 405,1);
143   
144   Float_t time[24], adc[24];
145   for (Int_t ipmt=0; ipmt<24; ipmt++) {
146     if(timeCFD->At(ipmt)>0 ){
147       Double_t qt0 = Double_t(chargeQT0->At(ipmt));
148       Double_t qt1 = Double_t(chargeQT1->At(ipmt));
149       if((qt1-qt0)>0)  adc[ipmt] = TMath::Exp( Double_t (channelWidth*(qt1-qt0)/1000));
150       time[ipmt] = calib-> WalkCorrection( ipmt, Int_t(qt1) , timeCFD->At(ipmt), "pdc" ) ;
151       
152       //LED
153       Double_t sl = (timeLED->At(ipmt) - time[ipmt])*channelWidth;
154       Double_t qt=((TGraph*)fAmpLEDrec.At(ipmt))->Eval(sl/1000.);
155       frecpoints->SetTime(ipmt,time[ipmt]);
156       frecpoints->SetAmp(ipmt,adc[ipmt]);
157       frecpoints->SetAmpLED(ipmt,qt);
158     }
159     else {
160       time[ipmt] = 0;
161       adc[ipmt] = 0;
162     }
163   }
164   
165   for (Int_t ipmt=0; ipmt<12; ipmt++){
166     if(time[ipmt] > 1 ) {
167       if(time[ipmt]<besttimeC){
168         besttimeC=time[ipmt]; //timeC
169         pmtBestC=ipmt;
170       }
171     }
172   }
173   for ( Int_t ipmt=12; ipmt<24; ipmt++){
174     if(time[ipmt] > 1) {
175       if(time[ipmt]<besttimeA) {
176         besttimeA=time[ipmt]; //timeA
177         pmtBestA=ipmt;}
178     }
179   }
180   if(besttimeA !=999999)  frecpoints->SetTimeBestA(Int_t(besttimeA));
181   if( besttimeC != 999999 ) frecpoints->SetTimeBestC(Int_t(besttimeC));
182   AliDebug(1,Form(" besttimeA %f ps,  besttimeC %f ps",besttimeA, besttimeC));
183   Float_t c = 0.0299792; // cm/ps
184   Float_t vertex = 0;
185   if(besttimeA !=999999 && besttimeC != 999999 ){
186     timeDiff =(besttimeC - besttimeA)*channelWidth;
187     meanTime = (meanT0 - (besttimeA + besttimeC)/2) * channelWidth;
188     vertex = c*(timeDiff)/2. + (fdZonA - fdZonC)/2; //-(lenr-lenl))/2;
189     AliDebug(1,Form("  timeDiff %f ps,  meanTime %f ps, vertex %f cm",timeDiff, meanTime,vertex ));
190     frecpoints->SetVertex(vertex);
191     frecpoints->SetMeanTime(Int_t(meanTime));
192     
193   }
194   //time in each channel as time[ipmt]-MeanTimeinThisChannel(with vertex=0)
195   for (Int_t ipmt=0; ipmt<24; ipmt++) {
196     if(time[ipmt]>1) {
197 //      time[ipmt] = (time[ipmt] - fTime0vertex[ipmt])*channelWidth;
198       time[ipmt] = time[ipmt] * channelWidth;
199       frecpoints->SetTime(ipmt,time[ipmt]);
200     }
201   }
202   clustersTree->Fill();
203
204   delete timeCFD;
205   delete timeLED;
206   delete chargeQT0; 
207   delete chargeQT1; 
208 }
209
210
211 //_______________________________________________________________________
212
213 void AliT0Reconstructor::Reconstruct(AliRawReader* rawReader, TTree*recTree) const
214 {
215   // T0 raw ->
216   // T0RecPoint writing 
217   
218   //Q->T-> coefficients !!!! should be measured!!!
219   Int_t allData[110][5];
220
221   Int_t timeCFD[24], timeLED[24], chargeQT0[24], chargeQT1[24];
222   TString option = GetOption(); 
223   AliDebug(1,Form("Option: %s\n", option.Data()));
224   
225   for (Int_t i0=0; i0<105; i0++)
226     {
227       for (Int_t j0=0; j0<5; j0++) allData[i0][j0]=0;   
228     }
229   
230   
231   AliDebug(10," before read data ");
232   AliT0RawReader myrawreader(rawReader);
233   if (!myrawreader.Next())
234     AliDebug(1,Form(" no raw data found!! %i", myrawreader.Next()));
235   
236   for (Int_t i=0; i<105; i++) {
237     for (Int_t iHit=0; iHit<5; iHit++) 
238       {
239         allData[i][iHit] = myrawreader.GetData(i,iHit);
240       }
241   }
242   AliT0Calibrator *calib = new AliT0Calibrator(); 
243   
244   //  Int_t mV2Mip = param->GetmV2Mip();     
245   //mV2Mip = param->GetmV2Mip();     
246   Float_t channelWidth = fParam->GetChannelWidth() ;  
247   
248   Int_t meanT0 = fParam->GetMeanT0();
249   if(option == "pdc"){
250     for (Int_t in=0; in<24; in++)  
251       {
252         
253         timeLED[in] = allData[in+1][0];
254         timeCFD[in] = allData[in+25][0];
255         chargeQT1[in] = allData[in+57][0];
256         chargeQT0[in] = allData[in+80][0];
257       }
258   }
259
260   if(option == "cosmic") {
261     for (Int_t in=0; in<12; in++)  
262       {
263         timeCFD[in] = allData[in+1][0];
264         timeCFD[in+12] = allData[in+1][0];
265         timeLED[in] = allData[in+12+1][0];
266         timeLED[in+12] = allData[in+68+1][0];
267       }
268     
269     for (Int_t in=0; in<24;  in=in+2)
270       {
271       Int_t cc=in/2;
272       chargeQT1[cc]=allData[in+25][0];
273       chargeQT0[cc]=allData[in+26][0];
274       }
275     for (Int_t in=24; in<48;  in=in+2)
276     {
277       Int_t cc=in/2;
278       chargeQT1[cc]=allData[in+57][0];
279       chargeQT0[cc]=allData[in+58][0];
280     }
281     
282   }
283     for (Int_t in=0; in<24; in++)  
284        AliDebug(10, Form(" readed Raw %i %i %i %i %i", in, timeLED[in],timeCFD[in],chargeQT0[in],chargeQT1[in]));
285      
286
287   Float_t besttimeA=9999999;
288   Float_t besttimeC=9999999;
289   Int_t pmtBestA=99999;
290   Int_t pmtBestC=99999;
291   Float_t timeDiff=9999999, meanTime=0;
292   Double_t qt=0;
293   
294   AliT0RecPoint* frecpoints= new AliT0RecPoint ();
295  
296   recTree->Branch( "T0", "AliT0RecPoint" ,&frecpoints, 405,1);
297   
298
299   Float_t time[24], adc[24];
300   for (Int_t ipmt=0; ipmt<24; ipmt++) {
301     if(timeCFD[ipmt]>0 ){
302      
303       if(option == "pdc"){
304         Double_t qt0 = Double_t(chargeQT0[ipmt]);
305         Double_t qt1 = Double_t(chargeQT1[ipmt]);
306         if((qt1-qt0)>0)  adc[ipmt] = TMath::Exp( Double_t (channelWidth*(qt1-qt0)/1000));
307
308         time[ipmt] = calib-> WalkCorrection( ipmt,Int_t(qt1) , timeCFD[ipmt] ) ;
309         Double_t sl = (timeLED[ipmt] - time[ipmt])*channelWidth;
310         if(fAmpLEDrec.At(ipmt)) 
311           qt=((TGraph*)fAmpLEDrec.At(ipmt))->Eval(sl/1000.);
312         frecpoints->SetTime(ipmt,time[ipmt]);
313         frecpoints->SetAmp(ipmt,adc[ipmt]);
314         frecpoints->SetAmpLED(ipmt,qt);
315         AliDebug(1,Form(" QTC %f mv,  time in chann %f ",adc[ipmt] ,time[ipmt]));
316       }
317       if(option == "cosmic") {
318         if(ipmt == 15) continue; //skip crashed PMT
319         Float_t qt0 = Float_t(chargeQT0[ipmt]);
320         Float_t qt1 = Float_t(chargeQT1[ipmt]);
321         if((qt1-qt0)>0)  adc[ipmt] = qt1-qt0;
322         
323         time[ipmt] = calib-> WalkCorrection( ipmt, Int_t(adc[ipmt]), timeCFD[ipmt] ) ;
324         Double_t sl = timeLED[ipmt] - time[ipmt];
325         if(fAmpLEDrec.At(ipmt)) 
326           qt=((TGraph*)fAmpLEDrec.At(ipmt))->Eval(sl/1000.);
327         frecpoints->SetTime(ipmt,time[ipmt]);
328         frecpoints->SetAmp(ipmt,adc[ipmt]);
329         frecpoints->SetAmpLED(ipmt,qt);
330         AliDebug(10,Form(" ipmt %i QTC %i , time in chann %i led %i ",
331                         ipmt, Int_t(adc[ipmt]) ,Int_t(time[ipmt]),Int_t( qt)));
332       }
333       
334     }
335     else {
336       time[ipmt] = 0;
337       adc[ipmt] = 0;
338     }
339   }
340   
341   for (Int_t ipmt=0; ipmt<12; ipmt++){
342     if(time[ipmt] > 1 ) {
343       if(time[ipmt]<besttimeC){
344         besttimeC=time[ipmt]; //timeC
345         pmtBestC=ipmt;
346       }
347     }
348   }
349   for ( Int_t ipmt=12; ipmt<24; ipmt++){
350     if(time[ipmt] > 1) {
351       if(time[ipmt]<besttimeA) {
352         besttimeA=time[ipmt]; //timeA
353         pmtBestA=ipmt;}
354     }
355   }
356   if(besttimeA !=9999999)  frecpoints->SetTimeBestA(Int_t(besttimeA));
357   if( besttimeC != 9999999 ) frecpoints->SetTimeBestC(Int_t(besttimeC));
358   AliDebug(1,Form(" besttimeA %f ps,  besttimeC %f ps",besttimeA, besttimeC));
359   Float_t c = 0.0299792; // cm/ps
360   Float_t vertex = 99999;
361   if(besttimeA <9999999 && besttimeC < 9999999 ){
362     timeDiff =( besttimeC - besttimeA) *channelWidth;
363     if(option == "pdc") 
364       meanTime = (meanT0 - (besttimeA + besttimeC)/2) * channelWidth;
365     if(option == "cosmic")   meanTime =  (besttimeA + besttimeC)/2;  
366     vertex = c*(timeDiff)/2. + (fdZonA - fdZonC)/2; 
367     AliDebug(1,Form("  timeDiff %f ps,  meanTime %f ps, vertex %f cm",timeDiff, meanTime,vertex ));
368     frecpoints->SetVertex(vertex);
369     frecpoints->SetMeanTime(Int_t(meanTime));
370     
371   }
372   //time in each channel as time[ipmt]-MeanTimeinThisChannel(with vertex=0)
373   /*
374   for (Int_t ipmt=0; ipmt<24; ipmt++) {
375     if(time[ipmt]>1) {
376       time[ipmt] = (time[ipmt] - fTime0vertex[ipmt])*channelWidth;
377       frecpoints->SetTime(ipmt,time[ipmt]);
378     }
379     }
380   */
381   recTree->Fill();
382 }
383 //____________________________________________________________
384
385 void AliT0Reconstructor::FillESD(TTree */*digitsTree*/, TTree *clustersTree, AliESDEvent *pESD) const
386 {
387
388   /***************************************************
389   Resonstruct digits to vertex position
390   ****************************************************/
391   
392   AliDebug(1,Form("Start FillESD T0"));
393
394   TTree *treeR = clustersTree;
395   
396    AliT0RecPoint* frecpoints= new AliT0RecPoint ();
397     if (!frecpoints) {
398     AliError("Reconstruct Fill ESD >> no recpoints found");
399     return;
400   }
401   
402   AliDebug(1,Form("Start FillESD T0"));
403   TBranch *brRec = treeR->GetBranch("T0");
404   if (brRec) {
405     brRec->SetAddress(&frecpoints);
406   }else{
407     AliError(Form("EXEC Branch T0 rec not found"));
408     return;
409   } 
410     
411     brRec->GetEntry(0);
412     Float_t amp[24], time[24];
413     Float_t  zPosition = frecpoints -> GetVertex();
414     Float_t timeStart = frecpoints -> GetMeanTime() ;
415     for ( Int_t i=0; i<24; i++) {
416       time[i] = Float_t (frecpoints -> GetTime(i)); // ps to ns
417       amp[i] = frecpoints -> GetAmp(i);
418     }
419     pESD->SetT0zVertex(zPosition); //vertex Z position 
420     pESD->SetT0(timeStart);        // interaction time 
421     pESD->SetT0time(time);         // best TOF on each PMT 
422     pESD->SetT0amplitude(amp);     // number of particles(MIPs) on each PMT
423  
424     AliDebug(1,Form(" Z position %f cm,  T0  %f ps",zPosition , timeStart));
425
426 } // vertex in 3 sigma
427
428
429
430
431
432