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

#ifndef ALIANALYSISTASKEMCALDIJETANA_H
#define ALIANALYSISTASKEMCALDIJETANA_H

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

#include "AliJetContainer.h"

#include "AliAnalysisTaskEmcalDiJetBase.h"

class AliAnalysisTaskEmcalDiJetAna : public AliAnalysisTaskEmcalDiJetBase {
 public:
  AliAnalysisTaskEmcalDiJetAna();
  AliAnalysisTaskEmcalDiJetAna(const char *name);
  virtual ~AliAnalysisTaskEmcalDiJetAna();

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

  //Setters
  void                        SetMatchFullCharged(Bool_t b)        { fDoMatchFullCharged = b;}
  void                        SetNKtBins(Int_t n)                  { fNKtBins = n;           }
  void                        SetNDiJetEtaBins(Int_t n)            { fNDiJetEtaBins = n;     }
  void                        SetNAjBins(Int_t n)                  { fNAjBins = n;           }

  //Getters
  Int_t                       GetPtTriggerBin(Double_t pt);

 protected:
  Bool_t                      Run()              ;
  void                        CorrelateJets(const Int_t type);
  void                        CorrelateAllJets(const Int_t type);
  void                        CorrelateTwoJets(const Int_t type);
  void                        CorrelateLeadingSubleadingJets(const Int_t type);
  AliEmcalJet                *GetLeadingJet(const Int_t type);
  AliEmcalJet                *GetLeadingAssociatedJet(const Int_t type, AliEmcalJet *jetTrig);
  AliEmcalJet                *GetSecondLeadingAssociatedJet(const Int_t type, AliEmcalJet *jetTrig);

  Bool_t                      FillHistograms()   ;
  void                        FillDiJetHistos(const AliEmcalJet *jet1 = 0, const AliEmcalJet *jet2 = 0, const Int_t mode = 0);
  void                        FillThreeJetHistos(const AliEmcalJet *jet1 = 0, const AliEmcalJet *jet2 = 0, const AliEmcalJet *jet3 = 0, const Int_t mode = 0);
  Bool_t                      RetrieveEventObjects();

  void                        FillMatchFullChargedHistos(Int_t cFull,Int_t cCharged);
  Int_t                       MatchFullAndChargedJets(Int_t cFull, Int_t cCharged);

 private:
  Bool_t            fDoMatchFullCharged;                  //  do full-charged matching histos
  Int_t             fNKtBins;                             // nbins on kT axis
  Int_t             fNDiJetEtaBins;                       // nbins on dijet eta axis
  Int_t             fNAjBins;                             // nbins on Aj axis
  TH2F             *fh2CentRhoCh;                         //! cent vs rho charged
  TH2F             *fh2CentRhoScaled;                     //! cent vs rho scaled
  TH3F             *fh3PtEtaPhiJetFull;                   //! pt,eta,phi of full jets
  TH3F             *fh3PtEtaPhiJetCharged;                //! pt,eta,phi of charged jets

  THnSparse        *fhnDiJetVarsFull;                     //! sparse with di-jet properties (full-full)
  THnSparse        *fhnDiJetVarsCh;                       //! sparse with di-jet properties (charged-charged)
  THnSparse        *fhnDiJetVarsFullCharged;              //! sparse with di-jet properties (full-charged)
  THnSparse        *fhnMatchingFullCharged;               //! sparse comparing full with matched charged jet

  TH3F             *fh3DiJetKtNEFPtAssoc[4];              //! dijet kt vs NEF vs pTassoc for 4 trigger intervals

  TH3F             *fCentCorrPtAssocCh[4];                //! default(V0A) vs ZNA centrality vs pT trigger assoc
  TH3F             *fCentCorrPtAssocFuCh[4];              //! default(V0A) vs ZNA centrality vs pT trigger assoc

  TH3F             *fAjPtAssocCentCh[4];                  //! Aj vs pT trigger assoc vs centrality
  TH3F             *fAjPtAssocCentFuCh[4];                //! Aj vs pT trigger assoc vs centrality

  TH3F             *fh3PtTrigKt1Kt2Ch;                    //! ptTrig vs kT1 vs kT2 for 3-jet events
  TH3F             *fh3PtTrigKt1Kt2FuCh;                  //! ptTrig vs kT1 vs kT2 for 3-jet events

  TH3F             *fh3PtTrigDPhi1DPhi2Ch;                //! ptTrig vs DPhi12 vs DPhi13 for 3-jet events
  TH3F             *fh3PtTrigDPhi1DPhi2FuCh;              //! ptTrig vs DPhi12 vs DPhi13 for 3-jet events

  TH3F             *fh3PtAssoc1PtAssoc2DPhi23Ch[4];       //! ptAssoc1 vs ptAssoc2 vs DPhi23 for 3-jet events
  TH3F             *fh3PtAssoc1PtAssoc2DPhi23FuCh[4];     //! ptAssoc1 vs ptAssoc2 vs DPhi23 for 3-jet events

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

  ClassDef(AliAnalysisTaskEmcalDiJetAna, 11) // dijet analysis task
};
#endif
Commit	Line	Data
6e8b6371	1	#ifndef ALIANALYSISTASKEMCALDIJETANA_H
	2	#define ALIANALYSISTASKEMCALDIJETANA_H
	3
	4	class TH1;
	5	class TH2;
	6	class TH3;
	7	class TH3F;
	8	class THnSparse;
	9	class TClonesArray;
	10	class TArrayI;
