[u/mrichter/AliRoot.git] / PWG4 / PartCorrBase / AliNeutralMesonSelection.h

#ifndef ALINEUTRALMESONSELECTION_H
#define ALINEUTRALMESONSELECTION_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice     */
/* $Id: AliNeutralMesonSelection.h 27413 2008-07-18 13:28:12Z gconesab $ */

//_________________________________________________________________________
// Class that contains methods to select candidate pairs to neutral meson 
// 2 main selections, invariant mass around pi0 (also any other mass),
// apperture angle to distinguish from combinatorial.
// There is a 3rd cut based on the gamma correlation on phi or pt.
//-- Author: Gustavo Conesa (INFN-LNF)

// --- ROOT system ---
#include<TObject.h>
#include<TArrayD.h>

class TLorentzVector ;
class TList ;
class TH2F ;

//--- ANALYSIS system ---
class AliLog ;

class AliNeutralMesonSelection : public TObject {
	
public: 
	
	AliNeutralMesonSelection() ; // default ctor
	AliNeutralMesonSelection(const AliNeutralMesonSelection & g) ; // cpy ctor
	AliNeutralMesonSelection & operator = (const AliNeutralMesonSelection & g) ;//cpy assignment
	virtual ~AliNeutralMesonSelection() ; //virtual dtor
	
	TList * GetCreateOutputObjects();
	
	Double_t GetAngleMaxParam(Int_t i) const {return fAngleMaxParam.At(i) ; }
	void SetAngleMaxParam(Int_t i, Double_t par){fAngleMaxParam.AddAt(par,i) ; }
	
	Double_t GetInvMassMaxCut() const {return fInvMassMaxCut ; }
	Double_t GetInvMassMinCut() const {return fInvMassMinCut ; }
	void SetInvMassCutRange(Double_t invmassmin, Double_t invmassmax)
	{fInvMassMaxCut =invmassmax;  fInvMassMinCut =invmassmin;}	
	
	Double_t GetMass() const {return fM ; }
	void SetMass(Double_t m) { fM =m ; }
	
	Bool_t AreNeutralMesonSelectionHistosKept() const { return fKeepNeutralMesonHistos ; }
	void KeepNeutralMesonSelectionHistos(Bool_t keep) { fKeepNeutralMesonHistos = keep ; }
	
	void InitParameters();	
	Bool_t IsAngleInWindow(const Float_t angle, const Float_t e) const ;
	void Print(const Option_t * opt) const;
	
	Bool_t  SelectPair(TLorentzVector particlei,  TLorentzVector particlej)  ;
	
	//Histogrammes setters and getters
	
	virtual void SetHistoERangeAndNBins(Float_t min, Float_t max, Int_t n) {
		fHistoNEBins = n ;
		fHistoEMax = max ;
		fHistoEMin = min ;
	}
	
	Int_t   GetHistoNEBins() const { return fHistoNEBins ; }
	Float_t GetHistoEMin()   const { return fHistoEMin ; }
	Float_t GetHistoEMax()   const { return fHistoEMax ; }

	virtual void SetHistoPtRangeAndNBins(Float_t min, Float_t max, Int_t n) {
		fHistoNPtBins = n ;
		fHistoPtMax = max ;
		fHistoPtMin = min ;
	}
	
	Int_t   GetHistoNPtBins() const { return fHistoNPtBins ; }
	Float_t GetHistoPtMin()   const { return fHistoPtMin ; }
	Float_t GetHistoPtMax()   const { return fHistoPtMax ; }

	virtual void SetHistoAngleRangeAndNBins(Float_t min, Float_t max, Int_t n) {
		fHistoNAngleBins = n ;
		fHistoAngleMax = max ;
		fHistoAngleMin = min ;
	}
	
	Int_t   GetHistoNAngleBins() const { return fHistoNAngleBins ; }
	Float_t GetHistoAngleMin()   const { return fHistoAngleMin ; }
	Float_t GetHistoAngleMax()   const { return fHistoAngleMax ; }

	virtual void SetHistoIMRangeAndNBins(Float_t min, Float_t max, Int_t n) {
		fHistoNIMBins = n ;
		fHistoIMMax = max ;
		fHistoIMMin = min ;
	}
	
	Int_t   GetHistoNIMBins() const { return fHistoNIMBins ; }
	Float_t GetHistoIMMin()   const { return fHistoIMMin ; }
	Float_t GetHistoIMMax()   const { return fHistoIMMax ; }
	
	
private:
	Double_t fM ; //mass of the neutral meson
	Double_t   fInvMassMaxCut ;  // Invariant Mass cut maximum
	Double_t   fInvMassMinCut ;  // Invariant Masscut minimun
	TArrayD    fAngleMaxParam ; //Max opening angle selection parameters
	
