[u/mrichter/AliRoot.git] / PWG2 / RESONANCES / AliRsnAnalysis.h

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

//-------------------------------------------------------------------------
//                      Class AliRsnAnalysis
//             Reconstruction and analysis of K* Rsn
// ........................................
// ........................................
// ........................................
// ........................................
// 
// author: A. Pulvirenti             (email: alberto.pulvirenti@ct.infn.it)
//-------------------------------------------------------------------------

#ifndef AliRsnANALYSIS_H
#define AliRsnANALYSIS_H

#include <Htypes.h>
#include <TNamed.h>

#include "AliPID.h"

class TH1D;
class TObjArray;
class AliRsnEvent;
class AliRsnDaughter;
class AliRsnDaughterCut;

class AliRsnAnalysis : public TObject
{
public:

	         AliRsnAnalysis();
			 AliRsnAnalysis(const AliRsnAnalysis &copy);
			 AliRsnAnalysis& operator=(const AliRsnAnalysis & /*copy*/) { return (*this); }
	virtual ~AliRsnAnalysis() {Clear();}
	
	void     AddCutPair(AliRsnDaughterCutPair *cut);
	void     AddCutSingle(AliPID::EParticleType type, AliRsnDaughterCut *cut);
	void     AddMixPairDef(AliPID::EParticleType p1, Char_t s1, AliPID::EParticleType p2, Char_t s2);
	void     AddPairDef(AliPID::EParticleType p1, Char_t s1, AliPID::EParticleType p2, Char_t s2, Bool_t onlyTrue = kFALSE);
	void     Clear(Option_t *option = "");	
	Stat_t   EventMix(Int_t nmix = 5, Int_t multDiffMax = 10, Double_t vzDiffMax = 0.01, Bool_t compareTotalMult = kFALSE);
	Stat_t   Process();
	void     SetBins(Int_t nbins, Double_t min, Double_t max) {fNBins=nbins;fHistoMin=min;fHistoMax=max;}
	void     SetEventsTree(TTree *tree)                       {fEventsTree = tree;}
	void     SetRejectFakes(Bool_t doit=kTRUE)                {fRejectFakes = doit;}		
	void     SetTrueMotherPDG(Int_t pdg)                      {fTrueMotherPDG = pdg;}
	void     WriteHistograms() const;
	
private:

	class AliPairDef : public TNamed
	{
	public:
	
		AliPairDef(AliPID::EParticleType p1, Char_t s1, AliPID::EParticleType p2, Char_t s2, Int_t pdg, Bool_t onlyTrue = kFALSE);
				   
		virtual ~AliPairDef()                 { }
		
		Char_t                 GetSign1() const {return fSign1;}
		Char_t                 GetSign2() const {return fSign2;}
		Bool_t                 GetOnlyTrue() const {return fOnlyTrue;}
		AliPID::EParticleType  GetParticle1() const {return fParticle1;}
		AliPID::EParticleType  GetParticle2() const {return fParticle2;}
		Double_t               GetMass1() const {return fMass1;}
		Double_t               GetMass2() const {return fMass2;}
				
	    void SetSign1(Char_t value)                 {fSign1 = value;}
		void SetSign2(Char_t value)                 {fSign2 = value;}
		void SetOnlyTrue(Bool_t value = kTRUE)      {fOnlyTrue = value;}
		void SetTrueMotherPDG(Int_t pdg)            {fTrueMotherPDG = pdg;}
		void SetParticle1(AliPID::EParticleType p)  {fParticle1 = p;}
		void SetParticle2(AliPID::EParticleType p)  {fParticle2 = p;}
		
		Text_t* ParticleName(AliPID::EParticleType part) const;
		
	private:
	
		Bool_t                 fOnlyTrue;	   // flag to be used for spectra of true pairs
		Int_t                  fTrueMotherPDG; // PDG code of true mother (if requested)
		
		Double_t               fMass1;     // info
		Char_t                 fSign1;     // about
		AliPID::EParticleType  fParticle1; // particle 1
	
