changed the order of call of endofcycle so that images are produced

[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / AliProtonFeedDownAnalysis.h

#ifndef ALIPROTONFEEDDOWNANALYSIS_H\r
#define ALIPROTONFEEDDOWNANALYSIS_H\r
\r
#include "TObject.h"\r
#include "TH1I.h"\r
#include "AliCFContainer.h"\r
class TF1;\r
class TH2D;\r
class TH1F;\r
class TList;\r
\r
//#include "AliPID.h"\r
class AliPID;\r
\r
class AliCFDataGrid;\r
class AliAODEvent;\r
class AliAODtrack;\r
class AliESDEvent;\r
class AliESDtrack;\r
class AliExternalTrackParam;\r
class AliStack;\r
class AliESDVertex;\r
class AliProtonAnalysisBase;\r
\r
class AliProtonFeedDownAnalysis : public TObject \r
{\r
 	public:\r
	enum \r
		{\r
    			kAll      = 0, //without protons reject the secondaries from hadronic inter. \r
    			kPrimary = 1,\r
    			kFromLambda        = 2,\r
   			kSelected = 3 ,\r
    			kSelectedfromLambda = 4,\r
			kNSteps=5\r
		};\r
		AliProtonFeedDownAnalysis();\r
		//AliProtonFeedDownAnalysis(Int_t nbinsY, Float_t fLowY, Float_t fHighY,Int_t nbinsPt, Float_t fLowPt, Float_t fHighPt);\r
		virtual ~AliProtonFeedDownAnalysis();\r
		\r
		void SetBaseAnalysis(AliProtonAnalysisBase * const baseAnalysis) {fProtonAnalysisBase = baseAnalysis;}\r
		AliProtonAnalysisBase *GetProtonAnalysisBaseObject() const {return fProtonAnalysisBase;}\r
		\r
		void InitAnalysisHistograms(Int_t nbinsY, Float_t fLowY, Float_t fHighY, Int_t nbinsPt, Float_t fLowPt, Float_t fHighPt);\r
		void Analyze(AliESDEvent *fESD, const AliESDVertex *vertex,AliStack *stack);\r
		void Analyze(AliAODEvent *fAOD);\r
		void Analyze(AliStack *stack);\r
		\r
		AliCFContainer *GetProtonContainer() const {return fProtonContainer;}\r
		AliCFContainer *GetAntiProtonContainer() const {return fAntiProtonContainer;}\r
		\r
		void SetWeightFunction(TF1* weightfunction){fweightfunction=weightfunction;}\r
		TF1* GetWeightFunction() const{return fweightfunction;}	\r
		TH2F* GetLambdaHist() const{return fLambda;}\r
		TH2F* GetLambdaweightedHist() const{return fLambdaweighted;}\r
		TH2F* GetAntiLambdaHist() const{return fAntiLambda;}\r
		TH2F* GetAntiLambdaweightedHist() const{return fAntiLambdaweighted;}\r
 	private:\r
		AliProtonFeedDownAnalysis(const AliProtonFeedDownAnalysis&); // Not implemented\r
		AliProtonFeedDownAnalysis& operator=(const AliProtonFeedDownAnalysis&); // Not implemented\r
		\r
		AliProtonAnalysisBase *fProtonAnalysisBase;//base analysis object\r
		Int_t LambdaIsMother(Int_t numbe, AliStack *stack); \r
		Float_t GetWeightforProton(Int_t number, AliStack *stack);\r
		Float_t GetWeightforLambda(Float_t pt);\r
		\r
		Int_t fNBinsY; //number of bins in y or eta\r
		Double_t fMinY, fMaxY; //min & max value of y or eta\r
		Int_t fNBinsPt;  //number of bins in pT\r
		Double_t fMinPt, fMaxPt; //min & max value of pT\r
		\r
		//Analysis containers\r
		AliCFContainer *fProtonContainer; //container for protons\r
		AliCFContainer *fAntiProtonContainer; //container for antiprotons\r
		\r
		TF1*fweightfunction; \r
		TH2F *fLambda;\r
		TH2F *fLambdaweighted;\r
		TH2F *fAntiLambda;\r
		TH2F *fAntiLambdaweighted;\r
		\r
  ClassDef(AliProtonFeedDownAnalysis,2);\r
};\r
\r
#endif\r
\r