init

src/FastGrayStep.cc

@@ -0,0 +1,71 @@
+// 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