FMD/analysis/AliFMDAnalysisTaskGenerateBackground.h

   1 #ifndef ALIFMDANALYSISTASKGENERATEBACKGROUND_H
   2 #define ALIFMDANALYSISTASKGENERATEBACKGROUND_H
   3
   4 #include "AliAnalysisTaskSE.h"
   5 #include "TList.h"
   6 #include "AliFMDFloatMap.h"
   7 #include "TH1F.h"
   8
   9 class AliFMDAnaCalibBackgroundCorrection;
  10
  11 class AliFMDAnalysisTaskGenerateBackground : public AliAnalysisTaskSE
  12 {
  13  public:
  14   AliFMDAnalysisTaskGenerateBackground();
  15     AliFMDAnalysisTaskGenerateBackground(const char* name);
  16     ~AliFMDAnalysisTaskGenerateBackground() {;}
  17  AliFMDAnalysisTaskGenerateBackground(const AliFMDAnalysisTaskGenerateBackground& o) : AliAnalysisTaskSE(),
  18       fListOfHits(),
  19       fListOfPrimaries(),
  20       fListOfCorrection(),
  21       fVertexBins(o.fVertexBins),
  22       fLastTrackByStrip(o.fLastTrackByStrip),
  23       fZvtxCut(o.fZvtxCut),
  24       fNvtxBins(o.fNvtxBins),
  25       fNbinsEta(o.fNbinsEta),
  26       fBackground(o.fBackground)
  27       {}
  28     AliFMDAnalysisTaskGenerateBackground& operator=(const AliFMDAnalysisTaskGenerateBackground&) { return *this; }
  29
  30     virtual void Init();
  31     virtual void UserCreateOutputObjects();
  32     virtual void UserExec(Option_t* /*option*/);
  33     void  Terminate(Option_t */*option*/);
  34     void SetZvtxCut(Float_t vtxcut) {fZvtxCut = vtxcut;}
  35     void SetNvtxBins(Int_t nvtxbins) {fNvtxBins = nvtxbins;}
  36     void SetNbinsEta(Int_t netabins) {fNbinsEta = netabins;}
  37     void ReadFromFile(const Char_t* filename = "background.root", Bool_t storeInOCDB = kFALSE, Int_t runNo=0);
  38  private:
  39
  40     void GenerateCorrection();
  41
  42     TList fListOfHits;
  43     TList fListOfPrimaries;
  44     TList fListOfCorrection;
  45     TH1F  fVertexBins;
  46     AliFMDFloatMap fLastTrackByStrip;
  47     Float_t fZvtxCut;
  48     Int_t fNvtxBins;
  49     Int_t fNbinsEta;
  50     AliFMDAnaCalibBackgroundCorrection* fBackground;
  51     ClassDef(AliFMDAnalysisTaskGenerateBackground, 1);
  52
  53 };
  54 #endif