1 #include "AliHMPIDTracker.h" //class header
2 #include "AliHMPIDtrack.h" //class header
3 #include "AliHMPIDCluster.h" //GetTrackPoint(),PropagateBack()
4 #include "AliHMPIDParam.h" //GetTrackPoint(),PropagateBack()
5 #include "AliHMPIDPid.h" //Recon(),reconHTA()
6 #include "AliHMPIDRecon.h" //Recon()
7 #include "AliHMPIDRecoParamV1.h" //Recon()
8 #include "AliHMPIDReconstructor.h"//Recon()
9 #include "AliHMPIDReconHTA.h" //ReconHTA()
10 #include <AliLog.h> //Recon()
11 #include <AliESDEvent.h> //PropagateBack(),Recon()
12 #include <AliESDtrack.h> //Intersect()
13 #include <AliTracker.h>
14 #include <AliRun.h> //GetTrackPoint(),PropagateBack()
15 #include <AliTrackPointArray.h> //GetTrackPoint()
16 #include <AliAlignObj.h> //GetTrackPoint()
17 #include <AliCDBManager.h> //PropageteBack()
18 #include <AliCDBEntry.h> //PropageteBack()
20 // HMPID base class fo tracking
24 ClassImp(AliHMPIDTracker)
25 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
26 AliHMPIDTracker::AliHMPIDTracker():
28 fClu(new TObjArray(AliHMPIDParam::kMaxCh+1))
30 // ctor. Create TObjArray of TClonesArray of AliHMPIDCluster
33 fClu->SetOwner(kTRUE);
34 for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) fClu->AddAt(new TClonesArray("AliHMPIDCluster"),i);
36 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
37 Bool_t AliHMPIDTracker::GetTrackPoint(Int_t idx, AliTrackPoint& point) const
39 // Interface callback methode invoked from AliReconstruction::WriteAlignmentData() to get position of MIP cluster in MARS associated to a current track.
40 // MIP cluster is reffered by index which is stored in AliESDtrack ???????
41 // Arguments: idx- cluster index which is stored by HMPID in AliESDtrack
42 // point- reference to the object where to store the point
43 // Returns: status of operation if FALSE then AliReconstruction::WriteAlignmentData() do not store this point to array of points for current track.
44 if(idx<0) return kFALSE; //no MIP cluster assigned to this track in PropagateBack()
45 Int_t iCham=idx/1000000; Int_t iClu=idx%1000000;
46 iClu = iClu%1000; //GetHMPIDcluIdx -> 1e+6*ch + 1e+3*clusize + cluIdx;
47 point.SetVolumeID(AliGeomManager::LayerToVolUID(AliGeomManager::kHMPID,iCham));//layer and chamber number
48 TClonesArray *pArr=(TClonesArray*)(*fClu)[iCham];
49 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pArr->UncheckedAt(iClu);//get pointer to cluster
51 pClu->GetGlobalXYZ(xyz);
53 pClu->GetGlobalCov(cov);
54 point.SetXYZ(xyz,cov);
57 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
58 Int_t AliHMPIDTracker::IntTrkCha(AliESDtrack *pTrk,Float_t &xPc,Float_t &yPc,Float_t &xRa,Float_t &yRa,Float_t &theta,Float_t &phi)
60 // Static method to find intersection in between given track and HMPID chambers
61 // Arguments: pTrk- ESD track; xPc,yPc- track intersection with PC in LORS [cm]
62 // Returns: intersected chamber ID or -1
63 AliHMPIDtrack *hmpTrk = new AliHMPIDtrack(*pTrk); //create a hmpid track to be used for propagation and matching
64 for(Int_t i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++){ //chambers loop
65 Int_t chInt = IntTrkCha(i,hmpTrk,xPc,yPc,xRa,yRa,theta,phi);
66 if(chInt>=0) {delete hmpTrk;hmpTrk=0x0;return chInt;}
68 delete hmpTrk; hmpTrk=0x0;
69 return -1; //no intersection with HMPID chambers
71 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
72 Int_t AliHMPIDTracker::IntTrkCha(Int_t ch,AliHMPIDtrack *pTrk,Float_t &xPc,Float_t &yPc,Float_t &xRa,Float_t &yRa,Float_t &theta,Float_t &phi)
74 // Static method to find intersection in between given track and HMPID chambers
75 // Arguments: pTrk- HMPID track; xPc,yPc- track intersection with PC in LORS [cm]
76 // Returns: intersected chamber ID or -1
77 AliHMPIDParam *pParam=AliHMPIDParam::Instance();
80 pParam->Point(ch,p1,AliHMPIDParam::kRad); //point & norm for middle of radiator plane
83 pParam->Point(ch,p2,AliHMPIDParam::kPc); //point & norm for entrance to PC plane
84 if(pTrk->Intersect(p1,n1)==kFALSE) return -1; //try to intersect track with the middle of radiator
85 if(pTrk->Intersect(p2,n2)==kFALSE) return -1;
86 pParam->Mars2LorsVec(ch,n1,theta,phi); //track angles at RAD
87 pParam->Mars2Lors (ch,p1,xRa,yRa); //TRKxRAD position
88 pParam->Mars2Lors (ch,p2,xPc,yPc); //TRKxPC position
89 if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kTRUE) return ch; //return intersected chamber
90 return -1; //no intersection with HMPID chambers
92 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
93 Int_t AliHMPIDTracker::LoadClusters(TTree *pCluTree)
95 // Interface callback methode invoked from AliReconstruction::RunTracking() to load HMPID clusters before PropagateBack() gets control. Done once per event.
96 // Arguments: pCluTree- pointer to clusters tree got by AliHMPIDLoader::LoadRecPoints("read") then AliHMPIDLoader::TreeR()
97 // Returns: error code (currently ignored in AliReconstruction::RunTraking())
98 for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) pCluTree->SetBranchAddress(Form("HMPID%d",i),&((*fClu)[i]));
99 pCluTree->GetEntry(0);
102 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
103 Int_t AliHMPIDTracker::PropagateBack(AliESDEvent *pEsd)
105 // Interface pure virtual in AliTracker. Invoked from AliReconstruction::RunTracking() after invocation of AliTracker::LoadClusters() once per event
106 // Agruments: pEsd - pointer to ESD
107 // Returns: error code
108 AliCDBEntry *pNmeanEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Nmean"); //contains TObjArray of 42 TF1 + 1 EPhotMean
109 AliCDBEntry *pQthreEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Qthre"); //contains TObjArray of 42 (7ch * 6sec) TF1
110 if(!pNmeanEnt) AliError("No Nmean C6F14 ");
111 if(!pQthreEnt) AliError("No Qthre");
113 return Recon(pEsd,fClu,(TObjArray*)pNmeanEnt->GetObject(),(TObjArray*)pQthreEnt->GetObject());
115 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
116 Int_t AliHMPIDTracker::Recon(AliESDEvent *pEsd,TObjArray *pClus,TObjArray *pNmean, TObjArray *pQthre)
118 // Static method to reconstruct Theta Ckov for all valid tracks of a given event.
119 // Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time)
120 // Returns: error code, 0 if no errors
122 AliHMPIDRecon recon; //instance of reconstruction class, nothing important in ctor
123 AliHMPIDParam *pParam = AliHMPIDParam::Instance(); //Instance of AliHMPIDParam
124 Float_t xPc,yPc,xRa,yRa,theta,phi;
125 Double_t cluLORS[2]={0};
126 // Double_t cluMARS[3]={0},trkMARS[3]={0};
127 // Double_t bestcluMARS[3]={0,0,0};
128 // Double_t radClu,radInitTrk;
131 Double_t qthre = 0; Double_t nmean=0; Int_t hvsec=0;
132 Int_t nClusCh[AliHMPIDParam::kMaxCh+1];
134 UInt_t tsmin = (UInt_t)((TF1*)pQthre->At(0))->GetXmin(); //
135 UInt_t tsmax = (UInt_t)((TF1*)pQthre->At(0))->GetXmax(); //
136 UInt_t ts = pEsd->GetTimeStamp();
138 if(ts<tsmin || ts>tsmax) {
139 AliWarning(Form(" in HMPID time stamp out of range!!! Please check!!! ts = %i",ts));
143 for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){ //loop on the ESD tracks in the event
145 // Double_t bestChi2=99999;chi2=99999; //init. track matching params
146 Double_t dmin=999999,bz=0,distCut=1,distParams[5]={1};
148 Bool_t isOkDcut=kFALSE;
149 Bool_t isOkQcut=kFALSE;
150 Bool_t isMatched=kFALSE;
152 AliHMPIDCluster *bestHmpCluster=0x0; //the best matching cluster
153 AliESDtrack *pTrk = pEsd->GetTrack(iTrk); //get reconstructed track
155 if(!pTrk->IsOn(AliESDtrack::kTPCout)) continue;
157 if(pTrk->IsOn(AliESDtrack::kTPCrefit)) continue;
159 AliHMPIDtrack *hmpTrk = new AliHMPIDtrack(*pTrk); //create a hmpid track to be used for propagation and matching
160 bz=AliTracker::GetBz();
161 //initial flags for HMPID ESD infos
162 pTrk->SetHMPIDtrk(0,0,0,0); //no intersection found
163 pTrk->SetHMPIDmip(0,0,0,0); //store mip info in any case
164 pTrk->SetHMPIDcluIdx(99,99999); //chamber not found, mip not yet considered
165 pTrk->SetHMPIDsignal(AliHMPIDRecon::kNotPerformed); //ring reconstruction not yet performed
167 Int_t ipCh=IntTrkCha(pTrk,xPc,yPc,xRa,yRa,theta,phi); //find the intersected chamber for this track
168 if(ipCh<0) continue; //no intersection at all, go after next track
170 pTrk->SetHMPIDtrk(xPc,yPc,theta,phi); //store initial infos
171 pTrk->SetHMPIDcluIdx(ipCh,9999); //set chamber, index of cluster + cluster size
173 // track intersects the chamber ipCh: find the MIP
175 TClonesArray *pMipCluLst=(TClonesArray *)pClus->At(ipCh); //get the list of clusters
176 nMipClusTot = pMipCluLst->GetEntries(); //total number of clusters in the given chamber
177 nClusCh[ipCh] = nMipClusTot;
179 if(nMipClusTot==0) continue;
181 Int_t index=-1; //index of the "best" matching cluster
183 for (Int_t iClu=0; iClu<nMipClusTot;iClu++) { //clusters loop
185 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pMipCluLst->UncheckedAt(iClu); //get the cluster
187 if(pQthre->GetEntriesFast()==pParam->kMaxCh+1) {
188 qthre=((TF1*)pQthre->At(pClu->Ch()))->Eval(ts); //
189 } else { // in the past just 1 qthre
190 hvsec = pParam->InHVSector(pClu->Y()); // per chamber
192 qthre=((TF1*)pQthre->At(6*ipCh+hvsec))->Eval(ts); //
197 if(pClu->Q()<qthre) continue; //charge compartible with MIP clusters
200 cluLORS[0]=pClu->X(); cluLORS[1]=pClu->Y(); //get the LORS coordinates of the cluster
201 Double_t dist = TMath::Sqrt((xPc-cluLORS[0])*(xPc-cluLORS[0])+(yPc-cluLORS[1])*(yPc-cluLORS[1]));
211 pTrk->SetHMPIDsignal(pParam->kMipQdcCut);
215 Double_t radius = (pParam->Lors2Mars(ipCh,pParam->SizeAllX()/2,pParam->SizeAllY()/2)).Mag();
217 if(!AliTracker::PropagateTrackToBxByBz(hmpTrk,radius,pTrk->GetMass(),1,kFALSE)) continue;
219 if(!hmpTrk->PropagateTo(bestHmpCluster)) continue;
221 Int_t cluSiz = bestHmpCluster->Size();
222 pTrk->SetHMPIDmip(bestHmpCluster->X(),bestHmpCluster->Y(),(Int_t)bestHmpCluster->Q(),0); //store mip info in any case
223 pTrk->SetHMPIDcluIdx(ipCh,index+1000*cluSiz); //set chamber, index of cluster + cluster size
225 if(AliHMPIDReconstructor::GetRecoParam()) //retrieve distance cut
227 if(AliHMPIDReconstructor::GetRecoParam()->IsFixedDistCut()==kTRUE) //distance cut is fixed number
229 distCut=AliHMPIDReconstructor::GetRecoParam()->GetHmpTrackMatchingDist();
233 for(Int_t ipar=0;ipar<5;ipar++) distParams[ipar]=AliHMPIDReconstructor::GetRecoParam()->GetHmpTrackMatchingDistParam(ipar); //prevision: distance cut is function of momentum
234 distCut=distParams[0]+distParams[1]*TMath::Power(distParams[2]*pTrk->GetP(),distParams[3]); //prevision: change functional form to be more precise
237 else {distCut=pParam->DistCut();}
246 pTrk->SetHMPIDsignal(pParam->kMipDistCut); //closest cluster with enough charge is still too far from intersection
249 if(isOkQcut*isOkDcut) isMatched = kTRUE; // MIP-Track matched !!
251 if(!isMatched) continue; // If matched continue...
253 Bool_t isOk = hmpTrk->Update(bestHmpCluster,0.1,0);
255 pTrk->SetOuterHmpParam(hmpTrk,AliESDtrack::kHMPIDout);
257 FillResiduals(hmpTrk,bestHmpCluster,kFALSE);
261 if(pNmean->GetEntries()==21) { //for backward compatibility
262 nmean=((TF1*)pNmean->At(3*ipCh))->Eval(ts); //C6F14 Nmean for this chamber
264 Int_t iRad = pParam->Radiator(yRa); //evaluate the radiator involved
268 Double_t tLow = ((TF1*)pNmean->At(6*ipCh+2*iRad ))->Eval(ts); //C6F14 low temp for this chamber
269 Double_t tHigh = ((TF1*)pNmean->At(6*ipCh+2*iRad+1))->Eval(ts); //C6F14 high temp for this chamber
270 Double_t tExp = pParam->FindTemp(tLow,tHigh,yRa); //estimated temp for that chamber at that y
271 nmean = pParam->NIdxRad(AliHMPIDParam::Instance()->GetEPhotMean(),tExp); //mean ref idx @ a given temp
273 if(nmean < 0){ //track didn' t pass through the radiator
274 pTrk->SetHMPIDsignal(AliHMPIDRecon::kNoRad); //set the appropriate flag
275 pTrk->SetHMPIDcluIdx(ipCh,index+1000*cluSiz); //set index of cluster
280 recon.SetImpPC(xPc,yPc); //store track impact to PC
281 recon.CkovAngle(pTrk,(TClonesArray *)pClus->At(ipCh),index,nmean,xRa,yRa); //search for Cerenkov angle of this track
283 if(pTrk->GetHMPIDsignal()<0) continue;
287 pID.FindPid(pTrk,5,prob);
288 pTrk->SetHMPIDpid(prob);
289 // Printf(" Prob e- %6.2f mu %6.2f pi %6.2f k %6.2f p %6.2f",prob[0]*100,prob[1]*100,prob[2]*100,prob[3]*100,prob[4]*100);
290 delete hmpTrk;hmpTrk=0x0;
293 return 0; // error code: 0=no error;
295 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
296 Int_t AliHMPIDTracker::ReconHiddenTrk(AliESDEvent *pEsd,TObjArray *pClus,TObjArray *pNmean, TObjArray *pQthre)
298 // Static method to reconstruct Theta Ckov for all valid tracks of a given event.
299 // Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time), pQthre - pointer to all function Qthre=f(time)
300 // Returns: error code, 0 if no errors
302 AliHMPIDReconHTA reconHTA; //instance of reconstruction class, nothing important in ctor
304 AliHMPIDParam *pParam = AliHMPIDParam::Instance(); //Instance of AliHMPIDParam
306 UInt_t tsmin = (UInt_t)((TF1*)pQthre->At(0))->GetXmin(); //
307 UInt_t tsmax = (UInt_t)((TF1*)pQthre->At(0))->GetXmax(); //
308 UInt_t ts = pEsd->GetTimeStamp();
310 if(ts<tsmin || ts>tsmax) {
311 AliWarning(Form(" in HMPID time stamp out of range!!! Please check!!! ts = %i",ts));
315 for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){ //loop on the ESD tracks in the event
317 AliESDtrack *pTrk = pEsd->GetTrack(iTrk); //here it is simulated or just empty track
319 ipCh = pTrk->GetHMPIDcluIdx();ipCh/=1000000;
322 TClonesArray *pMipCluLst=(TClonesArray *)pClus->At(ipCh); //get the list of clusters
323 Int_t nMipClusTot = pMipCluLst->GetEntries(); //total number of clusters in the given chamber
332 for (Int_t iClu=0; iClu<nMipClusTot;iClu++) { //clusters loop
334 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pMipCluLst->UncheckedAt(iClu); //get the cluster
335 Double_t qClus = pClu->Q();
342 cluMipSiz = pClu->Size();
346 if(chMip<0) return 1;
351 if(pQthre->GetEntriesFast()==pParam->kMaxCh+1) { // just for backward compatibility
352 qthre=((TF1*)pQthre->At(chMip))->Eval(ts); //
353 } else { // in the past just 1 qthre
354 hvsec = pParam->InHVSector(yMip); // per chamber
355 if(hvsec>=0) qthre=((TF1*)pQthre->At(6*chMip+hvsec))->Eval(ts); //
359 pTrk->SetHMPIDmip(xMip,yMip,(Int_t)qMip,0); //store mip info in any case
360 pTrk->SetHMPIDcluIdx(chMip,indMip+1000*cluMipSiz);
361 pTrk->SetHMPIDsignal(pParam->kMipQdcCut);
362 return 1; //charge compatible with MIP clusters
365 pTrk->SetHMPIDmip(xMip,yMip,(Int_t)qMip,0); //store mip info in any case
366 pTrk->SetHMPIDcluIdx(chMip,indMip+1000*cluMipSiz);
368 Double_t yRa = yMip; //just an approx...
371 if(pNmean->GetEntries()==21) { //for backward compatibility
372 nmean=((TF1*)pNmean->At(3*chMip))->Eval(ts); //C6F14 Nmean for this chamber
374 Int_t iRad = pParam->Radiator(yRa); //evaluate the radiator involved
378 Double_t tLow = ((TF1*)pNmean->At(6*chMip+2*iRad ))->Eval(ts); //C6F14 low temp for this chamber
379 Double_t tHigh = ((TF1*)pNmean->At(6*chMip+2*iRad+1))->Eval(ts); //C6F14 high temp for this chamber
380 Double_t tExp = pParam->FindTemp(tLow,tHigh,yRa); //estimated temp for that chamber at that y
381 nmean = pParam->NIdxRad(AliHMPIDParam::Instance()->GetEPhotMean(),tExp); //mean ref idx @ a given temp
383 if(nmean < 0){ //track didn' t pass through the radiator
384 pTrk->SetHMPIDsignal(AliHMPIDRecon::kNoRad); //set the appropriate flag
389 if(!reconHTA.CkovHiddenTrk(pTrk,(TClonesArray *)pClus->At(ipCh),indMip,nmean)) { //search for track parameters and Cerenkov angle of this track
392 pID.FindPid(pTrk,5,prob);
393 pTrk->SetHMPIDpid(prob);
395 // Printf(" Prob e- %6.2f mu %6.2f pi %6.2f k %6.2f p %6.2f",prob[0]*100,prob[1]*100,prob[2]*100,prob[3]*100,prob[4]*100);
398 return 0; // error code: 0=no error;
401 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
402 void AliHMPIDTracker::FillClusterArray(TObjArray* array) const {
404 // Publishes all pointers to clusters known to the tracker into the
405 // passed object array.
406 // The ownership is not transfered - the caller is not expected to delete
409 for(Int_t iCh=AliHMPIDParam::kMinCh;iCh<=AliHMPIDParam::kMaxCh;iCh++){
410 TClonesArray *pCluArr=(TClonesArray*)(*fClu)[iCh];
411 for (Int_t iClu=0; iClu<pCluArr->GetEntriesFast();iClu++){
412 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pCluArr->UncheckedAt(iClu);
413 array->AddLast(pClu);
414 }//cluster loop in iCh
420 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++