[u/mrichter/AliRoot.git] / PWG4 / JetTasks / AliUEHistograms.h

#ifndef AliUEHistograms_H
#define AliUEHistograms_H

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

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

// encapsulates several AliUEHist objects for a full UE analysis plus additional control histograms

#include "TObject.h"
#include "AliUEHist.h"

class AliVParticle;

class TList;
class TH1F;
class TH2F;
class TH3F;

class AliUEHistograms : public TObject
{
 public:
  AliUEHistograms();
  virtual ~AliUEHistograms();
  
  void Fill(Int_t eventType, AliUEHist::CFStep step, AliVParticle* leading, TList* toward, TList* away, TList* min, TList* max);
  void Fill(AliVParticle* leadingMC, AliVParticle* leadingReco);
  void FillEvent(Int_t eventType, Int_t step);
  void FillTrackingEfficiency(TObjArray* mc, TObjArray* recoPrim, TObjArray* recoAll, Int_t particleType);
  
  void CopyReconstructedData(AliUEHistograms* from);
  
  AliUEHist* GetNumberDensitypT() { return fNumberDensitypT; }
  AliUEHist* GetSumpT() { return fSumpT; }
  AliUEHist* GetNumberDensityPhi() { return fNumberDensityPhi; }
  
  TH2F* GetCorrelationpT()  { return fCorrelationpT; }
  TH2F* GetCorrelationEta() { return fCorrelationEta; }
  TH2F* GetCorrelationPhi() { return fCorrelationPhi; }
  TH2F* GetCorrelationR()   { return fCorrelationR; }
  TH2F* GetCorrelationLeading2Phi() { return fCorrelationLeading2Phi; }
  TH2F* GetCorrelationMultiplicity() { return fCorrelationMultiplicity; }
  
  TH2F* GetEventCount()     { return fEventCount; }
  TH3F* GetEventCountDifferential() { return fEventCountDifferential; }
  TH1F* GetVertexContributors() { return fVertexContributors; }
  
  void Correct(AliUEHistograms* corrections);
  
  void SetEtaRange(Float_t etaMin, Float_t etaMax);
  void SetPtRange(Float_t ptMin, Float_t ptMax);
  void SetContaminationEnhancement(TH1F* hist);
  void SetCombineMinMax(Bool_t flag);
  
  AliUEHistograms(const AliUEHistograms &c);
  AliUEHistograms& operator=(const AliUEHistograms& c);
  virtual void Copy(TObject& c) const;

  virtual Long64_t Merge(TCollection* list);
  
protected:
  void FillRegion(AliUEHist::Region region, AliUEHist::CFStep step, AliVParticle* leading, TList* list, Int_t multiplicity);
  Int_t CountParticles(TList* list, Float_t ptMin);

  AliUEHist* fNumberDensitypT;   // d^2N/dphideta vs pT,lead
  AliUEHist* fSumpT;             // d^2 sum(pT)/dphideta vs pT,lead
  AliUEHist* fNumberDensityPhi;  // d^2N/dphideta vs delta phi,lead (in pT,lead bins)
  
  TH2F* fCorrelationpT;         // pT,lead: true vs reco
  TH2F* fCorrelationEta;        // #eta,lead; true vs reco
  TH2F* fCorrelationPhi;        // #phi,lead; true vs reco
  TH2F* fCorrelationR;          // R = sqrt(delta eta^2 + delta phi^2) (true vs reco) vs pT,lead,MC
  TH2F* fCorrelationLeading2Phi;// delta phi (true vs reco) vs pT,lead,MC
  TH2F* fCorrelationMultiplicity; // number of mc particls vs reco particles (for pT > 0.5 GeV/c)
  
  TH2F* fEventCount;            // event count as function of step, event type (plus additional step -1 for all events without vertex range even in MC)
  TH3F* fEventCountDifferential;// event count as function of leading pT, step, event type
  
  TH1F* fVertexContributors;    // number of contributors to the vertex
  
  ClassDef(AliUEHistograms, 1)  // underlying event histogram container
};

