[u/mrichter/AliRoot.git] / PWGCF / EBYE / PIDFluctuation / task / AliAnalysisTaskPIDFluctuation.h

#ifndef ALIANALYSISTASKPIDFLUCTUATION_H
#define ALIANALYSISTASKPIDFLUCTUATION_H

/* 
 * updated by Zubayer Ahammed to instruct CINT about streaming
 * of data members.
 * Event by event PID fluctuation analysis
 * author: Roberto Preghenella (R+)
 * email:  preghenella@bo.infn.it
 *
 */
 
class TList;
class TH1F;
class TH2F;
class TH3F;
class AliVEvent;
class AliVTrack;
class AliESDtrackCuts;
class AliPIDResponse;

#include "AliAnalysisTaskSE.h"
#include "AliPID.h"
#include "THnSparse.h"

class AliAnalysisTaskPIDFluctuation : 
public AliAnalysisTaskSE 
{
  
 public:
  
  AliAnalysisTaskPIDFluctuation(const Char_t *name = "PIDFluctuation"); // default constructor
  virtual ~AliAnalysisTaskPIDFluctuation(); // default destructor
  
  void UserCreateOutputObjects();
  void UserExec(Option_t *option);

  /* setters */
  void SetPIDMethod(Int_t value) {fPIDMethod = value;}; // set PID method
  void SetESDtrackCuts(AliESDtrackCuts *value) {fESDtrackCuts = value;}; // set ESD track cuts
  void SetAODfilterBit(Int_t value) {fAODfilterBit = value;}; // set AOD filter bit
  void SetEtaRange(Float_t etaMin, Float_t etaMax) {fEtaMin = etaMin; fEtaMax = etaMax;}; // set eta range
  void SetPtRange(Float_t ptMin, Float_t ptMax) {fPtMin = ptMin; fPtMax = ptMax;}; // set pt range

  static const Int_t kNCentralityBins = 10; // N centrality bins

  enum EEventCounter_t {
    kAllEvents,
    kPhysicsSelection,
    kPrimaryVertex,
    kPrimaryVertexSPD,
    kVertexAccepted,
    kGoodCentrality,
    kAcceptedEvents,
    kNEventCounters
  };

  enum ESparseData_t {
    kCent_V0M,
    kCent_TRK,
    kNch,
    kNch_plus,
    kNch_minus,
    kNpi,
    kNpi_plus,
    kNpi_minus,
    kNka,
    kNka_plus,
    kNka_minus,
    kNpr,
    kNpr_plus,
    kNpr_minus,
    kNSparseData
  };

  enum EPIDMethod_t {
    kTPCTOF,
    kTPConly,
    kTOFonly,
    kNPIDMethods
  };

  static void MeasureNuDyn(const Char_t *filename, Int_t i1, Int_t i2, Int_t centralityEstimator = kCent_V0M);

 private:

  AliAnalysisTaskPIDFluctuation(const AliAnalysisTaskPIDFluctuation &); // not implemented
  AliAnalysisTaskPIDFluctuation &operator=(const AliAnalysisTaskPIDFluctuation &); // not implemented
  
  /*** event and track selection ***/
  Bool_t AcceptEvent(AliVEvent *event) const; // accept event
  Bool_t AcceptTrack(AliVTrack *track) const; // accept track

  /*** PID functions ***/
  Bool_t HasTPCPID(AliVTrack *track) const; // has TPC PID
  Bool_t HasTOFPID(AliVTrack *track) const; // has TOF PID
  Double_t MakeTPCPID(AliVTrack *track, Double_t *nSigma) const; // make TPC PID
  Double_t MakeTOFPID(AliVTrack *track, Double_t *nSigma) const; // make TOF PID
  void MakePID(AliVTrack *track, Bool_t *pidFlag, Float_t centrality) const; // make PID
  Bool_t InitPID(AliVEvent *event); // init PID

  /*** PID objects and flags ***/
  Int_t fPIDMethod; // PID method
  AliESDtrackCuts *fESDtrackCuts; // ESD track cuts
  Int_t fAODfilterBit; // AOD filter bit
  Float_t fEtaMin; // eta min
  Float_t fEtaMax; // eta max
  Float_t fPtMin; // pt min
  Float_t fPtMax; // pt max
  AliPIDResponse *fPID; //! PID

  /*** PID histos ***/
  TList *fHistoList; //! histo list
  TH1F *fHistoEventCounter; //! event counter
  TH2F *fHistoAcceptedTracks; //! accepted tracks
  TH2F *fHistoTOFMatchedTracks; //! TOF-matched tracks
  TH3F *fHistoTPCdEdx; //! TPC dEdx
  TH3F *fHistoTPCdEdx_inclusive; //! TPC dEdx
  TH3F *fHistoTOFbeta; //! TOF beta
  TH3F *fHistoTPCdEdx_selected[AliPID::kSPECIES]; //! TPC dEdx
  TH3F *fHistoTOFbeta_selected[AliPID::kSPECIES]; //! TOF beta
  TH3F *fHistoNSigmaTPC[AliPID::kSPECIES]; //! nsigma TPC
  TH3F *fHistoNSigmaTOF[AliPID::kSPECIES]; //! nsigma TOF

  /*** correlation histos */
  THnSparseI *fHistoCorrelation; // correlation THnSparse

  /*** labels, names and titles ***/
  static const Char_t *fgkEventCounterName[kNEventCounters]; // event couter name
  static const Char_t *fgkEventCounterTitle[kNEventCounters]; // event couter title
  static const Char_t *fgkSparseDataName[kNSparseData]; // sparse data name
  static const Char_t *fgkSparseDataTitle[kNSparseData]; // sparse data title

    
  ClassDef(AliAnalysisTaskPIDFluctuation, 1);
};

