Update of TRD code

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

diff --git a/TRD/AliTRDhit.h b/TRD/AliTRDhit.h
index 5b7bd4cda42f84b29cf665a8bef6b9aff671db1e..46749f066a3751c8c77d93c907fa23cd9900aa32 100644 (file)
--- a/TRD/AliTRDhit.h
+++ b/TRD/AliTRDhit.h
@@ -20,15 +20,32 @@ class AliTRDhit : public AliHit {
   AliTRDhit(Int_t shunt, Int_t track, Int_t det, Float_t *hits, Int_t q);
   virtual ~AliTRDhit();
 
-          Int_t GetDetector() const { return fDetector; };
-          Int_t GetCharge() const   { return fQ;        };
+          Int_t  GetDetector() const         { return fDetector; };
+          Int_t  GetCharge() const           { return fQ;        };
+
+          Bool_t FromDrift() const           { return TestBit(kDrift);         };
+          Bool_t FromAmplification() const   { return TestBit(kAmplification); };
+          Bool_t FromTRphoton() const        { return TestBit(kTRphoton);      };
+          Bool_t FromTest() const            { return TestBit(kTest);          };
+
+          void   SetDrift()                  { SetBit(kDrift);         };
+          void   SetAmplification()          { SetBit(kAmplification); };
+          void   SetTRphoton()               { SetBit(kTRphoton);      };
+          void   SetTest()                   { SetBit(kTest);          }; 
 
  protected:
 
-  UShort_t     fDetector;   // TRD detector number
-  Short_t      fQ;          // Charge created by a hit. TR signals are negative.
+  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.
 
-  ClassDef(AliTRDhit,3)     // Hit for the Transition Radiation Detector
+  ClassDef(AliTRDhit,3)             // Hit for the Transition Radiation Detector
 
 };