Net Particle updates: remove printout for PID check and TDataBasePDG for charge in...
[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;
c61e50c3 20
32d38de2 21class TTree;
22
c61e50c3 23
32d38de2 24//-------------------------------------------------------------------------
25class AliITSUTrackerGlo : public AliTracker {
26
27 public:
28 enum { // info from track extrapolation to layer for cluster check
29 kTrXIn ,kTrYIn ,kTrZIn ,kTrPhiIn , // entrance (outer) point on the layer from above
30 kTrXOut,kTrYOut,kTrZOut,kTrPhiOut, // exit (inner) point on the layer
31 kTrPhi0, kTrDPhi, kTrZ0, kTrDZ, // mean phi,dPhi, mean z, dZ (don't change this order)
c61e50c3 32 kNTrImpData};
32d38de2 33 //
f8832015 34 enum {kMissingCluster=0 // no cluster found on this layer
35 ,kTransportFailed=1 // seed did not reach target layer
32d38de2 36 ,kRWCheckFailed =2 // failed to rotate the seed to frame of the layer impact point
37 };
f8832015 38 enum {kStopSearchOnSensor,kClusterNotMatching,kClusterMatching}; // flags for track-to-cluster checks
32d38de2 39
40 AliITSUTrackerGlo(AliITSUReconstructor* rec);
41 virtual ~AliITSUTrackerGlo();
42
43 virtual Int_t Clusters2Tracks(AliESDEvent *event);
44 virtual Int_t PropagateBack(AliESDEvent *event);
45 virtual Int_t RefitInward(AliESDEvent *event);
46 virtual Int_t LoadClusters(TTree * treeRP=0);
47 virtual void UnloadClusters();
48 virtual AliCluster* GetCluster(Int_t index) const;
0091e9f0 49 //------------------------------------
50 AliITSURecoDet* GetITSInterface() const {return fITS;}
51 //
32d38de2 52 //------------------------------------
53 Bool_t NeedToProlong(AliESDtrack* estTr);
54 void Init(AliITSUReconstructor* rec);
716ccba7 55 void FindTrack(AliESDtrack* esdTr, Int_t esdID);
3e4e3c23 56 Bool_t InitHypothesis(AliESDtrack *esdTr, Int_t esdID);
32d38de2 57 Bool_t TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo);
3e4e3c23 58 Bool_t TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo);
44785f3e 59 Bool_t PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
3e4e3c23 60 Bool_t PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
44785f3e 61 //
c8d1f258 62 Bool_t NeedToKill(AliITSUSeed* seed, Int_t flag);
c61e50c3 63 Bool_t GetRoadWidth(AliITSUSeed* seed, int ilrA);
f8832015 64 Int_t CheckCluster(AliITSUSeed* seed, Int_t lr, Int_t clID);
65 void AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr);
c61e50c3 66 //
67 AliITSUSeed* NewSeedFromPool(const AliITSUSeed* src=0);
716ccba7 68 AliITSUTrackHyp* GetTrackHyp(Int_t id) const {return (AliITSUTrackHyp*)fHypStore.UncheckedAt(id);}
69 void SetTrackHyp(AliITSUTrackHyp* hyp,Int_t id) {fHypStore.AddAtAndExpand(hyp,id);}
70 void DeleteLastSeedFromPool() {fSeedsPool.RemoveLast();}
71 void SaveCurrentTrackHypotheses();
3e4e3c23 72 void FinalizeHypotheses();
716ccba7 73 //
44785f3e 74
32d38de2 75 private:
76
77 AliITSUTrackerGlo(const AliITSUTrackerGlo&);
78 AliITSUTrackerGlo &operator=(const AliITSUTrackerGlo &tr);
79 //
80 protected:
81 AliITSUReconstructor* fReconstructor; // ITS global reconstructor
82 AliITSURecoDet* fITS; // interface to ITS
0091e9f0 83 AliESDtrack* fCurrESDtrack; // current esd track in processing
32d38de2 84 Double_t fCurrMass; // current track mass
c61e50c3 85 Double_t fTrImpData[kNTrImpData]; // data on track impact on the layer
32d38de2 86 //
c61e50c3 87 // the seeds management to be optimized
716ccba7 88 TObjArray fHypStore; // storage for tracks hypotheses
89 AliITSUTrackHyp* fCurrHyp; // hypotheses container for current track
32d38de2 90 TClonesArray fSeedsPool; // pool for seeds
c8d1f258 91 //
92 AliITSUTrackCond fTrCond; // tmp, to be moved to recoparam
93 //
32d38de2 94 ClassDef(AliITSUTrackerGlo,1) //ITS upgrade tracker
95
96};
f8832015 97
98//_________________________________________________________________________
99inline void AliITSUTrackerGlo::AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr)
100{
101 // add new seed prolongation hypothesis
716ccba7 102 fCurrHyp->AddSeed(seed,lr);
c8d1f258 103 printf("*** Adding: "); seed->Print();
f8832015 104}
105
106
32d38de2 107#endif
108