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

//
// Header file for implementation of data analysis aft 900 GeV
//
// Author: A. Pulvirenti
//

#ifndef ALIRSNANALYSISMONITORTASK_H
#define ALIRSNANALYSISMONITORTASK_H

#include "AliAnalysisTaskSE.h"
#include "AliRsnCutSet.h"

class TH1I;
class TH1F;
class TTree;

class AliStack;
class AliESDEvent;
class AliESDVertex;
class AliESDpid;
class AliTOFT0maker;
class AliTOFcalib;

class AliRsnAnalysisMonitorTask : public AliAnalysisTaskSE {
public:

   AliRsnAnalysisMonitorTask(const char *name = "Phi7TeV");
   AliRsnAnalysisMonitorTask(const AliRsnAnalysisMonitorTask& copy);
   AliRsnAnalysisMonitorTask& operator=(const AliRsnAnalysisMonitorTask& copy);
   virtual ~AliRsnAnalysisMonitorTask();

   void             SetTPCpar(Double_t p0, Double_t p1, Double_t p2, Double_t p3, Double_t p4)
   {fTPCpar[0] = p0; fTPCpar[1] = p1; fTPCpar[2] = p2; fTPCpar[3] = p3; fTPCpar[4] = p4;}

   void             SetTOFcalibrateESD(Bool_t yn = kTRUE)  {fTOFcalibrateESD = yn;}
   void             SetTOFcorrectTExp(Bool_t yn = kTRUE)  {fTOFcorrectTExp = yn;}
   void             SetTOFuseT0(Bool_t yn = kTRUE)  {fTOFuseT0 = yn;}
   void             SetTOFtuneMC(Bool_t yn = kTRUE)  {fTOFtuneMC = yn;}
   void             SetTOFresolution(Double_t v = 100.0) {fTOFresolution = v;}

   virtual void     UserCreateOutputObjects();
   virtual void     UserExec(Option_t *option = "");
   virtual void     Terminate(Option_t *option = "");

   Int_t            EventEval(AliESDEvent *esd);
   Bool_t           IsTPCtrack(AliESDtrack *track);
   Bool_t           IsITSSAtrack(AliESDtrack *track);
   void             ProcessESD(AliESDEvent *esd, const AliESDVertex *v, AliStack *stack);

   AliRsnCutSet*    GetEventCuts() {return &fEventCuts;}
   AliRsnCutSet*    GetTrackCuts() {return &fTrackCuts;}

private:

   TTree              *fOut;              //  output TTree
   AliRsnMonitorTrack *fTrack;            //  branch object for output TTree

   Double_t            fTPCpar[5];        //  parameters for TPC bethe-Bloch

   AliESDpid          *fESDpid;           //! PID manager

   AliTOFT0maker      *fTOFmaker;         //! TOF time0 computator
   AliTOFcalib        *fTOFcalib;         //! TOF calibration
   Bool_t              fTOFcalibrateESD;  //  TOF settings
   Bool_t              fTOFcorrectTExp;   //  TOF settings
   Bool_t              fTOFuseT0;         //  TOF settings
   Bool_t              fTOFtuneMC;        //  TOF settings
   Double_t            fTOFresolution;    //  TOF settings

   AliRsnCutSet        fEventCuts;        //  event cuts
   AliRsnCutSet        fTrackCuts;        //  track cuts

   // ROOT dictionary
   ClassDef(AliRsnAnalysisMonitorTask, 1)
};

#endif
Commit	Line	Data
6aecf4fd	1	//
	2	// Header file for implementation of data analysis aft 900 GeV
	3	//
	4	// Author: A. Pulvirenti
	5	//
	6
	7	#ifndef ALIRSNANALYSISMONITORTASK_H
	8	#define ALIRSNANALYSISMONITORTASK_H
	9
	10	#include "AliAnalysisTaskSE.h"
69fbb331	11	#include "AliRsnCutSet.h"
6aecf4fd	12
	13	class TH1I;
	14	class TH1F;
	15	class TTree;
	16
	17	class AliStack;
	18	class AliESDEvent;
	19	class AliESDVertex;
	20	class AliESDpid;
	21	class AliTOFT0maker;
	22	class AliTOFcalib;
	23
2a1c7696	24	class AliRsnAnalysisMonitorTask : public AliAnalysisTaskSE {
	25	public:
	26
	27	AliRsnAnalysisMonitorTask(const char *name = "Phi7TeV");
	28	AliRsnAnalysisMonitorTask(const AliRsnAnalysisMonitorTask& copy);
	29	AliRsnAnalysisMonitorTask& operator=(const AliRsnAnalysisMonitorTask& copy);
	30	virtual ~AliRsnAnalysisMonitorTask();
	31
	32	void SetTPCpar(Double_t p0, Double_t p1, Double_t p2, Double_t p3, Double_t p4)
	33	{fTPCpar[0] = p0; fTPCpar[1] = p1; fTPCpar[2] = p2; fTPCpar[3] = p3; fTPCpar[4] = p4;}
	34
	35	void SetTOFcalibrateESD(Bool_t yn = kTRUE) {fTOFcalibrateESD = yn;}
	36	void SetTOFcorrectTExp(Bool_t yn = kTRUE) {fTOFcorrectTExp = yn;}
	37	void SetTOFuseT0(Bool_t yn = kTRUE) {fTOFuseT0 = yn;}
	38	void SetTOFtuneMC(Bool_t yn = kTRUE) {fTOFtuneMC = yn;}
	39	void SetTOFresolution(Double_t v = 100.0) {fTOFresolution = v;}
	40
	41	virtual void UserCreateOutputObjects();
	42	virtual void UserExec(Option_t *option = "");
	43	virtual void Terminate(Option_t *option = "");
	44
	45	Int_t EventEval(AliESDEvent *esd);
	46	Bool_t IsTPCtrack(AliESDtrack *track);
	47	Bool_t IsITSSAtrack(AliESDtrack *track);
	48	void ProcessESD(AliESDEvent esd, const AliESDVertex v, AliStack *stack);
	49
	50	AliRsnCutSet* GetEventCuts() {return &fEventCuts;}
	51	AliRsnCutSet* GetTrackCuts() {return &fTrackCuts;}
	52
	53	private:
	54
	55	TTree *fOut; // output TTree
	56	AliRsnMonitorTrack *fTrack; // branch object for output TTree
	57
	58	Double_t fTPCpar[5]; // parameters for TPC bethe-Bloch
	59
	60	AliESDpid *fESDpid; //! PID manager
	61
	62	AliTOFT0maker *fTOFmaker; //! TOF time0 computator
	63	AliTOFcalib *fTOFcalib; //! TOF calibration
	64	Bool_t fTOFcalibrateESD; // TOF settings
	65	Bool_t fTOFcorrectTExp; // TOF settings
	66	Bool_t fTOFuseT0; // TOF settings
	67	Bool_t fTOFtuneMC; // TOF settings
	68	Double_t fTOFresolution; // TOF settings
	69
	70	AliRsnCutSet fEventCuts; // event cuts
	71	AliRsnCutSet fTrackCuts; // track cuts
	72
	73	// ROOT dictionary
	74	ClassDef(AliRsnAnalysisMonitorTask, 1)
6aecf4fd	75	};
	76
	77	#endif