ITS/AliITStrackerSA.h

   1 #ifndef ALIITSTRACKERSA_H
   2 #define ALIITSTRACKERSA_H
   3
   4
   5
   6 #include "AliITSgeomTGeo.h"
   7 #include "AliITStrackerMI.h"
   8
   9 /* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
  10  * See cxx source for full Copyright notice                               */
  11
  12 /* $Id$ */
  13
  14
  15 ////////////////////////////////////////////////////
  16 //  Stand alone tracker class                     //
  17 //  Origin:  Elisabetta Crescio                   //
  18 //  e-mail:  crescio@to.infn.it                   //
  19 //  adapted for cosmics by A.Dainese              //
  20 ////////////////////////////////////////////////////
  21
  22 class AliITSclusterTable;
  23 class AliITStrackSA;
  24 class AliESDVertex;
  25 class AliESDEvent;
  26 class AliITSVertexer;
  27 class TTree;
  28 class TArrayD;
  29
  30 class AliITStrackerSA : public AliITStrackerMI {
  31
  32
  33  public:
  34
  35   AliITStrackerSA();
  36   AliITStrackerSA(const Char_t *geom);
  37   AliITStrackerSA(const Char_t *geom,AliESDVertex *vert);
  38   AliITStrackerSA(const Char_t *geom,AliITSVertexer *vertexer);
  39   AliITStrackerSA(const AliITStrackerSA& tracker);
  40   AliITStrackerSA& operator=(const AliITStrackerSA& source);
  41   virtual ~AliITStrackerSA();
  42   virtual Int_t Clusters2Tracks(AliESDEvent *event);
  43   Int_t FindTracks(AliESDEvent* event);
  44
  45   AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,Bool_t onePoint=kFALSE);
  46   Int_t FindTrackLowChiSquare() const;
  47   Int_t LoadClusters(TTree *cf) {Int_t rc=AliITStrackerMI::LoadClusters(cf); SetClusterTree(cf); SetSixPoints(kTRUE); return rc;}
  48   void SetVertex(AliESDVertex *vtx){fVert = vtx;}
  49   void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
  50   void SetSixPoints(Bool_t sp = kFALSE){fSixPoints = sp;}
  51   Bool_t GetSixPoints() const {return fSixPoints;}
  52   void SetOuterStartLayer(Int_t osl = 0) {if(osl>(AliITSgeomTGeo::GetNLayers()-2)) osl=AliITSgeomTGeo::GetNLayers()-2; fOuterStartLayer = osl;}
  53   Int_t GetOuterStartLayer() const {return fOuterStartLayer;}
  54   void SetSAFlag(Bool_t fl){fITSStandAlone=fl;}  // StandAlone flag setter
  55   Bool_t GetSAFlag() const {return fITSStandAlone;} // StandAlone flag getter
  56   void SetFixedWindowSizes(Int_t n=46, Double_t *phi=0, Double_t *lam=0);
  57   void SetCalculatedWindowSizes(Int_t n=10, Float_t phimin=0.002, Float_t phimax=0.0145, Float_t lambdamin=0.003, Float_t lambdamax=0.008);
  58
  59   enum {kSAflag=0x8000}; //flag to mark clusters used in the SA tracker
  60
  61  protected:
  62
  63   //Initialization
  64   void Init();
  65   void ResetForFinding();
  66   void UpdatePoints();
  67
  68   static Double_t Curvature(Double_t x1,Double_t y1,Double_t x2,Double_t y2,
  69                      Double_t x3,Double_t y3);
  70
  71   Double_t ChoosePoint(Double_t p1, Double_t p2, Double_t pp);
  72
  73   static Int_t   FindIntersection(Float_t a1, Float_t b1, Float_t c1, Float_t c2,
  74                            Float_t& x1,Float_t& y1, Float_t& x2, Float_t& y2);
  75   static Int_t   FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2,
  76                        Float_t x3, Float_t y3,Float_t& a, Float_t& b,
  77                        Float_t& c);
  78
  79   static Int_t FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6);
  80   static Int_t Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5,
  81               Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
  82               Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16,
  83               Int_t gl17, Int_t gl18, Int_t numberofpoints=6);
  84
  85   Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow,
  86                        AliITStrackSA* trs,Double_t zvertex,Int_t flagp);
  87
  88   void GetCoorAngles(AliITSRecPoint* cl,Double_t &phi,Double_t &lambda,Float_t &x,Float_t &y,Float_t &z,Double_t* vertex);
  89   void GetCoorErrors(AliITSRecPoint* cl,Float_t &sx,Float_t &sy, Float_t &sz);
  90
  91   AliITSclusterTable* GetClusterCoord(Int_t layer,Int_t n) const {return (AliITSclusterTable*)fCluCoord[layer]->UncheckedAt(n);}
  92   void RemoveClusterCoord(Int_t layer, Int_t n) {fCluCoord[layer]->RemoveAt(n);fCluCoord[layer]->Compress();}
  93
  94
  95   Double_t fPhiEstimate; //Estimation of phi angle on next layer
  96   Bool_t fITSStandAlone; //Tracking is performed in the ITS alone if kTRUE
  97   Float_t fPoint1[2];   //! coord. of 1-st point to evaluate the curvature
  98   Float_t fPoint2[2];   //! coord. of 2-nd point to evaluate the curvature
  99   Float_t fPoint3[2];   //! coord. of 3-rd point to evaluate the curvature
 100   Float_t fPointc[2];   //! current point coord (for curvature eval.)
 101   Double_t fLambdac;    //! current value of the Lambda angle in the window
 102   Double_t fPhic;       //! current value of the Phi angle in the window
 103   Float_t fCoef1;       //! param. of the equation of the circ. approx a layer
 104   Float_t fCoef2;       //! param. of the equation of the circ. approx a layer
 105   Float_t fCoef3;       //! param. of the equation of the circ. approx a layer
 106   Int_t fNloop;         //  Number of iterqations on phi and lambda windows
 107   Double_t *fPhiWin;    // phi window sizes
 108   Double_t *fLambdaWin; // lambda window sizes
 109   AliESDVertex *fVert;        //! primary vertex
 110   AliITSVertexer *fVertexer;  //! vertexer
 111   TObjArray *fListOfTracks;   //! container for found tracks
 112   TObjArray *fListOfSATracks; //! container for found SA tracks
 113   TTree *fITSclusters;        //! pointer to ITS tree of clusters
 114   Bool_t fSixPoints;          // If true 6/6 points are required (default). 5/6 otherwise
 115   Int_t fOuterStartLayer;     // Search for tracks with <6 points: outer layer to start from
 116
 117   TClonesArray** fCluLayer; //! array with clusters
 118   TClonesArray** fCluCoord; //! array with cluster info
 119
 120   ClassDef(AliITStrackerSA,7)
 121 };
 122
 123 #endif
 124