[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis / AliFMDAnalysisTaskGenerateCorrection.h

#ifndef ALIFMDANALYSISTASKGENERATECORRECTION_H
#define ALIFMDANALYSISTASKGENERATECORRECTION_H

#include "AliAnalysisTaskSE.h"
#include "TList.h"
#include "AliFMDFloatMap.h"
#include "TH1F.h"

class AliFMDAnaCalibBackgroundCorrection;
class AliFMDAnaCalibEventSelectionEfficiency;

class AliFMDAnalysisTaskGenerateCorrection : public AliAnalysisTaskSE
{
 public:
  AliFMDAnalysisTaskGenerateCorrection();
    AliFMDAnalysisTaskGenerateCorrection(const char* name);
    ~AliFMDAnalysisTaskGenerateCorrection() {;}
 AliFMDAnalysisTaskGenerateCorrection(const AliFMDAnalysisTaskGenerateCorrection& o) : AliAnalysisTaskSE(), 
      fListOfHits(), 
      fListOfPrimaries(),
      fListOfCorrection(),
      fVertexBins(o.fVertexBins),
      fLastTrackByStrip(o.fLastTrackByStrip),
      fHitsByStrip(o.fHitsByStrip),
      fZvtxCut(o.fZvtxCut),
      fNvtxBins(o.fNvtxBins),
      fNbinsEta(o.fNbinsEta),
      fBackground(o.fBackground),
      fEventSelectionEff(o.fEventSelectionEff)
      {}
    AliFMDAnalysisTaskGenerateCorrection& operator=(const AliFMDAnalysisTaskGenerateCorrection&) { return *this; }
    
    virtual void Init();
    virtual void UserCreateOutputObjects();
    virtual void UserExec(Option_t* /*option*/);
    void  Terminate(Option_t */*option*/);
    void SetZvtxCut(Float_t vtxcut) {fZvtxCut = vtxcut;}
    void SetNvtxBins(Int_t nvtxbins) {fNvtxBins = nvtxbins;}
    void SetNbinsEta(Int_t netabins) {fNbinsEta = netabins;}
    void ReadFromFile(const Char_t* filename = "background.root", Bool_t storeInOCDB = kFALSE, Int_t runNo=0);
 private:
    
    void GenerateCorrection();
    
    TList fListOfHits;
    TList fListOfPrimaries;
    TList fListOfCorrection;
    TH1F  fVertexBins;
    AliFMDFloatMap fLastTrackByStrip;
    AliFMDFloatMap fHitsByStrip;
    Float_t fZvtxCut;
    Int_t fNvtxBins;
    Int_t fNbinsEta;
    AliFMDAnaCalibBackgroundCorrection* fBackground;
    AliFMDAnaCalibEventSelectionEfficiency*     fEventSelectionEff;
    
    ClassDef(AliFMDAnalysisTaskGenerateCorrection, 1);

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