Continuation of TFlukaGeo based on TGeo.
[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(const AliTrack& t);            // Copy constructor
25   virtual TObject* Clone(const char* name="") const; // 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 hyp. 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() const;        // 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() const;              // 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() const;                // Provide number of decay products
42   AliTrack* GetDecayTrack(Int_t j) const; // 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() const;              // Provide number of related AliSignals
48   AliSignal* GetSignal(Int_t j) const;    // 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 SetReferencePoint(AliPosition& p); // Set the track reference-point for the 3-momentum vector
54   AliPosition* GetReferencePoint();       // Provide the track reference-point for the 3-momentum vector
55   void AddTrackHypothesis(AliTrack& t);   // Add track hypothesis
56   void AddTrackHypothesis(Double_t prob,Double_t m,Double_t dm=0); // Add track hypothesis with mass data
57   Int_t GetNhypotheses() const;           // Provide number of track hypotheses
58   AliTrack* GetTrackHypothesis(Int_t j=0) const; // Provide the j-th track hypothesis 
59   void RemoveTrackHypothesis(AliTrack& t);// Remove the specified track hypothesis 
60   void RemoveTrackHypotheses();           // Remove all track hypotheses 
61   Double_t GetPt();                       // Provide trans. momentum w.r.t. z-axis
62   Double_t GetPl();                       // Provide long. momentum w.r.t. z-axis
63   Double_t GetEt();                       // Provide trans. energy w.r.t. z-axis
64   Double_t GetEl();                       // Provide long. energy w.r.t. z-axis
65   Double_t GetMt();                       // Provide trans. mass w.r.t. z-axis
66   Double_t GetRapidity();                 // Provide rapidity value w.r.t. z-axis
67   void SetImpactPoint(AliPosition& p,TString q); // Set the impact-point in plane "q=0"
68   AliPosition* GetImpactPoint(TString q);        // Provide the impact-point in plane "q=0"
69   void SetId(Int_t id);                   // Set the user defined unique track identifier
70   Int_t GetId() const;                    // Provide the user defined unique track identifier
71   void SetClosestPoint(AliPosition& p);   // Set position p as point of closest approach w.r.t. some reference
72   AliPosition* GetClosestPoint();         // Provide point of closest approach w.r.t. some reference
73   void SetChi2(Float_t chi2);             // Set the chi-squared value of the track fit
74   void SetNdf(Int_t ndf);                 // Set the number of degrees of freedom for the track fit
75   Float_t GetChi2() const;                // Provide the chi-squared value of the track fit
76   Int_t GetNdf() const;                   // Provide the number of degrees of freedom for the track fit
77   void SetParticleCode(Int_t code);       // Set the user defined particle id code (e.g. the PDF convention)
78   Int_t GetParticleCode() const;          // Provide the user defined particle id code
79   void SetParentTrack(AliTrack* t);       // Set pointer to the parent track
80   AliTrack* GetParentTrack();             // Provide pointer to the parent track
81   void SetProb(Double_t prob);            // Set the hypothesis probability for this track
82   Float_t GetProb() const;                // Provide the hypothesis probability for this track
83
84  
85  protected:
86   void Init();               // Initialisation of pointers etc...
87   Float_t fQ;                // The charge of the particle
88   TObjArray* fDecays;        // The array of decay produced tracks
89   TObjArray* fSignals;       // The array of related AliSignals
90   TObjArray* fHypotheses;    // The array of track hypotheses
91   AliPositionObj* fBegin;    // The begin-point of the track 
92   AliPositionObj* fEnd;      // The end-point of the track 
93   AliPositionObj* fRef;      // The reference-point of the track for the 3-momentum vector
94   AliPositionObj* fImpactXY; // The (extrapolated) impact-point in the plane z=0
95   AliPositionObj* fImpactXZ; // The (extrapolated) impact-point in the plane y=0
96   AliPositionObj* fImpactYZ; // The (extrapolated) impact-point in the plane x=0
97   Int_t fUserId;             // The user defined identifier
98   AliPositionObj* fClosest;  // The (extrapolated) point of closest approach w.r.t some reference
99   Float_t fChi2;             // The Chi-squared of the track fit
100   Int_t fNdf;                // The number of degrees of freedom of the track fit
101   Int_t fCode;               // The user defined particle id code
102   AliTrack* fParent;         // Pointer to the parent track
103   Float_t fProb;             // Probability for this track as a hypothesis
104
105  private:
106   void Dumps(AliTrack* t,Int_t n,TString f); // Recursively print all decay levels
107  
108  ClassDef(AliTrack,12) // Handling of the attributes of a reconstructed particle track.
109 };
110 #endif