16d18e73b5c515d1adbe2c3f2914a17888832cb4
[u/mrichter/AliRoot.git] / RALICE / AliTrack.h
1 #ifndef ALITRACK_H
2 #define ALITRACK_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 // $Id$
8
9
10 #include "TObject.h"
11 #include "TObjArray.h"
12 #include "TArrayD.h"
13  
14 #include "AliSignal.h"
15 #include "AliBoost.h"
16 #include "AliPosition.h"
17  
18 class AliTrack : public TObject,public Ali4Vector
19 {
20  public:
21   AliTrack();                       // Default constructor
22   ~AliTrack();                      // Destructor
23   AliTrack(AliTrack& t);            // Copy constructor
24   void Reset();                     // Reset all values to 0
25   void Set4Momentum(Ali4Vector& p); // Set track 4-momentum
26   void Set3Momentum(Ali3Vector& p); // Set track 3-momentum
27   void SetMass(Double_t m,Double_t dm=0); // Set particle mass and error
28   void SetMass();                   // Set mass and error to mass hypothesis with highest prob.
29   void SetCharge(Float_t q);        // Set particle charge
30   void Info(TString f="car");       // Print track information for coord. frame f
31   void List(TString f="car");       // Print track and decay level 1 information for coord. frame f
32   void ListAll(TString f="car");    // Print track and all decay level information for coord. frame f
33   Ali3Vector Get3Momentum();        // Provide track 3-momentum
34   Double_t GetMomentum();           // Provide value of track 3-momentum
35   Double_t GetMass();               // Provide particle mass
36   Float_t GetCharge();              // Provide particle charge
37   Double_t GetEnergy();             // Provide particle total energy
38   void Decay(Double_t m1,Double_t m2,Double_t thcms,Double_t phicms); // Perform 2-body decay
39   Int_t GetNdecay();                // Provide number of decay products
40   AliTrack* GetDecayTrack(Int_t j); // Access to decay produced track number j
41   void AddSignal(AliSignal& s);     // Relate an AliSignal to this track
42   void RemoveSignal(AliSignal& s);  // Remove related AliSignal from this track
43   Int_t GetNsignals();              // Provide number of related AliSignals
44   AliSignal* GetSignal(Int_t j);    // Access to the related AliSignal number j
45   void SetBeginPoint(AliPosition p);// Set the track begin-point
46   AliPosition GetBeginPoint();      // Provide the track begin-point
47   void SetEndPoint(AliPosition p);  // Set the track end-point
48   AliPosition GetEndPoint();        // Provide the track end-point
49   void AddMassHypothesis(Double_t prob,Double_t m,Double_t dm=0); // Add mass hypothesis data
50   Int_t GetNMassHypotheses();                // Provide number of mass hypotheses
51   Double_t GetMassHypothesis(Int_t j=0);     // Provide mass of jth hypothesis 
52   Double_t GetMassHypothesisProb(Int_t j=0); // Provide prob. of jth mass hypothesis 
53   void RemoveMassHypothesis(Int_t j);        // Remove the jth mass hypothesis 
54   Double_t GetPt();                 // Provide trans. momentum w.r.t. z-axis
55   Double_t GetPl();                 // Provide long. momentum w.r.t. z-axis
56   Double_t GetEt();                 // Provide trans. energy w.r.t. z-axis
57   Double_t GetEl();                 // Provide long. energy w.r.t. z-axis
58   Double_t GetMt();                 // Provide trans. mass w.r.t. z-axis
59   Double_t GetMt(Int_t j);          // Provide trans. mass w.r.t. z-axis and jth mass hypothesis
60   Double_t GetRapidity();           // Provide rapidity value w.r.t. z-axis
61   void SetImpactPoint(AliPosition p,TString q); // Set the impact-point in plane "q=0"
62   AliPosition GetImpactPoint(TString q);        // Provide the impact-point in plane "q=0"
63   void SetId(Int_t id);             // Set the user defined unique track identifier
64   Int_t GetId();                    // Provide the user defined unique track identifier
65   void SetClosestPoint(AliPosition p); // Set position p as point of closest approach w.r.t. some reference
66   AliPosition GetClosestPoint();       // Provide point of closest approach w.r.t. some reference
67   void SetChi2(Float_t chi2);       // Set the chi-squared value of the track fit
68   void SetNdf(Int_t ndf);           // Set the number of degrees of freedom for the track fit
69   Float_t GetChi2();                // Provide the chi-squared value of the track fit
70   Int_t GetNdf();                   // Provide the number of degrees of freedom for the track fit
71   void SetParticleCode(Int_t code); // Set the user defined particle id code (e.g. the PDF convention)
72   Int_t GetParticleCode();          // Provide the user defined particle id code
73
74  
75  protected:
76   void Init();           // Initialisation of pointers etc...
77   Float_t fQ;            // The charge of the particle
78   Int_t fNdec;           // The number of decay products
79   TObjArray* fDecays;    // The array of decay produced tracks
80   Int_t fNsig;           // The number of related AliSignals
81   TObjArray* fSignals;   // The array of related AliSignals
82   AliPosition fBegin;    // The begin-point of the track 
83   AliPosition fEnd;      // The end-point of the track 
84   Int_t fNmasses;        // The number of mass hypotheses
85   TArrayD* fMasses;      // The various mass hypotheses
86   TArrayD* fDmasses;     // The errors on the various masses
87   TArrayD* fPmasses;     // The probabilities of the various mass hypotheses
88   AliPosition fImpactXY; // The (extrapolated) impact-point in the plane z=0
89   AliPosition fImpactXZ; // The (extrapolated) impact-point in the plane y=0
90   AliPosition fImpactYZ; // The (extrapolated) impact-point in the plane x=0
91   Int_t fUserId;         // The user defined identifier
92   AliPosition fClosest;  // The (extrapolated) point of closest approach w.r.t some reference
93   Float_t fChi2;         // The Chi-squared of the track fit
94   Int_t fNdf;            // The number of degrees of freedom of the track fit
95   Int_t fCode;           // The user defined particle id code
96
97  private:
98   void Dump(AliTrack* t,Int_t n,TString f); // Recursively print all decay levels
99  
100  ClassDef(AliTrack,2) // Handling of the attributes of a reconstructed particle track.
101 };
102 #endif