OpenCV/opencv/modules/videoio/perf/perf_output.cpp

// 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