/cvmfs/atlas.cern.ch/repo/sw/ASG/AnalysisBase/2.4.31/TrackVertexAssociationTool/TrackVertexAssociationTool/LooseTrackVertexAssociationTool.h

#ifndef LOOSETRACKVERTEXASSOCIATIONTOOL_H
#define LOOSETRACKVERTEXASSOCIATIONTOOL_H

#include "AsgTools/AsgTool.h"
#include "TrackVertexAssociationTool/ITrackVertexAssociationTool.h"
#include "AthContainers/AuxElement.h"
#include "AthLinks/ElementLink.h"

#include "xAODTracking/TrackParticleFwd.h"
#include "xAODTracking/TrackParticleContainerFwd.h"
#include "xAODTracking/VertexFwd.h"
#include "xAODTracking/VertexContainerFwd.h"
#include "TrackVertexAssociationTool/TrackVertexAssociationMap.h"

#include <string>

namespace CP
{

  class LooseTrackVertexAssociationTool : virtual public asg::AsgTool, virtual public ITrackVertexAssociationTool
  {

    ASG_TOOL_CLASS( LooseTrackVertexAssociationTool, ITrackVertexAssociationTool)
    public:
      LooseTrackVertexAssociationTool(std::string name);
//      ~TrackToVertexTool(){};

      StatusCode initialize();

      void print() const;

      bool isCompatible( const xAOD::TrackParticle &trk, const xAOD::Vertex &vx ) const;  // return the decision of with the track match to the vertex
      xAOD::TrackVertexAssociationMap getMatchMap( std::vector< const xAOD::TrackParticle *> &trk_list, std::vector< const xAOD::Vertex *> &vx_list) const;
      xAOD::TrackVertexAssociationMap getMatchMap( const xAOD::TrackParticleContainer &trkCont, const xAOD::VertexContainer &vxCont ) const;
 

      ElementLink< xAOD::VertexContainer> getUniqueMatchVertexLink(const xAOD::TrackParticle &, const xAOD::VertexContainer &vxCont) const;
      
      const xAOD::Vertex* getUniqueMatchVertex( const xAOD::TrackParticle &trk, std::vector< const xAOD::Vertex *> &vx_list) const;  // return the  vertex matched with the tracks in the vx_list

      xAOD::TrackVertexAssociationMap getUniqueMatchMap( std::vector< const xAOD::TrackParticle *> &trk_list , std::vector< const xAOD::Vertex *> &vx_list) const;  // return the map of vertex matched with the tracks in the 

      xAOD::TrackVertexAssociationMap getUniqueMatchMap( const xAOD::TrackParticleContainer &trkCont , const xAOD::VertexContainer &vxCont) const;  // return the map of vertex matched with the tracks in the 

    private:

      bool isMatch(const xAOD::TrackParticle &trk, const xAOD::Vertex &vx, float & dz) const;
      const xAOD::Vertex* getUniqueMatchVx( const xAOD::TrackParticle &trk, const xAOD::VertexContainer &vxCont) const;  // return the  vertex matched with the tracks in the vx_list

      float m_dz_cut;
      float m_d0_cut;
  };
}

#endif

---