AliAnalysisTaskEmcalDijetImbalance.h

Go to the documentation of this file.

#ifndef AliAnalysisTaskEmcalDijetImbalance_H
#define AliAnalysisTaskEmcalDijetImbalance_H

/* Copyright(c) 1998-2016, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

#include "THistManager.h"

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskEmcalDijetImbalance : public AliAnalysisTaskEmcalJet {
 public:
  
  // Struct to store all relevant info of a di-jet pair
  struct Dijet_t {
    Dijet_t() : leadingHadronCutType(0), trigJetMinPtType(0), assJetMinPtType(0), trigJet(0), trigJetPt(0), trigJetPhi(0), trigJetEta(0),
                assJet(0), assJetPt(0), assJetPhi(0), assJetEta(0), isAccepted(kFALSE), deltaPhi(0), deltaEta(0), AJ(0), xJ(0), kTy(0) {}
    
    Int_t leadingHadronCutType;
    Int_t trigJetMinPtType;
    Int_t assJetMinPtType;
    
    AliEmcalJet* trigJet;
    Double_t trigJetPt;
    Double_t trigJetPhi;
    Double_t trigJetEta;
    
    AliEmcalJet* assJet;
    Double_t assJetPt;
    Double_t assJetPhi;
    Double_t assJetEta;
    
    Bool_t isAccepted;
    
    Double_t deltaPhi;
    Double_t deltaEta;
    Double_t AJ;
    Double_t xJ;
    Double_t kTy;
    
    void clear() {leadingHadronCutType=0; trigJetMinPtType=0; assJetMinPtType=0; trigJet=0; trigJetPt=0; trigJetPhi=0; trigJetEta=0;
                    assJet=0; assJetPt=0; assJetPhi=0; assJetEta=0; isAccepted=kFALSE; deltaPhi=0; deltaEta=0; AJ=0; xJ=0; kTy=0;}
  };

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

  void UserCreateOutputObjects()                                                ;
  
  // Setters
  void SetDeltaPhiCut(Double_t d)                           { fDeltaPhiMin = d; }
  void SetMaxPt(Double_t d)                                 { fMaxPt = d; }
  void SetPlotJetHistograms(Bool_t b)                       { fPlotJetHistograms = b; }
  void SetPlotDijetJetHistograms(Bool_t b)                  { fPlotDijetJetHistograms = b; }
  void SetPlotDijetImbalanceHistograms(Bool_t b)            { fPlotDijetImbalanceHistograms = b; }
  void SetDoMomentumBalance(Bool_t b)                       { fDoMomentumBalance = b; }
  void SetDoGeometricalMatching(Bool_t b, Double_t r, Double_t trackThresh, Double_t clusThresh)
    { fDoGeometricalMatching = b; fMatchingJetR = r; fTrackConstituentThreshold = trackThresh; fClusterConstituentThreshold = clusThresh;}
  void SetNDijetPtThresholds(Int_t n)                       { fNDijetPtThresholds = n; }
  void SetMinTrigJetPt(Double_t* arr)                       { fMinTrigJetPt = arr; }
  void SetMinAssJetPt(Double_t* arr)                        { fMinAssJetPt = arr; }
  void SetDijetLeadingHadronPt(Double_t pt)                 { fDijetLeadingHadronPt = pt; }

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

  // Analysis and plotting functions
  void                        GenerateHistoBins()                               ;
  void                        AllocateJetHistograms()                           ;
  void                        AllocateDijetJetHistograms()                      ;
  void                        AllocateDijetImbalanceHistograms()                ;
  void                        AllocateMomentumBalanceHistograms()               ;
  void                        AllocateGeometricalMatchingHistograms()           ;
  void                        FindDijet(AliJetContainer* jetCont, Int_t leadingHadronCutBin, Int_t trigJetMinPtBin, Int_t assJetMinPtBin);
  void                        DoMomentumBalance(TString histname)               ;
  void                        DoGeometricalMatching();
  void                        FindMatchingDijet(AliJetContainer* jetCont, Int_t assJetMinPtBin);
  void                        FillJetHistograms()                               ;
  void                        FillDijetJetHistograms(TString histname, Int_t isAccepted, Int_t IsAssJet, Double_t jetPt, Double_t jetPhi,
                                                     Double_t jetEta, Int_t nTracksJet, Double_t jetArea);
  void                        FillDijetImbalanceHistograms(TString histname)    ;
  void                        FillMomentumBalanceHistograms(TString histname, Double_t deltaPhi, Double_t trackPt, Double_t balancePt);
  void                        FillGeometricalMatchingHistograms();
  
  // Utility functions
  Double_t                    GetJetPt(AliJetContainer* jetCont, AliEmcalJet* jet);
  Double_t                    GetDeltaR(AliEmcalJet* jet1, AliEmcalJet* jet2);
  
  // Analysis parameters
  Double_t                    fDeltaPhiMin;                         
  Int_t                       fNDijetPtThresholds;                  
  Double_t*                   fMinTrigJetPt;                        //[fNDijetPtThresholds] array of leading jet min pT's in a dijet pair
  Double_t*                   fMinAssJetPt;                         //[fNDijetPtThresholds] array of subleading jet min pT's in a dijet pair
  Double_t                    fDijetLeadingHadronPt;                
  Double_t                    fMatchingJetR;                        
  Double_t                    fTrackConstituentThreshold;           
  Double_t                    fClusterConstituentThreshold;         
  Dijet_t                     fDijet;                               
  Dijet_t                     fMatchingDijet;                       

  // Analysis configuration and plotting options
  Bool_t                      fPlotJetHistograms;                   
  Bool_t                      fPlotDijetJetHistograms;              
  Bool_t                      fPlotDijetImbalanceHistograms;        
  Bool_t                      fDoMomentumBalance;                   
  Bool_t                      fDoGeometricalMatching;               

  // Plotting parameters
  Float_t                     fMaxPt;                               
  Int_t                       fNCentHistBins;                       
  Double_t*                   fCentHistBins;                        
  
  THistManager                fHistManager;                         

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

  ClassDef(AliAnalysisTaskEmcalDijetImbalance, 3);
};
#endif