RLAPSolverJV.hpp

Go to the documentation of this file.

#ifndef RLAP_SOLVER_JV_H
#define RLAP_SOLVER_JV_H

#include <list>
#include <unordered_set>
#include <vector>
#include <algorithm>
#include "Tensor.hpp"
#include "RLAPSolver.hpp"

#define BIG 100000

class RLAPSolverJV : public RLAPSolver {
public:

    RLAPSolverJV(const Tensor<int>& mat);

    void solve(Tensor<unsigned>& assignments) override;

private:

    const unsigned rows;

    const unsigned cols;

    const unsigned size;

    std::vector<std::vector<double>> costMat;

    std::vector<int> rowsol;

    std::vector<int> colsol;

    void jvlap(int dim, const std::vector<std::vector<double>>& assigncost,
               std::vector<int>& rowsol,
               std::vector<int>& colsol, std::vector<double>& u, std::vector<double>& v);

    void assignMatching(Tensor<unsigned>& assignments);
};

#endif // RLAP_SOLVER_JV_H