Tiny (< 0.1 mm) overlaps corrected.
[u/mrichter/AliRoot.git] / HMPID / AliHMPIDTracker.cxx
CommitLineData
3c6274c1 1#include "AliHMPIDTracker.h" //class header
2#include "AliHMPIDCluster.h" //GetTrackPoint(),PropagateBack()
3#include "AliHMPIDParam.h" //GetTrackPoint(),PropagateBack()
59d9d4b3 4#include "AliHMPIDRecon.h" //Recon()
57fcdc2b 5#include "AliHMPIDReconHTA.h" //ReconHTA()
6#include <AliESDEvent.h> //PropagateBack(),Recon()
7#include <AliESDtrack.h> //Intersect()
d3da6dc4 8#include <AliRun.h> //GetTrackPoint(),PropagateBack()
9#include <AliTrackPointArray.h> //GetTrackPoint()
10#include <AliAlignObj.h> //GetTrackPoint()
59d9d4b3 11#include <AliCDBManager.h> //PropageteBack()
12#include <AliCDBEntry.h> //PropageteBack()
423554a3 13//.
14// HMPID base class fo tracking
15//.
16//.
17//.
d3da6dc4 18ClassImp(AliHMPIDTracker)
94b1fbfa 19//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
c61a7285 20AliHMPIDTracker::AliHMPIDTracker():
21 AliTracker(),
22 fClu(new TObjArray(AliHMPIDParam::kMaxCh+1))
94b1fbfa 23{
59d9d4b3 24// ctor. Create TObjArray of TClonesArray of AliHMPIDCluster
25//
26//
c61a7285 27 fClu->SetOwner(kTRUE);
ae5a42aa 28 for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) fClu->AddAt(new TClonesArray("AliHMPIDCluster"),i);
59d9d4b3 29}//ctor
d3da6dc4 30//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
31Bool_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()
59d9d4b3 39 Int_t iCham=idx/1000000; Int_t iClu=idx%1000000;
ae079791 40 point.SetVolumeID(AliGeomManager::LayerToVolUID(AliGeomManager::kHMPID,iCham-1));//layer and chamber number
94b1fbfa 41 TClonesArray *pArr=(TClonesArray*)(*fClu)[iCham];
42 AliHMPIDCluster *pClu=(AliHMPIDCluster*)pArr->UncheckedAt(iClu);//get pointer to cluster
d3da6dc4 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;
59d9d4b3 47}//GetTrackPoint()
48//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
39cd22e6 49Int_t AliHMPIDTracker::IntTrkCha(AliESDtrack *pTrk,Float_t &xPc,Float_t &yPc,Float_t &xRa,Float_t &yRa,Float_t &theta,Float_t &phi)
59d9d4b3 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();
ae5a42aa 55 for(Int_t i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++){ //chambers loop
59d9d4b3 56 Double_t p1[3],n1[3]; pParam->Norm(i,n1); pParam->Point(i,p1,AliHMPIDParam::kRad); //point & norm for middle of radiator plane
57 Double_t p2[3],n2[3]; pParam->Norm(i,n2); pParam->Point(i,p2,AliHMPIDParam::kPc); //point & norm for entrance to PC plane
58 if(pTrk->Intersect(p1,n1,-GetBz())==kFALSE) continue; //try to intersect track with the middle of radiator
59 if(pTrk->Intersect(p2,n2,-GetBz())==kFALSE) continue; //try to intersect track with PC
60 pParam->Mars2LorsVec(i,n1,theta,phi); //track angles at RAD
61 pParam->Mars2Lors (i,p1,xRa,yRa); //TRKxRAD position
62 pParam->Mars2Lors (i,p2,xPc,yPc); //TRKxPC position
39cd22e6 63 if(AliHMPIDParam::IsInside(xPc,yPc,pParam->DistCut())==kTRUE) return i; //return intersected chamber
59d9d4b3 64 } //chambers loop
39cd22e6 65 return -1; //no intersection with HMPID chambers
59d9d4b3 66}//IntTrkCha()
d3da6dc4 67//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
68Int_t AliHMPIDTracker::LoadClusters(TTree *pCluTree)
69{
59d9d4b3 70// Interface callback methode invoked from AliReconstruction::RunTracking() to load HMPID clusters before PropagateBack() gets control. Done once per event.
d3da6dc4 71// Arguments: pCluTree- pointer to clusters tree got by AliHMPIDLoader::LoadRecPoints("read") then AliHMPIDLoader::TreeR()
72// Returns: error code (currently ignored in AliReconstruction::RunTraking())
ae5a42aa 73 for(int i=AliHMPIDParam::kMinCh;i<=AliHMPIDParam::kMaxCh;i++) pCluTree->SetBranchAddress(Form("HMPID%d",i),&((*fClu)[i]));
94b1fbfa 74 pCluTree->GetEntry(0);
94b1fbfa 75 return 0;
59d9d4b3 76}//LoadClusters()
d3da6dc4 77//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
af885e0f 78Int_t AliHMPIDTracker::PropagateBack(AliESDEvent *pEsd)
abb5f786 79{
59d9d4b3 80// Interface pure virtual in AliTracker. Invoked from AliReconstruction::RunTracking() after invocation of AliTracker::LoadClusters() once per event
abb5f786 81// Agruments: pEsd - pointer to ESD
82// Returns: error code
f6998a13 83 AliCDBEntry *pNmeanEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Nmean"); //contains TObjArray of 21 TF1
49881df7 84 AliCDBEntry *pQthreEnt =AliCDBManager::Instance()->Get("HMPID/Calib/Qthre"); //contains TObjArray of 42 (7ch * 6sec) TF1
abb5f786 85 if(!pNmeanEnt) AliFatal("No Nmean C6F14 ");
86 if(!pQthreEnt) AliFatal("No Qthre");
94b1fbfa 87
afe12692 88 return Recon(pEsd,fClu,(TObjArray*)pNmeanEnt->GetObject(),(TObjArray*)pQthreEnt->GetObject());
611e810d 89}//PropagateBack()
abb5f786 90//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
afe12692 91Int_t AliHMPIDTracker::Recon(AliESDEvent *pEsd,TObjArray *pClus,TObjArray *pNmean, TObjArray *pQthre)
d3da6dc4 92{
59d9d4b3 93// Static method to reconstruct Theta Ckov for all valid tracks of a given event.
94// Arguments: pEsd- pointer ESD; pClu- pointer to clusters for all chambers; pNmean - pointer to all function Nmean=f(time)
d3da6dc4 95// Returns: error code, 0 if no errors
abb5f786 96 AliHMPIDRecon recon; //instance of reconstruction class, nothing important in ctor
39cd22e6 97 Float_t xPc,yPc,xRa,yRa,theta,phi;
59d9d4b3 98 for(Int_t iTrk=0;iTrk<pEsd->GetNumberOfTracks();iTrk++){ //ESD tracks loop
99 AliESDtrack *pTrk = pEsd->GetTrack(iTrk); //get reconstructed track
39cd22e6 100 Int_t cham=IntTrkCha(pTrk,xPc,yPc,xRa,yRa,theta,phi); //get chamber intersected by this track
101 if(cham<0) { //no intersection at all, go after next track
102 pTrk->SetHMPIDtrk(0,0,0,0); //no intersection found
103 pTrk->SetHMPIDcluIdx (99,99999); //chamber not found, mip not yet considered
104 pTrk->SetHMPIDsignal(AliHMPIDRecon::kNotPerformed); //ring reconstruction not yet performed
105 continue;
106 }
107 pTrk->SetHMPIDtrk(xRa,yRa,theta,phi); //store initial infos
a591e55f 108 Double_t nmean=((TF1*)pNmean->At(3*cham))->Eval(pEsd->GetTimeStamp()); //C6F14 Nmean for this chamber
6b7f1827 109 Double_t qthre = 0;
110 if(pQthre->GetEntriesFast()==AliHMPIDParam::kMaxCh+1) // just for backward compatibility
111 qthre=((TF1*)pQthre->At(cham))->Eval(pEsd->GetTimeStamp()); //
112 else { // in the past just 1 qthre
7656d8ac 113 Int_t hvsec = AliHMPIDParam::InHVSector(yPc); // per chamber
44193e0f 114 if (hvsec>=0)
115 qthre=((TF1*)pQthre->At(6*cham+hvsec))->Eval(pEsd->GetTimeStamp()); //
6b7f1827 116 } //
a591e55f 117 recon.SetImpPC(xPc,yPc); //store track impact to PC
afe12692 118 recon.CkovAngle(pTrk,(TClonesArray *)pClus->At(cham),nmean,qthre); //search for Cerenkov angle of this track
8d6043fc 119// Printf("AliHMPIDTracker::Recon: nmean %f, qthre %f",nmean,qthre);
59280a5a 120 } //ESD tracks loop
d3da6dc4 121 return 0; // error code: 0=no error;
611e810d 122}//Recon()
123//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
49881df7 124Int_t AliHMPIDTracker::ReconHiddenTrk(Int_t iCh,Int_t iHVsec,AliESDtrack *pTrk,TClonesArray *pCluLst,TObjArray *pNmean,TObjArray *pQthre)
611e810d 125{
126// Static method to reconstruct Theta Ckov for all valid tracks of a given event.
afe12692 127// 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)
611e810d 128// Returns: error code, 0 if no errors
57fcdc2b 129 AliHMPIDReconHTA reconHTA; //instance of reconstruction class, nothing important in ctor
611e810d 130 Double_t nmean=((TF1*)pNmean->At(3*iCh))->Eval(0); //C6F14 Nmean for this chamber
6b7f1827 131 Double_t qthre = 0;
fd47aebf 132 if(pQthre->GetEntriesFast()==AliHMPIDParam::kMaxCh+1) //
133 qthre=((TF1*)pQthre->At(iCh))->Eval(0); //just for backward compatibi
134 else qthre=((TF1*)pQthre->At(6*iCh+iHVsec))->Eval(0); //
611e810d 135 if(pCluLst->GetEntriesFast()<4) return 1; //min 4 clusters (3 + 1 mip) to find a ring!
57fcdc2b 136 if(reconHTA.CkovHiddenTrk(pTrk,pCluLst,nmean,qthre)) return 0; //search for track parameters and Cerenkov angle of this track
5b2b2013 137 else return 1; // error code: 0=no error,1=fit not performed;
611e810d 138}//Recon()
d3da6dc4 139//++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++