48 lines
1.5 KiB
C++
48 lines
1.5 KiB
C++
|
// This file is part of OpenCV project.
|
||
|
|
// It is subject to the license terms in the LICENSE file found in the top-level directory
|
||
|
|
// of this distribution and at http://opencv.org/license.html
|
||
|
|
#include "perf_precomp.hpp"
|
||
|
|
|
||
|
|
namespace opencv_test
|
||
|
|
{
|
||
|
|
using namespace perf;
|
||
|
|
|
||
|
|
typedef tuple<std::string, bool> VideoWriter_Writing_t;
|
||
|
|
typedef perf::TestBaseWithParam<VideoWriter_Writing_t> VideoWriter_Writing;
|
||
|
|
|
||
|
|
const string image_files[] = {
|
||
|
|
"python/images/QCIF_00.bmp",
|
||
|
|
"python/images/QCIF_01.bmp",
|
||
|
|
"python/images/QCIF_02.bmp",
|
||
|
|
"python/images/QCIF_03.bmp",
|
||
|
|
"python/images/QCIF_04.bmp",
|
||
|
|
"python/images/QCIF_05.bmp"
|
||
|
|
};
|
||
|
|
|
||
|
|
PERF_TEST_P(VideoWriter_Writing, WriteFrame,
|
||
|
|
testing::Combine(
|
||
|
|
testing::ValuesIn(image_files),
|
||
|
|
testing::Bool()))
|
||
|
|
{
|
||
|
|
const string filename = getDataPath(get<0>(GetParam()));
|
||
|
|
const bool isColor = get<1>(GetParam());
|
||
|
|
Mat image = imread(filename, isColor ? IMREAD_COLOR : IMREAD_GRAYSCALE );
|
||
|
|
#if defined(HAVE_MSMF) && !defined(HAVE_FFMPEG)
|
||
|
|
const string outfile = cv::tempfile(".wmv");
|
||
|
|
const int fourcc = VideoWriter::fourcc('W', 'M', 'V', '3');
|
||
|
|
#else
|
||
|
|
const string outfile = cv::tempfile(".avi");
|
||
|
|
const int fourcc = VideoWriter::fourcc('X', 'V', 'I', 'D');
|
||
|
|
#endif
|
||
|
|
|
||
|
|
VideoWriter writer(outfile, fourcc, 25, cv::Size(image.cols, image.rows), isColor);
|
||
|
|
if (!writer.isOpened())
|
||
|
|
throw SkipTestException("Video file can not be opened");
|
||
|
|
|
||
|
|
TEST_CYCLE_N(100) { writer << image; }
|
||
|
|
SANITY_CHECK_NOTHING();
|
||
|
|
remove(outfile.c_str());
|
||
|
|
}
|
||
|
|
|
||
|
|
} // namespace
|