effc++ warnings corrected
[u/mrichter/AliRoot.git] / HMPID / AliHMPIDTracker.cxx
1 #include "AliHMPIDTracker.h"     //class header
2 #include "AliHMPIDCluster.h"     //GetTrackPoint(),PropagateBack() 
3 #include "AliHMPIDParam.h"       //GetTrackPoint(),PropagateBack()
4 #include "AliHMPIDRecon.h"       //Recon()
5 #include "AliHMPIDReconHTA.h"    //ReconHTA()
6 #include <AliESDEvent.h>         //PropagateBack(),Recon()  
7 #include <AliESDtrack.h>         //Intersect()  
8 #include <AliRun.h>              //GetTrackPoint(),PropagateBack()  
9 #include <AliTrackPointArray.h>  //GetTrackPoint()
10 #include <AliAlignObj.h>         //GetTrackPoint()
11 #include <AliCDBManager.h>       //PropageteBack()
12 #include <AliCDBEntry.h>         //PropageteBack()
13 //.
14 // HMPID base class fo tracking
15 //.
16 //.
17 //.
18 ClassImp(AliHMPIDTracker)
19 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
20 AliHMPIDTracker::AliHMPIDTracker():
21   AliTracker(),
22   fClu(new TObjArray(AliHMPIDParam::kMaxCh+1))  
23 {
24 // ctor. Create TObjArray of TClonesArray of AliHMPIDCluster  
25 // 
26 //  
27   fClu->SetOwner(kTRUE);
28   for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) fClu->AddAt(new TClonesArray("AliHMPIDCluster"),i);
29 }//ctor
30 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++    
31 Bool_t AliHMPIDTracker::GetTrackPoint(Int_t idx, AliTrackPoint& point) const
32 {
33 // Interface callback methode invoked from AliReconstruction::WriteAlignmentData() to get position of MIP cluster in MARS associated to a current track.
34 // MIP cluster is reffered by index which is stored in AliESDtrack  ???????
35 // Arguments: idx- cluster index which is stored by HMPID in AliESDtrack
36 //            point- reference to the object where to store the point     
37 //   Returns: status of operation  if FALSE then AliReconstruction::WriteAlignmentData() do not store this point to array of points for current track. 
38   if(idx<0) return kFALSE; //no MIP cluster assigned to this track in PropagateBack()
39   Int_t iCham=idx/1000000; Int_t iClu=idx%1000000;
40   point.SetVolumeID(AliGeomManager::LayerToVolUID(AliGeomManager::kHMPID,iCham-1));//layer and chamber number
41   TClonesArray *pArr=(TClonesArray*)(*fClu)[iCham];
42   AliHMPIDCluster *pClu=(AliHMPIDCluster*)pArr->UncheckedAt(iClu);//get pointer to cluster
43   Double_t mars[3];
44   AliHMPIDParam::Instance()->Lors2Mars(iCham,pClu->X(),pClu->Y(),mars);
45   point.SetXYZ(mars[0],mars[1],mars[2]);
46   return kTRUE;
47 }//GetTrackPoint()
48 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
49 Int_t AliHMPIDTracker::IntTrkCha(AliESDtrack *pTrk,Float_t &xPc,Float_t &yPc)
50 {
51 // Static method to find intersection in between given track and HMPID chambers
52 // Arguments: pTrk- ESD track; xPc,yPc- track intersection with PC in LORS [cm]
53 //   Returns: intersected chamber ID or -1
54   AliHMPIDParam *pParam=AliHMPIDParam::Instance();
55   Float_t xRa=0,yRa=0,theta=0,phi=0;                                                            //track intersection at PC and angles at RAD, LORS  
56   for(Int_t i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++){                              //chambers loop
57     Double_t p1[3],n1[3]; pParam->Norm(i,n1); pParam->Point(i,p1,AliHMPIDParam::kRad);          //point & norm  for middle of radiator plane
58     Double_t p2[3],n2[3]; pParam->Norm(i,n2); pParam->Point(i,p2,AliHMPIDParam::kPc);           //point & norm  for entrance to PC plane
59     if(pTrk->Intersect(p1,n1,-GetBz())==kFALSE) continue;                                       //try to intersect track with the middle of radiator
60     if(pTrk->Intersect(p2,n2,-GetBz())==kFALSE) continue;                                       //try to intersect track with PC
61     pParam->Mars2LorsVec(i,n1,theta,phi);                                                       //track angles at RAD
62     pParam->Mars2Lors   (i,p1,xRa,yRa);                                                         //TRKxRAD position
63     pParam->Mars2Lors   (i,p2,xPc,yPc);                                                         //TRKxPC position
64     if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kFALSE) continue;                    //not in active area  
65     pTrk->SetHMPIDtrk      (xRa,yRa,theta,phi);                                                 //store track intersection info
66     pTrk->SetHMPIDcluIdx   (i,0);
67     return i;
68   }                                                                                             //chambers loop
69   return -1; //no intersection with HMPID chambers
70 }//IntTrkCha()
71 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
72 Int_t AliHMPIDTracker::LoadClusters(TTree *pCluTree)
73 {
74 // Interface callback methode invoked from AliReconstruction::RunTracking() to load HMPID clusters before PropagateBack() gets control. Done once per event.
75 // Arguments: pCluTree- pointer to clusters tree got by AliHMPIDLoader::LoadRecPoints("read") then AliHMPIDLoader::TreeR()
76 //   Returns: error code (currently ignored in AliReconstruction::RunTraking())    
77   for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) pCluTree->SetBranchAddress(Form("HMPID%d",i),&((*fClu)[i]));
78   pCluTree->GetEntry(0);
79   return 0;  
80 }//LoadClusters()
81 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
82 Int_t AliHMPIDTracker::PropagateBack(AliESDEvent *pEsd)
83 {
84 // Interface pure virtual in AliTracker. Invoked from AliReconstruction::RunTracking() after invocation of AliTracker::LoadClusters() once per event
85 // Agruments: pEsd - pointer to ESD
86 //   Returns: error code    
87   AliCDBEntry *pNmeanEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Nmean"); //contains TObjArray of 21 TF1
88   AliCDBEntry *pQthreEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Qthre"); //contains TObjArray of 7 TF1
89   if(!pNmeanEnt) AliFatal("No Nmean C6F14 ");
90   if(!pQthreEnt) AliFatal("No Qthre");
91     
92   return Recon(pEsd,fClu,(TObjArray*)pNmeanEnt->GetObject(),(TObjArray*)pQthreEnt->GetObject());  
93 }//PropagateBack()
94 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
95 Int_t AliHMPIDTracker::Recon(AliESDEvent *pEsd,TObjArray *pClus,TObjArray *pNmean, TObjArray *pQthre)
96 {
97 // Static method to reconstruct Theta Ckov for all valid tracks of a given event.
98 // Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time)
99 //   Returns: error code, 0 if no errors   
100   AliHMPIDRecon recon;                                                                       //instance of reconstruction class, nothing important in ctor
101   Float_t xPc,yPc;
102   for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){                                       //ESD tracks loop
103     AliESDtrack *pTrk = pEsd->GetTrack(iTrk);                                                    //get reconstructed track    
104     Int_t cham=IntTrkCha(pTrk,xPc,yPc);                                                          //get chamber intersected by this track 
105     if(cham<0) continue;                                                                         //no intersection at all, go after next track
106     Double_t nmean=((TF1*)pNmean->At(3*cham))->Eval(pEsd->GetTimeStamp());                       //C6F14 Nmean for this chamber
107     Double_t qthre=((TF1*)pQthre->At(cham))  ->Eval(pEsd->GetTimeStamp());                       //Qthre for this chamber
108     recon.SetImpPC(xPc,yPc);                                                                     //store track impact to PC
109     recon.CkovAngle(pTrk,(TClonesArray *)pClus->At(cham),nmean,qthre);                           //search for Cerenkov angle of this track
110 //    Printf("AliHMPIDTracker::Recon: nmean %f, qthre %f",nmean,qthre);
111   }                                                                                              //ESD tracks loop
112   return 0; // error code: 0=no error;
113 }//Recon()
114 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
115 Int_t AliHMPIDTracker::ReconHiddenTrk(Int_t iCh,AliESDtrack *pTrk,TClonesArray *pCluLst,TObjArray *pNmean,TObjArray *pQthre)
116 {
117 // Static method to reconstruct Theta Ckov for all valid tracks of a given event.
118 // 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)
119 //   Returns: error code, 0 if no errors
120   AliHMPIDReconHTA reconHTA;                                                                          //instance of reconstruction class, nothing important in ctor
121   Double_t nmean=((TF1*)pNmean->At(3*iCh))->Eval(0);                                            //C6F14 Nmean for this chamber
122   Double_t qthre=((TF1*)pQthre->At(iCh))  ->Eval(0);                                            //C6F14 Nmean for this chamber
123   if(pCluLst->GetEntriesFast()<4) return 1;                                                     //min 4 clusters (3 + 1 mip) to find a ring! 
124   if(reconHTA.CkovHiddenTrk(pTrk,pCluLst,nmean,qthre)) return 0;                                   //search for track parameters and Cerenkov angle of this track
125   else return 1;                                                                                // error code: 0=no error,1=fit not performed;
126 }//Recon()
127 //++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++