MC labeling added
[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSUTrackerGlo.h
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"
12 #include "AliITSUSeed.h"
13 #include "AliITSUTrackCond.h"
14 #include "AliITSUTrackHyp.h"
15
16 class AliITSUReconstructor;
17 class AliITSURecoDet;
18 class AliITSUClusterPix;
19 class AliESDtrack;
20 class AliITSURecoLayer;
21 class TTree;
22
23
24 //-------------------------------------------------------------------------
25 class AliITSUTrackerGlo : public AliTracker {
26
27   public:
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)
33     kNTrImpData};
34   //
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
38   };
39   enum {kStopSearchOnSensor,kClusterNotMatching,kClusterMatching}; // flags for track-to-cluster checks
40   //
41   enum {kDummyLabel=-3141593};
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;
51   //------------------------------------
52   AliITSURecoDet*        GetITSInterface()       const {return fITS;}
53   //
54   //------------------------------------
55   Bool_t                 NeedToProlong(AliESDtrack* estTr);
56   void                   Init(AliITSUReconstructor* rec);
57   void                   FindTrack(AliESDtrack* esdTr, Int_t esdID);
58   Bool_t                 InitHypothesis(AliESDtrack *esdTr, Int_t esdID);
59   Bool_t                 TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo);
60   Bool_t                 TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo);
61   Bool_t                 GoToExitFromLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
62   Bool_t                 GoToExitFromLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
63   Bool_t                 GoToEntranceToLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
64   Bool_t                 GoToEntranceToLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
65   Bool_t                 PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
66   Bool_t                 PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
67   Bool_t                 RefitTrack(AliITSUTrackHyp* trc, Double_t r);
68   //
69   void                   KillSeed(AliITSUSeed* seed, Bool_t branch=kFALSE);
70   Bool_t                 NeedToKill(AliITSUSeed* seed, Int_t flag);
71   Bool_t                 GetRoadWidth(AliITSUSeed* seed, int ilrA);
72   Int_t                  CheckCluster(AliITSUSeed* seed, Int_t lr, Int_t clID);
73   void                   AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr);
74   //
75   AliITSUSeed*           NewSeedFromPool(const AliITSUSeed* src=0);
76   void                   ResetSeedsPool();
77   void                   MarkSeedFree(AliITSUSeed* seed );
78
79   AliITSUTrackHyp*       GetTrackHyp(Int_t id)               const  {return (AliITSUTrackHyp*)fHypStore.UncheckedAt(id);}
80   void                   SetTrackHyp(AliITSUTrackHyp* hyp,Int_t id) {fHypStore.AddAtAndExpand(hyp,id);}
81   void                   DeleteLastSeedFromPool()                   {fSeedsPool.RemoveLast();}
82   void                   SaveCurrentTrackHypotheses();
83   void                   FinalizeHypotheses();
84   void                   UpdateESDTrack(AliITSUTrackHyp* hyp);
85   void                   CookMCLabel(AliITSUTrackHyp* hyp);
86  //
87  protected:
88   TObject*&              NextFreeSeed();
89   //
90  private:
91   //
92   AliITSUTrackerGlo(const AliITSUTrackerGlo&);
93   AliITSUTrackerGlo &operator=(const AliITSUTrackerGlo &tr);
94   //
95  protected:
96   AliITSUReconstructor*           fReconstructor;  // ITS global reconstructor 
97   AliITSURecoDet*                 fITS;            // interface to ITS
98   AliESDtrack*                    fCurrESDtrack;   // current esd track in processing
99   Double_t                        fCurrMass;       // current track mass
100   Double_t                        fTrImpData[kNTrImpData];  // data on track impact on the layer
101   //
102   // the seeds management to be optimized
103   TObjArray                       fHypStore;       // storage for tracks hypotheses
104   AliITSUTrackHyp*                fCurrHyp;        // hypotheses container for current track
105   TClonesArray                    fSeedsPool;      //! pool for seeds
106   TArrayI                         fFreeSeedsID;    //! array of ID's of freed seeds
107   Int_t                           fNFreeSeeds;     //! number of seeds freed in the pool
108   Int_t                           fLastSeedID;     //! id of the pool seed on which is returned by the NextFreeSeed method
109   //
110   AliITSUTrackCond                fTrCond;         // tmp, to be moved to recoparam
111   Int_t                           fTrackPhase;     // tracking phase
112   Int_t*                          fClInfo;         //! auxiliary track cluster info
113   //
114   static const Double_t           fgkToler;        // tracking tolerance
115   //
116   ClassDef(AliITSUTrackerGlo,1)   //ITS upgrade tracker
117     
118 };
119
120 //_________________________________________________________________________
121 inline void AliITSUTrackerGlo::AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr)
122 {
123   // add new seed prolongation hypothesis 
124   fCurrHyp->AddSeed(seed,lr);
125   printf("*** Adding: "); seed->Print();
126 }
127
128 //________________________________________
129 inline TObject *&AliITSUTrackerGlo::NextFreeSeed()
130 {
131   // return next free slot where the seed can be created
132   fLastSeedID = fNFreeSeeds ? fFreeSeedsID.GetArray()[--fNFreeSeeds] : fSeedsPool.GetEntriesFast();
133   //  AliInfo(Form("%d",fLastSeedID));
134   return fSeedsPool[ fLastSeedID ];
135   //
136 }
137
138 //_________________________________________________________________________
139 inline AliITSUSeed* AliITSUTrackerGlo::NewSeedFromPool(const AliITSUSeed* src)
140 {
141   // create new seed, optionally copying from the source
142   AliITSUSeed* sd =  src ? new( NextFreeSeed() ) AliITSUSeed(*src) : new( NextFreeSeed() ) AliITSUSeed();
143   sd->SetPoolID(fLastSeedID);
144   return sd;
145 }
146
147 //_________________________________________________________________________
148 inline void AliITSUTrackerGlo::KillSeed(AliITSUSeed* seed, Bool_t branch)
149 {
150   // flag seed as killed, if requested, kill recursively its parents whose sole child is the seed being killed
151   seed->Kill();
152   seed = (AliITSUSeed*)seed->GetParent();
153   if (seed && !seed->DecChildren() && branch) KillSeed(seed,branch);
154 }
155
156 #endif
157