[u/mrichter/AliRoot.git] / ITS / AliITStrackerSA.h

#ifndef ALIITSTRACKERSA_H
#define ALIITSTRACKERSA_H 


#include "AliITStrackerV2.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                   //
////////////////////////////////////////////////////

class AliITSclusterTable;
class AliITStrackSA;
class AliESDVertex;
class AliITSVertexer;
class TTree;

class AliITStrackerSA : public AliITStrackerV2 {


 public:

  AliITStrackerSA();
  AliITStrackerSA(AliITSgeom *geom);
  AliITStrackerSA(AliITSgeom *geom,AliESDVertex *vert);
  AliITStrackerSA(AliITSgeom *geom,AliITSVertexer *vertexer);
  AliITStrackerSA(AliITStrackerSA& tracker);
  virtual ~AliITStrackerSA();  
  virtual Int_t Clusters2Tracks(AliESD *event){Int_t rc = AliITStrackerV2::Clusters2Tracks(event); if(!rc)rc=FindTracks(event); return rc;}
  Int_t FindTracks(AliESD* event);
  void  FindTracks(TTree *out,Int_t evnumber=0);
  AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,
                          Double_t *errorprimvert);

  AliITStrackV2* FindTrackLowChiSquare(TObjArray* tracklist, Int_t dim) const;
  Int_t LoadClusters(TTree *cf) {Int_t rc=AliITStrackerV2::LoadClusters(cf); SetClusterTree(cf); return rc;}
  void SetVertex(AliESDVertex *vtx){fVert = vtx;}
  void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
  void SetSixPoints(Bool_t sp = kTRUE){fSixPoints = sp;}
  Bool_t GetSixPoints() const {return fSixPoints;}
  void SetWindowSizes(Int_t n=46, Double_t *phi=0, Double_t *lam=0);
  void UseFoundTracksV2(Int_t evnum,TTree* treev2, TTree* clustertree);
  void UseFoundTracksV2(Int_t evnum,AliESD *event, TTree* clustertree);

 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     ResetForFinding();
  void     SetFlagLoadedClusters(Int_t d) {fFlagLoad=d;}

  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 ChoosePoint(Double_t p1, Double_t p2, Double_t pp); 

  static Int_t   FindIntersection(Float_t a1, Float_t b1, Float_t c1, Float_t c2, 
                           Float_t& x1,Float_t& y1, Float_t& x2, Float_t& y2);
  static Int_t   FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2, 
                       Float_t x3, Float_t y3,Float_t& a, Float_t& b, 
                       Float_t& c);
 
  static Int_t FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6);
  static Int_t Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5, 
              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,Int_t flagp, AliITSclusterTable* table); 
 

  Double_t fPhiEstimate; //Estimation of phi angle on next layer
  Double_t fPoint1[2];   //! coord. of 1-st point to evaluate the curvature
  Double_t fPoint2[2];   //! coord. of 2-nd point to evaluate the curvature
  Double_t fPoint3[2];   //! coord. of 3-rd point to evaluate the curvature
  Double_t fPointc[2];   //! current point coord (for curvature eval.)
  Double_t fLambdac;    //! current value of the Lambda angle in the window
  Double_t fPhic;       //! current value of the Phi angle in the window
  Float_t fCoef1;       //! param. of the equation of the circ. approx a layer
  Float_t fCoef2;       //! param. of the equation of the circ. approx a layer
  Float_t fCoef3;       //! param. of the equation of the circ. approx a layer
  Int_t fNloop;         //  Number of iterqations on phi and lambda windows
  Double_t *fPhiWin;    // phi window sizes
  Double_t *fLambdaWin; // lambda window sizes
  AliESDVertex *fVert;        //! primary vertex
  AliITSVertexer *fVertexer;  //! vertexer 
  AliITSgeom *fGeom;          //! ITS geometry
  TTree *fITSclusters;        //! pointer to ITS tree of clusters
  Int_t fFlagLoad;            //  flag for loaded clusters (1==already loaded)
  Bool_t fSixPoints;          // If true 6/6 points are required (default). 5/6 otherwise
  AliITSclusterTable* fTable; //  table with clusters
  ClassDef(AliITStrackerSA,1)
};

#endif
Commit	Line	Data
13918578	1	#ifndef ALIITSTRACKERSA_H
	2	#define ALIITSTRACKERSA_H
	3
	4
	5
	6	#include "AliITStrackerV2.h"
	7
	8	/* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
	9	* See cxx source for full Copyright notice */
	10
	11	////////////////////////////////////////////////////
	12	// Stand alone tracker class //
	13	// Origin: Elisabetta Crescio //
	14	// e-mail: crescio@to.infn.it //
	15	////////////////////////////////////////////////////
	16
	17	class AliITSclusterTable;
	18	class AliITStrackSA;
