Optional scoring for background studies in HALL.

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

diff --git a/TRD/AliTRDpidESD.h b/TRD/AliTRDpidESD.h
index 974c6e5431eeeb8443f35637203220d16840a777..9fd25f375d3b467b142b2f2fd4edd9b591d0d963 100644 (file)
--- a/TRD/AliTRDpidESD.h
+++ b/TRD/AliTRDpidESD.h
@@ -7,15 +7,21 @@
 
 ////////////////////////////////////////////////////////////////////////////
 //                                                                        //
-//  Assigns the PID probabilities based on TRD information to the ESDs    //
+// 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 <Rtypes.h>
 
 #include <TObject.h>
 
-class AliESD;
+class AliESDEvent;
+class AliESDtrack;
+class AliExternalTrackParam;
 
 class AliTRDpidESD : public TObject {
 
@@ -23,28 +29,35 @@ class AliTRDpidESD : public TObject {
 
   AliTRDpidESD();
   AliTRDpidESD(const AliTRDpidESD &p);
-  virtual ~AliTRDpidESD() {}
+  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);
 
-  static  Int_t   MakePID(AliESD *event);
+          void    SetCheckTrackStatus(Bool_t status = kTRUE) { fgCheckTrackStatus = status; };
+          void    SetCheckKinkStatus(Bool_t status = kTRUE)  { fgCheckKinkStatus  = status; };
+          void    SetMinPlane(Int_t plane)                   { fgMinPlane         = plane;  };
 
-          void    SetCheckTrackStatus(Bool_t status = kTRUE) { fCheckTrackStatus = status; };
-          void    SetCheckKinkStatus(Bool_t status = kTRUE)  { fCheckKinkStatus  = status; };
-          void    SetMinPlane(Int_t plane)                   { fMinPlane         = plane;  };
+         Bool_t  GetCheckTrackStatus() const                { return fgCheckTrackStatus;   };      
+         Bool_t  GetCheckKinkStatus() const                 { return fgCheckKinkStatus;    };      
+          Int_t   GetMinPlane() const                        { return fgMinPlane;           };
 
-         Bool_t  GetCheckTrackStatus()                      { return fCheckTrackStatus;   };      
-         Bool_t  GetCheckKinkStatus()                       { return fCheckKinkStatus;    };      
-          Int_t   GetMinPlane()                              { return fMinPlane;           };
+private:
 
- private:
+          Bool_t  RecalculateTrackSegmentKine(AliESDtrack * const t
+                                            , Int_t plan
+                                            , Float_t &mom
+                                            , Float_t &length);
 
-  static  Bool_t  fCheckTrackStatus;    // Enable check on ESD track status
-  static  Bool_t  fCheckKinkStatus;     // Enable check on ESD kink track
-  static  Int_t   fMinPlane;            // Minimum number of planes
+  static  Bool_t  fgCheckTrackStatus;           //  Enable check on ESD track status
+  static  Bool_t  fgCheckKinkStatus;            //  Enable check on ESD kink track
+  static  Int_t   fgMinPlane;                   //  Minimum number of planes
 
-  ClassDef(AliTRDpidESD,2)              // TRD PID class
+  AliExternalTrackParam *fTrack;                //! Memory holder for Track segment calculations
+       
+  ClassDef(AliTRDpidESD,2)                      //  TRD PID class
 
 };