f5add2e39afde07a8713cdc6e7a32be5c82ca346
[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 #include "TArrayI.h"
14  
15 #include "AliSignal.h"
16 #include "AliBoost.h"
17 #include "AliPositionObj.h"
18  
19 class AliTrack : public TNamed,public Ali4Vector
20 {
21  public:
22   AliTrack();                           // Default constructor
23   virtual ~AliTrack();                  // Destructor
24   AliTrack(AliTrack& t);                // Copy constructor
25   virtual TObject* Clone(const char* name=""); // Make a deep copy and provide its pointer
26   virtual void Reset();                 // Reset all values to 0
27   void Set4Momentum(Ali4Vector& p);     // Set track 4-momentum
28   void Set3Momentum(Ali3Vector& p);     // Set track 3-momentum
29   void SetMass(Double_t m,Double_t dm=0); // Set particle mass and error
30   void SetMass();                       // Set mass and error to the values of the hypothesis with highest prob.
31   void SetCharge(Float_t q);            // Set particle charge
32   virtual void Data(TString f="car");   // Print track information for coord. frame f
33   virtual void List(TString f="car");   // Print track and decay level 1 information for coord. frame f
34   virtual void ListAll(TString f="car");// Print track and all decay level information for coord. frame f
35   Ali3Vector Get3Momentum();            // Provide track 3-momentum
36   Double_t GetMomentum();               // Provide value of track 3-momentum
37   Double_t GetMass();                   // Provide particle mass
38   Float_t GetCharge();                  // Provide particle charge
39   Double_t GetEnergy();                 // Provide particle total energy
40   void Decay(Double_t m1,Double_t m2,Double_t thcms,Double_t phicms); // Perform 2-body decay
41   Int_t GetNdecay();                    // Provide number of decay products
42   AliTrack* GetDecayTrack(Int_t j);     // Access to decay produced track number j
43   void RemoveDecays();                  // Remove all the decay products of this track
44   void AddSignal(AliSignal& s);         // Relate an AliSignal to this track
45   void RemoveSignal(AliSignal& s);      // Remove related AliSignal from this track
46   void RemoveSignals();                 // Remove all related AliSignals from this track
47   Int_t GetNsignals();                  // Provide number of related AliSignals
48   AliSignal* GetSignal(Int_t j);        // Access to the related AliSignal number j
49   void SetBeginPoint(AliPosition& p);   // Set the track begin-point
50   AliPosition* GetBeginPoint();         // Provide the track begin-point
51   void SetEndPoint(AliPosition& p);     // Set the track end-point
52   AliPosition* GetEndPoint();           // Provide the track end-point
53   void AddTrackHypothesis(AliTrack& t); // Add track hypothesis
54   void AddTrackHypothesis(Double_t prob,Double_t m,Double_t dm=0); // Add track hypothesis with mass data
55   Int_t GetNhypotheses();               // Provide number of track hypotheses
56   AliTrack* GetTrackHypothesis(Int_t j=0);  // Provide the j-th track hypothesis 
57   void RemoveTrackHypothesis(AliTrack& t);  // Remove the specified track hypothesis 
58   void RemoveTrackHypotheses();         // Remove all track hypotheses 
59   Double_t GetPt();                     // Provide trans. momentum w.r.t. z-axis
60   Double_t GetPl();                     // Provide long. momentum w.r.t. z-axis
61   Double_t GetEt();                     // Provide trans. energy w.r.t. z-axis
62   Double_t GetEl();                     // Provide long. energy w.r.t. z-axis
63   Double_t GetMt();                     // Provide trans. mass w.r.t. z-axis
64   Double_t GetRapidity();               // Provide rapidity value w.r.t. z-axis
65   void SetImpactPoint(AliPosition& p,TString q); // Set the impact-point in plane "q=0"
66   AliPosition* GetImpactPoint(TString q);        // Provide the impact-point in plane "q=0"
67   void SetId(Int_t id);                 // Set the user defined unique track identifier
68   Int_t GetId();                        // Provide the user defined unique track identifier
69   void SetClosestPoint(AliPosition& p); // Set position p as point of closest approach w.r.t. some reference
70   AliPosition* GetClosestPoint();       // Provide point of closest approach w.r.t. some reference
71   void SetChi2(Float_t chi2);           // Set the chi-squared value of the track fit
72   void SetNdf(Int_t ndf);               // Set the number of degrees of freedom for the track fit
73   Float_t GetChi2();                    // Provide the chi-squared value of the track fit
74   Int_t GetNdf();                       // Provide the number of degrees of freedom for the track fit
75   void SetParticleCode(Int_t code);     // Set the user defined particle id code (e.g. the PDF convention)
76   Int_t GetParticleCode();              // Provide the user defined particle id code
77   void SetParentTrack(AliTrack* t);     // Set pointer to the parent track
78   AliTrack* GetParentTrack();           // Provide pointer to the parent track
79   void SetProb(Double_t prob);          // Set the hypothesis probability for this track
80   Float_t GetProb();                    // Provide the hypothesis probability for this track
81
82  
83  protected:
84   void Init();               // Initialisation of pointers etc...
85   Float_t fQ;                // The charge of the particle
86   TObjArray* fDecays;        // The array of decay produced tracks
87   TObjArray* fSignals;       // The array of related AliSignals
88   TObjArray* fHypotheses;    // The array of track hypotheses
89   AliPositionObj* fBegin;    // The begin-point of the track 
90   AliPositionObj* fEnd;      // The end-point of the track 
91   AliPositionObj* fImpactXY; // The (extrapolated) impact-point in the plane z=0
92   AliPositionObj* fImpactXZ; // The (extrapolated) impact-point in the plane y=0
93   AliPositionObj* fImpactYZ; // The (extrapolated) impact-point in the plane x=0
94   Int_t fUserId;             // The user defined identifier
95   AliPositionObj* fClosest;  // The (extrapolated) point of closest approach w.r.t some reference
96   Float_t fChi2;             // The Chi-squared of the track fit
97   Int_t fNdf;                // The number of degrees of freedom of the track fit
98   Int_t fCode;               // The user defined particle id code
99   AliTrack* fParent;         // Pointer to the parent track
100   Float_t fProb;             // Probability for this track as a hypothesis
101
102  private:
103   void Dumps(AliTrack* t,Int_t n,TString f); // Recursively print all decay levels
104  
105  ClassDef(AliTrack,10) // Handling of the attributes of a reconstructed particle track.
106 };
107 #endif