[u/mrichter/AliRoot.git] / PWG2 / AliAnalysisTaskProtons.h

#ifndef AliAnalysisTaskProtons_cxx
#define AliAnalysisTaskProtons_cxx

// Analysis task creating a the 2d y-p_t spectrum of p and antip
// Author: Panos Cristakoglou

class TString;
class TList;
class AliESDEvent;
class AliAODEvent;
#include "TF1.h"

#include "PWG2spectra/SPECTRA/AliProtonAnalysis.h"
#include "AliAnalysisTask.h"

class AliAnalysisTaskProtons : public AliAnalysisTask {
 public:
  AliAnalysisTaskProtons(const char *name = "AliAnalysisTaskProtons");
  virtual ~AliAnalysisTaskProtons() {}
  
  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);

  void SetType(const char* type) {fAnalysisType = type;}
  void SetPriorProbabilityFunctions(TF1 *felectrons, 
				    TF1 *fmuons,
				    TF1 *fpions,
				    TF1 *fkaons,
				    TF1 *fprotons) {
    fFunctionUsed = kTRUE;
    fElectronFunction = felectrons;
    fMuonFunction = fmuons;
    fPionFunction = fpions;
    fKaonFunction = fkaons;
    fProtonFunction = fprotons;
  }
  
 private:
  AliESDEvent *fESD;    //ESD object
  AliAODEvent *fAOD;    //AOD object
  TString fAnalysisType;//"ESD" or "AOD"
  TList  *fList; //TList output object
  AliProtonAnalysis *fAnalysis; //analysis object
  TF1 *fElectronFunction; //TF1 for e
  TF1 *fMuonFunction; //TF1 for mu
  TF1 *fPionFunction; //TF1 for pi
  TF1 *fKaonFunction; //TF1 for K
  TF1 *fProtonFunction; //TF1 for p

  Bool_t fFunctionUsed; //kTRUE if Functions are used

  AliAnalysisTaskProtons(const AliAnalysisTaskProtons&); // not implemented
  AliAnalysisTaskProtons& operator=(const AliAnalysisTaskProtons&); // not implemented
  
  ClassDef(AliAnalysisTaskProtons, 1); // example of analysis
};

#endif
Commit	Line	Data
734d2c12	1	#ifndef AliAnalysisTaskProtons_cxx
	2	#define AliAnalysisTaskProtons_cxx
	3
	4	// Analysis task creating a the 2d y-p_t spectrum of p and antip
	5	// Author: Panos Cristakoglou
	6
b620b667	7	class TString;
734d2c12	8	class TList;
734d2c12	9	class AliESDEvent;
b620b667	10	class AliAODEvent;
aafecd8b	11	#include "TF1.h"
734d2c12	12
	13	#include "PWG2spectra/SPECTRA/AliProtonAnalysis.h"
	14	#include "AliAnalysisTask.h"
	15
	16	class AliAnalysisTaskProtons : public AliAnalysisTask {
	17	public:
	18	AliAnalysisTaskProtons(const char *name = "AliAnalysisTaskProtons");
	19	virtual ~AliAnalysisTaskProtons() {}
	20
	21	virtual void ConnectInputData(Option_t *);
	22	virtual void CreateOutputObjects();
	23	virtual void Exec(Option_t *option);
	24	virtual void Terminate(Option_t *);
aafecd8b	25
b620b667	26	void SetType(const char* type) {fAnalysisType = type;}
aafecd8b	27	void SetPriorProbabilityFunctions(TF1 *felectrons,
	28	TF1 *fmuons,
	29	TF1 *fpions,
	30	TF1 *fkaons,
	31	TF1 *fprotons) {
	32	fFunctionUsed = kTRUE;
	33	fElectronFunction = felectrons;
	34	fMuonFunction = fmuons;
	35	fPionFunction = fpions;
	36	fKaonFunction = fkaons;
	37	fProtonFunction = fprotons;
	38	}
734d2c12	39
	40	private:
	41	AliESDEvent *fESD; //ESD object
b620b667	42	AliAODEvent *fAOD; //AOD object
b620b667	43	TString fAnalysisType;//"ESD" or "AOD"
734d2c12	44	TList *fList; //TList output object
734d2c12	45	AliProtonAnalysis *fAnalysis; //analysis object
aafecd8b	46	TF1 *fElectronFunction; //TF1 for e
	47	TF1 *fMuonFunction; //TF1 for mu
	48	TF1 *fPionFunction; //TF1 for pi
	49	TF1 *fKaonFunction; //TF1 for K
	50	TF1 *fProtonFunction; //TF1 for p
	51
	52	Bool_t fFunctionUsed; //kTRUE if Functions are used
	53
734d2c12	54	AliAnalysisTaskProtons(const AliAnalysisTaskProtons&); // not implemented
	55	AliAnalysisTaskProtons& operator=(const AliAnalysisTaskProtons&); // not implemented
	56
	57	ClassDef(AliAnalysisTaskProtons, 1); // example of analysis
	58	};
	59
	60	#endif