PuppiWeightTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
#ifndef PUPPIWeightTool_h
#define PUPPIWeightTool_h
 
#include "JetRecTools/JetConstituentModifierBase.h"
#include "xAODBase/IParticleContainer.h"
#include "AsgDataHandles/ReadHandleKey.h"
 
#include "xAODCaloEvent/CaloCluster.h"
#include "xAODCaloEvent/CaloClusterContainer.h"
#include "xAODTracking/VertexContainer.h"
#include "xAODPFlow/PFOContainer.h"
#include "xAODPFlow/FlowElementContainer.h"
#include "JetRecTools/Puppi.h"
#include <string>
 
class PuppiWeightTool: public JetConstituentModifierBase {
  ASG_TOOL_CLASS(PuppiWeightTool, IJetConstituentModifier)
 
 public:
 
  PuppiWeightTool(const std::string& name);
 
  virtual StatusCode initialize() override final;
  virtual StatusCode finalize() override;
 
  virtual StatusCode process_impl(xAOD::IParticleContainer* cont) const override final;
  StatusCode applyPuppiWeights(xAOD::PFOContainer* cont) const;
  StatusCode applyPuppiWeights(xAOD::FlowElementContainer* cont) const;
 
 private:
 
  // puppi parameters
  double m_R0;
  double m_Rmin;
  double m_beta;
  double m_centralPTCutOffset;
  double m_centralPTCutSlope;
  double m_forwardPTCutOffset;
  double m_forwardPTCutSlope;
  double m_etaBoundary;
 
  bool m_includeCentralNeutralsInAlpha;
  bool m_applyWeight;
 
  SG::ReadHandleKey<xAOD::VertexContainer> m_vertexContainer_key{"PrimaryVertices"};
 
};
 
#endif