PWG3/hfe/AliAnalysisTaskHFEpidQA.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 // Task for PID QA
  17 // Using AliHFEpidQA and AliHFEMCpidQA
  18 // More information can be found in the source file
  19 //
  20 #ifndef ALIANALYSISTASKHFEPIDQA_H
  21 #define ALIANALYSISTASKHFEPIDQA_H
  22
  23 #ifndef ALIANALYSISTASKSE_H
  24 #include "AliAnalysisTaskSE.h"
  25 #endif
  26
  27 class TH1;
  28 class TList;
  29 class TFile;
  30
  31 class AliLog;
  32 class AliMCEvent;
  33
  34 class AliHFEpidQA;
  35
  36 class AliAnalysisTaskHFEpidQA : public AliAnalysisTaskSE{
  37   public:
  38     AliAnalysisTaskHFEpidQA();
  39     AliAnalysisTaskHFEpidQA(const Char_t *name);
  40     ~AliAnalysisTaskHFEpidQA();
  41
  42     virtual void UserCreateOutputObjects();
  43     virtual void UserExec(Option_t *);
  44     virtual void Terminate(Option_t *);
  45     virtual Bool_t UserNotify();
  46
  47     Bool_t HasV0pidQA() const { return TestBit(kV0pidQA); };
  48     Bool_t HasRecalculateTRDpid() const { return TestBit(kRecalculateTRDpid); };
  49     void SetV0pidQA(Bool_t v0pidQA = kTRUE) { SetBit(kV0pidQA, v0pidQA); };
  50     void SetRecalculateTRDpid(Bool_t recal = kTRUE) { SetBit(kRecalculateTRDpid, recal); };
  51     void SetTRDTotalChargeInSlice0() { fTRDTotalChargeInSlice0 = kTRUE; }
  52
  53     void SetNNref(TFile *f) { fNNref = f; };
  54
  55   private:
  56     enum{
  57       kV0pidQA = BIT(22),
  58       kRecalculateTRDpid = BIT(23)
  59     };
  60     AliAnalysisTaskHFEpidQA(const AliAnalysisTaskHFEpidQA &ref);
  61     AliAnalysisTaskHFEpidQA &operator=(const AliAnalysisTaskHFEpidQA &ref);
  62     AliHFEpidQA *fPIDqa;    //! The heart of the analysis
  63     TList *fOutput;         //! Container for output histos
  64     TH1 *fEvents;           //! Number of Events
  65     TFile  *fNNref;         //  reference file for NN
  66     Bool_t fTRDTotalChargeInSlice0;   // Fix for Foreware/Backward compatibility
  67
  68     ClassDef(AliAnalysisTaskHFEpidQA, 1)
  69 };
  70
  71 #endif
  72