Bug corrected.
[u/mrichter/AliRoot.git] / STEER / AliTrackPointArray.h
CommitLineData
98937d93 1#ifndef ALITRACKPOINTARRAY_H
2#define ALITRACKPOINTARRAY_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6//////////////////////////////////////////////////////////////////////////////
7// Class AliTrackPoint //
8// This class represent a single track space-point. //
9// It is used to access the points array defined in AliTrackPointArray. //
10// Note that the space point coordinates are given in the global frame. //
11// //
12// cvetan.cheshkov@cern.ch 3/11/2005 //
13//////////////////////////////////////////////////////////////////////////////
14
15#include <TObject.h>
ec9e17f9 16#include <TMatrixDSym.h>
98937d93 17
082050e1 18class TGeoRotation;
19
98937d93 20class AliTrackPoint : public TObject {
21
22 public:
23 AliTrackPoint();
e3901fd8 24 AliTrackPoint(Float_t x, Float_t y, Float_t z, const Float_t *cov, UShort_t volid, Float_t charge = 0, Float_t drifttime = 0,Float_t chargeratio = 0, Int_t clutype = 0);
25 AliTrackPoint(const Float_t *xyz, const Float_t *cov, UShort_t volid, Float_t charge = 0, Float_t drifttime = 0,Float_t chargeratio = 0, Int_t clutype=0);
98937d93 26 AliTrackPoint(const AliTrackPoint &p);
27 AliTrackPoint& operator= (const AliTrackPoint& p);
28 virtual ~AliTrackPoint() {}
29
30 // Multiplication with TGeoMatrix and distance between points (chi2) to be implemented
31
32 void SetXYZ(Float_t x, Float_t y, Float_t z, const Float_t *cov = 0);
33 void SetXYZ(const Float_t *xyz, const Float_t *cov = 0);
90a845c7 34 void SetCov(const Float_t *cov);
98937d93 35 void SetVolumeID(UShort_t volid) { fVolumeID = volid; }
6d016ab4 36 void SetCharge(Float_t charge) { fCharge = charge; }
cb0800cc 37 void SetDriftTime(Float_t time) { fDriftTime = time; }
b7bcc8ed 38 void SetChargeRatio(Float_t ratio) { fChargeRatio= ratio; }
e3901fd8 39 void SetClusterType(Int_t clutype) { fClusterType= clutype; }
eb2d90b9 40 void SetExtra(Bool_t flag=kTRUE) { fIsExtra = flag; }
98937d93 41
42 Float_t GetX() const { return fX; }
43 Float_t GetY() const { return fY; }
44 Float_t GetZ() const { return fZ; }
45 void GetXYZ(Float_t *xyz, Float_t *cov = 0) const;
46 const Float_t *GetCov() const { return &fCov[0]; }
47 UShort_t GetVolumeID() const { return fVolumeID; }
6d016ab4 48 Float_t GetCharge() const { return fCharge; }
cb0800cc 49 Float_t GetDriftTime() const { return fDriftTime;}
b7bcc8ed 50 Float_t GetChargeRatio() const { return fChargeRatio;}
e3901fd8 51 Int_t GetClusterType() const { return fClusterType;}
eb2d90b9 52 Bool_t IsExtra() const { return fIsExtra;}
98937d93 53
46ae650f 54 Float_t GetResidual(const AliTrackPoint &p, Bool_t weighted = kFALSE) const;
4dcdc747 55 Bool_t GetPCA(const AliTrackPoint &p, AliTrackPoint &out) const;
46ae650f 56
57 Float_t GetAngle() const;
082050e1 58 Bool_t GetRotMatrix(TGeoRotation& rot) const;
ec9e17f9 59 void SetAlignCovMatrix(const TMatrixDSym alignparmtrx);
60
46ae650f 61 AliTrackPoint& Rotate(Float_t alpha) const;
62 AliTrackPoint& MasterToLocal() const;
63
64 void Print(Option_t *) const;
65
98937d93 66 private:
67
68 Float_t fX; // X coordinate
69 Float_t fY; // Y coordinate
70 Float_t fZ; // Z coordinate
6d016ab4 71 Float_t fCharge; // Cluster charge in arbitrary units
cb0800cc 72 Float_t fDriftTime;// Drift time in SDD (in ns)
b7bcc8ed 73 Float_t fChargeRatio; // Charge ratio in SSD
e3901fd8 74 Int_t fClusterType; // Cluster Type (encoded info on size and shape)
98937d93 75 Float_t fCov[6]; // Cov matrix
eb2d90b9 76 Bool_t fIsExtra; // attached by tracker but not used in fit
98937d93 77 UShort_t fVolumeID; // Volume ID
78
e3901fd8 79 ClassDef(AliTrackPoint,7)
98937d93 80};
81
82//////////////////////////////////////////////////////////////////////////////
83// Class AliTrackPointArray //
84// This class contains the ESD track space-points which are used during //
85// the alignment procedures. Each space-point consist of 3 coordinates //
86// (and their errors) and the index of the sub-detector which contains //
87// the space-point. //
88// cvetan.cheshkov@cern.ch 3/11/2005 //
89//////////////////////////////////////////////////////////////////////////////
90
91class AliTrackPointArray : public TObject {
92
93 public:
94
95 AliTrackPointArray();
96 AliTrackPointArray(Int_t npoints);
97 AliTrackPointArray(const AliTrackPointArray &array);
98 AliTrackPointArray& operator= (const AliTrackPointArray& array);
99 virtual ~AliTrackPointArray();
100
101 // Bool_t AddPoint(Int_t i, AliCluster *cl, UShort_t volid);
102 Bool_t AddPoint(Int_t i, const AliTrackPoint *p);
103
104 Int_t GetNPoints() const { return fNPoints; }
105 Int_t GetCovSize() const { return fSize; }
106 Bool_t GetPoint(AliTrackPoint &p, Int_t i) const;
107 // Getters for fast access to the coordinate arrays
108 const Float_t* GetX() const { return &fX[0]; }
109 const Float_t* GetY() const { return &fY[0]; }
110 const Float_t* GetZ() const { return &fZ[0]; }
6d016ab4 111 const Float_t* GetCharge() const { return &fCharge[0]; }
cb0800cc 112 const Float_t* GetDriftTime() const { return &fDriftTime[0]; }
b7bcc8ed 113 const Float_t* GetChargeRatio() const { return &fChargeRatio[0]; }
e3901fd8 114 const Int_t* GetClusterType() const { return &fClusterType[0]; }
eb2d90b9 115 const Bool_t* GetExtra() const { return &fIsExtra[0]; }
98937d93 116 const Float_t* GetCov() const { return &fCov[0]; }
117 const UShort_t* GetVolumeID() const { return &fVolumeID[0]; }
118
119 Bool_t HasVolumeID(UShort_t volid) const;
46ae650f 120
bead9796 121 void Sort(Bool_t down=kTRUE);
122
98937d93 123 private:
bead9796 124 Bool_t fSorted; // Sorted flag
98937d93 125
126 Int_t fNPoints; // Number of space points
127 Float_t *fX; //[fNPoints] Array with space point X coordinates
128 Float_t *fY; //[fNPoints] Array with space point Y coordinates
129 Float_t *fZ; //[fNPoints] Array with space point Z coordinates
6d016ab4 130 Float_t *fCharge; //[fNPoints] Array with clusters charge
cb0800cc 131 Float_t *fDriftTime; //[fNPoints] Array with drift times
b7bcc8ed 132 Float_t *fChargeRatio; //[fNPoints] Array with charge ratio
e3901fd8 133 Int_t *fClusterType; //[fNPoints] Array with cluster type
eb2d90b9 134 Bool_t *fIsExtra; //[fNPoints] Array with extra flags
98937d93 135 Int_t fSize; // Size of array with cov matrices = 6*N of points
136 Float_t *fCov; //[fSize] Array with space point coordinates cov matrix
137 UShort_t *fVolumeID; //[fNPoints] Array of space point volume IDs
138
e3901fd8 139 ClassDef(AliTrackPointArray,7)
98937d93 140};
141
142#endif
eb2d90b9 143