Function floodFill

src/backend/opencl/kernel/flood_fill.hpp:71–112

template<typename T>
void floodFill(Param out, const Param image, const Param seedsx,
               const Param seedsy, const T newValue, const T lowValue,
               const T highValue, const af::connectivity nlookup) {
    constexpr int RADIUS = 1;

    UNUSED(nlookup);
    std::array<std::string, 11> options = {
        DefineKeyValue(T, dtype_traits<T>::getName()),
        DefineValue(RADIUS),
        DefineValue(VALID),
        DefineValue(INVALID),
        DefineValue(ZERO),
        DefineKey(FLOOD_FILL_STEP),
        DefineKeyValue(LMEM_WIDTH, (THREADS_X + 2 * RADIUS)),
        DefineKeyValue(LMEM_HEIGHT, (THREADS_Y + 2 * RADIUS)),
        DefineKeyValue(GROUP_SIZE, (THREADS_Y * THREADS_X)),
        getTypeBuildDefinition<T>()};

    auto floodStep =
        common::getKernel("flood_step", {{flood_fill_cl_src}},
                          TemplateArgs(TemplateTypename<T>()), options);
    cl::NDRange local(kernel::THREADS_X, kernel::THREADS_Y, 1);
    cl::NDRange global(divup(out.info.dims[0], local[0]) * local[0],
                       divup(out.info.dims[1], local[1]) * local[1], 1);

    initSeeds<T>(out, seedsx, seedsy);

    int notFinished       = 1;
    cl::Buffer* dContinue = bufferAlloc(sizeof(int));

    while (notFinished) {
        notFinished = 0;
        floodStep.setFlag(dContinue, &notFinished);
        floodStep(cl::EnqueueArgs(getQueue(), global, local), *out.data,
                  out.info, *image.data, image.info, lowValue, highValue,
                  *dContinue);
        CL_DEBUG_FINISH(getQueue());
        notFinished = floodStep.getFlag(dContinue);
    }
    bufferFree(dContinue);
    finalizeOutput<T>(out, newValue);
}

}  // namespace kernel
}  // namespace opencl