6e8b6371	11	class AliAnalysisManager;
	12	class AliGenPythiaEventHeader;
	13
	14	#include "AliJetContainer.h"
	15
	16	#include "AliAnalysisTaskEmcalDiJetBase.h"
	17
	18	class AliAnalysisTaskEmcalDiJetAna : public AliAnalysisTaskEmcalDiJetBase {
	19	public:
	20	AliAnalysisTaskEmcalDiJetAna();
	21	AliAnalysisTaskEmcalDiJetAna(const char *name);
	22	virtual ~AliAnalysisTaskEmcalDiJetAna();
	23
	24	void UserCreateOutputObjects();
	25	void Terminate(Option_t *option);
	26
	27	//Setters
bb84b374	28	void SetMatchFullCharged(Bool_t b) { fDoMatchFullCharged = b;}
677e90b9	29	void SetNKtBins(Int_t n) { fNKtBins = n; }
a29bc208 CL	30	void SetNDiJetEtaBins(Int_t n) { fNDiJetEtaBins = n; }
a29bc208 CL	31	void SetNAjBins(Int_t n) { fNAjBins = n; }
6e8b6371	32
6e8b6371	33	//Getters
bb84b374	34	Int_t GetPtTriggerBin(Double_t pt);
6e8b6371	35
	36	protected:
	37	Bool_t Run() ;
	38	void CorrelateJets(const Int_t type);
c66364a6	39	void CorrelateAllJets(const Int_t type);
4bd08270	40	void CorrelateTwoJets(const Int_t type);
c66364a6	41	void CorrelateLeadingSubleadingJets(const Int_t type);
	42	AliEmcalJet *GetLeadingJet(const Int_t type);
	43	AliEmcalJet GetLeadingAssociatedJet(const Int_t type, AliEmcalJet jetTrig);
8e49a788	44	AliEmcalJet GetSecondLeadingAssociatedJet(const Int_t type, AliEmcalJet jetTrig);
4bd08270	45
6e8b6371	46	Bool_t FillHistograms() ;
6e8b6371	47	void FillDiJetHistos(const AliEmcalJet jet1 = 0, const AliEmcalJet jet2 = 0, const Int_t mode = 0);
8e49a788	48	void FillThreeJetHistos(const AliEmcalJet jet1 = 0, const AliEmcalJet jet2 = 0, const AliEmcalJet *jet3 = 0, const Int_t mode = 0);
6e8b6371	49	Bool_t RetrieveEventObjects();
	50
	51	void FillMatchFullChargedHistos(Int_t cFull,Int_t cCharged);
	52	Int_t MatchFullAndChargedJets(Int_t cFull, Int_t cCharged);
	53
	54	private:
	55	Bool_t fDoMatchFullCharged; // do full-charged matching histos
677e90b9	56	Int_t fNKtBins; // nbins on kT axis
	57	Int_t fNDiJetEtaBins; // nbins on dijet eta axis
	58	Int_t fNAjBins; // nbins on Aj axis
6e8b6371	59	TH2F *fh2CentRhoCh; //! cent vs rho charged
	60	TH2F *fh2CentRhoScaled; //! cent vs rho scaled
	61	TH3F *fh3PtEtaPhiJetFull; //! pt,eta,phi of full jets
	62	TH3F *fh3PtEtaPhiJetCharged; //! pt,eta,phi of charged jets
	63
	64	THnSparse *fhnDiJetVarsFull; //! sparse with di-jet properties (full-full)
	65	THnSparse *fhnDiJetVarsCh; //! sparse with di-jet properties (charged-charged)
	66	THnSparse *fhnDiJetVarsFullCharged; //! sparse with di-jet properties (full-charged)
	67	THnSparse *fhnMatchingFullCharged; //! sparse comparing full with matched charged jet
6e8b6371	68
bb84b374	69	TH3F *fh3DiJetKtNEFPtAssoc[4]; //! dijet kt vs NEF vs pTassoc for 4 trigger intervals
bb84b374	70
eca9052c	71	TH3F *fCentCorrPtAssocCh[4]; //! default(V0A) vs ZNA centrality vs pT trigger assoc
	72	TH3F *fCentCorrPtAssocFuCh[4]; //! default(V0A) vs ZNA centrality vs pT trigger assoc
	73
0327741d	74	TH3F *fAjPtAssocCentCh[4]; //! Aj vs pT trigger assoc vs centrality
	75	TH3F *fAjPtAssocCentFuCh[4]; //! Aj vs pT trigger assoc vs centrality
	76
8e49a788	77	TH3F *fh3PtTrigKt1Kt2Ch; //! ptTrig vs kT1 vs kT2 for 3-jet events
	78	TH3F *fh3PtTrigKt1Kt2FuCh; //! ptTrig vs kT1 vs kT2 for 3-jet events
	79
	80	TH3F *fh3PtTrigDPhi1DPhi2Ch; //! ptTrig vs DPhi12 vs DPhi13 for 3-jet events
	81	TH3F *fh3PtTrigDPhi1DPhi2FuCh; //! ptTrig vs DPhi12 vs DPhi13 for 3-jet events
	82
	83	TH3F *fh3PtAssoc1PtAssoc2DPhi23Ch[4]; //! ptAssoc1 vs ptAssoc2 vs DPhi23 for 3-jet events
	84	TH3F *fh3PtAssoc1PtAssoc2DPhi23FuCh[4]; //! ptAssoc1 vs ptAssoc2 vs DPhi23 for 3-jet events
6e8b6371	85
	86	AliAnalysisTaskEmcalDiJetAna(const AliAnalysisTaskEmcalDiJetAna&); // not implemented
	87	AliAnalysisTaskEmcalDiJetAna &operator=(const AliAnalysisTaskEmcalDiJetAna&); // not implemented
	88
677e90b9	89	ClassDef(AliAnalysisTaskEmcalDiJetAna, 11) // dijet analysis task
6e8b6371	90	};
6e8b6371	91	#endif