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     kITSpid = 2,
  38     kTPCpid = 3,
  39     kTRDpid = 4,
  40     kTOFpid = 5,
  41     kNdetectorPID = 6
  42   };
  43   enum{
  44     kCombinedTPCTRD=0
  45   };
  46   enum{
  47     kTRDSignal = 0,
  48     kITSSignal = 1
  49   };
  50   public:
  51     AliHFEpid();
  52     AliHFEpid(const AliHFEpid &c);
  53     AliHFEpid &operator=(const AliHFEpid &c);
  54     ~AliHFEpid();
  55
  56     Bool_t InitializePID(TString detectors);
  57     Bool_t IsSelected(AliVParticle *track);
  58     void SetMCEvent(AliMCEvent *mc);
  59
  60     Bool_t IsQAOn() const { return TestBit(kIsQAOn); };
  61     Bool_t HasMCData() const { return TestBit(kHasMCData); };
  62     void SetQAOn();
  63     void SetHasMCData(Bool_t hasMCdata = kTRUE) { SetBit(kHasMCData, hasMCdata); };
  64     TList *GetQAhistograms() const { return fQAlist; };
  65
  66   protected:
  67     Bool_t MakePidTpcTof(AliESDtrack *track);
  68     Bool_t MakePidTpcTrd(AliESDtrack *track);
  69     void MakePlotsItsTpc(AliESDtrack *track);
  70   private:
  71     AliHFEpidBase *fDetectorPID[kNdetectorPID];    //! Detector PID classes
  72     UInt_t fEnabledDetectors;             // Enabled Detectors
  73     TList *fQAlist;                       //! QA histograms
  74
  75   ClassDef(AliHFEpid, 1)      // Steering class for Electron ID
  76 };
  77
  78 #endif