]>
Commit | Line | Data |
---|---|---|
1 | /************************************************************************** | |
2 | * Copyright(c) 2005-2007, 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 | // ITS PID class --- method # 2 // | |
20 | // // | |
21 | // // | |
22 | //The PID is based on the likelihood of all the four ITS' layers, // | |
23 | //without using the truncated mean for the dE/dx. The response // | |
24 | //functions for each layer are convoluted Landau-Gaussian functions. // | |
25 | // Origin: Elena Bruna bruna@to.infn.it, Massimo Masera masera@to.infn.it// | |
26 | //-----------------------------------------------------------------------// | |
27 | ||
28 | #include "AliITSpidESD2.h" | |
29 | #include "AliESDEvent.h" | |
30 | #include "AliESDtrack.h" | |
31 | #include "AliITStrackV2.h" | |
32 | #include "AliITSRecPoint.h" | |
33 | #include "AliITStrackerMI.h" | |
34 | #include "AliITSLoader.h" | |
35 | #include "AliITSPident.h" | |
36 | #include "AliITSSteerPid.h" | |
37 | #include "AliLog.h" | |
38 | ||
39 | ClassImp(AliITSpidESD2) | |
40 | //_________________________________________________________________________ | |
41 | AliITSpidESD2::AliITSpidESD2():AliITSpidESD(), | |
42 | fTracker(0), | |
43 | fSp(0) | |
44 | { // | |
45 | // The main constructor | |
46 | } | |
47 | //_________________________________________________________________________ | |
48 | AliITSpidESD2::AliITSpidESD2(AliITStrackerMI* tracker):AliITSpidESD(), | |
49 | fTracker(tracker), | |
50 | fSp(0) | |
51 | { // | |
52 | // The main constructor | |
53 | fSp=new AliITSSteerPid(); | |
54 | fSp->InitLayer(); | |
55 | } | |
56 | //_________________________________________________________________________ | |
57 | AliITSpidESD2::~AliITSpidESD2(){ | |
58 | //destructor | |
59 | ||
60 | if(fSp)delete fSp; | |
61 | ||
62 | } | |
63 | //______________________________________________________________________ | |
64 | AliITSpidESD2::AliITSpidESD2(const AliITSpidESD2 &ob) :AliITSpidESD(ob), | |
65 | fTracker(ob.fTracker), | |
66 | fSp(ob.fSp) | |
67 | { | |
68 | // Copy constructor | |
69 | ||
70 | } | |
71 | ||
72 | //______________________________________________________________________ | |
73 | AliITSpidESD2& AliITSpidESD2::operator=(const AliITSpidESD2& ob ){ | |
74 | // Assignment operator | |
75 | this->~AliITSpidESD2(); | |
76 | new(this) AliITSpidESD2(ob); | |
77 | return *this; | |
78 | } | |
79 | ||
80 | //_________________________________________________________________________ | |
81 | Int_t AliITSpidESD2::MakePID(TTree *clustersTree, AliESDEvent *event) | |
82 | { | |
83 | ||
84 | // | |
85 | // This function calculates the "detector response" PID probabilities | |
86 | // | |
87 | Double_t xr,par[5]; | |
88 | AliITStrackV2* track=0; | |
89 | fTracker->LoadClusters(clustersTree); | |
90 | printf("==== Landau Fit PID ITS ====== \n"); | |
91 | Int_t ntrk=event->GetNumberOfTracks(); | |
92 | Double_t momits; | |
93 | // for (Int_t i=0; i<ntrk; i++) { | |
94 | for (Int_t i=0; i<ntrk; i++) { | |
95 | AliESDtrack *esdtr=event->GetTrack(i); | |
96 | if ((esdtr->GetStatus()&AliESDtrack::kITSin )==0) | |
97 | if ((esdtr->GetStatus()&AliESDtrack::kITSout)==0) continue; | |
98 | ||
99 | track = new AliITStrackV2(*esdtr); | |
100 | Double_t dEdxsignal=track->GetdEdx(); | |
101 | track->GetExternalParameters(xr,par); | |
102 | if (par[4]!=0) { | |
103 | Float_t lamb=TMath::ATan(par[3]); | |
104 | momits=1/(TMath::Abs(par[4])*TMath::Cos(lamb)); | |
105 | } | |
106 | else { | |
107 | AliWarning("Null particle momentum in ITS"); | |
108 | momits = 0.; | |
109 | } | |
110 | Double_t snp=track->GetSnp(); | |
111 | Double_t tgl=track->GetTgl(); | |
112 | const Int_t kns=AliPID::kSPECIES; | |
113 | Double_t condprobfun[kns]; | |
114 | for(Int_t ii=0;ii<kns;ii++)condprobfun[ii]=0; | |
115 | Int_t cluind[12]; | |
116 | for(Int_t ii=0;ii<12;ii++){ | |
117 | cluind[ii]=track->GetClusterIndex(ii); | |
118 | } | |
119 | AliITSRecPoint* cluarr[12]; | |
120 | Float_t qclu[8],qclucorr[8],nlay[8]; | |
121 | for(Int_t ii=0;ii<8;ii++){ | |
122 | qclu[ii]=0; | |
123 | qclucorr[ii]=0; | |
124 | nlay[ii]=0; | |
125 | } | |
126 | Int_t jj=0; | |
127 | for(Int_t ij=0;ij<12;ij++){ | |
128 | cluind[ij]=track->GetClusterIndex(ij); | |
129 | if(cluind[ij]>0){ | |
130 | cluarr[ij]=(AliITSRecPoint*)fTracker->GetCluster(cluind[ij]); | |
131 | Int_t lay=cluarr[ij]->GetLayer(); | |
132 | if(lay>1){//sdd+ssd only | |
133 | qclu[jj]=cluarr[ij]->GetQ(); | |
134 | qclucorr[jj]=qclu[jj]*TMath::Sqrt((1-snp*snp)/(1+tgl*tgl)); | |
135 | nlay[jj]=lay; | |
136 | jj++; | |
137 | } | |
138 | else qclucorr[jj]=-1; | |
139 | } | |
140 | } | |
141 | ||
142 | Float_t prip=0.33; | |
143 | Float_t prik=0.33; | |
144 | Float_t pripi=0.33; | |
145 | Float_t prie=0.; | |
146 | AliITSPident mypid(momits,dEdxsignal,fSp,qclucorr,nlay,prip,prik,pripi,prie); | |
147 | condprobfun[0]=mypid.GetProdCondFunPi();//el --PID in the ITS does not distinguish among Pi,el,mu | |
148 | condprobfun[1]=mypid.GetProdCondFunPi();//mu | |
149 | condprobfun[2]=mypid.GetProdCondFunPi();//pi | |
150 | condprobfun[3]=mypid.GetProdCondFunK();//kaon | |
151 | condprobfun[4]=mypid.GetProdCondFunPro();//pro | |
152 | ||
153 | esdtr->SetITSpid(condprobfun); | |
154 | ||
155 | delete track; | |
156 | } | |
157 | fTracker->UnloadClusters(); | |
158 | return 0; | |
159 | } |