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

#ifndef ALIANALYSISTASKEMCALJETMASSBKG_H
#define ALIANALYSISTASKEMCALJETMASSBKG_H

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

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskEmcalJetMassBkg : public AliAnalysisTaskEmcalJet {
 public:

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

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

  //Setters
  void                        SetJetContainerBase(Int_t c)                         { fContainerBase           = c          ; }

  void                        SetJetMinRC2LJ(Float_t d)                            { fMinRC2LJ                = d          ; }
  void                        SetRCperEvent(Int_t n)                               { fRCperEvent              = n          ; }
  void                        SetConeRadius(Double_t r)                            { fConeRadius              = r          ; }
  void                        SetConeEtaPhiEMCAL() ;
  void                        SetConeEtaPhiTPC()   ;
  void                        SetConeEtaLimits(Float_t min, Float_t max)           { fConeMinEta = min, fConeMaxEta = max  ; }
  void                        SetConePhiLimits(Float_t min, Float_t max)           { fConeMinPhi = min, fConeMaxPhi = max  ; }

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

  void                        GetRandomCone(TLorentzVector& lvRC, Float_t &pt, Float_t &eta, Float_t &phi, AliParticleContainer* tracks, AliClusterContainer* clusters, AliEmcalJet *jet = 0) const;
  void                        GetCone(TLorentzVector& lvRC,Float_t &pt, Float_t eta, Float_t phi, AliParticleContainer* tracks, AliClusterContainer* clusters) const;
  void                        GetPerpCone(TLorentzVector& lvRC, Float_t &pt, Float_t &eta, Float_t &phi, AliParticleContainer* tracks, AliClusterContainer* clusters, AliEmcalJet *jet = 0) const;

  Int_t                       fContainerBase;              // jets to be tagged

  Float_t                     fMinRC2LJ;                   // Minimum distance random cone to leading jet
  Int_t                       fRCperEvent;                 // No. of random cones per event
  Double_t                    fConeRadius;                 // Radius of the random cones
  Float_t                     fConeMinEta;                 // Minimum eta of the random cones
  Float_t                     fConeMaxEta;                 // Maximum eta of the random cones
  Float_t                     fConeMinPhi;                 // Minimum phi of the random cones
  Float_t                     fConeMaxPhi;                 // Maximum phi of the random cones
  
  AliJetContainer            *fJetsCont;                   //!Jets
  AliParticleContainer       *fTracksCont;                 //!Tracks
  AliClusterContainer        *fCaloClustersCont;           //!Clusters  

 private:
  TH2F            **fh2PtVsMassRC;                //!pT vs mass of RC
  TH3F            **fh2PtVsMassRCExLJDPhi;        //!pT vs mass of RC excluding area around leading jet
  TH2F            **fh2PtVsMassPerpConeLJ;        //!pT vs mass of cone perpendicular to leading jet
  TH2F            **fh2PtVsMassPerpConeTJ;        //!pT vs mass of cone perpendicular to all tagged jets

  TH2F            **fh2PtVsERC;                   //!E vs mass of RC
  TH3F            **fh2PtVsERCExLJDPhi;           //!E vs mass of RC excluding area around leading jet
  TH2F            **fh2PtVsEPerpConeLJ;           //!E vs mass of cone perpendicular to leading jet
  TH2F            **fh2PtVsEPerpConeTJ;           //!E vs mass of cone perpendicular to all tagged jets

  TProfile        **fpPtVsMassRC;                 //!pT vs Avg mass of RC
  TProfile        **fpPtVsMassRCExLJ;             //!pT vs Avg mass of RC excluding area around leading jet
  TProfile        **fpPtVsMassPerpConeLJ;         //!pT vs Avg mass of cone perpendicular to leading jet
  TProfile        **fpPtVsMassPerpConeTJ;         //!pT vs Avg mass of cone perpendicular to all tagged jets

  TH2F            **fh2EtaVsMassRC;               //!eta vs mass of RC
  TH2F            **fh2EtaVsMassRCExLJ;           //!eta vs mass of RC excluding area around leading jet
  TH2F            **fh2EtaVsMassPerpConeLJ;       //!eta vs mass of cone perpendicular to leading jet
  TH2F            **fh2EtaVsMassPerpConeTJ;       //!eta vs mass of cone perpendicular to all tagged jets

  TH2F             *fh2CentVsMassRC;              //!cent vs mass of RC
  TH2F             *fh2CentVsMassRCExLJ;          //!cent vs mass of RC excluding area around leading jet
  TH2F             *fh2CentVsMassPerpConeLJ;      //!cent vs mass of RC perpendicular to leading jet
  TH2F             *fh2CentVsMassPerpConeTJ;      //!cent vs mass of RC perpendicular to tagged jets

  TH2F             *fh2MultVsMassRC;              //!track multiplicity vs mass of RC
  TH2F             *fh2MultVsMassRCExLJ;          //!track multiplicity vs mass of RC excluding area around leading jet
  TH2F             *fh2MultVsMassPerpConeLJ;      //!track multiplicity vs mass of RC perpendicular to leading jet
  TH2F             *fh2MultVsMassPerpConeTJ;      //!track multiplicity vs mass of RC perpendicular to tagged jets

  TH2F             *fh2CentVsMedianMassRC;        //!cent vs median mass of all RC in event
  TH2F             *fh2CentVsMedianMassRCExLJ;    //!cent vs meidan mass mass of all RC in event excluding area around leading jet

  TH2F             *fh2MultVsMedianMassRC;        //!cent vs median mass of all RC in event
  TH2F             *fh2MultVsMedianMassRCExLJ;    //!cent vs meidan mass mass of all RC in event excluding area around leading jet

  TH2F             *fh2CentVsMeanMassRC;          //!cent vs median mass of all RC in event
  TH2F             *fh2CentVsMeanMassRCExLJ;      //!cent vs meidan mass mass of all RC in event excluding area around leading jet

  TH2F             *fh2MultVsMeanMassRC;          //!cent vs median mass of all RC in event
  TH2F             *fh2MultVsMeanMassRCExLJ;      //!cent vs meidan mass mass of all RC in event excluding area around leading jet

  TH2F             *fh2CentVsMedianMassPerAreaRC;     //!cent vs median mass of all RC in event
  TH2F             *fh2CentVsMedianMassPerAreaRCExLJ; //!cent vs meidan mass mass of all RC in event excluding area around leading jet

  TH2F             *fh2MultVsMedianMassPerAreaRC;     //!cent vs median mass of all RC in event
  TH2F             *fh2MultVsMedianMassPerAreaRCExLJ; //!cent vs meidan mass mass of all RC in event excluding area around leading jet

  AliAnalysisTaskEmcalJetMassBkg(const AliAnalysisTaskEmcalJetMassBkg&);            // not implemented
  AliAnalysisTaskEmcalJetMassBkg &operator=(const AliAnalysisTaskEmcalJetMassBkg&); // not implemented

  ClassDef(AliAnalysisTaskEmcalJetMassBkg, 4)
};
#endif
Commit	Line	Data
8612dfc8	1	#ifndef ALIANALYSISTASKEMCALJETMASSBKG_H
	2	#define ALIANALYSISTASKEMCALJETMASSBKG_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 AliAnalysisTaskEmcalJetMassBkg : public AliAnalysisTaskEmcalJet {
	17	public:
	18
	19	AliAnalysisTaskEmcalJetMassBkg();
	20	AliAnalysisTaskEmcalJetMassBkg(const char *name);
	21	virtual ~AliAnalysisTaskEmcalJetMassBkg();
	22
	23	void UserCreateOutputObjects();
	24	void Terminate(Option_t *option);
	25
	26	//Setters
	27	void SetJetContainerBase(Int_t c) { fContainerBase = c ; }
	28
	29	void SetJetMinRC2LJ(Float_t d) { fMinRC2LJ = d ; }
	30	void SetRCperEvent(Int_t n) { fRCperEvent = n ; }
	31	void SetConeRadius(Double_t r) { fConeRadius = r ; }
	32	void SetConeEtaPhiEMCAL() ;
	33	void SetConeEtaPhiTPC() ;
	34	void SetConeEtaLimits(Float_t min, Float_t max) { fConeMinEta = min, fConeMaxEta = max ; }
	35	void SetConePhiLimits(Float_t min, Float_t max) { fConeMinPhi = min, fConeMaxPhi = max ; }
	36
	37	protected:
	38	void ExecOnce();
	39	Bool_t RetrieveEventObjects();
	40	Bool_t Run();
	41	Bool_t FillHistograms();
	42
	43	void GetRandomCone(TLorentzVector& lvRC, Float_t &pt, Float_t &eta, Float_t &phi, AliParticleContainer* tracks, AliClusterContainer* clusters, AliEmcalJet *jet = 0) const;
	44	void GetCone(TLorentzVector& lvRC,Float_t &pt, Float_t eta, Float_t phi, AliParticleContainer* tracks, AliClusterContainer* clusters) const;
	45	void GetPerpCone(TLorentzVector& lvRC, Float_t &pt, Float_t &eta, Float_t &phi, AliParticleContainer* tracks, AliClusterContainer* clusters, AliEmcalJet *jet = 0) const;
	46
	47	Int_t fContainerBase; // jets to be tagged
	48
	49	Float_t fMinRC2LJ; // Minimum distance random cone to leading jet
	50	Int_t fRCperEvent; // No. of random cones per event
	51	Double_t fConeRadius; // Radius of the random cones
	52	Float_t fConeMinEta; // Minimum eta of the random cones
	53	Float_t fConeMaxEta; // Maximum eta of the random cones
	54	Float_t fConeMinPhi; // Minimum phi of the random cones
	55	Float_t fConeMaxPhi; // Maximum phi of the random cones
	56
	57	AliJetContainer *fJetsCont; //!Jets
	58	AliParticleContainer *fTracksCont; //!Tracks
	59	AliClusterContainer *fCaloClustersCont; //!Clusters
	60
	61	private:
	62	TH2F **fh2PtVsMassRC; //!pT vs mass of RC
