AliAnalysisTaskEmcalJetHMEC.h

Go to the documentation of this file.

#ifndef AliAnalysisTaskEmcalJetHMEC_H
#define AliAnalysisTaskEmcalJetHMEC_H

class TH1;
class TH2;
class TH3;
class THnSparse;
class AliEmcalJet;
class AliEventPoolManager;
class AliTLorentzVector;

#include "AliAnalysisTaskEmcalJet.h"

class AliAnalysisTaskEmcalJetHMEC : public AliAnalysisTaskEmcalJet {
 public:
  enum jetBias_t {
    kDisableBias = 10000            
  };

  AliAnalysisTaskEmcalJetHMEC();
  AliAnalysisTaskEmcalJetHMEC(const char *name);
  virtual ~AliAnalysisTaskEmcalJetHMEC() {}

  Double_t                GetTrackBias() const                       { return fTrackBias; }
  Double_t                GetClusterBias() const                     { return fClusterBias; }

  // Jet bias - setters
  virtual void            SetTrackBias(Double_t b)                   { fTrackBias    = b; }
  virtual void            SetClusterBias(Double_t b)                 { fClusterBias = b; }

  // Event trigger/mixed selection - setters
  virtual void            SetTriggerType(UInt_t te)                  { fTriggerType = te; }
  virtual void            SetMixedEventTriggerType(UInt_t me)        { fMixingEventType = me; }
  void                    SetDisableFastPartition(Bool_t b = kTRUE)  { fDisableFastPartition = b; }
  Bool_t                  GetDisableFastPartition() const            { return fDisableFastPartition; }
  // Mixed events
  virtual void            SetEventMixing(Bool_t enable)              { fDoEventMixing = enable;}
  virtual void            SetNumberOfMixingTracks(Int_t tracks)      { fNMixingTracks = tracks; }
  virtual void            SetMinNTracksForMixedEvents(Int_t nmt)     { fMinNTracksMixedEvents = nmt; }
  virtual void            SetMinNEventsForMixedEvents(Int_t nme)     { fMinNEventsMixedEvents = nme; }
  virtual void            SetNCentBinsMixedEvent(Bool_t centbins)    { fNCentBinsMixedEvent = centbins; }
  // Switch to cut out some unneeded sparse axis
  void                    SetDoLessSparseAxes(Bool_t dlsa)           { fDoLessSparseAxes = dlsa; }
  void                    SetDoWiderTrackBin(Bool_t wtrbin)          { fDoWiderTrackBin = wtrbin; }
  // set efficiency correction
  void                    SetDoEffCorr(Int_t effcorr)                { fDoEffCorrection = effcorr; }
  // Setup JES correction
  void                    SetJESCorrectionHist(TH2D * hist)          { fJESCorrectionHist = hist; }
  void                    SetNoMixedEventJESCorrection(Bool_t b) { fNoMixedEventJESCorrection = b; }

  Bool_t                  RetrieveAndInitializeJESCorrectionHist(TString filename, TString histName, Double_t trackBias = AliAnalysisTaskEmcalJetHMEC::kDisableBias, Double_t clusterBias = AliAnalysisTaskEmcalJetHMEC::kDisableBias);

  // Corrections
  // Public so that it can be tested externally
  Double_t       EffCorrection(Double_t trkETA, Double_t trkPT, AliAnalysisTaskEmcal::BeamType beamType) const;

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

  // AddTask
  static AliAnalysisTaskEmcalJetHMEC * AddTaskEmcalJetHMEC(
     const char *nTracks              = "usedefault",
     const char *nCaloClusters        = "usedefault",
     // Jet options
     const Double_t trackBias         = 5,
     const Double_t clusterBias       = 5,
     const Double_t minJetArea        = 0.4,
     // Mixed event options
     const Int_t nTracksMixedEvent    = 0,  // Additionally acts as a switch for enabling mixed events
     const Int_t minNTracksMixedEvent = 5000,
     const Int_t minNEventsMixedEvent = 5,
     const UInt_t nCentBinsMixedEvent = 10,
     // Triggers
     UInt_t trigEvent                 = AliVEvent::kAny,
     UInt_t mixEvent                  = AliVEvent::kAny,
     // Options
     const Double_t trackEta          = 0.9,
     const Bool_t lessSparseAxes      = 0,
     const Bool_t widerTrackBin       = 0,
     // Corrections
     const Int_t doEffCorrSW          = 0,
     const Bool_t JESCorrection = kFALSE,
     const char * JESCorrectionFilename = "alien:///alice/cern.ch/user/r/rehlersi/JESCorrection.root",
     const char * JESCorrectionHistName = "JESCorrection",
     const char *suffix               = "biased"
   );

