///////////////////////////////////////////////////////////////////////////////
// hungarian.hpp: Header file for Class HungarianAlgorithm.
//
// This is a C++ wrapper with slight modification of a hungarian algorithm
// implementation by Markus Buehren. The original implementation is a few
// mex-functions for use in MATLAB, found here:
// http://www.mathworks.com/matlabcentral/fileexchange/6543-functions-for-the-rectangular-assignment-problem
//
// Both this code and the original code are published under the BSD license.
// by Cong Ma, 2016
//
// (Modified by Desh Raj, Johns Hopkins University)

#ifndef SPYDER_HUNGARIAN_H
#define SPYDER_HUNGARIAN_H

#include <iostream>
#include <vector>

namespace spyder {

class HungarianAlgorithm {
 public:
  HungarianAlgorithm();
  ~HungarianAlgorithm();
  double Solve(std::vector<std::vector<double>> &DistMatrix, std::vector<int> &Assignment);

 private:
  void assignmentoptimal(int *assignment, double *cost, double *distMatrix, int nOfRows,
                         int nOfColumns);
  void buildassignmentvector(int *assignment, bool *starMatrix, int nOfRows, int nOfColumns);
  void computeassignmentcost(int *assignment, double *cost, double *distMatrix, int nOfRows);
  void step2a(int *assignment, double *distMatrix, bool *starMatrix, bool *newStarMatrix,
              bool *primeMatrix, bool *coveredColumns, bool *coveredRows, int nOfRows,
              int nOfColumns, int minDim);
  void step2b(int *assignment, double *distMatrix, bool *starMatrix, bool *newStarMatrix,
              bool *primeMatrix, bool *coveredColumns, bool *coveredRows, int nOfRows,
              int nOfColumns, int minDim);
  void step3(int *assignment, double *distMatrix, bool *starMatrix, bool *newStarMatrix,
             bool *primeMatrix, bool *coveredColumns, bool *coveredRows, int nOfRows,
             int nOfColumns, int minDim);
  void step4(int *assignment, double *distMatrix, bool *starMatrix, bool *newStarMatrix,
             bool *primeMatrix, bool *coveredColumns, bool *coveredRows, int nOfRows,
             int nOfColumns, int minDim, int row, int col);
  void step5(int *assignment, double *distMatrix, bool *starMatrix, bool *newStarMatrix,
             bool *primeMatrix, bool *coveredColumns, bool *coveredRows, int nOfRows,
             int nOfColumns, int minDim);
};
}  // end namespace spyder

#endif
