| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // This file input format is based loosely on | 5 // This file input format is based loosely on |
| 6 // Tools/DumpRenderTree/ImageDiff.m | 6 // Tools/DumpRenderTree/ImageDiff.m |
| 7 | 7 |
| 8 // The exact format of this tool's output to stdout is important, to match | 8 // The exact format of this tool's output to stdout is important, to match |
| 9 // what the run-webkit-tests script expects. | 9 // what the run-webkit-tests script expects. |
| 10 | 10 |
| 11 #include <algorithm> | 11 #include <algorithm> |
| 12 #include <iostream> | 12 #include <iostream> |
| 13 #include <string> | 13 #include <string> |
| 14 #include <vector> | 14 #include <vector> |
| 15 | 15 |
| 16 #include "base/basictypes.h" | 16 #include "base/basictypes.h" |
| 17 #include "base/command_line.h" | 17 #include "base/command_line.h" |
| 18 #include "base/file_util.h" | 18 #include "base/file_util.h" |
| 19 #include "base/files/file_path.h" | 19 #include "base/files/file_path.h" |
| 20 #include "base/logging.h" | 20 #include "base/logging.h" |
| 21 #include "base/memory/scoped_ptr.h" | 21 #include "base/memory/scoped_ptr.h" |
| 22 #include "base/process/memory.h" | 22 #include "base/process/memory.h" |
| 23 #include "base/safe_numerics.h" | 23 #include "base/safe_numerics.h" |
| 24 #include "base/strings/string_util.h" | 24 #include "base/strings/string_util.h" |
| 25 #include "base/strings/utf_string_conversions.h" | 25 #include "base/strings/utf_string_conversions.h" |
| 26 #include "webkit/support/webkit_support_gfx.h" | 26 #include "tools/imagediff/image_diff_png.h" |
| 27 | 27 |
| 28 #if defined(OS_WIN) | 28 #if defined(OS_WIN) |
| 29 #include "windows.h" | 29 #include "windows.h" |
| 30 #endif | 30 #endif |
| 31 | 31 |
| 32 // Causes the app to remain open, waiting for pairs of filenames on stdin. | 32 // Causes the app to remain open, waiting for pairs of filenames on stdin. |
| 33 // The caller is then responsible for terminating this app. | 33 // The caller is then responsible for terminating this app. |
| 34 static const char kOptionPollStdin[] = "use-stdin"; | 34 static const char kOptionPollStdin[] = "use-stdin"; |
| 35 static const char kOptionGenerateDiff[] = "diff"; | 35 static const char kOptionGenerateDiff[] = "diff"; |
| 36 | 36 |
| (...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 73 // Creates the image from stdin with the given data length. On success, it | 73 // Creates the image from stdin with the given data length. On success, it |
| 74 // will return true. On failure, no other methods should be accessed. | 74 // will return true. On failure, no other methods should be accessed. |
| 75 bool CreateFromStdin(size_t byte_length) { | 75 bool CreateFromStdin(size_t byte_length) { |
| 76 if (byte_length == 0) | 76 if (byte_length == 0) |
| 77 return false; | 77 return false; |
| 78 | 78 |
| 79 scoped_ptr<unsigned char[]> source(new unsigned char[byte_length]); | 79 scoped_ptr<unsigned char[]> source(new unsigned char[byte_length]); |
| 80 if (fread(source.get(), 1, byte_length, stdin) != byte_length) | 80 if (fread(source.get(), 1, byte_length, stdin) != byte_length) |
| 81 return false; | 81 return false; |
| 82 | 82 |
| 83 if (!webkit_support::DecodePNG(source.get(), byte_length, | 83 if (!image_diff_png::DecodePNG(source.get(), byte_length, |
| 84 &data_, &w_, &h_)) { | 84 &data_, &w_, &h_)) { |
| 85 Clear(); | 85 Clear(); |
| 86 return false; | 86 return false; |
| 87 } | 87 } |
| 88 return true; | 88 return true; |
| 89 } | 89 } |
| 90 | 90 |
| 91 // Creates the image from the given filename on disk, and returns true on | 91 // Creates the image from the given filename on disk, and returns true on |
| 92 // success. | 92 // success. |
| 93 bool CreateFromFilename(const base::FilePath& path) { | 93 bool CreateFromFilename(const base::FilePath& path) { |
| 94 FILE* f = file_util::OpenFile(path, "rb"); | 94 FILE* f = file_util::OpenFile(path, "rb"); |
| 95 if (!f) | 95 if (!f) |
| 96 return false; | 96 return false; |
| 97 | 97 |
| 98 std::vector<unsigned char> compressed; | 98 std::vector<unsigned char> compressed; |
| 99 const int buf_size = 1024; | 99 const int buf_size = 1024; |
| 100 unsigned char buf[buf_size]; | 100 unsigned char buf[buf_size]; |
| 101 size_t num_read = 0; | 101 size_t num_read = 0; |
| 102 while ((num_read = fread(buf, 1, buf_size, f)) > 0) { | 102 while ((num_read = fread(buf, 1, buf_size, f)) > 0) { |
| 103 compressed.insert(compressed.end(), buf, buf + num_read); | 103 compressed.insert(compressed.end(), buf, buf + num_read); |
| 104 } | 104 } |
| 105 | 105 |
| 106 file_util::CloseFile(f); | 106 file_util::CloseFile(f); |
| 107 | 107 |
| 108 if (!webkit_support::DecodePNG(&compressed[0], compressed.size(), | 108 if (!image_diff_png::DecodePNG(&compressed[0], compressed.size(), |
| 109 &data_, &w_, &h_)) { | 109 &data_, &w_, &h_)) { |
| 110 Clear(); | 110 Clear(); |
| 111 return false; | 111 return false; |
| 112 } | 112 } |
| 113 return true; | 113 return true; |
| 114 } | 114 } |
| 115 | 115 |
| 116 void Clear() { | 116 void Clear() { |
| 117 w_ = h_ = 0; | 117 w_ = h_ = 0; |
| 118 data_.clear(); | 118 data_.clear(); |
| (...skipping 189 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 308 file2.value().c_str()); | 308 file2.value().c_str()); |
| 309 return kStatusError; | 309 return kStatusError; |
| 310 } | 310 } |
| 311 | 311 |
| 312 Image diff_image; | 312 Image diff_image; |
| 313 bool same = CreateImageDiff(baseline_image, actual_image, &diff_image); | 313 bool same = CreateImageDiff(baseline_image, actual_image, &diff_image); |
| 314 if (same) | 314 if (same) |
| 315 return kStatusSame; | 315 return kStatusSame; |
| 316 | 316 |
| 317 std::vector<unsigned char> png_encoding; | 317 std::vector<unsigned char> png_encoding; |
| 318 webkit_support::EncodeRGBAPNG( | 318 image_diff_png::EncodeRGBAPNG( |
| 319 diff_image.data(), diff_image.w(), diff_image.h(), | 319 diff_image.data(), diff_image.w(), diff_image.h(), |
| 320 diff_image.w() * 4, &png_encoding); | 320 diff_image.w() * 4, &png_encoding); |
| 321 if (file_util::WriteFile(out_file, | 321 if (file_util::WriteFile(out_file, |
| 322 reinterpret_cast<char*>(&png_encoding.front()), | 322 reinterpret_cast<char*>(&png_encoding.front()), |
| 323 base::checked_numeric_cast<int>(png_encoding.size())) < 0) | 323 base::checked_numeric_cast<int>(png_encoding.size())) < 0) |
| 324 return kStatusError; | 324 return kStatusError; |
| 325 | 325 |
| 326 return kStatusDifferent; | 326 return kStatusDifferent; |
| 327 } | 327 } |
| 328 | 328 |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 372 base::FilePath(args[1]), | 372 base::FilePath(args[1]), |
| 373 base::FilePath(args[2])); | 373 base::FilePath(args[2])); |
| 374 } | 374 } |
| 375 } else if (args.size() == 2) { | 375 } else if (args.size() == 2) { |
| 376 return CompareImages(base::FilePath(args[0]), base::FilePath(args[1])); | 376 return CompareImages(base::FilePath(args[0]), base::FilePath(args[1])); |
| 377 } | 377 } |
| 378 | 378 |
| 379 PrintHelp(); | 379 PrintHelp(); |
| 380 return kStatusError; | 380 return kStatusError; |
| 381 } | 381 } |
| OLD | NEW |