1 /*
2 * Copyright 2017 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
8 #include "TestUtils.h"
9
10 #include "GrContextPriv.h"
11 #include "GrSurfaceContext.h"
12 #include "GrSurfaceContextPriv.h"
13 #include "GrSurfaceProxy.h"
14 #include "GrTextureProxy.h"
15 #include "ProxyUtils.h"
16 #include "SkGr.h"
17 #include "SkBase64.h"
18 #include "SkPngEncoder.h"
19
test_read_pixels(skiatest::Reporter * reporter,GrSurfaceContext * srcContext,uint32_t expectedPixelValues[],const char * testName)20 void test_read_pixels(skiatest::Reporter* reporter,
21 GrSurfaceContext* srcContext, uint32_t expectedPixelValues[],
22 const char* testName) {
23 int pixelCnt = srcContext->width() * srcContext->height();
24 SkAutoTMalloc<uint32_t> pixels(pixelCnt);
25 memset(pixels.get(), 0, sizeof(uint32_t)*pixelCnt);
26
27 SkImageInfo ii = SkImageInfo::Make(srcContext->width(), srcContext->height(),
28 kRGBA_8888_SkColorType, kPremul_SkAlphaType);
29 bool read = srcContext->readPixels(ii, pixels.get(), 0, 0, 0);
30 if (!read) {
31 ERRORF(reporter, "%s: Error reading from texture.", testName);
32 }
33
34 for (int i = 0; i < pixelCnt; ++i) {
35 if (pixels.get()[i] != expectedPixelValues[i]) {
36 ERRORF(reporter, "%s: Error, pixel value %d should be 0x%08x, got 0x%08x.",
37 testName, i, expectedPixelValues[i], pixels.get()[i]);
38 break;
39 }
40 }
41 }
42
test_write_pixels(skiatest::Reporter * reporter,GrSurfaceContext * dstContext,bool expectedToWork,const char * testName)43 void test_write_pixels(skiatest::Reporter* reporter,
44 GrSurfaceContext* dstContext, bool expectedToWork,
45 const char* testName) {
46 int pixelCnt = dstContext->width() * dstContext->height();
47 SkAutoTMalloc<uint32_t> pixels(pixelCnt);
48 for (int y = 0; y < dstContext->width(); ++y) {
49 for (int x = 0; x < dstContext->height(); ++x) {
50 pixels.get()[y * dstContext->width() + x] =
51 SkColorToPremulGrColor(SkColorSetARGB(2*y, x, y, x + y));
52 }
53 }
54
55 SkImageInfo ii = SkImageInfo::Make(dstContext->width(), dstContext->height(),
56 kRGBA_8888_SkColorType, kPremul_SkAlphaType);
57 bool write = dstContext->writePixels(ii, pixels.get(), 0, 0, 0);
58 if (!write) {
59 if (expectedToWork) {
60 ERRORF(reporter, "%s: Error writing to texture.", testName);
61 }
62 return;
63 }
64
65 if (write && !expectedToWork) {
66 ERRORF(reporter, "%s: writePixels succeeded when it wasn't supposed to.", testName);
67 return;
68 }
69
70 test_read_pixels(reporter, dstContext, pixels.get(), testName);
71 }
72
test_copy_from_surface(skiatest::Reporter * reporter,GrContext * context,GrSurfaceProxy * proxy,uint32_t expectedPixelValues[],bool onlyTestRTConfig,const char * testName)73 void test_copy_from_surface(skiatest::Reporter* reporter, GrContext* context,
74 GrSurfaceProxy* proxy, uint32_t expectedPixelValues[],
75 bool onlyTestRTConfig, const char* testName) {
76 GrSurfaceDesc copyDstDesc;
77 copyDstDesc.fWidth = proxy->width();
78 copyDstDesc.fHeight = proxy->height();
79 copyDstDesc.fConfig = kRGBA_8888_GrPixelConfig;
80
81 for (auto flags : { kNone_GrSurfaceFlags, kRenderTarget_GrSurfaceFlag }) {
82 if (kNone_GrSurfaceFlags == flags && onlyTestRTConfig) {
83 continue;
84 }
85
86 copyDstDesc.fFlags = flags;
87 auto origin = (kNone_GrSurfaceFlags == flags) ? kTopLeft_GrSurfaceOrigin
88 : kBottomLeft_GrSurfaceOrigin;
89
90 sk_sp<GrSurfaceContext> dstContext(
91 GrSurfaceProxy::TestCopy(context, copyDstDesc, origin, proxy));
92
93 test_read_pixels(reporter, dstContext.get(), expectedPixelValues, testName);
94 }
95 }
96
test_copy_to_surface(skiatest::Reporter * reporter,GrContext * context,GrSurfaceContext * dstContext,const char * testName)97 void test_copy_to_surface(skiatest::Reporter* reporter,
98 GrContext* context,
99 GrSurfaceContext* dstContext,
100 const char* testName) {
101
102 int pixelCnt = dstContext->width() * dstContext->height();
103 SkAutoTMalloc<uint32_t> pixels(pixelCnt);
104 for (int y = 0; y < dstContext->width(); ++y) {
105 for (int x = 0; x < dstContext->height(); ++x) {
106 pixels.get()[y * dstContext->width() + x] =
107 SkColorToPremulGrColor(SkColorSetARGB(2*y, y, x, x * y));
108 }
109 }
110
111 for (auto isRT : {false, true}) {
112 for (auto origin : {kTopLeft_GrSurfaceOrigin, kBottomLeft_GrSurfaceOrigin}) {
113 auto src = sk_gpu_test::MakeTextureProxyFromData(
114 context, isRT, dstContext->width(),
115 dstContext->height(), GrColorType::kRGBA_8888, origin, pixels.get(), 0);
116 dstContext->copy(src.get());
117 test_read_pixels(reporter, dstContext, pixels.get(), testName);
118 }
119 }
120 }
121
fill_pixel_data(int width,int height,GrColor * data)122 void fill_pixel_data(int width, int height, GrColor* data) {
123 for (int j = 0; j < height; ++j) {
124 for (int i = 0; i < width; ++i) {
125 unsigned int red = (unsigned int)(256.f * (i / (float)width));
126 unsigned int green = (unsigned int)(256.f * (j / (float)height));
127 data[i + j * width] = GrColorPackRGBA(red - (red >> 8), green - (green >> 8),
128 0xff, 0xff);
129 }
130 }
131 }
132
does_full_buffer_contain_correct_color(GrColor * srcBuffer,GrColor * dstBuffer,int width,int height)133 bool does_full_buffer_contain_correct_color(GrColor* srcBuffer,
134 GrColor* dstBuffer,
135 int width,
136 int height) {
137 GrColor* srcPtr = srcBuffer;
138 GrColor* dstPtr = dstBuffer;
139 for (int j = 0; j < height; ++j) {
140 for (int i = 0; i < width; ++i) {
141 if (srcPtr[i] != dstPtr[i]) {
142 return false;
143 }
144 }
145 srcPtr += width;
146 dstPtr += width;
147 }
148 return true;
149 }
150
bitmap_to_base64_data_uri(const SkBitmap & bitmap,SkString * dst)151 bool bitmap_to_base64_data_uri(const SkBitmap& bitmap, SkString* dst) {
152 SkPixmap pm;
153 if (!bitmap.peekPixels(&pm)) {
154 dst->set("peekPixels failed");
155 return false;
156 }
157
158 // We're going to embed this PNG in a data URI, so make it as small as possible
159 SkPngEncoder::Options options;
160 options.fFilterFlags = SkPngEncoder::FilterFlag::kAll;
161 options.fZLibLevel = 9;
162
163 SkDynamicMemoryWStream wStream;
164 if (!SkPngEncoder::Encode(&wStream, pm, options)) {
165 dst->set("SkPngEncoder::Encode failed");
166 return false;
167 }
168
169 sk_sp<SkData> pngData = wStream.detachAsData();
170 size_t len = SkBase64::Encode(pngData->data(), pngData->size(), nullptr);
171
172 // The PNG can be almost arbitrarily large. We don't want to fill our logs with enormous URLs.
173 // Infra says these can be pretty big, as long as we're only outputting them on failure.
174 static const size_t kMaxBase64Length = 1024 * 1024;
175 if (len > kMaxBase64Length) {
176 dst->printf("Encoded image too large (%u bytes)", static_cast<uint32_t>(len));
177 return false;
178 }
179
180 dst->resize(len);
181 SkBase64::Encode(pngData->data(), pngData->size(), dst->writable_str());
182 dst->prepend("data:image/png;base64,");
183 return true;
184 }
185