PileupAreaResidualCalibStep.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETCALIBTOOLS_PILEUPAREARESIDUALCALIBSTEP_H
#define JETCALIBTOOLS_PILEUPAREARESIDUALCALIBSTEP_H 1
 
/* PileupAreaResidualCalibStep performs the first step of the jet calibration
 *
 *  - area subtraction in the form  pT_corr = pT - rho x pT_area 
 *  - residual correction (1D version)
 *
 */
 
#include "xAODTracking/VertexContainer.h"
#include "xAODEventShape/EventShape.h"
#include "xAODEventInfo/EventInfo.h"
 
#include "AsgTools/AsgTool.h"
#include "AsgTools/ToolHandle.h"
#include "AsgDataHandles/ReadHandleKey.h"
#include "AsgDataHandles/ReadDecorHandleKey.h"
#include "AsgTools/PropertyWrapper.h"
#include "JetAnalysisInterfaces/IJetCalibStep.h"
#include "JetAnalysisInterfaces/IVarTool.h"
 
class PileupAreaResidualCalibStep   : public asg::AsgTool,
                    virtual public IJetCalibStep
{
 
  ASG_TOOL_CLASS(PileupAreaResidualCalibStep, IJetCalibStep) 
 
 public:
  PileupAreaResidualCalibStep(const std::string& name="PileupAreaResidualCalibStep");
 
  virtual StatusCode initialize() override;
 
  // Apply calibration to jet
  virtual StatusCode calibrate(xAOD::JetContainer& jetCont) const override;
 
 private:
 
  double getResidualOffset(const xAOD::Jet& jet, const JetHelper::JetContext& jc, double mu, double NPV, bool MuOnly, bool NOnly) const;
 
  Gaudi::Property<bool> m_isData{this, "IsData", false, ""};
  Gaudi::Property<bool> m_doJetArea{this, "DoJetArea", false, "doc"};

  SG::ReadHandleKey<xAOD::EventShape> m_rhoKey{this, "RhoKey", "auto"};
  SG::ReadHandleKey<xAOD::VertexContainer> m_pvKey{this, "PrimaryVerticesContainerName", "PrimaryVertices"};
  SG::ReadDecorHandleKey<xAOD::EventInfo> m_muKey {this, "averageInteractionsPerCrossingKey",
    "EventInfo.averageInteractionsPerCrossing","Decoration for Average Interaction Per Crossing"};

  ToolHandle<JetHelper::IVarTool> m_histTool_mu = {this , "histTool_mu", "HistoInput1D", "mu histo reader" };
  ToolHandle<JetHelper::IVarTool> m_histTool_NPV = {this , "histTool_NPV", "HistoInput1D", "npv histo reader" };

  Gaudi::Property<bool> m_doMuOnly{this, "ApplyOnlyMuResidual", false, "doc"};
  Gaudi::Property<bool> m_doNPVOnly{this, "ApplyOnlyNPVResidual", false, "doc"};
 
  Gaudi::Property<float> m_NPV_ref{this , "DefaultNPVRef", -99., ""};
  Gaudi::Property<float> m_mu_ref{this , "DefaultMuRef", -99., ""};
  Gaudi::Property<float> m_muSF{this , "MuScaleFactor", 1., ""};
 
  Gaudi::Property<bool> m_doSequentialResidual{this, "DoSequentialResidual", false, "doc"};

  Gaudi::Property<std::string> m_jetInScale {this, "InScale", "JetConstitScaleMomentum", "Starting jet scale" };
  Gaudi::Property<std::string> m_jetOutScale {this, "OutScale", "JetPileupScaleMomentum", "Ending jet scale" };
 
};
 
#endif