PWGHF/vertexingHF/AliAnalysisTaskSEHFQA.h

   1 #ifndef ALIANALYSISTASKSEHFQUALITYASSURANCE_H
   2 #define ALIANALYSISTASKSEHFQUALITYASSURANCE_H
   3
   4 /* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
   5  * See cxx source for full Copyright notice                               */
   6
   7 /* $Id$ */
   8
   9 //*************************************************************************
  10 // Class AliAnalysisTaskSEHFQA
  11 // AliAnalysisTaskSE for HF quality assurance
  12 // Authors: C.Bianchin, chiara.bianchin@pd.infn.it
  13 //*************************************************************************
  14
  15 #include <TROOT.h>
  16 #include <TSystem.h>
  17
  18 #include "AliAnalysisTaskSE.h"
  19
  20 class AliRDHFCuts;
  21 class TH1F;
  22
  23 class AliAnalysisTaskSEHFQA : public AliAnalysisTaskSE
  24 {
  25
  26  public:
  27
  28   enum DecChannel {kDplustoKpipi,kD0toKpi,kDstartoKpipi,kDstoKKpi,kD0toKpipipi,kLambdactopKpi};
  29
  30   AliAnalysisTaskSEHFQA();
  31   AliAnalysisTaskSEHFQA(const char *name, DecChannel ch, AliRDHFCuts* cuts);
  32   virtual ~AliAnalysisTaskSEHFQA();
  33
  34   // Implementation of interface methods
  35   virtual void UserCreateOutputObjects();
  36   virtual void Init();
  37   virtual void LocalInit() {Init();}
  38   virtual void UserExec(Option_t *option);
  39   virtual void Terminate(Option_t *option);
  40
  41   //setters
  42   void SetReadMC(Bool_t mcflag){fReadMC=mcflag;}
  43   void SetSimpleMode(Bool_t flag){fSimpleMode=flag;}
  44   void SetTrackOn(Bool_t trackon=kTRUE){fOnOff[0]=trackon;}
  45   void SetPIDOn(Bool_t pidon=kTRUE){fOnOff[1]=pidon;}
  46   void SetCentralityOn(Bool_t centron=kTRUE){fOnOff[2]=centron;}
  47   void SetEvSelectionOn(Bool_t evselon=kTRUE){fOnOff[3]=evselon;}
  48
  49   //getters
  50   AliRDHFCuts* GetCutObject() const {return fCuts;}
  51   DecChannel GetDecayChannel()const {return fDecayChannel;}
  52   Bool_t GetTrackStatus() const {return fOnOff[0];}
  53   Bool_t GetPIDStatus() const {return fOnOff[1];}
  54   Bool_t GetCentralityStatus() const {return fOnOff[2];}
  55   Bool_t GetEvSelStatus() const {return fOnOff[3];}
  56
  57  private:
  58   AliAnalysisTaskSEHFQA(const AliAnalysisTaskSEHFQA &source);
  59   AliAnalysisTaskSEHFQA operator=(const AliAnalysisTaskSEHFQA &source);
  60
  61  TH1F*  fNEntries;         //! histogram with number of events on output slot 1
  62  TList* fOutputPID;        //! list sent on output slot 2
  63  TList* fOutputTrack;      //! list sent on output slot 3
  64  TList* fOutputCounters;   //! list sent on output slot 5
  65  TList* fOutputCheckCentrality;   //! list sent on output slot 6
  66  TList* fOutputEvSelection; //! list sent on output slot 7
  67  DecChannel fDecayChannel; //identify the decay channel
  68  AliRDHFCuts* fCuts;       // object containing cuts
  69  AliRDHFCuts::ECentrality fEstimator; //2nd estimator for centrality
  70  Bool_t fReadMC;           // flag to read MC
  71  Bool_t fSimpleMode;       // if true, don't do candidates (much faster in PbPb)
  72  Bool_t fOnOff[4];         // on-off the QA on tracks (0), PID (1), centrality (2), event selection -- default is {kTRUE,kTRUE,kTRUE,kTRUE}
  73  ClassDef(AliAnalysisTaskSEHFQA,6); //AnalysisTaskSE for the quality assurance of HF in hadrons
  74
  75 };
  76
  77 #endif
  78