#endif
Commit	Line	Data
a75aacd6	1	#ifndef AliUEHistograms_H
	2	#define AliUEHistograms_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: AliUEHistograms.h 20164 2007-08-14 15:31:50Z morsch $ */
	8
	9	// encapsulates several AliUEHist objects for a full UE analysis plus additional control histograms
	10
	11	#include "TObject.h"
	12	#include "AliUEHist.h"
	13
	14	class AliVParticle;
	15
	16	class TList;
	17	class TH1F;
	18	class TH2F;
	19	class TH3F;
	20
	21	class AliUEHistograms : public TObject
	22	{
	23	public:
	24	AliUEHistograms();
	25	virtual ~AliUEHistograms();
	26
	27	void Fill(Int_t eventType, AliUEHist::CFStep step, AliVParticle* leading, TList* toward, TList* away, TList* min, TList* max);
	28	void Fill(AliVParticle* leadingMC, AliVParticle* leadingReco);
	29	void FillEvent(Int_t eventType, Int_t step);
b1831bcb	30	void FillTrackingEfficiency(TObjArray* mc, TObjArray* recoPrim, TObjArray* recoAll, Int_t particleType);
	31
	32	void CopyReconstructedData(AliUEHistograms* from);
a75aacd6	33
	34	AliUEHist* GetNumberDensitypT() { return fNumberDensitypT; }
	35	AliUEHist* GetSumpT() { return fSumpT; }
	36	AliUEHist* GetNumberDensityPhi() { return fNumberDensityPhi; }
	37
	38	TH2F* GetCorrelationpT() { return fCorrelationpT; }
	39	TH2F* GetCorrelationEta() { return fCorrelationEta; }
	40	TH2F* GetCorrelationPhi() { return fCorrelationPhi; }
	41	TH2F* GetCorrelationR() { return fCorrelationR; }
	42	TH2F* GetCorrelationLeading2Phi() { return fCorrelationLeading2Phi; }
	43	TH2F* GetCorrelationMultiplicity() { return fCorrelationMultiplicity; }
	44
	45	TH2F* GetEventCount() { return fEventCount; }
	46	TH3F* GetEventCountDifferential() { return fEventCountDifferential; }
	47	TH1F* GetVertexContributors() { return fVertexContributors; }
	48
	49	void Correct(AliUEHistograms* corrections);
	50
	51	void SetEtaRange(Float_t etaMin, Float_t etaMax);
	52	void SetPtRange(Float_t ptMin, Float_t ptMax);
144bd037	53	void SetContaminationEnhancement(TH1F* hist);
a75aacd6	54	void SetCombineMinMax(Bool_t flag);
a75aacd6	55
d1c75d06	56	AliUEHistograms(const AliUEHistograms &c);
a75aacd6	57	AliUEHistograms& operator=(const AliUEHistograms& c);
	58	virtual void Copy(TObject& c) const;
	59
	60	virtual Long64_t Merge(TCollection* list);
	61
	62	protected:
	63	void FillRegion(AliUEHist::Region region, AliUEHist::CFStep step, AliVParticle* leading, TList* list, Int_t multiplicity);
	64	Int_t CountParticles(TList* list, Float_t ptMin);
	65
	66	AliUEHist* fNumberDensitypT; // d^2N/dphideta vs pT,lead
	67	AliUEHist* fSumpT; // d^2 sum(pT)/dphideta vs pT,lead
	68	AliUEHist* fNumberDensityPhi; // d^2N/dphideta vs delta phi,lead (in pT,lead bins)
	69
	70	TH2F* fCorrelationpT; // pT,lead: true vs reco
	71	TH2F* fCorrelationEta; // #eta,lead; true vs reco
	72	TH2F* fCorrelationPhi; // #phi,lead; true vs reco
	73	TH2F* fCorrelationR; // R = sqrt(delta eta^2 + delta phi^2) (true vs reco) vs pT,lead,MC
	74	TH2F* fCorrelationLeading2Phi;// delta phi (true vs reco) vs pT,lead,MC
	75	TH2F* fCorrelationMultiplicity; // number of mc particls vs reco particles (for pT > 0.5 GeV/c)
	76
	77	TH2F* fEventCount; // event count as function of step, event type (plus additional step -1 for all events without vertex range even in MC)
	78	TH3F* fEventCountDifferential;// event count as function of leading pT, step, event type
	79
	80	TH1F* fVertexContributors; // number of contributors to the vertex
	81
	82	ClassDef(AliUEHistograms, 1) // underlying event histogram container
	83	};
	84
	85	#endif