New package for heavy flavour electrons analysis (M.Fasel)
[u/mrichter/AliRoot.git] / PWG3 / hfe / AliHFEpid.h
1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3 *                                                                        *
4 * Author: The ALICE Off-line Project.                                    *
5 * Contributors are mentioned in the code where appropriate.              *
6 *                                                                        *
7 * Permission to use, copy, modify and distribute this software and its   *
8 * documentation strictly for non-commercial purposes is hereby granted   *
9 * without fee, provided that the above copyright notice appears in all   *
10 * copies and that both the copyright notice and this permission notice   *
11 * appear in the supporting documentation. The authors make no claims     *
12 * about the suitability of this software for any purpose. It is          *
13 * provided "as is" without express or implied warranty.                  *
14 **************************************************************************/
15 #ifndef __ALIHFEPID_H__
16 #define __ALIHFEPID_H__
17
18 #ifndef ROOT_TObject
19 #include <TObject.h>
20 #endif
21
22 class AliHFEpidBase;
23 class AliESDtrack;
24 class AliVParticle;
25 class AliMCEvent;
26
27 class TList;
28
29 class AliHFEpid : public TObject{
30   enum{
31     kIsQAOn = BIT(14),
32     kHasMCData = BIT(15)
33   };
34   enum{
35     kMCpid = 0,
36     kESDpid = 1,
37     kTPCpid = 2,
38     kTRDpid = 3,
39     kTOFpid = 4,
40     kNdetectorPID = 5
41   };
42   public:
43     AliHFEpid();
44     AliHFEpid(const AliHFEpid &c);
45     AliHFEpid &operator=(const AliHFEpid &c);
46     ~AliHFEpid();
47     
48     Bool_t InitializePID(TString detectors);
49     Bool_t IsSelected(AliVParticle *track);
50     void SetMCEvent(AliMCEvent *mc);
51
52     Bool_t IsQAOn() const { return TestBit(kIsQAOn); };
53     Bool_t HasMCData() const { return TestBit(kHasMCData); };
54     void SetQAOn();
55     void SetHasMCData(Bool_t hasMCdata = kTRUE) { SetBit(kHasMCData, hasMCdata); };
56     TList *GetQAhistograms() const { return fQAlist; };
57
58   protected:
59     Bool_t MakePID_TPC_TOF(AliESDtrack *track);
60   private:
61     AliHFEpidBase *fDetectorPID[kNdetectorPID];    //! Detector PID classes
62     UInt_t fEnabledDetectors;             // Enabled Detectors
63     TList *fQAlist;                       //! QA histograms
64
65   ClassDef(AliHFEpid, 1)      // Steering class for Electron ID
66 };
67
68 #endif