d681bb2d	19	class AliESDVertex;
13918578	20	class AliITSVertexer;
2257f27e	21	class TTree;
13918578	22
	23	class AliITStrackerSA : public AliITStrackerV2 {
	24
	25
	26	public:
	27
	28	AliITStrackerSA();
2257f27e	29	AliITStrackerSA(AliITSgeom *geom);
d681bb2d	30	AliITStrackerSA(AliITSgeom geom,AliESDVertex vert);
13918578	31	AliITStrackerSA(AliITSgeom geom,AliITSVertexer vertexer);
	32	AliITStrackerSA(AliITStrackerSA& tracker);
	33	virtual ~AliITStrackerSA();
2257f27e	34	virtual Int_t Clusters2Tracks(AliESD *event){Int_t rc = AliITStrackerV2::Clusters2Tracks(event); if(!rc)rc=FindTracks(event); return rc;}
	35	Int_t FindTracks(AliESD* event);
	36	void FindTracks(TTree *out,Int_t evnumber=0);
13918578	37	AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,
2257f27e	38	Double_t *errorprimvert);
13918578	39
13918578	40	AliITStrackV2* FindTrackLowChiSquare(TObjArray* tracklist, Int_t dim) const;
2257f27e	41	Int_t LoadClusters(TTree *cf) {Int_t rc=AliITStrackerV2::LoadClusters(cf); SetClusterTree(cf); return rc;}
d681bb2d	42	void SetVertex(AliESDVertex *vtx){fVert = vtx;}
2257f27e	43	void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
	44	void SetSixPoints(Bool_t sp = kTRUE){fSixPoints = sp;}
	45	Bool_t GetSixPoints() const {return fSixPoints;}
13918578	46	void SetWindowSizes(Int_t n=46, Double_t phi=0, Double_t lam=0);
13918578	47	void UseFoundTracksV2(Int_t evnum,TTree* treev2, TTree* clustertree);
bef31448	48	void UseFoundTracksV2(Int_t evnum,AliESD event, TTree clustertree);
13918578	49
	50	protected:
	51
	52	// copy constructor (NO copy allowed: the constructor is protected
	53	// to avoid misuse)
	54	AliITStrackerSA(const AliITStrackerSA& trkr);
	55	// assignment operator (NO assignment allowed)
	56	AliITStrackerSA& operator=(const AliITStrackerSA& /* trkr */);
	57	//Initialization
	58	void Init();
	59	Int_t GetFlagLoadedClusters() const {return fFlagLoad;}
bef31448	60
13918578	61	void ResetForFinding();
	62	void SetFlagLoadedClusters(Int_t d) {fFlagLoad=d;}
	63
	64	void UpdatePoints();
	65
	66	static Double_t Curvature(Double_t x1,Double_t y1,Double_t x2,Double_t y2,
bef31448	67	Double_t x3,Double_t y3);
13918578	68
	69	Double_t ChoosePoint(Double_t p1, Double_t p2, Double_t pp);
	70
	71	static Int_t FindIntersection(Float_t a1, Float_t b1, Float_t c1, Float_t c2,
	72	Float_t& x1,Float_t& y1, Float_t& x2, Float_t& y2);
	73	static Int_t FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2,
	74	Float_t x3, Float_t y3,Float_t& a, Float_t& b,
	75	Float_t& c);
	76
	77	static Int_t FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6);
	78	static Int_t Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5,
	79	Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
	80	Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16,
	81	Int_t gl17, Int_t gl18, Int_t numberofpoints=6);
bef31448	82
13918578	83	Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow,
bef31448	84	AliITStrackSA* trs,Double_t zvertex,Int_t flagp, AliITSclusterTable* table);
13918578	85
	86
	87	Double_t fPhiEstimate; //Estimation of phi angle on next layer
bef31448	88	Double_t fPoint1[2]; //! coord. of 1-st point to evaluate the curvature
	89	Double_t fPoint2[2]; //! coord. of 2-nd point to evaluate the curvature
	90	Double_t fPoint3[2]; //! coord. of 3-rd point to evaluate the curvature
	91	Double_t fPointc[2]; //! current point coord (for curvature eval.)
13918578	92	Double_t fLambdac; //! current value of the Lambda angle in the window
	93	Double_t fPhic; //! current value of the Phi angle in the window
	94	Float_t fCoef1; //! param. of the equation of the circ. approx a layer
	95	Float_t fCoef2; //! param. of the equation of the circ. approx a layer
	96	Float_t fCoef3; //! param. of the equation of the circ. approx a layer
	97	Int_t fNloop; // Number of iterqations on phi and lambda windows
	98	Double_t *fPhiWin; // phi window sizes
	99	Double_t *fLambdaWin; // lambda window sizes
d681bb2d	100	AliESDVertex *fVert; //! primary vertex
13918578	101	AliITSVertexer *fVertexer; //! vertexer
13918578	102	AliITSgeom *fGeom; //! ITS geometry
2257f27e	103	TTree *fITSclusters; //! pointer to ITS tree of clusters
13918578	104	Int_t fFlagLoad; // flag for loaded clusters (1==already loaded)
2257f27e	105	Bool_t fSixPoints; // If true 6/6 points are required (default). 5/6 otherwise
13918578	106	AliITSclusterTable* fTable; // table with clusters
	107	ClassDef(AliITStrackerSA,1)
	108	};
	109
	110	#endif
	111