// Copyright (C) 2011  Dmitri Nikulin
// Copyright (C) 2011  Monash University
//
// Permission is hereby granted, free of charge, to any person
// obtaining a copy of this software and associated documentation
// files (the "Software"), to deal in the Software without
// restriction, including without limitation the rights to use,
// copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the
// Software is furnished to do so, subject to the following
// conditions:
//
// The above copyright notice and this permission notice shall be
// included in all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
// EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
// OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
// NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
// HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
// WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
// OTHER DEALINGS IN THE SOFTWARE.

#include "cvd-cl/steps/FastGrayStep.hh"
#include "kernels/fast-gray.hh"

namespace CVD {
namespace CL  {

FastGrayStep::FastGrayStep(GrayImageState & i_image, PointListState & i_points, PointListState & o_points, cl_int threshold, cl_int ring) :
    WorkerStep (i_image.worker),
    i_image    (i_image),
    i_points   (i_points),
    o_points   (o_points),
    threshold  (threshold),
    ring       (ring)
{
    char opt[256] = {0,};
    snprintf(opt, sizeof(opt) - 1, "-DFAST_THRESH=%d -DFAST_RING=%d -DFAST_COUNT=%d", int(threshold), int(ring), int(o_points.size));
    worker.compile(&program, &kernel, OCL_FAST_GRAY, "fast_gray", opt);
}

FastGrayStep::~FastGrayStep() {
    // Do nothing.
}

void FastGrayStep::execute() {
    // Read number of input points.
    size_t const np = i_points.getCount();

    // Assign kernel parameters.
    kernel.setArg(0, i_image.image);
    kernel.setArg(1, i_points.buffer);
    kernel.setArg(2, o_points.buffer);
    kernel.setArg(3, o_points.count);
    kernel.setArg(4, (cl_uint) np);

    // Reset number of output points.
    o_points.setCount(0);

    // Prepare global and local size.
    cl::NDRange const global(worker.padGlobal(np));
    cl::NDRange const local(worker.defaultLocal);

    // Queue kernel with global size set to number of input points.
    worker.queue.enqueueNDRangeKernel(kernel, cl::NullRange, global, local);
}

} // namespace CL
} // namespace CVD