	Bool_t  fKeepNeutralMesonHistos ; // Keep neutral meson selection histograms
	
	//Histograms
	TH2F * fhAnglePairNoCut  ;  //Aperture angle of decay photons, no cuts
	TH2F * fhAnglePairOpeningAngleCut   ;  //Aperture angle of decay photons, cut on opening angle
	TH2F * fhAnglePairAllCut   ;  //Aperture angle of decay photons, all cuts
	TH2F * fhInvMassPairNoCut    ;  //Invariant mass of decay photons, no cuts
	TH2F * fhInvMassPairOpeningAngleCut  ;  //Invariant mass of decay photons, cut on opening angle
	TH2F * fhInvMassPairAllCut   ;  //Invariant mass of decay photons, all cuts
	
  	//Histograms binning and range    
	Int_t   fHistoNEBins ;  //Number of bins in pi0 E axis
	Float_t fHistoEMax ;    //Maximum value of pi0 E histogram range
	Float_t fHistoEMin ;    //Minimum value of pi0 E histogram range
	Int_t   fHistoNPtBins ;  //Number of bins in Pt trigger axis
	Float_t fHistoPtMax ;    //Maximum value of Pt trigger histogram range
	Float_t fHistoPtMin ;    //Minimum value of Pt trigger histogram range		
	Int_t   fHistoNAngleBins ; //Number of bins in angle axis
	Float_t fHistoAngleMax ;//Maximum value of angle histogram range
	Float_t fHistoAngleMin ;//Minimum value of angle histogram range
	Int_t   fHistoNIMBins ; //Number of bins in Invariant Mass axis
	Float_t fHistoIMMax ;   //Maximum value of Invariant Mass histogram range
	Float_t fHistoIMMin ;   //Minimum value of Invariant Mass histogram range
	
	ClassDef(AliNeutralMesonSelection,2)
    
} ;


