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

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

#ifndef ALIRSNANALYSISPHI7TEVNOPID_H
#define ALIRSNANALYSISPHI7TEVNOPID_H

#include "AliAnalysisTaskSE.h"
#include "AliRsnTOFT0maker.h"
#include "AliESDtrackCuts.h"

class TH1I;
class TH1F;
class TTree;

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

class AliRsnAnalysisPhi7TeVNoPID : public AliAnalysisTaskSE
{
  public:
  
    enum
    {
      kGoodTracksPrimaryVertex = 0,
      kGoodSPDPrimaryVertex    = 1,
      kFarTracksPrimaryVertex  = 2,
      kFarSPDPrimaryVertex     = 3,
      kNoGoodPrimaryVertex     = 4,
      kEvaluations             = 5
    };

    AliRsnAnalysisPhi7TeVNoPID(const char *name = "Phi7TeVNoPID");
    AliRsnAnalysisPhi7TeVNoPID(const AliRsnAnalysisPhi7TeVNoPID& copy);
    AliRsnAnalysisPhi7TeVNoPID& operator=(const AliRsnAnalysisPhi7TeVNoPID& copy);
    virtual ~AliRsnAnalysisPhi7TeVNoPID();

    void             SetUseMC(Bool_t yn = kTRUE) {fUseMC = yn;}
    
    void             SetMaxVz(Double_t v)   {fMaxVz = v;}
    
    void             SetITSband(Double_t v) {fMaxITSband = v;}
    
    void             SetTPCpLimit(Double_t v) {fTPCpLimit = v;}
    void             SetTPCrange(Double_t min, Double_t max) {fMinTPCband = min; fMaxTPCband = max;}
    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);
    AliESDtrackCuts* GetCutsTPC() {return &fESDtrackCutsTPC;}
    AliESDtrackCuts* GetCutsITS() {return &fESDtrackCutsITS;}

  private:

    void     ProcessESD(AliESDEvent *esd, AliStack *stack);
    void     ProcessMC(AliStack *stack);

    Bool_t   fUseMC;        // use MC or data?
    
    Short_t  fPDG;          // PDG code
    Short_t  fCh;           // control flag for like/unlike sign
    Short_t  fITS[2];       // check flag to know if one or both candidates are ITS standalone
    Float_t  fIM;           // inv mass
    Float_t  fPt;           // transv momentum
    Float_t  fY;            // rapidity
    Float_t  fEta;          // pseudo-rapidity
    Float_t  fTPCnsigma[2]; // number of sigma in TPC
    Float_t  fITSnsigma[2]; // number of sigma in ITS
    Float_t  fTOFdiff[2];   // relative PID signal in TOF
    Float_t  fP[2];         // total momentum at vertex
    Float_t  fPTPC[2];      // total momentum at inner TPC wall
    
    Double_t fMaxVz;      // range in Z of primary vertex w.r. to origin
    
    Double_t fMaxITSband; // range for ITS de/dx band

    Double_t fTPCpLimit;  // limit to choose what band to apply
    Double_t fTPCpar[5];  // parameters for TPC bethe-Bloch
    Double_t fMinTPCband; // range for TPC de/dx band - min
    Double_t fMaxTPCband; // range for TPC de/dx band - max

    TTree     *fRsnTreeComp;    // output tree of computed pairs
    TTree     *fRsnTreeTrue;    // output tree of true pairs
    TList     *fOutList;        // list for monitoring histograms
    TH1I      *fHEvents;        // histogram of event types
    TH1F      *fVertexX[2];     // histogram of X coordinate of primary vertex ([0] = tracks, [1] = SPD)
    TH1F      *fVertexY[2];     // histogram of Y coordinate of primary vertex ([0] = tracks, [1] = SPD)
    TH1F      *fVertexZ[2];     // histogram of Z coordinate of primary vertex ([0] = tracks, [1] = SPD)
    
    AliESDtrackCuts  fESDtrackCutsTPC;  //  ESD standard defined track cuts for TPC tracks
    AliESDtrackCuts  fESDtrackCutsITS;  //  ESD standard defined track cuts for ITS-SA tracks
    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

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

