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 */
6 //-------------------------------------------------------------------------
7 // ITS upgrade tracker base class
8 //-------------------------------------------------------------------------
10 #include "AliTracker.h"
11 #include "AliESDEvent.h"
12 #include "AliITSUSeed.h"
13 #include "AliITSUTrackCond.h"
14 #include "AliITSUTrackHyp.h"
16 class AliITSUReconstructor;
18 class AliITSUClusterPix;
20 class AliITSURecoLayer;
24 //-------------------------------------------------------------------------
25 class AliITSUTrackerGlo : public AliTracker {
28 enum {kClus2Tracks,kPropBack,kRefitInw}; // tracking phases
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)
35 enum {kMissingCluster=0 // no cluster found on this layer
36 ,kTransportFailed=1 // seed did not reach target layer
37 ,kRWCheckFailed =2 // failed to rotate the seed to frame of the layer impact point
39 enum {kStopSearchOnSensor,kClusterNotMatching,kClusterMatching}; // flags for track-to-cluster checks
41 enum {kDummyLabel=-3141593};
42 AliITSUTrackerGlo(AliITSUReconstructor* rec);
43 virtual ~AliITSUTrackerGlo();
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;
51 //------------------------------------
52 AliITSURecoDet* GetITSInterface() const {return fITS;}
54 //------------------------------------
55 Bool_t NeedToProlong(AliESDtrack* estTr);
56 void Init(AliITSUReconstructor* rec);
57 void FindTrack(AliESDtrack* esdTr, Int_t esdID);
58 void CreateDefaultTrackCond();
59 Bool_t InitHypothesis(AliESDtrack *esdTr, Int_t esdID);
60 Bool_t TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo);
61 Bool_t TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo);
62 Bool_t GoToExitFromLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kTRUE);
63 Bool_t GoToExitFromLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kTRUE);
64 Bool_t GoToEntranceToLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
65 Bool_t GoToEntranceToLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
66 Bool_t PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
67 Bool_t PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
68 Bool_t RefitTrack(AliITSUTrackHyp* trc, Double_t r, Bool_t stopAtLastCl=kFALSE);
69 Int_t GetTrackingPhase() const {return fTrackPhase;}
72 void KillSeed(AliITSUSeed* seed, Bool_t branch=kFALSE);
73 Bool_t NeedToKill(AliITSUSeed* seed, Int_t flag);
74 Bool_t GetRoadWidth(AliITSUSeed* seed, int ilrA);
75 Int_t CheckCluster(AliITSUSeed* seed, Int_t lr, Int_t clID);
76 void AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr);
78 AliITSUSeed* NewSeedFromPool(const AliITSUSeed* src=0);
79 void ResetSeedsPool();
80 void MarkSeedFree(AliITSUSeed* seed );
82 AliITSUTrackHyp* GetTrackHyp(Int_t id) const {return (AliITSUTrackHyp*)fHypStore.UncheckedAt(id);}
83 void SetTrackHyp(AliITSUTrackHyp* hyp,Int_t id) {fHypStore.AddAtAndExpand(hyp,id);}
84 void DeleteLastSeedFromPool() {fSeedsPool.RemoveLast();}
85 void SaveCurrentTrackHypotheses();
86 void FinalizeHypotheses();
87 void UpdateESDTrack(AliITSUTrackHyp* hyp,Int_t flag);
88 void CookMCLabel(AliITSUTrackHyp* hyp);
89 void SetTrackingPhase(Int_t p) {fTrackPhase = p;}
92 TObject*& NextFreeSeed();
96 AliITSUTrackerGlo(const AliITSUTrackerGlo&);
97 AliITSUTrackerGlo &operator=(const AliITSUTrackerGlo &tr);
100 AliITSUReconstructor* fReconstructor; // ITS global reconstructor
101 AliITSURecoDet* fITS; // interface to ITS, borrowed from reconstructor
102 AliESDtrack* fCurrESDtrack; // current esd track in processing
103 Double_t fCurrMass; // current track mass
104 Double_t fTrImpData[kNTrImpData]; // data on track impact on the layer
106 // the seeds management to be optimized
107 TObjArray fHypStore; // storage for tracks hypotheses
108 AliITSUTrackHyp* fCurrHyp; // hypotheses container for current track
109 TClonesArray fSeedsPool; //! pool for seeds
110 TArrayI fFreeSeedsID; //! array of ID's of freed seeds
111 Int_t fNFreeSeeds; //! number of seeds freed in the pool
112 Int_t fLastSeedID; //! id of the pool seed on which is returned by the NextFreeSeed method
114 TObjArray fDefTrackConds; //! default tracking conditions
115 AliITSUTrackCond* fCurrTrackCond; // current tracking condition
116 Int_t fTrackPhase; // tracking phase
117 Int_t* fClInfo; //! auxiliary track cluster info
119 static const Double_t fgkToler; // tracking tolerance
121 ClassDef(AliITSUTrackerGlo,1) //ITS upgrade tracker
125 //_________________________________________________________________________
126 inline void AliITSUTrackerGlo::AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr)
128 // add new seed prolongation hypothesis
129 fCurrHyp->AddSeed(seed,lr);
130 printf("*** Adding: "); seed->Print();
133 //________________________________________
134 inline TObject *&AliITSUTrackerGlo::NextFreeSeed()
136 // return next free slot where the seed can be created
137 fLastSeedID = fNFreeSeeds ? fFreeSeedsID.GetArray()[--fNFreeSeeds] : fSeedsPool.GetEntriesFast();
138 // AliInfo(Form("%d",fLastSeedID));
139 return fSeedsPool[ fLastSeedID ];
143 //_________________________________________________________________________
144 inline AliITSUSeed* AliITSUTrackerGlo::NewSeedFromPool(const AliITSUSeed* src)
146 // create new seed, optionally copying from the source
147 AliITSUSeed* sd = src ? new( NextFreeSeed() ) AliITSUSeed(*src) : new( NextFreeSeed() ) AliITSUSeed();
148 sd->SetPoolID(fLastSeedID);
152 //_________________________________________________________________________
153 inline void AliITSUTrackerGlo::KillSeed(AliITSUSeed* seed, Bool_t branch)
155 // flag seed as killed, if requested, kill recursively its parents whose sole child is the seed being killed
157 seed = (AliITSUSeed*)seed->GetParent();
158 if (seed && !seed->DecChildren() && branch) KillSeed(seed,branch);