]>
Commit | Line | Data |
---|---|---|
998b831f | 1 | #include "AliRICHTracker.h" |
2 | #include <AliESD.h> | |
3 | #include <TVector3.h> | |
4 | #include <TTree.h> | |
5 | #include "AliRICH.h" | |
6 | #include "AliRICHHelix.h" | |
7 | #include <AliMagF.h> | |
8 | #include "AliRICHRecon.h" | |
9 | #include <AliStack.h> | |
10 | #include <TParticle.h> | |
30c60010 | 11 | #include <TMath.h> |
0fe8fa07 | 12 | #include <AliRun.h> |
998b831f | 13 | ClassImp(AliRICHTracker) |
30c60010 | 14 | //__________________________________________________________________________________________________ |
998b831f | 15 | Int_t AliRICHTracker::PropagateBack(AliESD *pESD) |
16 | { | |
a25b3368 | 17 | // Interface callback methode invoked by AliRecontruction during tracking after TOF |
18 | // It steers to different way to provide the final reconstructed information sutable for analisys: | |
19 | // 1. AliESD - reconstructed tracks are used | |
20 | // 2. RICH private ntuple for debug- stack particles used instead of reconstructed tracks | |
30c60010 | 21 | AliDebug(1,"Start pattern recognition"); |
910735ae | 22 | if(pESD->GetNumberOfTracks()) { |
23 | Int_t iNtracks=pESD->GetNumberOfTracks(); | |
24 | AliDebug(1,Form("Start with %i tracks",iNtracks)); | |
25 | AliRICH *pRich=((AliRICH*)gAlice->GetDetector("RICH")); | |
26 | for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){//ESD tracks loop | |
27 | RecWithESD(pESD,pRich,iTrackN); | |
28 | } | |
29 | } | |
30 | else RecWithStack(0); | |
30c60010 | 31 | AliDebug(1,"Stop pattern recognition"); |
30c60010 | 32 | return 0; // error code: 0=no error; |
a25b3368 | 33 | }//PropagateBack() |
30c60010 | 34 | //__________________________________________________________________________________________________ |
910735ae | 35 | void AliRICHTracker::RecWithESD(AliESD *pESD,AliRICH *pRich,Int_t iTrackN) |
30c60010 | 36 | { |
d3eb6079 | 37 | //recontruction from ESD- primary way to reconstruct particle ID signal from tracks provided by core detectors |
fab9e039 | 38 | fnPhotBKG = 0; |
d3eb6079 | 39 | |
910735ae | 40 | Double_t fField=GetFieldMap()->SolenoidField()/10;// magnetic field in Tesla |
998b831f | 41 | AliESDtrack *pTrack = pESD->GetTrack(iTrackN);// get next reconstructed track |
42 | // if((pTrack->GetStatus()&AliESDtrack::kTOFout)==0) continue; //ignore tracks not recontructed by TOF | |
d3eb6079 | 43 | // pTrack->GetXYZ(xb); |
44 | // pTrack->GetPxPyPz(pb); | |
1cb5a97c | 45 | Int_t status=pTrack->GetStatus()&AliESDtrack::kTOFout;//get running track parameters |
910735ae | 46 | Int_t charge = (Int_t)(-TMath::Sign(1.,pTrack->GetSign()*fField)); |
39853df5 | 47 | AliDebug(1,Form("Track %i pmod=%f charge=%i stat=%i",iTrackN,pTrack->GetP(),charge,status)); |
910735ae | 48 | AliRICHHelix helix(pTrack->X3(),pTrack->P3(),charge,fField); |
998b831f | 49 | Int_t iChamber=helix.RichIntersect(pRich->P()); |
50 | AliDebug(1,Form("intersection with %i chamber found",iChamber)); | |
01c81d9d | 51 | if(!iChamber) { |
52 | pTrack->SetRICHsignal(-999); //to be improved by flags... | |
53 | return;//intersection with no chamber found | |
54 | } | |
55 | //find MIP cluster candidate (cluster which is closest to track intersection point) | |
d3eb6079 | 56 | Double_t distMip=9999,distX=0,distY=0; //min distance between clusters and track position on PC |
f770edb5 | 57 | Int_t iMipId=0; //index of that min distance cluster |
58 | Double_t chargeMip=0; //charge of the MIP | |
01c81d9d | 59 | Bool_t kFound = kFALSE; |
998b831f | 60 | for(Int_t iClusN=0;iClusN<pRich->Clusters(iChamber)->GetEntries();iClusN++){//clusters loop for intersected chamber |
0fe8fa07 | 61 | AliRICHCluster *pClus=(AliRICHCluster*)pRich->Clusters(iChamber)->UncheckedAt(iClusN);//get pointer to current cluster |
01c81d9d | 62 | if(pClus->Q()<AliRICHParam::QthMIP()) continue; |
d3eb6079 | 63 | Double_t distCurrent=pClus->DistTo(helix.PosPc());//distance between current cluster and helix intersection with PC |
a25b3368 | 64 | if(distCurrent<distMip){ |
01c81d9d | 65 | kFound = kTRUE; |
a25b3368 | 66 | distMip=distCurrent; |
67 | iMipId=iClusN; | |
d3eb6079 | 68 | distX=pClus->DistX(helix.PosPc()); |
69 | distY=pClus->DistY(helix.PosPc()); | |
f770edb5 | 70 | chargeMip=pClus->Q(); |
a25b3368 | 71 | }//find cluster nearest to the track |
998b831f | 72 | AliDebug(1,Form("Ploc (%f,%f,%f) dist= %f",helix.Ploc().Mag(),helix.Ploc().Theta()*TMath::RadToDeg(), |
a25b3368 | 73 | helix.Ploc().Phi()*TMath::RadToDeg(),pClus->DistTo(helix.PosPc()))); |
74 | }//clusters loop for intersected chamber | |
01c81d9d | 75 | |
76 | if(!kFound) { | |
77 | pTrack->SetRICHsignal(-999); | |
78 | return; | |
79 | } | |
998b831f | 80 | AliDebug(1,Form("Min distance cluster: %i dist is %f",iMipId,distMip)); |
01c81d9d | 81 | pTrack->SetRICHcluster(((Int_t)chargeMip)+1000000*iChamber); |
82 | pTrack->SetRICHdxdy(distX,distY); | |
83 | pTrack->SetRICHthetaPhi(helix.Ploc().Theta(),helix.Ploc().Phi()); | |
f770edb5 | 84 | // |
85 | // HERE CUTS ON GOLD RINGS.... | |
86 | // | |
01c81d9d | 87 | if(distMip>AliRICHParam::DmatchMIP()) { |
f770edb5 | 88 | //track not accepted for pattern recognition |
01c81d9d | 89 | pTrack->SetRICHsignal(-990); //to be improved by flags... |
910735ae | 90 | return; |
f770edb5 | 91 | } |
92 | // | |
d3eb6079 | 93 | AliRICHRecon recon(&helix,pRich->Clusters(iChamber),iMipId); //actual job is done there |
101624cd | 94 | |
95 | Double_t thetaCerenkov=recon.ThetaCerenkov(); //search for mean Cerenkov angle for this track | |
d3eb6079 | 96 | |
101624cd | 97 | pTrack->SetRICHsignal(thetaCerenkov); |
d3eb6079 | 98 | pTrack->SetRICHnclusters(recon.GetHoughPhotons()); |
fab9e039 | 99 | |
100 | fnPhotBKG = recon.GetPhotBKG(); | |
101 | ||
d3eb6079 | 102 | AliDebug(1,Form("FINAL Theta Cerenkov=%f",pTrack->GetRICHsignal())); |
101624cd | 103 | // |
104 | if(pTrack->GetRICHsignal()>0) { | |
105 | AliDebug(1,Form("Start to assign the probabilities")); | |
106 | Double_t sigmaPID[AliPID::kSPECIES]; | |
107 | Double_t richPID[AliPID::kSPECIES]; | |
108 | for (Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++) { | |
109 | sigmaPID[iPart] = 0; | |
068217e3 | 110 | fErrPar[iPart] = 0; |
101624cd | 111 | for(Int_t iphot=0;iphot<pRich->Clusters(iChamber)->GetEntries();iphot++) { |
112 | recon.SetPhotonIndex(iphot); | |
113 | if(recon.GetPhotonFlag() == 2) { | |
fab9e039 | 114 | Double_t theta_g=recon.GetTrackTheta(); |
115 | Double_t phi_g=(recon.GetPhiPoint()-recon.GetTrackPhi()); | |
01c81d9d | 116 | Double_t sigma2 = AliRICHParam::SigmaSinglePhoton(iPart,pTrack->GetP(),theta_g,phi_g).Mag2(); |
117 | if(sigma2>0) sigmaPID[iPart] += 1/sigma2; | |
101624cd | 118 | } |
119 | } | |
d1be0d8a | 120 | if (sigmaPID[iPart]>0) |
01c81d9d | 121 | sigmaPID[iPart] *= (Double_t)(recon.GetHoughPhotons()-fnPhotBKG)/(Double_t)(recon.GetHoughPhotons()); // n total phots, m are background...the sigma are scaled.. |
122 | if(sigmaPID[iPart]>0) sigmaPID[iPart] = 1/TMath::Sqrt(sigmaPID[iPart])*0.001; // sigma from parametrization are in mrad... | |
123 | else sigmaPID[iPart] = 0; | |
124 | fErrPar[iPart]=sigmaPID[iPart]; | |
101624cd | 125 | AliDebug(1,Form("sigma for %s is %f rad",AliPID::ParticleName(iPart),sigmaPID[iPart])); |
126 | } | |
127 | CalcProb(thetaCerenkov,pTrack->GetP(),sigmaPID,richPID); | |
128 | pTrack->SetRICHpid(richPID); | |
129 | AliDebug(1,Form("PROBABILITIES ---> %f - %f - %f - %f - %f",richPID[0],richPID[1],richPID[2],richPID[3],richPID[4])); | |
910735ae | 130 | } |
998b831f | 131 | AliDebug(1,"Stop."); |
30c60010 | 132 | } //RecWithESD |
133 | //__________________________________________________________________________________________________ | |
1cb5a97c | 134 | void AliRICHTracker::RecWithStack(TNtupleD *hn) |
30c60010 | 135 | { |
d3eb6079 | 136 | //Reconstruction for particles from STACK. This methode is to be used for RICH standalone when no other detectors are switched on, |
137 | //so normal tracking is not available | |
138 | AliDebug(1,"Start."); | |
30c60010 | 139 | AliRICH *pRich=((AliRICH*)gAlice->GetDetector("RICH")); |
140 | ||
1cb5a97c | 141 | // pRich->GetLoader()->GetRunLoader()->LoadHeader(); |
d3eb6079 | 142 | if(!pRich->GetLoader()->GetRunLoader()->TreeK()) pRich->GetLoader()->GetRunLoader()->LoadKinematics(); |
30c60010 | 143 | AliStack *pStack = pRich->GetLoader()->GetRunLoader()->Stack(); |
39853df5 | 144 | if(!pStack) {AliDebug(1,Form("No STACK found in AliRoot"));return;} |
30c60010 | 145 | Int_t iNtracks=pStack->GetNtrack(); |
1cb5a97c | 146 | AliDebug(1,Form(" Start reconstruction with %i track(s) from Stack",iNtracks)); |
147 | ||
ab92fbe6 | 148 | Double_t hnvec[20]; |
1cb5a97c | 149 | |
30c60010 | 150 | Double_t b=GetFieldMap()->SolenoidField()/10;// magnetic field in Tesla |
151 | AliDebug(1,Form("Start with simulated %i tracks in %f Tesla field",iNtracks,b)); | |
152 | TVector3 x0(0,0,0); TVector3 p0(0,0,0);//tmp storage for AliRICHHelix | |
153 | ||
154 | ||
1cb5a97c | 155 | if(pRich->GetLoader()->LoadRecPoints()) {AliDebug(1,Form("No clusters found in RICH"));return;} |
156 | pRich->GetLoader()->TreeR()->GetEntry(0); | |
157 | ||
a25b3368 | 158 | for(Int_t iTrackN=0;iTrackN<iNtracks;iTrackN++){//stack particles loop |
30c60010 | 159 | TParticle *pParticle = pStack->Particle(iTrackN); |
39853df5 | 160 | if(!pParticle) {AliDebug(1,Form("Not a valid TParticle pointer. Track skipped"));continue;} |
161 | AliDebug(1,Form(" PDG code : %i",pParticle->GetPdgCode())); | |
ab92fbe6 | 162 | // |
163 | // problem of PDG code of some extra particles to be solved!!!!!!!!! | |
164 | // | |
165 | // found problem! Look in TRD directory : codes from Fluka are : | |
166 | // | |
167 | // if ((pdg_code == 10010020) || | |
168 | // (pdg_code == 10010030) || | |
169 | // (pdg_code == 50000050) || | |
170 | // (pdg_code == 50000051) || | |
171 | // (pdg_code == 10020040)) { | |
172 | // | |
173 | if(pParticle->GetPdgCode()>=50000050||pParticle->GetPdgCode()==0||pParticle->GetPdgCode()>10000) {AliDebug(1,Form("A photon as track... Track skipped"));continue;} | |
174 | // | |
175 | // to be updated for us!! | |
176 | // | |
9f873529 | 177 | AliDebug(1,Form("Track %i is a %s with charge %i and momentum %f", |
178 | iTrackN,pParticle->GetPDG()->GetName(),Int_t(pParticle->GetPDG()->Charge()),pParticle->P())); | |
179 | // if(pParticle->GetMother(0)!=-1) continue; //consider only primaries | |
180 | if(pParticle->GetPDG()->Charge()==0||TMath::Abs(Int_t(pParticle->GetPDG()->Charge()))!=3) continue; //to avoid photons from stack... | |
1cb5a97c | 181 | hnvec[0]=pParticle->P(); |
182 | hnvec[1]=pParticle->GetPDG()->Charge(); | |
30c60010 | 183 | p0.SetMagThetaPhi(pParticle->P(),pParticle->Theta(),pParticle->Phi()); |
184 | x0.SetXYZ(pParticle->Vx(),pParticle->Vy(),pParticle->Vz()); | |
9f873529 | 185 | AliRICHHelix helix(x0,p0,TMath::Sign(1,(Int_t)pParticle->GetPDG()->Charge()),b); |
30c60010 | 186 | Int_t iChamber=helix.RichIntersect(pRich->P()); |
ef210ba6 | 187 | hnvec[2]=helix.Ploc().Theta(); |
188 | hnvec[3]=helix.Ploc().Phi(); | |
30c60010 | 189 | AliDebug(1,Form("intersection with %i chamber found",iChamber)); |
1cb5a97c | 190 | if(!iChamber) continue;// no intersection with RICH found |
191 | hnvec[4]=helix.PosPc().X(); | |
192 | hnvec[5]=helix.PosPc().Y(); | |
193 | Double_t distMip=9999; //min distance between clusters and track position on PC | |
0fe8fa07 | 194 | Double_t mipX=-1; //min distance between clusters and track position on PC |
195 | Double_t mipY=-1; //min distance between clusters and track position on PC | |
196 | Double_t chargeMip=-1; // charge MIP to find | |
197 | Int_t iMipId=-1; //index of that min distance cluster | |
30c60010 | 198 | for(Int_t iClusN=0;iClusN<pRich->Clusters(iChamber)->GetEntries();iClusN++){//clusters loop for intersected chamber |
0fe8fa07 | 199 | AliRICHCluster *pClus=(AliRICHCluster*)pRich->Clusters(iChamber)->UncheckedAt(iClusN);//get pointer to current cluster |
30c60010 | 200 | Double_t distCurrent=pClus->DistTo(helix.PosPc());//ditance between current cluster and helix intersection with PC |
1cb5a97c | 201 | if(distCurrent<distMip){distMip=distCurrent;mipX=pClus->X(); |
202 | mipY=pClus->Y(); | |
203 | chargeMip=pClus->Q();iMipId=1000000*iChamber+iClusN;}//find cluster nearest to the track | |
30c60010 | 204 | |
205 | AliDebug(1,Form("Ploc (%f,%f,%f) dist= %f",helix.Ploc().Mag(),helix.Ploc().Theta()*TMath::RadToDeg(), | |
206 | helix.Ploc().Phi()*TMath::RadToDeg(),pClus->DistTo(helix.PosPc()))); | |
a25b3368 | 207 | }//clusters loop for intersected chamber |
30c60010 | 208 | |
209 | AliDebug(1,Form("Min distance cluster: %i dist is %f",iMipId,distMip)); | |
1cb5a97c | 210 | hnvec[6]=mipX;hnvec[7]=mipY; |
211 | hnvec[8]=chargeMip; | |
30c60010 | 212 | AliRICHRecon recon(&helix,pRich->Clusters(iChamber),iMipId); |
213 | Double_t thetaCerenkov=recon.ThetaCerenkov(); //search for mean Cerenkov angle for this track | |
1cb5a97c | 214 | hnvec[9]=thetaCerenkov; |
215 | hnvec[10]=recon.GetHoughPhotons(); | |
216 | hnvec[11]=(Double_t)iMipId; | |
ab92fbe6 | 217 | hnvec[12]=(Double_t)iChamber; |
218 | hnvec[13]=(Double_t)pParticle->GetPdgCode(); | |
1cb5a97c | 219 | if(hn) hn->Fill(hnvec); |
30c60010 | 220 | AliDebug(1,Form("FINAL Theta Cerenkov=%f",thetaCerenkov)); |
a25b3368 | 221 | }//stack particles loop |
1cb5a97c | 222 | |
39853df5 | 223 | pRich->GetLoader()->UnloadRecPoints(); |
30c60010 | 224 | AliDebug(1,"Stop."); |
a25b3368 | 225 | }//RecWithStack |
226 | //__________________________________________________________________________________________________ | |
998b831f | 227 | Int_t AliRICHTracker::LoadClusters(TTree *pTree) |
228 | { | |
229 | // Load clusters for RICH | |
230 | AliDebug(1,"Start."); pTree->GetEntry(0); AliDebug(1,"Stop."); return 0; | |
231 | } | |
30c60010 | 232 | //__________________________________________________________________________________________________ |
101624cd | 233 | void AliRICHTracker::CalcProb(Double_t thetaCer,Double_t pmod, Double_t *sigmaPID, Double_t *richPID) |
30c60010 | 234 | { |
12b0e5f4 | 235 | // Calculates probability to be a electron-muon-pion-kaon-proton |
a25b3368 | 236 | // from the given Cerenkov angle and momentum assuming no initial particle composition |
12b0e5f4 | 237 | // (i.e. apriory probability to be the particle of the given sort is the same for all sorts) |
304864ab | 238 | Double_t height[AliPID::kSPECIES];Double_t totalHeight=0; |
101624cd | 239 | Double_t thetaTh[AliPID::kSPECIES]; |
304864ab | 240 | for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++){ |
101624cd | 241 | height[iPart]=0; |
068217e3 | 242 | Double_t mass = AliRICHParam::fgMass[iPart]; |
101624cd | 243 | Double_t refIndex=AliRICHParam::RefIdxC6F14(AliRICHParam::MeanCkovEnergy()); |
30c60010 | 244 | Double_t cosThetaTh = TMath::Sqrt(mass*mass+pmod*pmod)/(refIndex*pmod); |
101624cd | 245 | thetaTh[iPart]=0; |
246 | if(cosThetaTh>=1) continue; | |
247 | thetaTh[iPart] = TMath::ACos(cosThetaTh); | |
248 | // Double_t sinThetaThNorm = TMath::Sin(thetaTh)/TMath::Sqrt(1-1/(refIndex*refIndex)); | |
249 | // Double_t sigmaThetaTh = (0.014*(1/sinThetaThNorm-1) + 0.0043)*1.25; | |
250 | // height[iPart] = TMath::Gaus(thetaCer,thetaTh,sigmaThetaTh); | |
01c81d9d | 251 | if(sigmaPID[iPart]>0) height[iPart] = TMath::Gaus(thetaCer,thetaTh[iPart],sigmaPID[iPart],kTRUE); |
252 | else height[iPart] = 0; | |
30c60010 | 253 | totalHeight +=height[iPart]; |
101624cd | 254 | AliDebug(1,Form(" Particle %s with mass %f with height %f and thetaTH %f",AliPID::ParticleName(iPart),mass,height[iPart],thetaTh[iPart])); |
12b0e5f4 | 255 | AliDebug(1,Form(" partial height %15.14f total height %15.14f",height[iPart],totalHeight)); |
30c60010 | 256 | } |
12b0e5f4 | 257 | if(totalHeight<1e-5) {for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++)richPID[iPart]=1.0/AliPID::kSPECIES;return;} |
101624cd | 258 | for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++) richPID[iPart] = height[iPart]/totalHeight; |
259 | Int_t iPartNear = TMath::LocMax(AliPID::kSPECIES,richPID); | |
01c81d9d | 260 | if(TMath::Abs(thetaCer-thetaTh[iPartNear])>5.*sigmaPID[iPartNear]) for(Int_t iPart=0;iPart<AliPID::kSPECIES;iPart++)richPID[iPart]=1.0/AliPID::kSPECIES; |
07625543 | 261 | //last line is to check if the nearest thetacerenkov to the teorethical one is within 5 sigma, otherwise no response (equal prob to every particle |
101624cd | 262 | |
30c60010 | 263 | }//CalcProb |
a25b3368 | 264 | //__________________________________________________________________________________________________ |