| OLD | NEW |
| 1 // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 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 // WebKitTools/DumpRenderTree/ImageDiff.m | 6 // WebKitTools/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 <vector> | 12 #include <vector> |
| 13 #include <string> | 13 #include <string> |
| 14 #include <iostream> | 14 #include <iostream> |
| 15 | 15 |
| 16 #include "app/gfx/codec/png_codec.h" |
| 16 #include "base/basictypes.h" | 17 #include "base/basictypes.h" |
| 17 #include "base/command_line.h" | 18 #include "base/command_line.h" |
| 18 #include "base/file_util.h" | 19 #include "base/file_util.h" |
| 19 #include "base/gfx/png_decoder.h" | |
| 20 #include "base/gfx/png_encoder.h" | |
| 21 #include "base/logging.h" | 20 #include "base/logging.h" |
| 22 #include "base/process_util.h" | 21 #include "base/process_util.h" |
| 23 #include "base/scoped_ptr.h" | 22 #include "base/scoped_ptr.h" |
| 24 #include "base/string_util.h" | 23 #include "base/string_util.h" |
| 25 | 24 |
| 26 #if defined(OS_WIN) | 25 #if defined(OS_WIN) |
| 27 #include "windows.h" | 26 #include "windows.h" |
| 28 #endif | 27 #endif |
| 29 | 28 |
| 30 // Causes the app to remain open, waiting for pairs of filenames on stdin. | 29 // Causes the app to remain open, waiting for pairs of filenames on stdin. |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 71 // Creates the image from stdin with the given data length. On success, it | 70 // Creates the image from stdin with the given data length. On success, it |
| 72 // will return true. On failure, no other methods should be accessed. | 71 // will return true. On failure, no other methods should be accessed. |
| 73 bool CreateFromStdin(size_t byte_length) { | 72 bool CreateFromStdin(size_t byte_length) { |
| 74 if (byte_length == 0) | 73 if (byte_length == 0) |
| 75 return false; | 74 return false; |
| 76 | 75 |
| 77 scoped_array<unsigned char> source(new unsigned char[byte_length]); | 76 scoped_array<unsigned char> source(new unsigned char[byte_length]); |
| 78 if (fread(source.get(), 1, byte_length, stdin) != byte_length) | 77 if (fread(source.get(), 1, byte_length, stdin) != byte_length) |
| 79 return false; | 78 return false; |
| 80 | 79 |
| 81 if (!PNGDecoder::Decode(source.get(), byte_length, PNGDecoder::FORMAT_RGBA, | 80 if (!gfx::PNGCodec::Decode(source.get(), byte_length, |
| 82 &data_, &w_, &h_)) { | 81 gfx::PNGCodec::FORMAT_RGBA, |
| 82 &data_, &w_, &h_)) { |
| 83 Clear(); | 83 Clear(); |
| 84 return false; | 84 return false; |
| 85 } | 85 } |
| 86 return true; | 86 return true; |
| 87 } | 87 } |
| 88 | 88 |
| 89 // Creates the image from the given filename on disk, and returns true on | 89 // Creates the image from the given filename on disk, and returns true on |
| 90 // success. | 90 // success. |
| 91 bool CreateFromFilename(const char* filename) { | 91 bool CreateFromFilename(const char* filename) { |
| 92 FILE* f = file_util::OpenFile(std::string(filename), "rb"); | 92 FILE* f = file_util::OpenFile(std::string(filename), "rb"); |
| 93 if (!f) | 93 if (!f) |
| 94 return false; | 94 return false; |
| 95 | 95 |
| 96 std::vector<unsigned char> compressed; | 96 std::vector<unsigned char> compressed; |
| 97 const int buf_size = 1024; | 97 const int buf_size = 1024; |
| 98 unsigned char buf[buf_size]; | 98 unsigned char buf[buf_size]; |
| 99 size_t num_read = 0; | 99 size_t num_read = 0; |
| 100 while ((num_read = fread(buf, 1, buf_size, f)) > 0) { | 100 while ((num_read = fread(buf, 1, buf_size, f)) > 0) { |
| 101 std::copy(buf, &buf[num_read], std::back_inserter(compressed)); | 101 std::copy(buf, &buf[num_read], std::back_inserter(compressed)); |
| 102 } | 102 } |
| 103 | 103 |
| 104 file_util::CloseFile(f); | 104 file_util::CloseFile(f); |
| 105 | 105 |
| 106 if (!PNGDecoder::Decode(&compressed[0], compressed.size(), | 106 if (!gfx::PNGCodec::Decode(&compressed[0], compressed.size(), |
| 107 PNGDecoder::FORMAT_RGBA, &data_, &w_, &h_)) { | 107 gfx::PNGCodec::FORMAT_RGBA, &data_, &w_, &h_)) { |
| 108 Clear(); | 108 Clear(); |
| 109 return false; | 109 return false; |
| 110 } | 110 } |
| 111 return true; | 111 return true; |
| 112 } | 112 } |
| 113 | 113 |
| 114 void Clear() { | 114 void Clear() { |
| 115 w_ = h_ = 0; | 115 w_ = h_ = 0; |
| 116 data_.clear(); | 116 data_.clear(); |
| 117 } | 117 } |
| (...skipping 183 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 301 fprintf(stderr, "image_diff: Unable to open file \"%s\"\n", file2); | 301 fprintf(stderr, "image_diff: Unable to open file \"%s\"\n", file2); |
| 302 return kStatusError; | 302 return kStatusError; |
| 303 } | 303 } |
| 304 | 304 |
| 305 Image diff_image; | 305 Image diff_image; |
| 306 bool same = CreateImageDiff(baseline_image, actual_image, &diff_image); | 306 bool same = CreateImageDiff(baseline_image, actual_image, &diff_image); |
| 307 if (same) | 307 if (same) |
| 308 return kStatusSame; | 308 return kStatusSame; |
| 309 | 309 |
| 310 std::vector<unsigned char> png_encoding; | 310 std::vector<unsigned char> png_encoding; |
| 311 PNGEncoder::Encode(diff_image.data(), PNGEncoder::FORMAT_RGBA, | 311 gfx::PNGCodec::Encode(diff_image.data(), gfx::PNGCodec::FORMAT_RGBA, |
| 312 diff_image.w(), diff_image.h(), diff_image.w() * 4, | 312 diff_image.w(), diff_image.h(), diff_image.w() * 4, |
| 313 false, &png_encoding); | 313 false, &png_encoding); |
| 314 if (file_util::WriteFile(UTF8ToWide(out_file), | 314 if (file_util::WriteFile(UTF8ToWide(out_file), |
| 315 reinterpret_cast<char*>(&png_encoding.front()), png_encoding.size()) < 0) | 315 reinterpret_cast<char*>(&png_encoding.front()), png_encoding.size()) < 0) |
| 316 return kStatusError; | 316 return kStatusError; |
| 317 | 317 |
| 318 return kStatusDifferent; | 318 return kStatusDifferent; |
| 319 } | 319 } |
| 320 | 320 |
| 321 int main(int argc, const char* argv[]) { | 321 int main(int argc, const char* argv[]) { |
| 322 base::EnableTerminationOnHeapCorruption(); | 322 base::EnableTerminationOnHeapCorruption(); |
| 323 CommandLine::Init(argc, argv); | 323 CommandLine::Init(argc, argv); |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 355 WideToUTF8(values[1]).c_str(), | 355 WideToUTF8(values[1]).c_str(), |
| 356 WideToUTF8(values[2]).c_str()); | 356 WideToUTF8(values[2]).c_str()); |
| 357 } | 357 } |
| 358 } else if (values.size() == 2) { | 358 } else if (values.size() == 2) { |
| 359 return CompareImages(argv[1], argv[2]); | 359 return CompareImages(argv[1], argv[2]); |
| 360 } | 360 } |
| 361 | 361 |
| 362 PrintHelp(); | 362 PrintHelp(); |
| 363 return kStatusError; | 363 return kStatusError; |
| 364 } | 364 } |
| OLD | NEW |