#endif
Commit	Line	Data
2dab9030	1	//
	2	// Header file for implementation of data analysis aft 900 GeV
	3	//
	4	// Author: A. Pulvirenti
	5	//
	6
	7	#ifndef ALIRSNANALYSISPHI7TEVNOPID_H
	8	#define ALIRSNANALYSISPHI7TEVNOPID_H
	9
	10	#include "AliAnalysisTaskSE.h"
	11	#include "AliRsnTOFT0maker.h"
	12	#include "AliESDtrackCuts.h"
	13
	14	class TH1I;
	15	class TH1F;
	16	class TTree;
	17
	18	class AliStack;
	19	class AliESDEvent;
	20	class AliESDVertex;
	21	class AliESDpid;
	22	class AliTOFT0maker;
	23	class AliTOFcalib;
	24
	25	class AliRsnAnalysisPhi7TeVNoPID : public AliAnalysisTaskSE
	26	{
	27	public:
	28
	29	enum
	30	{
	31	kGoodTracksPrimaryVertex = 0,
	32	kGoodSPDPrimaryVertex = 1,
	33	kFarTracksPrimaryVertex = 2,
	34	kFarSPDPrimaryVertex = 3,
	35	kNoGoodPrimaryVertex = 4,
	36	kEvaluations = 5
	37	};
	38
	39	AliRsnAnalysisPhi7TeVNoPID(const char *name = "Phi7TeVNoPID");
	40	AliRsnAnalysisPhi7TeVNoPID(const AliRsnAnalysisPhi7TeVNoPID& copy);
	41	AliRsnAnalysisPhi7TeVNoPID& operator=(const AliRsnAnalysisPhi7TeVNoPID& copy);
	42	virtual ~AliRsnAnalysisPhi7TeVNoPID();
	43
	44	void SetUseMC(Bool_t yn = kTRUE) {fUseMC = yn;}
	45
	46	void SetMaxVz(Double_t v) {fMaxVz = v;}
	47
	48	void SetITSband(Double_t v) {fMaxITSband = v;}
	49
	50	void SetTPCpLimit(Double_t v) {fTPCpLimit = v;}
	51	void SetTPCrange(Double_t min, Double_t max) {fMinTPCband = min; fMaxTPCband = max;}
	52	void SetTPCpar(Double_t p0, Double_t p1, Double_t p2, Double_t p3, Double_t p4)
	53	{fTPCpar[0]=p0;fTPCpar[1]=p1;fTPCpar[2]=p2;fTPCpar[3]=p3;fTPCpar[4]=p4;}
	54
	55	void SetTOFcalibrateESD(Bool_t yn = kTRUE) {fTOFcalibrateESD = yn;}
	56	void SetTOFcorrectTExp (Bool_t yn = kTRUE) {fTOFcorrectTExp = yn;}
	57	void SetTOFuseT0 (Bool_t yn = kTRUE) {fTOFuseT0 = yn;}
	58	void SetTOFtuneMC (Bool_t yn = kTRUE) {fTOFtuneMC = yn;}
	59	void SetTOFresolution (Double_t v = 100.0) {fTOFresolution = v;}
	60
	61	virtual void UserCreateOutputObjects();
	62	virtual void UserExec(Option_t *option = "");
	63	virtual void Terminate(Option_t *option = "");
	64
65	Int_t EventEval(AliESDEvent *esd);
66	AliESDtrackCuts* GetCutsTPC() {return &fESDtrackCutsTPC;}
67	AliESDtrackCuts* GetCutsITS() {return &fESDtrackCutsITS;}
68
69	private:
70
71	void ProcessESD(AliESDEvent esd, AliStack stack);
72	void ProcessMC(AliStack *stack);
73
74	Bool_t fUseMC; // use MC or data?
75
76	Short_t fPDG; // PDG code
77	Short_t fCh; // control flag for like/unlike sign
78	Short_t fITS[2]; // check flag to know if one or both candidates are ITS standalone
79	Float_t fIM; // inv mass
80	Float_t fPt; // transv momentum
81	Float_t fY; // rapidity
82	Float_t fEta; // pseudo-rapidity
83	Float_t fTPCnsigma[2]; // number of sigma in TPC
84	Float_t fITSnsigma[2]; // number of sigma in ITS
85	Float_t fTOFdiff[2]; // relative PID signal in TOF
86	Float_t fP[2]; // total momentum at vertex
87	Float_t fPTPC[2]; // total momentum at inner TPC wall
88
89	Double_t fMaxVz; // range in Z of primary vertex w.r. to origin
90
91	Double_t fMaxITSband; // range for ITS de/dx band
92
93	Double_t fTPCpLimit; // limit to choose what band to apply
94	Double_t fTPCpar[5]; // parameters for TPC bethe-Bloch
95	Double_t fMinTPCband; // range for TPC de/dx band - min
96	Double_t fMaxTPCband; // range for TPC de/dx band - max
97
98	TTree *fRsnTreeComp; // output tree of computed pairs
99	TTree *fRsnTreeTrue; // output tree of true pairs
100	TList *fOutList; // list for monitoring histograms
101	TH1I *fHEvents; // histogram of event types
102	TH1F *fVertexX[2]; // histogram of X coordinate of primary vertex ([0] = tracks, [1] = SPD)
103	TH1F *fVertexY[2]; // histogram of Y coordinate of primary vertex ([0] = tracks, [1] = SPD)
104	TH1F *fVertexZ[2]; // histogram of Z coordinate of primary vertex ([0] = tracks, [1] = SPD)
105
106	AliESDtrackCuts fESDtrackCutsTPC; // ESD standard defined track cuts for TPC tracks
107	AliESDtrackCuts fESDtrackCutsITS; // ESD standard defined track cuts for ITS-SA tracks
108	AliESDpid *fESDpid; //! PID manager
109	AliTOFT0maker *fTOFmaker; //! TOF time0 computator
110	AliTOFcalib *fTOFcalib; //! TOF calibration
111	Bool_t fTOFcalibrateESD; // TOF settings
112	Bool_t fTOFcorrectTExp; // TOF settings
113	Bool_t fTOFuseT0; // TOF settings
114	Bool_t fTOFtuneMC; // TOF settings
115	Double_t fTOFresolution; // TOF settings
116
117	// ROOT dictionary
118	ClassDef(AliRsnAnalysisPhi7TeVNoPID,1)
119	};
120
121	#endif