]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - ITS/AliITStrackerSA.h
Possibility to reconstruct tracks with 1 point + vertex, and possibility to reuse...
[u/mrichter/AliRoot.git] / ITS / AliITStrackerSA.h
index 3f21f9a2f81792c5de85cf7b2d3403a6369cd801..fb568294919578db8c0f0f0883d445dba35f73ed 100644 (file)
@@ -3,7 +3,8 @@
 
 
 
-#include "AliITStrackerV2.h"
+#include "AliITSgeomTGeo.h"
+#include "AliITStrackerMI.h"
 
 /* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 //  Stand alone tracker class                     //
 //  Origin:  Elisabetta Crescio                   //
 //  e-mail:  crescio@to.infn.it                   //
+//  adapted for cosmics by A.Dainese              //
 ////////////////////////////////////////////////////
 
 class AliITSclusterTable;
 class AliITStrackSA;
-class AliITSVertex;
+class AliESDVertex;
+class AliESDEvent;
 class AliITSVertexer;
+class TTree;
+class TArrayD;
 
-class AliITStrackerSA : public AliITStrackerV2 {
+class AliITStrackerSA : public AliITStrackerMI {
 
 
  public:
 
   AliITStrackerSA();
-  AliITStrackerSA(AliITSgeom *geom,AliITSVertex *vert);
-  AliITStrackerSA(AliITSgeom *geom,AliITSVertexer *vertexer);
-  AliITStrackerSA(AliITStrackerSA& tracker);
+  AliITStrackerSA(const Char_t *geom);
+  AliITStrackerSA(const Char_t *geom,AliESDVertex *vert);
+  AliITStrackerSA(const Char_t *geom,AliITSVertexer *vertexer);
+  AliITStrackerSA(const AliITStrackerSA& tracker);
+  AliITStrackerSA& operator=(const AliITStrackerSA& source);
   virtual ~AliITStrackerSA();  
-  void     FindTracks(TTree *clusterTree, TTree *out,Int_t evnumber=0,
-                     char *opt="6/6");
-
-  AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,
-                         Double_t *errorprimvert,char *opt="6/6");
-
-  AliITStrackV2* FindTrackLowChiSquare(TObjArray* tracklist, Int_t dim) const;
-  void SetVertex(AliITSVertex *vtx){fVert = vtx;}
+  virtual Int_t Clusters2Tracks(AliESDEvent *event);
+  Int_t FindTracks(AliESDEvent* event);
+
+  AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,Bool_t onePoint=kFALSE);
+  Int_t FindTrackLowChiSquare(TObjArray* tracklist, Int_t dim) const;
+  Int_t LoadClusters(TTree *cf) {Int_t rc=AliITStrackerMI::LoadClusters(cf); SetClusterTree(cf); SetSixPoints(kTRUE); return rc;}
+  void SetVertex(AliESDVertex *vtx){fVert = vtx;}
+  void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
+  void SetSixPoints(Bool_t sp = kFALSE){fSixPoints = sp;}
+  Bool_t GetSixPoints() const {return fSixPoints;}
+  void SetOuterStartLayer(Int_t osl = 0) {if(osl>(AliITSgeomTGeo::GetNLayers()-2)) osl=AliITSgeomTGeo::GetNLayers()-2; fOuterStartLayer = osl;}
+  Int_t GetOuterStartLayer() const {return fOuterStartLayer;}
+  void SetSAFlag(Bool_t fl){fITSStandAlone=fl;}  // StandAlone flag setter
+  Bool_t GetSAFlag() const {return fITSStandAlone;} // StandAlone flag getter
   void SetWindowSizes(Int_t n=46, Double_t *phi=0, Double_t *lam=0);
-  void UseFoundTracksV2(Int_t evnum,TTree* treev2, TTree* clustertree);
+
+  enum {kSAflag=0x8000}; //flag to mark clusters used in the SA tracker
 
  protected:
 
-  // copy constructor (NO copy allowed: the constructor is protected
-  // to avoid misuse)
-  AliITStrackerSA(const AliITStrackerSA& trkr);
-  // assignment operator (NO assignment allowed)
-  AliITStrackerSA& operator=(const AliITStrackerSA& /* trkr */);
   //Initialization
   void Init();
