[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskPIDCombined.h

#ifndef ALIPIDCOMBINEDTASK_H
#define ALIPIDCOMBINEDTASK_H
/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

//#########################################################
//#                                                       # 
//#        Task for testing the combined PID              #
//#                                                       #
//#  Pietro Antonioli, INFN / Pietro.Antonioli@bo.infn.it #
//#  Jens Wiechula, Uni Tübingen / Jens.Wiechula@cern.ch  #
//#                                                       #
//#########################################################

#include <AliPID.h>
#include <AliPIDResponse.h>
#include <AliPIDCombined.h>

#include <AliESDtrackCuts.h>
#include <AliAnalysisFilter.h>

#include "AliAnalysisTaskSE.h"

class TH1F;
class TH2D;
class fPIDResponse;
class fPIDCombined;

class AliAnalysisTaskPIDCombined : public AliAnalysisTaskSE {
  
public:
  static const Int_t kPtBins = 6;


  AliAnalysisTaskPIDCombined();
  AliAnalysisTaskPIDCombined(const char *name);
  virtual ~AliAnalysisTaskPIDCombined(){;}
  
  virtual void  UserExec(Option_t *option);
  virtual void  UserCreateOutputObjects();
    

private:

  TList fHistList;                   //! list of histograms
  TH2D *fProbTPCnSigma[AliPID::kSPECIES];    //! probabilities vs nSigma in the TPC
  TH2D *fProbTOFnSigma[AliPID::kSPECIES];    //! probabilities vs nSigma  the TOF
  TH2D *fProbTPCTOFnSigmaTPC[AliPID::kSPECIES];    //! comb. probabilities vs nSigma TPC
  TH2D *fProbTPC[AliPID::kSPECIES];          //! probabilities vs mom in the TPC
  TH2D *fProbTOF[AliPID::kSPECIES];          //! probabilities vs mom in the TOF
  TH2D *fProbTPCTOF[AliPID::kSPECIES];       //! combined probabilities vs mom TPC-TOF
  TH1F *fPriors[AliPID::kSPECIES];           //! priors

  TH2D *fProbTPCTOFnSigTPCMom[kPtBins][AliPID::kSPECIES];  // prob. x mom. bins
  TH2D *fProbTPCnSigTPCMom[kPtBins][AliPID::kSPECIES];     // prob. x mom. bins
  TH2D *fProbTOFnSigTOFMom[kPtBins][AliPID::kSPECIES];     // prob. x mom. bins

  TH2D *fPriorsUsed[AliPID::kSPECIES];       //! priors used

  const AliPIDResponse *fPIDResponse;     //! PID response object
  AliPIDCombined       *fPIDCombined;     //! combined PID object
  AliESDtrackCuts      *fTrackCuts;            //! track selection
  AliAnalysisFilter    *fTrackFilter;         //! track filter

  TH2D *fDeDx;                              //! histo with the dedx
  TH2D *fDeDxTuned;                         //! histo to check the dedx tuning in MC


  AliAnalysisTaskPIDCombined(const AliAnalysisTaskPIDCombined &c);
  AliAnalysisTaskPIDCombined& operator= (const AliAnalysisTaskPIDCombined &c);

  void FillHistogram(const char* name, Double_t x, Double_t weight=1.);
  void FillHistogram(const char* name, Double_t x, Double_t y, Double_t weight=1.);
  Int_t GetMomBin(Float_t mom);
  static const char* fgkBinMomDesc[kPtBins];
  
  ClassDef(AliAnalysisTaskPIDCombined, 2);
};
#endif
Commit	Line	Data
a65a7e70	1	#ifndef ALIPIDCOMBINEDTASK_H
	2	#define ALIPIDCOMBINEDTASK_H
	3	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	4	* See cxx source for full Copyright notice */
	5
	6	//#########################################################
	7	//# #
	8	//# Task for testing the combined PID #
	9	//# #
	10	//# Pietro Antonioli, INFN / Pietro.Antonioli@bo.infn.it #
	11	//# Jens Wiechula, Uni Tübingen / Jens.Wiechula@cern.ch #
	12	//# #
	13	//#########################################################
	14
	15	#include <AliPID.h>
	16	#include <AliPIDResponse.h>
	17	#include <AliPIDCombined.h>
	18
	19	#include <AliESDtrackCuts.h>
	20	#include <AliAnalysisFilter.h>
	21
	22	#include "AliAnalysisTaskSE.h"
	23
	24	class TH1F;
	25	class TH2D;
	26	class fPIDResponse;
	27	class fPIDCombined;
	28
	29	class AliAnalysisTaskPIDCombined : public AliAnalysisTaskSE {
	30
	31	public:
	32	static const Int_t kPtBins = 6;
	33
	34
	35	AliAnalysisTaskPIDCombined();
	36	AliAnalysisTaskPIDCombined(const char *name);
	37	virtual ~AliAnalysisTaskPIDCombined(){;}
	38
	39	virtual void UserExec(Option_t *option);
	40	virtual void UserCreateOutputObjects();
	41
	42
	43	private:
	44
	45	TList fHistList; //! list of histograms
	46	TH2D *fProbTPCnSigma[AliPID::kSPECIES]; //! probabilities vs nSigma in the TPC
	47	TH2D *fProbTOFnSigma[AliPID::kSPECIES]; //! probabilities vs nSigma the TOF
	48	TH2D *fProbTPCTOFnSigmaTPC[AliPID::kSPECIES]; //! comb. probabilities vs nSigma TPC
	49	TH2D *fProbTPC[AliPID::kSPECIES]; //! probabilities vs mom in the TPC
	50	TH2D *fProbTOF[AliPID::kSPECIES]; //! probabilities vs mom in the TOF
	51	TH2D *fProbTPCTOF[AliPID::kSPECIES]; //! combined probabilities vs mom TPC-TOF
	52	TH1F *fPriors[AliPID::kSPECIES]; //! priors
	53
	54	TH2D *fProbTPCTOFnSigTPCMom[kPtBins][AliPID::kSPECIES]; // prob. x mom. bins
	55	TH2D *fProbTPCnSigTPCMom[kPtBins][AliPID::kSPECIES]; // prob. x mom. bins
	56	TH2D *fProbTOFnSigTOFMom[kPtBins][AliPID::kSPECIES]; // prob. x mom. bins
	57
	58	TH2D *fPriorsUsed[AliPID::kSPECIES]; //! priors used
	59
	60	const AliPIDResponse *fPIDResponse; //! PID response object
	61	AliPIDCombined *fPIDCombined; //! combined PID object
	62	AliESDtrackCuts *fTrackCuts; //! track selection
	63	AliAnalysisFilter *fTrackFilter; //! track filter
539a5a59	64
	65	TH2D *fDeDx; //! histo with the dedx
	66	TH2D *fDeDxTuned; //! histo to check the dedx tuning in MC
	67
a65a7e70	68
	69	AliAnalysisTaskPIDCombined(const AliAnalysisTaskPIDCombined &c);
	70	AliAnalysisTaskPIDCombined& operator= (const AliAnalysisTaskPIDCombined &c);
	71
	72	void FillHistogram(const char* name, Double_t x, Double_t weight=1.);
	73	void FillHistogram(const char* name, Double_t x, Double_t y, Double_t weight=1.);
	74	Int_t GetMomBin(Float_t mom);
	75	static const char* fgkBinMomDesc[kPtBins];
	76
	77	ClassDef(AliAnalysisTaskPIDCombined, 2);
	78	};
	79	#endif