Pyhtia comparison extended

[u/mrichter/AliRoot.git] / TRD / AliTRDhit.h

diff --git a/TRD/AliTRDhit.h b/TRD/AliTRDhit.h
index 8b87a1a87dafad906e8865d0c22d829ef7f2751a..3fa50278d0339d3d27723f857aa0c2850934d8f2 100644 (file)
--- a/TRD/AliTRDhit.h
+++ b/TRD/AliTRDhit.h
@@ -29,8 +29,23 @@ class AliTRDhit : public AliHit {
           void   SetDetector(Int_t det)      { fDetector = det; }
           void   SetQ(Int_t q)               { fQ        =   q; }
 
+          Bool_t FromDrift() const           { return TestBit(kDrift);         };
+          Bool_t FromAmplification() const   { return TestBit(kAmplification); };
+          Bool_t FromTRphoton() const        { return (fQ < 0);                };
+
+          void   SetDrift()                  { SetBit(kDrift);         };
+          void   SetAmplification()          { SetBit(kAmplification); };
+          void   SetTRphoton()               { SetBit(kTRphoton);      };
+
  protected:
 
+  enum {
+    kDrift         = 0x00000001,    // Hit is from the drift region
+    kAmplification = 0x00000002,    // Hit is from the amplification region
+    kTRphoton      = 0x00000004,    // Hit is from a TR photon
+    kTest          = 0x00000008     // Hit is a special test hit
+  };
+
   UShort_t     fDetector;           // TRD detector number
   Short_t      fQ;                  // Charge created by a hit. TR signals are negative.