-  Int_t     GetFlagLoadedClusters() const {return fFlagLoad;}
-  
-  void     GetClusterGCoordinates(AliITSclusterV2* cluster,
-                                 Int_t module,Double_t& x, Double_t& y,Double_t& z);
-  void     GetClusterGErrors(AliITSclusterV2* cluster,
-                            Int_t module,Double_t& sigmax, Double_t& sigmay,
-                            Double_t& sigmaz);
-  void     ResetForFinding();
-  void     SetFlagLoadedClusters(Int_t d) {fFlagLoad=d;}
-
-  void     UpdatePoints();
+  void ResetForFinding();
+  void UpdatePoints();
 
   static Double_t Curvature(Double_t x1,Double_t y1,Double_t x2,Double_t y2,
-                    Double_t x3,Double_t y3);
+                     Double_t x3,Double_t y3);
 
   Double_t ChoosePoint(Double_t p1, Double_t p2, Double_t pp); 
 
@@ -77,13 +77,19 @@ class AliITStrackerSA : public AliITStrackerV2 {
               Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
               Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16, 
               Int_t gl17, Int_t gl18, Int_t numberofpoints=6);
-
-  Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow, 
-                      AliITStrackSA* trs,Double_t zvertex,Double_t yvertex,
-                      Double_t xvertex, Int_t flagp, AliITSclusterTable* table); 
  
+  Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow, 
+                       AliITStrackSA* trs,Double_t zvertex,Int_t flagp); 
+
+  void GetCoorAngles(AliITSRecPoint* cl,Double_t &phi,Double_t &lambda,Float_t &x,Float_t &y,Float_t &z,Double_t* vertex);
+  void GetCoorErrors(AliITSRecPoint* cl,Float_t &sx,Float_t &sy, Float_t &sz);
+
+  AliITSclusterTable* GetClusterCoord(Int_t layer,Int_t n) const {return (AliITSclusterTable*)fCluCoord[layer]->UncheckedAt(n);}
+  void RemoveClusterCoord(Int_t layer, Int_t n) {fCluCoord[layer]->RemoveAt(n);fCluCoord[layer]->Compress();}
+
 
   Double_t fPhiEstimate; //Estimation of phi angle on next layer
+  Bool_t fITSStandAlone; //Tracking is performed in the ITS alone if kTRUE
   Float_t fPoint1[2];   //! coord. of 1-st point to evaluate the curvature
   Float_t fPoint2[2];   //! coord. of 2-nd point to evaluate the curvature
   Float_t fPoint3[2];   //! coord. of 3-rd point to evaluate the curvature
@@ -96,15 +102,18 @@ class AliITStrackerSA : public AliITStrackerV2 {
   Int_t fNloop;         //  Number of iterqations on phi and lambda windows
   Double_t *fPhiWin;    // phi window sizes
   Double_t *fLambdaWin; // lambda window sizes
-  AliITSVertex *fVert;        //! primary vertex
+  AliESDVertex *fVert;        //! primary vertex
   AliITSVertexer *fVertexer;  //! vertexer 
-  AliITSgeom *fGeom;          //! ITS geometry
-  Int_t fFlagLoad;            //  flag for loaded clusters (1==already loaded)
-  AliITSclusterTable* fTable; //  table with clusters
-  ClassDef(AliITStrackerSA,1)
-};
+  TObjArray *fListOfTracks;   //! container for found tracks 
+  TTree *fITSclusters;        //! pointer to ITS tree of clusters
+  Bool_t fSixPoints;          // If true 6/6 points are required (default). 5/6 otherwise
+  Int_t fOuterStartLayer;     // Search for tracks with <6 points: outer layer to start from
 
-#endif
+  TClonesArray** fCluLayer; //! array with clusters 
+  TClonesArray** fCluCoord; //! array with cluster info
 
+  ClassDef(AliITStrackerSA,6)
+};
 
+#endif