#endif /* ALIANALYSISTASKPIDFLUCTUATION_H */
Commit	Line	Data
a65a7e70	1	#ifndef ALIANALYSISTASKPIDFLUCTUATION_H
	2	#define ALIANALYSISTASKPIDFLUCTUATION_H
	3
	4	/*
	5	* updated by Zubayer Ahammed to instruct CINT about streaming
	6	* of data members.
	7	* Event by event PID fluctuation analysis
	8	* author: Roberto Preghenella (R+)
	9	* email: preghenella@bo.infn.it
	10	*
	11	*/
	12
	13	class TList;
	14	class TH1F;
	15	class TH2F;
	16	class TH3F;
	17	class AliVEvent;
	18	class AliVTrack;
	19	class AliESDtrackCuts;
	20	class AliPIDResponse;
	21
	22	#include "AliAnalysisTaskSE.h"
	23	#include "AliPID.h"
	24	#include "THnSparse.h"
	25
	26	class AliAnalysisTaskPIDFluctuation :
	27	public AliAnalysisTaskSE
	28	{
	29
	30	public:
	31
	32	AliAnalysisTaskPIDFluctuation(const Char_t *name = "PIDFluctuation"); // default constructor
	33	virtual ~AliAnalysisTaskPIDFluctuation(); // default destructor
	34
	35	void UserCreateOutputObjects();
	36	void UserExec(Option_t *option);
	37
	38	/* setters */
	39	void SetPIDMethod(Int_t value) {fPIDMethod = value;}; // set PID method
	40	void SetESDtrackCuts(AliESDtrackCuts *value) {fESDtrackCuts = value;}; // set ESD track cuts
	41	void SetAODfilterBit(Int_t value) {fAODfilterBit = value;}; // set AOD filter bit
	42	void SetEtaRange(Float_t etaMin, Float_t etaMax) {fEtaMin = etaMin; fEtaMax = etaMax;}; // set eta range
	43	void SetPtRange(Float_t ptMin, Float_t ptMax) {fPtMin = ptMin; fPtMax = ptMax;}; // set pt range
	44
	45	static const Int_t kNCentralityBins = 10; // N centrality bins
	46
	47	enum EEventCounter_t {
	48	kAllEvents,
	49	kPhysicsSelection,
	50	kPrimaryVertex,
	51	kPrimaryVertexSPD,
	52	kVertexAccepted,
	53	kGoodCentrality,
	54	kAcceptedEvents,
	55	kNEventCounters
	56	};
	57
	58	enum ESparseData_t {
	59	kCent_V0M,
	60	kCent_TRK,
	61	kNch,
	62	kNch_plus,
	63	kNch_minus,
	64	kNpi,
65	kNpi_plus,
66	kNpi_minus,
67	kNka,
68	kNka_plus,
69	kNka_minus,
70	kNpr,
71	kNpr_plus,
72	kNpr_minus,
73	kNSparseData
74	};
75
76	enum EPIDMethod_t {
77	kTPCTOF,
78	kTPConly,
79	kTOFonly,
80	kNPIDMethods
81	};
82
83	static void MeasureNuDyn(const Char_t *filename, Int_t i1, Int_t i2, Int_t centralityEstimator = kCent_V0M);
84
85	private:
86
87	AliAnalysisTaskPIDFluctuation(const AliAnalysisTaskPIDFluctuation &); // not implemented
88	AliAnalysisTaskPIDFluctuation &operator=(const AliAnalysisTaskPIDFluctuation &); // not implemented
89
90	/* event and track selection */
91	Bool_t AcceptEvent(AliVEvent *event) const; // accept event
92	Bool_t AcceptTrack(AliVTrack *track) const; // accept track
93
94	/* PID functions */
95	Bool_t HasTPCPID(AliVTrack *track) const; // has TPC PID
96	Bool_t HasTOFPID(AliVTrack *track) const; // has TOF PID
97	Double_t MakeTPCPID(AliVTrack track, Double_t nSigma) const; // make TPC PID
98	Double_t MakeTOFPID(AliVTrack track, Double_t nSigma) const; // make TOF PID
99	void MakePID(AliVTrack track, Bool_t pidFlag, Float_t centrality) const; // make PID
100	Bool_t InitPID(AliVEvent *event); // init PID
101
102	/* PID objects and flags */
103	Int_t fPIDMethod; // PID method
104	AliESDtrackCuts *fESDtrackCuts; // ESD track cuts
105	Int_t fAODfilterBit; // AOD filter bit
106	Float_t fEtaMin; // eta min
107	Float_t fEtaMax; // eta max
108	Float_t fPtMin; // pt min
109	Float_t fPtMax; // pt max
110	AliPIDResponse *fPID; //! PID
111
112	/* PID histos */
113	TList *fHistoList; //! histo list
114	TH1F *fHistoEventCounter; //! event counter
115	TH2F *fHistoAcceptedTracks; //! accepted tracks
116	TH2F *fHistoTOFMatchedTracks; //! TOF-matched tracks
117	TH3F *fHistoTPCdEdx; //! TPC dEdx
118	TH3F *fHistoTPCdEdx_inclusive; //! TPC dEdx
119	TH3F *fHistoTOFbeta; //! TOF beta
120	TH3F *fHistoTPCdEdx_selected[AliPID::kSPECIES]; //! TPC dEdx
121	TH3F *fHistoTOFbeta_selected[AliPID::kSPECIES]; //! TOF beta
122	TH3F *fHistoNSigmaTPC[AliPID::kSPECIES]; //! nsigma TPC
123	TH3F *fHistoNSigmaTOF[AliPID::kSPECIES]; //! nsigma TOF
124
125	/*** correlation histos */
126	THnSparseI *fHistoCorrelation; // correlation THnSparse
127
128	/* labels, names and titles */
129	static const Char_t *fgkEventCounterName[kNEventCounters]; // event couter name
130	static const Char_t *fgkEventCounterTitle[kNEventCounters]; // event couter title
131	static const Char_t *fgkSparseDataName[kNSparseData]; // sparse data name
132	static const Char_t *fgkSparseDataTitle[kNSparseData]; // sparse data title
133
134
135	ClassDef(AliAnalysisTaskPIDFluctuation, 1);
136	};
137
138	#endif /* ALIANALYSISTASKPIDFLUCTUATION_H */