// OpenCV_Exam08_Mat.cpp : Defines the entry point for the console application.

//

#include "stdafx.h"

#include <opencv.hpp>

#ifdef _DEBUG

#pragma comment (lib, "opencv_highgui248d.lib")

#pragma comment (lib, "opencv_imgproc248d.lib")

#pragma comment (lib, "opencv_core248d.lib")

#pragma comment (lib, "opencv_calib3d248d.lib")

#pragma comment (lib, "opencv_features2d248d.lib")

#pragma comment (lib, "opencv_ocl248d.lib")

#pragma comment (lib, "opencv_video248d.lib")

#else

#pragma comment (lib, "opencv_core248.lib")

#pragma comment (lib, "opencv_imgproc248.lib")

#pragma comment (lib, "opencv_highgui248.lib")

#pragma comment (lib, "opencv_calib3d248.lib")

#pragma comment (lib, "opencv_ocl248.lib")

#pragma comment (lib, "opencv_video248.lib")

#endif

#include <iostream>

#include <string>

#include "opencv2/core/core.hpp"

#include "opencv2/ocl/ocl.hpp"

#include "opencv2/highgui/highgui.hpp"

using namespace std;

using namespace cv;

using namespace cv::ocl;

#define M_MOG  1

#define M_MOG2 2

int main(int argc, const char** argv)

{

        /*cv::CommandLineParser cmd(argc, argv,

                "{ c | camera | false       | use camera }"

                "{ f | file   | 768x576.avi | input video file }"

                "{ m | method | mog         | method (mog, mog2) }"

                "{ h | help   | false       | print help message }");

        if (cmd.get<bool>("help"))

        {

                cout << "Usage : bgfg_segm [options]" << endl;

                cout << "Available options:" << endl;

                cmd.printParams();

                return EXIT_SUCCESS;

        }*/

        bool useCamera = false;//cmd.get<bool>("camera");

        string file = "..\\768x576.avi";//cmd.get<string>("file");

        string method = "mog2";//cmd.get<string>("method");

        if (method != "mog" && method != "mog2")

        {

                cerr << "Incorrect method" << endl;

                return EXIT_FAILURE;

        }

        int m = method == "mog" ? M_MOG : M_MOG2;

        VideoCapture cap;

        if (useCamera)

                cap.open(0);

        else

                cap.open(file);

        if (!cap.isOpened())

        {

                cout << "can not open camera or video file" << endl;

                return EXIT_FAILURE;

        }

        Mat frame;

        cap >> frame;

#define USE_OCL

#ifdef USE_OCL

        oclMat d_frame(frame);

        cv::ocl::MOG mog;

        cv::ocl::MOG2 mog2;

        oclMat d_fgmask, d_fgimg, d_bgimg;

        d_fgimg.create(d_frame.size(), d_frame.type());

        Mat fgmask, fgimg, bgimg;

        switch (m)

        {

        case M_MOG:

                mog(d_frame, d_fgmask, 0.01f);

                break;

        case M_MOG2:

                mog2(d_frame, d_fgmask);

                break;

        }

#else

        BackgroundSubtractorMOG mog;

        BackgroundSubtractorMOG2 mog2;

        Mat fgmask, fgimg, bgimg;

        switch (m)

        {

        case M_MOG:

                mog(frame, fgmask, 0.01f);

                break;

        case M_MOG2:

                mog2(frame, fgmask);

                break;

        }

#endif

#ifdef USE_OCL

        int64 start = cv::getTickCount();

        for (;;)

        {

                cap >> frame;

                if (frame.empty())

                        break;

                d_frame.upload(frame);

                //int64 start = cv::getTickCount();

                //update the model

                switch (m)

                {

                case M_MOG:

                        mog(d_frame, d_fgmask, 0.01f);

                        mog.getBackgroundImage(d_bgimg);

                        break;

                case M_MOG2:

                        mog2(d_frame, d_fgmask);

                        mog2.getBackgroundImage(d_bgimg);

                        break;

                }

                //double fps = cv::getTickFrequency() / (cv::getTickCount() - start);

                //std::cout << "FPS : " << fps << std::endl;

                d_fgimg.setTo(Scalar::all(0));

                d_frame.copyTo(d_fgimg, d_fgmask);

                d_fgmask.download(fgmask);

                d_fgimg.download(fgimg);

                if (!d_bgimg.empty())

                        d_bgimg.download(bgimg);

                imshow("image", frame);

                imshow("foreground mask", fgmask);

                imshow("foreground image", fgimg);

                if (!bgimg.empty())

                        imshow("mean background image", bgimg);

                if (27 == waitKey(1))

                        break;

        }

        double t = (cv::getTickCount() - start) / cv::getTickFrequency();

        std::cout << "t : " << t << std::endl;

        getchar();

#else

        int64 start = cv::getTickCount();

        for (;;)

        {

                cap >> frame;

                if (frame.empty())

                        break;

                //int64 start = cv::getTickCount();

                //update the model

                switch (m)

                {

                case M_MOG:

                        mog(frame, fgmask, 0.01f);

                        mog.getBackgroundImage(bgimg);

                        break;

                case M_MOG2:

                        mog2(frame, fgmask);

                        mog2.getBackgroundImage(bgimg);

                        break;

                }

                //double fps = cv::getTickFrequency() / (cv::getTickCount() - start);

                //std::cout << "FPS : " << fps << std::endl;

                fgimg.setTo(Scalar::all(0));

                frame.copyTo(fgimg, fgmask);

                imshow("image", frame);

                imshow("foreground mask", fgmask);

                imshow("foreground image", fgimg);

                if (!bgimg.empty())

                        imshow("mean background image", bgimg);

                if (27 == waitKey(1))

                        break;

        }

        double t = (cv::getTickCount() - start) / cv::getTickFrequency();

        std::cout << "t : " << t << std::endl;

        getchar();

#endif

        return EXIT_SUCCESS;

}