Classes moved to STEERBase.
[u/mrichter/AliRoot.git] / 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