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

#ifndef ALIITSTRACKERSA_H
#define ALIITSTRACKERSA_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                               */

/* $Id$ */


////////////////////////////////////////////////////
//  Stand alone tracker class                     //
//  Origin:  Elisabetta Crescio                   //
//  e-mail:  crescio@to.infn.it                   //
//  adapted for cosmics by A.Dainese              //
////////////////////////////////////////////////////

class AliITSclusterTable;
class AliITStrackSA;
class AliESDVertex;
class AliESDEvent;
class AliITSVertexer;
class TTree;
class TArrayD;

class AliITStrackerSA : public AliITStrackerMI {


 public:

  AliITStrackerSA();
  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();  
  virtual Int_t Clusters2Tracks(AliESDEvent *event);
  Int_t FindTracks(AliESDEvent* event);

  AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,Bool_t onePoint=kFALSE);
  void StoreTrack(AliITStrackV2 *t,AliESDEvent *event) const; 
  Int_t FindTrackLowChiSquare() const;
  Int_t LoadClusters(TTree *cf) {Int_t rc=AliITStrackerMI::LoadClusters(cf); SetClusterTree(cf); return rc;}
  void SetVertex(AliESDVertex *vtx){fVert = vtx;}
  void SetClusterTree(TTree * itscl){fITSclusters = itscl;}

  void SetOutwardFinding() {fInwardFlag=kFALSE;}
  void SetInwardFinding() {fInwardFlag=kTRUE;}
  void SetOuterStartLayer(Int_t osl = 0) {
    if(osl>(AliITSgeomTGeo::GetNLayers()-2)) AliWarning("Minimum Number of layers is 2. OuterStartLayer set to Nlayers-2");
    fOuterStartLayer = TMath::Min(AliITSgeomTGeo::GetNLayers()-2,osl);
  }
  Int_t GetOuterStartLayer() const {return fOuterStartLayer;}
  void SetInnerStartLayer(Int_t isl = 5) {
    if(isl<1) AliWarning("Minimum Number of layers is 2. InnerStartLayer set to 1");
    fInnerStartLayer = TMath::Max(1,isl);
  }
  Int_t GetInnerStartLayer() const {return fInnerStartLayer;}

  void SetSAFlag(Bool_t fl){fITSStandAlone=fl;}  // StandAlone flag setter
  Bool_t GetSAFlag() const {return fITSStandAlone;} // StandAlone flag getter
  void SetFixedWindowSizes(Int_t n=46, Double_t *phi=0, Double_t *lam=0);
  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);

  void SetMinNPoints(Int_t np){fMinNPoints=np;}
  Int_t GetMinNPoints() const {return fMinNPoints;}
  void SetMinimumChargeSDDSSD(Float_t minq=0.){fMinQ=minq;}
  enum {kSAflag=0x8000}; //flag to mark clusters used in the SA tracker
 protected:

  //Initialization
  void Init();
  void ResetForFinding();
  void UpdatePoints();
  Bool_t SetFirstPoint(Int_t lay, Int_t clu, Double_t* primaryVertex);
  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); 

  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
  Float_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 
  TObjArray *fListOfTracks;   //! container for found tracks 
  TObjArray *fListOfSATracks; //! container for found SA tracks 
  TTree *fITSclusters;        //! pointer to ITS tree of clusters
  Bool_t fInwardFlag;       // set to kTRUE for inward track finding
  Int_t fOuterStartLayer;     // Outward search for tracks with <6 points: outer layer to start from
  Int_t fInnerStartLayer;     // Inward search for tracks with <6 points: inner layer to start from
  Int_t fMinNPoints;        // minimum number of clusters for a track
  Float_t fMinQ;              // lower cut on cluster charge (SDD and SSD)

  TClonesArray** fCluLayer; //! array with clusters 
  TClonesArray** fCluCoord; //! array with cluster info

  ClassDef(AliITStrackerSA,9)
};

#endif
Commit	Line	Data
13918578	1	#ifndef ALIITSTRACKERSA_H
	2	#define ALIITSTRACKERSA_H
	3
	4
	5
36e140f1	6	#include "AliITSgeomTGeo.h"
e43c066c	7	#include "AliITStrackerMI.h"
13918578	8
	9	/* Copyright(c) 1998-2003, ALICE Experiment at CERN, All rights reserved. *
	10	* See cxx source for full Copyright notice */
	11
deae0246	12	/* $Id$ */
	13
	14
