[u/mrichter/AliRoot.git] / FMD / analysis / AliFMDAnalysisTaskGenerateBackground.h

#ifndef ALIFMDANALYSISTASKGENERATEBACKGROUND_H
#define ALIFMDANALYSISTASKGENERATEBACKGROUND_H

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

/**
 * Make a background distribution from simulated data
 * @ingroup FMD_ana
 * 
 * 
 */

class AliFMDAnaCalibBackgroundCorrection;

class AliFMDAnalysisTaskGenerateBackground : public AliAnalysisTaskSE
{
 public:
  AliFMDAnalysisTaskGenerateBackground();
    AliFMDAnalysisTaskGenerateBackground(const char* name);
    ~AliFMDAnalysisTaskGenerateBackground() {;}
 AliFMDAnalysisTaskGenerateBackground(const AliFMDAnalysisTaskGenerateBackground& 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)
      {}
    AliFMDAnalysisTaskGenerateBackground& operator=(const AliFMDAnalysisTaskGenerateBackground&) { 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;
    ClassDef(AliFMDAnalysisTaskGenerateBackground, 1);

};
#endif
// Local Variables:
//   mode: C++ 
// End:
Commit	Line	Data
2a667047	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
cfe59e45	9	/**
	10	* Make a background distribution from simulated data
	11	* @ingroup FMD_ana
	12	*
	13	*
	14	*/
	15
2a667047	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),
41bad769	30	fHitsByStrip(o.fHitsByStrip),
2a667047	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;
f58a4769	55	AliFMDFloatMap fHitsByStrip;
2a667047	56	Float_t fZvtxCut;
	57	Int_t fNvtxBins;
	58	Int_t fNbinsEta;
	59	AliFMDAnaCalibBackgroundCorrection* fBackground;
	60	ClassDef(AliFMDAnalysisTaskGenerateBackground, 1);
	61
	62	};
	63	#endif
cfe59e45	64	// Local Variables:
	65	// mode: C++
	66	// End: