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