AliAnalysisTaskEmcalDiJetAna.h

Go to the documentation of this file.

#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;                         
  TH2F             *fh2CentRhoScaled;                     
  TH3F             *fh3PtEtaPhiJetFull;                   
  TH3F             *fh3PtEtaPhiJetCharged;                

  THnSparse        *fhnDiJetVarsFull;                     
  THnSparse        *fhnDiJetVarsCh;                       
  THnSparse        *fhnDiJetVarsFullCharged;              
  THnSparse        *fhnMatchingFullCharged;               

  TH3F             *fh3DiJetKtNEFPtAssoc[4];              

  TH3F             *fCentCorrPtAssocCh[4];                
  TH3F             *fCentCorrPtAssocFuCh[4];              

  TH3F             *fAjPtAssocCentCh[4];                  
  TH3F             *fAjPtAssocCentFuCh[4];                

  TH3F             *fh3PtTrigKt1Kt2Ch;                    
  TH3F             *fh3PtTrigKt1Kt2FuCh;                  

  TH3F             *fh3PtTrigDPhi1DPhi2Ch;                
  TH3F             *fh3PtTrigDPhi1DPhi2FuCh;              

  TH3F             *fh3PtAssoc1PtAssoc2DPhi23Ch[4];       
  TH3F             *fh3PtAssoc1PtAssoc2DPhi23FuCh[4];     

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

  ClassDef(AliAnalysisTaskEmcalDiJetAna, 11) // dijet analysis task
};
#endif