 protected:

  // NOTE: This is not an ideal way to resolve the size of histogram initialization.
  //       Will be resolved when we move fully to the THnSparse
  enum binArrayLimits_t {
    kMaxJetPtBins = 5,              
    kMaxTrackPtBins = 7,            
    kMaxCentralityBins = 5,         
    kMixedEventMulitplictyBins = 8, 
    kMaxEtaBins = 3                 
  };

  // EMCal framework functions
  void                   ExecOnce();
  Bool_t                 Run();

  // Reduce event mixing memory usage
  TObjArray*             CloneAndReduceTrackList();
  // Histogram helper functions
  virtual THnSparse*     NewTHnSparseF(const char* name, UInt_t entries);
  virtual void           GetDimParams(Int_t iEntry,TString &label, Int_t &nbins, Double_t &xmin, Double_t &xmax);
  // Binning helper functions
  Int_t                  GetEtaBin(Double_t eta) const;
  Int_t                  GetTrackPtBin(Double_t pt) const;
  Int_t                  GetJetPtBin(Double_t pt) const;
  // Helper functions
  void                   InitializeArraysToZero();
  void                   GetDeltaEtaDeltaPhiDeltaR(AliTLorentzVector & particleOne, AliVParticle * particleTwo, Double_t & deltaEta, Double_t & deltaPhi, Double_t & deltaR);
  // Test for biased jet
  Bool_t                 BiasedJet(AliEmcalJet * jet);
  // Corrections
  Double_t               EffCorrection(Double_t trkETA, Double_t trkPT) const;

  // Fill methods which allow for the JES correction
  void                   FillHist(TH1 * hist, Double_t fillValue, Double_t weight = 1.0, Bool_t noCorrection = kFALSE);
  void                   FillHist(THnSparse * hist, Double_t *fillValue, Double_t weight = 1.0, Bool_t noCorrection = kFALSE);
  void                   AccessSetOfYBinValues(TH2D * hist, Int_t xBin, std::vector <Double_t> & yBinsContent, Double_t scaleFactor = -1.0);
  
  // Jet bias
  Double_t               fTrackBias;               
  Double_t               fClusterBias;             
  // Event Mixing
  Bool_t                 fDoEventMixing;           
  Int_t                  fNMixingTracks;           
  Int_t                  fMinNTracksMixedEvents;   
  Int_t                  fMinNEventsMixedEvents;   
  UInt_t                 fNCentBinsMixedEvent;     
  AliEventPoolManager   *fPoolMgr;                 
  // Event selection types
  UInt_t                 fTriggerType;             
  UInt_t                 fMixingEventType;         
  Bool_t                 fDisableFastPartition;    
  // Efficiency correction
  Int_t                  fDoEffCorrection;         
  // JES correction
  Bool_t                 fNoMixedEventJESCorrection; 
  TH2D                  *fJESCorrectionHist;       
  // Histogram binning variables
  Bool_t                 fDoLessSparseAxes;        
  Bool_t                 fDoWiderTrackBin;         

  // TODO: Consider moving to THistManager
  TH1                   *fHistTrackPt;             
  TH2                   *fHistJetEtaPhi;           
  TH2                   *fHistTrackEtaPhi[7];      
  TH2                   *fHistJetHEtaPhi;          

  TH1                   *fHistJetPt[6];            
  TH1                   *fHistJetPtBias[6];        
  TH2                   *fHistJetH[6][5][3];       
  TH2                   *fHistJetHBias[6][5][3];   
  TH3                   *fHistJHPsi;               
  THnSparse             *fhnMixedEvents;           
  THnSparse             *fhnJH;                    

  // Pb-Pb Efficiency correction coefficients
  static Double_t p0_10SG[17];                    
  static Double_t p10_30SG[17];                   
  static Double_t p30_50SG[17];                   
  static Double_t p50_90SG[17];                   
  // Good Runs
  static Double_t p0_10G[17];                     
  static Double_t p10_30G[17];                    
  static Double_t p30_50G[17];                    
  static Double_t p50_90G[17];                    

 private:

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

  ClassDef(AliAnalysisTaskEmcalJetHMEC, 12);
};
#endif