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 "AliHMPIDReconHTA.h" //ReconHTA()
8 #include <AliESDEvent.h> //PropagateBack(),Recon()
9 #include <AliESDtrack.h> //Intersect()
10 #include <AliRun.h> //GetTrackPoint(),PropagateBack()
11 #include <AliTrackPointArray.h> //GetTrackPoint()
12 #include <AliAlignObj.h> //GetTrackPoint()
13 #include <AliCDBManager.h> //PropageteBack()
14 #include <AliCDBEntry.h> //PropageteBack()
16 // HMPID base class fo tracking
20 ClassImp(AliHMPIDTracker)
21 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
22 AliHMPIDTracker::AliHMPIDTracker():
24 fClu(new TObjArray(AliHMPIDParam::kMaxCh+1))
26 // ctor. Create TObjArray of TClonesArray of AliHMPIDCluster
29 fClu->SetOwner(kTRUE);
30 for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) fClu->AddAt(new TClonesArray("AliHMPIDCluster"),i);
32 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
33 Bool_t AliHMPIDTracker::GetTrackPoint(Int_t idx, AliTrackPoint& point) const
35 // Interface callback methode invoked from AliReconstruction::WriteAlignmentData() to get position of MIP cluster in MARS associated to a current track.
36 // MIP cluster is reffered by index which is stored in AliESDtrack ???????
37 // Arguments: idx- cluster index which is stored by HMPID in AliESDtrack
38 // point- reference to the object where to store the point
39 // Returns: status of operation if FALSE then AliReconstruction::WriteAlignmentData() do not store this point to array of points for current track.
40 if(idx<0) return kFALSE; //no MIP cluster assigned to this track in PropagateBack()
41 Int_t iCham=idx/1000000; Int_t iClu=idx%1000000;
42 point.SetVolumeID(AliGeomManager::LayerToVolUID(AliGeomManager::kHMPID,iCham-1));//layer and chamber number
43 TClonesArray *pArr=(TClonesArray*)(*fClu)[iCham];
44 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pArr->UncheckedAt(iClu);//get pointer to cluster
46 AliHMPIDParam::Instance()->Lors2Mars(iCham,pClu->X(),pClu->Y(),mars);
47 point.SetXYZ(mars[0],mars[1],mars[2]);
50 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
51 Int_t AliHMPIDTracker::IntTrkCha(AliESDtrack *pTrk,Float_t &xPc,Float_t &yPc,Float_t &xRa,Float_t &yRa,Float_t &theta,Float_t &phi)
53 // Static method to find intersection in between given track and HMPID chambers
54 // Arguments: pTrk- ESD track; xPc,yPc- track intersection with PC in LORS [cm]
55 // Returns: intersected chamber ID or -1
56 AliHMPIDParam *pParam=AliHMPIDParam::Instance();
57 for(Int_t i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++){ //chambers loop
58 Double_t p1[3],n1[3]; pParam->Norm(i,n1); pParam->Point(i,p1,AliHMPIDParam::kRad); //point & norm for middle of radiator plane
59 Double_t p2[3],n2[3]; pParam->Norm(i,n2); pParam->Point(i,p2,AliHMPIDParam::kPc); //point & norm for entrance to PC plane
60 if(pTrk->Intersect(p1,n1,-GetBz())==kFALSE) continue; //try to intersect track with the middle of radiator
61 if(pTrk->Intersect(p2,n2,-GetBz())==kFALSE) continue; //try to intersect track with PC
62 pParam->Mars2LorsVec(i,n1,theta,phi); //track angles at RAD
63 pParam->Mars2Lors (i,p1,xRa,yRa); //TRKxRAD position
64 pParam->Mars2Lors (i,p2,xPc,yPc); //TRKxPC position
65 if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kTRUE) return i; //return intersected chamber
67 return -1; //no intersection with HMPID chambers
69 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
70 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)
72 // Static method to find intersection in between given track and HMPID chambers
73 // Arguments: pTrk- HMPID track; xPc,yPc- track intersection with PC in LORS [cm]
74 // Returns: intersected chamber ID or -1
75 AliHMPIDParam *pParam=AliHMPIDParam::Instance();
78 pParam->Point(ch,p1,AliHMPIDParam::kRad); //point & norm for middle of radiator plane
81 pParam->Point(ch,p2,AliHMPIDParam::kPc); //point & norm for entrance to PC plane
82 if(pTrk->Intersect(pTrk,p1,n1)==kFALSE) return -1; //try to intersect track with the middle of radiator
83 if(pTrk->Intersect(pTrk,p2,n2)==kFALSE) return -1;
84 pParam->Mars2LorsVec(ch,n1,theta,phi); //track angles at RAD
85 pParam->Mars2Lors (ch,p1,xRa,yRa); //TRKxRAD position
86 pParam->Mars2Lors (ch,p2,xPc,yPc); //TRKxPC position
87 if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kTRUE) return ch; //return intersected chamber
88 return -1; //no intersection with HMPID chambers
90 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
91 Int_t AliHMPIDTracker::LoadClusters(TTree *pCluTree)
93 // Interface callback methode invoked from AliReconstruction::RunTracking() to load HMPID clusters before PropagateBack() gets control. Done once per event.
94 // Arguments: pCluTree- pointer to clusters tree got by AliHMPIDLoader::LoadRecPoints("read") then AliHMPIDLoader::TreeR()
95 // Returns: error code (currently ignored in AliReconstruction::RunTraking())
96 for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) pCluTree->SetBranchAddress(Form("HMPID%d",i),&((*fClu)[i]));
97 pCluTree->GetEntry(0);
100 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
101 Int_t AliHMPIDTracker::PropagateBack(AliESDEvent *pEsd)
103 // Interface pure virtual in AliTracker. Invoked from AliReconstruction::RunTracking() after invocation of AliTracker::LoadClusters() once per event
104 // Agruments: pEsd - pointer to ESD
105 // Returns: error code
106 AliCDBEntry *pNmeanEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Nmean"); //contains TObjArray of 42 TF1 + 1 EPhotMean
107 AliCDBEntry *pQthreEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Qthre"); //contains TObjArray of 42 (7ch * 6sec) TF1
108 if(!pNmeanEnt) AliError("No Nmean C6F14 ");
109 if(!pQthreEnt) AliError("No Qthre");
111 return Recon(pEsd,fClu,(TObjArray*)pNmeanEnt->GetObject(),(TObjArray*)pQthreEnt->GetObject());
113 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
114 Int_t AliHMPIDTracker::Recon(AliESDEvent *pEsd,TObjArray *pClus,TObjArray *pNmean, TObjArray *pQthre)
116 // Static method to reconstruct Theta Ckov for all valid tracks of a given event.
117 // Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time)
118 // Returns: error code, 0 if no errors
120 AliHMPIDRecon recon; //instance of reconstruction class, nothing important in ctor
121 Float_t xPc,yPc,xRa,yRa,theta,phi;
122 Double_t cluLORS[2]={0},cluMARS[3]={0},trkMARS[3]={0};
123 // Double_t bestcluMARS[3]={0,0,0};
124 Double_t radClu,radInitTrk;
126 Double_t d3d=0,dmin=999999,bz=0;
127 Bool_t isMatched=kFALSE;
129 Double_t qthre = 0; Double_t nmean=0; Int_t cham=0; Int_t hvsec=0;
130 Int_t index=0; //index of the "best" matching cluster
131 Double_t bestChi2=-1; //Chi2 of the "best" matching cluster
133 Int_t nClusCh[AliHMPIDParam::kMaxCh+1];
134 Bool_t isOkQcut=kFALSE;
135 Bool_t isOkDcut=kFALSE;
137 AliHMPIDParam *pParam = AliHMPIDParam::Instance(); //Instance of AliHMPIDParam
139 for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){ //loop on the ESD tracks in the event
140 isMatched=kFALSE;dmin=999999;bestChi2=99999;chi2=99999;cluSiz=0; //init. track matching params
142 AliHMPIDCluster *bestHmpCluster=0x0; //the best matching cluster
143 AliESDtrack *pTrk = pEsd->GetTrack(iTrk); //get reconstructed track
144 AliHMPIDtrack *hmpTrk = new AliHMPIDtrack(*pTrk); //create a hmpid track to be used for propagation and matching
145 bz=AliTracker::GetBz();
147 Int_t ipCh=IntTrkCha(pTrk,xPc,yPc,xRa,yRa,theta,phi); //find the intersected chamber for this track
148 if(ipCh<0) { //no intersection at all, go after next track
149 pTrk->SetHMPIDtrk(0,0,0,0); //no intersection found
150 pTrk->SetHMPIDcluIdx (99,99999); //chamber not found, mip not yet considered
151 pTrk->SetHMPIDsignal(AliHMPIDRecon::kNotPerformed); //ring reconstruction not yet performed
155 // track intersects the chamber ipCh: find the MIP
157 TClonesArray *pMipCluLst=(TClonesArray *)pClus->At(ipCh); //get the list of clusters
158 nMipClusTot = pMipCluLst->GetEntries(); //total number of clusters in the given chamber
159 nClusCh[ipCh] = nMipClusTot;
161 for (Int_t iClu=0; iClu<nMipClusTot;iClu++) { //clusters loop
163 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pMipCluLst->UncheckedAt(iClu); //get the cluster
165 if(pQthre->GetEntriesFast()==pParam->kMaxCh+1) { // just for backward compatibility
166 qthre=((TF1*)pQthre->At(pClu->Ch()))->Eval(pEsd->GetTimeStamp()); //
167 } else { // in the past just 1 qthre
168 hvsec = pParam->InHVSector(pClu->Y()); // per chamber
170 qthre=((TF1*)pQthre->At(6*cham+hvsec))->Eval(pEsd->GetTimeStamp()); //
173 if(pClu->Q()<qthre) continue; //charge compartible with MIP clusters
176 cluLORS[0]=pClu->X(); cluLORS[1]=pClu->Y(); //get the LORS coordinates of the cluster
177 pParam->Lors2Mars(ipCh,cluLORS[0],cluLORS[1],cluMARS); //convert cluster coors. from LORS to MARS
178 radClu=TMath::Sqrt(cluMARS[0]*cluMARS[0]+cluMARS[1]*cluMARS[1]); //radial distance of candidate cluster in MARS
180 hmpTrk->GetXYZ(trkx0); //get track position in MARS
181 radInitTrk=TMath::Sqrt(trkx0[0]*trkx0[0]+trkx0[1]*trkx0[1]);
182 hmpTrk->PropagateToR(radClu,10);
183 hmpTrk->GetXYZ(trkx0); //get track position in MARS
184 hmpTrk->GetXYZAt(radClu,bz,trkMARS); //get the track coordinates at the rad distance after prop.
185 d3d=TMath::Sqrt((cluMARS[0]-trkMARS[0])*(cluMARS[0]-trkMARS[0])+(cluMARS[1]-trkMARS[1])*(cluMARS[1]-trkMARS[1])+(cluMARS[2]-trkMARS[2])*(cluMARS[2]-trkMARS[2]));
186 chi2=hmpTrk->GetPredictedChi2(pClu);
187 if(dmin > d3d ) { //to be saved for the moment...
188 cluSiz = pClu->Size();
193 cluLORS[0]=pClu->X(); cluLORS[1]=pClu->Y();
194 // pParam->Lors2Mars(ipCh,cluLORS[0],cluLORS[1],bestcluMARS);
198 pTrk->SetHMPIDmip(0,0,0,0); //store mip info in any case
201 pTrk->SetHMPIDcluIdx(ipCh,9999);
202 pTrk->SetHMPIDsignal(pParam->kMipQdcCut);
206 if(dmin < pParam->DistCut()) {
211 pTrk->SetHMPIDmip(bestHmpCluster->X(),bestHmpCluster->Y(),(Int_t)bestHmpCluster->Q(),0); //store mip info in any case
212 pTrk->SetHMPIDcluIdx(ipCh,index+1000*cluSiz); //set chamber, index of cluster + cluster size
213 pTrk->SetHMPIDsignal(pParam->kMipDistCut); //closest cluster with enough charge is still too far from intersection
216 if(isOkQcut*isOkDcut) isMatched = kTRUE; // MIP-Track matched !!
218 if(!isMatched) continue; // If matched continue...
221 for(Int_t iC=0;iC<ipCh;iC++) indexAll+=nClusCh[iC]; indexAll+=index; //to be verified...
223 Bool_t isOk = hmpTrk->Update(bestHmpCluster,bestChi2,indexAll);
225 pTrk->SetOuterParam(hmpTrk,AliESDtrack::kHMPIDout);
227 // cham=IntTrkCha(ipCh,hmpTrk,xPc,yPc,xRa,yRa,theta,phi);
228 cham=IntTrkCha(pTrk,xPc,yPc,xRa,yRa,theta,phi);
229 if(cham<0) { //no intersection at all, go after next track
230 pTrk->SetHMPIDtrk(0,0,0,0); //no intersection found
231 pTrk->SetHMPIDcluIdx (99,99999); //chamber not found, mip not yet considered
232 pTrk->SetHMPIDsignal(AliHMPIDRecon::kNotPerformed); //ring reconstruction not yet performed
236 pTrk->SetHMPIDtrk(xRa,yRa,theta,phi); //store initial infos
238 if(pNmean->GetEntries()==21) { //for backward compatibility
239 nmean=((TF1*)pNmean->At(3*cham))->Eval(pEsd->GetTimeStamp()); //C6F14 Nmean for this chamber
241 Int_t iRad = pParam->Radiator(yRa); //evaluate the radiator involved
245 Double_t tLow = ((TF1*)pNmean->At(6*cham+2*iRad ))->Eval(pEsd->GetTimeStamp()); //C6F14 low temp for this chamber
246 Double_t tHigh = ((TF1*)pNmean->At(6*cham+2*iRad+1))->Eval(pEsd->GetTimeStamp()); //C6F14 high temp for this chamber
247 Double_t tExp = pParam->FindTemp(tLow,tHigh,yRa); //estimated temp for that chamber at that y
248 nmean = pParam->NIdxRad(AliHMPIDParam::Instance()->GetEPhotMean(),tExp); //mean ref idx @ a given temp
250 if(nmean < 0){ //track didn' t pass through the radiator
251 pTrk->SetHMPIDsignal(AliHMPIDRecon::kNoRad); //set the appropriate flag
252 pTrk->SetHMPIDcluIdx(ipCh,index+1000*cluSiz); //set index of cluster
257 recon.SetImpPC(xPc,yPc); //store track impact to PC
258 recon.CkovAngle(pTrk,(TClonesArray *)pClus->At(cham),index,nmean); //search for Cerenkov angle of this track
262 pID.FindPid(pTrk,5,prob);
263 pTrk->SetHMPIDpid(prob);
264 // 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);
268 return 0; // error code: 0=no error;
270 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
271 Int_t AliHMPIDTracker::ReconHiddenTrk(AliESDEvent *pEsd,TObjArray *pClus,TObjArray *pNmean, TObjArray *pQthre)
273 // Static method to reconstruct Theta Ckov for all valid tracks of a given event.
274 // 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)
275 // Returns: error code, 0 if no errors
277 AliHMPIDReconHTA reconHTA; //instance of reconstruction class, nothing important in ctor
279 AliHMPIDParam *pParam = AliHMPIDParam::Instance(); //Instance of AliHMPIDParam
281 for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){ //loop on the ESD tracks in the event
283 AliESDtrack *pTrk = pEsd->GetTrack(iTrk); //here it is simulated or just empty track
285 ipCh = pTrk->GetHMPIDcluIdx();ipCh/=1000000;
288 TClonesArray *pMipCluLst=(TClonesArray *)pClus->At(ipCh); //get the list of clusters
289 Int_t nMipClusTot = pMipCluLst->GetEntries(); //total number of clusters in the given chamber
298 for (Int_t iClu=0; iClu<nMipClusTot;iClu++) { //clusters loop
300 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pMipCluLst->UncheckedAt(iClu); //get the cluster
301 Double_t qClus = pClu->Q();
308 cluMipSiz = pClu->Size();
312 if(chMip<0) return 1;
317 if(pQthre->GetEntriesFast()==pParam->kMaxCh+1) { // just for backward compatibility
318 qthre=((TF1*)pQthre->At(chMip))->Eval(pEsd->GetTimeStamp()); //
319 } else { // in the past just 1 qthre
320 hvsec = pParam->InHVSector(yMip); // per chamber
321 if(hvsec>=0) qthre=((TF1*)pQthre->At(6*chMip+hvsec))->Eval(pEsd->GetTimeStamp()); //
325 pTrk->SetHMPIDmip(xMip,yMip,(Int_t)qMip,0); //store mip info in any case
326 pTrk->SetHMPIDcluIdx(chMip,indMip+1000*cluMipSiz);
327 pTrk->SetHMPIDsignal(pParam->kMipQdcCut);
328 return 1; //charge compatible with MIP clusters
331 pTrk->SetHMPIDmip(xMip,yMip,(Int_t)qMip,0); //store mip info in any case
332 pTrk->SetHMPIDcluIdx(chMip,indMip+1000*cluMipSiz);
334 Double_t yRa = yMip; //just an approx...
337 if(pNmean->GetEntries()==21) { //for backward compatibility
338 nmean=((TF1*)pNmean->At(3*chMip))->Eval(pEsd->GetTimeStamp()); //C6F14 Nmean for this chamber
340 Int_t iRad = pParam->Radiator(yRa); //evaluate the radiator involved
344 Double_t tLow = ((TF1*)pNmean->At(6*chMip+2*iRad ))->Eval(pEsd->GetTimeStamp()); //C6F14 low temp for this chamber
345 Double_t tHigh = ((TF1*)pNmean->At(6*chMip+2*iRad+1))->Eval(pEsd->GetTimeStamp()); //C6F14 high temp for this chamber
346 Double_t tExp = pParam->FindTemp(tLow,tHigh,yRa); //estimated temp for that chamber at that y
347 nmean = pParam->NIdxRad(AliHMPIDParam::Instance()->GetEPhotMean(),tExp); //mean ref idx @ a given temp
349 if(nmean < 0){ //track didn' t pass through the radiator
350 pTrk->SetHMPIDsignal(AliHMPIDRecon::kNoRad); //set the appropriate flag
355 Printf(" qthre %f nmean %f index %i cham %i",qthre,nmean,indMip,chMip);
356 if(!reconHTA.CkovHiddenTrk(pTrk,(TClonesArray *)pClus->At(ipCh),indMip,nmean)) { //search for track parameters and Cerenkov angle of this track
359 pID.FindPid(pTrk,5,prob);
360 pTrk->SetHMPIDpid(prob);
362 // 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);
365 return 0; // error code: 0=no error;
368 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
369 void AliHMPIDTracker::FillClusterArray(TObjArray* array) const {
371 // Publishes all pointers to clusters known to the tracker into the
372 // passed object array.
373 // The ownership is not transfered - the caller is not expected to delete
376 for(Int_t iCh=AliHMPIDParam::kMinCh;iCh<=AliHMPIDParam::kMaxCh;iCh++){
377 TClonesArray *pCluArr=(TClonesArray*)(*fClu)[iCh];
378 for (Int_t iClu=0; iClu<pCluArr->GetEntriesFast();iClu++){
379 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pCluArr->UncheckedAt(iClu);
380 array->AddLast(pClu);
381 }//cluster loop in iCh
387 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++