CorrectPFOTool.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration
*/
 
// CorrectPFOTool.h
 
#ifndef JETRECTOOLS_CORRECTPFOTOOL_H
#define JETRECTOOLS_CORRECTPFOTOOL_H
 
 
#include "JetRecTools/JetConstituentModifierBase.h"
#include "xAODTracking/VertexContainer.h" 
#include "xAODPFlow/PFOContainer.h"
#include "xAODPFlow/FlowElementContainer.h"
#include "AsgTools/ToolHandle.h"
#include "PFlowUtils/IWeightPFOTool.h"
 
#include "AsgDataHandles/ReadHandleKey.h"
 
 
class CorrectPFOTool : public JetConstituentModifierBase{
  ASG_TOOL_CLASS(CorrectPFOTool, IJetConstituentModifier)
 
  public:
 
  CorrectPFOTool(const std::string& name);
  
  // Check that the configuration is reasonable
  StatusCode initialize() override final;
 
 private:
 
  // Implement the correction
  StatusCode process_impl(xAOD::IParticleContainer* cont) const override final;
  // Type-specific operation
  StatusCode correctPFO(xAOD::PFOContainer& cont) const;
  StatusCode correctPFO(xAOD::FlowElementContainer& cont) const;
  StatusCode correctPFOByVertex(xAOD::PFOContainer& cont) const;
  StatusCode correctPFOByVertex(xAOD::FlowElementContainer& cont) const;
 
  const xAOD::Vertex* getPrimaryVertex() const;
  StatusCode applyNeutralCorrection(xAOD::PFO& pfo, const xAOD::Vertex& vtx) const;
  StatusCode applyNeutralCorrection(xAOD::FlowElement& pfo, const xAOD::Vertex& vtx) const;
  StatusCode applyChargedCorrection(xAOD::PFO& pfo) const;
  StatusCode applyChargedCorrection(xAOD::FlowElement& pfo) const;
 
  bool m_inputIsEM;   
  bool m_calibrate;   
  bool m_correctneutral;   //If true, correct neutral PFO
  bool m_correctcharged;   //If true, correct charged PFO
  bool m_useChargedWeights; //If true, them we make use of the charged PFO weighting scheme
  bool m_doByVertex;  // If true, then we add vertex-by-vertex corrections as auxdata for neutral PFOs
  ToolHandle<CP::IWeightPFOTool> m_weightPFOTool;
 
  SG::ReadHandleKey<xAOD::VertexContainer> m_vertexContainer_key;
  
};
 
#endif