		Double_t               fMass2;     // info
		Char_t                 fSign2;     // about
		AliPID::EParticleType  fParticle2; // particle 2
	};
	
	Stat_t     Compute(AliPairDef *pd, TH1D* &h, AliRsnEvent *ev1, AliRsnEvent *ev2);
	Bool_t     SingleCutCheck(Int_t itype, AliRsnDaughter *track) const;
	Bool_t     PairCutCheck(AliRsnDaughter *track1, AliRsnDaughter *track2) const;
	
	Bool_t     fRejectFakes;             // reject particles labeled as fake
	
	Int_t      fNBins;                   // number of histogram bins
	Double_t   fHistoMin;                // minimum of the histograms
	Double_t   fHistoMax;                // maximum of the histograms
	
	Int_t      fTrueMotherPDG;           // PDG code of true mother (used to create 'true' histos)
	
	TObjArray *fMixPairDefs;             //  list of pair definitions for histograms (event mixing)
	TObjArray *fMixHistograms;           //! list of invmass histograms created (event mixing)
	
	TObjArray *fPairDefs;                //  list of pair definitions for histograms
	TObjArray *fHistograms;              //! list of invmass histograms created

	TObjArray *fCuts[AliPID::kSPECIES];  //! list of single particle cuts for each particle type
	TObjArray *fPairCuts;                //! list of pair cuts
	TTree     *fEventsTree;              //! TTree of events (can not be created here, must be passed)
	
	// Rsn analysis implementation
	ClassDef(AliRsnAnalysis,1)
};

#endif
Commit	Line	Data
2f1637fb	1	/**************************************************************************
	2	* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	3	* See cxx source for full Copyright notice *
	4	**************************************************************************/
	5
	6	//-------------------------------------------------------------------------
	7	// Class AliRsnAnalysis
	8	// Reconstruction and analysis of K* Rsn
2f769150	9	// ........................................
	10	// ........................................
	11	// ........................................
	12	// ........................................
2f1637fb	13	//
	14	// author: A. Pulvirenti (email: alberto.pulvirenti@ct.infn.it)
	15	//-------------------------------------------------------------------------
	16
	17	#ifndef AliRsnANALYSIS_H
	18	#define AliRsnANALYSIS_H
	19
