1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "cc/test/pixel_test_utils.h"
#include <string>
#include <vector>
#include "base/base64.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/logging.h"
#include "base/strings/strcat.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "ui/gfx/codec/png_codec.h"
namespace cc {
bool WritePNGFile(const SkBitmap& bitmap, const base::FilePath& file_path,
bool discard_transparency) {
std::optional<std::vector<uint8_t>> png_data =
gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, discard_transparency);
if (png_data && base::CreateDirectory(file_path.DirName())) {
return base::WriteFile(file_path, png_data.value());
}
return false;
}
std::string GetPNGDataUrl(const SkBitmap& bitmap) {
std::optional<std::vector<uint8_t>> png_data =
gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, /*discard_transparency=*/false);
return base::StrCat(
{"data:image/png;base64,",
base::Base64Encode(png_data.value_or(std::vector<uint8_t>()))});
}
SkBitmap ReadPNGFile(const base::FilePath& file_path) {
std::optional<std::vector<uint8_t>> png_data =
base::ReadFileToBytes(file_path);
if (!png_data) {
return SkBitmap();
}
return gfx::PNGCodec::Decode(png_data.value());
}
bool MatchesBitmap(const SkBitmap& gen_bmp,
const SkBitmap& ref_bmp,
const PixelComparator& comparator) {
bool pixels_match = true;
// Check if images size matches
if (gen_bmp.width() != ref_bmp.width() ||
gen_bmp.height() != ref_bmp.height()) {
LOG(ERROR)
<< "Dimensions do not match! "
<< "Actual: " << gen_bmp.width() << "x" << gen_bmp.height()
<< "; "
<< "Expected: " << ref_bmp.width() << "x" << ref_bmp.height();
pixels_match = false;
}
// Shortcut for empty images. They are always equal.
if (pixels_match && (gen_bmp.width() == 0 || gen_bmp.height() == 0))
return true;
if (pixels_match && !comparator.Compare(gen_bmp, ref_bmp)) {
LOG(ERROR) << "Pixels do not match!";
pixels_match = false;
}
if (!pixels_match) {
LOG(ERROR) << "Actual pixels (open in browser):\n"
<< GetPNGDataUrl(gen_bmp);
LOG(ERROR) << "Expected pixels (open in browser):\n"
<< GetPNGDataUrl(ref_bmp);
}
return pixels_match;
}
bool MatchesPNGFile(const SkBitmap& gen_bmp,
base::FilePath ref_img_path,
const PixelComparator& comparator) {
SkBitmap ref_bmp = ReadPNGFile(ref_img_path);
if (ref_bmp.isNull()) {
LOG(ERROR) << "Cannot read reference image: " << ref_img_path.value();
return false;
}
return MatchesBitmap(gen_bmp, ref_bmp, comparator);
}
} // namespace cc
|
