AliEmcalJetByJetCorrection.h

Go to the documentation of this file.

#ifndef ALIEMCALJETBYJETCORRECTION_H
#define ALIEMCALJETBYJETCORRECTION_H

#include <TNamed.h>
#include <TObjArray.h>
#include <TArrayF.h>
#include <TClonesArray.h>

class TH3;
class TProfile;
class AliEmcalJet;
class TList;
class THnSparse;

class AliEmcalJetByJetCorrection : public TNamed
{
 public:

  AliEmcalJetByJetCorrection();
  AliEmcalJetByJetCorrection(const char* name);
  AliEmcalJetByJetCorrection(const AliEmcalJetByJetCorrection &jet);
  AliEmcalJetByJetCorrection& operator=(const AliEmcalJetByJetCorrection &jet);
  ~AliEmcalJetByJetCorrection() {;}

  void SetTemplate(TH3 *h)          { fh3JetPtDRTrackPt = h; }
  void SetJetPtBinWidth(Double_t w) { fBinWidthJetPt    = w; }
  void SetJetPtRange(Double_t min, Double_t max) {fJetPtMin = min; fJetPtMax = max;}
  void SetFixedTrackEfficiency(Double_t eff) { fEfficiencyFixed = eff; }
  void SetEfficiencyHist(TH1 *h)             { fhEfficiency     = h  ; }
  void SetCorrectTemplateTrackpT(Bool_t correct=kTRUE) {fCorrectpTtrack = correct;}
  void SetPoissonianNmissing(Bool_t set=kTRUE) {fNpPoisson=set;}
  void SetNMissedTracks(Double_t number) {fNMissedTracks = number;}
  void SetExternalDefinitionOfNmissed(Bool_t set=kTRUE) {fExternalNmissed=set;}
  
  Int_t              GetJetPtBin(const Double_t jetpt) const;
  Double_t           GetEfficiency(const Double_t pt) const;
  Double_t           GetMeanPtConstituents(const AliEmcalJet *jet, TClonesArray *fTracks) const;
                    
  TProfile          *GetAppliedEfficiency() const      {return fpAppliedEfficiency;}
  Bool_t             GetCorrectTemplateTrackpT() const {return fCorrectpTtrack;}
  Bool_t             GetPoissonianNmissing() const     {return fNpPoisson;}
  Bool_t             GetExternalDefinitionOfNmissed() const {return fExternalNmissed;}
  TList             *GetListOfOutput()       const     {return fListOfOutput;}
  void               Init();
  AliEmcalJet       *Eval(const AliEmcalJet *jet, TClonesArray *fTracks);
  TClonesArray      *GetAddedTrackArray()   const      {return fArrayTrackCorr;}
  
 protected:
  TH3      *fh3JetPtDRTrackPt;                 
  Double_t  fBinWidthJetPt;                    
  Double_t  fJetPtMin;                         
  Double_t  fJetPtMax;                         
  TObjArray fCollTemplates;                    
  Bool_t    fInitialized;                      
  Double_t  fEfficiencyFixed;                  
  TH1      *fhEfficiency;                      
  TH1D     *fhSmoothEfficiency;                
  Bool_t   fCorrectpTtrack;                    
  Bool_t   fNpPoisson;                         
  Bool_t   fExternalNmissed;                   
  TRandom3 *fRndm;                             
  Int_t    fNMissedTracks;                     
  //  -- now done in the analysis task from the difference of the integral of the pt distr of the constituents of the particle level jet and the reco level jet
  TClonesArray *fArrayTrackCorr;               
  TString  fPartArrayN;                        
  //book-keeping object filled inside Eval()
  TProfile *fpAppliedEfficiency;               
  THnSparse*fhNmissing;                        
  TList    *fListOfOutput;                     
 private:
  ClassDef(AliEmcalJetByJetCorrection, 8) // jet-by-jet correction class
};
#endif