[u/mrichter/AliRoot.git] / PWGCF / Correlations / Base / AliUEHist.h

#ifndef AliUEHist_H
#define AliUEHist_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id: AliUEHist.h 20164 2007-08-14 15:31:50Z morsch $ */

// encapsulate histogram and corrections for one underlying event histogram

#include "TObject.h"
#include "TString.h"

class AliCFContainer;
class TH1;
class TH1F;
class TH3;
class TH3F;
class TH1D;
class TH2;
class TH2D;
class TCollection;
class AliCFGridSparse;
class THnSparse;
class THnBase;

class AliUEHist : public TObject
{
 public:
  AliUEHist(const char* reqHist = "", const char* binning = 0);
  virtual ~AliUEHist();
  
  const UInt_t fkRegions;
  enum Region { kToward = 0, kAway, kMin, kMax };
  
  static const Int_t fgkCFSteps;
  enum CFStep { kCFStepAll = 0, kCFStepTriggered, kCFStepVertex, kCFStepAnaTopology, kCFStepTrackedOnlyPrim, kCFStepTracked, kCFStepReconstructed, kCFStepRealLeading, kCFStepBiasStudy, kCFStepBiasStudy2, kCFStepCorrected };
  
  const char* GetRegionTitle(Region region);
  const char* GetStepTitle(CFStep step);
  
  AliCFContainer* GetTrackHist(Region region) { return fTrackHist[region]; }
  AliCFContainer* GetEventHist() { return fEventHist; }
  AliCFContainer* GetTrackHistEfficiency()     { return fTrackHistEfficiency; }
  TH3F* GetMCRecoPtCorrelation() { return fFakePt; } 
 
  void SetTrackHist(Region region, AliCFContainer* hist) { fTrackHist[region] = hist; }
  void SetEventHist(AliCFContainer* hist) { fEventHist = hist; }
  void SetTrackHistEfficiency(AliCFContainer* hist) { fTrackHistEfficiency = hist; }
  
  void CopyReconstructedData(AliUEHist* from);
  void DeepCopy(AliUEHist* from);
  
  TH1* GetUEHist(CFStep step, Region region, Float_t ptLeadMin = -1, Float_t ptLeadMax = -1, Int_t multBinBegin = 0, Int_t multBinEnd = -1, Int_t twoD = 0, Bool_t etaNorm = kTRUE, Long64_t* normEvents = 0);
  TH1* GetPtHist(CFStep step, Region region, Float_t ptLeadMin, Float_t ptLeadMax, Int_t multBinBegin, Int_t multBinEnd, Float_t phiMin, Float_t phiMax, Float_t etaMin, Float_t etaMax, Bool_t skipPhiNormalization = kFALSE);
  TH2* GetSumOfRatios(AliUEHist* mixed, CFStep step, Region region, Float_t ptLeadMin, Float_t ptLeadMax, Int_t multBinBegin, Int_t multBinEnd, Bool_t etaNorm = kTRUE, Bool_t useVertexBins = kFALSE);
  
  void GetHistsZVtx(AliUEHist::CFStep step, AliUEHist::Region region, Float_t ptLeadMin, Float_t ptLeadMax, Int_t multBinBegin, Int_t multBinEnd, TH3** trackHist, TH1** eventHist);
  void GetHistsZVtxMult(AliUEHist::CFStep step, AliUEHist::Region region, Float_t ptLeadMin, Float_t ptLeadMax, THnBase** trackHist, TH2** eventHist);
  
  TH2* GetSumOfRatios2(AliUEHist* mixed, AliUEHist::CFStep step, AliUEHist::Region region, Float_t ptLeadMin, Float_t ptLeadMax, Int_t multBinBegin, Int_t multBinEnd, Bool_t normalizePerTrigger = kTRUE, Int_t stepForMixed = -1);
  
  TH1* GetTriggersAsFunctionOfMultiplicity(AliUEHist::CFStep step, Float_t ptLeadMin, Float_t ptLeadMax);

