229 lines
7.5 KiB
C++
Executable File
229 lines
7.5 KiB
C++
Executable File
/*M///////////////////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
|
|
//
|
|
// By downloading, copying, installing or using the software you agree to this license.
|
|
// If you do not agree to this license, do not download, install,
|
|
// copy or use the software.
|
|
//
|
|
//
|
|
// License Agreement
|
|
// For Open Source Computer Vision Library
|
|
//
|
|
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
|
|
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
|
|
// Third party copyrights are property of their respective owners.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without modification,
|
|
// are permitted provided that the following conditions are met:
|
|
//
|
|
// * Redistribution's of source code must retain the above copyright notice,
|
|
// this list of conditions and the following disclaimer.
|
|
//
|
|
// * Redistribution's in binary form must reproduce the above copyright notice,
|
|
// this list of conditions and the following disclaimer in the documentation
|
|
// and/or other materials provided with the distribution.
|
|
//
|
|
// * The name of the copyright holders may not be used to endorse or promote products
|
|
// derived from this software without specific prior written permission.
|
|
//
|
|
// This software is provided by the copyright holders and contributors "as is" and
|
|
// any express or implied warranties, including, but not limited to, the implied
|
|
// warranties of merchantability and fitness for a particular purpose are disclaimed.
|
|
// In no event shall the Intel Corporation or contributors be liable for any direct,
|
|
// indirect, incidental, special, exemplary, or consequential damages
|
|
// (including, but not limited to, procurement of substitute goods or services;
|
|
// loss of use, data, or profits; or business interruption) however caused
|
|
// and on any theory of liability, whether in contract, strict liability,
|
|
// or tort (including negligence or otherwise) arising in any way out of
|
|
// the use of this software, even if advised of the possibility of such damage.
|
|
//
|
|
//M*/
|
|
|
|
#include "test_precomp.hpp"
|
|
#include "opencv2/highgui/highgui.hpp"
|
|
|
|
#ifdef HAVE_FFMPEG
|
|
|
|
#include "ffmpeg_codecs.hpp"
|
|
|
|
using namespace cv;
|
|
using namespace std;
|
|
|
|
class CV_PositioningTest : public cvtest::BaseTest
|
|
{
|
|
public:
|
|
void CreateTestVideo(const string& format, int codec, int framecount = 125);
|
|
void run(int);
|
|
};
|
|
|
|
void CV_PositioningTest::CreateTestVideo(const string& format, int codec, int framecount)
|
|
{
|
|
stringstream s; s << codec;
|
|
|
|
cv::VideoWriter writer("test_video_"+s.str()+"."+format, codec, 25, cv::Size(640, 480), false);
|
|
|
|
for (int i = 0; i < framecount; ++i)
|
|
{
|
|
cv::Mat mat(480, 640, CV_8UC1);
|
|
size_t n = 8, tmp = i;
|
|
|
|
vector<char> tmp_code; tmp_code.clear();
|
|
|
|
while ( tmp > 1 )
|
|
{
|
|
tmp_code.push_back(tmp%2);
|
|
tmp /= 2;
|
|
}
|
|
tmp_code.push_back(tmp);
|
|
|
|
vector<char> i_code;
|
|
|
|
for (size_t j = 0; j < n; ++j)
|
|
{
|
|
char val = j < n - tmp_code.size() ? 0 : tmp_code.at(n-1-j);
|
|
i_code.push_back(val);
|
|
}
|
|
|
|
const size_t w = 480/n;
|
|
|
|
for (size_t j = 0; j < n; ++j)
|
|
{
|
|
cv::Scalar color = i_code[j] ? 255 : 0;
|
|
rectangle(mat, Rect(0, w*j, 640, w), color, -1);
|
|
}
|
|
|
|
writer << mat;
|
|
}
|
|
|
|
writer.~VideoWriter();
|
|
}
|
|
|
|
void CV_PositioningTest::run(int)
|
|
{
|
|
#if defined WIN32 || (defined __linux__ && !defined ANDROID)
|
|
#if !defined HAVE_GSTREAMER || defined HAVE_GSTREAMER_APP
|
|
|
|
const string format[] = {"avi", "mov", "mp4", "mpg", "wmv", "3gp"};
|
|
|
|
const char codec[][4] = { {'X', 'V', 'I', 'D'},
|
|
{'M', 'P', 'G', '2'},
|
|
{'M', 'J', 'P', 'G'} };
|
|
|
|
size_t n_format = sizeof(format)/sizeof(format[0]),
|
|
n_codec = sizeof(codec)/sizeof(codec[0]);
|
|
|
|
int n_frames = 256;
|
|
|
|
for (size_t i = 0; i < n_format; ++i)
|
|
for (size_t j = 0; j < n_codec; ++j)
|
|
{
|
|
CreateTestVideo(format[i], CV_FOURCC(codec[j][0], codec[j][1], codec[j][2], codec[j][3]), n_frames);
|
|
|
|
stringstream s; s << CV_FOURCC(codec[j][0], codec[j][1], codec[j][2], codec[j][3]); //codec_bmp_tags[j].tag;
|
|
|
|
const string file_path = "test_video_"+s.str()+"."+format[i];
|
|
|
|
bool error = false; int failed = 0;
|
|
|
|
cv::VideoCapture cap(file_path);
|
|
|
|
if (!cap.isOpened())
|
|
{
|
|
ts->printf(ts->LOG, "\n\nFile: %s\n", file_path.c_str());
|
|
ts->printf(ts->LOG, "\nVideo codec: %s\n", string(&codec[j][0], 4).c_str());
|
|
ts->printf(ts->LOG, "\nError: cannot read video file.");
|
|
ts->set_failed_test_info(ts->FAIL_INVALID_TEST_DATA);
|
|
error = true;
|
|
}
|
|
|
|
cap.set(CV_CAP_PROP_POS_FRAMES, 0);
|
|
|
|
int N = cap.get(CV_CAP_PROP_FRAME_COUNT);
|
|
|
|
if (N != n_frames)
|
|
{
|
|
if (!error)
|
|
{
|
|
ts->printf(ts->LOG, "\n\nFile: %s\n", file_path.c_str());
|
|
ts->printf(ts->LOG, "\nVideo codec: %s\n", string(&codec[j][0], 4).c_str());
|
|
error = true;
|
|
}
|
|
ts->printf(ts->LOG, "\nError: returned frame count in clip is incorrect.\n");
|
|
ts->set_failed_test_info(ts->FAIL_INVALID_OUTPUT);
|
|
}
|
|
|
|
vector <int> idx;
|
|
|
|
RNG rng(N);
|
|
idx.clear();
|
|
for( int k = 0; k < N-1; k++ )
|
|
idx.push_back(rng.uniform(0, N));
|
|
idx.push_back(N-1);
|
|
std::swap(idx.at(rng.uniform(0, N-1)), idx.at(N-1));
|
|
|
|
for (int k = 0; k < N; ++k)
|
|
{
|
|
cap.set(CV_CAP_PROP_POS_FRAMES, (double)idx[k]);
|
|
|
|
cv::Mat img; cap.retrieve(img);
|
|
|
|
if (img.empty())
|
|
{
|
|
if (!error)
|
|
{
|
|
ts->printf(ts->LOG, "\n\nFile: %s\n", file_path.c_str());
|
|
ts->printf(ts->LOG, "\nVideo codec: %s\n", string(&codec[j][0], 4).c_str());
|
|
error = true;
|
|
}
|
|
ts->printf(ts->LOG, "\nError: cannot read a frame in position %d.\n", idx[k]);
|
|
ts->set_failed_test_info(ts->FAIL_INVALID_OUTPUT);
|
|
}
|
|
|
|
const double thresh = 100;
|
|
|
|
const size_t n = 8, w = img.rows/n;
|
|
|
|
int index = 0, deg = n-1;
|
|
|
|
for (size_t l = 0; l < n; ++l)
|
|
{
|
|
cv::Mat mat = img.rowRange(w*l, w*(l+1)-1);
|
|
|
|
Scalar mat_mean = cv::mean(mat);
|
|
|
|
if (mat_mean[0] > thresh) index += (int)std::pow(2.0, 1.0*deg);
|
|
|
|
deg--;
|
|
}
|
|
|
|
if (index != idx[k])
|
|
{
|
|
if (!error)
|
|
{
|
|
ts->printf(ts->LOG, "\n\nFile: %s\n", file_path.c_str());
|
|
ts->printf(ts->LOG, "\nVideo codec: %s\n\n", string(&codec[j][0], 4).c_str());
|
|
error = true;
|
|
}
|
|
ts->printf(ts->LOG, "Required position: %d Returned position: %d\n", idx[k], index);
|
|
ts->set_failed_test_info(ts->FAIL_INVALID_OUTPUT);
|
|
failed++;
|
|
}
|
|
}
|
|
|
|
if (!error) { ts->printf(ts->LOG, "\n\nFile: %s\n", file_path.c_str());
|
|
ts->printf(ts->LOG, "\nVideo codec: %s\n", string(&codec[j][0], 4).c_str()); }
|
|
|
|
const string status = failed ? "FAILED" : "OK";
|
|
ts->printf(ts->LOG, "\nSuccessfull iterations: %d(%d%%) Failed iterations: %d(%d%%) %s\n", N-failed, (N-failed)*100/N, failed, failed*100/N, status.c_str());
|
|
if( i < n_format-1 || j < n_codec-1 ) ts->printf(ts->LOG, "\n----------");
|
|
}
|
|
|
|
#endif
|
|
#endif
|
|
}
|
|
|
|
TEST(Highgui_Positioning, regression) { CV_PositioningTest test; test.safe_run(); }
|
|
|
|
#endif
|