13918578	15	////////////////////////////////////////////////////
	16	// Stand alone tracker class //
	17	// Origin: Elisabetta Crescio //
	18	// e-mail: crescio@to.infn.it //
36e140f1	19	// adapted for cosmics by A.Dainese //
13918578	20	////////////////////////////////////////////////////
	21
	22	class AliITSclusterTable;
	23	class AliITStrackSA;
d681bb2d	24	class AliESDVertex;
b8ed1a92	25	class AliESDEvent;
13918578	26	class AliITSVertexer;
2257f27e	27	class TTree;
7e5bf5af	28	class TArrayD;
13918578	29
e43c066c	30	class AliITStrackerSA : public AliITStrackerMI {
13918578	31
	32
	33	public:
	34
	35	AliITStrackerSA();
cc088660	36	AliITStrackerSA(const Char_t *geom);
	37	AliITStrackerSA(const Char_t geom,AliESDVertex vert);
	38	AliITStrackerSA(const Char_t geom,AliITSVertexer vertexer);
7d62fb64	39	AliITStrackerSA(const AliITStrackerSA& tracker);
e399f961	40	AliITStrackerSA& operator=(const AliITStrackerSA& source);
13918578	41	virtual ~AliITStrackerSA();
af885e0f	42	virtual Int_t Clusters2Tracks(AliESDEvent *event);
af885e0f	43	Int_t FindTracks(AliESDEvent* event);
13918578	44
5a03f353	45	AliITStrackV2* FitTrack(AliITStrackSA* tr,Double_t* primaryVertex,Bool_t onePoint=kFALSE);
c421499f	46	void StoreTrack(AliITStrackV2 t,AliESDEvent event) const;
deae0246	47	Int_t FindTrackLowChiSquare() const;
12b1afb7	48	Int_t LoadClusters(TTree *cf) {Int_t rc=AliITStrackerMI::LoadClusters(cf); SetClusterTree(cf); return rc;}
d681bb2d	49	void SetVertex(AliESDVertex *vtx){fVert = vtx;}
2257f27e	50	void SetClusterTree(TTree * itscl){fITSclusters = itscl;}
12b1afb7	51
	52	void SetOutwardFinding() {fInwardFlag=kFALSE;}
	53	void SetInwardFinding() {fInwardFlag=kTRUE;}
	54	void SetOuterStartLayer(Int_t osl = 0) {
	55	if(osl>(AliITSgeomTGeo::GetNLayers()-2)) AliWarning("Minimum Number of layers is 2. OuterStartLayer set to Nlayers-2");
	56	fOuterStartLayer = TMath::Min(AliITSgeomTGeo::GetNLayers()-2,osl);
	57	}
b8ed1a92	58	Int_t GetOuterStartLayer() const {return fOuterStartLayer;}
12b1afb7	59	void SetInnerStartLayer(Int_t isl = 5) {
	60	if(isl<1) AliWarning("Minimum Number of layers is 2. InnerStartLayer set to 1");
	61	fInnerStartLayer = TMath::Max(1,isl);
	62	}
	63	Int_t GetInnerStartLayer() const {return fInnerStartLayer;}
	64
1966f03c	65	void SetSAFlag(Bool_t fl){fITSStandAlone=fl;} // StandAlone flag setter
1966f03c	66	Bool_t GetSAFlag() const {return fITSStandAlone;} // StandAlone flag getter
c7d6d7b7	67	void SetFixedWindowSizes(Int_t n=46, Double_t phi=0, Double_t lam=0);
c7d6d7b7	68	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);
13918578	69
12b1afb7	70	void SetMinNPoints(Int_t np){fMinNPoints=np;}
12b1afb7	71	Int_t GetMinNPoints() const {return fMinNPoints;}
3733ccd2	72	void SetMinimumChargeSDDSSD(Float_t minq=0.){fMinQ=minq;}
627b6db6	73	enum {kSAflag=0x8000}; //flag to mark clusters used in the SA tracker
13918578	74	protected:
13918578	75
13918578	76	//Initialization
13918578	77	void Init();
4e05ab9a	78	void ResetForFinding();
4e05ab9a	79	void UpdatePoints();
12b1afb7	80	Bool_t SetFirstPoint(Int_t lay, Int_t clu, Double_t* primaryVertex);
13918578	81	static Double_t Curvature(Double_t x1,Double_t y1,Double_t x2,Double_t y2,
bef31448	82	Double_t x3,Double_t y3);
13918578	83
	84	Double_t ChoosePoint(Double_t p1, Double_t p2, Double_t pp);
	85
	86	static Int_t FindIntersection(Float_t a1, Float_t b1, Float_t c1, Float_t c2,
	87	Float_t& x1,Float_t& y1, Float_t& x2, Float_t& y2);
	88	static Int_t FindEquation(Float_t x1, Float_t y1, Float_t x2, Float_t y2,
	89	Float_t x3, Float_t y3,Float_t& a, Float_t& b,
	90	Float_t& c);
	91
	92	static Int_t FindLabel(Int_t l1, Int_t l2, Int_t l3, Int_t l4, Int_t l5, Int_t l6);
	93	static Int_t Label(Int_t gl1, Int_t gl2, Int_t gl3, Int_t gl4, Int_t gl5,
	94	Int_t gl6,Int_t gl7, Int_t gl8, Int_t gl9, Int_t gl10,Int_t gl11,
	95	Int_t gl12, Int_t gl13, Int_t gl14,Int_t gl15, Int_t gl16,
	96	Int_t gl17, Int_t gl18, Int_t numberofpoints=6);