  TH1* GetTrackEfficiency(CFStep step1, CFStep step2, Int_t axis1, Int_t axis2 = -1, Int_t source = 1, Int_t axis3 = -1);
  THnBase* GetTrackEfficiencyND(CFStep step1, CFStep step2);
  TH1* GetEventEfficiency(CFStep step1, CFStep step2, Int_t axis1, Int_t axis2 = -1, Float_t ptLeadMin = -1, Float_t ptLeadMax = -1);
  TH1* GetBias(CFStep step1, CFStep step2, Int_t region, const char* axis, Float_t leadPtMin = 0, Float_t leadPtMax = -1, Int_t weighting = 0);
  
  TH1D* GetTrackingEfficiency(Int_t axis);
  TH2D* GetTrackingEfficiency();
  TH2D* GetTrackingEfficiencyCentrality();
  
  TH2D* GetFakeRate();
  TH1D* GetFakeRate(Int_t axis);

  TH1D* GetTrackingContamination(Int_t axis);
  TH2D* GetTrackingContamination();
  TH2D* GetTrackingContaminationCentrality();
  
  TH1D* GetTrackingCorrection(Int_t axis);
  TH2D* GetTrackingCorrection();
  
  TH1D* GetTrackingEfficiencyCorrection(Int_t axis);
  TH2D* GetTrackingEfficiencyCorrection();
  TH2D* GetTrackingEfficiencyCorrectionCentrality();
  
  TH2* GetCorrelatedContamination();

  void ExtendTrackingEfficiency(Bool_t verbose = kFALSE);
  
  void Correct(AliUEHist* corrections);
  void CorrectTracks(CFStep step1, CFStep step2, TH1* trackCorrection, Int_t var1, Int_t var2 = -1);
  void CorrectTracks(CFStep step1, CFStep step2, Int_t region, TH1* trackCorrection, Int_t var1, Int_t var2 = -1);
  void CorrectEvents(CFStep step1, CFStep step2, TH1* eventCorrection, Int_t var1, Int_t var2 = -1);
  void CorrectCorrelatedContamination(CFStep step1, Int_t region, TH1* trackCorrection);
  
  void CondenseBin(THnSparse* grid, THnSparse* target, Int_t axis, Float_t targetValue, Float_t from, Float_t to);
  void CondenseBin(CFStep step, Int_t trackAxis, Int_t eventAxis, Float_t targetValue, Float_t from = 0, Float_t to = -1, CFStep tmpStep = AliUEHist::kCFStepBiasStudy2);
  void SymmetrizepTBins();
  
  void SetCombineMinMax(Bool_t flag) { fCombineMinMax = flag; }
  
  void SetEtaRange(Float_t etaMin, Float_t etaMax) { fEtaMin = etaMin; fEtaMax = etaMax; }
  void SetPtRange(Float_t ptMin, Float_t ptMax)    { fPtMin = ptMin; fPtMax = ptMax; }
  void SetPartSpecies(Int_t species)    { fPartSpecies = species;}
  void SetCentralityRange(Float_t min, Float_t max)    { fCentralityMin = min; fCentralityMax = max; }
  void SetZVtxRange(Float_t min, Float_t max)          { fZVtxMin = min; fZVtxMax = max; }
  void SetPt2Min(Float_t ptMin)			       { fPt2Min = ptMin; }
  
  Float_t GetTrackEtaCut() { return fTrackEtaCut; }
  void SetTrackEtaCut(Float_t value) { fTrackEtaCut = value; }
  void SetWeightPerEvent(Bool_t flag)   { fWeightPerEvent = flag; }
  void SetSkipScaleMixedEvent(Bool_t flag)  { fSkipScaleMixedEvent = flag; }
  
  void SetContaminationEnhancement(TH1F* hist)    { fContaminationEnhancement = hist; }
  
  void SetHistogramType(const char* histogramType)  { fHistogramType = histogramType; }
  
  void CountEmptyBins(AliUEHist::CFStep step, Float_t ptLeadMin, Float_t ptLeadMax);
  
  void AdditionalDPhiCorrection(Int_t step);
  
  void SetBinLimits(AliCFGridSparse* grid);
  void SetBinLimits(THnBase* grid);

  void ResetBinLimits(AliCFGridSparse* grid);
  void ResetBinLimits(THnBase* grid);
  