ef7f972e	63	TH3F **fh2PtVsMassRCExLJDPhi; //!pT vs mass of RC excluding area around leading jet
	64	TH2F **fh2PtVsMassPerpConeLJ; //!pT vs mass of cone perpendicular to leading jet
	65	TH2F **fh2PtVsMassPerpConeTJ; //!pT vs mass of cone perpendicular to all tagged jets
	66
bda6a83f	67	TH2F **fh2PtVsERC; //!E vs mass of RC
	68	TH3F **fh2PtVsERCExLJDPhi; //!E vs mass of RC excluding area around leading jet
	69	TH2F **fh2PtVsEPerpConeLJ; //!E vs mass of cone perpendicular to leading jet
	70	TH2F **fh2PtVsEPerpConeTJ; //!E vs mass of cone perpendicular to all tagged jets
	71
8612dfc8	72	TProfile **fpPtVsMassRC; //!pT vs Avg mass of RC
8612dfc8	73	TProfile **fpPtVsMassRCExLJ; //!pT vs Avg mass of RC excluding area around leading jet
8612dfc8	74	TProfile **fpPtVsMassPerpConeLJ; //!pT vs Avg mass of cone perpendicular to leading jet
8612dfc8	75	TProfile **fpPtVsMassPerpConeTJ; //!pT vs Avg mass of cone perpendicular to all tagged jets
8612dfc8	76
ef7f972e	77	TH2F **fh2EtaVsMassRC; //!eta vs mass of RC
	78	TH2F **fh2EtaVsMassRCExLJ; //!eta vs mass of RC excluding area around leading jet
	79	TH2F **fh2EtaVsMassPerpConeLJ; //!eta vs mass of cone perpendicular to leading jet
	80	TH2F **fh2EtaVsMassPerpConeTJ; //!eta vs mass of cone perpendicular to all tagged jets
	81
