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 //
  16 // Steering class for electron identification
  17 // Combines detector PID objects
  18 // For more information please check the implementation file
  19 //
  20 #ifndef ALIHFEPID_H
  21 #define ALIHFEPID_H
  22
  23 #ifndef ROOT_TObject
  24 #include <TObject.h>
  25 #endif
  26
  27 #ifndef ALIHFEPIDBASE_H
  28 #include "AliHFEpidBase.h"
  29 #endif
  30
  31 class AliESDtrack;
  32 class AliVParticle;
  33 class AliMCParticle;
  34
  35 class TList;
  36
  37 class AliHFEpid : public TObject{
  38  public:
  39     AliHFEpid();
  40     AliHFEpid(const AliHFEpid &c);
  41     AliHFEpid &operator=(const AliHFEpid &c);
  42     ~AliHFEpid();
  43
  44     Bool_t InitializePID(TString argument);
  45     Bool_t IsSelected(AliHFEpidObject *track);
  46
  47     Bool_t IsQAOn() const { return TestBit(kIsQAOn); };
  48     Bool_t HasMCData() const { return TestBit(kHasMCData); };
  49     void SetDebugLevel(Int_t debugLevel) { fDebugLevel = debugLevel; }
  50     void SetQAOn();
  51     void SetHasMCData(Bool_t hasMCdata = kTRUE) { SetBit(kHasMCData, hasMCdata); };
  52     TList *GetQAhistograms() const { return fQAlist; };
  53
  54   protected:
  55     Bool_t MakePidTpcTof(AliHFEpidObject *track);
  56     Bool_t MakePidTpcTrd(AliHFEpidObject *track);
  57     void MakePlotsItsTpc(AliHFEpidObject *track);
  58
  59     // Stratgies
  60     void InitStrategy1();
  61     void InitStrategy2();
  62     void InitStrategy3();
  63     void InitStrategy4();
  64     void InitStrategy5();
  65     Bool_t IdentifyStrategy1(AliHFEpidObject *track);
  66     Bool_t IdentifyStrategy2(AliHFEpidObject *track);
  67     Bool_t IdentifyStrategy3(AliHFEpidObject *track);
  68     Bool_t IdentifyStrategy4(AliHFEpidObject *track);
  69     Bool_t IdentifyStrategy5(AliHFEpidObject *track);
  70   private:
  71     enum{
  72       kIsQAOn = BIT(14),
  73       kHasMCData = BIT(15)
  74     };
  75     enum{
  76       kMCpid = 0,
  77       kESDpid = 1,
  78       kITSpid = 2,
  79       kTPCpid = 3,
  80       kTRDpid = 4,
  81       kTOFpid = 5,
  82       kNdetectorPID = 6
  83     };
  84     enum{
  85       kCombinedTPCTRD=0
  86     };
  87     enum{
  88       kTRDSignal = 0,
  89       kITSSignal = 1
  90     };
  91     AliHFEpidBase *fDetectorPID[kNdetectorPID];     //! Detector PID classes
  92     UInt_t fEnabledDetectors;                       //  Enabled Detectors
  93     UInt_t fPIDstrategy;                            //  PID Strategy
  94     TList *fQAlist;                                 //! QA histograms
  95     Int_t fDebugLevel;                              //  Debug Level
  96
  97   ClassDef(AliHFEpid, 1)      // Steering class for Electron ID
  98 };
  99
 100 #endif