TauAxisSetter.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2026 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_TAUAXISSETTER_H
#define TAURECTOOLS_TAUAXISSETTER_H
 
#include "tauRecTools/TauRecToolBase.h"
 
#include "AsgTools/ToolHandle.h"
 
#include "xAODPFlow/FlowElement.h"
#include "PFlowUtils/FEHelpers.h"

 
class TauAxisSetter : public TauRecToolBase {
 
public:
  
  ASG_TOOL_CLASS2(TauAxisSetter, TauRecToolBase, ITauToolBase)
  
   
  TauAxisSetter(const std::string& name);

  virtual ~TauAxisSetter() = default;
 
  using TauRecToolBase::executeTool;
  virtual StatusCode executeTool(xAOD::TauJet& tau,
                 const EventContext& ctx) const override;
 
private:
  
  TLorentzVector getVertexCorrectedP4(const xAOD::JetConstituent& constituent, 
                                      const Amg::Vector3D& position) const; 

  TLorentzVector getVertexCorrectedP4(const xAOD::FlowElement& fe,
                                      const Amg::Vector3D& position) const;
 
  Gaudi::Property<double> m_clusterCone {this, "ClusterCone", 0.2, "cone of tau candidate"};
  Gaudi::Property<bool> m_doVertexCorrection {this, "VertexCorrection", true, "switch of tau vertex correction"};
};
 
#endif // TAURECTOOLS_TAUAXISSETTER_H