[u/mrichter/AliRoot.git] / TPC / AliTPCtracker.h

#ifndef ALITPCTRACKER_H
#define ALITPCTRACKER_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//-------------------------------------------------------
//                       TPC tracker
//
//   Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
//-------------------------------------------------------
#include "AliTracker.h"
#include "AliTPCtrack.h"
#include "AliTPCreco.h"
#include "AliTPCcluster.h"
#include "AliBarrelTrack.h"

class TFile;
class TTree;
class TBranch;
class AliTPCParam;
class TObjArray;
class TClonesArray;
class AliESD;

class AliTPCtracker : public AliTracker {
public:
   AliTPCtracker():AliTracker(),fkNIS(0),fkNOS(0) {
      fInnerSec=fOuterSec=0; fSeeds=0; 
      fParam = 0;
   }
   AliTPCtracker(const AliTPCParam *par);
  ~AliTPCtracker();

   Int_t ReadSeeds(const TFile *in);

   Int_t LoadClusters(TTree *c);
   void UnloadClusters();

   AliCluster *GetCluster(Int_t index) const;
   Int_t Clusters2Tracks(TTree *in, TTree *out);
   Int_t Clusters2Tracks(AliESD *event);
   Int_t PropagateBack(TTree *in, TTree *out);
   Int_t PropagateBack(AliESD *event);
   Int_t RefitInward(TTree *inTracks, TTree *outTracks);
   Int_t RefitInward(AliESD *event);

   virtual void  CookLabel(AliKalmanTrack *t,Float_t wrong) const; 

public:
//**************** Internal tracker class ********************** 
   class AliTPCRow {
   public:
     AliTPCRow() {
       fN=0; 
       fSize=kMaxClusterPerRow/8;
       fClusterArray=new AliTPCcluster[fSize];
     }
     ~AliTPCRow() {delete[] fClusterArray;}
     void InsertCluster(const AliTPCcluster *c, Int_t sec, Int_t row);
     void ResetClusters() {fN=0; delete[] fClusterArray; fClusterArray=0;}
     operator int() const {return fN;}
     const AliTPCcluster *operator[](Int_t i) const {return fClusters[i];}
     const AliTPCcluster *GetUnsortedCluster(Int_t i) const {
       if ((i < 0) || (i >= fN)) return NULL;
       return fClusterArray+i;
     }
     UInt_t GetIndex(Int_t i) const {return fIndex[i];}
     Int_t Find(Double_t y) const; 
     void SetX(Double_t x) {fX=x;}
     Double_t GetX() const {return fX;}

   private:
     Int_t fN;                                          //number of clusters 
     const AliTPCcluster *fClusters[kMaxClusterPerRow]; //pointers to clusters
     Int_t fSize;                                 //size of array of clusters
     AliTPCcluster *fClusterArray;                      //array of clusters
     UInt_t fIndex[kMaxClusterPerRow];                  //indeces of clusters
     Double_t fX;                                 //X-coordinate of this row

   private:
     AliTPCRow(const AliTPCRow& r);            //dummy copy constructor
     AliTPCRow &operator=(const AliTPCRow& r); //dummy assignment operator
   };

//**************** Internal tracker class ********************** 
   class AliTPCSector {
   public:
     AliTPCSector() { fN=0; fRow = 0; }
    ~AliTPCSector() { delete[] fRow; }
     AliTPCRow& operator[](Int_t i) const { return *(fRow+i); }
     Int_t GetNRows() const { return fN; }
     void Setup(const AliTPCParam *par, Int_t flag);
     Double_t GetX(Int_t l) const {return fRow[l].GetX();}
     Double_t GetMaxY(Int_t l) const {
         return GetX(l)*TMath::Tan(0.5*GetAlpha());
     } 
     Double_t GetAlpha() const {return fAlpha;}
     Double_t GetAlphaShift() const {return fAlphaShift;}
     Int_t GetRowNumber(Double_t x) const {
        //return pad row number for this x
       Double_t r;
       if (fN < 64){
	 r=fRow[fN-1].GetX();
	 if (x > r) return fN;
	 r=fRow[0].GetX();
	 if (x < r) return -1;
	 return Int_t((x-r)/f1PadPitchLength + 0.5);}
       else{	
	   r=fRow[fN-1].GetX();
	   if (x > r) return fN;
	   r=fRow[0].GetX();
	   if (x < r) return -1;
	  Double_t r1=fRow[64].GetX();
	  if(x<r1){	  
	    return Int_t((x-r)/f1PadPitchLength + 0.5);}
	  else{
	    return (Int_t((x-r1)/f2PadPitchLength + 0.5)+64);} 
       }
     }
     Double_t GetPadPitchWidth()  const {return fPadPitchWidth;}

