71 lines
2.5 KiB
C++
71 lines
2.5 KiB
C++
// 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/PreFastGrayStep.hh"
|
|
#include "kernels/prefast-gray.hh"
|
|
|
|
namespace CVD {
|
|
namespace CL {
|
|
|
|
PreFastGrayStep::PreFastGrayStep(GrayImageState & i_image, PointListState & o_points, cl_int threshold) :
|
|
WorkerStep (i_image.worker),
|
|
i_image (i_image),
|
|
o_points (o_points),
|
|
threshold (threshold)
|
|
{
|
|
char opt[256] = {0,};
|
|
snprintf(opt, sizeof(opt) - 1, "-DFAST_THRESH=%d -DFAST_COUNT=%d", int(threshold), int(o_points.size));
|
|
worker.compile(&program, &kernel, OCL_PRE_FAST_GRAY, "prefast_gray", opt);
|
|
}
|
|
|
|
PreFastGrayStep::~PreFastGrayStep() {
|
|
// Do nothing.
|
|
}
|
|
|
|
void PreFastGrayStep::execute() {
|
|
// Reset number of output points.
|
|
o_points.setCount(0);
|
|
|
|
// Assign kernel parameters.
|
|
kernel.setArg(0, i_image.image);
|
|
kernel.setArg(1, o_points.buffer);
|
|
kernel.setArg(2, o_points.count);
|
|
|
|
// Read image dimensions.
|
|
size_t const nx = i_image.nx - 16;
|
|
size_t const ny = i_image.ny - 16;
|
|
|
|
// Construct global, local and offset with a safety boundary.
|
|
// 16x16 appears to give good performance on most devices.
|
|
cl::NDRange const global (nx - 16, ny - 16);
|
|
cl::NDRange const local ( 8, 8);
|
|
cl::NDRange const offset ( 8, 8);
|
|
|
|
// Queue kernel.
|
|
worker.queue.enqueueNDRangeKernel(kernel, offset, global, local);
|
|
}
|
|
|
|
} // namespace CL
|
|
} // namespace CVD
|