#include "test_precomp.hpp"
#include "opencv2/highgui/highgui.hpp"

using namespace cv;
using namespace std;

class CV_FramecountTest: public cvtest::BaseTest
{
public:
    void run(int);
};

void CV_FramecountTest::run(int)
{
    const int time_sec = 5, fps = 25;

    const string ext[] = {"avi", "mov", "mp4", "mpg", "wmv"};

    const size_t n = sizeof(ext)/sizeof(ext[0]);

	const string src_dir = ts->get_data_path();

	ts->printf(cvtest::TS::LOG, "\n\nSource files directory: %s\n", (src_dir+"../perf/video/").c_str());

    int failed = 0;

    for (size_t i = 0; i < n; ++i)
    {
        int code = cvtest::TS::OK;

        string file_path = src_dir+"../perf/video/big_buck_bunny."+ext[i];

        CvCapture *cap = cvCreateFileCapture(file_path.c_str());
        if (!cap)
        {
            ts->printf(cvtest::TS::LOG, "\nFile information (video %d): \n\nName: big_buck_bunny.%s\nFAILED\n\n", i+1, ext[i].c_str());
            ts->printf(cvtest::TS::LOG, "Error: cannot read source video file.\n");
            ts->set_failed_test_info(cvtest::TS::FAIL_INVALID_TEST_DATA);
            failed++; continue;
        }

        cvSetCaptureProperty(cap, CV_CAP_PROP_POS_FRAMES, 0);
        IplImage* frame; int FrameCount = -1;

        do
        {
            FrameCount++;
            frame = cvQueryFrame(cap);
        }
        while (frame);

        int framecount = (int)cvGetCaptureProperty(cap, CV_CAP_PROP_FRAME_COUNT);

        ts->printf(cvtest::TS::LOG, "\nFile information (video %d): \n"\
                   "\nName: big_buck_bunny.%s\nActual frame count: %d\n"\
                   "Frame count computed in the cycle of queries of frames: %d\n"\
                   "Frame count returned by cvGetCaptureProperty function: %d\n",
                   i+1, ext[i].c_str(), time_sec*fps, FrameCount, framecount);

        code = FrameCount != time_sec*fps ? cvtest::TS::FAIL_INVALID_OUTPUT : FrameCount != framecount ? cvtest::TS::FAIL_INVALID_OUTPUT : code;

        if (code)
        {
            ts->printf(cvtest::TS::LOG, "FAILED\n");
            ts->printf(cvtest::TS::LOG, "\nError: actual frame count and returned frame count are not matched.\n");
            ts->set_failed_test_info(code);
            failed++;
        }
        else
        {
            ts->printf(cvtest::TS::LOG, "OK\n");
            ts->set_failed_test_info(ts->OK);
        }

        cvReleaseImage(&frame);
        cvReleaseCapture(&cap);
    }

    ts->printf(cvtest::TS::LOG, "\nSuccessfull experiments: %d (%d%%)\n", n-failed, (n - failed)*100/n);
    ts->printf(cvtest::TS::LOG, "Failed experiments: %d (%d%%)\n", failed, failed*100/n);
}

TEST(HighguiFramecount, regression) {CV_FramecountTest test; test.safe_run();}