bef31448	97
13918578	98	Int_t SearchClusters(Int_t layer,Double_t phiwindow,Double_t lambdawindow,
cfe729e0	99	AliITStrackSA* trs,Double_t zvertex,Int_t flagp);
7e5bf5af	100
cc088660	101	void GetCoorAngles(AliITSRecPoint* cl,Double_t &phi,Double_t &lambda,Float_t &x,Float_t &y,Float_t &z,Double_t* vertex);
cc088660	102	void GetCoorErrors(AliITSRecPoint* cl,Float_t &sx,Float_t &sy, Float_t &sz);
7e5bf5af	103
	104	AliITSclusterTable* GetClusterCoord(Int_t layer,Int_t n) const {return (AliITSclusterTable*)fCluCoord[layer]->UncheckedAt(n);}
	105	void RemoveClusterCoord(Int_t layer, Int_t n) {fCluCoord[layer]->RemoveAt(n);fCluCoord[layer]->Compress();}
	106
	107
13918578	108	Double_t fPhiEstimate; //Estimation of phi angle on next layer
1966f03c	109	Bool_t fITSStandAlone; //Tracking is performed in the ITS alone if kTRUE
4e05ab9a	110	Float_t fPoint1[2]; //! coord. of 1-st point to evaluate the curvature
	111	Float_t fPoint2[2]; //! coord. of 2-nd point to evaluate the curvature
	112	Float_t fPoint3[2]; //! coord. of 3-rd point to evaluate the curvature
	113	Float_t fPointc[2]; //! current point coord (for curvature eval.)
13918578	114	Double_t fLambdac; //! current value of the Lambda angle in the window
	115	Double_t fPhic; //! current value of the Phi angle in the window
	116	Float_t fCoef1; //! param. of the equation of the circ. approx a layer
	117	Float_t fCoef2; //! param. of the equation of the circ. approx a layer
	118	Float_t fCoef3; //! param. of the equation of the circ. approx a layer
	119	Int_t fNloop; // Number of iterqations on phi and lambda windows
	120	Double_t *fPhiWin; // phi window sizes
	121	Double_t *fLambdaWin; // lambda window sizes
d681bb2d	122	AliESDVertex *fVert; //! primary vertex
13918578	123	AliITSVertexer *fVertexer; //! vertexer
4e05ab9a	124	TObjArray *fListOfTracks; //! container for found tracks
deae0246	125	TObjArray *fListOfSATracks; //! container for found SA tracks
2257f27e	126	TTree *fITSclusters; //! pointer to ITS tree of clusters
12b1afb7	127	Bool_t fInwardFlag; // set to kTRUE for inward track finding
	128	Int_t fOuterStartLayer; // Outward search for tracks with <6 points: outer layer to start from
	129	Int_t fInnerStartLayer; // Inward search for tracks with <6 points: inner layer to start from
	130	Int_t fMinNPoints; // minimum number of clusters for a track
3733ccd2	131	Float_t fMinQ; // lower cut on cluster charge (SDD and SSD)
7e5bf5af	132
	133	TClonesArray** fCluLayer; //! array with clusters
	134	TClonesArray** fCluCoord; //! array with cluster info
	135
12b1afb7	136	ClassDef(AliITStrackerSA,9)
13918578	137	};
	138
	139	#endif