Function morphImageTest

test/morph.cpp:152–219

template<typename T, bool isDilation, bool isColor>
void morphImageTest(string pTestFile, dim_t seLen) {
    SUPPORTED_TYPE_CHECK(T);
    IMAGEIO_ENABLED_CHECK();

    vector<dim4> inDims;
    vector<string> inFiles;
    vector<dim_t> outSizes;
    vector<string> outFiles;

    readImageTests(pTestFile, inDims, inFiles, outSizes, outFiles);

    size_t testCount = inDims.size();

    for (size_t testId = 0; testId < testCount; ++testId) {
        af_array _inArray   = 0;
        af_array inArray    = 0;
        af_array maskArray  = 0;
        af_array outArray   = 0;
        af_array _goldArray = 0;
        af_array goldArray  = 0;
        dim_t nElems        = 0;

        inFiles[testId].insert(0, string(TEST_DIR "/morph/"));
        outFiles[testId].insert(0, string(TEST_DIR "/morph/"));

        af_dtype targetType = static_cast<af_dtype>(dtype_traits<T>::af_type);

        dim4 mdims(seLen, seLen, 1, 1);
        ASSERT_SUCCESS(af_constant(&maskArray, 1.0, mdims.ndims(), mdims.get(),
                                   targetType));

        ASSERT_SUCCESS(
            af_load_image(&_inArray, inFiles[testId].c_str(), isColor));
        ASSERT_SUCCESS(af_cast(&inArray, _inArray, targetType));

        ASSERT_SUCCESS(
            af_load_image(&_goldArray, outFiles[testId].c_str(), isColor));
        ASSERT_SUCCESS(af_cast(&goldArray, _goldArray, targetType));

        ASSERT_SUCCESS(af_get_elements(&nElems, goldArray));

        af_err error_code = AF_SUCCESS;
        if (isDilation) {
            error_code = af_dilate(&outArray, inArray, maskArray);
        } else {
            error_code = af_erode(&outArray, inArray, maskArray);
        }

#if defined(AF_CPU)
        ASSERT_SUCCESS(error_code);
        ASSERT_IMAGES_NEAR(goldArray, outArray, 0.018f);
#else
        if (targetType != b8 && seLen > 19) {
            ASSERT_EQ(error_code, AF_ERR_NOT_SUPPORTED);
        } else {
            ASSERT_SUCCESS(error_code);
            ASSERT_IMAGES_NEAR(goldArray, outArray, 0.018f);
        }