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 #define _FILL_CONTROL_HISTOS_
   7
   8 //-------------------------------------------------------------------------
   9 //                ITS upgrade tracker base class
  10 //-------------------------------------------------------------------------
  11
  12 #include "AliTracker.h"
  13 #include "AliESDEvent.h"
  14 #include "AliITSUSeed.h"
  15 #include "AliITSUTrackCond.h"
  16 #include "AliITSUTrackHyp.h"
  17
  18 class AliITSUReconstructor;
  19 class AliITSURecoDet;
  20 class AliITSUClusterPix;
  21 class AliESDtrack;
  22 class AliITSURecoLayer;
  23 class TTree;
  24
  25
  26 //-------------------------------------------------------------------------
  27 class AliITSUTrackerGlo : public AliTracker {
  28
  29   public:
  30   enum {kClus2Tracks,kPropBack,kRefitInw,kNTrackingPhases};   // tracking phases
  31   enum { // info from track extrapolation to layer for cluster check
  32     kTrXIn ,kTrYIn ,kTrZIn ,kTrPhiIn , // entrance (outer) point on the layer from above
  33     kTrXOut,kTrYOut,kTrZOut,kTrPhiOut, // exit (inner) point on the layer
  34     kTrPhi0, kTrDPhi, kTrZ0, kTrDZ,     // mean phi,dPhi, mean z, dZ (don't change this order)
  35     kNTrImpData};
  36   //
  37   enum {kMissingCluster=0  // no cluster found on this layer
  38         ,kTransportFailed=1  // seed did not reach target layer
  39         ,kRWCheckFailed =2  // failed to rotate the seed to frame of the layer impact point
  40   };
  41   enum {kStopSearchOnSensor,kClusterNotMatching,kClusterMatching}; // flags for track-to-cluster checks
  42   //
  43   enum {kDummyLabel=-3141593};
  44   //
  45   AliITSUTrackerGlo(AliITSUReconstructor* rec);
  46   virtual ~AliITSUTrackerGlo();
  47
  48   virtual Int_t          Clusters2Tracks(AliESDEvent *event);
  49   virtual Int_t          PropagateBack(AliESDEvent *event);
  50   virtual Int_t          RefitInward(AliESDEvent *event);
  51   virtual Int_t          LoadClusters(TTree * treeRP=0);
  52   virtual void           UnloadClusters();
  53   virtual AliCluster*    GetCluster(Int_t index) const;
  54   void                   FlagSplitClusters();
  55   Bool_t                 ContainsSplitCluster(const AliITSUSeed* seed, Int_t maxSize=99999);
  56   void                   PrintSeedClusters(const AliITSUSeed* seed, Option_t* option="");
  57   //
  58   Int_t                  GetCountPronlongationTrials() const {return fCountProlongationTrials;}
  59   Int_t                  GetCountITSin()               const {return fCountITSin;}
  60   Int_t                  GetCountITSout()              const {return fCountITSout;}
  61   Int_t                  GetCountITSrefit()            const {return fCountITSrefit;}
  62
  63   //------------------------------------
  64   AliITSURecoDet*        GetITSInterface()       const {return fITS;}
  65   //
  66   //------------------------------------
  67   Bool_t                 NeedToProlong(AliESDtrack* estTr);
  68   void                   Init(AliITSUReconstructor* rec);
  69   void                   FindTrack(AliESDtrack* esdTr, Int_t esdID);
  70   void                   CreateDefaultTrackCond();
  71   Bool_t                 InitHypothesis(AliESDtrack *esdTr, Int_t esdID);
  72   Bool_t                 TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo);
  73   Bool_t                 TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo);
  74   Bool_t                 TransportToLayerX(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t xStop);
  75   Bool_t                 GoToExitFromLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kTRUE);
  76   Bool_t                 GoToExitFromLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kTRUE);
  77   Bool_t                 GoToEntranceToLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
  78   Bool_t                 GoToEntranceToLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
  79   Bool_t                 PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
  80   Bool_t                 PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
  81   Bool_t                 RefitTrack(AliITSUTrackHyp* trc, Double_t r, Bool_t stopAtLastCl=kFALSE);
  82   Int_t                  GetTrackingPhase()                 const {return fTrackPhase;}
  83
  84   //
  85   void                   KillSeed(AliITSUSeed* seed, Bool_t branch=kFALSE);
  86   Bool_t                 NeedToKill(AliITSUSeed* seed, Int_t flag);
  87   Bool_t                 GetRoadWidth(AliITSUSeed* seed, int ilrA);
  88   Int_t                  CheckCluster(AliITSUSeed* seed, Int_t lr, Int_t clID);
  89   void                   AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr);
  90   Bool_t                 AddSeedBranch(AliITSUSeed* seed);
  91   void                   ValidateAllowedBranches(Int_t accMax);
  92   void                   ValidateAllowedCandidates(Int_t ilr, Int_t accMax);
  93   void                   FlagSeedClusters(const AliITSUSeed* seed, Bool_t flg);
  94   //
  95   AliITSUSeed*           NewSeedFromPool(const AliITSUSeed* src=0);
  96   void                   ResetSeedsPool();
  97   void                   MarkSeedFree(AliITSUSeed* seed );
  98
  99   AliITSUTrackHyp*       GetTrackHyp(Int_t id)               const  {return (AliITSUTrackHyp*)fHypStore.UncheckedAt(id);}
 100   void                   SetTrackHyp(AliITSUTrackHyp* hyp,Int_t id) {fHypStore.AddAtAndExpand(hyp,id);}
 101   void                   DeleteLastSeedFromPool()                   {fSeedsPool.RemoveLast();}
 102   void                   SaveCurrentTrackHypotheses();
 103   void                   FinalizeHypotheses();
 104   void                   UpdateESDTrack(AliITSUTrackHyp* hyp,Int_t flag);
 105   void                   CookMCLabel(AliITSUTrackHyp* hyp);
 106   void                   SetTrackingPhase(Int_t p)        {fTrackPhase = p;}
 107  //
 108  protected:
 109   TObject*&              NextFreeSeed();
 110   //
 111  private:
 112   //
 113   AliITSUTrackerGlo(const AliITSUTrackerGlo&);
 114   AliITSUTrackerGlo &operator=(const AliITSUTrackerGlo &tr);
 115   //
 116  protected:
 117   AliITSUReconstructor*           fReconstructor;  // ITS global reconstructor
 118   AliITSURecoDet*                 fITS;            // interface to ITS, borrowed from reconstructor
 119   AliESDtrack*                    fCurrESDtrack;   // current esd track in processing
 120   Int_t                           fCurrESDtrMClb;  // its eventual mc label
 121   Double_t                        fCurrMass;       // current track mass
 122   Double_t                        fTrImpData[kNTrImpData];  // data on track impact on the layer
 123   //
 124   Int_t                           fCountProlongationTrials;   // number of TPC seeds
 125   Int_t                           fCountITSin;     // number of successful ITSin
 126   Int_t                           fCountITSout;    // number of successful ITSout
 127   Int_t                           fCountITSrefit;  // number of successful ITSrefit
 128   //
 129   // the seeds management to be optimized
 130   TObjArray                       fHypStore;       // storage for tracks hypotheses
 131   AliITSUSeed*                    fLayerCandidates[AliITSUTrackCond::kMaxCandidates]; // array for branches of current track prolongation
 132   Int_t                           fNBranchesAdded; // number of branches created for current seed in prolongation
 133   Int_t                           fNCandidatesAdded; // number of candidates added for current seed in prolongation
 134   AliITSUTrackHyp*                fCurrHyp;        // hypotheses container for current track
 135   TClonesArray                    fSeedsPool;      //! pool for seeds
 136   TArrayI                         fFreeSeedsID;    //! array of ID's of freed seeds
 137   Int_t                           fNFreeSeeds;     //! number of seeds freed in the pool
 138   Int_t                           fLastSeedID;     //! id of the pool seed on which is returned by the NextFreeSeed method
 139   //
 140   TObjArray                       fDefTrackConds;  //! default tracking conditions
 141   AliITSUTrackCond*               fCurrTrackCond;  //! current tracking condition
 142   Int_t                           fCurrActLrID;    //! current active layer ID being processed (set only when needed, not guaranteed)
 143   AliITSURecoLayer*               fCurrLayer;      //! current layer being processed  (set only when needed, not guaranteed)
 144   Int_t                           fTrackPhase;     //! tracking phase
 145   Int_t*                          fClInfo;         //! auxiliary track cluster info
 146   //
 147   static const Double_t           fgkToler;        // tracking tolerance
 148   //
 149 #ifdef  _FILL_CONTROL_HISTOS_
 150   // this code is only for special histos needed to extract some control parameters
 151   void BookControlHistos();
 152   TObjArray* fCHistoArr;
 153   enum {kHResY=0,kHResYP=10,kHResZ=20,kHResZP=30,kHChi2Cl=40,kHChi2Nrm=50,kHBestInBranch=60,kHBestInCand=70};
 154   enum {kHistosPhase=100};
 155   //
 156 #endif
 157
 158   ClassDef(AliITSUTrackerGlo,1)   //ITS upgrade tracker
 159
 160 };
 161
 162 //________________________________________
 163 inline TObject *&AliITSUTrackerGlo::NextFreeSeed()
 164 {
 165   // return next free slot where the seed can be created
 166   fLastSeedID = fNFreeSeeds ? fFreeSeedsID.GetArray()[--fNFreeSeeds] : fSeedsPool.GetEntriesFast();
 167   //  AliInfo(Form("%d",fLastSeedID));
 168   return fSeedsPool[ fLastSeedID ];
 169   //
 170 }
 171
 172 //_________________________________________________________________________
 173 inline AliITSUSeed* AliITSUTrackerGlo::NewSeedFromPool(const AliITSUSeed* src)
 174 {
 175   // create new seed, optionally copying from the source
 176   AliITSUSeed* sd =  src ? new( NextFreeSeed() ) AliITSUSeed(*src) : new( NextFreeSeed() ) AliITSUSeed();
 177   sd->SetPoolID(fLastSeedID);
 178   return sd;
 179 }
 180
 181 //_________________________________________________________________________
 182 inline void AliITSUTrackerGlo::KillSeed(AliITSUSeed* seed, Bool_t branch)
 183 {
 184   // flag seed as killed, if requested, kill recursively its parents whose sole child is the seed being killed
 185   seed->Kill();
 186   seed = (AliITSUSeed*)seed->GetParent();
 187   if (seed && !seed->DecChildren() && branch) KillSeed(seed,branch);
 188 }
 189
 190 //_________________________________________________________________________
 191 inline void AliITSUTrackerGlo::AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr)
 192 {
 193   // add new seed prolongation hypothesis
 194   fCurrHyp->AddSeed(seed,lr);
 195 }
 196
 197 #endif
 198