OpenCV OpenEXR test (issue #10114)

test_io_exr.cpp

#include "test_precomp.hpp"

using namespace cv;

TEST(Imgcodecs, EXR_test)
{
    Mat m(cv::Size(2048, 1024), CV_32FC1);
    for (int y = 0; y < m.rows; y++)
    {
        for (int x = 0; x < m.cols; x++)
        {
            m.at<float>(y, x) = y + x*1e-3;
        }
    }

    Rect roi(498, 64, 8, 4);
    std::cout << "test ROI: " << roi << std::endl;
    std::cout << "source: " << std::endl;
    std::cout << m.type() << ": " << m.depth() << " x " << m.channels() << " " << m.size() << std::endl;
    std::cout << m(roi) << std::endl;

    {
        imwrite("test.exr", m);
        Mat m2 = imread("test.exr");
        std::cout << "imread (default): " << std::endl;
        std::cout << m2.type() << ": " << m2.depth() << " x " << m2.channels() << " " << m2.size() << std::endl;
        std::cout << m2(roi) << std::endl;
    }

    {
        imwrite("test.exr", m);
        Mat m2 = imread("test.exr", IMREAD_UNCHANGED);
        std::cout << "imread IMREAD_UNCHANGED: " << std::endl;
        std::cout << m2.type() << ": " << m2.depth() << " x " << m2.channels() << " " << m2.size() << std::endl;
        std::cout << m2(roi) << std::endl;
    }

    if (0) // crashed
    {
        imwrite("test.exr", m);
        Mat m2 = imread("test.exr", IMREAD_GRAYSCALE);
        std::cout << "imread IMREAD_GRAYSCALE: " << std::endl;
        std::cout << m2.type() << ": " << m2.depth() << " x " << m2.channels() << " " << m2.size() << std::endl;
        std::cout << m2(roi) << std::endl;
    }

    {
        int parameters_[] = {IMWRITE_EXR_TYPE, IMWRITE_EXR_TYPE_HALF};
        std::vector<int> parameters(parameters_, parameters_ + 2);
        imwrite("test_fp16.exr", m, parameters);
        Mat m2 = imread("test_fp16.exr");
        std::cout << "imread from fp16 (default): " << std::endl;
        std::cout << m2.type() << ": " << m2.depth() << " x " << m2.channels() << " " << m2.size() << std::endl;
        std::cout << m2(roi) << std::endl;
    }

    {
        int parameters_[] = {IMWRITE_EXR_TYPE, IMWRITE_EXR_TYPE_HALF};
        std::vector<int> parameters(parameters_, parameters_ + 2);
        imwrite("test_fp16.exr", m, parameters);
        Mat m2 = imread("test_fp16.exr", IMREAD_UNCHANGED);
        std::cout << "imread from fp16 IMREAD_UNCHANGED: " << std::endl;
        std::cout << m2.type() << ": " << m2.depth() << " x " << m2.channels() << " " << m2.size() << std::endl;
        std::cout << m2(roi) << std::endl;
    }
}