TRD/AliTRDpidESD.h

   1 #ifndef ALITRDPIDESD_H
   2 #define ALITRDPIDESD_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6 /* $Id$ */
   7
   8 ////////////////////////////////////////////////////////////////////////////
   9 //                                                                        //
  10 //  Assigns the PID probabilities based on TRD information to the ESDs    //
  11 //                                                                        //
  12 ////////////////////////////////////////////////////////////////////////////
  13
  14 #include <Rtypes.h>
  15
  16 #include <TObject.h>
  17
  18 class AliESDEvent;
  19 class AliESDtrack;
  20 class AliExternalTrackParam;
  21 class AliTRDpidESD : public TObject {
  22
  23  public:
  24
  25   AliTRDpidESD();
  26   AliTRDpidESD(const AliTRDpidESD &p);
  27   virtual ~AliTRDpidESD();
  28   AliTRDpidESD &operator=(const AliTRDpidESD &p);
  29
  30   virtual void    Copy(TObject &p) const;
  31   static  Bool_t  CheckTrack(AliESDtrack *t);
  32           Int_t   MakePID(AliESDEvent *event);
  33
  34           void    SetCheckTrackStatus(Bool_t status = kTRUE) { fCheckTrackStatus = status; };
  35           void    SetCheckKinkStatus(Bool_t status = kTRUE)  { fCheckKinkStatus  = status; };
  36           void    SetMinPlane(Int_t plane)                   { fMinPlane         = plane;  };
  37
  38           Bool_t  GetCheckTrackStatus()                      { return fCheckTrackStatus;   };
  39           Bool_t  GetCheckKinkStatus()                       { return fCheckKinkStatus;    };
  40           Int_t   GetMinPlane()                              { return fMinPlane;           };
  41
  42 private:
  43   Bool_t  RecalculateTrackSegmentKine(AliESDtrack *t, Int_t plan, Float_t &mom, Float_t &length);
  44
  45 private:
  46
  47   static  Bool_t  fCheckTrackStatus;    // Enable check on ESD track status
  48   static  Bool_t  fCheckKinkStatus;     // Enable check on ESD kink track
  49   static  Int_t   fMinPlane;            // Minimum number of planes
  50
  51         AliExternalTrackParam *fTrack;                          //! Memory holder for Track segment calculations
  52
  53   ClassDef(AliTRDpidESD,2)              // TRD PID class
  54
  55 };
  56
  57 #endif
  58
  59