8612dfc8	82	TH2F *fh2CentVsMassRC; //!cent vs mass of RC
	83	TH2F *fh2CentVsMassRCExLJ; //!cent vs mass of RC excluding area around leading jet
	84	TH2F *fh2CentVsMassPerpConeLJ; //!cent vs mass of RC perpendicular to leading jet
	85	TH2F *fh2CentVsMassPerpConeTJ; //!cent vs mass of RC perpendicular to tagged jets
	86
	87	TH2F *fh2MultVsMassRC; //!track multiplicity vs mass of RC
	88	TH2F *fh2MultVsMassRCExLJ; //!track multiplicity vs mass of RC excluding area around leading jet
	89	TH2F *fh2MultVsMassPerpConeLJ; //!track multiplicity vs mass of RC perpendicular to leading jet
	90	TH2F *fh2MultVsMassPerpConeTJ; //!track multiplicity vs mass of RC perpendicular to tagged jets
	91
396ffc82 MV	92	TH2F *fh2CentVsMedianMassRC; //!cent vs median mass of all RC in event
	93	TH2F *fh2CentVsMedianMassRCExLJ; //!cent vs meidan mass mass of all RC in event excluding area around leading jet
	94
	95	TH2F *fh2MultVsMedianMassRC; //!cent vs median mass of all RC in event
	96	TH2F *fh2MultVsMedianMassRCExLJ; //!cent vs meidan mass mass of all RC in event excluding area around leading jet
	97
	98	TH2F *fh2CentVsMeanMassRC; //!cent vs median mass of all RC in event
	99	TH2F *fh2CentVsMeanMassRCExLJ; //!cent vs meidan mass mass of all RC in event excluding area around leading jet
	100
	101	TH2F *fh2MultVsMeanMassRC; //!cent vs median mass of all RC in event
	102	TH2F *fh2MultVsMeanMassRCExLJ; //!cent vs meidan mass mass of all RC in event excluding area around leading jet
	103
	104	TH2F *fh2CentVsMedianMassPerAreaRC; //!cent vs median mass of all RC in event
	105	TH2F *fh2CentVsMedianMassPerAreaRCExLJ; //!cent vs meidan mass mass of all RC in event excluding area around leading jet
	106
	107	TH2F *fh2MultVsMedianMassPerAreaRC; //!cent vs median mass of all RC in event
	108	TH2F *fh2MultVsMedianMassPerAreaRCExLJ; //!cent vs meidan mass mass of all RC in event excluding area around leading jet
	109
8612dfc8	110	AliAnalysisTaskEmcalJetMassBkg(const AliAnalysisTaskEmcalJetMassBkg&); // not implemented
	111	AliAnalysisTaskEmcalJetMassBkg &operator=(const AliAnalysisTaskEmcalJetMassBkg&); // not implemented
	112
bda6a83f	113	ClassDef(AliAnalysisTaskEmcalJetMassBkg, 4)
8612dfc8	114	};
	115	#endif
	116