-//-------------------------------------------------------
-// TRD PID class
-// A very naive design... And the implementation is even poorer...
-// Should be made better by the detector experts...
-//-------------------------------------------------------
-#include <Rtypes.h>
-
-class AliESD;
-
-class AliTRDpidESD {
-public:
- AliTRDpidESD(Double_t *param);
- Int_t MakePID(AliESD *event);
- static Double_t Bethe(Double_t bg);
+/* $Id$ */
+
+////////////////////////////////////////////////////////////////////////////
+// //
+// Assigns the PID probabilities based on TRD information to the ESDs //
+// //
+// Authors : //
+// Prashant Shukla <shukla@pi0.physi.uni-heidelberg.de> (orig. version) //
+// Alex Bercuci (a.bercuci@gsi.de) //
+// //
+////////////////////////////////////////////////////////////////////////////
+
+//#include <Rtypes.h>
+
+#include <TObject.h>
+
+class AliESDEvent;
+class AliESDtrack;
+class AliExternalTrackParam;
+
+class AliTRDpidESD : public TObject {
+
+ public:
+
+ AliTRDpidESD();
+ AliTRDpidESD(const AliTRDpidESD &p);
+ virtual ~AliTRDpidESD();
+ AliTRDpidESD &operator=(const AliTRDpidESD &p);
+
+ virtual void Copy(TObject &p) const;
+ static Bool_t CheckTrack(AliESDtrack * const t);
+ Int_t MakePID(AliESDEvent * const event);
+
+ void SetCheckTrackStatus(Bool_t status = kTRUE) { fgCheckTrackStatus = status; };
+ void SetCheckKinkStatus(Bool_t status = kTRUE) { fgCheckKinkStatus = status; };
+ void SetMinPlane(Int_t plane) { fgMinPlane = plane; };
+
+ Bool_t GetCheckTrackStatus() const { return fgCheckTrackStatus; };
+ Bool_t GetCheckKinkStatus() const { return fgCheckKinkStatus; };
+ Int_t GetMinPlane() const { return fgMinPlane; };
+