   private:
     Int_t fN;                        //number of pad rows 
     AliTPCRow *fRow;                    //array of pad rows
     Double_t fAlpha;                    //opening angle
     Double_t fAlphaShift;               //shift angle;
     Double_t fPadPitchWidth;            //pad pitch width
     Double_t f1PadPitchLength;          //pad pitch length
     Double_t f2PadPitchLength;          //pad pitch length
   private:
     AliTPCSector(const AliTPCSector &s);           //dummy copy contructor
     AliTPCSector& operator=(const AliTPCSector &s);//dummy assignment operator
   };

//**************** Internal tracker class ********************** 
   class AliTPCseed : public AliTPCtrack {
   public:
     AliTPCseed():AliTPCtrack(){}
     AliTPCseed(const AliTPCtrack &t):AliTPCtrack(t){}
     AliTPCseed(const AliKalmanTrack &t, Double_t a):AliTPCtrack(t,a){}
     AliTPCseed(UInt_t index, const Double_t xx[5], 
                const Double_t cc[15], Double_t xr, Double_t alpha): 
                AliTPCtrack(index, xx, cc, xr, alpha) {}
     void SetSampledEdx(Float_t q, Int_t i) {
        Double_t s=GetSnp(), t=GetTgl();
        q *= TMath::Sqrt((1-s*s)/(1+t*t));
        fdEdxSample[i]=q;
     }
     void CookdEdx(Double_t low=0.05, Double_t up=0.70);

   private:
     Float_t fdEdxSample[200]; //array of dE/dx samples 
   };

private:

   void MakeSeeds(Int_t i1, Int_t i2);
   Int_t FollowProlongation(AliTPCseed& t, Int_t rf=0);
   Int_t FollowBackProlongation(AliTPCseed &s, const AliTPCtrack &t);
   Int_t FollowRefitInward(AliTPCseed *seed, AliTPCtrack *track);

   AliTPCtracker(const AliTPCtracker& r);           //dummy copy constructor
   AliTPCtracker &operator=(const AliTPCtracker& r);//dummy assignment operator

   const Int_t fkNIS;        //number of inner sectors
   AliTPCSector *fInnerSec;  //array of inner sectors
   const Int_t fkNOS;        //number of outer sectors
   AliTPCSector *fOuterSec;  //array of outer sectors

   Int_t fN;               //number of "active" sectors
   AliTPCSector *fSectors; //pointer to "active" sectors;
   
   AliTPCParam *fParam;      //! TPC parameters for outer reference plane [SR, GSI, 18.02.2003]
   TObjArray *fSeeds;        //array of track seeds

   // [SR, 01.04.2003]
   void SetBarrelTree(const char *mode);
   void StoreBarrelTrack(AliTPCtrack *ps, Int_t refPlane, Int_t isIn);

   // [SR, 01.04.2003]
   TFile *fBarrelFile;             // file with "barrel" tracks
   TTree *fBarrelTree;             // tree with "barrel" tracks
   TBranch *fBarrelBranch;
   TClonesArray *fBarrelArray;
   AliBarrelTrack *fBarrelTrack;

  ClassDef(AliTPCtracker,1)   // Time Projection Chamber tracker
};

