Reconstruction/Jet/JetRecTools/JetRecTools/TrackVertexAssociationTool.h

Go to the documentation of this file.

// -*- C++ -*- 
 
/*
  Copyright (C) 2002-2021 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef JETRECTOOLS_TRACKVERTEXASSOCIATIONTOOL_H
#define JETRECTOOLS_TRACKVERTEXASSOCIATIONTOOL_H
 
#include "AsgTools/AsgTool.h"
 
#include "JetInterface/IJetExecuteTool.h"
#include "JetEDM/TrackVertexAssociation.h"
 
#include "xAODTracking/TrackParticleContainer.h" 
#include "xAODTracking/VertexContainer.h" 
#include "TrackVertexAssociationTool/ITrackVertexAssociationTool.h"
#include "AsgTools/ToolHandle.h"
#include "AsgDataHandles/ReadHandleKey.h"
#include "AsgDataHandles/WriteHandleKey.h"
 
#include <string>
 
class TrackVertexAssociationTool :  public asg::AsgTool,
                                    virtual public IJetExecuteTool
{
  ASG_TOOL_CLASS(TrackVertexAssociationTool,IJetExecuteTool)
  
  public:
  TrackVertexAssociationTool(const std::string & t);
  
  
  StatusCode initialize();
 
  // Retrieve necessary inputs, call building function, store result in evtStore
  int execute() const;
  
private:
 
  std::unique_ptr<jet::TrackVertexAssociation>
  buildTrackVertexAssociation_withTool(const xAOD::TrackParticleContainer*, 
                                       const xAOD::VertexContainer*) const;
  // Constuct a TrackVertexAssociation using the older custom method
  // jet::TrackVertexAssociation* 
  std::unique_ptr<jet::TrackVertexAssociation>
  buildTrackVertexAssociation_custom(const xAOD::TrackParticleContainer*, 
                                     const xAOD::VertexContainer*) const;
 
  std::unique_ptr<jet::TrackVertexAssociation>
  makeTrackVertexAssociation(const xAOD::TrackParticleContainer*, 
                             const xAOD::VertexContainer*,
                             bool useCustom) const;
 
  ToolHandle<CP::ITrackVertexAssociationTool> m_tvaTool;
 
  // old Configurable parameters
 
  float m_transDistMax;
  float m_longDistMax;
  float m_maxZ0SinTheta;
 
  SG::ReadHandleKey<xAOD::TrackParticleContainer> m_trackContainer_key;
  SG::ReadHandleKey<xAOD::VertexContainer> m_vertexContainer_key;
  SG::WriteHandleKey<jet::TrackVertexAssociation> m_tva_key;
 
};
 
 
#endif