StableMatchingBase.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef INDETTRACKPERFMON_STABLEMATCHINGBASE_H
#define INDETTRACKPERFMON_STABLEMATCHINGBASE_H


#include "AsgTools/AsgTool.h"

#include "ITrackMatchingTool.h"
#include "TrackAnalysisCollections.h"
#include "TrackMatchingLookup.h"
 
#include <vector>
#include <algorithm>
 
namespace IDTPM {

  template< typename T, typename R=T > 
  class StableMatchingBase : public asg::AsgTool {
  public:

    StableMatchingBase( const std::string& name ) :
        asg::AsgTool( name ) { };

    virtual StatusCode matchVectors(const std::vector< const T* >& vTest,
                    const std::vector< const R* >& vRef,
                    ITrackMatchingLookup& matches ) const;
 
    // Override this in derived classes
    virtual float distance(const T& t, const R& r) const = 0;
 
  protected:
 
    // Helper methods (implemented in StableMatchingBase.icc)
    void buildDistanceMatrix(const std::vector<const T*>& vTest,
                 const std::vector<const R*>& vRef,
                 std::vector<std::vector<float>>& dist) const;
 
    void buildPreferenceLists(const std::vector<std::vector<float>>& dist,
                  std::vector<std::vector<int>>& testPrefs,
                  std::vector<std::vector<int>>& refRankings) const;
 
    void galeShapley(const std::vector<std::vector<int>>& testPrefs,
             const std::vector<std::vector<int>>& refRankings,
                     const std::vector<std::vector<float>>& dist,
                     std::vector<int>& testMatch,
                     std::vector<int>& refMatch) const;
  };
 
} // namespace IDTPM
 
#include "StableMatchingBase.icc"
#endif // > !INDETTRACKPERFMON_STABLEMATCHINGBASE_H