  void SetGetMultCache(Bool_t flag = kTRUE) { fGetMultCacheOn = flag; }
  
  AliUEHist(const AliUEHist &c);
  AliUEHist& operator=(const AliUEHist& corr);
  virtual void Copy(TObject& c) const;

  virtual Long64_t Merge(TCollection* list);
  void Scale(Double_t factor);
  void Reset();
  THnBase* ChangeToThn(THnBase* sparse);
  
  static TString CombineBinning(TString defaultBinning, TString customBinning);
  
protected:
  Double_t* GetBinning(const char* configuration, const char* tag, Int_t& nBins);
  void SetStepNames(AliCFContainer* container);
  void WeightHistogram(TH3* hist1, TH1* hist2);
  void MultiplyHistograms(THnSparse* grid, THnSparse* target, TH1* histogram, Int_t var1, Int_t var2);

  AliCFContainer* fTrackHist[4];      // container for track level distributions in four regions (toward, away, min, max) and at all analysis steps
  AliCFContainer* fEventHist;         // container for event level distribution at all analysis steps
  AliCFContainer* fTrackHistEfficiency; // container for tracking efficiency and contamination (all particles filled including leading one): axes: eta, pT, particle species
  TH3F* fFakePt;
 
  Float_t fEtaMin;                    // eta min for projections
  Float_t fEtaMax;                    // eta max for projections
  Float_t fPtMin;                     // pT min for projections (for track pT, not pT,lead)
  Float_t fPtMax;                     // pT max for projections (for track pT, not pT,lead)
  Int_t fPartSpecies;                   // Particle species for projections 
  Float_t fCentralityMin;             // centrality min for projections
  Float_t fCentralityMax;             // centrality max for projections
  Float_t fZVtxMin;                   // z vtx min for projections
  Float_t fZVtxMax;                   // z vtx max for projections
  Float_t fPt2Min;		      // pT min for projections (for pT,2 (only 2+1 corr case))
  
  TH1F* fContaminationEnhancement;    // histogram that contains the underestimation of secondaries in the MC as function of pT
  
  Bool_t fCombineMinMax;              // flag to combine min and max to a general towards region
  Float_t fTrackEtaCut;               // cut used during production of histograms (needed for finite bin correction in GetSumOfRatios)
  Bool_t fWeightPerEvent;	      // weight with the number of trigger particles per event
  Bool_t fSkipScaleMixedEvent;        // scale the mixed event with (0, 0) plus finite bin correction (default: kTRUE)
  
  AliCFContainer* fCache;             //! cache variable for GetTrackEfficiency
  
  Bool_t fGetMultCacheOn;             //! cache for GetHistsZVtxMult function active
  THnBase* fGetMultCache;             //! cache for GetHistsZVtxMult function
  
  TString fHistogramType;             // what is stored in this histogram
  
  ClassDef(AliUEHist, 15) // underlying event histogram container
};

#endif
Commit	Line	Data
a75aacd6	1	#ifndef AliUEHist_H
	2	#define AliUEHist_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id: AliUEHist.h 20164 2007-08-14 15:31:50Z morsch $ */
	8
	9	// encapsulate histogram and corrections for one underlying event histogram
	10
	11	#include "TObject.h"
c7245604	12	#include "TString.h"
a75aacd6	13
	14	class AliCFContainer;
	15	class TH1;
144bd037	16	class TH1F;
a75aacd6	17	class TH3;
b591fb9c	18	class TH3F;
a75aacd6	19	class TH1D;
2a910c25	20	class TH2;
a75aacd6	21	class TH2D;
	22	class TCollection;
	23	class AliCFGridSparse;
