template for performance study task
[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSUTrackerGlo.h
CommitLineData
32d38de2 1#ifndef ALIITSTRACKERU_H
2#define ALIITSTRACKERU_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6//-------------------------------------------------------------------------
7// ITS upgrade tracker base class
8//-------------------------------------------------------------------------
9
10#include "AliTracker.h"
11#include "AliESDEvent.h"
f8832015 12#include "AliITSUSeed.h"
c8d1f258 13#include "AliITSUTrackCond.h"
716ccba7 14#include "AliITSUTrackHyp.h"
32d38de2 15
16class AliITSUReconstructor;
17class AliITSURecoDet;
c61e50c3 18class AliITSUClusterPix;
0091e9f0 19class AliESDtrack;
e2d2481c 20class AliITSURecoLayer;
32d38de2 21class TTree;
22
c61e50c3 23
32d38de2 24//-------------------------------------------------------------------------
25class AliITSUTrackerGlo : public AliTracker {
26
27 public:
c51c3124 28 enum {kClus2Tracks,kPropBack,kRefitInw}; // tracking phases
32d38de2 29 enum { // info from track extrapolation to layer for cluster check
30 kTrXIn ,kTrYIn ,kTrZIn ,kTrPhiIn , // entrance (outer) point on the layer from above
31 kTrXOut,kTrYOut,kTrZOut,kTrPhiOut, // exit (inner) point on the layer
32 kTrPhi0, kTrDPhi, kTrZ0, kTrDZ, // mean phi,dPhi, mean z, dZ (don't change this order)
c61e50c3 33 kNTrImpData};
32d38de2 34 //
f8832015 35 enum {kMissingCluster=0 // no cluster found on this layer
36 ,kTransportFailed=1 // seed did not reach target layer
32d38de2 37 ,kRWCheckFailed =2 // failed to rotate the seed to frame of the layer impact point
38 };
f8832015 39 enum {kStopSearchOnSensor,kClusterNotMatching,kClusterMatching}; // flags for track-to-cluster checks
dd2117a2 40 //
41 enum {kDummyLabel=-3141593};
32d38de2 42 AliITSUTrackerGlo(AliITSUReconstructor* rec);
43 virtual ~AliITSUTrackerGlo();
44
45 virtual Int_t Clusters2Tracks(AliESDEvent *event);
46 virtual Int_t PropagateBack(AliESDEvent *event);
47 virtual Int_t RefitInward(AliESDEvent *event);
48 virtual Int_t LoadClusters(TTree * treeRP=0);
49 virtual void UnloadClusters();
50 virtual AliCluster* GetCluster(Int_t index) const;
0ddbf657 51 //
52 Int_t GetCountPronlongationTrials() const {return fCountProlongationTrials;}
53 Int_t GetCountITSin() const {return fCountITSin;}
54 Int_t GetCountITSout() const {return fCountITSout;}
55 Int_t GetCountITSrefit() const {return fCountITSrefit;}
56
0091e9f0 57 //------------------------------------
58 AliITSURecoDet* GetITSInterface() const {return fITS;}
59 //
32d38de2 60 //------------------------------------
61 Bool_t NeedToProlong(AliESDtrack* estTr);
62 void Init(AliITSUReconstructor* rec);
716ccba7 63 void FindTrack(AliESDtrack* esdTr, Int_t esdID);
42c3d4bd 64 void CreateDefaultTrackCond();
3e4e3c23 65 Bool_t InitHypothesis(AliESDtrack *esdTr, Int_t esdID);
32d38de2 66 Bool_t TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo);
3e4e3c23 67 Bool_t TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo);
e7d83d38 68 Bool_t GoToExitFromLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kTRUE);
69 Bool_t GoToExitFromLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kTRUE);
e2d2481c 70 Bool_t GoToEntranceToLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
71 Bool_t GoToEntranceToLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
44785f3e 72 Bool_t PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
3e4e3c23 73 Bool_t PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
68a0f687 74 Bool_t RefitTrack(AliITSUTrackHyp* trc, Double_t r, Bool_t stopAtLastCl=kFALSE);
e7d83d38 75 Int_t GetTrackingPhase() const {return fTrackPhase;}
76
44785f3e 77 //
b8b59e05 78 void KillSeed(AliITSUSeed* seed, Bool_t branch=kFALSE);
c8d1f258 79 Bool_t NeedToKill(AliITSUSeed* seed, Int_t flag);
c61e50c3 80 Bool_t GetRoadWidth(AliITSUSeed* seed, int ilrA);
f8832015 81 Int_t CheckCluster(AliITSUSeed* seed, Int_t lr, Int_t clID);
82 void AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr);
c61e50c3 83 //
84 AliITSUSeed* NewSeedFromPool(const AliITSUSeed* src=0);
b8b59e05 85 void ResetSeedsPool();
86 void MarkSeedFree(AliITSUSeed* seed );
87
716ccba7 88 AliITSUTrackHyp* GetTrackHyp(Int_t id) const {return (AliITSUTrackHyp*)fHypStore.UncheckedAt(id);}
89 void SetTrackHyp(AliITSUTrackHyp* hyp,Int_t id) {fHypStore.AddAtAndExpand(hyp,id);}
90 void DeleteLastSeedFromPool() {fSeedsPool.RemoveLast();}
91 void SaveCurrentTrackHypotheses();
3e4e3c23 92 void FinalizeHypotheses();
68a0f687 93 void UpdateESDTrack(AliITSUTrackHyp* hyp,Int_t flag);
dd2117a2 94 void CookMCLabel(AliITSUTrackHyp* hyp);
e7d83d38 95 void SetTrackingPhase(Int_t p) {fTrackPhase = p;}
716ccba7 96 //
b8b59e05 97 protected:
98 TObject*& NextFreeSeed();
99 //
32d38de2 100 private:
b8b59e05 101 //
32d38de2 102 AliITSUTrackerGlo(const AliITSUTrackerGlo&);
103 AliITSUTrackerGlo &operator=(const AliITSUTrackerGlo &tr);
104 //
105 protected:
106 AliITSUReconstructor* fReconstructor; // ITS global reconstructor
3d4dc3e2 107 AliITSURecoDet* fITS; // interface to ITS, borrowed from reconstructor
0091e9f0 108 AliESDtrack* fCurrESDtrack; // current esd track in processing
0ddbf657 109 Int_t fCurrESDtrMClb; // its eventual mc label
32d38de2 110 Double_t fCurrMass; // current track mass
c61e50c3 111 Double_t fTrImpData[kNTrImpData]; // data on track impact on the layer
32d38de2 112 //
0ddbf657 113 Int_t fCountProlongationTrials; // number of TPC seeds
114 Int_t fCountITSin; // number of successful ITSin
115 Int_t fCountITSout; // number of successful ITSout
116 Int_t fCountITSrefit; // number of successful ITSrefit
117 //
c61e50c3 118 // the seeds management to be optimized
716ccba7 119 TObjArray fHypStore; // storage for tracks hypotheses
120 AliITSUTrackHyp* fCurrHyp; // hypotheses container for current track
b8b59e05 121 TClonesArray fSeedsPool; //! pool for seeds
122 TArrayI fFreeSeedsID; //! array of ID's of freed seeds
123 Int_t fNFreeSeeds; //! number of seeds freed in the pool
124 Int_t fLastSeedID; //! id of the pool seed on which is returned by the NextFreeSeed method
c8d1f258 125 //
42c3d4bd 126 TObjArray fDefTrackConds; //! default tracking conditions
127 AliITSUTrackCond* fCurrTrackCond; // current tracking condition
c51c3124 128 Int_t fTrackPhase; // tracking phase
8b16dbae 129 Int_t* fClInfo; //! auxiliary track cluster info
c8d1f258 130 //
e2d2481c 131 static const Double_t fgkToler; // tracking tolerance
132 //
32d38de2 133 ClassDef(AliITSUTrackerGlo,1) //ITS upgrade tracker
134
135};
f8832015 136
137//_________________________________________________________________________
138inline void AliITSUTrackerGlo::AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr)
139{
140 // add new seed prolongation hypothesis
716ccba7 141 fCurrHyp->AddSeed(seed,lr);
f8832015 142}
143
b8b59e05 144//________________________________________
145inline TObject *&AliITSUTrackerGlo::NextFreeSeed()
146{
147 // return next free slot where the seed can be created
148 fLastSeedID = fNFreeSeeds ? fFreeSeedsID.GetArray()[--fNFreeSeeds] : fSeedsPool.GetEntriesFast();
149 // AliInfo(Form("%d",fLastSeedID));
150 return fSeedsPool[ fLastSeedID ];
151 //
152}
153
154//_________________________________________________________________________
155inline AliITSUSeed* AliITSUTrackerGlo::NewSeedFromPool(const AliITSUSeed* src)
156{
157 // create new seed, optionally copying from the source
158 AliITSUSeed* sd = src ? new( NextFreeSeed() ) AliITSUSeed(*src) : new( NextFreeSeed() ) AliITSUSeed();
159 sd->SetPoolID(fLastSeedID);
160 return sd;
161}
162
163//_________________________________________________________________________
164inline void AliITSUTrackerGlo::KillSeed(AliITSUSeed* seed, Bool_t branch)
165{
166 // flag seed as killed, if requested, kill recursively its parents whose sole child is the seed being killed
167 seed->Kill();
168 seed = (AliITSUSeed*)seed->GetParent();
169 if (seed && !seed->DecChildren() && branch) KillSeed(seed,branch);
170}
f8832015 171
32d38de2 172#endif
173