junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2012 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 | |
commit-bot@chromium.org | 56799e2 | 2013-07-16 18:21:46 +0000 | [diff] [blame] | 8 | #include "LazyDecodeBitmap.h" |
scroggo@google.com | 4a26d9d | 2012-11-07 18:01:46 +0000 | [diff] [blame] | 9 | #include "CopyTilesRenderer.h" |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 10 | #include "SkBitmap.h" |
keyar@chromium.org | 472b379 | 2012-07-20 22:34:27 +0000 | [diff] [blame] | 11 | #include "SkDevice.h" |
scroggo@google.com | d9ba9a0 | 2013-03-21 19:43:15 +0000 | [diff] [blame] | 12 | #include "SkCommandLineFlags.h" |
borenet@google.com | 10ef79e | 2012-09-10 17:19:06 +0000 | [diff] [blame] | 13 | #include "SkGraphics.h" |
scroggo@google.com | 5a7c6be | 2012-10-04 21:46:08 +0000 | [diff] [blame] | 14 | #include "SkImageDecoder.h" |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 15 | #include "SkImageEncoder.h" |
keyar@chromium.org | f4959ab | 2012-08-23 20:53:25 +0000 | [diff] [blame] | 16 | #include "SkMath.h" |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 17 | #include "SkOSFile.h" |
| 18 | #include "SkPicture.h" |
robertphillips@google.com | 770963f | 2014-04-18 18:04:41 +0000 | [diff] [blame] | 19 | #include "SkPictureRecorder.h" |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 20 | #include "SkStream.h" |
| 21 | #include "SkString.h" |
commit-bot@chromium.org | 90c0fbd | 2014-05-09 03:18:41 +0000 | [diff] [blame] | 22 | |
| 23 | #include "image_expectations.h" |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 24 | #include "PictureRenderer.h" |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 25 | #include "PictureRenderingFlags.h" |
twiz@google.com | a31b8bb | 2012-06-22 18:24:56 +0000 | [diff] [blame] | 26 | #include "picture_utils.h" |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 27 | |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 28 | // Flags used by this file, alphabetically: |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 29 | DECLARE_bool(deferImageDecoding); |
| 30 | DEFINE_int32(maxComponentDiff, 256, "Maximum diff on a component, 0 - 256. Components that differ " |
| 31 | "by more than this amount are considered errors, though all diffs are reported. " |
| 32 | "Requires --validate."); |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 33 | DEFINE_string(mismatchPath, "", "Write images for tests that failed due to " |
| 34 | "pixel mismatches into this directory."); |
commit-bot@chromium.org | 205ce48 | 2014-05-12 15:37:20 +0000 | [diff] [blame] | 35 | DEFINE_string(readJsonSummaryPath, "", "JSON file to read image expectations from."); |
scroggo@google.com | 604e0c2 | 2013-04-09 21:25:46 +0000 | [diff] [blame] | 36 | DECLARE_string(readPath); |
commit-bot@chromium.org | f5e315c | 2014-03-19 17:26:07 +0000 | [diff] [blame] | 37 | DEFINE_bool(writeChecksumBasedFilenames, false, |
| 38 | "When writing out images, use checksum-based filenames."); |
scroggo@google.com | 1125d39 | 2013-05-03 20:43:37 +0000 | [diff] [blame] | 39 | DEFINE_bool(writeEncodedImages, false, "Any time the skp contains an encoded image, write it to a " |
| 40 | "file rather than decoding it. Requires writePath to be set. Skips drawing the full " |
| 41 | "skp to a file. Not compatible with deferImageDecoding."); |
commit-bot@chromium.org | 4610a46 | 2014-04-29 19:39:22 +0000 | [diff] [blame] | 42 | DEFINE_string(writeJsonSummaryPath, "", "File to write a JSON summary of image results to."); |
| 43 | DEFINE_string2(writePath, w, "", "Directory to write the rendered images into."); |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 44 | DEFINE_bool(writeWholeImage, false, "In tile mode, write the entire rendered image to a " |
| 45 | "file, instead of an image for each tile."); |
| 46 | DEFINE_bool(validate, false, "Verify that the rendered image contains the same pixels as " |
junov@chromium.org | e286e84 | 2013-03-13 17:27:16 +0000 | [diff] [blame] | 47 | "the picture rendered in simple mode. When used in conjunction with --bbh, results " |
| 48 | "are validated against the picture rendered in the same mode, but without the bbh."); |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 49 | |
commit-bot@chromium.org | ff36a1d | 2013-07-24 20:37:30 +0000 | [diff] [blame] | 50 | DEFINE_bool(bench_record, false, "If true, drop into an infinite loop of recording the picture."); |
| 51 | |
commit-bot@chromium.org | 145d1c0 | 2014-03-16 19:46:36 +0000 | [diff] [blame] | 52 | DEFINE_bool(preprocess, false, "If true, perform device specific preprocessing before rendering."); |
| 53 | |
scroggo@google.com | 1125d39 | 2013-05-03 20:43:37 +0000 | [diff] [blame] | 54 | //////////////////////////////////////////////////////////////////////////////////////////////////// |
| 55 | |
| 56 | /** |
| 57 | * Table for translating from format of data to a suffix. |
| 58 | */ |
| 59 | struct Format { |
| 60 | SkImageDecoder::Format fFormat; |
| 61 | const char* fSuffix; |
| 62 | }; |
| 63 | static const Format gFormats[] = { |
| 64 | { SkImageDecoder::kBMP_Format, ".bmp" }, |
| 65 | { SkImageDecoder::kGIF_Format, ".gif" }, |
| 66 | { SkImageDecoder::kICO_Format, ".ico" }, |
| 67 | { SkImageDecoder::kJPEG_Format, ".jpg" }, |
| 68 | { SkImageDecoder::kPNG_Format, ".png" }, |
| 69 | { SkImageDecoder::kWBMP_Format, ".wbmp" }, |
| 70 | { SkImageDecoder::kWEBP_Format, ".webp" }, |
| 71 | { SkImageDecoder::kUnknown_Format, "" }, |
| 72 | }; |
| 73 | |
| 74 | /** |
| 75 | * Get an appropriate suffix for an image format. |
| 76 | */ |
| 77 | static const char* get_suffix_from_format(SkImageDecoder::Format format) { |
| 78 | for (size_t i = 0; i < SK_ARRAY_COUNT(gFormats); i++) { |
| 79 | if (gFormats[i].fFormat == format) { |
| 80 | return gFormats[i].fSuffix; |
| 81 | } |
| 82 | } |
| 83 | return ""; |
| 84 | } |
| 85 | |
| 86 | /** |
| 87 | * Base name for an image file created from the encoded data in an skp. |
| 88 | */ |
| 89 | static SkString gInputFileName; |
| 90 | |
| 91 | /** |
| 92 | * Number to be appended to the image file name so that it is unique. |
| 93 | */ |
| 94 | static uint32_t gImageNo; |
| 95 | |
| 96 | /** |
| 97 | * Set up the name for writing encoded data to a file. |
| 98 | * Sets gInputFileName to name, minus any extension ".*" |
| 99 | * Sets gImageNo to 0, so images from file "X.skp" will |
| 100 | * look like "X_<gImageNo>.<suffix>", beginning with 0 |
| 101 | * for each new skp. |
| 102 | */ |
| 103 | static void reset_image_file_base_name(const SkString& name) { |
| 104 | gImageNo = 0; |
| 105 | // Remove ".skp" |
| 106 | const char* cName = name.c_str(); |
| 107 | const char* dot = strrchr(cName, '.'); |
| 108 | if (dot != NULL) { |
| 109 | gInputFileName.set(cName, dot - cName); |
| 110 | } else { |
| 111 | gInputFileName.set(name); |
| 112 | } |
| 113 | } |
| 114 | |
| 115 | /** |
| 116 | * Write the raw encoded bitmap data to a file. |
| 117 | */ |
| 118 | static bool write_image_to_file(const void* buffer, size_t size, SkBitmap* bitmap) { |
| 119 | SkASSERT(!FLAGS_writePath.isEmpty()); |
| 120 | SkMemoryStream memStream(buffer, size); |
| 121 | SkString outPath; |
| 122 | SkImageDecoder::Format format = SkImageDecoder::GetStreamFormat(&memStream); |
| 123 | SkString name = SkStringPrintf("%s_%d%s", gInputFileName.c_str(), gImageNo++, |
| 124 | get_suffix_from_format(format)); |
| 125 | SkString dir(FLAGS_writePath[0]); |
tfarina | da4ed32 | 2014-06-12 08:50:56 -0700 | [diff] [blame] | 126 | outPath = SkOSPath::SkPathJoin(dir.c_str(), name.c_str()); |
scroggo@google.com | 1125d39 | 2013-05-03 20:43:37 +0000 | [diff] [blame] | 127 | SkFILEWStream fileStream(outPath.c_str()); |
| 128 | if (!(fileStream.isValid() && fileStream.write(buffer, size))) { |
| 129 | SkDebugf("Failed to write encoded data to \"%s\"\n", outPath.c_str()); |
| 130 | } |
| 131 | // Put in a dummy bitmap. |
reed | bfefc7c | 2014-06-12 17:40:00 -0700 | [diff] [blame] | 132 | return SkImageDecoder::DecodeStream(&memStream, bitmap, kUnknown_SkColorType, |
scroggo@google.com | 1125d39 | 2013-05-03 20:43:37 +0000 | [diff] [blame] | 133 | SkImageDecoder::kDecodeBounds_Mode); |
| 134 | } |
| 135 | |
| 136 | //////////////////////////////////////////////////////////////////////////////////////////////////// |
| 137 | |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 138 | /** |
| 139 | * Called only by render_picture(). |
| 140 | */ |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 141 | static bool render_picture_internal(const SkString& inputPath, const SkString* writePath, |
| 142 | const SkString* mismatchPath, |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 143 | sk_tools::PictureRenderer& renderer, |
| 144 | SkBitmap** out) { |
tfarina | 1c99ea8 | 2014-06-11 08:58:50 -0700 | [diff] [blame] | 145 | SkString inputFilename = SkOSPath::SkBasename(inputPath.c_str()); |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 146 | SkString writePathString; |
| 147 | if (NULL != writePath && writePath->size() > 0 && !FLAGS_writeEncodedImages) { |
| 148 | writePathString.set(*writePath); |
| 149 | } |
| 150 | SkString mismatchPathString; |
| 151 | if (NULL != mismatchPath && mismatchPath->size() > 0) { |
| 152 | mismatchPathString.set(*mismatchPath); |
commit-bot@chromium.org | f5e315c | 2014-03-19 17:26:07 +0000 | [diff] [blame] | 153 | } |
twiz@google.com | a31b8bb | 2012-06-22 18:24:56 +0000 | [diff] [blame] | 154 | |
keyar@chromium.org | d1dc920 | 2012-07-09 18:32:08 +0000 | [diff] [blame] | 155 | SkFILEStream inputStream; |
twiz@google.com | a31b8bb | 2012-06-22 18:24:56 +0000 | [diff] [blame] | 156 | inputStream.setPath(inputPath.c_str()); |
| 157 | if (!inputStream.isValid()) { |
| 158 | SkDebugf("Could not open file %s\n", inputPath.c_str()); |
borenet@google.com | 66bcbd1 | 2012-09-17 18:26:06 +0000 | [diff] [blame] | 159 | return false; |
twiz@google.com | a31b8bb | 2012-06-22 18:24:56 +0000 | [diff] [blame] | 160 | } |
| 161 | |
scroggo@google.com | f1754ec | 2013-06-28 21:32:00 +0000 | [diff] [blame] | 162 | SkPicture::InstallPixelRefProc proc; |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 163 | if (FLAGS_deferImageDecoding) { |
commit-bot@chromium.org | 56799e2 | 2013-07-16 18:21:46 +0000 | [diff] [blame] | 164 | proc = &sk_tools::LazyDecodeBitmap; |
scroggo@google.com | 1125d39 | 2013-05-03 20:43:37 +0000 | [diff] [blame] | 165 | } else if (FLAGS_writeEncodedImages) { |
| 166 | SkASSERT(!FLAGS_writePath.isEmpty()); |
| 167 | reset_image_file_base_name(inputFilename); |
scroggo@google.com | f1754ec | 2013-06-28 21:32:00 +0000 | [diff] [blame] | 168 | proc = &write_image_to_file; |
scroggo@google.com | f8d7d27 | 2013-02-22 21:38:35 +0000 | [diff] [blame] | 169 | } else { |
scroggo@google.com | f1754ec | 2013-06-28 21:32:00 +0000 | [diff] [blame] | 170 | proc = &SkImageDecoder::DecodeMemory; |
scroggo@google.com | f8d7d27 | 2013-02-22 21:38:35 +0000 | [diff] [blame] | 171 | } |
scroggo@google.com | 1125d39 | 2013-05-03 20:43:37 +0000 | [diff] [blame] | 172 | |
scroggo@google.com | f1754ec | 2013-06-28 21:32:00 +0000 | [diff] [blame] | 173 | SkDebugf("deserializing... %s\n", inputPath.c_str()); |
| 174 | |
| 175 | SkPicture* picture = SkPicture::CreateFromStream(&inputStream, proc); |
| 176 | |
| 177 | if (NULL == picture) { |
borenet@google.com | 66bcbd1 | 2012-09-17 18:26:06 +0000 | [diff] [blame] | 178 | SkDebugf("Could not read an SkPicture from %s\n", inputPath.c_str()); |
| 179 | return false; |
| 180 | } |
keyar@chromium.org | 451bb9f | 2012-07-26 17:27:57 +0000 | [diff] [blame] | 181 | |
commit-bot@chromium.org | ff36a1d | 2013-07-24 20:37:30 +0000 | [diff] [blame] | 182 | while (FLAGS_bench_record) { |
robertphillips@google.com | 84b18c7 | 2014-04-13 19:09:42 +0000 | [diff] [blame] | 183 | SkPictureRecorder recorder; |
commit-bot@chromium.org | 5fb2ce3 | 2014-04-17 23:35:06 +0000 | [diff] [blame] | 184 | picture->draw(recorder.beginRecording(picture->width(), picture->height(), NULL, 0)); |
robertphillips@google.com | 84b18c7 | 2014-04-13 19:09:42 +0000 | [diff] [blame] | 185 | SkAutoTUnref<SkPicture> other(recorder.endRecording()); |
commit-bot@chromium.org | ff36a1d | 2013-07-24 20:37:30 +0000 | [diff] [blame] | 186 | } |
| 187 | |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 188 | SkDebugf("drawing... [%i %i] %s\n", picture->width(), picture->height(), |
borenet@google.com | 03fcee8 | 2012-09-10 18:18:38 +0000 | [diff] [blame] | 189 | inputPath.c_str()); |
skia.committer@gmail.com | 1d225f2 | 2012-09-14 02:01:10 +0000 | [diff] [blame] | 190 | |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 191 | renderer.init(picture, &writePathString, &mismatchPathString, &inputFilename, |
| 192 | FLAGS_writeChecksumBasedFilenames); |
skia.committer@gmail.com | eb849e5 | 2014-03-17 03:02:17 +0000 | [diff] [blame] | 193 | |
commit-bot@chromium.org | 145d1c0 | 2014-03-16 19:46:36 +0000 | [diff] [blame] | 194 | if (FLAGS_preprocess) { |
| 195 | if (NULL != renderer.getCanvas()) { |
commit-bot@chromium.org | 8ddc26b | 2014-03-31 17:55:12 +0000 | [diff] [blame] | 196 | renderer.getCanvas()->EXPERIMENTAL_optimize(renderer.getPicture()); |
commit-bot@chromium.org | 145d1c0 | 2014-03-16 19:46:36 +0000 | [diff] [blame] | 197 | } |
| 198 | } |
| 199 | |
scroggo@google.com | b4773b4 | 2012-10-01 20:06:09 +0000 | [diff] [blame] | 200 | renderer.setup(); |
commit-bot@chromium.org | 8991c67 | 2014-05-22 00:36:05 +0000 | [diff] [blame] | 201 | renderer.enableWrites(); |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 202 | |
commit-bot@chromium.org | f5e315c | 2014-03-19 17:26:07 +0000 | [diff] [blame] | 203 | bool success = renderer.render(out); |
| 204 | if (!success) { |
| 205 | SkDebugf("Failed to render %s\n", inputFilename.c_str()); |
scroggo@google.com | 81f9d2e | 2012-09-20 14:54:21 +0000 | [diff] [blame] | 206 | } |
scroggo@google.com | 9a41252 | 2012-09-07 15:21:18 +0000 | [diff] [blame] | 207 | |
keyar@chromium.org | 9d696c0 | 2012-08-07 17:11:33 +0000 | [diff] [blame] | 208 | renderer.end(); |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 209 | |
| 210 | SkDELETE(picture); |
borenet@google.com | 66bcbd1 | 2012-09-17 18:26:06 +0000 | [diff] [blame] | 211 | return success; |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 212 | } |
| 213 | |
edisonn@google.com | ca1b3ff | 2013-01-16 18:18:48 +0000 | [diff] [blame] | 214 | static inline int getByte(uint32_t value, int index) { |
| 215 | SkASSERT(0 <= index && index < 4); |
| 216 | return (value >> (index * 8)) & 0xFF; |
| 217 | } |
| 218 | |
| 219 | static int MaxByteDiff(uint32_t v1, uint32_t v2) { |
| 220 | return SkMax32(SkMax32(abs(getByte(v1, 0) - getByte(v2, 0)), abs(getByte(v1, 1) - getByte(v2, 1))), |
| 221 | SkMax32(abs(getByte(v1, 2) - getByte(v2, 2)), abs(getByte(v1, 3) - getByte(v2, 3)))); |
| 222 | } |
| 223 | |
junov@chromium.org | e286e84 | 2013-03-13 17:27:16 +0000 | [diff] [blame] | 224 | class AutoRestoreBbhType { |
| 225 | public: |
| 226 | AutoRestoreBbhType() { |
| 227 | fRenderer = NULL; |
junov@chromium.org | d34fda1 | 2013-03-13 19:03:26 +0000 | [diff] [blame] | 228 | fSavedBbhType = sk_tools::PictureRenderer::kNone_BBoxHierarchyType; |
junov@chromium.org | e286e84 | 2013-03-13 17:27:16 +0000 | [diff] [blame] | 229 | } |
| 230 | |
| 231 | void set(sk_tools::PictureRenderer* renderer, |
| 232 | sk_tools::PictureRenderer::BBoxHierarchyType bbhType) { |
| 233 | fRenderer = renderer; |
| 234 | fSavedBbhType = renderer->getBBoxHierarchyType(); |
| 235 | renderer->setBBoxHierarchyType(bbhType); |
| 236 | } |
| 237 | |
| 238 | ~AutoRestoreBbhType() { |
| 239 | if (NULL != fRenderer) { |
| 240 | fRenderer->setBBoxHierarchyType(fSavedBbhType); |
| 241 | } |
| 242 | } |
| 243 | |
| 244 | private: |
| 245 | sk_tools::PictureRenderer* fRenderer; |
| 246 | sk_tools::PictureRenderer::BBoxHierarchyType fSavedBbhType; |
| 247 | }; |
junov@chromium.org | e286e84 | 2013-03-13 17:27:16 +0000 | [diff] [blame] | 248 | |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 249 | /** |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 250 | * Render the SKP file(s) within inputPath. |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 251 | * |
| 252 | * @param inputPath path to an individual SKP file, or a directory of SKP files |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 253 | * @param writePath if not NULL, write all image(s) generated into this directory |
| 254 | * @param mismatchPath if not NULL, write any image(s) not matching expectations into this directory |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 255 | * @param renderer PictureRenderer to use to render the SKPs |
| 256 | * @param jsonSummaryPtr if not NULL, add the image(s) generated to this summary |
| 257 | */ |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 258 | static bool render_picture(const SkString& inputPath, const SkString* writePath, |
| 259 | const SkString* mismatchPath, sk_tools::PictureRenderer& renderer, |
commit-bot@chromium.org | 205ce48 | 2014-05-12 15:37:20 +0000 | [diff] [blame] | 260 | sk_tools::ImageResultsAndExpectations *jsonSummaryPtr) { |
edisonn@google.com | ca1b3ff | 2013-01-16 18:18:48 +0000 | [diff] [blame] | 261 | int diffs[256] = {0}; |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 262 | SkBitmap* bitmap = NULL; |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 263 | renderer.setJsonSummaryPtr(jsonSummaryPtr); |
| 264 | bool success = render_picture_internal(inputPath, |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 265 | FLAGS_writeWholeImage ? NULL : writePath, |
| 266 | FLAGS_writeWholeImage ? NULL : mismatchPath, |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 267 | renderer, |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 268 | FLAGS_validate || FLAGS_writeWholeImage ? &bitmap : NULL); |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 269 | |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 270 | if (!success || ((FLAGS_validate || FLAGS_writeWholeImage) && bitmap == NULL)) { |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 271 | SkDebugf("Failed to draw the picture.\n"); |
| 272 | SkDELETE(bitmap); |
| 273 | return false; |
| 274 | } |
| 275 | |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 276 | if (FLAGS_validate) { |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 277 | SkBitmap* referenceBitmap = NULL; |
junov@chromium.org | e286e84 | 2013-03-13 17:27:16 +0000 | [diff] [blame] | 278 | sk_tools::PictureRenderer* referenceRenderer; |
| 279 | // If the renderer uses a BBoxHierarchy, then the reference renderer |
skia.committer@gmail.com | 03682be | 2013-03-14 07:02:51 +0000 | [diff] [blame] | 280 | // will be the same renderer, without the bbh. |
junov@chromium.org | e286e84 | 2013-03-13 17:27:16 +0000 | [diff] [blame] | 281 | AutoRestoreBbhType arbbh; |
| 282 | if (sk_tools::PictureRenderer::kNone_BBoxHierarchyType != |
| 283 | renderer.getBBoxHierarchyType()) { |
| 284 | referenceRenderer = &renderer; |
| 285 | referenceRenderer->ref(); // to match auto unref below |
| 286 | arbbh.set(referenceRenderer, sk_tools::PictureRenderer::kNone_BBoxHierarchyType); |
| 287 | } else { |
| 288 | referenceRenderer = SkNEW(sk_tools::SimplePictureRenderer); |
| 289 | } |
| 290 | SkAutoTUnref<sk_tools::PictureRenderer> aurReferenceRenderer(referenceRenderer); |
| 291 | |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 292 | success = render_picture_internal(inputPath, NULL, NULL, *referenceRenderer, |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 293 | &referenceBitmap); |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 294 | |
junov@chromium.org | c19c191 | 2013-03-12 19:56:49 +0000 | [diff] [blame] | 295 | if (!success || NULL == referenceBitmap || NULL == referenceBitmap->getPixels()) { |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 296 | SkDebugf("Failed to draw the reference picture.\n"); |
| 297 | SkDELETE(bitmap); |
| 298 | SkDELETE(referenceBitmap); |
| 299 | return false; |
| 300 | } |
| 301 | |
| 302 | if (success && (bitmap->width() != referenceBitmap->width())) { |
| 303 | SkDebugf("Expected image width: %i, actual image width %i.\n", |
| 304 | referenceBitmap->width(), bitmap->width()); |
| 305 | SkDELETE(bitmap); |
| 306 | SkDELETE(referenceBitmap); |
| 307 | return false; |
| 308 | } |
| 309 | if (success && (bitmap->height() != referenceBitmap->height())) { |
| 310 | SkDebugf("Expected image height: %i, actual image height %i", |
| 311 | referenceBitmap->height(), bitmap->height()); |
| 312 | SkDELETE(bitmap); |
| 313 | SkDELETE(referenceBitmap); |
| 314 | return false; |
| 315 | } |
skia.committer@gmail.com | a7d8e3e | 2012-12-19 02:01:38 +0000 | [diff] [blame] | 316 | |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 317 | for (int y = 0; success && y < bitmap->height(); y++) { |
| 318 | for (int x = 0; success && x < bitmap->width(); x++) { |
edisonn@google.com | ca1b3ff | 2013-01-16 18:18:48 +0000 | [diff] [blame] | 319 | int diff = MaxByteDiff(*referenceBitmap->getAddr32(x, y), |
| 320 | *bitmap->getAddr32(x, y)); |
| 321 | SkASSERT(diff >= 0 && diff <= 255); |
| 322 | diffs[diff]++; |
skia.committer@gmail.com | 4d28d98 | 2013-01-17 07:06:06 +0000 | [diff] [blame] | 323 | |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 324 | if (diff > FLAGS_maxComponentDiff) { |
edisonn@google.com | ca1b3ff | 2013-01-16 18:18:48 +0000 | [diff] [blame] | 325 | SkDebugf("Expected pixel at (%i %i) exceedds maximum " |
| 326 | "component diff of %i: 0x%x, actual 0x%x\n", |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 327 | x, y, FLAGS_maxComponentDiff, |
edisonn@google.com | 01754bf | 2013-01-11 16:08:07 +0000 | [diff] [blame] | 328 | *referenceBitmap->getAddr32(x, y), |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 329 | *bitmap->getAddr32(x, y)); |
| 330 | SkDELETE(bitmap); |
| 331 | SkDELETE(referenceBitmap); |
| 332 | return false; |
| 333 | } |
| 334 | } |
| 335 | } |
| 336 | SkDELETE(referenceBitmap); |
edisonn@google.com | ca1b3ff | 2013-01-16 18:18:48 +0000 | [diff] [blame] | 337 | |
| 338 | for (int i = 1; i <= 255; ++i) { |
| 339 | if(diffs[i] > 0) { |
| 340 | SkDebugf("Number of pixels with max diff of %i is %i\n", i, diffs[i]); |
| 341 | } |
| 342 | } |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 343 | } |
| 344 | |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 345 | if (FLAGS_writeWholeImage) { |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 346 | sk_tools::force_all_opaque(*bitmap); |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 347 | |
tfarina | 1c99ea8 | 2014-06-11 08:58:50 -0700 | [diff] [blame] | 348 | SkString inputFilename = SkOSPath::SkBasename(inputPath.c_str()); |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 349 | SkString outputFilename(inputFilename); |
| 350 | sk_tools::replace_char(&outputFilename, '.', '_'); |
| 351 | outputFilename.append(".png"); |
commit-bot@chromium.org | 24c568c | 2014-04-10 15:39:02 +0000 | [diff] [blame] | 352 | |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 353 | if (NULL != jsonSummaryPtr) { |
commit-bot@chromium.org | 205ce48 | 2014-05-12 15:37:20 +0000 | [diff] [blame] | 354 | sk_tools::ImageDigest imageDigest(*bitmap); |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 355 | jsonSummaryPtr->add(inputFilename.c_str(), outputFilename.c_str(), imageDigest); |
| 356 | if ((NULL != mismatchPath) && !mismatchPath->isEmpty() && |
| 357 | !jsonSummaryPtr->matchesExpectation(inputFilename.c_str(), imageDigest)) { |
| 358 | success &= sk_tools::write_bitmap_to_disk(*bitmap, *mismatchPath, NULL, |
| 359 | outputFilename); |
| 360 | } |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 361 | } |
| 362 | |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 363 | if ((NULL != writePath) && !writePath->isEmpty()) { |
| 364 | success &= sk_tools::write_bitmap_to_disk(*bitmap, *writePath, NULL, outputFilename); |
edisonn@google.com | 84f548c | 2012-12-18 22:24:03 +0000 | [diff] [blame] | 365 | } |
| 366 | } |
| 367 | SkDELETE(bitmap); |
| 368 | |
| 369 | return success; |
| 370 | } |
| 371 | |
| 372 | |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 373 | static int process_input(const char* input, const SkString* writePath, |
| 374 | const SkString* mismatchPath, sk_tools::PictureRenderer& renderer, |
commit-bot@chromium.org | 205ce48 | 2014-05-12 15:37:20 +0000 | [diff] [blame] | 375 | sk_tools::ImageResultsAndExpectations *jsonSummaryPtr) { |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 376 | SkOSFile::Iter iter(input, "skp"); |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 377 | SkString inputFilename; |
borenet@google.com | 66bcbd1 | 2012-09-17 18:26:06 +0000 | [diff] [blame] | 378 | int failures = 0; |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 379 | SkDebugf("process_input, %s\n", input); |
keyar@chromium.org | d1dc920 | 2012-07-09 18:32:08 +0000 | [diff] [blame] | 380 | if (iter.next(&inputFilename)) { |
| 381 | do { |
tfarina | da4ed32 | 2014-06-12 08:50:56 -0700 | [diff] [blame] | 382 | SkString inputPath = SkOSPath::SkPathJoin(input, inputFilename.c_str()); |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 383 | if (!render_picture(inputPath, writePath, mismatchPath, renderer, jsonSummaryPtr)) { |
borenet@google.com | 57837bf | 2012-09-19 17:28:29 +0000 | [diff] [blame] | 384 | ++failures; |
| 385 | } |
keyar@chromium.org | d1dc920 | 2012-07-09 18:32:08 +0000 | [diff] [blame] | 386 | } while(iter.next(&inputFilename)); |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 387 | } else if (SkStrEndsWith(input, ".skp")) { |
keyar@chromium.org | d1dc920 | 2012-07-09 18:32:08 +0000 | [diff] [blame] | 388 | SkString inputPath(input); |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 389 | if (!render_picture(inputPath, writePath, mismatchPath, renderer, jsonSummaryPtr)) { |
borenet@google.com | 57837bf | 2012-09-19 17:28:29 +0000 | [diff] [blame] | 390 | ++failures; |
| 391 | } |
| 392 | } else { |
| 393 | SkString warning; |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 394 | warning.printf("Warning: skipping %s\n", input); |
borenet@google.com | 57837bf | 2012-09-19 17:28:29 +0000 | [diff] [blame] | 395 | SkDebugf(warning.c_str()); |
keyar@chromium.org | 1cbd47c | 2012-07-13 18:22:59 +0000 | [diff] [blame] | 396 | } |
borenet@google.com | 66bcbd1 | 2012-09-17 18:26:06 +0000 | [diff] [blame] | 397 | return failures; |
keyar@chromium.org | 1cbd47c | 2012-07-13 18:22:59 +0000 | [diff] [blame] | 398 | } |
| 399 | |
caryclark@google.com | 5987f58 | 2012-10-02 18:33:14 +0000 | [diff] [blame] | 400 | int tool_main(int argc, char** argv); |
| 401 | int tool_main(int argc, char** argv) { |
scroggo@google.com | d9ba9a0 | 2013-03-21 19:43:15 +0000 | [diff] [blame] | 402 | SkCommandLineFlags::SetUsage("Render .skp files."); |
| 403 | SkCommandLineFlags::Parse(argc, argv); |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 404 | |
scroggo@google.com | 604e0c2 | 2013-04-09 21:25:46 +0000 | [diff] [blame] | 405 | if (FLAGS_readPath.isEmpty()) { |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 406 | SkDebugf(".skp files or directories are required.\n"); |
| 407 | exit(-1); |
| 408 | } |
| 409 | |
| 410 | if (FLAGS_maxComponentDiff < 0 || FLAGS_maxComponentDiff > 256) { |
| 411 | SkDebugf("--maxComponentDiff must be between 0 and 256\n"); |
| 412 | exit(-1); |
| 413 | } |
| 414 | |
| 415 | if (FLAGS_maxComponentDiff != 256 && !FLAGS_validate) { |
| 416 | SkDebugf("--maxComponentDiff requires --validate\n"); |
| 417 | exit(-1); |
| 418 | } |
| 419 | |
scroggo@google.com | 1125d39 | 2013-05-03 20:43:37 +0000 | [diff] [blame] | 420 | if (FLAGS_writeEncodedImages) { |
| 421 | if (FLAGS_writePath.isEmpty()) { |
| 422 | SkDebugf("--writeEncodedImages requires --writePath\n"); |
| 423 | exit(-1); |
| 424 | } |
| 425 | if (FLAGS_deferImageDecoding) { |
| 426 | SkDebugf("--writeEncodedImages is not compatible with --deferImageDecoding\n"); |
| 427 | exit(-1); |
| 428 | } |
| 429 | } |
| 430 | |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 431 | SkString errorString; |
| 432 | SkAutoTUnref<sk_tools::PictureRenderer> renderer(parseRenderer(errorString, |
| 433 | kRender_PictureTool)); |
| 434 | if (errorString.size() > 0) { |
| 435 | SkDebugf("%s\n", errorString.c_str()); |
| 436 | } |
| 437 | |
| 438 | if (renderer.get() == NULL) { |
| 439 | exit(-1); |
| 440 | } |
| 441 | |
borenet@google.com | 66bcbd1 | 2012-09-17 18:26:06 +0000 | [diff] [blame] | 442 | SkAutoGraphics ag; |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 443 | |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 444 | SkString writePath; |
scroggo@google.com | 604e0c2 | 2013-04-09 21:25:46 +0000 | [diff] [blame] | 445 | if (FLAGS_writePath.count() == 1) { |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 446 | writePath.set(FLAGS_writePath[0]); |
| 447 | } |
| 448 | SkString mismatchPath; |
| 449 | if (FLAGS_mismatchPath.count() == 1) { |
| 450 | mismatchPath.set(FLAGS_mismatchPath[0]); |
scroggo@google.com | 161e1ba | 2013-03-04 16:41:06 +0000 | [diff] [blame] | 451 | } |
commit-bot@chromium.org | 205ce48 | 2014-05-12 15:37:20 +0000 | [diff] [blame] | 452 | sk_tools::ImageResultsAndExpectations jsonSummary; |
| 453 | sk_tools::ImageResultsAndExpectations* jsonSummaryPtr = NULL; |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 454 | if (FLAGS_writeJsonSummaryPath.count() == 1) { |
| 455 | jsonSummaryPtr = &jsonSummary; |
commit-bot@chromium.org | 205ce48 | 2014-05-12 15:37:20 +0000 | [diff] [blame] | 456 | if (FLAGS_readJsonSummaryPath.count() == 1) { |
| 457 | SkASSERT(jsonSummary.readExpectationsFile(FLAGS_readJsonSummaryPath[0])); |
| 458 | } |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 459 | } |
keyar@chromium.org | d1dc920 | 2012-07-09 18:32:08 +0000 | [diff] [blame] | 460 | |
borenet@google.com | 66bcbd1 | 2012-09-17 18:26:06 +0000 | [diff] [blame] | 461 | int failures = 0; |
scroggo@google.com | 604e0c2 | 2013-04-09 21:25:46 +0000 | [diff] [blame] | 462 | for (int i = 0; i < FLAGS_readPath.count(); i ++) { |
commit-bot@chromium.org | 3f04517 | 2014-05-15 15:10:48 +0000 | [diff] [blame] | 463 | failures += process_input(FLAGS_readPath[i], &writePath, &mismatchPath, *renderer.get(), |
| 464 | jsonSummaryPtr); |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 465 | } |
borenet@google.com | 66bcbd1 | 2012-09-17 18:26:06 +0000 | [diff] [blame] | 466 | if (failures != 0) { |
| 467 | SkDebugf("Failed to render %i pictures.\n", failures); |
| 468 | return 1; |
| 469 | } |
robertphillips@google.com | 163c84b | 2012-09-13 15:40:37 +0000 | [diff] [blame] | 470 | #if SK_SUPPORT_GPU |
| 471 | #if GR_CACHE_STATS |
| 472 | if (renderer->isUsingGpuDevice()) { |
| 473 | GrContext* ctx = renderer->getGrContext(); |
robertphillips@google.com | 163c84b | 2012-09-13 15:40:37 +0000 | [diff] [blame] | 474 | ctx->printCacheStats(); |
commit-bot@chromium.org | 03e3e89 | 2013-10-02 18:19:17 +0000 | [diff] [blame] | 475 | #ifdef SK_DEVELOPER |
| 476 | ctx->dumpFontCache(); |
| 477 | #endif |
robertphillips@google.com | 163c84b | 2012-09-13 15:40:37 +0000 | [diff] [blame] | 478 | } |
| 479 | #endif |
| 480 | #endif |
commit-bot@chromium.org | a3f882c | 2013-12-13 20:52:36 +0000 | [diff] [blame] | 481 | if (FLAGS_writeJsonSummaryPath.count() == 1) { |
| 482 | jsonSummary.writeToFile(FLAGS_writeJsonSummaryPath[0]); |
| 483 | } |
caryclark@google.com | 868e1f6 | 2012-10-02 20:00:03 +0000 | [diff] [blame] | 484 | return 0; |
junov@chromium.org | 777442d | 2012-06-12 14:56:36 +0000 | [diff] [blame] | 485 | } |
caryclark@google.com | 5987f58 | 2012-10-02 18:33:14 +0000 | [diff] [blame] | 486 | |
| 487 | #if !defined SK_BUILD_FOR_IOS |
| 488 | int main(int argc, char * const argv[]) { |
| 489 | return tool_main(argc, (char**) argv); |
| 490 | } |
| 491 | #endif |