epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 1 | |
| 2 | /* |
| 3 | * Copyright 2011 Google Inc. |
| 4 | * |
| 5 | * Use of this source code is governed by a BSD-style license that can be |
| 6 | * found in the LICENSE file. |
| 7 | */ |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 8 | #include "SkColorPriv.h" |
| 9 | #include "SkImageDecoder.h" |
| 10 | #include "SkImageEncoder.h" |
| 11 | #include "SkOSFile.h" |
| 12 | #include "SkStream.h" |
| 13 | #include "SkTDArray.h" |
| 14 | #include "SkTemplates.h" |
| 15 | #include "SkTime.h" |
| 16 | #include "SkTSearch.h" |
| 17 | #include "SkTypes.h" |
| 18 | |
| 19 | /** |
| 20 | * skdiff |
| 21 | * |
| 22 | * Given three directory names, expects to find identically-named files in |
| 23 | * each of the first two; the first are treated as a set of baseline, |
| 24 | * the second a set of variant images, and a diff image is written into the |
| 25 | * third directory for each pair. |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 26 | * Creates an index.html in the current third directory to compare each |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 27 | * pair that does not match exactly. |
| 28 | * Does *not* recursively descend directories. |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 29 | * |
| 30 | * With the --chromium flag, *does* recursively descend the first directory |
| 31 | * named, comparing *-expected.png with *-actual.png and writing diff |
| 32 | * images into the second directory, also writing index.html there. |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 33 | */ |
| 34 | |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 35 | #if SK_BUILD_FOR_WIN32 |
| 36 | #define PATH_DIV_STR "\\" |
| 37 | #define PATH_DIV_CHAR '\\' |
| 38 | #else |
| 39 | #define PATH_DIV_STR "/" |
| 40 | #define PATH_DIV_CHAR '/' |
| 41 | #endif |
| 42 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 43 | struct DiffRecord { |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 44 | DiffRecord (const SkString filename, |
| 45 | const SkString basePath, |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 46 | const SkString comparisonPath, |
| 47 | bool baseMissing = false, |
| 48 | bool comparisonMissing = false) |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 49 | : fFilename (filename) |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 50 | , fBasePath (basePath) |
| 51 | , fComparisonPath (comparisonPath) |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 52 | , fBaseBitmap (new SkBitmap ()) |
| 53 | , fComparisonBitmap (new SkBitmap ()) |
| 54 | , fDifferenceBitmap (new SkBitmap ()) |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 55 | , fWhiteBitmap (new SkBitmap ()) |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 56 | , fBaseHeight (0) |
| 57 | , fBaseWidth (0) |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 58 | , fFractionDifference (0) |
| 59 | , fWeightedFraction (0) |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 60 | , fAverageMismatchR (0) |
| 61 | , fAverageMismatchG (0) |
| 62 | , fAverageMismatchB (0) |
| 63 | , fMaxMismatchR (0) |
| 64 | , fMaxMismatchG (0) |
tomhudson@google.com | 8b08c3e | 2011-11-30 17:01:00 +0000 | [diff] [blame] | 65 | , fMaxMismatchB (0) |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 66 | , fDoImageSizesMismatch (false) |
| 67 | , fBaseMissing(baseMissing) |
| 68 | , fComparisonMissing(comparisonMissing) { |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 69 | // These asserts are valid for GM, but not for --chromium |
| 70 | //SkASSERT(basePath.endsWith(filename.c_str())); |
| 71 | //SkASSERT(comparisonPath.endsWith(filename.c_str())); |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 72 | }; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 73 | |
| 74 | SkString fFilename; |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 75 | SkString fBasePath; |
| 76 | SkString fComparisonPath; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 77 | |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 78 | SkBitmap* fBaseBitmap; |
| 79 | SkBitmap* fComparisonBitmap; |
| 80 | SkBitmap* fDifferenceBitmap; |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 81 | SkBitmap* fWhiteBitmap; |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 82 | |
| 83 | int fBaseHeight; |
| 84 | int fBaseWidth; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 85 | |
| 86 | /// Arbitrary floating-point metric to be used to sort images from most |
| 87 | /// to least different from baseline; values of 0 will be omitted from the |
| 88 | /// summary webpage. |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 89 | float fFractionDifference; |
| 90 | float fWeightedFraction; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 91 | |
| 92 | float fAverageMismatchR; |
| 93 | float fAverageMismatchG; |
| 94 | float fAverageMismatchB; |
| 95 | |
| 96 | uint32_t fMaxMismatchR; |
| 97 | uint32_t fMaxMismatchG; |
| 98 | uint32_t fMaxMismatchB; |
tomhudson@google.com | 8b08c3e | 2011-11-30 17:01:00 +0000 | [diff] [blame] | 99 | |
| 100 | /// By the time we need to report image size mismatch, we've already |
| 101 | /// released the bitmaps, so we need to remember it when we detect it. |
| 102 | bool fDoImageSizesMismatch; |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 103 | |
| 104 | bool fBaseMissing; |
| 105 | bool fComparisonMissing; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 106 | }; |
| 107 | |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 108 | #define MAX2(a,b) (((b) < (a)) ? (a) : (b)) |
| 109 | #define MAX3(a,b,c) (((b) < (a)) ? MAX2((a), (c)) : MAX2((b), (c))) |
| 110 | |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 111 | const SkPMColor PMCOLOR_WHITE = SkPreMultiplyColor(SK_ColorWHITE); |
| 112 | const SkPMColor PMCOLOR_BLACK = SkPreMultiplyColor(SK_ColorBLACK); |
| 113 | |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 114 | typedef SkTDArray<SkString*> StringArray; |
| 115 | typedef StringArray FileArray; |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 116 | |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 117 | struct DiffSummary { |
| 118 | DiffSummary () |
| 119 | : fNumMatches (0) |
| 120 | , fNumMismatches (0) |
| 121 | , fMaxMismatchV (0) |
| 122 | , fMaxMismatchPercent (0) { }; |
| 123 | |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 124 | ~DiffSummary() { |
| 125 | fBaseMissing.deleteAll(); |
| 126 | fComparisonMissing.deleteAll(); |
| 127 | } |
| 128 | |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 129 | uint32_t fNumMatches; |
| 130 | uint32_t fNumMismatches; |
| 131 | uint32_t fMaxMismatchV; |
| 132 | float fMaxMismatchPercent; |
| 133 | |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 134 | FileArray fBaseMissing; |
| 135 | FileArray fComparisonMissing; |
| 136 | |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 137 | void print () { |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 138 | int n = fBaseMissing.count(); |
| 139 | if (n > 0) { |
| 140 | printf("Missing in baseDir:\n"); |
| 141 | for (int i = 0; i < n; ++i) { |
| 142 | printf("\t%s\n", fBaseMissing[i]->c_str()); |
| 143 | } |
| 144 | } |
| 145 | n = fComparisonMissing.count(); |
| 146 | if (n > 0) { |
| 147 | printf("Missing in comparisonDir:\n"); |
| 148 | for (int i = 0; i < n; ++i) { |
| 149 | printf("\t%s\n", fComparisonMissing[i]->c_str()); |
| 150 | } |
| 151 | } |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 152 | printf("%d of %d images matched.\n", fNumMatches, |
| 153 | fNumMatches + fNumMismatches); |
| 154 | if (fNumMismatches > 0) { |
| 155 | printf("Maximum pixel intensity mismatch %d\n", fMaxMismatchV); |
| 156 | printf("Largest area mismatch was %.2f%% of pixels\n", |
| 157 | fMaxMismatchPercent); |
| 158 | } |
| 159 | |
| 160 | } |
| 161 | |
| 162 | void add (DiffRecord* drp) { |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 163 | if (drp->fBaseMissing) { |
| 164 | fBaseMissing.push(new SkString(drp->fFilename)); |
| 165 | fNumMismatches++; |
| 166 | } else if (drp->fComparisonMissing) { |
| 167 | fComparisonMissing.push(new SkString(drp->fFilename)); |
| 168 | fNumMismatches++; |
| 169 | } else if (0 == drp->fFractionDifference) { |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 170 | fNumMatches++; |
| 171 | } else { |
| 172 | fNumMismatches++; |
| 173 | if (drp->fFractionDifference * 100 > fMaxMismatchPercent) { |
| 174 | fMaxMismatchPercent = drp->fFractionDifference * 100; |
| 175 | } |
tomhudson@google.com | 88a0e05 | 2011-06-09 18:54:01 +0000 | [diff] [blame] | 176 | uint32_t value = MAX3(drp->fMaxMismatchR, drp->fMaxMismatchG, |
| 177 | drp->fMaxMismatchB); |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 178 | if (value > fMaxMismatchV) { |
| 179 | fMaxMismatchV = value; |
| 180 | } |
| 181 | } |
| 182 | } |
| 183 | }; |
| 184 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 185 | typedef SkTDArray<DiffRecord*> RecordArray; |
| 186 | |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 187 | /// Comparison routine for qsort; sorts by fFractionDifference |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 188 | /// from largest to smallest. |
| 189 | static int compare_diff_metrics (DiffRecord** lhs, DiffRecord** rhs) { |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 190 | if ((*lhs)->fFractionDifference < (*rhs)->fFractionDifference) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 191 | return 1; |
| 192 | } |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 193 | if ((*rhs)->fFractionDifference < (*lhs)->fFractionDifference) { |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 194 | return -1; |
| 195 | } |
| 196 | return 0; |
| 197 | } |
| 198 | |
| 199 | static int compare_diff_weighted (DiffRecord** lhs, DiffRecord** rhs) { |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 200 | if ((*lhs)->fWeightedFraction < (*rhs)->fWeightedFraction) { |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 201 | return 1; |
| 202 | } |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 203 | if ((*lhs)->fWeightedFraction > (*rhs)->fWeightedFraction) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 204 | return -1; |
| 205 | } |
| 206 | return 0; |
| 207 | } |
| 208 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 209 | /// Comparison routine for qsort; sorts by max(fAverageMismatch{RGB}) |
| 210 | /// from largest to smallest. |
| 211 | static int compare_diff_mean_mismatches (DiffRecord** lhs, DiffRecord** rhs) { |
| 212 | float leftValue = MAX3((*lhs)->fAverageMismatchR, |
| 213 | (*lhs)->fAverageMismatchG, |
| 214 | (*lhs)->fAverageMismatchB); |
| 215 | float rightValue = MAX3((*rhs)->fAverageMismatchR, |
| 216 | (*rhs)->fAverageMismatchG, |
| 217 | (*rhs)->fAverageMismatchB); |
| 218 | if (leftValue < rightValue) { |
| 219 | return 1; |
| 220 | } |
| 221 | if (rightValue < leftValue) { |
| 222 | return -1; |
| 223 | } |
| 224 | return 0; |
| 225 | } |
| 226 | |
| 227 | /// Comparison routine for qsort; sorts by max(fMaxMismatch{RGB}) |
| 228 | /// from largest to smallest. |
| 229 | static int compare_diff_max_mismatches (DiffRecord** lhs, DiffRecord** rhs) { |
bsalomon@google.com | 8e06dab | 2011-10-07 20:03:39 +0000 | [diff] [blame] | 230 | uint32_t leftValue = MAX3((*lhs)->fMaxMismatchR, |
| 231 | (*lhs)->fMaxMismatchG, |
| 232 | (*lhs)->fMaxMismatchB); |
| 233 | uint32_t rightValue = MAX3((*rhs)->fMaxMismatchR, |
| 234 | (*rhs)->fMaxMismatchG, |
| 235 | (*rhs)->fMaxMismatchB); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 236 | if (leftValue < rightValue) { |
| 237 | return 1; |
| 238 | } |
| 239 | if (rightValue < leftValue) { |
| 240 | return -1; |
| 241 | } |
| 242 | return compare_diff_mean_mismatches(lhs, rhs); |
| 243 | } |
| 244 | |
| 245 | |
| 246 | |
| 247 | /// Parameterized routine to compute the color of a pixel in a difference image. |
| 248 | typedef SkPMColor (*DiffMetricProc)(SkPMColor, SkPMColor); |
| 249 | |
tomhudson@google.com | 8b08c3e | 2011-11-30 17:01:00 +0000 | [diff] [blame] | 250 | static void expand_and_copy (int width, int height, SkBitmap** dest) { |
| 251 | SkBitmap* temp = new SkBitmap (); |
| 252 | temp->reset(); |
| 253 | temp->setConfig((*dest)->config(), width, height); |
| 254 | temp->allocPixels(); |
| 255 | (*dest)->copyPixelsTo(temp->getPixels(), temp->getSize(), |
| 256 | temp->rowBytes()); |
| 257 | *dest = temp; |
| 258 | } |
| 259 | |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 260 | static bool get_bitmaps (DiffRecord* diffRecord) { |
| 261 | SkFILEStream compareStream(diffRecord->fComparisonPath.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 262 | if (!compareStream.isValid()) { |
| 263 | SkDebugf("WARNING: couldn't open comparison file <%s>\n", |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 264 | diffRecord->fComparisonPath.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 265 | return false; |
| 266 | } |
| 267 | |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 268 | SkFILEStream baseStream(diffRecord->fBasePath.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 269 | if (!baseStream.isValid()) { |
| 270 | SkDebugf("ERROR: couldn't open base file <%s>\n", |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 271 | diffRecord->fBasePath.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 272 | return false; |
| 273 | } |
| 274 | |
| 275 | SkImageDecoder* codec = SkImageDecoder::Factory(&baseStream); |
| 276 | if (NULL == codec) { |
| 277 | SkDebugf("ERROR: no codec found for <%s>\n", |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 278 | diffRecord->fBasePath.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 279 | return false; |
| 280 | } |
| 281 | |
tomhudson@google.com | 8b08c3e | 2011-11-30 17:01:00 +0000 | [diff] [blame] | 282 | // In debug, the DLL will automatically be unloaded when this is deleted, |
| 283 | // but that shouldn't be a problem in release mode. |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 284 | SkAutoTDelete<SkImageDecoder> ad(codec); |
| 285 | |
| 286 | baseStream.rewind(); |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 287 | if (!codec->decode(&baseStream, diffRecord->fBaseBitmap, |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 288 | SkBitmap::kARGB_8888_Config, |
| 289 | SkImageDecoder::kDecodePixels_Mode)) { |
| 290 | SkDebugf("ERROR: codec failed for <%s>\n", |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 291 | diffRecord->fBasePath.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 292 | return false; |
| 293 | } |
| 294 | |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 295 | diffRecord->fBaseWidth = diffRecord->fBaseBitmap->width(); |
| 296 | diffRecord->fBaseHeight = diffRecord->fBaseBitmap->height(); |
| 297 | |
| 298 | if (!codec->decode(&compareStream, diffRecord->fComparisonBitmap, |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 299 | SkBitmap::kARGB_8888_Config, |
| 300 | SkImageDecoder::kDecodePixels_Mode)) { |
| 301 | SkDebugf("ERROR: codec failed for <%s>\n", |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 302 | diffRecord->fComparisonPath.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 303 | return false; |
| 304 | } |
| 305 | |
| 306 | return true; |
| 307 | } |
| 308 | |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 309 | static bool get_bitmap_height_width(const SkString& path, |
| 310 | int *height, int *width) { |
| 311 | SkFILEStream stream(path.c_str()); |
| 312 | if (!stream.isValid()) { |
| 313 | SkDebugf("ERROR: couldn't open file <%s>\n", |
| 314 | path.c_str()); |
| 315 | return false; |
| 316 | } |
| 317 | |
| 318 | SkImageDecoder* codec = SkImageDecoder::Factory(&stream); |
| 319 | if (NULL == codec) { |
| 320 | SkDebugf("ERROR: no codec found for <%s>\n", |
| 321 | path.c_str()); |
| 322 | return false; |
| 323 | } |
| 324 | |
| 325 | SkAutoTDelete<SkImageDecoder> ad(codec); |
| 326 | SkBitmap bm; |
| 327 | |
| 328 | stream.rewind(); |
| 329 | if (!codec->decode(&stream, &bm, |
| 330 | SkBitmap::kARGB_8888_Config, |
| 331 | SkImageDecoder::kDecodePixels_Mode)) { |
| 332 | SkDebugf("ERROR: codec failed for <%s>\n", |
| 333 | path.c_str()); |
| 334 | return false; |
| 335 | } |
| 336 | |
| 337 | *height = bm.height(); |
| 338 | *width = bm.width(); |
| 339 | |
| 340 | return true; |
| 341 | } |
| 342 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 343 | // from gm - thanks to PNG, we need to force all pixels 100% opaque |
| 344 | static void force_all_opaque(const SkBitmap& bitmap) { |
| 345 | SkAutoLockPixels lock(bitmap); |
| 346 | for (int y = 0; y < bitmap.height(); y++) { |
| 347 | for (int x = 0; x < bitmap.width(); x++) { |
| 348 | *bitmap.getAddr32(x, y) |= (SK_A32_MASK << SK_A32_SHIFT); |
| 349 | } |
| 350 | } |
| 351 | } |
| 352 | |
| 353 | // from gm |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 354 | static bool write_bitmap(const SkString& path, const SkBitmap* bitmap) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 355 | SkBitmap copy; |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 356 | bitmap->copyTo(©, SkBitmap::kARGB_8888_Config); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 357 | force_all_opaque(copy); |
| 358 | return SkImageEncoder::EncodeFile(path.c_str(), copy, |
| 359 | SkImageEncoder::kPNG_Type, 100); |
| 360 | } |
| 361 | |
| 362 | // from gm |
| 363 | static inline SkPMColor compute_diff_pmcolor(SkPMColor c0, SkPMColor c1) { |
| 364 | int dr = SkGetPackedR32(c0) - SkGetPackedR32(c1); |
| 365 | int dg = SkGetPackedG32(c0) - SkGetPackedG32(c1); |
| 366 | int db = SkGetPackedB32(c0) - SkGetPackedB32(c1); |
| 367 | |
| 368 | return SkPackARGB32(0xFF, SkAbs32(dr), SkAbs32(dg), SkAbs32(db)); |
| 369 | } |
| 370 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 371 | static inline bool colors_match_thresholded(SkPMColor c0, SkPMColor c1, |
| 372 | const int threshold) { |
| 373 | int da = SkGetPackedA32(c0) - SkGetPackedA32(c1); |
| 374 | int dr = SkGetPackedR32(c0) - SkGetPackedR32(c1); |
| 375 | int dg = SkGetPackedG32(c0) - SkGetPackedG32(c1); |
| 376 | int db = SkGetPackedB32(c0) - SkGetPackedB32(c1); |
| 377 | |
| 378 | return ((SkAbs32(da) <= threshold) && |
| 379 | (SkAbs32(dr) <= threshold) && |
| 380 | (SkAbs32(dg) <= threshold) && |
| 381 | (SkAbs32(db) <= threshold)); |
| 382 | } |
| 383 | |
| 384 | // based on gm |
| 385 | static void compute_diff(DiffRecord* dr, |
| 386 | DiffMetricProc diffFunction, |
| 387 | const int colorThreshold) { |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 388 | SkAutoLockPixels alpDiff(*dr->fDifferenceBitmap); |
| 389 | SkAutoLockPixels alpWhite(*dr->fWhiteBitmap); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 390 | |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 391 | const int w = dr->fComparisonBitmap->width(); |
| 392 | const int h = dr->fComparisonBitmap->height(); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 393 | int mismatchedPixels = 0; |
| 394 | int totalMismatchR = 0; |
| 395 | int totalMismatchG = 0; |
| 396 | int totalMismatchB = 0; |
tomhudson@google.com | 8b08c3e | 2011-11-30 17:01:00 +0000 | [diff] [blame] | 397 | |
| 398 | if (w != dr->fBaseWidth || h != dr->fBaseHeight) { |
| 399 | dr->fDoImageSizesMismatch = true; |
| 400 | dr->fFractionDifference = 1; |
| 401 | return; |
| 402 | } |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 403 | // Accumulate fractionally different pixels, then divide out |
| 404 | // # of pixels at the end. |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 405 | dr->fWeightedFraction = 0; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 406 | for (int y = 0; y < h; y++) { |
| 407 | for (int x = 0; x < w; x++) { |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 408 | SkPMColor c0 = *dr->fBaseBitmap->getAddr32(x, y); |
| 409 | SkPMColor c1 = *dr->fComparisonBitmap->getAddr32(x, y); |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 410 | SkPMColor trueDifference = compute_diff_pmcolor(c0, c1); |
| 411 | SkPMColor outputDifference = diffFunction(c0, c1); |
| 412 | uint32_t thisR = SkGetPackedR32(trueDifference); |
| 413 | uint32_t thisG = SkGetPackedG32(trueDifference); |
| 414 | uint32_t thisB = SkGetPackedB32(trueDifference); |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 415 | totalMismatchR += thisR; |
| 416 | totalMismatchG += thisG; |
| 417 | totalMismatchB += thisB; |
| 418 | // In HSV, value is defined as max RGB component. |
| 419 | int value = MAX3(thisR, thisG, thisB); |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 420 | dr->fWeightedFraction += ((float) value) / 255; |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 421 | if (thisR > dr->fMaxMismatchR) { |
| 422 | dr->fMaxMismatchR = thisR; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 423 | } |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 424 | if (thisG > dr->fMaxMismatchG) { |
| 425 | dr->fMaxMismatchG = thisG; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 426 | } |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 427 | if (thisB > dr->fMaxMismatchB) { |
| 428 | dr->fMaxMismatchB = thisB; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 429 | } |
| 430 | if (!colors_match_thresholded(c0, c1, colorThreshold)) { |
| 431 | mismatchedPixels++; |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 432 | *dr->fDifferenceBitmap->getAddr32(x, y) = outputDifference; |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 433 | *dr->fWhiteBitmap->getAddr32(x, y) = PMCOLOR_WHITE; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 434 | } else { |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 435 | *dr->fDifferenceBitmap->getAddr32(x, y) = 0; |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 436 | *dr->fWhiteBitmap->getAddr32(x, y) = PMCOLOR_BLACK; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 437 | } |
| 438 | } |
| 439 | } |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 440 | int pixelCount = w * h; |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 441 | dr->fFractionDifference = ((float) mismatchedPixels) / pixelCount; |
| 442 | dr->fWeightedFraction /= pixelCount; |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 443 | dr->fAverageMismatchR = ((float) totalMismatchR) / pixelCount; |
| 444 | dr->fAverageMismatchG = ((float) totalMismatchG) / pixelCount; |
| 445 | dr->fAverageMismatchB = ((float) totalMismatchB) / pixelCount; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 446 | } |
| 447 | |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 448 | static SkString filename_to_derived_filename (const SkString& filename, |
| 449 | const char *suffix) { |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 450 | SkString diffName (filename); |
| 451 | const char* cstring = diffName.c_str(); |
| 452 | int dotOffset = strrchr(cstring, '.') - cstring; |
| 453 | diffName.remove(dotOffset, diffName.size() - dotOffset); |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 454 | diffName.append(suffix); |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 455 | return diffName; |
| 456 | } |
| 457 | |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 458 | /// Given a image filename, returns the name of the file containing the |
| 459 | /// associated difference image. |
| 460 | static SkString filename_to_diff_filename (const SkString& filename) { |
| 461 | return filename_to_derived_filename(filename, "-diff.png"); |
| 462 | } |
| 463 | |
| 464 | /// Given a image filename, returns the name of the file containing the |
| 465 | /// "white" difference image. |
| 466 | static SkString filename_to_white_filename (const SkString& filename) { |
| 467 | return filename_to_derived_filename(filename, "-white.png"); |
| 468 | } |
| 469 | |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 470 | /// Convert a chromium/WebKit LayoutTest "foo-expected.png" to "foo-actual.png" |
| 471 | static SkString chrome_expected_path_to_actual (const SkString& expected) { |
| 472 | SkString actualPath (expected); |
| 473 | actualPath.remove(actualPath.size() - 13, 13); |
| 474 | actualPath.append("-actual.png"); |
| 475 | return actualPath; |
| 476 | } |
| 477 | |
| 478 | /// Convert a chromium/WebKit LayoutTest "foo-expected.png" to "foo.png" |
| 479 | static SkString chrome_expected_name_to_short (const SkString& expected) { |
| 480 | SkString shortName (expected); |
| 481 | shortName.remove(shortName.size() - 13, 13); |
| 482 | shortName.append(".png"); |
| 483 | return shortName; |
| 484 | } |
| 485 | |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 486 | static void release_bitmaps(DiffRecord* drp) { |
| 487 | delete drp->fBaseBitmap; |
| 488 | drp->fBaseBitmap = NULL; |
| 489 | delete drp->fComparisonBitmap; |
| 490 | drp->fComparisonBitmap = NULL; |
| 491 | delete drp->fDifferenceBitmap; |
| 492 | drp->fDifferenceBitmap = NULL; |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 493 | delete drp->fWhiteBitmap; |
| 494 | drp->fWhiteBitmap = NULL; |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 495 | } |
| 496 | |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 497 | |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 498 | /// If outputDir.isEmpty(), don't write out diff files. |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 499 | static void create_and_write_diff_image(DiffRecord* drp, |
| 500 | DiffMetricProc dmp, |
| 501 | const int colorThreshold, |
| 502 | const SkString& outputDir, |
| 503 | const SkString& filename) { |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 504 | const int w = drp->fBaseWidth; |
| 505 | const int h = drp->fBaseHeight; |
| 506 | drp->fDifferenceBitmap->setConfig(SkBitmap::kARGB_8888_Config, w, h); |
| 507 | drp->fDifferenceBitmap->allocPixels(); |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 508 | drp->fWhiteBitmap->setConfig(SkBitmap::kARGB_8888_Config, w, h); |
| 509 | drp->fWhiteBitmap->allocPixels(); |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 510 | compute_diff(drp, dmp, colorThreshold); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 511 | |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 512 | if (!outputDir.isEmpty()) { |
| 513 | SkString differencePath (outputDir); |
| 514 | differencePath.append(filename_to_diff_filename(filename)); |
| 515 | write_bitmap(differencePath, drp->fDifferenceBitmap); |
| 516 | SkString whitePath (outputDir); |
| 517 | whitePath.append(filename_to_white_filename(filename)); |
| 518 | write_bitmap(whitePath, drp->fWhiteBitmap); |
| 519 | } |
| 520 | |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 521 | release_bitmaps(drp); |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 522 | } |
| 523 | |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 524 | /// Returns true if string contains any of these substrings. |
| 525 | static bool string_contains_any_of(const SkString& string, |
| 526 | const StringArray& substrings) { |
| 527 | for (int i = 0; i < substrings.count(); i++) { |
| 528 | if (string.contains(substrings[i]->c_str())) { |
| 529 | return true; |
| 530 | } |
| 531 | } |
| 532 | return false; |
| 533 | } |
| 534 | |
| 535 | /// Iterate over dir and get all files that: |
| 536 | /// - match any of the substrings in matchSubstrings, but... |
| 537 | /// - DO NOT match any of the substrings in nomatchSubstrings |
| 538 | /// Returns the list of files in *files. |
| 539 | static void get_file_list(const SkString& dir, |
| 540 | const StringArray& matchSubstrings, |
| 541 | const StringArray& nomatchSubstrings, |
| 542 | FileArray *files) { |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 543 | SkOSFile::Iter it(dir.c_str()); |
| 544 | SkString filename; |
| 545 | while (it.next(&filename)) { |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 546 | if (string_contains_any_of(filename, matchSubstrings) && |
| 547 | !string_contains_any_of(filename, nomatchSubstrings)) { |
| 548 | files->push(new SkString(filename)); |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 549 | } |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 550 | } |
| 551 | } |
| 552 | |
| 553 | static void release_file_list(FileArray *files) { |
| 554 | files->deleteAll(); |
| 555 | } |
| 556 | |
| 557 | /// Comparison routines for qsort, sort by file names. |
| 558 | static int compare_file_name_metrics(SkString **lhs, SkString **rhs) { |
| 559 | return strcmp((*lhs)->c_str(), (*rhs)->c_str()); |
| 560 | } |
| 561 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 562 | /// Creates difference images, returns the number that have a 0 metric. |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 563 | /// If outputDir.isEmpty(), don't write out diff files. |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 564 | static void create_diff_images (DiffMetricProc dmp, |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 565 | const int colorThreshold, |
| 566 | RecordArray* differences, |
| 567 | const SkString& baseDir, |
| 568 | const SkString& comparisonDir, |
| 569 | const SkString& outputDir, |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 570 | const StringArray& matchSubstrings, |
| 571 | const StringArray& nomatchSubstrings, |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 572 | DiffSummary* summary) { |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 573 | SkASSERT(!baseDir.isEmpty()); |
| 574 | SkASSERT(!comparisonDir.isEmpty()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 575 | |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 576 | FileArray baseFiles; |
| 577 | FileArray comparisonFiles; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 578 | |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 579 | get_file_list(baseDir, matchSubstrings, nomatchSubstrings, &baseFiles); |
| 580 | get_file_list(comparisonDir, matchSubstrings, nomatchSubstrings, |
| 581 | &comparisonFiles); |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 582 | |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 583 | if (!baseFiles.isEmpty()) { |
reed@google.com | c7a67cb | 2012-05-07 14:52:12 +0000 | [diff] [blame] | 584 | qsort(baseFiles.begin(), baseFiles.count(), sizeof(SkString*), |
| 585 | SkCastForQSort(compare_file_name_metrics)); |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 586 | } |
| 587 | if (!comparisonFiles.isEmpty()) { |
reed@google.com | c7a67cb | 2012-05-07 14:52:12 +0000 | [diff] [blame] | 588 | qsort(comparisonFiles.begin(), comparisonFiles.count(), |
| 589 | sizeof(SkString*), SkCastForQSort(compare_file_name_metrics)); |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 590 | } |
| 591 | |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 592 | int i = 0; |
| 593 | int j = 0; |
| 594 | |
| 595 | while (i < baseFiles.count() && |
| 596 | j < comparisonFiles.count()) { |
| 597 | |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 598 | SkString basePath (baseDir); |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 599 | basePath.append(*baseFiles[i]); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 600 | SkString comparisonPath (comparisonDir); |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 601 | comparisonPath.append(*comparisonFiles[j]); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 602 | |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 603 | DiffRecord *drp = NULL; |
| 604 | int v = strcmp(baseFiles[i]->c_str(), |
| 605 | comparisonFiles[j]->c_str()); |
| 606 | |
| 607 | if (v < 0) { |
| 608 | // in baseDir, but not in comparisonDir |
| 609 | drp = new DiffRecord(*baseFiles[i], |
| 610 | basePath, comparisonPath, false, true); |
| 611 | ++i; |
| 612 | } else if (v > 0) { |
| 613 | // in comparisonDir, but not in baseDir |
| 614 | drp = new DiffRecord(*comparisonFiles[j], |
| 615 | basePath, comparisonPath, true, false); |
| 616 | ++j; |
| 617 | } else { |
| 618 | // let's diff! |
| 619 | drp = new DiffRecord(*baseFiles[i], basePath, comparisonPath); |
| 620 | |
| 621 | if (get_bitmaps(drp)) { |
| 622 | create_and_write_diff_image(drp, dmp, colorThreshold, |
| 623 | outputDir, *baseFiles[i]); |
| 624 | } |
| 625 | |
| 626 | ++i; |
| 627 | ++j; |
| 628 | } |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 629 | |
| 630 | differences->push(drp); |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 631 | summary->add(drp); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 632 | } |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 633 | |
| 634 | for (; i < baseFiles.count(); ++i) { |
| 635 | // files only in baseDir |
| 636 | SkString basePath (baseDir); |
| 637 | basePath.append(*baseFiles[i]); |
| 638 | SkString comparisonPath; |
| 639 | DiffRecord *drp = new DiffRecord(*baseFiles[i], basePath, |
| 640 | comparisonPath, false, true); |
| 641 | differences->push(drp); |
| 642 | summary->add(drp); |
| 643 | } |
| 644 | |
| 645 | for (; j < comparisonFiles.count(); ++j) { |
| 646 | // files only in comparisonDir |
| 647 | SkString basePath; |
| 648 | SkString comparisonPath(comparisonDir); |
| 649 | comparisonPath.append(*comparisonFiles[j]); |
| 650 | DiffRecord *drp = new DiffRecord(*comparisonFiles[j], basePath, |
| 651 | comparisonPath, true, false); |
| 652 | differences->push(drp); |
| 653 | summary->add(drp); |
| 654 | } |
| 655 | |
| 656 | release_file_list(&baseFiles); |
| 657 | release_file_list(&comparisonFiles); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 658 | } |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 659 | |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 660 | static void create_diff_images_chromium (DiffMetricProc dmp, |
| 661 | const int colorThreshold, |
| 662 | RecordArray* differences, |
| 663 | const SkString& dirname, |
| 664 | const SkString& outputDir, |
| 665 | DiffSummary* summary) { |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 666 | SkASSERT(!outputDir.isEmpty()); |
| 667 | |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 668 | SkOSFile::Iter baseIterator (dirname.c_str()); |
| 669 | SkString filename; |
| 670 | while (baseIterator.next(&filename)) { |
| 671 | if (filename.endsWith(".pdf")) { |
| 672 | continue; |
| 673 | } |
| 674 | if (filename.endsWith("-expected.png")) { |
| 675 | SkString expectedPath (dirname); |
| 676 | expectedPath.append(filename); |
| 677 | SkString shortName (chrome_expected_name_to_short(filename)); |
| 678 | SkString actualPath (chrome_expected_path_to_actual(expectedPath)); |
| 679 | DiffRecord * drp = |
| 680 | new DiffRecord (shortName, expectedPath, actualPath); |
| 681 | if (!get_bitmaps(drp)) { |
| 682 | continue; |
| 683 | } |
| 684 | create_and_write_diff_image(drp, dmp, colorThreshold, |
| 685 | outputDir, shortName); |
| 686 | |
| 687 | differences->push(drp); |
| 688 | summary->add(drp); |
| 689 | } |
| 690 | } |
| 691 | } |
| 692 | |
| 693 | static void analyze_chromium(DiffMetricProc dmp, |
| 694 | const int colorThreshold, |
| 695 | RecordArray* differences, |
| 696 | const SkString& dirname, |
| 697 | const SkString& outputDir, |
| 698 | DiffSummary* summary) { |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 699 | SkASSERT(!outputDir.isEmpty()); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 700 | create_diff_images_chromium(dmp, colorThreshold, differences, |
| 701 | dirname, outputDir, summary); |
| 702 | SkOSFile::Iter dirIterator(dirname.c_str()); |
| 703 | SkString newdirname; |
| 704 | while (dirIterator.next(&newdirname, true)) { |
| 705 | if (newdirname.startsWith(".")) { |
| 706 | continue; |
| 707 | } |
| 708 | SkString fullname (dirname); |
| 709 | fullname.append(newdirname); |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 710 | if (!fullname.endsWith(PATH_DIV_STR)) { |
| 711 | fullname.append(PATH_DIV_STR); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 712 | } |
| 713 | analyze_chromium(dmp, colorThreshold, differences, |
| 714 | fullname, outputDir, summary); |
| 715 | } |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 716 | } |
| 717 | |
| 718 | /// Make layout more consistent by scaling image to 240 height, 360 width, |
| 719 | /// or natural size, whichever is smallest. |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 720 | static int compute_image_height (int height, int width) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 721 | int retval = 240; |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 722 | if (height < retval) { |
| 723 | retval = height; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 724 | } |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 725 | float scale = (float) retval / height; |
| 726 | if (width * scale > 360) { |
| 727 | scale = (float) 360 / width; |
bsalomon@google.com | 8e06dab | 2011-10-07 20:03:39 +0000 | [diff] [blame] | 728 | retval = static_cast<int>(height * scale); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 729 | } |
| 730 | return retval; |
| 731 | } |
| 732 | |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 733 | static void print_table_header (SkFILEWStream* stream, |
| 734 | const int matchCount, |
| 735 | const int colorThreshold, |
| 736 | const RecordArray& differences, |
| 737 | const SkString &baseDir, |
| 738 | const SkString &comparisonDir) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 739 | SkTime::DateTime dt; |
| 740 | SkTime::GetDateTime(&dt); |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 741 | stream->writeText("<table>\n"); |
| 742 | stream->writeText("<tr><th>"); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 743 | stream->writeText("SkDiff run at "); |
| 744 | stream->writeDecAsText(dt.fHour); |
| 745 | stream->writeText(":"); |
| 746 | if (dt.fMinute < 10) { |
| 747 | stream->writeText("0"); |
| 748 | } |
| 749 | stream->writeDecAsText(dt.fMinute); |
| 750 | stream->writeText(":"); |
| 751 | if (dt.fSecond < 10) { |
| 752 | stream->writeText("0"); |
| 753 | } |
| 754 | stream->writeDecAsText(dt.fSecond); |
| 755 | stream->writeText("<br>"); |
| 756 | stream->writeDecAsText(matchCount); |
| 757 | stream->writeText(" of "); |
| 758 | stream->writeDecAsText(differences.count()); |
| 759 | stream->writeText(" images matched "); |
| 760 | if (colorThreshold == 0) { |
| 761 | stream->writeText("exactly"); |
| 762 | } else { |
| 763 | stream->writeText("within "); |
| 764 | stream->writeDecAsText(colorThreshold); |
| 765 | stream->writeText(" color units per component"); |
| 766 | } |
| 767 | stream->writeText(".<br>"); |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 768 | stream->writeText("</th>\n<th>"); |
| 769 | stream->writeText("every different pixel shown in white"); |
| 770 | stream->writeText("</th>\n<th>"); |
| 771 | stream->writeText("color difference at each pixel"); |
| 772 | stream->writeText("</th>\n<th>"); |
| 773 | stream->writeText(baseDir.c_str()); |
| 774 | stream->writeText("</th>\n<th>"); |
| 775 | stream->writeText(comparisonDir.c_str()); |
| 776 | stream->writeText("</th>\n"); |
| 777 | stream->writeText("</tr>\n"); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 778 | } |
| 779 | |
| 780 | static void print_pixel_count (SkFILEWStream* stream, |
| 781 | const DiffRecord& diff) { |
| 782 | stream->writeText("<br>("); |
bsalomon@google.com | 8e06dab | 2011-10-07 20:03:39 +0000 | [diff] [blame] | 783 | stream->writeDecAsText(static_cast<int>(diff.fFractionDifference * |
| 784 | diff.fBaseWidth * |
| 785 | diff.fBaseHeight)); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 786 | stream->writeText(" pixels)"); |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 787 | /* |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 788 | stream->writeDecAsText(diff.fWeightedFraction * |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 789 | diff.fBaseWidth * |
| 790 | diff.fBaseHeight); |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 791 | stream->writeText(" weighted pixels)"); |
| 792 | */ |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 793 | } |
| 794 | |
| 795 | static void print_label_cell (SkFILEWStream* stream, |
| 796 | const DiffRecord& diff) { |
| 797 | stream->writeText("<td>"); |
| 798 | stream->writeText(diff.fFilename.c_str()); |
| 799 | stream->writeText("<br>"); |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 800 | if (diff.fBaseMissing || diff.fComparisonMissing) { |
| 801 | stream->writeText("</td>"); |
| 802 | return; |
| 803 | } |
tomhudson@google.com | 8b08c3e | 2011-11-30 17:01:00 +0000 | [diff] [blame] | 804 | if (diff.fDoImageSizesMismatch) { |
| 805 | stream->writeText("Image sizes differ"); |
| 806 | stream->writeText("</td>"); |
| 807 | return; |
| 808 | } |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 809 | char metricBuf [20]; |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 810 | sprintf(metricBuf, "%12.4f%%", 100 * diff.fFractionDifference); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 811 | stream->writeText(metricBuf); |
| 812 | stream->writeText(" of pixels differ"); |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 813 | stream->writeText("\n ("); |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 814 | sprintf(metricBuf, "%12.4f%%", 100 * diff.fWeightedFraction); |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 815 | stream->writeText(metricBuf); |
| 816 | stream->writeText(" weighted)"); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 817 | // Write the actual number of pixels that differ if it's < 1% |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 818 | if (diff.fFractionDifference < 0.01) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 819 | print_pixel_count(stream, diff); |
| 820 | } |
| 821 | stream->writeText("<br>Average color mismatch "); |
bsalomon@google.com | 8e06dab | 2011-10-07 20:03:39 +0000 | [diff] [blame] | 822 | stream->writeDecAsText(static_cast<int>(MAX3(diff.fAverageMismatchR, |
| 823 | diff.fAverageMismatchG, |
| 824 | diff.fAverageMismatchB))); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 825 | stream->writeText("<br>Max color mismatch "); |
| 826 | stream->writeDecAsText(MAX3(diff.fMaxMismatchR, |
| 827 | diff.fMaxMismatchG, |
| 828 | diff.fMaxMismatchB)); |
| 829 | stream->writeText("</td>"); |
| 830 | } |
| 831 | |
| 832 | static void print_image_cell (SkFILEWStream* stream, |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 833 | const SkString& path, |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 834 | int height) { |
| 835 | stream->writeText("<td><a href=\""); |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 836 | stream->writeText(path.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 837 | stream->writeText("\"><img src=\""); |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 838 | stream->writeText(path.c_str()); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 839 | stream->writeText("\" height=\""); |
| 840 | stream->writeDecAsText(height); |
| 841 | stream->writeText("px\"></a></td>"); |
| 842 | } |
| 843 | |
epoger@google.com | 01f7870 | 2012-04-12 16:32:04 +0000 | [diff] [blame] | 844 | static void print_text_cell (SkFILEWStream* stream, const char* text) { |
| 845 | stream->writeText("<td align=center>"); |
| 846 | if (NULL != text) { |
| 847 | stream->writeText(text); |
| 848 | } |
| 849 | stream->writeText("</td>"); |
| 850 | } |
| 851 | |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 852 | static void print_diff_with_missing_file(SkFILEWStream* stream, |
| 853 | DiffRecord& diff, |
| 854 | const SkString& relativePath) { |
| 855 | stream->writeText("<tr>\n"); |
| 856 | print_label_cell(stream, diff); |
| 857 | stream->writeText("<td>N/A</td>"); |
| 858 | stream->writeText("<td>N/A</td>"); |
| 859 | if (!diff.fBaseMissing) { |
| 860 | int h, w; |
| 861 | if (!get_bitmap_height_width(diff.fBasePath, &h, &w)) { |
| 862 | stream->writeText("<td>N/A</td>"); |
| 863 | } else { |
| 864 | int height = compute_image_height(h, w); |
| 865 | if (!diff.fBasePath.startsWith(PATH_DIV_STR)) { |
| 866 | diff.fBasePath.prepend(relativePath); |
| 867 | } |
| 868 | print_image_cell(stream, diff.fBasePath, height); |
| 869 | } |
| 870 | } else { |
| 871 | stream->writeText("<td>N/A</td>"); |
| 872 | } |
| 873 | if (!diff.fComparisonMissing) { |
| 874 | int h, w; |
| 875 | if (!get_bitmap_height_width(diff.fComparisonPath, &h, &w)) { |
| 876 | stream->writeText("<td>N/A</td>"); |
| 877 | } else { |
| 878 | int height = compute_image_height(h, w); |
| 879 | if (!diff.fComparisonPath.startsWith(PATH_DIV_STR)) { |
| 880 | diff.fComparisonPath.prepend(relativePath); |
| 881 | } |
| 882 | print_image_cell(stream, diff.fComparisonPath, height); |
| 883 | } |
| 884 | } else { |
| 885 | stream->writeText("<td>N/A</td>"); |
| 886 | } |
| 887 | stream->writeText("</tr>\n"); |
| 888 | stream->flush(); |
| 889 | } |
| 890 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 891 | static void print_diff_page (const int matchCount, |
| 892 | const int colorThreshold, |
| 893 | const RecordArray& differences, |
| 894 | const SkString& baseDir, |
| 895 | const SkString& comparisonDir, |
| 896 | const SkString& outputDir) { |
| 897 | |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 898 | SkASSERT(!baseDir.isEmpty()); |
| 899 | SkASSERT(!comparisonDir.isEmpty()); |
| 900 | SkASSERT(!outputDir.isEmpty()); |
| 901 | |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 902 | SkString outputPath (outputDir); |
| 903 | outputPath.append("index.html"); |
| 904 | //SkFILEWStream outputStream ("index.html"); |
| 905 | SkFILEWStream outputStream (outputPath.c_str()); |
| 906 | |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 907 | // Need to convert paths from relative-to-cwd to relative-to-outputDir |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 908 | // FIXME this doesn't work if there are '..' inside the outputDir |
| 909 | unsigned int ui; |
| 910 | SkString relativePath; |
| 911 | for (ui = 0; ui < outputDir.size(); ui++) { |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 912 | if (outputDir[ui] == PATH_DIV_CHAR) { |
| 913 | relativePath.append(".." PATH_DIV_STR); |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 914 | } |
| 915 | } |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 916 | |
| 917 | outputStream.writeText("<html>\n<body>\n"); |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 918 | print_table_header(&outputStream, matchCount, colorThreshold, differences, |
| 919 | baseDir, comparisonDir); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 920 | int i; |
| 921 | for (i = 0; i < differences.count(); i++) { |
| 922 | DiffRecord* diff = differences[i]; |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 923 | |
| 924 | if (diff->fBaseMissing || diff->fComparisonMissing) { |
| 925 | print_diff_with_missing_file(&outputStream, *diff, relativePath); |
| 926 | continue; |
| 927 | } else if (0 == diff->fFractionDifference) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 928 | continue; |
| 929 | } |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 930 | |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 931 | if (!diff->fBasePath.startsWith(PATH_DIV_STR)) { |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 932 | diff->fBasePath.prepend(relativePath); |
| 933 | } |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 934 | if (!diff->fComparisonPath.startsWith(PATH_DIV_STR)) { |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 935 | diff->fComparisonPath.prepend(relativePath); |
| 936 | } |
epoger@google.com | 5fd5385 | 2012-03-22 18:20:06 +0000 | [diff] [blame] | 937 | |
tomhudson@google.com | 9b540ce | 2011-08-02 14:10:04 +0000 | [diff] [blame] | 938 | int height = compute_image_height(diff->fBaseHeight, diff->fBaseWidth); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 939 | outputStream.writeText("<tr>\n"); |
| 940 | print_label_cell(&outputStream, *diff); |
epoger@google.com | 01f7870 | 2012-04-12 16:32:04 +0000 | [diff] [blame] | 941 | if (diff->fDoImageSizesMismatch) { |
| 942 | print_text_cell(&outputStream, |
| 943 | "[image size mismatch, so no diff to display]"); |
| 944 | print_text_cell(&outputStream, |
| 945 | "[image size mismatch, so no diff to display]"); |
| 946 | } else { |
| 947 | print_image_cell(&outputStream, |
| 948 | filename_to_white_filename(diff->fFilename), height); |
| 949 | print_image_cell(&outputStream, |
| 950 | filename_to_diff_filename(diff->fFilename), height); |
| 951 | } |
epoger@google.com | 25d961c | 2012-02-02 20:50:36 +0000 | [diff] [blame] | 952 | print_image_cell(&outputStream, diff->fBasePath, height); |
tomhudson@google.com | 4e30598 | 2011-07-13 17:42:46 +0000 | [diff] [blame] | 953 | print_image_cell(&outputStream, diff->fComparisonPath, height); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 954 | outputStream.writeText("</tr>\n"); |
| 955 | outputStream.flush(); |
| 956 | } |
| 957 | outputStream.writeText("</table>\n"); |
| 958 | outputStream.writeText("</body>\n</html>\n"); |
| 959 | outputStream.flush(); |
| 960 | } |
| 961 | |
| 962 | static void usage (char * argv0) { |
| 963 | SkDebugf("Skia baseline image diff tool\n"); |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 964 | SkDebugf("\n" |
| 965 | "Usage: \n" |
| 966 | " %s <baseDir> <comparisonDir> [outputDir] \n" |
| 967 | "or \n" |
| 968 | " %s --chromium-release|--chromium-debug <baseDir> <outputDir> \n", |
| 969 | argv0, argv0); |
| 970 | SkDebugf("\n" |
| 971 | "Arguments: \n" |
| 972 | " --nodiffs: don't write out image diffs or index.html, just generate \n" |
| 973 | " report on stdout \n" |
| 974 | " --threshold <n>: only report differences > n (per color channel) [default 0]\n" |
| 975 | " --match: compare files whose filenames contain this substring; if \n" |
| 976 | " unspecified, compare ALL files. \n" |
| 977 | " this flag may be repeated to add more matching substrings. \n" |
| 978 | " --nomatch: regardless of --match, DO NOT compare files whose filenames \n" |
| 979 | " contain this substring. \n" |
| 980 | " this flag may be repeated to add more forbidden substrings. \n" |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 981 | " --sortbymismatch: sort by average color channel mismatch\n"); |
| 982 | SkDebugf( |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 983 | " --sortbymaxmismatch: sort by worst color channel mismatch;\n" |
| 984 | " break ties with -sortbymismatch\n" |
| 985 | " [default sort is by fraction of pixels mismatching]\n"); |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 986 | SkDebugf( |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 987 | " --weighted: sort by # pixels different weighted by color difference\n"); |
| 988 | SkDebugf( |
| 989 | " --chromium-release: process Webkit LayoutTests results instead of gm\n" |
| 990 | " --chromium-debug: process Webkit LayoutTests results instead of gm\n"); |
| 991 | SkDebugf( |
| 992 | " baseDir: directory to read baseline images from,\n" |
| 993 | " or chromium/src directory for --chromium.\n"); |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 994 | SkDebugf( |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 995 | " comparisonDir: directory to read comparison images from\n"); |
| 996 | SkDebugf( |
| 997 | " outputDir: directory to write difference images and index.html to; \n" |
| 998 | " defaults to comparisonDir when not running --chromium \n"); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 999 | } |
| 1000 | |
| 1001 | int main (int argc, char ** argv) { |
| 1002 | DiffMetricProc diffProc = compute_diff_pmcolor; |
reed@google.com | c7a67cb | 2012-05-07 14:52:12 +0000 | [diff] [blame] | 1003 | int (*sortProc)(const void*, const void*) = SkCastForQSort(compare_diff_metrics); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1004 | |
| 1005 | // Maximum error tolerated in any one color channel in any one pixel before |
| 1006 | // a difference is reported. |
| 1007 | int colorThreshold = 0; |
| 1008 | SkString baseDir; |
| 1009 | SkString comparisonDir; |
| 1010 | SkString outputDir; |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1011 | StringArray matchSubstrings; |
| 1012 | StringArray nomatchSubstrings; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1013 | |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1014 | bool analyzeChromium = false; |
| 1015 | bool chromiumDebug = false; |
| 1016 | bool chromiumRelease = false; |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1017 | bool generateDiffs = true; |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1018 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1019 | RecordArray differences; |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 1020 | DiffSummary summary; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1021 | |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1022 | int i; |
| 1023 | int numUnflaggedArguments = 0; |
| 1024 | for (i = 1; i < argc; i++) { |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1025 | if (!strcmp(argv[i], "--help")) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1026 | usage(argv[0]); |
| 1027 | return 0; |
| 1028 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1029 | if (!strcmp(argv[i], "--nodiffs")) { |
| 1030 | generateDiffs = false; |
| 1031 | continue; |
| 1032 | } |
| 1033 | if (!strcmp(argv[i], "--threshold")) { |
| 1034 | colorThreshold = atoi(argv[++i]); |
| 1035 | continue; |
| 1036 | } |
| 1037 | if (!strcmp(argv[i], "--match")) { |
| 1038 | matchSubstrings.push(new SkString(argv[++i])); |
| 1039 | continue; |
| 1040 | } |
| 1041 | if (!strcmp(argv[i], "--nomatch")) { |
| 1042 | nomatchSubstrings.push(new SkString(argv[++i])); |
| 1043 | continue; |
| 1044 | } |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1045 | if (!strcmp(argv[i], "--sortbymismatch")) { |
reed@google.com | c7a67cb | 2012-05-07 14:52:12 +0000 | [diff] [blame] | 1046 | sortProc = SkCastForQSort(compare_diff_mean_mismatches); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1047 | continue; |
| 1048 | } |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1049 | if (!strcmp(argv[i], "--sortbymaxmismatch")) { |
reed@google.com | c7a67cb | 2012-05-07 14:52:12 +0000 | [diff] [blame] | 1050 | sortProc = SkCastForQSort(compare_diff_max_mismatches); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1051 | continue; |
| 1052 | } |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1053 | if (!strcmp(argv[i], "--weighted")) { |
reed@google.com | c7a67cb | 2012-05-07 14:52:12 +0000 | [diff] [blame] | 1054 | sortProc = SkCastForQSort(compare_diff_weighted); |
tomhudson@google.com | 5b32529 | 2011-05-24 19:41:13 +0000 | [diff] [blame] | 1055 | continue; |
| 1056 | } |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1057 | if (!strcmp(argv[i], "--chromium-release")) { |
| 1058 | analyzeChromium = true; |
| 1059 | chromiumRelease = true; |
| 1060 | continue; |
| 1061 | } |
| 1062 | if (!strcmp(argv[i], "--chromium-debug")) { |
| 1063 | analyzeChromium = true; |
| 1064 | chromiumDebug = true; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1065 | continue; |
| 1066 | } |
| 1067 | if (argv[i][0] != '-') { |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1068 | switch (numUnflaggedArguments++) { |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1069 | case 0: |
| 1070 | baseDir.set(argv[i]); |
| 1071 | continue; |
| 1072 | case 1: |
| 1073 | comparisonDir.set(argv[i]); |
| 1074 | continue; |
| 1075 | case 2: |
| 1076 | outputDir.set(argv[i]); |
| 1077 | continue; |
| 1078 | default: |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1079 | SkDebugf("extra unflagged argument <%s>\n", argv[i]); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1080 | usage(argv[0]); |
| 1081 | return 0; |
| 1082 | } |
| 1083 | } |
| 1084 | |
| 1085 | SkDebugf("Unrecognized argument <%s>\n", argv[i]); |
| 1086 | usage(argv[0]); |
| 1087 | return 0; |
| 1088 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1089 | |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1090 | if (analyzeChromium) { |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1091 | if (numUnflaggedArguments != 2) { |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1092 | usage(argv[0]); |
| 1093 | return 0; |
| 1094 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1095 | outputDir = comparisonDir; |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1096 | if (chromiumRelease && chromiumDebug) { |
| 1097 | SkDebugf( |
| 1098 | "--chromium must be either -release or -debug, not both!\n"); |
| 1099 | return 0; |
| 1100 | } |
| 1101 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1102 | |
| 1103 | if (numUnflaggedArguments == 2) { |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 1104 | outputDir = comparisonDir; |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1105 | } else if (numUnflaggedArguments != 3) { |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 1106 | usage(argv[0]); |
| 1107 | return 0; |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1108 | } |
| 1109 | |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 1110 | if (!baseDir.endsWith(PATH_DIV_STR)) { |
| 1111 | baseDir.append(PATH_DIV_STR); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1112 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1113 | printf("baseDir is [%s]\n", baseDir.c_str()); |
| 1114 | |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 1115 | if (!comparisonDir.endsWith(PATH_DIV_STR)) { |
| 1116 | comparisonDir.append(PATH_DIV_STR); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1117 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1118 | printf("comparisonDir is [%s]\n", comparisonDir.c_str()); |
| 1119 | |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 1120 | if (!outputDir.endsWith(PATH_DIV_STR)) { |
| 1121 | outputDir.append(PATH_DIV_STR); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1122 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1123 | if (generateDiffs) { |
| 1124 | printf("writing diffs to outputDir is [%s]\n", outputDir.c_str()); |
| 1125 | } else { |
| 1126 | printf("not writing any diffs to outputDir [%s]\n", outputDir.c_str()); |
| 1127 | outputDir.set(""); |
| 1128 | } |
| 1129 | |
| 1130 | // Default substring matching: |
| 1131 | // - No matter what, don't match any PDF files. |
| 1132 | // We may want to change this later, but for now this maintains the filter |
| 1133 | // that get_file_list() used to always apply. |
| 1134 | // - If no matchSubstrings were specified, match ALL strings. |
| 1135 | nomatchSubstrings.push(new SkString(".pdf")); |
| 1136 | if (matchSubstrings.isEmpty()) { |
| 1137 | matchSubstrings.push(new SkString("")); |
| 1138 | } |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1139 | |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1140 | if (analyzeChromium) { |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 1141 | baseDir.append("webkit" PATH_DIV_STR); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1142 | if (chromiumRelease) { |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 1143 | baseDir.append("Release" PATH_DIV_STR); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1144 | } |
| 1145 | if (chromiumDebug) { |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 1146 | baseDir.append("Debug" PATH_DIV_STR); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1147 | } |
bsalomon@google.com | 1a315fe | 2011-09-23 14:56:37 +0000 | [diff] [blame] | 1148 | baseDir.append("layout-test-results" PATH_DIV_STR); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1149 | analyze_chromium(diffProc, colorThreshold, &differences, |
| 1150 | baseDir, outputDir, &summary); |
| 1151 | } else { |
| 1152 | create_diff_images(diffProc, colorThreshold, &differences, |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1153 | baseDir, comparisonDir, outputDir, |
| 1154 | matchSubstrings, nomatchSubstrings, &summary); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1155 | } |
tomhudson@google.com | 9dc527b | 2011-06-09 15:47:10 +0000 | [diff] [blame] | 1156 | summary.print(); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1157 | |
| 1158 | if (differences.count()) { |
reed@google.com | c7a67cb | 2012-05-07 14:52:12 +0000 | [diff] [blame] | 1159 | qsort(differences.begin(), differences.count(), |
| 1160 | sizeof(DiffRecord*), sortProc); |
tomhudson@google.com | 7d04280 | 2011-07-14 13:15:55 +0000 | [diff] [blame] | 1161 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1162 | |
| 1163 | if (generateDiffs) { |
| 1164 | print_diff_page(summary.fNumMatches, colorThreshold, differences, |
| 1165 | baseDir, comparisonDir, outputDir); |
| 1166 | } |
| 1167 | |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1168 | for (i = 0; i < differences.count(); i++) { |
| 1169 | delete differences[i]; |
| 1170 | } |
epoger@google.com | a5f406e | 2012-05-01 13:26:16 +0000 | [diff] [blame] | 1171 | matchSubstrings.deleteAll(); |
| 1172 | nomatchSubstrings.deleteAll(); |
tomhudson@google.com | 4b33d28 | 2011-04-27 15:39:30 +0000 | [diff] [blame] | 1173 | } |