d560b581	20	#include <Htypes.h>
2f1637fb	21	#include <TNamed.h>
	22
	23	#include "AliPID.h"
	24
	25	class TH1D;
	26	class TObjArray;
	27	class AliRsnEvent;
	28	class AliRsnDaughter;
	29	class AliRsnDaughterCut;
	30
	31	class AliRsnAnalysis : public TObject
	32	{
	33	public:
	34
	35	AliRsnAnalysis();
c37c6481	36	AliRsnAnalysis(const AliRsnAnalysis &copy);
2f769150	37	AliRsnAnalysis& operator=(const AliRsnAnalysis & /copy/) { return (*this); }
2f1637fb	38	virtual ~AliRsnAnalysis() {Clear();}
2f1637fb	39
c37c6481	40	void AddCutPair(AliRsnDaughterCutPair *cut);
2f1637fb	41	void AddCutSingle(AliPID::EParticleType type, AliRsnDaughterCut *cut);
	42	void AddMixPairDef(AliPID::EParticleType p1, Char_t s1, AliPID::EParticleType p2, Char_t s2);
	43	void AddPairDef(AliPID::EParticleType p1, Char_t s1, AliPID::EParticleType p2, Char_t s2, Bool_t onlyTrue = kFALSE);
	44	void Clear(Option_t *option = "");
	45	Stat_t EventMix(Int_t nmix = 5, Int_t multDiffMax = 10, Double_t vzDiffMax = 0.01, Bool_t compareTotalMult = kFALSE);
	46	Stat_t Process();
	47	void SetBins(Int_t nbins, Double_t min, Double_t max) {fNBins=nbins;fHistoMin=min;fHistoMax=max;}
	48	void SetEventsTree(TTree *tree) {fEventsTree = tree;}
	49	void SetRejectFakes(Bool_t doit=kTRUE) {fRejectFakes = doit;}
	50	void SetTrueMotherPDG(Int_t pdg) {fTrueMotherPDG = pdg;}
2f769150	51	void WriteHistograms() const;
2f1637fb	52
	53	private:
	54
	55	class AliPairDef : public TNamed
	56	{
	57	public:
	58
	59	AliPairDef(AliPID::EParticleType p1, Char_t s1, AliPID::EParticleType p2, Char_t s2, Int_t pdg, Bool_t onlyTrue = kFALSE);
	60
	61	virtual ~AliPairDef() { }
	62
2f769150	63	Char_t GetSign1() const {return fSign1;}
	64	Char_t GetSign2() const {return fSign2;}
	65	Bool_t GetOnlyTrue() const {return fOnlyTrue;}
	66	AliPID::EParticleType GetParticle1() const {return fParticle1;}
	67	AliPID::EParticleType GetParticle2() const {return fParticle2;}
	68	Double_t GetMass1() const {return fMass1;}
	69	Double_t GetMass2() const {return fMass2;}
2f1637fb	70
	71	void SetSign1(Char_t value) {fSign1 = value;}
	72	void SetSign2(Char_t value) {fSign2 = value;}
	73	void SetOnlyTrue(Bool_t value = kTRUE) {fOnlyTrue = value;}
	74	void SetTrueMotherPDG(Int_t pdg) {fTrueMotherPDG = pdg;}
	75	void SetParticle1(AliPID::EParticleType p) {fParticle1 = p;}
	76	void SetParticle2(AliPID::EParticleType p) {fParticle2 = p;}
	77
2f769150	78	Text_t* ParticleName(AliPID::EParticleType part) const;
2f1637fb	79
	80	private:
	81
	82	Bool_t fOnlyTrue; // flag to be used for spectra of true pairs
	83	Int_t fTrueMotherPDG; // PDG code of true mother (if requested)
	84
2f769150	85	Double_t fMass1; // info
	86	Char_t fSign1; // about
	87	AliPID::EParticleType fParticle1; // particle 1
2f1637fb	88
2f769150	89	Double_t fMass2; // info
	90	Char_t fSign2; // about
	91	AliPID::EParticleType fParticle2; // particle 2
2f1637fb	92	};
	93
	94	Stat_t Compute(AliPairDef pd, TH1D &h, AliRsnEvent ev1, AliRsnEvent ev2);
2f769150	95	Bool_t SingleCutCheck(Int_t itype, AliRsnDaughter *track) const;
2f769150	96	Bool_t PairCutCheck(AliRsnDaughter track1, AliRsnDaughter track2) const;
2f1637fb	97
	98	Bool_t fRejectFakes; // reject particles labeled as fake
	99
	100	Int_t fNBins; // number of histogram bins
	101	Double_t fHistoMin; // minimum of the histograms
	102	Double_t fHistoMax; // maximum of the histograms
	103
	104	Int_t fTrueMotherPDG; // PDG code of true mother (used to create 'true' histos)
	105
	106	TObjArray *fMixPairDefs; // list of pair definitions for histograms (event mixing)
	107	TObjArray *fMixHistograms; //! list of invmass histograms created (event mixing)
	108
	109	TObjArray *fPairDefs; // list of pair definitions for histograms
	110	TObjArray *fHistograms; //! list of invmass histograms created
	111
	112	TObjArray *fCuts[AliPID::kSPECIES]; //! list of single particle cuts for each particle type
	113	TObjArray *fPairCuts; //! list of pair cuts
	114	TTree *fEventsTree; //! TTree of events (can not be created here, must be passed)
	115
	116	// Rsn analysis implementation
	117	ClassDef(AliRsnAnalysis,1)
	118	};
	119
	120	#endif