c32a0ca9	24	class THnSparse;
d4b3dbfc	25	class THnBase;
a75aacd6	26
	27	class AliUEHist : public TObject
	28	{
	29	public:
3f3f12d9	30	AliUEHist(const char* reqHist = "", const char* binning = 0);
a75aacd6	31	virtual ~AliUEHist();
a75aacd6	32
ff458f69	33	const UInt_t fkRegions;
a75aacd6	34	enum Region { kToward = 0, kAway, kMin, kMax };
	35
	36	static const Int_t fgkCFSteps;
408d1ac9	37	enum CFStep { kCFStepAll = 0, kCFStepTriggered, kCFStepVertex, kCFStepAnaTopology, kCFStepTrackedOnlyPrim, kCFStepTracked, kCFStepReconstructed, kCFStepRealLeading, kCFStepBiasStudy, kCFStepBiasStudy2, kCFStepCorrected };
a75aacd6	38
	39	const char* GetRegionTitle(Region region);
	40	const char* GetStepTitle(CFStep step);
	41
	42	AliCFContainer* GetTrackHist(Region region) { return fTrackHist[region]; }
	43	AliCFContainer* GetEventHist() { return fEventHist; }
b1831bcb	44	AliCFContainer* GetTrackHistEfficiency() { return fTrackHistEfficiency; }
b591fb9c	45	TH3F* GetMCRecoPtCorrelation() { return fFakePt; }
b591fb9c	46
a75aacd6	47	void SetTrackHist(Region region, AliCFContainer* hist) { fTrackHist[region] = hist; }
a75aacd6	48	void SetEventHist(AliCFContainer* hist) { fEventHist = hist; }
ada1a03f	49	void SetTrackHistEfficiency(AliCFContainer* hist) { fTrackHistEfficiency = hist; }
a75aacd6	50
b1831bcb	51	void CopyReconstructedData(AliUEHist* from);
0ffdaf17	52	void DeepCopy(AliUEHist* from);
b1831bcb	53
3f3f12d9	54	TH1* GetUEHist(CFStep step, Region region, Float_t ptLeadMin = -1, Float_t ptLeadMax = -1, Int_t multBinBegin = 0, Int_t multBinEnd = -1, Int_t twoD = 0, Bool_t etaNorm = kTRUE, Long64_t* normEvents = 0);
c7245604	55	TH1* GetPtHist(CFStep step, Region region, Float_t ptLeadMin, Float_t ptLeadMax, Int_t multBinBegin, Int_t multBinEnd, Float_t phiMin, Float_t phiMax, Float_t etaMin, Float_t etaMax, Bool_t skipPhiNormalization = kFALSE);
44af28f9	56	TH2* GetSumOfRatios(AliUEHist* mixed, CFStep step, Region region, Float_t ptLeadMin, Float_t ptLeadMax, Int_t multBinBegin, Int_t multBinEnd, Bool_t etaNorm = kTRUE, Bool_t useVertexBins = kFALSE);
52c83678	57
52c83678	58	void GetHistsZVtx(AliUEHist::CFStep step, AliUEHist::Region region, Float_t ptLeadMin, Float_t ptLeadMax, Int_t multBinBegin, Int_t multBinEnd, TH3 trackHist, TH1 eventHist);
d4b3dbfc	59	void GetHistsZVtxMult(AliUEHist::CFStep step, AliUEHist::Region region, Float_t ptLeadMin, Float_t ptLeadMax, THnBase trackHist, TH2 eventHist);
d4b3dbfc	60
c9ae910e	61	TH2* GetSumOfRatios2(AliUEHist* mixed, AliUEHist::CFStep step, AliUEHist::Region region, Float_t ptLeadMin, Float_t ptLeadMax, Int_t multBinBegin, Int_t multBinEnd, Bool_t normalizePerTrigger = kTRUE, Int_t stepForMixed = -1);
b4263273	62
b4263273	63	TH1* GetTriggersAsFunctionOfMultiplicity(AliUEHist::CFStep step, Float_t ptLeadMin, Float_t ptLeadMax);
44af28f9	64
2a910c25	65	TH1* GetTrackEfficiency(CFStep step1, CFStep step2, Int_t axis1, Int_t axis2 = -1, Int_t source = 1, Int_t axis3 = -1);
408d1ac9	66	THnBase* GetTrackEfficiencyND(CFStep step1, CFStep step2);
a75aacd6	67	TH1* GetEventEfficiency(CFStep step1, CFStep step2, Int_t axis1, Int_t axis2 = -1, Float_t ptLeadMin = -1, Float_t ptLeadMax = -1);
c7245604	68	TH1* GetBias(CFStep step1, CFStep step2, Int_t region, const char* axis, Float_t leadPtMin = 0, Float_t leadPtMax = -1, Int_t weighting = 0);
a75aacd6	69
	70	TH1D* GetTrackingEfficiency(Int_t axis);
	71	TH2D* GetTrackingEfficiency();
2a910c25	72	TH2D* GetTrackingEfficiencyCentrality();
a75aacd6	73
b591fb9c	74	TH2D* GetFakeRate();
46848f0b	75	TH1D* GetFakeRate(Int_t axis);
b591fb9c	76
a75aacd6	77	TH1D* GetTrackingContamination(Int_t axis);
a75aacd6	78	TH2D* GetTrackingContamination();
2a910c25	79	TH2D* GetTrackingContaminationCentrality();
a75aacd6	80
	81	TH1D* GetTrackingCorrection(Int_t axis);
	82	TH2D* GetTrackingCorrection();
	83
144bd037	84	TH1D* GetTrackingEfficiencyCorrection(Int_t axis);
144bd037	85	TH2D* GetTrackingEfficiencyCorrection();
2a910c25	86	TH2D* GetTrackingEfficiencyCorrectionCentrality();
	87
	88	TH2* GetCorrelatedContamination();
6f803f6c	89
df269636	90	void ExtendTrackingEfficiency(Bool_t verbose = kFALSE);
144bd037	91
a75aacd6	92	void Correct(AliUEHist* corrections);
a75aacd6	93	void CorrectTracks(CFStep step1, CFStep step2, TH1* trackCorrection, Int_t var1, Int_t var2 = -1);
144bd037	94	void CorrectTracks(CFStep step1, CFStep step2, Int_t region, TH1* trackCorrection, Int_t var1, Int_t var2 = -1);
2a910c25	95	void CorrectEvents(CFStep step1, CFStep step2, TH1* eventCorrection, Int_t var1, Int_t var2 = -1);
2a910c25	96	void CorrectCorrelatedContamination(CFStep step1, Int_t region, TH1* trackCorrection);
a75aacd6	97
9894bedd	98	void CondenseBin(THnSparse* grid, THnSparse* target, Int_t axis, Float_t targetValue, Float_t from, Float_t to);
9894bedd	99	void CondenseBin(CFStep step, Int_t trackAxis, Int_t eventAxis, Float_t targetValue, Float_t from = 0, Float_t to = -1, CFStep tmpStep = AliUEHist::kCFStepBiasStudy2);
4da05e3e	100	void SymmetrizepTBins();
a5d12d24	101
a75aacd6	102	void SetCombineMinMax(Bool_t flag) { fCombineMinMax = flag; }
	103
	104	void SetEtaRange(Float_t etaMin, Float_t etaMax) { fEtaMin = etaMin; fEtaMax = etaMax; }
	105	void SetPtRange(Float_t ptMin, Float_t ptMax) { fPtMin = ptMin; fPtMax = ptMax; }
c824c30e	106	void SetPartSpecies(Int_t species) { fPartSpecies = species;}
2a910c25	107	void SetCentralityRange(Float_t min, Float_t max) { fCentralityMin = min; fCentralityMax = max; }
85bfac17	108	void SetZVtxRange(Float_t min, Float_t max) { fZVtxMin = min; fZVtxMax = max; }
c9ae910e	109	void SetPt2Min(Float_t ptMin) { fPt2Min = ptMin; }
a75aacd6	110
b4263273	111	Float_t GetTrackEtaCut() { return fTrackEtaCut; }
defad170	112	void SetTrackEtaCut(Float_t value) { fTrackEtaCut = value; }
20b7a005	113	void SetWeightPerEvent(Bool_t flag) { fWeightPerEvent = flag; }
7e9608f2	114	void SetSkipScaleMixedEvent(Bool_t flag) { fSkipScaleMixedEvent = flag; }
defad170	115
144bd037	116	void SetContaminationEnhancement(TH1F* hist) { fContaminationEnhancement = hist; }
144bd037	117
2a910c25	118	void SetHistogramType(const char* histogramType) { fHistogramType = histogramType; }
2a910c25	119
144bd037	120	void CountEmptyBins(AliUEHist::CFStep step, Float_t ptLeadMin, Float_t ptLeadMax);
144bd037	121
e741fadd	122	void AdditionalDPhiCorrection(Int_t step);
e741fadd	123
ada1a03f	124	void SetBinLimits(AliCFGridSparse* grid);
d4b3dbfc	125	void SetBinLimits(THnBase* grid);
d4b3dbfc	126
ada1a03f	127	void ResetBinLimits(AliCFGridSparse* grid);
d4b3dbfc	128	void ResetBinLimits(THnBase* grid);
	129
	130	void SetGetMultCache(Bool_t flag = kTRUE) { fGetMultCacheOn = flag; }
ada1a03f	131
d1c75d06	132	AliUEHist(const AliUEHist &c);
a75aacd6	133	AliUEHist& operator=(const AliUEHist& corr);
	134	virtual void Copy(TObject& c) const;
	135
	136	virtual Long64_t Merge(TCollection* list);
c7245604	137	void Scale(Double_t factor);
c7245604	138	void Reset();
d4b3dbfc	139	THnBase* ChangeToThn(THnBase* sparse);
a75aacd6	140
c9ae910e	141	static TString CombineBinning(TString defaultBinning, TString customBinning);
c9ae910e	142
a75aacd6	143	protected:
3f3f12d9	144	Double_t* GetBinning(const char* configuration, const char* tag, Int_t& nBins);
a75aacd6	145	void SetStepNames(AliCFContainer* container);
a75aacd6	146	void WeightHistogram(TH3* hist1, TH1* hist2);
c32a0ca9	147	void MultiplyHistograms(THnSparse* grid, THnSparse* target, TH1* histogram, Int_t var1, Int_t var2);
a75aacd6	148
e93687b1 JFGO	149	AliCFContainer* fTrackHist[4]; // container for track level distributions in four regions (toward, away, min, max) and at all analysis steps
e93687b1 JFGO	150	AliCFContainer* fEventHist; // container for event level distribution at all analysis steps
b1831bcb	151	AliCFContainer* fTrackHistEfficiency; // container for tracking efficiency and contamination (all particles filled including leading one): axes: eta, pT, particle species
b591fb9c	152	TH3F* fFakePt;
b591fb9c	153
a75aacd6	154	Float_t fEtaMin; // eta min for projections
	155	Float_t fEtaMax; // eta max for projections
	156	Float_t fPtMin; // pT min for projections (for track pT, not pT,lead)
	157	Float_t fPtMax; // pT max for projections (for track pT, not pT,lead)
c824c30e	158	Int_t fPartSpecies; // Particle species for projections
2a910c25	159	Float_t fCentralityMin; // centrality min for projections
2a910c25	160	Float_t fCentralityMax; // centrality max for projections
85bfac17	161	Float_t fZVtxMin; // z vtx min for projections
85bfac17	162	Float_t fZVtxMax; // z vtx max for projections
c9ae910e	163	Float_t fPt2Min; // pT min for projections (for pT,2 (only 2+1 corr case))
a75aacd6	164
144bd037	165	TH1F* fContaminationEnhancement; // histogram that contains the underestimation of secondaries in the MC as function of pT
144bd037	166
a75aacd6	167	Bool_t fCombineMinMax; // flag to combine min and max to a general towards region
defad170	168	Float_t fTrackEtaCut; // cut used during production of histograms (needed for finite bin correction in GetSumOfRatios)
7e9608f2	169	Bool_t fWeightPerEvent; // weight with the number of trigger particles per event
7e9608f2	170	Bool_t fSkipScaleMixedEvent; // scale the mixed event with (0, 0) plus finite bin correction (default: kTRUE)
a75aacd6	171
	172	AliCFContainer* fCache; //! cache variable for GetTrackEfficiency
	173
d4b3dbfc	174	Bool_t fGetMultCacheOn; //! cache for GetHistsZVtxMult function active
	175	THnBase* fGetMultCache; //! cache for GetHistsZVtxMult function
	176
c7245604	177	TString fHistogramType; // what is stored in this histogram
c7245604	178
c9ae910e	179	ClassDef(AliUEHist, 15) // underlying event histogram container
a75aacd6	180	};
	181
	182	#endif