Function floodFill

src/backend/cuda/kernel/flood_fill.hpp:36–81

template<typename T>
void floodFill(Param<T> out, CParam<T> image, CParam<uint> seedsx,
               CParam<uint> seedsy, const T newValue, const T lowValue,
               const T highValue, const af::connectivity nlookup) {
    UNUSED(nlookup);
    if (sharedMemRequiredByFloodFill<T>() >
        getDeviceProp(getActiveDeviceId()).sharedMemPerBlock) {
        char errMessage[256];
        snprintf(errMessage, sizeof(errMessage),
                 "\nCurrent thread's CUDA device doesn't have sufficient "
                 "shared memory required by FloodFill\n");
        CUDA_NOT_SUPPORTED(errMessage);
    }

    auto initSeeds =
        common::getKernel("arrayfire::cuda::initSeeds", {{flood_fill_cuh_src}},
                          TemplateArgs(TemplateTypename<T>()));
    auto floodStep =
        common::getKernel("arrayfire::cuda::floodStep", {{flood_fill_cuh_src}},
                          TemplateArgs(TemplateTypename<T>()),
                          {{DefineValue(THREADS_X), DefineValue(THREADS_Y)}});
    auto finalizeOutput = common::getKernel(
        "arrayfire::cuda::finalizeOutput", {{flood_fill_cuh_src}},
        TemplateArgs(TemplateTypename<T>()));

    EnqueueArgs qArgs(dim3(divup(seedsx.elements(), THREADS)), dim3(THREADS),
                      getActiveStream());
    initSeeds(qArgs, out, seedsx, seedsy);
    POST_LAUNCH_CHECK();

    dim3 threads(THREADS_X, THREADS_Y);
    dim3 blocks(divup(image.dims[0], threads.x),
                divup(image.dims[1], threads.y));
    EnqueueArgs fQArgs(blocks, threads, getActiveStream());

    auto continueFlagPtr = floodStep.getDevPtr("doAnotherLaunch");

    for (int doAnotherLaunch = 1; doAnotherLaunch > 0;) {
        doAnotherLaunch = 0;
        floodStep.setFlag(continueFlagPtr, &doAnotherLaunch);
        floodStep(fQArgs, out, image, lowValue, highValue);
        POST_LAUNCH_CHECK();
        doAnotherLaunch = floodStep.getFlag(continueFlagPtr);
    }
    finalizeOutput(fQArgs, out, newValue);
    POST_LAUNCH_CHECK();
}

}  // namespace kernel
}  // namespace cuda