14-jun-2001 NvE Memberfunction GetRapidity() introduced for AliTrack and AliJet
[u/mrichter/AliRoot.git] / RALICE / AliTrack.h
index 42f551e4ce92af70dc2ffc23cb8be52065cf7b8f..eff2b9eadda20d49db27b408a82cccfc8243e1ff 100644 (file)
@@ -1,52 +1,19 @@
 #ifndef ALITRACK_H
 #define ALITRACK_H
-///////////////////////////////////////////////////////////////////////////
-// Class AliTrack
-// Handling of the attributes of a reconstructed particle track.
-//
-// Coding example :
-// ----------------
-//
-// Float_t a[4]={195.,1.2,-0.04,8.5};
-// Ali4Vector pmu;
-// pmu.SetVector(a,"car");
-// AliTrack t1;
-// t1.Set4Momentum(pmu);
-//
-// Float_t b[3]={1.2,-0.04,8.5};
-// Ali3Vector p;
-// p.SetVector(b,"car");
-// AliTrack t2;
-// t2.Set3Momentum(p);
-// t2.SetCharge(0);
-// t2.SetMass(1.115);
-//
-// t1.Info();
-// t2.Info();
-//
-// Float_t pi=acos(-1.);
-// Float_t thcms=0.2*pi; // decay theta angle in cms
-// Float_t phicms=pi/4.; // decay theta angle in cms
-// Float_t m1=0.938;
-// Float_t m2=0.140;
-// t2.Decay(m1,m2,thcms,phicms); // Track t2 decay : Lambda -> proton + pion
-//
-// t2.List();
-//
-// Int_t ndec=t2.GetNdecay();
-// AliTrack* d1=t2.GetDecayTrack(1); // Access to decay track number 1
-// AliTrack* d2=t2.GetDecayTrack(2); // Access to decay track number 2
-//
-// Note : All quantities are in GeV, GeV/c or GeV/c**2
-//
-//--- NvE 10-jul-1997 UU-SAP Utrecht
-//--- Modified : NvE 06-apr-1999 UU-SAP Utrecht to inherit from Ali4Vector
-///////////////////////////////////////////////////////////////////////////
+
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
+
+// $Id$
+
+
 #include "TObject.h"
 #include "TObjArray.h"
+#include "TArrayD.h"
  
+#include "AliSignal.h"
 #include "AliBoost.h"
+#include "AliPosition.h"
  
 class AliTrack : public TObject,public Ali4Vector
 {
@@ -56,7 +23,8 @@ class AliTrack : public TObject,public Ali4Vector
   void Reset();                     // Reset all values to 0
   void Set4Momentum(Ali4Vector& p); // Set track 4-momentum
   void Set3Momentum(Ali3Vector& p); // Set track 3-momentum
-  void SetMass(Double_t m);         // Set particle mass
+  void SetMass(Double_t m,Double_t dm=0); // Set particle mass and error
+  void SetMass();                   // Set mass and error to mass hypothesis with highest prob.
   void SetCharge(Float_t q);        // Set particle charge
   void Info(TString f="car");       // Print track information for coord. frame f
   void List(TString f="car");       // Print track and decay level 1 information for coord. frame f
@@ -69,16 +37,43 @@ class AliTrack : public TObject,public Ali4Vector
   void Decay(Double_t m1,Double_t m2,Double_t thcms,Double_t phicms); // Perform 2-body decay
   Int_t GetNdecay();                // Provide number of decay products
   AliTrack* GetDecayTrack(Int_t j); // Access to decay produced track number j
+  void AddSignal(AliSignal& s);     // Relate an AliSignal to this track
+  void RemoveSignal(AliSignal& s);  // Remove related AliSignal from this track
+  Int_t GetNsignals();              // Provide number of related AliSignals
+  AliSignal* GetSignal(Int_t j);    // Access to the related AliSignal number j
+  void SetBeginPoint(AliPosition p);// Set the track begin-point
+  AliPosition GetBeginPoint();      // Provide the track begin-point
+  void SetEndPoint(AliPosition p);  // Set the track end-point
+  AliPosition GetEndPoint();        // Provide the track end-point
+  void AddMassHypothesis(Double_t prob,Double_t m,Double_t dm=0); // Add mass hypothesis data
+  Int_t GetNMassHypotheses();                // Provide number of mass hypotheses
+  Double_t GetMassHypothesis(Int_t j=0);     // Provide mass of jth hypothesis 
+  Double_t GetMassHypothesisProb(Int_t j=0); // Provide prob. of jth mass hypothesis 
+  void RemoveMassHypothesis(Int_t j);        // Remove the jth mass hypothesis 
+  Double_t GetPt();                 // Provide trans. momentum w.r.t. z-axis
+  Double_t GetPl();                 // Provide long. momentum w.r.t. z-axis
+  Double_t GetEt();                 // Provide trans. energy w.r.t. z-axis
+  Double_t GetEl();                 // Provide long. energy w.r.t. z-axis
+  Double_t GetMt();                 // Provide trans. mass w.r.t. z-axis
+  Double_t GetMt(Int_t j);          // Provide trans. mass w.r.t. z-axis and jth mass hypothesis
+  Double_t GetRapidity();           // Provide rapidity value w.r.t. z-axis
  
  protected:
-  Double_t fM;        // The mass of the particle
-  Float_t fQ;         // The charge of the particle
-  Int_t fNdec;        // The number of decay products
-  TObjArray* fDecays; // The array of decay produced tracks for output
+  Float_t fQ;          // The charge of the particle
+  Int_t fNdec;         // The number of decay products
+  TObjArray* fDecays;  // The array of decay produced tracks
+  Int_t fNsig;         // The number of related AliSignals
+  TObjArray* fSignals; // The array of related AliSignals
+  AliPosition fBegin;  // The begin-point of the track 
+  AliPosition fEnd;    // The end-point of the track 
+  Int_t fNmasses;      // The number of mass hypotheses
+  TArrayD* fMasses;    // The various mass hypotheses
+  TArrayD* fDmasses;   // The errors on the various masses
+  TArrayD* fPmasses;   // The probabilities of the various mass hypotheses
 
  private:
   void Dump(AliTrack* t,Int_t n,TString f); // Recursively print all decay levels
  
- ClassDef(AliTrack,1) // Class definition to enable ROOT I/O
+ ClassDef(AliTrack,1) // Handling of the attributes of a reconstructed particle track.
 };
 #endif