[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / IdentifiedHighPt / lib / AliHighPtDeDxSpectra.h

#ifndef ALIHIGHPTDEDXSPECTRA_H
#define ALIHIGHPTDEDXSPECTRA_H

#include <TNamed.h>
#include <TH1.h>

#include "AliHighPtDeDxData.h"
#include "AliHighPtDeDxMc.h"

class AliHighPtDeDxSpectra : public TNamed {
 public:
  AliHighPtDeDxSpectra(); // default constructor  
  AliHighPtDeDxSpectra(const char* name, const char* title); // named constructor  
  virtual ~AliHighPtDeDxSpectra(); // default destructor

  void GetCorrectedSpectra(AliHighPtDeDxData* data,
			   AliHighPtDeDxMc* mc);

  void SetUseMcNoVtxCorrection(Bool_t value) { fUseMcNoVtxCorrection = value; };
  void SetUseFittedEfficiency(Bool_t value) { fUseFittedEfficiency = value; };
  void SetUseBinCorrection(Bool_t value) { fUseBinCorrection = value; };
  void SetDebugLevel(Int_t value)  { fDebugLevel = value; };

  Double_t GetNevents()        { return fNevents; };
  TH1D* GetHistPt()            { return hPt; };
  TH1D* GetHistNevents()       { return hNevents; };
  TH1D* GetHistTriggerEfficiency() { return hTriggerEfficiency; };
  TH1D* GetHistEfficiency()    { return hEfficiency; };
  TH1D* GetHistBinCorrection() { return hBinCorrection; };
  TH1D* GetHistMcNoVtxCorrection() { return hMcNoVtxCorrection; };
  TH1D* GetHistMeanPt()        { return hMeanPt; };
  TF1*  GetFuncEfficiency()    { return fEfficiency; };
  TF1*  GetFuncBinFit()        { return fBinFit; };


 private:

  TH1D* ConstructBinCorrection(TH1D* histPt, TProfile* histMeanPt);
  TH1D* ConstructTriggerEfficiency(AliHighPtDeDxMc* mc);
  TH1D* GetEventCount(AliHighPtDeDxData* data, AliHighPtDeDxMc* mc);
  TH1D* ConstructTrackCorrection(AliHighPtDeDxMc* mc);
  TF1*  FitEfficiency(TH1D* histEff);
  void NormalizePt(TH1* hist);

  // members
  Int_t    fDebugLevel;     // debug level
  Bool_t   fUseMcNoVtxCorrection; 
  Bool_t   fUseFittedEfficiency; 
  Bool_t   fUseBinCorrection; 
  // use MC to correct down the novtx fraction outside the vtx cut
  // see AliHighPtDeDxMc::FillEventInfo() for details
  Double_t fNevents;        // corrected number of events

  // histograms
  TH1D* hPt;                // pt spectrum for 
  TH1D* hNevents;           // events
  TH1D* hTriggerEfficiency; // trigger efficency
  TH1D* hEfficiency;        // track correction
  TH1D* hMcNoVtxCorrection; // mc no vtx correction
  TH1D* hBinCorrection;     // bin correction
  TH1D* hMeanPt;            // mean pt of data
  TF1*  fEfficiency;        // fitted efficiency
  TF1*  fBinFit;            // fit used for bin correction
  
  ClassDef(AliHighPtDeDxSpectra, 1)  // AliHighPtDeDxSpectra information
    };

#endif
Commit	Line	Data
4ebdd20e	1	#ifndef ALIHIGHPTDEDXSPECTRA_H
	2	#define ALIHIGHPTDEDXSPECTRA_H
	3
	4	#include <TNamed.h>
	5	#include <TH1.h>
	6
	7	#include "AliHighPtDeDxData.h"
	8	#include "AliHighPtDeDxMc.h"
	9
	10	class AliHighPtDeDxSpectra : public TNamed {
	11	public:
	12	AliHighPtDeDxSpectra(); // default constructor
	13	AliHighPtDeDxSpectra(const char* name, const char* title); // named constructor
	14	virtual ~AliHighPtDeDxSpectra(); // default destructor
	15
	16	void GetCorrectedSpectra(AliHighPtDeDxData* data,
	17	AliHighPtDeDxMc* mc);
	18
	19	void SetUseMcNoVtxCorrection(Bool_t value) { fUseMcNoVtxCorrection = value; };
	20	void SetUseFittedEfficiency(Bool_t value) { fUseFittedEfficiency = value; };
	21	void SetUseBinCorrection(Bool_t value) { fUseBinCorrection = value; };
	22	void SetDebugLevel(Int_t value) { fDebugLevel = value; };
	23
	24	Double_t GetNevents() { return fNevents; };
	25	TH1D* GetHistPt() { return hPt; };
	26	TH1D* GetHistNevents() { return hNevents; };
	27	TH1D* GetHistTriggerEfficiency() { return hTriggerEfficiency; };
	28	TH1D* GetHistEfficiency() { return hEfficiency; };
	29	TH1D* GetHistBinCorrection() { return hBinCorrection; };
	30	TH1D* GetHistMcNoVtxCorrection() { return hMcNoVtxCorrection; };
	31	TH1D* GetHistMeanPt() { return hMeanPt; };
	32	TF1* GetFuncEfficiency() { return fEfficiency; };
	33	TF1* GetFuncBinFit() { return fBinFit; };
	34
	35
	36	private:
	37
	38	TH1D* ConstructBinCorrection(TH1D* histPt, TProfile* histMeanPt);
	39	TH1D* ConstructTriggerEfficiency(AliHighPtDeDxMc* mc);
	40	TH1D* GetEventCount(AliHighPtDeDxData* data, AliHighPtDeDxMc* mc);
	41	TH1D* ConstructTrackCorrection(AliHighPtDeDxMc* mc);
	42	TF1* FitEfficiency(TH1D* histEff);
	43	void NormalizePt(TH1* hist);
	44
	45	// members
	46	Int_t fDebugLevel; // debug level
	47	Bool_t fUseMcNoVtxCorrection;
	48	Bool_t fUseFittedEfficiency;
	49	Bool_t fUseBinCorrection;
	50	// use MC to correct down the novtx fraction outside the vtx cut
	51	// see AliHighPtDeDxMc::FillEventInfo() for details
	52	Double_t fNevents; // corrected number of events
	53
	54	// histograms
	55	TH1D* hPt; // pt spectrum for
	56	TH1D* hNevents; // events
	57	TH1D* hTriggerEfficiency; // trigger efficency
	58	TH1D* hEfficiency; // track correction
	59	TH1D* hMcNoVtxCorrection; // mc no vtx correction
	60	TH1D* hBinCorrection; // bin correction
	61	TH1D* hMeanPt; // mean pt of data
	62	TF1* fEfficiency; // fitted efficiency
	63	TF1* fBinFit; // fit used for bin correction
	64
65	ClassDef(AliHighPtDeDxSpectra, 1) // AliHighPtDeDxSpectra information
66	};
67
68	#endif
69