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 /**
  10  * Make a background distribution from simulated data
  11  * @ingroup FMD_ana
  12  *
  13  *
  14  */
  15
  16 class AliFMDAnaCalibBackgroundCorrection;
  17
  18 class AliFMDAnalysisTaskGenerateBackground : public AliAnalysisTaskSE
  19 {
  20  public:
  21   AliFMDAnalysisTaskGenerateBackground();
  22     AliFMDAnalysisTaskGenerateBackground(const char* name);
  23     ~AliFMDAnalysisTaskGenerateBackground() {;}
  24  AliFMDAnalysisTaskGenerateBackground(const AliFMDAnalysisTaskGenerateBackground& o) : AliAnalysisTaskSE(),
  25       fListOfHits(),
  26       fListOfPrimaries(),
  27       fListOfCorrection(),
  28       fVertexBins(o.fVertexBins),
  29       fLastTrackByStrip(o.fLastTrackByStrip),
  30       fHitsByStrip(o.fHitsByStrip),
  31       fZvtxCut(o.fZvtxCut),
  32       fNvtxBins(o.fNvtxBins),
  33       fNbinsEta(o.fNbinsEta),
  34       fBackground(o.fBackground)
  35       {}
  36     AliFMDAnalysisTaskGenerateBackground& operator=(const AliFMDAnalysisTaskGenerateBackground&) { return *this; }
  37
  38     virtual void Init();
  39     virtual void UserCreateOutputObjects();
  40     virtual void UserExec(Option_t* /*option*/);
  41     void  Terminate(Option_t */*option*/);
  42     void SetZvtxCut(Float_t vtxcut) {fZvtxCut = vtxcut;}
  43     void SetNvtxBins(Int_t nvtxbins) {fNvtxBins = nvtxbins;}
  44     void SetNbinsEta(Int_t netabins) {fNbinsEta = netabins;}
  45     void ReadFromFile(const Char_t* filename = "background.root", Bool_t storeInOCDB = kFALSE, Int_t runNo=0);
  46  private:
  47
  48     void GenerateCorrection();
  49
  50     TList fListOfHits;
  51     TList fListOfPrimaries;
  52     TList fListOfCorrection;
  53     TH1F  fVertexBins;
  54     AliFMDFloatMap fLastTrackByStrip;
  55     AliFMDFloatMap fHitsByStrip;
  56     Float_t fZvtxCut;
  57     Int_t fNvtxBins;
  58     Int_t fNbinsEta;
  59     AliFMDAnaCalibBackgroundCorrection* fBackground;
  60     ClassDef(AliFMDAnalysisTaskGenerateBackground, 1);
  61
  62 };
  63 #endif
  64 // Local Variables:
  65 //   mode: C++
  66 // End: