[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / UserTasks / AliAnalysisTaskSAJF.h

#ifndef ALIANALYSISTASKSAJF_H
#define ALIANALYSISTASKSAJF_H

// $Id$

class TH2;
class THnSparse;

#include <TH3F.h>

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskSAJF : public AliAnalysisTaskEmcalJet {
 public:

  AliAnalysisTaskSAJF();
  AliAnalysisTaskSAJF(const char *name);
  virtual ~AliAnalysisTaskSAJF() {;}

  void                        UserCreateOutputObjects();

  void                        SetHistoType(Int_t t) { fHistoType = t; }

 protected:
  void                        AllocateTHX();
  void                        AllocateTHnSparse();

  Bool_t                      FillHistograms();
  void                        FillJetHisto(Double_t cent, Double_t ep, Double_t eta, Double_t phi, Double_t pt, Double_t MCpt, Double_t corrpt, Double_t area, 
					   Double_t NEF, Double_t z, Int_t n, Double_t leadingpt);

  Int_t                       fHistoType;                      // histogram type (0=TH2, 1=THnSparse)

  // Inclusive jets histograms
  THnSparse                  *fHistJetObservables;             //!Jet-wise observables

  // TH2/TH3 versions
  TH3                        *fHistJetPtEtaPhi[4];             //!Jet Pt vs. Eta vs. Phi
  TH2                        *fHistJetPtArea[4];               //!Jet Pt vs. Area
  TH2                        *fHistJetPtEP[4];                 //!Jet Pt vs. event plane
  TH2                        *fHistJetPtNEF[4];                //!Jet Pt vs. neutral energy fraction
  TH2                        *fHistJetPtZ[4];                  //!Jet Pt vs. z
  TH2                        *fHistJetPtLeadingPartPt[4];      //!Jet Pt vs. leading particle pt
  TH3                        *fHistJetCorrPtEtaPhi[4];         //!Jet corrPt vs. Eta vs. Phi
  TH2                        *fHistJetCorrPtArea[4];           //!Jet corrPt vs. Area
  TH2                        *fHistJetCorrPtEP[4];             //!Jet corrPt vs. event plane
  TH2                        *fHistJetCorrPtNEF[4];            //!Jet corrPt vs. neutral energy fraction
  TH2                        *fHistJetCorrPtZ[4];              //!Jet corrPt vs. z
  TH2                        *fHistJetCorrPtLeadingPartPt[4];  //!Jet corrPt vs. leading particle pt
  TH2                        *fHistJetPtCorrPt[4];             //!Jet Pt vs. corrPt
  TH2                        *fHistJetPtMCPt[4];               //!Jet Pt vs. MCPt
  TH2                        *fHistJetMCPtCorrPt[4];           //!Jet MCPt vs. corrPt

  TH2                        *fHistTracksJetPt[4];             //!Track pt vs. jet pt
  TH2                        *fHistClustersJetPt[4];           //!Cluster pt vs. jet pt
  TH2                        *fHistTracksPtDist[4];            //!Track pt vs. distance form jet axis
  TH2                        *fHistClustersPtDist[4];          //!Cluster pt vs. distance form jet axis

 private:
  AliAnalysisTaskSAJF(const AliAnalysisTaskSAJF&);            // not implemented
  AliAnalysisTaskSAJF &operator=(const AliAnalysisTaskSAJF&); // not implemented

  ClassDef(AliAnalysisTaskSAJF, 17) // jet analysis task
};
#endif
Commit	Line	Data
00514d01	1	#ifndef ALIANALYSISTASKSAJF_H
00514d01	2	#define ALIANALYSISTASKSAJF_H
25283b37	3
00514d01	4	// $Id$
25283b37	5
43032ce2	6	class TH2;
	7	class THnSparse;
	8
85d48773	9	#include <TH3F.h>
85d48773	10
9239b066	11	#include "AliAnalysisTaskEmcalJet.h"
25283b37	12
9239b066	13	class AliAnalysisTaskSAJF : public AliAnalysisTaskEmcalJet {
25283b37	14	public:
91f4b7c5	15
00514d01	16	AliAnalysisTaskSAJF();
00514d01	17	AliAnalysisTaskSAJF(const char *name);
43032ce2	18	virtual ~AliAnalysisTaskSAJF() {;}
25283b37	19
25283b37	20	void UserCreateOutputObjects();
05077f28	21
85d48773	22	void SetHistoType(Int_t t) { fHistoType = t; }
85d48773	23
25283b37	24	protected:
85d48773	25	void AllocateTHX();
	26	void AllocateTHnSparse();
	27
	28	Bool_t FillHistograms();
3fe08cdb	29	void FillJetHisto(Double_t cent, Double_t ep, Double_t eta, Double_t phi, Double_t pt, Double_t MCpt, Double_t corrpt, Double_t area,
3fe08cdb	30	Double_t NEF, Double_t z, Int_t n, Double_t leadingpt);
a487deae	31
85d48773	32	Int_t fHistoType; // histogram type (0=TH2, 1=THnSparse)
85d48773	33
a487deae	34	// Inclusive jets histograms
85d48773	35	THnSparse *fHistJetObservables; //!Jet-wise observables
	36
	37	// TH2/TH3 versions
	38	TH3 *fHistJetPtEtaPhi[4]; //!Jet Pt vs. Eta vs. Phi
	39	TH2 *fHistJetPtArea[4]; //!Jet Pt vs. Area
	40	TH2 *fHistJetPtEP[4]; //!Jet Pt vs. event plane
	41	TH2 *fHistJetPtNEF[4]; //!Jet Pt vs. neutral energy fraction
	42	TH2 *fHistJetPtZ[4]; //!Jet Pt vs. z
	43	TH2 *fHistJetPtLeadingPartPt[4]; //!Jet Pt vs. leading particle pt
	44	TH3 *fHistJetCorrPtEtaPhi[4]; //!Jet corrPt vs. Eta vs. Phi
	45	TH2 *fHistJetCorrPtArea[4]; //!Jet corrPt vs. Area
	46	TH2 *fHistJetCorrPtEP[4]; //!Jet corrPt vs. event plane
	47	TH2 *fHistJetCorrPtNEF[4]; //!Jet corrPt vs. neutral energy fraction
	48	TH2 *fHistJetCorrPtZ[4]; //!Jet corrPt vs. z
	49	TH2 *fHistJetCorrPtLeadingPartPt[4]; //!Jet corrPt vs. leading particle pt
	50	TH2 *fHistJetPtCorrPt[4]; //!Jet Pt vs. corrPt
	51	TH2 *fHistJetPtMCPt[4]; //!Jet Pt vs. MCPt
	52	TH2 *fHistJetMCPtCorrPt[4]; //!Jet MCPt vs. corrPt
	53
	54	TH2 *fHistTracksJetPt[4]; //!Track pt vs. jet pt
	55	TH2 *fHistClustersJetPt[4]; //!Cluster pt vs. jet pt
	56	TH2 *fHistTracksPtDist[4]; //!Track pt vs. distance form jet axis
	57	TH2 *fHistClustersPtDist[4]; //!Cluster pt vs. distance form jet axis
25283b37	58
25283b37	59	private:
00514d01	60	AliAnalysisTaskSAJF(const AliAnalysisTaskSAJF&); // not implemented
00514d01	61	AliAnalysisTaskSAJF &operator=(const AliAnalysisTaskSAJF&); // not implemented
25283b37	62
7cd832c7	63	ClassDef(AliAnalysisTaskSAJF, 17) // jet analysis task
25283b37	64	};
25283b37	65	#endif