3e98e449451a75dd956622cb39b6163107c6de41
[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 <Riostream.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                                              fQTC(0),
48                                              fAmpLED(0),
49                                              fCalib()
50 {
51   //constructor
52
53   fParam = AliT0Parameters::Instance();
54   fParam->Init();
55  
56   for (Int_t i=0; i<24; i++){
57         TGraph* gr = fParam ->GetAmpLEDRec(i);
58         if (gr) fAmpLEDrec.AddAtAndExpand(gr,i) ; 
59           TGraph* gr1 = fParam ->GetAmpLED(i);
60           if (gr1) fAmpLED.AddAtAndExpand(gr1,i) ; 
61           TGraph* gr2 = fParam ->GetQTC(i);
62           if (gr2) fQTC.AddAtAndExpand(gr2,i) ;         
63   }
64
65   
66   // fdZonC = TMath::Abs(fParam->GetZPositionShift("T0/C/PMT1"));
67   //fdZonA = TMath::Abs(fParam->GetZPositionShift("T0/A/PMT15"));
68
69
70   fCalib = new AliT0Calibrator();
71
72 }
73
74 //_____________________________________________________________________________
75 void AliT0Reconstructor::Reconstruct(TTree*digitsTree, TTree*clustersTree) const
76   
77 {
78   // T0 digits reconstruction
79   Int_t refAmp = GetRecoParam()->GetRefAmp();
80   
81   TArrayI * timeCFD = new TArrayI(24); 
82   TArrayI * timeLED = new TArrayI(24); 
83   TArrayI * chargeQT0 = new TArrayI(24); 
84   TArrayI * chargeQT1 = new TArrayI(24); 
85
86  
87   Float_t channelWidth = fParam->GetChannelWidth() ;  
88   Float_t meanVertex = fParam->GetMeanVertex();
89   Float_t c = 0.0299792; // cm/ps
90   Double32_t vertex = 9999999;
91   Double32_t timeDiff=999999, meanTime=999999, timeclock=999999;
92
93   
94   AliDebug(1,Form("Start DIGITS reconstruction "));
95   
96
97   TBranch *brDigits=digitsTree->GetBranch("T0");
98   AliT0digit *fDigits = new AliT0digit() ;
99   if (brDigits) {
100     brDigits->SetAddress(&fDigits);
101   }else{
102     AliError(Form("EXEC Branch T0 digits not found"));
103      return;
104   }
105   
106   digitsTree->GetEvent(0);
107   digitsTree->GetEntry(0);
108   brDigits->GetEntry(0);
109   fDigits->GetTimeCFD(*timeCFD);
110   fDigits->GetTimeLED(*timeLED);
111   fDigits->GetQT0(*chargeQT0);
112   fDigits->GetQT1(*chargeQT1);
113   Int_t onlineMean =  fDigits->MeanTime();
114
115   Bool_t tr[5];
116   for (Int_t i=0; i<5; i++) tr[i]=false; 
117   
118   Double32_t besttimeA=999999;
119   Double32_t besttimeC=999999;
120   Int_t pmtBestA=99999;
121   Int_t pmtBestC=99999;
122   
123   AliT0RecPoint* frecpoints= new AliT0RecPoint ();
124   clustersTree->Branch( "T0", "AliT0RecPoint" ,&frecpoints, 405,1);
125   
126   Float_t time[24], adc[24];
127   for (Int_t ipmt=0; ipmt<24; ipmt++) {
128     if(timeCFD->At(ipmt)>0 ){
129      if(( chargeQT1->At(ipmt) - chargeQT0->At(ipmt))>0)  
130         adc[ipmt] = chargeQT1->At(ipmt) - chargeQT0->At(ipmt);
131       else
132         adc[ipmt] = 0;
133       
134      // time[ipmt] = fCalib-> WalkCorrection(refAmp, ipmt, adc[ipmt],  timeCFD->At(ipmt)) ;
135              
136       Double_t sl = Double_t(timeLED->At(ipmt) - timeCFD->At(ipmt));
137       time[ipmt] = fCalib-> WalkCorrection( refAmp,ipmt, Int_t(sl),  timeCFD->At(ipmt) ) ;
138       AliDebug(10,Form(" ipmt %i QTC %i , time in chann %i (led-cfd) %i ",
139                        ipmt, Int_t(adc[ipmt]) ,Int_t(time[ipmt]),Int_t( sl)));
140
141       Double_t ampMip =((TGraph*)fAmpLED.At(ipmt))->Eval(sl);
142       Double_t qtMip = ((TGraph*)fQTC.At(ipmt))->Eval(adc[ipmt]);
143       AliDebug(10,Form("  Amlitude in MIPS LED %f ,  QTC %f in channels %i\n ",ampMip,qtMip, adc[ipmt]));
144       
145       frecpoints->SetTime(ipmt, Float_t(time[ipmt]) );
146       frecpoints->SetAmp(ipmt, Float_t( ampMip)); //for cosmic &pp beam 
147       frecpoints->SetAmpLED(ipmt, Float_t(qtMip));
148       
149     }
150     else {
151       time[ipmt] = 0;
152       adc[ipmt] = 0;
153     }
154   }
155   
156   for (Int_t ipmt=0; ipmt<12; ipmt++){
157     if(time[ipmt] > 1 ) {
158       if(time[ipmt]<besttimeC){
159         besttimeC=time[ipmt]; //timeC
160         pmtBestC=ipmt;
161       }
162     }
163   }
164   for ( Int_t ipmt=12; ipmt<24; ipmt++){
165     if(time[ipmt] > 1) {
166       if(time[ipmt]<besttimeA) {
167         besttimeA=time[ipmt]; //timeA
168         pmtBestA=ipmt;}
169     }
170   }
171   if(besttimeA < 999999) {
172     frecpoints->SetTimeBestA(Int_t(besttimeA *channelWidth));
173     tr[1]=true;
174   }
175   if( besttimeC < 999999 ) {
176     frecpoints->SetTimeBestC(Int_t(besttimeC *channelWidth));
177     tr[2]=true;
178   }
179   AliDebug(10,Form(" besttimeA %f ch,  besttimeC %f ch",besttimeA, besttimeC));
180   if(besttimeA <999999 && besttimeC < 999999 ){
181     //    timeDiff = (besttimeC - besttimeA)*channelWidth;
182     timeDiff = (besttimeA - besttimeC)*channelWidth;
183     meanTime = (besttimeA + besttimeC)/2;// * channelWidth); 
184     timeclock = meanTime *channelWidth ;
185     vertex = meanVertex - c*(timeDiff)/2.;// + (fdZonA - fdZonC)/2;
186     tr[0]=true; 
187   }
188   frecpoints->SetVertex(vertex);
189   frecpoints->SetMeanTime(meanTime);
190   frecpoints->SetT0clock(timeclock);
191   frecpoints->SetT0Trig(tr);
192
193   for (Int_t i=0; i<5; i++) {
194     printf(" T0 trigers %i ",tr[i]);
195   }
196     printf(" \n ");
197
198   //online mean
199   frecpoints->SetOnlineMean(Int_t(onlineMean));
200   AliDebug(10,Form("  timeDiff %i #channel,  meanTime %i #channel, vertex %f cm online mean %i timeclock %i ps",timeDiff, meanTime,vertex, Int_t(onlineMean), timeclock));
201   
202   
203
204    
205   
206   clustersTree->Fill();
207
208   delete timeCFD;
209   delete timeLED;
210   delete chargeQT0; 
211   delete chargeQT1; 
212 }
213
214
215 //_______________________________________________________________________
216
217 void AliT0Reconstructor::Reconstruct(AliRawReader* rawReader, TTree*recTree) const
218 {
219   // T0 raw ->
220   //
221   // reference amplitude and time ref. point from reco param
222
223   Int_t refAmp = GetRecoParam()->GetRefAmp();
224   Int_t refPoint = GetRecoParam()->GetRefPoint();
225   Float_t latencyL1 = GetRecoParam()->GetLatencyL1();
226   Float_t latencyHPTDC = GetRecoParam()->GetLatencyHPTDC();
227   Float_t vertexshift = GetRecoParam()->GetVertexShift();
228
229   Int_t allData[110][5];
230   
231   Int_t timeCFD[24], timeLED[24], chargeQT0[24], chargeQT1[24];
232   Double32_t timeDiff=999999, meanTime=999999, timeclock=999999;
233   Float_t c = 29.9792458; // cm/ns
234   Double32_t vertex = 9999999;
235   Int_t onlineMean=0;
236
237   for (Int_t i0=0; i0<105; i0++)
238     {
239       for (Int_t j0=0; j0<5; j0++) allData[i0][j0]=0;   
240     }
241    
242   Double32_t besttimeA=9999999;
243   Double32_t besttimeC=9999999;
244   Int_t pmtBestA=99999;
245   Int_t pmtBestC=99999;
246   Float_t meanVertex = fParam->GetMeanVertex();
247    
248   AliT0RecPoint* frecpoints= new AliT0RecPoint ();
249   
250   recTree->Branch( "T0", "AliT0RecPoint" ,&frecpoints, 405,1);
251    
252   AliDebug(10," before read data ");
253   AliT0RawReader myrawreader(rawReader);
254
255   UInt_t type =rawReader->GetType();
256
257   if (!myrawreader.Next())
258     AliDebug(1,Form(" no raw data found!!"));
259   else
260     {  
261       if(type == 7) {  //only physics 
262       for (Int_t i=0; i<105; i++) {
263         for (Int_t iHit=0; iHit<5; iHit++) 
264           {
265             allData[i][iHit] = myrawreader.GetData(i,iHit);
266           }
267       }
268       Int_t ref=0;
269       if (refPoint>0) 
270       ref = allData[refPoint][0]-5000;
271
272       Float_t channelWidth = fParam->GetChannelWidth() ;  
273       
274       //       Int_t meanT0 = fParam->GetMeanT0();
275        
276           
277           for (Int_t in=0; in<12; in++)  
278             {
279               timeCFD[in] = allData[in+1][0] ;
280               timeCFD[in+12] = allData[in+56+1][0] ;
281               timeLED[in] = allData[in+12+1][0] ;
282               timeLED[in+12] = allData[in+68+1][0] ;
283               AliDebug(10, Form(" readed i %i cfdC %i cfdA %i ledC %i ledA%i ",
284                                 in, timeCFD[in],timeCFD[in+12],timeLED[in], 
285                                 timeLED[in+12]));   
286             }
287           
288           for (Int_t in=0; in<12;  in++)
289             {
290               chargeQT0[in]=allData[2*in+25][0];
291               chargeQT1[in]=allData[2*in+26][0];
292             }
293           
294            for (Int_t in=12; in<24;  in++)
295              {
296                chargeQT0[in]=allData[2*in+57][0];
297                chargeQT1[in]=allData[2*in+58][0];
298              }
299            
300            //    } //cosmic with physics event
301        for (Int_t in=0; in<24; in++)  
302          AliDebug(10, Form(" readed Raw %i %i %i %i %i",
303                            in, timeLED[in],timeCFD[in],chargeQT0[in],chargeQT1[in]));
304         onlineMean = allData[49][0];       
305        
306        Double32_t time[24], adc[24];
307        for (Int_t ipmt=0; ipmt<24; ipmt++) {
308          if(timeCFD[ipmt]>0 && timeLED[ipmt]>0){
309            //for simulated data
310              //for physics  data
311            if(( chargeQT1[ipmt] - chargeQT0[ipmt])>0)  
312              adc[ipmt] = chargeQT0[ipmt] - chargeQT1[ipmt];
313            else
314              adc[ipmt] = 0;
315            
316
317            //      time[ipmt] = fCalib-> WalkCorrection(refAmp, ipmt, adc[ipmt], timeCFD[ipmt] ) ;
318            
319            Double_t sl = timeLED[ipmt] - timeCFD[ipmt];
320            time[ipmt] = fCalib-> WalkCorrection( refAmp,ipmt, Int_t(sl), timeCFD[ipmt] ) ;
321            AliDebug(10,Form(" ipmt %i QTC %i , time in chann %i (led-cfd) %i ",
322                             ipmt, Int_t(adc[ipmt]) ,Int_t(time[ipmt]),Int_t( sl)));
323            Double_t ampMip =( (TGraph*)fAmpLED.At(ipmt))->Eval(sl);
324            Double_t qtMip = ((TGraph*)fQTC.At(ipmt))->Eval(adc[ipmt]);
325            AliDebug(10,Form("  Amlitude in MIPS LED %f ; QTC %f;  in channels %i\n ",ampMip,qtMip, adc[ipmt]));
326              
327            //      frecpoints->SetTime(ipmt, Float_t(time[ipmt]) );
328            frecpoints->SetTime(ipmt,Double32_t(timeCFD[ipmt]));
329            frecpoints->SetAmpLED(ipmt, Double32_t( qtMip)); //for cosmic &pp beam 
330            frecpoints->SetAmp(ipmt, Double32_t(ampMip));
331              
332          }
333          else {
334            time[ipmt] = 0;
335            adc[ipmt] = 0;
336          }
337        }
338        
339        for (Int_t ipmt=0; ipmt<12; ipmt++){
340          if(time[ipmt] > 1 ) {
341            if(time[ipmt]<besttimeC){
342              besttimeC=time[ipmt]; //timeC
343              pmtBestC=ipmt;
344            }
345          }
346        }
347        for ( Int_t ipmt=12; ipmt<24; ipmt++){
348          if(time[ipmt] > 1) {
349            if(time[ipmt]<besttimeA) {
350              besttimeA=time[ipmt]; //timeA
351              pmtBestA=ipmt;}
352          }
353        }
354        if(besttimeA < 999999) 
355          frecpoints->SetTimeBestA( 0.001 * besttimeA * channelWidth - latencyHPTDC + latencyL1);
356        if( besttimeC < 999999 ) 
357          frecpoints->SetTimeBestC(0.001 * besttimeC * channelWidth - latencyHPTDC + latencyL1);
358        AliDebug(10,Form(" pmtA %i besttimeA %f ps, pmtC %i besttimeC %f ps",
359                        pmtBestA,besttimeA, pmtBestC,  besttimeC));
360         if(besttimeA <999999 && besttimeC < 999999 ){
361          timeDiff = ( besttimeA - besttimeC) *0.001 * channelWidth;
362          timeclock =  0.001 * channelWidth * Float_t(besttimeA+besttimeC)/2. - latencyHPTDC + latencyL1;  
363          meanTime = (besttimeA+besttimeC-2.*Float_t(ref))/2.;
364          vertex =  meanVertex -  c*(timeDiff)/2. + vertexshift; //+ (fdZonA - fdZonC)/2; 
365         }
366       }  //if phys event       
367       AliDebug(5,Form("  timeDiff %f #channel,  meanTime %f #channel, TOFmean%f  vertex %f cm meanVertex %f online mean %i \n",timeDiff, meanTime,timeclock, vertex,meanVertex, onlineMean));
368       frecpoints->SetT0clock(timeclock);
369       frecpoints->SetVertex(vertex);
370       frecpoints->SetMeanTime(meanTime);
371       frecpoints->SetOnlineMean(Int_t(onlineMean));
372         // Set triggers
373       
374       Bool_t tr[5];
375       Int_t trchan[5]= {50,51,52,55,56};
376       for (Int_t i=0; i<5; i++) tr[i]=false; 
377       for (Int_t itr=0; itr<5; itr++) {
378         if(allData[trchan[itr]][0]>0) tr[itr]=true;
379         frecpoints->SetT0Trig(tr);
380       }
381       for (Int_t i=0; i<5; i++) 
382         printf(" T0 trigers %i ",tr[i]);
383       printf(" \n ");
384       
385       
386       } // if (else )raw data
387       recTree->Fill();
388       if(frecpoints) delete frecpoints;
389     }
390   
391   
392   //____________________________________________________________
393   
394   void AliT0Reconstructor::FillESD(TTree */*digitsTree*/, TTree *clustersTree, AliESDEvent *pESD) const
395   {
396
397   /***************************************************
398   Resonstruct digits to vertex position
399   ****************************************************/
400   
401   AliDebug(1,Form("Start FillESD T0"));
402   Float_t channelWidth = 0.0244;
403   Float_t c = 29.9792458; // cm/ns
404   Float_t currentVertex=0, shift=0;
405   Int_t ncont=0;
406   const AliESDVertex* vertex = pESD->GetPrimaryVertex();
407   if (!vertex)        vertex = pESD->GetPrimaryVertexSPD();
408   if (!vertex)        vertex = pESD->GetPrimaryVertexTPC();
409   if (!vertex)        vertex = pESD->GetVertex();
410
411   if (vertex) {
412     AliDebug(2, Form("Got %s (%s) from ESD: %f", 
413                     vertex->GetName(), vertex->GetTitle(), vertex->GetZ()));
414     currentVertex = vertex->GetZ();
415     
416     ncont = vertex->GetNContributors();
417     // cout<<" spdver "<<spdver<<" ncont "<<ncont<<endl;
418     if(ncont>3 ) {
419       //     hVertexSPD->Fill(spdver);
420       shift = currentVertex/c;
421       //          cout<<" vertex shif "<<shift<<" vertex "<<spdver<<" IsFromVertexer3D  "<<fverSPD->IsFromVertexer3D()<<endl;
422     }
423   }
424   TTree *treeR = clustersTree;
425   
426    AliT0RecPoint* frecpoints= new AliT0RecPoint ();
427     if (!frecpoints) {
428     AliError("Reconstruct Fill ESD >> no recpoints found");
429     return;
430   }
431   
432   AliDebug(1,Form("Start FillESD T0"));
433   TBranch *brRec = treeR->GetBranch("T0");
434   if (brRec) {
435     brRec->SetAddress(&frecpoints);
436   }else{
437     AliError(Form("EXEC Branch T0 rec not found"));
438     return;
439   } 
440     
441     brRec->GetEntry(0);
442     Double32_t amp[24], time[24];  
443     Double32_t timeClock[3];
444     Double32_t zPosition = frecpoints -> GetVertex();
445     Double32_t timeStart = frecpoints -> GetMeanTime();
446     timeClock[0] = frecpoints -> GetT0clock() ;
447     timeClock[1] = frecpoints -> GetBestTimeA() + shift;
448     timeClock[2] = frecpoints -> GetBestTimeC() - shift;
449     for ( Int_t i=0; i<24; i++) {
450       time[i] =  frecpoints -> GetTime(i); // ps to ns
451       amp[i] = frecpoints -> GetAmp(i);
452     }
453     Int_t trig= frecpoints ->GetT0Trig();
454     pESD->SetT0Trig(trig);
455     
456     pESD->SetT0zVertex(zPosition); //vertex Z position 
457     pESD->SetT0(timeStart);        // interaction time 
458     for(Int_t i=0; i<3; i++) 
459       pESD->SetT0TOF(i,timeClock[i]);   // interaction time (ns) 
460     pESD->SetT0time(time);         // best TOF on each PMT 
461     pESD->SetT0amplitude(amp);     // number of particles(MIPs) on each PMT
462     
463     AliDebug(1,Form("T0: Vertex %f (T0A+T0C)/2 %f #channels T0signal %f ns OrA %f ns OrC %f T0trig %i\n",zPosition, timeStart, timeClock[0], timeClock[1], timeClock[2], trig));
464     
465     
466 } // vertex in 3 sigma
467
468
469
470
471
472