TauVertexFinder.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024  CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef TAURECTOOLS_TAUVERTEXFINDER_H
#define TAURECTOOLS_TAUVERTEXFINDER_H
 
#include "tauRecTools/TauRecToolBase.h"
#include "GaudiKernel/ToolHandle.h"
#include "StoreGate/ReadHandle.h"
#include "xAODTracking/VertexContainer.h"
#include "JetEDM/TrackVertexAssociation.h"
#include "InDetTrackSelectionTool/IInDetTrackSelectionTool.h"
#include "TrackVertexAssociationTool/ITrackVertexAssociationTool.h"
 
 
class TauVertexFinder : public TauRecToolBase {
public:
  //-------------------------------------------------------------
  //-------------------------------------------------------------
  TauVertexFinder(const std::string& name);
  ASG_TOOL_CLASS2(TauVertexFinder, TauRecToolBase, ITauToolBase);
  ~TauVertexFinder();
 
  //-------------------------------------------------------------
  //-------------------------------------------------------------
  StatusCode initialize() override;
  StatusCode executeVertexFinder(xAOD::TauJet& pTau,
                                const xAOD::VertexContainer* vertexContainer = nullptr) const override;
 
private:
  ElementLink<xAOD::VertexContainer>
  getPV_TJVA(const xAOD::TauJet& tauJet,
             const xAOD::VertexContainer& vertices,
             float& maxJVF) const;
 
  std::pair<float, float> getVertexScores(const std::vector<const xAOD::TrackParticle*>& tracks, float vx_z) const;
  
private:
  
  Gaudi::Property<bool> m_useTJVA {this, "UseTJVA", true};
  Gaudi::Property<std::string> m_assocTracksName {this, "AssociatedTracks", ""};
  
  ToolHandle< InDet::IInDetTrackSelectionTool > m_TrackSelectionToolForTJVA {this, "InDetTrackSelectionToolForTJVA", ""};
  ToolHandle<CP::ITrackVertexAssociationTool> m_trkVertexAssocTool{this, "TVATool", "Tau track-vertex association tool"};
  
  SG::ReadHandleKey<xAOD::VertexContainer> m_vertexInputContainer{this,"Key_vertexInputContainer", "PrimaryVertices", "input vertex container key"};
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trackPartInputContainer{this,"Key_trackPartInputContainer", "InDetTrackParticles", "input track particle container key"};
 
};
 
#endif // TAURECTOOLS_TAUVERTEXFINDER_H