Chromium Code Reviews| 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 #ifndef SkCLImageDiffer_DEFINED | |
| 9 #define SkCLImageDiffer_DEFINED | |
| 10 | |
| 11 #include <CL/cl.h> | |
| 12 #include "SkTDArray.h" | |
| 13 | |
| 14 #include "SkImageDiffer.h" | |
| 15 | |
| 16 class SkStream; | |
| 17 | |
| 18 /** | |
| 19 * An SkImageDiffer that requires initialization with an OpenCL device and conte xt. | |
| 20 */ | |
| 21 class SkCLImageDiffer : public SkImageDiffer { | |
| 22 public: | |
| 23 SkCLImageDiffer(); | |
| 24 | |
| 25 /** | |
| 26 * Initializes the OpenCL resources this differ needs to work | |
| 27 * @param device An OpenCL device | |
| 28 * @param context An OpenCL context of the given device | |
| 29 * @return True on success, false otherwise | |
| 30 */ | |
| 31 virtual bool init(cl_device_id device, cl_context context) SK_OVERRIDE; | |
|
djsollen
2013/06/13 13:33:02
this doesn't appear to be an override.
| |
| 32 | |
| 33 protected: | |
| 34 /** | |
| 35 * Called by init after fDevice, fContext, and fCommandQueue are successfull y initialized | |
| 36 * @return True on success, false otherwise | |
| 37 */ | |
| 38 virtual bool onInit() = 0; | |
| 39 | |
| 40 /** | |
| 41 * Loads an OpenCL kernel from the file with the given named entry point. Th is only works after | |
| 42 * init is called. | |
| 43 * @param file The file path of the kernel | |
| 44 * @param name The name of the entry point of the desired kernel in the f ile | |
| 45 * @param kernel A pointer to return the loaded kernel into | |
| 46 * @return True on success, false otherwise | |
| 47 */ | |
| 48 bool loadKernelFile(const char file[], const char name[], cl_kernel* kernel) ; | |
| 49 | |
| 50 /** | |
| 51 * Loads an OpenCL kernel from the stream with the given named entry point. This only works | |
| 52 * after init is called. | |
| 53 * @param stream The stream that contains the kernel | |
| 54 * @param name The name of the entry point of the desired kernel in the stream | |
| 55 * @param kernel A pointer to return the loaded kernel into | |
| 56 * @return True on success, false otherwise | |
| 57 */ | |
| 58 bool loadKernelStream(SkStream* stream, const char name[], cl_kernel* kernel ); | |
| 59 | |
| 60 /** | |
| 61 * Loads an OpenCL kernel from the source string with the given named entry point. This only | |
| 62 * works after init is called. | |
| 63 * @param source The string that contains the kernel | |
| 64 * @param name The name of the entry point of the desired kernel in the source string | |
| 65 * @param kernel A pointer to return the loaded kernel into | |
| 66 * @return True on success, false otherwise | |
| 67 */ | |
| 68 bool loadKernelSource(const char source[], const char name[], cl_kernel* ker nel); | |
| 69 | |
| 70 /** | |
| 71 * Loads a read only copy of the given bitmap into device memory and returns the block of | |
| 72 * memory. This only works after init is called. | |
| 73 * @param bitmap The bitmap to load into memory | |
| 74 * @param image A pointer to return the allocated image to | |
| 75 * @return True on success, false otherwise | |
| 76 */ | |
| 77 bool makeImage2D(SkBitmap* bitmap, cl_mem* image); | |
| 78 | |
| 79 cl_device_id fDevice; | |
| 80 cl_context fContext; | |
| 81 cl_command_queue fCommandQueue; | |
| 82 | |
| 83 private: | |
| 84 | |
| 85 typedef SkImageDiffer INHERITED; | |
| 86 }; | |
| 87 | |
| 88 /** | |
| 89 * A OpenCL differ that measures the percentage of different corresponding pixel s. If the two images | |
| 90 * are not the same size or have no pixels, the result will always be zero. | |
| 91 */ | |
| 92 class SkDifferentPixelsImageDiffer : public SkCLImageDiffer { | |
| 93 public: | |
| 94 virtual const char* getName() SK_OVERRIDE; | |
| 95 virtual int queueDiff(SkBitmap* baseline, SkBitmap* test) SK_OVERRIDE; | |
| 96 virtual bool isFinished(int id) SK_OVERRIDE; | |
| 97 virtual double getResult(int id) SK_OVERRIDE; | |
| 98 | |
| 99 protected: | |
| 100 virtual bool onInit() SK_OVERRIDE; | |
| 101 | |
| 102 private: | |
| 103 struct QueuedDiff { | |
| 104 bool finished; | |
| 105 double result; | |
| 106 cl_mem baseline; | |
| 107 cl_mem test; | |
| 108 cl_mem resultsBuffer; | |
| 109 }; | |
| 110 | |
| 111 SkTDArray<QueuedDiff> fQueuedDiffs; | |
| 112 cl_kernel fKernel; | |
| 113 | |
| 114 typedef SkCLImageDiffer INHERITED; | |
| 115 }; | |
| 116 | |
| 117 #endif | |
| OLD | NEW |