#endif
Commit	Line	Data
	1	#ifndef ALITPCTRACKER_H
	2	#define ALITPCTRACKER_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	/* $Id$ */
	7
	8	//-------------------------------------------------------
	9	// TPC tracker
	10	//
	11	// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
	12	//-------------------------------------------------------
	13	#include "AliTracker.h"
	14	#include "AliTPCtrack.h"
	15	#include "AliTPCreco.h"
	16	#include "AliTPCcluster.h"
	17	#include "AliBarrelTrack.h"
	18
	19	class TFile;
	20	class TTree;
	21	class TBranch;
	22	class AliTPCParam;
	23	class TObjArray;
	24	class TClonesArray;
	25	class AliESD;
	26
	27	class AliTPCtracker : public AliTracker {
	28	public:
	29	AliTPCtracker():AliTracker(),fkNIS(0),fkNOS(0) {
	30	fInnerSec=fOuterSec=0; fSeeds=0;
	31	fParam = 0;
	32	}
	33	AliTPCtracker(const AliTPCParam *par);
	34	~AliTPCtracker();
	35
	36	Int_t ReadSeeds(const TFile *in);
	37
	38	Int_t LoadClusters(TTree *c);
	39	void UnloadClusters();
	40
	41	AliCluster *GetCluster(Int_t index) const;
	42	Int_t Clusters2Tracks(TTree in, TTree out);
	43	Int_t Clusters2Tracks(AliESD *event);
	44	Int_t PropagateBack(TTree in, TTree out);
	45	Int_t PropagateBack(AliESD *event);
	46	Int_t RefitInward(TTree inTracks, TTree outTracks);
	47	Int_t RefitInward(AliESD *event);
	48
	49	virtual void CookLabel(AliKalmanTrack *t,Float_t wrong) const;
	50
	51	public:
	52	//************** Internal tracker class ********************
	53	class AliTPCRow {
	54	public:
	55	AliTPCRow() {
	56	fN=0;
	57	fSize=kMaxClusterPerRow/8;
	58	fClusterArray=new AliTPCcluster[fSize];
	59	}
	60	~AliTPCRow() {delete[] fClusterArray;}
	61	void InsertCluster(const AliTPCcluster *c, Int_t sec, Int_t row);
	62	void ResetClusters() {fN=0; delete[] fClusterArray; fClusterArray=0;}
	63	operator int() const {return fN;}
	64	const AliTPCcluster *operator[](Int_t i) const {return fClusters[i];}
	65	const AliTPCcluster *GetUnsortedCluster(Int_t i) const {
	66	if ((i < 0) \|\| (i >= fN)) return NULL;
	67	return fClusterArray+i;
	68	}
	69	UInt_t GetIndex(Int_t i) const {return fIndex[i];}
	70	Int_t Find(Double_t y) const;
	71	void SetX(Double_t x) {fX=x;}
	72	Double_t GetX() const {return fX;}
	73
	74	private:
	75	Int_t fN; //number of clusters
	76	const AliTPCcluster *fClusters[kMaxClusterPerRow]; //pointers to clusters
	77	Int_t fSize; //size of array of clusters
	78	AliTPCcluster *fClusterArray; //array of clusters
	79	UInt_t fIndex[kMaxClusterPerRow]; //indeces of clusters
	80	Double_t fX; //X-coordinate of this row
	81
	82	private:
	83	AliTPCRow(const AliTPCRow& r); //dummy copy constructor
	84	AliTPCRow &operator=(const AliTPCRow& r); //dummy assignment operator
	85	};
	86
	87	//************** Internal tracker class ********************
	88	class AliTPCSector {
	89	public:
	90	AliTPCSector() { fN=0; fRow = 0; }
	91	~AliTPCSector() { delete[] fRow; }
	92	AliTPCRow& operator[](Int_t i) const { return *(fRow+i); }
	93	Int_t GetNRows() const { return fN; }
	94	void Setup(const AliTPCParam *par, Int_t flag);
	95	Double_t GetX(Int_t l) const {return fRow[l].GetX();}
	96	Double_t GetMaxY(Int_t l) const {
	97	return GetX(l)TMath::Tan(0.5GetAlpha());
	98	}
	99	Double_t GetAlpha() const {return fAlpha;}
	100	Double_t GetAlphaShift() const {return fAlphaShift;}
	101	Int_t GetRowNumber(Double_t x) const {
	102	//return pad row number for this x
	103	Double_t r;
	104	if (fN < 64){
	105	r=fRow[fN-1].GetX();
	106	if (x > r) return fN;
	107	r=fRow[0].GetX();
	108	if (x < r) return -1;
	109	return Int_t((x-r)/f1PadPitchLength + 0.5);}
	110	else{
	111	r=fRow[fN-1].GetX();
	112	if (x > r) return fN;
	113	r=fRow[0].GetX();
	114	if (x < r) return -1;
	115	Double_t r1=fRow[64].GetX();
	116	if(x<r1){
	117	return Int_t((x-r)/f1PadPitchLength + 0.5);}
	118	else{
	119	return (Int_t((x-r1)/f2PadPitchLength + 0.5)+64);}
	120	}
	121	}
	122	Double_t GetPadPitchWidth() const {return fPadPitchWidth;}
	123
	124	private:
	125	Int_t fN; //number of pad rows
	126	AliTPCRow *fRow; //array of pad rows
	127	Double_t fAlpha; //opening angle
	128	Double_t fAlphaShift; //shift angle;
	129	Double_t fPadPitchWidth; //pad pitch width
	130	Double_t f1PadPitchLength; //pad pitch length
	131	Double_t f2PadPitchLength; //pad pitch length
	132	private:
	133	AliTPCSector(const AliTPCSector &s); //dummy copy contructor
	134	AliTPCSector& operator=(const AliTPCSector &s);//dummy assignment operator
	135	};
	136
	137	//************** Internal tracker class ********************
	138	class AliTPCseed : public AliTPCtrack {
	139	public:
	140	AliTPCseed():AliTPCtrack(){}
	141	AliTPCseed(const AliTPCtrack &t):AliTPCtrack(t){}
	142	AliTPCseed(const AliKalmanTrack &t, Double_t a):AliTPCtrack(t,a){}
	143	AliTPCseed(UInt_t index, const Double_t xx[5],
	144	const Double_t cc[15], Double_t xr, Double_t alpha):
	145	AliTPCtrack(index, xx, cc, xr, alpha) {}
	146	void SetSampledEdx(Float_t q, Int_t i) {
	147	Double_t s=GetSnp(), t=GetTgl();
	148	q = TMath::Sqrt((1-ss)/(1+t*t));
	149	fdEdxSample[i]=q;
	150	}
	151	void CookdEdx(Double_t low=0.05, Double_t up=0.70);
	152
	153	private:
	154	Float_t fdEdxSample[200]; //array of dE/dx samples
	155	};
	156
	157	private:
	158
	159	void MakeSeeds(Int_t i1, Int_t i2);
	160	Int_t FollowProlongation(AliTPCseed& t, Int_t rf=0);
	161	Int_t FollowBackProlongation(AliTPCseed &s, const AliTPCtrack &t);
	162	Int_t FollowRefitInward(AliTPCseed seed, AliTPCtrack track);
	163
	164	AliTPCtracker(const AliTPCtracker& r); //dummy copy constructor
	165	AliTPCtracker &operator=(const AliTPCtracker& r);//dummy assignment operator
	166
	167	const Int_t fkNIS; //number of inner sectors
	168	AliTPCSector *fInnerSec; //array of inner sectors
	169	const Int_t fkNOS; //number of outer sectors
	170	AliTPCSector *fOuterSec; //array of outer sectors
	171
	172	Int_t fN; //number of "active" sectors
	173	AliTPCSector *fSectors; //pointer to "active" sectors;
	174
	175	AliTPCParam *fParam; //! TPC parameters for outer reference plane [SR, GSI, 18.02.2003]
	176	TObjArray *fSeeds; //array of track seeds
	177
	178	// [SR, 01.04.2003]
	179	void SetBarrelTree(const char *mode);
	180	void StoreBarrelTrack(AliTPCtrack *ps, Int_t refPlane, Int_t isIn);
	181
	182	// [SR, 01.04.2003]
	183	TFile *fBarrelFile; // file with "barrel" tracks
	184	TTree *fBarrelTree; // tree with "barrel" tracks
	185	TBranch *fBarrelBranch;
	186	TClonesArray *fBarrelArray;
	187	AliBarrelTrack *fBarrelTrack;
	188
	189	ClassDef(AliTPCtracker,1) // Time Projection Chamber tracker
	190	};
	191
	192	#endif
	193
	194