#endif //ALINEUTRALMESONSELECTION_H
Commit	Line	Data
1c5acb87	1	#ifndef ALINEUTRALMESONSELECTION_H
	2	#define ALINEUTRALMESONSELECTION_H
	3	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5	/* $Id: AliNeutralMesonSelection.h 27413 2008-07-18 13:28:12Z gconesab $ */
	6
	7	//_________________________________________________________________________
	8	// Class that contains methods to select candidate pairs to neutral meson
	9	// 2 main selections, invariant mass around pi0 (also any other mass),
	10	// apperture angle to distinguish from combinatorial.
	11	// There is a 3rd cut based on the gamma correlation on phi or pt.
	12	//-- Author: Gustavo Conesa (INFN-LNF)
	13
	14	// --- ROOT system ---
	15	#include<TObject.h>
	16	#include<TArrayD.h>
	17
	18	class TLorentzVector ;
	19	class TList ;
	20	class TH2F ;
	21
	22	//--- ANALYSIS system ---
	23	class AliLog ;
	24
	25	class AliNeutralMesonSelection : public TObject {
	26
	27	public:
	28
	29	AliNeutralMesonSelection() ; // default ctor
	30	AliNeutralMesonSelection(const AliNeutralMesonSelection & g) ; // cpy ctor
	31	AliNeutralMesonSelection & operator = (const AliNeutralMesonSelection & g) ;//cpy assignment
	32	virtual ~AliNeutralMesonSelection() ; //virtual dtor
	33
	34	TList * GetCreateOutputObjects();
	35
	36	Double_t GetAngleMaxParam(Int_t i) const {return fAngleMaxParam.At(i) ; }
	37	void SetAngleMaxParam(Int_t i, Double_t par){fAngleMaxParam.AddAt(par,i) ; }
	38
	39	Double_t GetInvMassMaxCut() const {return fInvMassMaxCut ; }
	40	Double_t GetInvMassMinCut() const {return fInvMassMinCut ; }
	41	void SetInvMassCutRange(Double_t invmassmin, Double_t invmassmax)
	42	{fInvMassMaxCut =invmassmax; fInvMassMinCut =invmassmin;}
	43
	44	Double_t GetMass() const {return fM ; }
	45	void SetMass(Double_t m) { fM =m ; }
	46
	47	Bool_t AreNeutralMesonSelectionHistosKept() const { return fKeepNeutralMesonHistos ; }
	48	void KeepNeutralMesonSelectionHistos(Bool_t keep) { fKeepNeutralMesonHistos = keep ; }
	49
	50	void InitParameters();
	51	Bool_t IsAngleInWindow(const Float_t angle, const Float_t e) const ;
	52	void Print(const Option_t * opt) const;
	53
	54	Bool_t SelectPair(TLorentzVector particlei, TLorentzVector particlej) ;
	55
	56	//Histogrammes setters and getters
	57
	58	virtual void SetHistoERangeAndNBins(Float_t min, Float_t max, Int_t n) {
	59	fHistoNEBins = n ;
	60	fHistoEMax = max ;
	61	fHistoEMin = min ;
	62	}
	63
	64	Int_t GetHistoNEBins() const { return fHistoNEBins ; }
65	Float_t GetHistoEMin() const { return fHistoEMin ; }
66	Float_t GetHistoEMax() const { return fHistoEMax ; }
67
68	virtual void SetHistoPtRangeAndNBins(Float_t min, Float_t max, Int_t n) {
69	fHistoNPtBins = n ;
70	fHistoPtMax = max ;
71	fHistoPtMin = min ;
72	}
73
74	Int_t GetHistoNPtBins() const { return fHistoNPtBins ; }
75	Float_t GetHistoPtMin() const { return fHistoPtMin ; }
76	Float_t GetHistoPtMax() const { return fHistoPtMax ; }
77
78	virtual void SetHistoAngleRangeAndNBins(Float_t min, Float_t max, Int_t n) {
79	fHistoNAngleBins = n ;
80	fHistoAngleMax = max ;
81	fHistoAngleMin = min ;
82	}
83
84	Int_t GetHistoNAngleBins() const { return fHistoNAngleBins ; }
85	Float_t GetHistoAngleMin() const { return fHistoAngleMin ; }
86	Float_t GetHistoAngleMax() const { return fHistoAngleMax ; }
87
88	virtual void SetHistoIMRangeAndNBins(Float_t min, Float_t max, Int_t n) {
89	fHistoNIMBins = n ;
90	fHistoIMMax = max ;
91	fHistoIMMin = min ;
92	}
93
94	Int_t GetHistoNIMBins() const { return fHistoNIMBins ; }
95	Float_t GetHistoIMMin() const { return fHistoIMMin ; }
96	Float_t GetHistoIMMax() const { return fHistoIMMax ; }
97
98
99	private:
100	Double_t fM ; //mass of the neutral meson
101	Double_t fInvMassMaxCut ; // Invariant Mass cut maximum
102	Double_t fInvMassMinCut ; // Invariant Masscut minimun
103	TArrayD fAngleMaxParam ; //Max opening angle selection parameters
104
105	Bool_t fKeepNeutralMesonHistos ; // Keep neutral meson selection histograms
106
107	//Histograms
108	TH2F * fhAnglePairNoCut ; //Aperture angle of decay photons, no cuts
109	TH2F * fhAnglePairOpeningAngleCut ; //Aperture angle of decay photons, cut on opening angle
110	TH2F * fhAnglePairAllCut ; //Aperture angle of decay photons, all cuts
111	TH2F * fhInvMassPairNoCut ; //Invariant mass of decay photons, no cuts
112	TH2F * fhInvMassPairOpeningAngleCut ; //Invariant mass of decay photons, cut on opening angle
113	TH2F * fhInvMassPairAllCut ; //Invariant mass of decay photons, all cuts
114
115	//Histograms binning and range
116	Int_t fHistoNEBins ; //Number of bins in pi0 E axis
117	Float_t fHistoEMax ; //Maximum value of pi0 E histogram range
118	Float_t fHistoEMin ; //Minimum value of pi0 E histogram range
119	Int_t fHistoNPtBins ; //Number of bins in Pt trigger axis
120	Float_t fHistoPtMax ; //Maximum value of Pt trigger histogram range
121	Float_t fHistoPtMin ; //Minimum value of Pt trigger histogram range
122	Int_t fHistoNAngleBins ; //Number of bins in angle axis
123	Float_t fHistoAngleMax ;//Maximum value of angle histogram range
124	Float_t fHistoAngleMin ;//Minimum value of angle histogram range
125	Int_t fHistoNIMBins ; //Number of bins in Invariant Mass axis
126	Float_t fHistoIMMax ; //Maximum value of Invariant Mass histogram range
127	Float_t fHistoIMMin ; //Minimum value of Invariant Mass histogram range
128
129	ClassDef(AliNeutralMesonSelection,2)
130
131	} ;
132
133
134	#endif //ALINEUTRALMESONSELECTION_H
135
136
137