[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;
class AliMCEvent;
class AliProtonAnalysis;
class TF1;

#include "AliAnalysisTask.h"

class AliAnalysisTaskProtons : public AliAnalysisTask {
 public:
  AliAnalysisTaskProtons();
  AliAnalysisTaskProtons(const char *name);
  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                                                                                  
  AliMCEvent  *fMC;     //MC object                                                                                   
  
  TString fAnalysisType;//"ESD", "AOD" or "MC"                                                                        

  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
b620b667	6	class TString;
734d2c12	7	class TList;
734d2c12	8	class AliESDEvent;
b620b667	9	class AliAODEvent;
e4358d7f	10	class AliMCEvent;
c5ba3680	11	class AliProtonAnalysis;
c5ba3680	12	class TF1;
734d2c12	13
734d2c12	14	#include "AliAnalysisTask.h"
	15
	16	class AliAnalysisTaskProtons : public AliAnalysisTask {
	17	public:
db10bcb0	18	AliAnalysisTaskProtons();
db10bcb0	19	AliAnalysisTaskProtons(const char *name);
734d2c12	20	virtual ~AliAnalysisTaskProtons() {}
	21
	22	virtual void ConnectInputData(Option_t *);
	23	virtual void CreateOutputObjects();
	24	virtual void Exec(Option_t *option);
	25	virtual void Terminate(Option_t *);
aafecd8b	26
b620b667	27	void SetType(const char* type) {fAnalysisType = type;}
df01f00b	28	void SetPriorProbabilityFunctions(TF1 *felectrons,
	29	TF1 *fmuons,
	30	TF1 *fpions,
	31	TF1 *fkaons,
	32	TF1 *fprotons) {
aafecd8b	33	fFunctionUsed = kTRUE;
	34	fElectronFunction = felectrons;
	35	fMuonFunction = fmuons;
	36	fPionFunction = fpions;
	37	fKaonFunction = fkaons;
	38	fProtonFunction = fprotons;
	39	}
734d2c12	40
734d2c12	41	private:
df01f00b	42	AliESDEvent *fESD; //ESD object
	43	AliAODEvent *fAOD; //AOD object
	44	AliMCEvent *fMC; //MC object
	45
	46	TString fAnalysisType;//"ESD", "AOD" or "MC"
aafecd8b	47
df01f00b	48	TList *fList; //TList output object
	49
	50	AliProtonAnalysis *fAnalysis; //analysis object
	51
	52	TF1 *fElectronFunction; //TF1 for e
	53	TF1 *fMuonFunction; //TF1 for mu
	54	TF1 *fPionFunction; //TF1 for pi
	55	TF1 *fKaonFunction; //TF1 for K
	56	TF1 *fProtonFunction; //TF1 for p
	57
	58	Bool_t fFunctionUsed; //kTRUE if Functions are used
	59
	60	AliAnalysisTaskProtons(const AliAnalysisTaskProtons&); // not implemented
	61	AliAnalysisTaskProtons& operator=(const AliAnalysisTaskProtons&); // not implemented
734d2c12	62
	63	ClassDef(AliAnalysisTaskProtons, 1); // example of analysis
	64	};
	65
	66	#endif