#include <opencv2/opencv.hpp>

#include <opencv2/ml/ml.hpp>

#include <iostream>

using namespace cv;

using namespace std;

Mat catMat(Mat const& a, Mat const& b) {

        assert(a.cols == b.cols);

        Mat ret(a.rows + b.rows, a.cols, a.type());

        Mat d = ret.rowRange(0, a.rows);

        a.copyTo(d);

        d = ret.rowRange(a.rows, a.rows + b.rows);

        b.copyTo(d);

        return ret;

}

void outMat(Mat& out) {

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

                for (int j = 0; j < out.cols; j++) {

                        cout << out.at<float>(i, j) << " ";

                }

                cout << endl;

        }

}

void classify_svm(Mat const& data, Mat const& labels, Mat& out) {

        CvSVMParams params;

        params.svm_type    = CvSVM::C_SVC;

        params.kernel_type = CvSVM::LINEAR;

        params.p           = 1;

        CvSVM classifier;

        classifier.train(data, labels, Mat(), Mat(), params);

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

                out.at<float>(i, 0) = -classifier.predict(data.row(i), true);

        }

        params = classifier.get_params();

        cout << "p: " << params.p << endl;

}

int main(int, char**) {

        int npos = 2474;

               int nneg = 12180;

        int dim = 3780;

        Mat pos(npos, dim, CV_32FC1);

        randn(pos, 1, 1);

        Mat neg(nneg, dim, CV_32FC1);

        randn(neg, 0, 1);

        Mat data = catMat(pos, neg);

        Mat labels = catMat(Mat::ones(npos, 1, CV_32FC1), Mat::ones(nneg, 1, CV_32FC1) * -1);

        cout << data.rows << " " << data.cols << endl;

        Mat out(labels.rows, 1, CV_32FC1);

        classify_svm(data, labels, out);

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

                cout << labels.at<float>(i, 0) << " " << out.at<float>(i, 0) << endl;

        }

}