| Index: experimental/skpdiff/SkCLImageDiffer.cpp
|
| diff --git a/experimental/skpdiff/SkCLImageDiffer.cpp b/experimental/skpdiff/SkCLImageDiffer.cpp
|
| index 0c27e0989b6c852fdfbf169821cfc0a5fbb2a721..d1e6958ab9f77cb1324ab545d19bc3699f87fe85 100644
|
| --- a/experimental/skpdiff/SkCLImageDiffer.cpp
|
| +++ b/experimental/skpdiff/SkCLImageDiffer.cpp
|
| @@ -1,4 +1,3 @@
|
| -
|
| /*
|
| * Copyright 2013 Google Inc.
|
| *
|
| @@ -18,7 +17,6 @@ SkCLImageDiffer::SkCLImageDiffer() {
|
| fIsGood = false;
|
| }
|
|
|
| -
|
| bool SkCLImageDiffer::init(cl_device_id device, cl_context context) {
|
| fContext = context;
|
| fDevice = device;
|
| @@ -122,139 +120,3 @@ bool SkCLImageDiffer::makeImage2D(SkBitmap* bitmap, cl_mem* image) {
|
|
|
| return true;
|
| }
|
| -
|
| -
|
| -////////////////////////////////////////////////////////////////
|
| -
|
| -struct SkDifferentPixelsImageDiffer::QueuedDiff {
|
| - bool finished;
|
| - double result;
|
| - int numDiffPixels;
|
| - SkIPoint* poi;
|
| - cl_mem baseline;
|
| - cl_mem test;
|
| - cl_mem resultsBuffer;
|
| - cl_mem poiBuffer;
|
| -};
|
| -
|
| -const char* SkDifferentPixelsImageDiffer::getName() {
|
| - return "different_pixels";
|
| -}
|
| -
|
| -int SkDifferentPixelsImageDiffer::queueDiff(SkBitmap * baseline, SkBitmap * test) {
|
| - int diffID = fQueuedDiffs.count();
|
| - double startTime = get_seconds();
|
| - QueuedDiff* diff = fQueuedDiffs.push();
|
| -
|
| - // If we never end up running the kernel, include some safe defaults in the result.
|
| - diff->finished = false;
|
| - diff->result = -1.0;
|
| - diff->numDiffPixels = 0;
|
| - diff->poi = NULL;
|
| -
|
| - // Ensure the images are comparable
|
| - if (baseline->width() != test->width() || baseline->height() != test->height() ||
|
| - baseline->width() <= 0 || baseline->height() <= 0) {
|
| - diff->finished = true;
|
| - return diffID;
|
| - }
|
| -
|
| - // Upload images to the CL device
|
| - if (!this->makeImage2D(baseline, &diff->baseline) || !this->makeImage2D(test, &diff->test)) {
|
| - diff->finished = true;
|
| - fIsGood = false;
|
| - return -1;
|
| - }
|
| -
|
| - // A small hack that makes calculating percentage difference easier later on.
|
| - diff->result = 1.0 / ((double)baseline->width() * baseline->height());
|
| -
|
| - // Make a buffer to store results into. It must be initialized with pointers to memory.
|
| - static const int kZero = 0;
|
| - // We know OpenCL won't write to it because we use CL_MEM_COPY_HOST_PTR
|
| - diff->resultsBuffer = clCreateBuffer(fContext, CL_MEM_READ_WRITE | CL_MEM_COPY_HOST_PTR,
|
| - sizeof(int), (int*)&kZero, NULL);
|
| -
|
| - diff->poiBuffer = clCreateBuffer(fContext, CL_MEM_WRITE_ONLY,
|
| - sizeof(int) * 2 * baseline->width() * baseline->height(),
|
| - NULL, NULL);
|
| -
|
| - // Set all kernel arguments
|
| - cl_int setArgErr = clSetKernelArg(fKernel, 0, sizeof(cl_mem), &diff->baseline);
|
| - setArgErr |= clSetKernelArg(fKernel, 1, sizeof(cl_mem), &diff->test);
|
| - setArgErr |= clSetKernelArg(fKernel, 2, sizeof(cl_mem), &diff->resultsBuffer);
|
| - setArgErr |= clSetKernelArg(fKernel, 3, sizeof(cl_mem), &diff->poiBuffer);
|
| - if (CL_SUCCESS != setArgErr) {
|
| - SkDebugf("Set arg failed: %s\n", cl_error_to_string(setArgErr));
|
| - fIsGood = false;
|
| - return -1;
|
| - }
|
| -
|
| - // Queue this diff on the CL device
|
| - cl_event event;
|
| - const size_t workSize[] = { baseline->width(), baseline->height() };
|
| - cl_int enqueueErr;
|
| - enqueueErr = clEnqueueNDRangeKernel(fCommandQueue, fKernel, 2, NULL, workSize,
|
| - NULL, 0, NULL, &event);
|
| - if (CL_SUCCESS != enqueueErr) {
|
| - SkDebugf("Enqueue failed: %s\n", cl_error_to_string(enqueueErr));
|
| - fIsGood = false;
|
| - return -1;
|
| - }
|
| -
|
| - // This makes things totally synchronous. Actual queue is not ready yet
|
| - clWaitForEvents(1, &event);
|
| - diff->finished = true;
|
| -
|
| - // Immediate read back the results
|
| - clEnqueueReadBuffer(fCommandQueue, diff->resultsBuffer, CL_TRUE, 0,
|
| - sizeof(int), &diff->numDiffPixels, 0, NULL, NULL);
|
| - diff->result *= (double)diff->numDiffPixels;
|
| - diff->result = (1.0 - diff->result);
|
| -
|
| - diff->poi = SkNEW_ARRAY(SkIPoint, diff->numDiffPixels);
|
| - clEnqueueReadBuffer(fCommandQueue, diff->poiBuffer, CL_TRUE, 0,
|
| - sizeof(SkIPoint) * diff->numDiffPixels, diff->poi, 0, NULL, NULL);
|
| -
|
| - // Release all the buffers created
|
| - clReleaseMemObject(diff->poiBuffer);
|
| - clReleaseMemObject(diff->resultsBuffer);
|
| - clReleaseMemObject(diff->baseline);
|
| - clReleaseMemObject(diff->test);
|
| -
|
| - SkDebugf("Time: %f\n", (get_seconds() - startTime));
|
| -
|
| - return diffID;
|
| -}
|
| -
|
| -void SkDifferentPixelsImageDiffer::deleteDiff(int id) {
|
| - QueuedDiff* diff = &fQueuedDiffs[id];
|
| - if (NULL != diff->poi) {
|
| - SkDELETE_ARRAY(diff->poi);
|
| - diff->poi = NULL;
|
| - }
|
| -}
|
| -
|
| -bool SkDifferentPixelsImageDiffer::isFinished(int id) {
|
| - return fQueuedDiffs[id].finished;
|
| -}
|
| -
|
| -double SkDifferentPixelsImageDiffer::getResult(int id) {
|
| - return fQueuedDiffs[id].result;
|
| -}
|
| -
|
| -int SkDifferentPixelsImageDiffer::getPointsOfInterestCount(int id) {
|
| - return fQueuedDiffs[id].numDiffPixels;
|
| -}
|
| -
|
| -SkIPoint* SkDifferentPixelsImageDiffer::getPointsOfInterest(int id) {
|
| - return fQueuedDiffs[id].poi;
|
| -}
|
| -
|
| -bool SkDifferentPixelsImageDiffer::onInit() {
|
| - if (!loadKernelFile("experimental/skpdiff/diff_pixels.cl", "diff", &fKernel)) {
|
| - return false;
|
| - }
|
| -
|
| - return true;
|
| -}
|
|
|