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

#ifndef ALIANALYSISTASKEMCALJETMASS_H
#define ALIANALYSISTASKEMCALJETMASS_H

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

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskEmcalJetMass : public AliAnalysisTaskEmcalJet {
 public:

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

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

  //Setters
  void SetJetContainerBase(Int_t c)                             { fContainerBase     = c   ; }
  void SetMinFractionShared(Double_t f)                         { fMinFractionShared = f   ; }
  void SetJetMassAverage(Double_t m)                            { fJetMassAvg        = m   ; }

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

  Double_t                            GetJetMass(AliEmcalJet *jet);
  TLorentzVector                      GetSubtractedVector(AliEmcalJet *jet);
  TLorentzVector                      GetBkgVector(AliEmcalJet *jet, AliJetContainer *cont);
 
  Int_t                               fContainerBase;              // jets to be analyzed
  Double_t                            fMinFractionShared;          // only fill histos for jets if shared fraction larger than X
  Double_t                            fJetMassAvg;                 // average jet mass

  TH2F            **fh2PtJet1VsLeadPtAllSel;         //!all jets after std selection vs leading track pt
  TH2F            **fh2PtJet1VsLeadPtTagged;         //!tagged jets vs leading track pt
  TH2F            **fh2PtJet1VsLeadPtTaggedMatch;    //!tagged jets vs leading track pt matched to MC
  TH2F            **fh2PtVsMassJet1All;              //!pT vs mass of all jets
  TH2F            **fh2PtVsMassJet1Tagged;           //!pT vs mass of tagged jets
  TH2F            **fh2PtVsMassJet1TaggedMatch;      //!pT vs mass of tagged jets matched to MC
  TProfile        **fpPtVsMassJet1All;               //!pT vs avg mass of all jets
  TProfile        **fpPtVsMassJet1Tagged;            //!pT vs avg mass of tagged jets
  TProfile        **fpPtVsMassJet1TaggedMatch;       //!pT vs avg mass of tagged jets matched to MC
  TH2F            **fh2MassVsAreaJet1All;            //!mass vs area of all jets
  TH2F            **fh2MassVsAreaJet1Tagged;         //!mass vs area of tagged jets
  TH2F            **fh2MassVsAreaJet1TaggedMatch;    //!mass vs area of tagged jets matched to MC
  TH2F            **fh2MassVsNConstJet1All;          //!mass vs number of constituents of all jets
  TH2F            **fh2MassVsNConstJet1Tagged;       //!mass vs number of constituents of tagged jets
  TH2F            **fh2MassVsNConstJet1TaggedMatch;  //!mass vs number of constituents of tagged jets matched to MC
  TH2F            **fh2EtMassOverEtRSq;           //!Et vs (M/Et*R)^2

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

  ClassDef(AliAnalysisTaskEmcalJetMass, 4)
};
#endif
Commit	Line	Data
8612dfc8	1	#ifndef ALIANALYSISTASKEMCALJETMASS_H
	2	#define ALIANALYSISTASKEMCALJETMASS_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 AliAnalysisTaskEmcalJetMass : public AliAnalysisTaskEmcalJet {
	17	public:
	18
	19	AliAnalysisTaskEmcalJetMass();
	20	AliAnalysisTaskEmcalJetMass(const char *name);
	21	virtual ~AliAnalysisTaskEmcalJetMass();
	22
	23	void UserCreateOutputObjects();
	24	void Terminate(Option_t *option);
	25
	26	//Setters
2e31a073	27	void SetJetContainerBase(Int_t c) { fContainerBase = c ; }
ba7663ad	28	void SetMinFractionShared(Double_t f) { fMinFractionShared = f ; }
2e31a073	29	void SetJetMassAverage(Double_t m) { fJetMassAvg = m ; }
8612dfc8	30
	31	protected:
	32	Bool_t RetrieveEventObjects();
	33	Bool_t Run();
	34	Bool_t FillHistograms();
	35
2e31a073	36	Double_t GetJetMass(AliEmcalJet *jet);
4d3b366f	37	TLorentzVector GetSubtractedVector(AliEmcalJet *jet);
	38	TLorentzVector GetBkgVector(AliEmcalJet jet, AliJetContainer cont);
	39
ba7663ad	40	Int_t fContainerBase; // jets to be analyzed
ba7663ad	41	Double_t fMinFractionShared; // only fill histos for jets if shared fraction larger than X
2e31a073	42	Double_t fJetMassAvg; // average jet mass
2e31a073	43
4d3b366f	44	TH2F **fh2PtJet1VsLeadPtAllSel; //!all jets after std selection vs leading track pt
	45	TH2F **fh2PtJet1VsLeadPtTagged; //!tagged jets vs leading track pt
	46	TH2F **fh2PtJet1VsLeadPtTaggedMatch; //!tagged jets vs leading track pt matched to MC
	47	TH2F **fh2PtVsMassJet1All; //!pT vs mass of all jets
	48	TH2F **fh2PtVsMassJet1Tagged; //!pT vs mass of tagged jets
	49	TH2F **fh2PtVsMassJet1TaggedMatch; //!pT vs mass of tagged jets matched to MC
	50	TProfile **fpPtVsMassJet1All; //!pT vs avg mass of all jets
	51	TProfile **fpPtVsMassJet1Tagged; //!pT vs avg mass of tagged jets
	52	TProfile **fpPtVsMassJet1TaggedMatch; //!pT vs avg mass of tagged jets matched to MC
	53	TH2F **fh2MassVsAreaJet1All; //!mass vs area of all jets
	54	TH2F **fh2MassVsAreaJet1Tagged; //!mass vs area of tagged jets
	55	TH2F **fh2MassVsAreaJet1TaggedMatch; //!mass vs area of tagged jets matched to MC
	56	TH2F **fh2MassVsNConstJet1All; //!mass vs number of constituents of all jets
	57	TH2F **fh2MassVsNConstJet1Tagged; //!mass vs number of constituents of tagged jets
	58	TH2F **fh2MassVsNConstJet1TaggedMatch; //!mass vs number of constituents of tagged jets matched to MC
02e33ee8	59	TH2F *fh2EtMassOverEtRSq; //!Et vs (M/EtR)^2
8612dfc8	60
655c03b1	61	private:
8612dfc8	62	AliAnalysisTaskEmcalJetMass(const AliAnalysisTaskEmcalJetMass&); // not implemented
	63	AliAnalysisTaskEmcalJetMass &operator=(const AliAnalysisTaskEmcalJetMass&); // not implemented
	64
4d3b366f	65	ClassDef(AliAnalysisTaskEmcalJetMass, 4)
8612dfc8	66	};
	67	#endif
	68