| OLD | NEW |
|---|
| (Empty) |
| 1 /* | |
| 2 * Copyright 2013 Google Inc. | |
| 3 * | |
| 4 * Use of this source code is governed by a BSD-style license that can be | |
| 5 * found in the LICENSE file. | |
| 6 */ | |
| 7 | |
| 8 #include "SkBitmap.h" | |
| 9 | |
| 10 #include "SkDifferentPixelsMetric.h" | |
| 11 #include "skpdiff_util.h" | |
| 12 | |
| 13 static const char kDifferentPixelsKernelSource[] = | |
| 14 "#pragma OPENCL_EXTENSION cl_khr_global_int32_base_atomics
\n" | |
| 15 "
\n" | |
| 16 "const sampler_t gInSampler = CLK_NORMALIZED_COORDS_FALSE |
\n" | |
| 17 " CLK_ADDRESS_CLAMP_TO_EDGE |
\n" | |
| 18 " CLK_FILTER_NEAREST;
\n" | |
| 19 "
\n" | |
| 20 "__kernel void diff(read_only image2d_t baseline, read_only image2d_t test,
\n" | |
| 21 " __global int* result, __global int2* poi) {
\n" | |
| 22 " int2 coord = (int2)(get_global_id(0), get_global_id(1));
\n" | |
| 23 " uint4 baselinePixel = read_imageui(baseline, gInSampler, coord);
\n" | |
| 24 " uint4 testPixel = read_imageui(test, gInSampler, coord);
\n" | |
| 25 " int4 pixelCompare = baselinePixel == testPixel;
\n" | |
| 26 " if (baselinePixel.x != testPixel.x ||
\n" | |
| 27 " baselinePixel.y != testPixel.y ||
\n" | |
| 28 " baselinePixel.z != testPixel.z ||
\n" | |
| 29 " baselinePixel.w != testPixel.w) {
\n" | |
| 30 "
\n" | |
| 31 " int poiIndex = atomic_inc(result);
\n" | |
| 32 " poi[poiIndex] = coord;
\n" | |
| 33 " }
\n" | |
| 34 "}
\n"; | |
| 35 | |
| 36 struct SkDifferentPixelsMetric::QueuedDiff { | |
| 37 bool finished; | |
| 38 double result; | |
| 39 int numDiffPixels; | |
| 40 SkIPoint* poi; | |
| 41 cl_mem baseline; | |
| 42 cl_mem test; | |
| 43 cl_mem resultsBuffer; | |
| 44 cl_mem poiBuffer; | |
| 45 }; | |
| 46 | |
| 47 const char* SkDifferentPixelsMetric::getName() { | |
| 48 return "different_pixels"; | |
| 49 } | |
| 50 | |
| 51 int SkDifferentPixelsMetric::queueDiff(SkBitmap* baseline, SkBitmap* test) { | |
| 52 int diffID = fQueuedDiffs.count(); | |
| 53 double startTime = get_seconds(); | |
| 54 QueuedDiff* diff = fQueuedDiffs.push(); | |
| 55 | |
| 56 // If we never end up running the kernel, include some safe defaults in the
result. | |
| 57 diff->finished = false; | |
| 58 diff->result = -1.0; | |
| 59 diff->numDiffPixels = 0; | |
| 60 diff->poi = NULL; | |
| 61 | |
| 62 // Ensure the images are comparable | |
| 63 if (baseline->width() != test->width() || baseline->height() != test->height
() || | |
| 64 baseline->width() <= 0 || baseline->height() <= 0 || | |
| 65 baseline->config() != test->config()) { | |
| 66 diff->finished = true; | |
| 67 return diffID; | |
| 68 } | |
| 69 | |
| 70 // Upload images to the CL device | |
| 71 if (!this->makeImage2D(baseline, &diff->baseline) || !this->makeImage2D(test
, &diff->test)) { | |
| 72 diff->finished = true; | |
| 73 fIsGood = false; | |
| 74 return -1; | |
| 75 } | |
| 76 | |
| 77 // A small hack that makes calculating percentage difference easier later on
. | |
| 78 diff->result = 1.0 / ((double)baseline->width() * baseline->height()); | |
| 79 | |
| 80 // Make a buffer to store results into. It must be initialized with pointers
to memory. | |
| 81 static const int kZero = 0; | |
| 82 // We know OpenCL won't write to it because we use CL_MEM_COPY_HOST_PTR | |
| 83 diff->resultsBuffer = clCreateBuffer(fContext, CL_MEM_READ_WRITE | CL_MEM_CO
PY_HOST_PTR, | |
| 84 sizeof(int), (int*)&kZero, NULL); | |
| 85 | |
| 86 diff->poiBuffer = clCreateBuffer(fContext, CL_MEM_WRITE_ONLY, | |
| 87 sizeof(int) * 2 * baseline->width() * basel
ine->height(), | |
| 88 NULL, NULL); | |
| 89 | |
| 90 // Set all kernel arguments | |
| 91 cl_int setArgErr = clSetKernelArg(fKernel, 0, sizeof(cl_mem), &diff->baselin
e); | |
| 92 setArgErr |= clSetKernelArg(fKernel, 1, sizeof(cl_mem), &diff->test); | |
| 93 setArgErr |= clSetKernelArg(fKernel, 2, sizeof(cl_mem), &diff->results
Buffer); | |
| 94 setArgErr |= clSetKernelArg(fKernel, 3, sizeof(cl_mem), &diff->poiBuff
er); | |
| 95 if (CL_SUCCESS != setArgErr) { | |
| 96 SkDebugf("Set arg failed: %s\n", cl_error_to_string(setArgErr)); | |
| 97 fIsGood = false; | |
| 98 return -1; | |
| 99 } | |
| 100 | |
| 101 // Queue this diff on the CL device | |
| 102 cl_event event; | |
| 103 const size_t workSize[] = { baseline->width(), baseline->height() }; | |
| 104 cl_int enqueueErr; | |
| 105 enqueueErr = clEnqueueNDRangeKernel(fCommandQueue, fKernel, 2, NULL, workSiz
e, | |
| 106 NULL, 0, NULL, &event); | |
| 107 if (CL_SUCCESS != enqueueErr) { | |
| 108 SkDebugf("Enqueue failed: %s\n", cl_error_to_string(enqueueErr)); | |
| 109 fIsGood = false; | |
| 110 return -1; | |
| 111 } | |
| 112 | |
| 113 // This makes things totally synchronous. Actual queue is not ready yet | |
| 114 clWaitForEvents(1, &event); | |
| 115 diff->finished = true; | |
| 116 | |
| 117 // Immediate read back the results | |
| 118 clEnqueueReadBuffer(fCommandQueue, diff->resultsBuffer, CL_TRUE, 0, | |
| 119 sizeof(int), &diff->numDiffPixels, 0, NULL, NULL); | |
| 120 diff->result *= (double)diff->numDiffPixels; | |
| 121 diff->result = (1.0 - diff->result); | |
| 122 | |
| 123 diff->poi = SkNEW_ARRAY(SkIPoint, diff->numDiffPixels); | |
| 124 clEnqueueReadBuffer(fCommandQueue, diff->poiBuffer, CL_TRUE, 0, | |
| 125 sizeof(SkIPoint) * diff->numDiffPixels, diff->poi, 0, NU
LL, NULL); | |
| 126 | |
| 127 // Release all the buffers created | |
| 128 clReleaseMemObject(diff->poiBuffer); | |
| 129 clReleaseMemObject(diff->resultsBuffer); | |
| 130 clReleaseMemObject(diff->baseline); | |
| 131 clReleaseMemObject(diff->test); | |
| 132 | |
| 133 SkDebugf("Time: %f\n", (get_seconds() - startTime)); | |
| 134 | |
| 135 return diffID; | |
| 136 } | |
| 137 | |
| 138 void SkDifferentPixelsMetric::deleteDiff(int id) { | |
| 139 QueuedDiff* diff = &fQueuedDiffs[id]; | |
| 140 if (NULL != diff->poi) { | |
| 141 SkDELETE_ARRAY(diff->poi); | |
| 142 diff->poi = NULL; | |
| 143 } | |
| 144 } | |
| 145 | |
| 146 bool SkDifferentPixelsMetric::isFinished(int id) { | |
| 147 return fQueuedDiffs[id].finished; | |
| 148 } | |
| 149 | |
| 150 double SkDifferentPixelsMetric::getResult(int id) { | |
| 151 return fQueuedDiffs[id].result; | |
| 152 } | |
| 153 | |
| 154 int SkDifferentPixelsMetric::getPointsOfInterestCount(int id) { | |
| 155 return fQueuedDiffs[id].numDiffPixels; | |
| 156 } | |
| 157 | |
| 158 SkIPoint* SkDifferentPixelsMetric::getPointsOfInterest(int id) { | |
| 159 return fQueuedDiffs[id].poi; | |
| 160 } | |
| 161 | |
| 162 bool SkDifferentPixelsMetric::onInit() { | |
| 163 if (!this->loadKernelSource(kDifferentPixelsKernelSource, "diff", &fKernel))
{ | |
| 164 return false; | |
| 165 } | |
| 166 | |
| 167 return true; | |
| 168 } | |
| OLD | NEW |
|---|
Chromium Code Reviews
