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
  21 class TTree;
  22
  23
  24 //-------------------------------------------------------------------------
  25 class 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)
  32     kNTrImpData};
  33   //
  34   enum {kMissingCluster=0  // no cluster found on this layer
  35         ,kTransportFailed=1  // seed did not reach target layer
  36         ,kRWCheckFailed =2  // failed to rotate the seed to frame of the layer impact point
  37   };
  38   enum {kStopSearchOnSensor,kClusterNotMatching,kClusterMatching}; // flags for track-to-cluster checks
  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;
  49   //------------------------------------
  50   AliITSURecoDet*        GetITSInterface()       const {return fITS;}
  51   //
  52   //------------------------------------
  53   Bool_t                 NeedToProlong(AliESDtrack* estTr);
  54   void                   Init(AliITSUReconstructor* rec);
  55   void                   FindTrack(AliESDtrack* esdTr, Int_t esdID);
  56   Bool_t                 InitHypothesis(AliESDtrack *esdTr, Int_t esdID);
  57   Bool_t                 TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo);
  58   Bool_t                 TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo);
  59   Bool_t                 PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
  60   Bool_t                 PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
  61   //
  62   Bool_t                 NeedToKill(AliITSUSeed* seed, Int_t flag);
  63   Bool_t                 GetRoadWidth(AliITSUSeed* seed, int ilrA);
  64   Int_t                  CheckCluster(AliITSUSeed* seed, Int_t lr, Int_t clID);
  65   void                   AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr);
  66   //
  67   AliITSUSeed*           NewSeedFromPool(const AliITSUSeed* src=0);
  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();
  72   void                   FinalizeHypotheses();
  73  //
  74
  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
  83   AliESDtrack*                    fCurrESDtrack;   // current esd track in processing
  84   Double_t                        fCurrMass;       // current track mass
  85   Double_t                        fTrImpData[kNTrImpData];  // data on track impact on the layer
  86   //
  87   // the seeds management to be optimized
  88   TObjArray                       fHypStore;       // storage for tracks hypotheses
  89   AliITSUTrackHyp*                fCurrHyp;        // hypotheses container for current track
  90   TClonesArray                    fSeedsPool;      // pool for seeds
  91   //
  92   AliITSUTrackCond                fTrCond;         // tmp, to be moved to recoparam
  93   //
  94   ClassDef(AliITSUTrackerGlo,1)   //ITS upgrade tracker
  95
  96 };
  97
  98 //_________________________________________________________________________
  99 inline void AliITSUTrackerGlo::AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr)
 100 {
 101   // add new seed prolongation hypothesis
 102   fCurrHyp->AddSeed(seed,lr);
 103   printf("*** Adding: "); seed->Print();
 104 }
 105
 106
 107 #endif
 108