#include "test_precomp.hpp" #include #include #include using namespace cv; using namespace std; using namespace cvtest; TEST(Imgcodecs_Image, read_write_bmp) { const size_t IMAGE_COUNT = 10; const double thresDbell = 32; for (size_t i = 0; i < IMAGE_COUNT; ++i) { stringstream s; s << i; const string digit = s.str(); const string src_name = TS::ptr()->get_data_path() + "../python/images/QCIF_0" + digit + ".bmp"; const string dst_name = cv::tempfile((digit + ".bmp").c_str()); Mat image = imread(src_name); ASSERT_FALSE(image.empty()); resize(image, image, Size(968, 757), 0.0, 0.0, INTER_CUBIC); imwrite(dst_name, image); Mat loaded = imread(dst_name); ASSERT_FALSE(loaded.empty()); double psnr = cvtest::PSNR(loaded, image); EXPECT_GT(psnr, thresDbell); vector from_file; FILE *f = fopen(dst_name.c_str(), "rb"); fseek(f, 0, SEEK_END); long len = ftell(f); from_file.resize((size_t)len); fseek(f, 0, SEEK_SET); from_file.resize(fread(&from_file[0], 1, from_file.size(), f)); fclose(f); vector buf; imencode(".bmp", image, buf); ASSERT_EQ(buf, from_file); Mat buf_loaded = imdecode(Mat(buf), 1); ASSERT_FALSE(buf_loaded.empty()); psnr = cvtest::PSNR(buf_loaded, image); EXPECT_GT(psnr, thresDbell); EXPECT_EQ(0, remove(dst_name.c_str())); } } //================================================================================================== typedef string Ext; typedef testing::TestWithParam Imgcodecs_Image; TEST_P(Imgcodecs_Image, read_write) { const string ext = this->GetParam(); const string full_name = cv::tempfile(ext.c_str()); const string _name = TS::ptr()->get_data_path() + "../cv/shared/baboon.png"; const double thresDbell = 32; Mat image = imread(_name); image.convertTo(image, CV_8UC3); ASSERT_FALSE(image.empty()); imwrite(full_name, image); Mat loaded = imread(full_name); ASSERT_FALSE(loaded.empty()); double psnr = cvtest::PSNR(loaded, image); EXPECT_GT(psnr, thresDbell); vector from_file; FILE *f = fopen(full_name.c_str(), "rb"); fseek(f, 0, SEEK_END); long len = ftell(f); from_file.resize((size_t)len); fseek(f, 0, SEEK_SET); from_file.resize(fread(&from_file[0], 1, from_file.size(), f)); fclose(f); vector buf; imencode("." + ext, image, buf); ASSERT_EQ(buf, from_file); Mat buf_loaded = imdecode(Mat(buf), 1); ASSERT_FALSE(buf_loaded.empty()); psnr = cvtest::PSNR(buf_loaded, image); EXPECT_GT(psnr, thresDbell); EXPECT_EQ(0, remove(full_name.c_str())); } const string exts[] = { #ifdef HAVE_PNG "png", #endif #ifdef HAVE_TIFF "tiff", #endif #ifdef HAVE_JPEG "jpg", #endif #ifdef HAVE_JASPER "jp2", #endif #if 0 /*defined HAVE_OPENEXR && !defined __APPLE__*/ "exr", #endif "bmp", "ppm", "ras" }; INSTANTIATE_TEST_CASE_P(imgcodecs, Imgcodecs_Image, testing::ValuesIn(exts));