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 #if SK_SUPPORT_OPENCL | |
9 | |
10 #define __NO_STD_VECTOR // Uses cl::vectpr instead of std::vectpr | |
11 #define __NO_STD_STRING // Uses cl::STRING_CLASS instead of std::string | |
12 #if SK_BUILD_FOR_MAC | |
13 // Note that some macs don't have this header and it can be downloaded from the
Khronos registry | |
14 # include <OpenCL/cl.hpp> | |
15 #else | |
16 # include <CL/cl.hpp> | |
17 #endif | |
18 | |
19 #endif | |
20 | |
21 #include "SkCommandLineFlags.h" | |
22 #include "SkGraphics.h" | |
23 #include "SkStream.h" | |
24 #include "SkTDArray.h" | |
25 | |
26 #include "SkDifferentPixelsMetric.h" | |
27 #include "SkDiffContext.h" | |
28 #include "SkImageDiffer.h" | |
29 #include "SkPMetric.h" | |
30 #include "skpdiff_util.h" | |
31 | |
32 #include "SkForceLinking.h" | |
33 __SK_FORCE_IMAGE_DECODER_LINKING; | |
34 | |
35 // Command line argument definitions go here | |
36 DEFINE_bool2(list, l, false, "List out available differs"); | |
37 DEFINE_string2(differs, d, "", "The names of the differs to use or all of them b
y default"); | |
38 DEFINE_string2(folders, f, "", "Compare two folders with identical subfile names
: <baseline folder> <test folder>"); | |
39 DEFINE_string2(patterns, p, "", "Use two patterns to compare images: <baseline>
<test>"); | |
40 DEFINE_string2(output, o, "skpdiff_output.json", "Writes the output of these dif
fs to output: <output>"); | |
41 DEFINE_bool(jsonp, true, "Output JSON with padding"); | |
42 DEFINE_string(csv, "", "Writes the output of these diffs to a csv file"); | |
43 | |
44 #if SK_SUPPORT_OPENCL | |
45 /// A callback for any OpenCL errors | |
46 static void CL_CALLBACK error_notify(const char* errorInfo, const void* privateI
nfoSize, ::size_t cb, void* userData) { | |
47 SkDebugf("OpenCL error notify: %s\n", errorInfo); | |
48 exit(1); | |
49 } | |
50 | |
51 /// Creates a device and context with OpenCL | |
52 static bool init_device_and_context(cl::Device* device, cl::Context* context) { | |
53 // Query for a platform | |
54 cl::vector<cl::Platform> platformList; | |
55 cl::Platform::get(&platformList); | |
56 SkDebugf("The number of platforms is %u\n", platformList.size()); | |
57 | |
58 // Print some information about the platform for debugging | |
59 cl::Platform& platform = platformList[0]; | |
60 cl::STRING_CLASS platformName; | |
61 platform.getInfo(CL_PLATFORM_NAME, &platformName); | |
62 SkDebugf("Platform index 0 is named %s\n", platformName.c_str()); | |
63 | |
64 // Query for a device | |
65 cl::vector<cl::Device> deviceList; | |
66 platform.getDevices(CL_DEVICE_TYPE_ALL, &deviceList); | |
67 SkDebugf("The number of devices is %u\n", deviceList.size()); | |
68 | |
69 // Print some information about the device for debugging | |
70 *device = deviceList[0]; | |
71 cl::STRING_CLASS deviceName; | |
72 device->getInfo(CL_DEVICE_NAME, &deviceName); | |
73 SkDebugf("Device index 0 is named %s\n", deviceName.c_str()); | |
74 | |
75 // Create a CL context and check for all errors | |
76 cl_int contextErr = CL_SUCCESS; | |
77 *context = cl::Context(deviceList, NULL, error_notify, NULL, &contextErr); | |
78 if (contextErr != CL_SUCCESS) { | |
79 SkDebugf("Context creation failed: %s\n", cl_error_to_string(contextErr)
); | |
80 return false; | |
81 } | |
82 | |
83 return true; | |
84 } | |
85 | |
86 static bool init_cl_diff(SkImageDiffer* differ) { | |
87 // Setup OpenCL | |
88 cl::Device device; | |
89 cl::Context context; | |
90 if (!init_device_and_context(&device, &context)) { | |
91 return false; | |
92 } | |
93 | |
94 // Setup our differ of choice | |
95 SkCLImageDiffer* clDiffer = (SkCLImageDiffer*)differ; | |
96 return clDiffer->init(device(), context()); | |
97 } | |
98 #endif | |
99 | |
100 // TODO Find a better home for the diff registry. One possibility is to have the
differs self | |
101 // register. | |
102 | |
103 // List here every differ | |
104 SkDifferentPixelsMetric gDiffPixel; | |
105 SkPMetric gPDiff; | |
106 | |
107 // A null terminated array of pointer to every differ declared above | |
108 SkImageDiffer* gDiffers[] = { &gDiffPixel, &gPDiff, NULL }; | |
109 | |
110 int tool_main(int argc, char * argv[]); | |
111 int tool_main(int argc, char * argv[]) { | |
112 // Setup command line parsing | |
113 SkCommandLineFlags::SetUsage("Compare images using various metrics."); | |
114 SkCommandLineFlags::Parse(argc, argv); | |
115 | |
116 // Needed by various Skia components | |
117 SkAutoGraphics ag; | |
118 | |
119 if (FLAGS_list) { | |
120 SkDebugf("Available Metrics:\n"); | |
121 } | |
122 | |
123 // Figure which differs the user chose, and optionally print them if the use
r requests it | |
124 SkTDArray<SkImageDiffer*> chosenDiffers; | |
125 for (int differIndex = 0; NULL != gDiffers[differIndex]; differIndex++) { | |
126 SkImageDiffer* differ = gDiffers[differIndex]; | |
127 if (FLAGS_list) { | |
128 SkDebugf(" %s", differ->getName()); | |
129 SkDebugf("\n"); | |
130 } | |
131 | |
132 // Check if this differ was chosen by any of the flags. Initialize them
if they were chosen. | |
133 if (FLAGS_differs.isEmpty()) { | |
134 // If no differs were chosen, they all get added | |
135 if (differ->requiresOpenCL()) { | |
136 #if SK_SUPPORT_OPENCL | |
137 init_cl_diff(differ); | |
138 chosenDiffers.push(differ); | |
139 #endif | |
140 } else { | |
141 chosenDiffers.push(differ); | |
142 } | |
143 } else { | |
144 for (int flagIndex = 0; flagIndex < FLAGS_differs.count(); flagIndex
++) { | |
145 if (SkString(FLAGS_differs[flagIndex]).equals(differ->getName())
) { | |
146 // Initialize OpenCL for the differ if it needs it and suppo
rt was compiled in. | |
147 if (differ->requiresOpenCL()) { | |
148 #if SK_SUPPORT_OPENCL | |
149 init_cl_diff(differ); | |
150 chosenDiffers.push(differ); | |
151 #endif | |
152 } else { | |
153 chosenDiffers.push(differ); | |
154 } | |
155 break; | |
156 } | |
157 } | |
158 } | |
159 } | |
160 | |
161 // Don't attempt to initialize the differ if we aren't going to use it | |
162 if (FLAGS_folders.isEmpty() && FLAGS_patterns.isEmpty()) { | |
163 return 0; | |
164 } | |
165 | |
166 // Validate command line flags | |
167 if (!FLAGS_folders.isEmpty()) { | |
168 if (2 != FLAGS_folders.count()) { | |
169 SkDebugf("Folders flag expects two arguments: <baseline folder> <tes
t folder>\n"); | |
170 return 1; | |
171 } | |
172 } | |
173 | |
174 if (!FLAGS_patterns.isEmpty()) { | |
175 if (2 != FLAGS_patterns.count()) { | |
176 SkDebugf("Patterns flag expects two arguments: <baseline pattern> <t
est pattern>\n"); | |
177 return 1; | |
178 } | |
179 } | |
180 | |
181 if (!FLAGS_csv.isEmpty()) { | |
182 if (1 != FLAGS_csv.count()) { | |
183 SkDebugf("csv flag expects one argument: <csv file>\n"); | |
184 return 1; | |
185 } | |
186 } | |
187 | |
188 SkDiffContext ctx; | |
189 ctx.setDiffers(chosenDiffers); | |
190 | |
191 // Perform a folder diff if one is requested | |
192 if (!FLAGS_folders.isEmpty()) { | |
193 ctx.diffDirectories(FLAGS_folders[0], FLAGS_folders[1]); | |
194 } | |
195 | |
196 // Perform a pattern diff if one is requested | |
197 if (!FLAGS_patterns.isEmpty()) { | |
198 ctx.diffPatterns(FLAGS_patterns[0], FLAGS_patterns[1]); | |
199 } | |
200 | |
201 // Output to the file specified | |
202 if (!FLAGS_output.isEmpty()) { | |
203 SkFILEWStream outputStream(FLAGS_output[0]); | |
204 ctx.outputRecords(outputStream, FLAGS_jsonp); | |
205 } | |
206 | |
207 if (!FLAGS_csv.isEmpty()) { | |
208 SkFILEWStream outputStream(FLAGS_csv[0]); | |
209 ctx.outputCsv(outputStream); | |
210 } | |
211 | |
212 return 0; | |
213 } | |
214 | |
215 #if !defined(SK_BUILD_FOR_IOS) && !defined(SK_BUILD_FOR_NACL) | |
216 int main(int argc, char * argv[]) { | |
217 return tool_main(argc, (char**) argv); | |
218 } | |
219 #endif | |
OLD | NEW |