//g++ -Wall -Wextra `pkg-config --cflags --libs opencv` -o test test.cpp

#include <iostream>

#include <fstream>

#include <cassert>

#include <opencv2/core/core.hpp>

#include <opencv2/highgui/highgui.hpp>

#include <opencv2/ml/ml.hpp>

static

void

usage(char* argv0)

{

  std::cerr<<"Usage: "<<argv0<<" inputFilename inputModelFile"<<std::endl;

  std::cerr<<"inputFilename: input filename with data to classify & results obtained on linux\n";

  std::cerr<<"inputModelFile : input model file\n";

  std::cerr<<"\n";

}

static

void

processArgs(int argc, char *argv[],

            std::string &inputFilename,

            std::string &modelFilename)

{

  if (argc != 3) {

    usage(argv[0]);

    exit(EXIT_FAILURE);

  }

  inputFilename = argv[1];

  modelFilename = argv[2];

  if (inputFilename.empty()) {

    std::cerr << "Error: unspecified input filename\n";

    usage (argv[0]);

    exit (EXIT_FAILURE);

  }

  if (modelFilename.empty()) {

    std::cerr<<"Error: no model filename specified\n";

    usage(argv[0]);

    exit(EXIT_FAILURE);

  }

}

void

checkResults(const cv::Mat &res, 

             const cv::Mat &resMat, 

             const cv::Mat &data,

             const std::string &s)

{

  assert(res.cols == resMat.cols);

  assert(res.rows == resMat.rows);

  assert(res.type() == resMat.type());

  assert(resMat.isContinuous());

  assert(res.isContinuous());

  const float *r1 = res.ptr<float>(0);

  const float *r2 = resMat.ptr<float>(0);

  const float EPS = 0.000001;

  int nbWrongs = 0;

  for (int i=0; i<res.rows; ++i) {

    if (fabs(*r1 - *r2) > EPS) {

      if (nbWrongs < 6) {

        std::cerr<<"For data["<<i<<": ";

        const float *d = data.ptr<float>(i);

        for (int i=0; i<data.cols; ++i)  {

          std::cerr<<*d<<" ";

          ++d;

        }

        std::cerr<<"\n => classifier gives "<<*r1<<" instead of "<<*r2<<"\n";

      }

      ++nbWrongs;

    }

    ++r1;

    ++r2;

  }

  std::cerr<<s<<" ===> "<<nbWrongs<<"/"<<res.rows<<" errors\n";

}

void

test1(const cv::Mat &testData, 

      const CvSVM &svmClassifier, 

      const cv::Mat &resMat)

{

  assert(resMat.cols == 1);

  assert(resMat.type() == CV_32F);

  cv::Mat res(testData.rows, 1, CV_32F);

  assert(res.isContinuous());

  float *r = res.ptr<float>(0);

  for (int i=0; i<testData.rows; ++i) {

    *r = svmClassifier.predict(testData.row(i), true);

    ++r;

  }

  checkResults(res, resMat, testData, "test1");

}

void

test2(const cv::Mat &testData, 

      const CvSVM &svmClassifier, 

      const cv::Mat &resMat)

{

  assert(resMat.cols == 1);

  assert(resMat.type() == CV_32F);

  cv::Mat res(testData.rows, 1, CV_32F);

  svmClassifier.predict(testData, res);

  checkResults(res, resMat, testData, "test2");

}

int

main(int argc, char *argv[])

{

  std::string inputFilename, modelFilename;

  processArgs(argc, argv, inputFilename, modelFilename);

  cv::FileStorage fs2(inputFilename, cv::FileStorage::READ);

  cv::Mat dataMat, resMat, resCMat;

  fs2["data"] >> dataMat;

  fs2["results"] >> resMat;

  fs2["resultsC"] >> resCMat;

  if (dataMat.rows != resMat.rows) {

    std::cerr<<"ERROR: data & res do not have same number of rows ["<<dataMat.rows<<" vs "<<resMat.rows<<"] \n";

    exit(EXIT_FAILURE);

  }

  if (dataMat.rows != resCMat.rows) {

    std::cerr<<"ERROR: data & resC do not have same number of rows ["<<dataMat.rows<<" vs "<<resCMat.rows<<"] \n";

    exit(EXIT_FAILURE);

  }

  if (resMat.cols != 1) {

    std::cerr<<"ERROR: resMat.cols="<<resMat.cols<<" != 1\n";

    exit(EXIT_FAILURE);

  }

  if (resCMat.cols != 1) {

    std::cerr<<"ERROR: resCMat.cols="<<resCMat.cols<<" != 1\n";

    exit(EXIT_FAILURE);

  }

  CvSVM svmClassifier;

  svmClassifier.load(modelFilename.c_str());

  test1(dataMat, svmClassifier, resMat);

  test2(dataMat, svmClassifier, resCMat);

  return 0;

}