]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ITS/UPGRADE/AliITSUTrackerGlo.h
An example Config.C for ITSU pileup studies in pp
[u/mrichter/AliRoot.git] / ITS / UPGRADE / AliITSUTrackerGlo.h
index d444801754258057bb8e06a30b288e2b0d2f3a33..1e73ae23b54166fc9ba3df13091eddb6521ee36e 100644 (file)
@@ -6,26 +6,27 @@
 //-------------------------------------------------------------------------
 //                ITS upgrade tracker base class
 //-------------------------------------------------------------------------
-
 #include "AliTracker.h"
 #include "AliESDEvent.h"
 #include "AliITSUSeed.h"
 #include "AliITSUTrackCond.h"
 #include "AliITSUTrackHyp.h"
+#include "AliITSUAux.h"
+#include "AliITSUMatLUT.h"
+#include <TArrayI.h>
 
 class AliITSUReconstructor;
 class AliITSURecoDet;
 class AliITSUClusterPix;
 class AliESDtrack;
-
+class AliITSURecoLayer;
 class TTree;
 
-
 //-------------------------------------------------------------------------
 class AliITSUTrackerGlo : public AliTracker {
 
   public:
-  enum {kClus2Tracks,kPropBack,kRefitInw};   // tracking phases
+  enum {kClus2Tracks,kPropBack,kRefitInw,kNTrackingPhases};   // tracking phases
   enum { // info from track extrapolation to layer for cluster check
     kTrXIn ,kTrYIn ,kTrZIn ,kTrPhiIn , // entrance (outer) point on the layer from above 
     kTrXOut,kTrYOut,kTrZOut,kTrPhiOut, // exit (inner) point on the layer
@@ -37,7 +38,9 @@ class AliITSUTrackerGlo : public AliTracker {
        ,kRWCheckFailed =2  // failed to rotate the seed to frame of the layer impact point
   };
   enum {kStopSearchOnSensor,kClusterNotMatching,kClusterMatching}; // flags for track-to-cluster checks
-
+  //
+  enum {kDummyLabel=-3141593};
+  //
   AliITSUTrackerGlo(AliITSUReconstructor* rec);
   virtual ~AliITSUTrackerGlo();
 
@@ -47,25 +50,47 @@ class AliITSUTrackerGlo : public AliTracker {
   virtual Int_t          LoadClusters(TTree * treeRP=0);
   virtual void           UnloadClusters();
   virtual AliCluster*    GetCluster(Int_t index) const;
+  void                   PrintSeedClusters(const AliITSUSeed* seed, Option_t* option="");
+  //
+  Int_t                  GetCountPronlongationTrials() const {return fCountProlongationTrials;}
+  Int_t                  GetCountITSin()               const {return fCountITSin;}
+  Int_t                  GetCountITSout()              const {return fCountITSout;}
+  Int_t                  GetCountITSrefit()            const {return fCountITSrefit;}
+
   //------------------------------------
   AliITSURecoDet*        GetITSInterface()       const {return fITS;}
   //
   //------------------------------------
-  Bool_t                 NeedToProlong(AliESDtrack* estTr);
+  Bool_t                 NeedToProlong(AliESDtrack* estTr, Int_t esdID);
   void                   Init(AliITSUReconstructor* rec);
   void                   FindTrack(AliESDtrack* esdTr, Int_t esdID);
-  Bool_t                 InitHypothesis(AliESDtrack *esdTr, Int_t esdID);
-  Bool_t                 TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo);
-  Bool_t                 TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo);
+  void                   CreateDefaultTrackCond();
+  AliITSUTrackHyp*       InitHypothesis(AliESDtrack *esdTr, Int_t esdID);
+  Bool_t                 TransportToLayer(AliITSUSeed* seed, Int_t lFrom, Int_t lTo, Double_t rLim=-1);
+  Bool_t                 TransportToLayer(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t rLim=-1);
+  Bool_t                 TransportToLayerX(AliExternalTrackParam* seed, Int_t lFrom, Int_t lTo, Double_t xStop);  
+  Bool_t                 GoToExitFromLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kTRUE);
+  Bool_t                 GoToExitFromLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kTRUE);
+  Bool_t                 GoToEntranceToLayer(AliITSUSeed* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
+  Bool_t                 GoToEntranceToLayer(AliExternalTrackParam* seed, AliITSURecoLayer* lr, Int_t dir, Bool_t check=kFALSE);
   Bool_t                 PropagateSeed(AliITSUSeed *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
   Bool_t                 PropagateSeed(AliExternalTrackParam *seed, Double_t xToGo, Double_t mass, Double_t maxStep=1.0, Bool_t matCorr=kTRUE);
-  Bool_t                 RefitTrack(AliITSUTrackHyp* trc, Double_t r);
+  Double_t               RefitTrack(AliITSUTrackHyp* trc, Double_t r, Int_t& nclFit, Int_t stopCond=0);
+  Double_t               GetMaterialBudget(const double* pnt0, const double* pnt1, double& x2x0, double& rhol) const;
+
+  Int_t                  GetTrackingPhase()                 const {return fTrackPhaseID;}
+
   //
   void                   KillSeed(AliITSUSeed* seed, Bool_t branch=kFALSE);
   Bool_t                 NeedToKill(AliITSUSeed* seed, Int_t flag);
   Bool_t                 GetRoadWidth(AliITSUSeed* seed, int ilrA);
+  Bool_t                 CheckBackwardMatching(AliITSUSeed* seed);
   Int_t                  CheckCluster(AliITSUSeed* seed, Int_t lr, Int_t clID);
   void                   AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr);
+  Bool_t                 AddSeedBranch(AliITSUSeed* seed);
+  void                   ValidateAllowedBranches(Int_t accMax);
+  void                   ValidateAllowedCandidates(Int_t ilr, Int_t accMax);
+  void                   FlagSeedClusters(const AliITSUSeed* seed, Bool_t flg, UShort_t hypRef);
   //
   AliITSUSeed*           NewSeedFromPool(const AliITSUSeed* src=0);
   void                   ResetSeedsPool();
@@ -74,10 +99,20 @@ class AliITSUTrackerGlo : public AliTracker {
   AliITSUTrackHyp*       GetTrackHyp(Int_t id)               const  {return (AliITSUTrackHyp*)fHypStore.UncheckedAt(id);}
   void                   SetTrackHyp(AliITSUTrackHyp* hyp,Int_t id) {fHypStore.AddAtAndExpand(hyp,id);}
   void                   DeleteLastSeedFromPool()                   {fSeedsPool.RemoveLast();}
-  void                   SaveCurrentTrackHypotheses();
+  void                   CheckClusterSharingConflicts(AliITSUTrackHyp* hyp);
+  void                   SaveReducedHypothesesTree(AliITSUTrackHyp* dest);
+  void                   CleanHypothesis(AliITSUTrackHyp* hyp);
   void                   FinalizeHypotheses();
-  void                   UpdateESDTrack(AliITSUTrackHyp* hyp);
- //
+  Bool_t                 FinalizeHypothesis(AliITSUTrackHyp* hyp);
+  void                   UpdateESDTrack(AliITSUTrackHyp* hyp,Int_t flag);
+  void                   CookMCLabel(AliITSUTrackHyp* hyp);
+  void                   SetTrackingPhase(Int_t p)        {fTrackPhaseID = p;}
+  //
+  // monitoring stuff
+  void                   FlagSplitClusters();
+  Bool_t                 ContainsSplitCluster(const AliITSUSeed* seed, Int_t maxSize=99999);
+  void                   CheckClusterUsage();
+  //
  protected:
   TObject*&              NextFreeSeed();
   //
@@ -88,35 +123,60 @@ class AliITSUTrackerGlo : public AliTracker {
   //
  protected:
   AliITSUReconstructor*           fReconstructor;  // ITS global reconstructor 
-  AliITSURecoDet*                 fITS;            // interface to ITS
+  AliITSURecoDet*                 fITS;            // interface to ITS, borrowed from reconstructor
+  AliITSUMatLUT*                  fMatLUT;         // material lookup table
   AliESDtrack*                    fCurrESDtrack;   // current esd track in processing
+  Int_t                           fCurrESDtrMClb;  // its eventual mc label
   Double_t                        fCurrMass;       // current track mass
   Double_t                        fTrImpData[kNTrImpData];  // data on track impact on the layer
   //
+  Int_t                           fCountProlongationTrials;   // number of TPC seeds
+  Int_t                           fCountITSin;     // number of successful ITSin 
+  Int_t                           fCountITSout;    // number of successful ITSout
+  Int_t                           fCountITSrefit;  // number of successful ITSrefit 
+  Int_t                           fNTracksESD;     // number of esd tracks
+  //
   // the seeds management to be optimized
   TObjArray                       fHypStore;       // storage for tracks hypotheses
-  AliITSUTrackHyp*                fCurrHyp;        // hypotheses container for current track
+  Int_t                           fLayerMaxCandidates; //! size of tmp candidates array 
+  AliITSUSeed**                   fLayerCandidates;//! array for branches of current track prolongation
+  Int_t                           fNBranchesAdded; // number of branches created for current seed in prolongation
+  Int_t                           fNCandidatesAdded; // number of candidates added for current seed in prolongation
+  AliITSUTrackHyp*                fCurrHyp;        //! hypotheses container for current track
+  AliITSUTrackHyp*                fWorkHyp;        //! temporary hypothesis for track finding
   TClonesArray                    fSeedsPool;      //! pool for seeds
   TArrayI                         fFreeSeedsID;    //! array of ID's of freed seeds
+  TArrayI                         fESDIndex;       //! array of ID's of freed seeds
   Int_t                           fNFreeSeeds;     //! number of seeds freed in the pool
   Int_t                           fLastSeedID;     //! id of the pool seed on which is returned by the NextFreeSeed method
+  Int_t                           fNLrActive;      //! number of active layers
   //
-  AliITSUTrackCond                fTrCond;         // tmp, to be moved to recoparam
-  Int_t                           fTrackPhase;     // tracking phase
-  Int_t*                          fClInfo;         //! auxiliary track cluster info
+  TObjArray                       fDefTrackConds;  //! default tracking conditions
+  AliITSUTrackCond*               fCurrTrackCond;  //! current tracking condition
+  Int_t                           fCurrActLrID;    //! current active layer ID being processed (set only when needed, not guaranteed)
+  AliITSURecoLayer*               fCurrLayer;      //! current layer being processed  (set only when needed, not guaranteed)
+  Int_t                           fTrackPhaseID;   //! tracking phase (clusters2tracks, backward, inward)
+  Int_t                           fCurrPassID;     //! tracking pass (different tracking conditions)
+  Bool_t                          fUseMatLUT;      //! use material lookup table rather than TGeo
+  //
+  static const Double_t           fgkToler;        // tracking tolerance
+  //
+#ifdef  _ITSU_TUNING_MODE_
+  // this code is only for special histos needed to extract some control parameters
+  Int_t GetHistoID(Int_t lr, Int_t hid, Int_t pass=0, Int_t phase=0);
+  TObjArray* BookControlHistos(const char* pref);
+  TObjArray* fCHistoArrCorr; // set of histos for each tracking pass/phase: correct
+  TObjArray* fCHistoArrFake; // set of histos for each tracking pass/phase: fakse
+  enum {kHResY,kHResYP,kHResZ,kHResZP,kHChi2Cl,kHChi2Nrm,kHBestInBranch,kHBestInCand,kMaxHID=10};
+  enum {kHChiMatch,kHChiITSSA}; // custom histos 
+  enum {kHistosPhase=kMaxHID*(AliITSUAux::kMaxLayers+1),kHistosPass=kNTrackingPhases*kHistosPhase};
+  //
+#endif
   //
   ClassDef(AliITSUTrackerGlo,1)   //ITS upgrade tracker
     
 };
 
-//_________________________________________________________________________
-inline void AliITSUTrackerGlo::AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr)
-{
-  // add new seed prolongation hypothesis 
-  fCurrHyp->AddSeed(seed,lr);
-  printf("*** Adding: "); seed->Print();
-}
-
 //________________________________________
 inline TObject *&AliITSUTrackerGlo::NextFreeSeed()
 {
@@ -145,5 +205,33 @@ inline void AliITSUTrackerGlo::KillSeed(AliITSUSeed* seed, Bool_t branch)
   if (seed && !seed->DecChildren() && branch) KillSeed(seed,branch);
 }
 
+//_________________________________________________________________________
+inline void AliITSUTrackerGlo::AddProlongationHypothesis(AliITSUSeed* seed, Int_t lr)
+{
+  // add new seed prolongation hypothesis 
+#ifdef  _ITSU_TUNING_MODE_
+  seed->SetOrdCand(fCurrHyp->GetNSeeds(lr));
+#endif
+  fCurrHyp->AddSeed(seed,lr);  
+}
+
+//_________________________________________________________________________
+inline Double_t AliITSUTrackerGlo::GetMaterialBudget(const double* pnt0,const double* pnt1, double& x2x0, double& rhol) const
+{
+  double par[7];
+  if (fUseMatLUT && fMatLUT) {
+    double d = fMatLUT->GetMatBudget(pnt0,pnt1,par);
+    x2x0 = par[AliITSUMatLUT::kParX2X0];
+    rhol = par[AliITSUMatLUT::kParRhoL];    
+    return d;
+  }
+  else {
+    MeanMaterialBudget(pnt0,pnt1,par);
+    x2x0 = par[1];
+    rhol = par[0]*par[4];    
+    return par[4];
+  }
+}
+
 #endif