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

#ifndef TIGHTTRACKVERTEXASSOCIATIONTOOL_H
#define TIGHTTRACKVERTEXASSOCIATIONTOOL_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 TightTrackVertexAssociationTool : virtual public asg::AsgTool, virtual public ITrackVertexAssociationTool
  {
    enum MatchStatus{UsedInFit=1, Matched=2, UnMatch=0};  // MatchCode for the matching, UsedInFit for the tracks used in the vertex fitting process, Matched for the tracks not used in the fitting process. UnMatch for the tracks not match to the vertex.

    ASG_TOOL_CLASS( TightTrackVertexAssociationTool, ITrackVertexAssociationTool)
    public:
      TightTrackVertexAssociationTool(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:

      MatchStatus doTightMatch(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;
      bool m_doPV;
      bool m_dod0sel;
  };
}

#endif

---