Track reconstruction code for TOF: updating
[u/mrichter/AliRoot.git] / TOF / AliTOFpidESD.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 //-----------------------------------------------------------------
17 //           Implementation of the TOF PID class
18 // Very naive one... Should be made better by the detector experts...
19 //      Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
20 //-----------------------------------------------------------------
21 #include "TFile.h"
22 #include "TTree.h"
23 #include "TClonesArray.h"
24 #include "TError.h"
25
26 #include "AliTOFpidESD.h"
27 #include "AliESD.h"
28 #include "AliESDtrack.h"
29 #include "AliTOFdigit.h"
30 #include "AliTOFGeometry.h"
31 #include <Riostream.h>
32
33 #include <stdlib.h>
34
35
36 ClassImp(AliTOFpidESD)
37
38 //_________________________________________________________________________
39 AliTOFpidESD::AliTOFpidESD(Double_t *param) {
40   //
41   //  The main constructor
42   //
43   fN=0; fEventN=0;
44
45   fSigma=param[0];
46   fRange=param[1];
47
48 }
49
50 //_________________________________________________________________________
51 Int_t AliTOFpidESD::MakePID(AliESD *event)
52 {
53   //
54   //  This function calculates the "detector response" PID probabilities
55   //                Just for a bare hint... 
56
57   static const Double_t kMasses[]={
58     0.000511, 0.105658, 0.139570, 0.493677, 0.938272, 1.875613
59   };
60
61   Int_t nPID=0;
62   Int_t ntrk=event->GetNumberOfTracks();
63   AliESDtrack **tracks=new AliESDtrack*[ntrk];
64
65   Int_t i;
66   for (i=0; i<ntrk; i++) {
67     AliESDtrack *t=event->GetTrack(i);
68     tracks[i]=t;
69   }
70
71   for (i=0; i<ntrk; i++) {
72     AliESDtrack *t=tracks[i];
73     if ((t->GetStatus()&AliESDtrack::kTOFout)==0) continue;
74     if ((t->GetStatus()&AliESDtrack::kTIME)==0) continue;
75     Double_t tof=t->GetTOFsignal();
76     Double_t time[10]; t->GetIntegratedTimes(time);
77     Double_t p[10];
78     Double_t mom=t->GetP();
79     nPID++;
80     for (Int_t j=0; j<AliESDtrack::kSPECIES; j++) {
81       Double_t mass=kMasses[j];
82       Double_t dpp=0.01;      //mean relative pt resolution;
83       if (mom>0.5) dpp=0.01*mom;
84       Double_t sigma=dpp*time[j]/(1.+ mom*mom/(mass*mass));
85       sigma=TMath::Sqrt(sigma*sigma + fSigma*fSigma);
86       if (TMath::Abs(tof-time[j]) > fRange*sigma) {
87         p[j]=TMath::Exp(-0.5*fRange*fRange)/sigma;
88         continue;
89       }
90       p[j]=TMath::Exp(-0.5*(tof-time[j])*(tof-time[j])/(sigma*sigma))/sigma;
91     }
92     t->SetTOFpid(p);
93   }
94
95   delete[] tracks;
96   
97   return 0;
98 }
99