Function testImage

test/confidence_connected.cpp:45–112

template<typename T>
void testImage(const std::string pTestFile, const size_t numSeeds,
               const unsigned *seedx, const unsigned *seedy,
               const int multiplier, const unsigned neighborhood_radius,
               const int iter) {
    SUPPORTED_TYPE_CHECK(T);
    IMAGEIO_ENABLED_CHECK();

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

    readImageTests(std::string(TEST_DIR) + "/confidence_cc/" + pTestFile,
                   inDims, inFiles, outSizes, outFiles);

    size_t testCount = inDims.size();

    af_array seedxArr = 0, seedyArr = 0;
    dim4 seedDims(numSeeds);
    ASSERT_SUCCESS(af_create_array(&seedxArr, seedx, seedDims.ndims(),
                                   seedDims.get(), u32));
    ASSERT_SUCCESS(af_create_array(&seedyArr, seedy, seedDims.ndims(),
                                   seedDims.get(), u32));

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

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

        ASSERT_SUCCESS(
            af_load_image(&_inArray, inFiles[testId].c_str(), false));
        ASSERT_SUCCESS(
            af_load_image(&_goldArray, outFiles[testId].c_str(), false));

        // af_load_image always returns float array, so convert to output type
        ASSERT_SUCCESS(conv_image<T>(&inArray, _inArray));
        ASSERT_SUCCESS(conv_image<T>(&goldArray, _goldArray));

        CCCTestParams params;
        params.prefix     = "Image";
        params.radius     = neighborhood_radius;
        params.multiplier = multiplier;
        params.iterations = iter;
        params.replace    = 255.0;

        ASSERT_SUCCESS(af_confidence_cc(&outArray, inArray, seedxArr, seedyArr,
                                        params.radius, params.multiplier,
                                        params.iterations, params.replace));
        int device = 0;
        ASSERT_SUCCESS(af_get_device(&device));
        ASSERT_SUCCESS(af_sync(device));

        ASSERT_ARRAYS_EQ(outArray, goldArray);