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

#ifndef ALIANALYSISTASKJETMASSRESPONSEDET_H
#define ALIANALYSISTASKJETMASSRESPONSEDET_H

class TH1;
class TH2;
class TH3;
class TH3F;
class THnSparse;
class TClonesArray;
class TArrayI;
class AliAnalysisManager;
class AliJetContainer;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskJetMassResponseDet : public AliAnalysisTaskEmcalJet {
 public:
  enum JetMassType {
    kRaw   = 0,  //mass form anti-kt 4-vector
    kDeriv = 1   //area based subtracted jet mass
  };

  AliAnalysisTaskJetMassResponseDet();
  AliAnalysisTaskJetMassResponseDet(const char *name);
  virtual ~AliAnalysisTaskJetMassResponseDet();

  void                                UserCreateOutputObjects();
  void                                Terminate(Option_t *option);

  //Setters
  void SetJetContainerPart(Int_t c)                             { fContainerPart     = c   ; }
  void SetJetContainerDet(Int_t c)                              { fContainerDet      = c   ; }
  void SetJetMassType(JetMassType t)                            { fJetMassType       = t   ; }

 protected:
  Bool_t                              RetrieveEventObjects();
  Bool_t                              Run();
  Bool_t                              FillHistograms();

  Double_t                            GetJetMass(AliEmcalJet *jet);
 
  Int_t                               fContainerPart;              // particle level jets
  Int_t                               fContainerDet;               // detector level jets
  JetMassType                         fJetMassType;                // jet mass type to be used

  TH2F            *fh2PtVsMassJetPartAll;            //!pT vs mass of all particle level jets
  TH2F            *fh2PtVsMassJetPartMatch;          //!pT vs mass of all particle level jets matched to a detector level jet
  TH2F            *fh2PtVsMassJetPartTagged;         //!pT vs mass of tagged particle level jets
  TH2F            *fh2PtVsMassJetPartTaggedMatch;    //!pT vs mass of tagged particle level jets matched to a detector level jet
  TH2F            *fh2PtVsMassJetDetAll;             //!pT vs mass of all detector level jets
  TH2F            *fh2PtVsMassJetDetTagged;          //!pT vs mass of tagged detector level jets
  THnSparse       *fhnMassResponse;                  //!response matrix

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

  ClassDef(AliAnalysisTaskJetMassResponseDet, 1)
};
#endif
Commit	Line	Data
252cb00e	1	#ifndef ALIANALYSISTASKJETMASSRESPONSEDET_H
	2	#define ALIANALYSISTASKJETMASSRESPONSEDET_H
	3
	4	class TH1;
	5	class TH2;
	6	class TH3;
	7	class TH3F;
	8	class THnSparse;
	9	class TClonesArray;
	10	class TArrayI;
	11	class AliAnalysisManager;
	12	class AliJetContainer;
	13
	14	#include "AliAnalysisTaskEmcalJet.h"
	15
	16	class AliAnalysisTaskJetMassResponseDet : public AliAnalysisTaskEmcalJet {
	17	public:
	18	enum JetMassType {
	19	kRaw = 0, //mass form anti-kt 4-vector
	20	kDeriv = 1 //area based subtracted jet mass
	21	};
	22
	23	AliAnalysisTaskJetMassResponseDet();
	24	AliAnalysisTaskJetMassResponseDet(const char *name);
	25	virtual ~AliAnalysisTaskJetMassResponseDet();
	26
	27	void UserCreateOutputObjects();
	28	void Terminate(Option_t *option);
	29
	30	//Setters
	31	void SetJetContainerPart(Int_t c) { fContainerPart = c ; }
	32	void SetJetContainerDet(Int_t c) { fContainerDet = c ; }
	33	void SetJetMassType(JetMassType t) { fJetMassType = t ; }
	34
	35	protected:
	36	Bool_t RetrieveEventObjects();
	37	Bool_t Run();
	38	Bool_t FillHistograms();
	39
	40	Double_t GetJetMass(AliEmcalJet *jet);
	41
	42	Int_t fContainerPart; // particle level jets
	43	Int_t fContainerDet; // detector level jets
	44	JetMassType fJetMassType; // jet mass type to be used
	45
	46	TH2F *fh2PtVsMassJetPartAll; //!pT vs mass of all particle level jets
	47	TH2F *fh2PtVsMassJetPartMatch; //!pT vs mass of all particle level jets matched to a detector level jet
	48	TH2F *fh2PtVsMassJetPartTagged; //!pT vs mass of tagged particle level jets
	49	TH2F *fh2PtVsMassJetPartTaggedMatch; //!pT vs mass of tagged particle level jets matched to a detector level jet
	50	TH2F *fh2PtVsMassJetDetAll; //!pT vs mass of all detector level jets
	51	TH2F *fh2PtVsMassJetDetTagged; //!pT vs mass of tagged detector level jets
	52	THnSparse *fhnMassResponse; //!response matrix
	53
	54	private:
	55	AliAnalysisTaskJetMassResponseDet(const AliAnalysisTaskJetMassResponseDet&); // not implemented
	56	AliAnalysisTaskJetMassResponseDet &operator=(const AliAnalysisTaskJetMassResponseDet&); // not implemented
	57
	58	ClassDef(AliAnalysisTaskJetMassResponseDet, 1)
	59	};
	60	#endif
	61