Improving event printout

[u/mrichter/AliRoot.git] / STEER / AliTrackReference.h

diff --git a/STEER/AliTrackReference.h b/STEER/AliTrackReference.h
index 85d821b161904b6f7d9610c9a72a3818ee13198c..76042a3384444c51d284a54a7adb7b7af1b6838f 100644 (file)
--- a/STEER/AliTrackReference.h
+++ b/STEER/AliTrackReference.h
@@ -14,17 +14,29 @@
 #include "TObject.h"
 #include "TMath.h"
 
-class AliExternalTrackParam;
-
 class AliTrackReference : public TObject {
-
 public:
 
+    enum constants {kDisappeared = -1,
+                   kITS   = 0,
+                   kTPC   = 1,
+                   kFRAME = 2,
+                   kTRD   = 3,
+                   kTOF   = 4,
+                   kMUON  = 5,
+                   kHMPID = 6,
+                   kT0    = 7,
+                   kEMCAL = 8,
+                   kPMD   = 10,
+                   kFMD   = 12,
+                   kVZERO = 14
+    };
   AliTrackReference();
-  AliTrackReference(Int_t label);
+  AliTrackReference(Int_t label, Int_t id = -999);
+  AliTrackReference(const AliTrackReference &tr);
   virtual ~AliTrackReference() {}
 
-  static AliExternalTrackParam * MakeTrack(const AliTrackReference *ref, Double_t mass);
+//  static AliExternalTrackParam * MakeTrack(const AliTrackReference *ref, Double_t mass);
   virtual Int_t GetTrack() const {return fTrack;}
   virtual void SetTrack(Int_t track) {fTrack=track;}
   virtual void SetLength(Float_t length){fLength=length;}
@@ -45,10 +57,12 @@ public:
   virtual Float_t Pz() const {return fPz;}
   virtual Float_t P() const {return TMath::Sqrt(fPx*fPx+fPy*fPy+fPz*fPz);}
   virtual Int_t   UserId() const {return fUserId;}
+  virtual Int_t   DetectorId() const {return fDetectorId;}
+  virtual void SetDetectorId(Int_t id){fDetectorId = id;}
   virtual void SetPosition(Float_t x, Float_t y, Float_t z){fX=x; fY=y; fZ=z;}
   virtual void SetMomentum(Float_t px, Float_t py, Float_t pz){fPx=px; fPy=py; fPz=pz;}
   virtual void SetUserId(Int_t userId){fUserId=userId;}
-
+ 
   // Methods to get position of the track reference in 
   // in the TPC/TRD/TOF Tracking coordinate system
 
@@ -61,23 +75,25 @@ public:
   Bool_t IsSortable() const {return kTRUE;}
   Int_t Compare(const TObject *obj) const {
     Int_t ll = ((AliTrackReference*)obj)->GetTrack();
-    if (ll == fTrack) return 0;
     if (ll < fTrack) return 1;
     if (ll > fTrack) return -1;
+    return 0;
   }
 
+  virtual void Print(Option_t* opt="") const;
+    
 protected:
-  Int_t     fTrack;  // Track number
-  Float_t   fX;      // X reference position of the track
-  Float_t   fY;      // Y reference position of the track
-  Float_t   fZ;      // Z reference position of the track
-  Float_t   fPx;     // momentum
-  Float_t   fPy;     // momentum
-  Float_t   fPz;     // momentum
-  Float_t   fLength; // track lenght from its origin in cm
-  Float_t   fTime;   // time of flight in cm  
-  Int_t     fUserId; // optional Id defined by user
-
-  ClassDef(AliTrackReference,5)  //Base class for all Alice track references
+  Int_t     fTrack;      // Track number
+  Float_t   fX;          // X reference position of the track
+  Float_t   fY;          // Y reference position of the track
+  Float_t   fZ;          // Z reference position of the track
+  Float_t   fPx;         // momentum
+  Float_t   fPy;         // momentum
+  Float_t   fPz;         // momentum
+  Float_t   fLength;     // track lenght from its origin in cm
+  Float_t   fTime;       // time of flight in cm  
+  Int_t     fUserId;     // optional Id defined by user
+  Int_t     fDetectorId; // Detector Id
+  ClassDef(AliTrackReference,6)  //Base class for all Alice track references
 };
 #endif