Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2019 Google LLC |
| 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 "gm/gm.h" |
| 9 | #include "include/core/SkCanvas.h" |
| 10 | #include "include/core/SkColor.h" |
| 11 | #include "include/core/SkPaint.h" |
| 12 | #include "include/core/SkRect.h" |
| 13 | #include "include/core/SkSurface.h" |
Brian Salomon | 7db7139 | 2020-10-16 10:05:21 -0400 | [diff] [blame] | 14 | #include "include/core/SkYUVAInfo.h" |
| 15 | #include "include/core/SkYUVAPixmaps.h" |
Brian Salomon | bacbb92 | 2021-01-21 19:48:00 -0500 | [diff] [blame] | 16 | #include "include/effects/SkGradientShader.h" |
Robert Phillips | 2a4acf3 | 2020-07-06 15:50:15 -0400 | [diff] [blame] | 17 | #include "include/gpu/GrDirectContext.h" |
Robert Phillips | b7bfbc2 | 2020-07-01 12:55:01 -0400 | [diff] [blame] | 18 | #include "include/gpu/GrRecordingContext.h" |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 19 | #include "src/core/SkAutoPixmapStorage.h" |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 20 | #include "src/core/SkScopeExit.h" |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 21 | #include "tools/Resources.h" |
| 22 | #include "tools/ToolUtils.h" |
Brian Salomon | 7db7139 | 2020-10-16 10:05:21 -0400 | [diff] [blame] | 23 | #include "tools/gpu/YUVUtils.h" |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 24 | |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 25 | namespace { |
| 26 | struct AsyncContext { |
| 27 | bool fCalled = false; |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 28 | std::unique_ptr<const SkImage::AsyncReadResult> fResult; |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 29 | }; |
| 30 | } // anonymous namespace |
| 31 | |
| 32 | // Making this a lambda in the test functions caused: |
| 33 | // "error: cannot compile this forwarded non-trivially copyable parameter yet" |
| 34 | // on x86/Win/Clang bot, referring to 'result'. |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 35 | static void async_callback(void* c, std::unique_ptr<const SkImage::AsyncReadResult> result) { |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 36 | auto context = static_cast<AsyncContext*>(c); |
| 37 | context->fResult = std::move(result); |
| 38 | context->fCalled = true; |
| 39 | }; |
| 40 | |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 41 | // Draws the image to a surface, does a asyncRescaleAndReadPixels of the image, and then sticks |
| 42 | // the result in a raster image. |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 43 | template <typename Src> |
| 44 | static sk_sp<SkImage> do_read_and_scale(Src* src, |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 45 | GrDirectContext* direct, |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 46 | const SkIRect& srcRect, |
| 47 | const SkImageInfo& ii, |
| 48 | SkImage::RescaleGamma rescaleGamma, |
Mike Reed | 1efa14d | 2021-01-02 21:44:59 -0500 | [diff] [blame] | 49 | SkImage::RescaleMode rescaleMode) { |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 50 | auto* asyncContext = new AsyncContext(); |
Mike Reed | 1efa14d | 2021-01-02 21:44:59 -0500 | [diff] [blame] | 51 | src->asyncRescaleAndReadPixels(ii, srcRect, rescaleGamma, rescaleMode, async_callback, |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 52 | asyncContext); |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 53 | if (direct) { |
| 54 | direct->submit(); |
Brian Salomon | 3927802 | 2020-06-02 11:47:26 -0400 | [diff] [blame] | 55 | } |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 56 | while (!asyncContext->fCalled) { |
Brian Salomon | 9c21978 | 2019-10-02 22:51:08 +0000 | [diff] [blame] | 57 | // Only GPU should actually be asynchronous. |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 58 | SkASSERT(direct); |
| 59 | direct->checkAsyncWorkCompletion(); |
Brian Salomon | 9c21978 | 2019-10-02 22:51:08 +0000 | [diff] [blame] | 60 | } |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 61 | if (!asyncContext->fResult) { |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 62 | return nullptr; |
| 63 | } |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 64 | SkPixmap pixmap(ii, asyncContext->fResult->data(0), asyncContext->fResult->rowBytes(0)); |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 65 | auto releasePixels = [](const void*, void* c) { delete static_cast<AsyncContext*>(c); }; |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 66 | return SkImage::MakeFromRaster(pixmap, releasePixels, asyncContext); |
Ravi Mistry | cb55010 | 2019-10-03 09:06:25 +0000 | [diff] [blame] | 67 | } |
| 68 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 69 | template <typename Src> |
| 70 | static sk_sp<SkImage> do_read_and_scale_yuv(Src* src, |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 71 | GrDirectContext* direct, |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 72 | SkYUVColorSpace yuvCS, |
| 73 | const SkIRect& srcRect, |
| 74 | SkISize size, |
| 75 | SkImage::RescaleGamma rescaleGamma, |
Mike Reed | 1efa14d | 2021-01-02 21:44:59 -0500 | [diff] [blame] | 76 | SkImage::RescaleMode rescaleMode, |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 77 | SkScopeExit* cleanup) { |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 78 | SkASSERT(!(size.width() & 0b1) && !(size.height() & 0b1)); |
Ravi Mistry | cb55010 | 2019-10-03 09:06:25 +0000 | [diff] [blame] | 79 | |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 80 | SkISize uvSize = {size.width()/2, size.height()/2}; |
| 81 | SkImageInfo yII = SkImageInfo::Make(size, kGray_8_SkColorType, kPremul_SkAlphaType); |
| 82 | SkImageInfo uvII = SkImageInfo::Make(uvSize, kGray_8_SkColorType, kPremul_SkAlphaType); |
Ravi Mistry | cb55010 | 2019-10-03 09:06:25 +0000 | [diff] [blame] | 83 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 84 | AsyncContext asyncContext; |
| 85 | src->asyncRescaleAndReadPixelsYUV420(yuvCS, SkColorSpace::MakeSRGB(), srcRect, size, |
Mike Reed | 1efa14d | 2021-01-02 21:44:59 -0500 | [diff] [blame] | 86 | rescaleGamma, rescaleMode, async_callback, &asyncContext); |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 87 | if (direct) { |
| 88 | direct->submit(); |
Brian Salomon | 3927802 | 2020-06-02 11:47:26 -0400 | [diff] [blame] | 89 | } |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 90 | while (!asyncContext.fCalled) { |
Ravi Mistry | cb55010 | 2019-10-03 09:06:25 +0000 | [diff] [blame] | 91 | // Only GPU should actually be asynchronous. |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 92 | SkASSERT(direct); |
| 93 | direct->checkAsyncWorkCompletion(); |
Ravi Mistry | cb55010 | 2019-10-03 09:06:25 +0000 | [diff] [blame] | 94 | } |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 95 | if (!asyncContext.fResult) { |
Brian Salomon | cd5caa3 | 2019-06-11 17:02:19 -0400 | [diff] [blame] | 96 | return nullptr; |
| 97 | } |
Brian Salomon | e438738 | 2020-11-11 16:34:19 -0500 | [diff] [blame] | 98 | SkYUVAInfo yuvaInfo(size, |
| 99 | SkYUVAInfo::PlaneConfig::kY_U_V, |
| 100 | SkYUVAInfo::Subsampling::k420, |
| 101 | yuvCS); |
Brian Salomon | 7db7139 | 2020-10-16 10:05:21 -0400 | [diff] [blame] | 102 | SkPixmap yuvPMs[] = { |
| 103 | {yII, asyncContext.fResult->data(0), asyncContext.fResult->rowBytes(0)}, |
| 104 | {uvII, asyncContext.fResult->data(1), asyncContext.fResult->rowBytes(1)}, |
| 105 | {uvII, asyncContext.fResult->data(2), asyncContext.fResult->rowBytes(2)} |
Robert Phillips | 2e1142c | 2019-08-14 12:33:29 -0400 | [diff] [blame] | 106 | }; |
Brian Salomon | 7db7139 | 2020-10-16 10:05:21 -0400 | [diff] [blame] | 107 | auto pixmaps = SkYUVAPixmaps::FromExternalPixmaps(yuvaInfo, yuvPMs); |
| 108 | SkASSERT(pixmaps.isValid()); |
| 109 | auto lazyYUVImage = sk_gpu_test::LazyYUVImage::Make(pixmaps); |
| 110 | SkASSERT(lazyYUVImage); |
| 111 | return lazyYUVImage->refImage(direct, sk_gpu_test::LazyYUVImage::Type::kFromTextures); |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 112 | } |
| 113 | |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 114 | // Draws a grid of rescales. The columns are none, low, and high filter quality. The rows are |
| 115 | // rescale in src gamma and rescale in linear gamma. |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 116 | template <typename Src> |
| 117 | static skiagm::DrawResult do_rescale_grid(SkCanvas* canvas, |
| 118 | Src* src, |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 119 | GrDirectContext* direct, |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 120 | const SkIRect& srcRect, |
| 121 | SkISize newSize, |
| 122 | bool doYUV420, |
| 123 | SkString* errorMsg, |
| 124 | int pad = 0) { |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 125 | if (doYUV420 && !direct) { |
Adlai Holler | e3ad527 | 2020-07-07 10:27:55 -0400 | [diff] [blame] | 126 | errorMsg->printf("YUV420 only supported on direct GPU for now."); |
| 127 | return skiagm::DrawResult::kSkip; |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 128 | } |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 129 | if (canvas->imageInfo().colorType() == kUnknown_SkColorType) { |
| 130 | *errorMsg = "Not supported on recording/vector backends."; |
| 131 | return skiagm::DrawResult::kSkip; |
| 132 | } |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 133 | const auto ii = canvas->imageInfo().makeDimensions(newSize); |
Brian Salomon | a34e0fc | 2019-05-17 19:56:59 +0000 | [diff] [blame] | 134 | |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 135 | SkYUVColorSpace yuvColorSpace = kRec601_SkYUVColorSpace; |
Brian Salomon | a34e0fc | 2019-05-17 19:56:59 +0000 | [diff] [blame] | 136 | canvas->save(); |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 137 | for (auto gamma : {SkImage::RescaleGamma::kSrc, SkImage::RescaleGamma::kLinear}) { |
Brian Salomon | a34e0fc | 2019-05-17 19:56:59 +0000 | [diff] [blame] | 138 | canvas->save(); |
Mike Reed | 1efa14d | 2021-01-02 21:44:59 -0500 | [diff] [blame] | 139 | for (auto mode : { |
| 140 | SkImage::RescaleMode::kNearest, |
| 141 | SkImage::RescaleMode::kRepeatedLinear, |
| 142 | SkImage::RescaleMode::kRepeatedCubic}) { |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 143 | SkScopeExit cleanup; |
| 144 | sk_sp<SkImage> result; |
| 145 | if (doYUV420) { |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 146 | result = do_read_and_scale_yuv(src, direct, yuvColorSpace, srcRect, newSize, gamma, |
Mike Reed | 1efa14d | 2021-01-02 21:44:59 -0500 | [diff] [blame] | 147 | mode, &cleanup); |
Brian Salomon | cd5caa3 | 2019-06-11 17:02:19 -0400 | [diff] [blame] | 148 | if (!result) { |
| 149 | errorMsg->printf("YUV420 async call failed. Allowed for now."); |
| 150 | return skiagm::DrawResult::kSkip; |
| 151 | } |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 152 | int nextCS = static_cast<int>(yuvColorSpace + 1) % (kLastEnum_SkYUVColorSpace + 1); |
| 153 | yuvColorSpace = static_cast<SkYUVColorSpace>(nextCS); |
| 154 | } else { |
Mike Reed | 1efa14d | 2021-01-02 21:44:59 -0500 | [diff] [blame] | 155 | result = do_read_and_scale(src, direct, srcRect, ii, gamma, mode); |
Brian Salomon | cd5caa3 | 2019-06-11 17:02:19 -0400 | [diff] [blame] | 156 | if (!result) { |
| 157 | errorMsg->printf("async read call failed."); |
| 158 | return skiagm::DrawResult::kFail; |
| 159 | } |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 160 | } |
| 161 | canvas->drawImage(result, 0, 0); |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 162 | canvas->translate(newSize.width() + pad, 0); |
Brian Salomon | a34e0fc | 2019-05-17 19:56:59 +0000 | [diff] [blame] | 163 | } |
| 164 | canvas->restore(); |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 165 | canvas->translate(0, newSize.height() + pad); |
Brian Salomon | 451b01f | 2019-05-17 12:39:17 -0400 | [diff] [blame] | 166 | } |
Brian Salomon | a34e0fc | 2019-05-17 19:56:59 +0000 | [diff] [blame] | 167 | canvas->restore(); |
| 168 | return skiagm::DrawResult::kOk; |
Brian Salomon | 451b01f | 2019-05-17 12:39:17 -0400 | [diff] [blame] | 169 | } |
| 170 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 171 | static skiagm::DrawResult do_rescale_image_grid(SkCanvas* canvas, |
| 172 | const char* imageFile, |
| 173 | const SkIRect& srcRect, |
| 174 | SkISize newSize, |
| 175 | bool doSurface, |
| 176 | bool doYUV420, |
| 177 | SkString* errorMsg) { |
Brian Salomon | 286b557 | 2019-05-20 10:25:50 -0400 | [diff] [blame] | 178 | auto image = GetResourceAsImage(imageFile); |
| 179 | if (!image) { |
| 180 | errorMsg->printf("Could not load image file %s.", imageFile); |
| 181 | return skiagm::DrawResult::kFail; |
| 182 | } |
| 183 | if (canvas->imageInfo().colorType() == kUnknown_SkColorType) { |
| 184 | *errorMsg = "Not supported on recording/vector backends."; |
| 185 | return skiagm::DrawResult::kSkip; |
| 186 | } |
Robert Phillips | 41fc174 | 2020-10-15 10:04:03 -0400 | [diff] [blame] | 187 | |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 188 | auto dContext = GrAsDirectContext(canvas->recordingContext()); |
Robert Phillips | 41fc174 | 2020-10-15 10:04:03 -0400 | [diff] [blame] | 189 | if (!dContext && canvas->recordingContext()) { |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 190 | *errorMsg = "Not supported in DDL mode"; |
| 191 | return skiagm::DrawResult::kSkip; |
| 192 | } |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 193 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 194 | if (doSurface) { |
| 195 | // Turn the image into a surface in order to call the read and rescale API |
| 196 | auto surfInfo = image->imageInfo().makeDimensions(image->dimensions()); |
| 197 | auto surface = canvas->makeSurface(surfInfo); |
| 198 | if (!surface && surfInfo.colorType() == kBGRA_8888_SkColorType) { |
| 199 | surfInfo = surfInfo.makeColorType(kRGBA_8888_SkColorType); |
| 200 | surface = canvas->makeSurface(surfInfo); |
Brian Salomon | 1caf378 | 2020-06-26 17:06:31 +0000 | [diff] [blame] | 201 | } |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 202 | if (!surface) { |
| 203 | *errorMsg = "Could not create surface for image."; |
| 204 | // When testing abandoned GrContext we expect surface creation to fail. |
Robert Phillips | 9eb0002 | 2020-06-30 15:30:12 -0400 | [diff] [blame] | 205 | if (canvas->recordingContext() && canvas->recordingContext()->abandoned()) { |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 206 | return skiagm::DrawResult::kSkip; |
| 207 | } |
| 208 | return skiagm::DrawResult::kFail; |
| 209 | } |
| 210 | SkPaint paint; |
| 211 | paint.setBlendMode(SkBlendMode::kSrc); |
Mike Reed | 8d29ab6 | 2021-01-23 18:10:39 -0500 | [diff] [blame] | 212 | surface->getCanvas()->drawImage(image, 0, 0, SkSamplingOptions(), &paint); |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 213 | return do_rescale_grid(canvas, surface.get(), dContext, srcRect, newSize, |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 214 | doYUV420, errorMsg); |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 215 | } else if (dContext) { |
| 216 | image = image->makeTextureImage(dContext); |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 217 | if (!image) { |
| 218 | *errorMsg = "Could not create image."; |
| 219 | // When testing abandoned GrContext we expect surface creation to fail. |
Robert Phillips | 9eb0002 | 2020-06-30 15:30:12 -0400 | [diff] [blame] | 220 | if (canvas->recordingContext() && canvas->recordingContext()->abandoned()) { |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 221 | return skiagm::DrawResult::kSkip; |
| 222 | } |
| 223 | return skiagm::DrawResult::kFail; |
| 224 | } |
Brian Salomon | 1caf378 | 2020-06-26 17:06:31 +0000 | [diff] [blame] | 225 | } |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 226 | return do_rescale_grid(canvas, image.get(), dContext, srcRect, newSize, doYUV420, |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 227 | errorMsg); |
Brian Salomon | 286b557 | 2019-05-20 10:25:50 -0400 | [diff] [blame] | 228 | } |
| 229 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 230 | #define DEF_RESCALE_AND_READ_SURF_GM(IMAGE_FILE, TAG, SRC_RECT, W, H) \ |
| 231 | DEF_SIMPLE_GM_CAN_FAIL(async_rescale_and_read_##TAG, canvas, errorMsg, 3 * W, 2 * H) { \ |
| 232 | ToolUtils::draw_checkerboard(canvas, SK_ColorDKGRAY, SK_ColorLTGRAY, 25); \ |
| 233 | return do_rescale_image_grid(canvas, #IMAGE_FILE, SRC_RECT, {W, H}, true, false, \ |
| 234 | errorMsg); \ |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 235 | } |
| 236 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 237 | #define DEF_RESCALE_AND_READ_YUV_SURF_GM(IMAGE_FILE, TAG, SRC_RECT, W, H) \ |
| 238 | DEF_SIMPLE_GM_CAN_FAIL(async_rescale_and_read_yuv420_##TAG, canvas, errorMsg, 3 * W, 2 * H) { \ |
| 239 | ToolUtils::draw_checkerboard(canvas, SK_ColorDKGRAY, SK_ColorLTGRAY, 25); \ |
| 240 | return do_rescale_image_grid(canvas, #IMAGE_FILE, SRC_RECT, {W, H}, true, true, errorMsg); \ |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 241 | } |
| 242 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 243 | #define DEF_RESCALE_AND_READ_IMG_GM(IMAGE_FILE, TAG, SRC_RECT, W, H) \ |
| 244 | DEF_SIMPLE_GM_CAN_FAIL(async_rescale_and_read_##TAG, canvas, errorMsg, 3 * W, 2 * H) { \ |
| 245 | ToolUtils::draw_checkerboard(canvas, SK_ColorDKGRAY, SK_ColorLTGRAY, 25); \ |
| 246 | return do_rescale_image_grid(canvas, #IMAGE_FILE, SRC_RECT, {W, H}, false, false, \ |
| 247 | errorMsg); \ |
| 248 | } |
Brian Salomon | 024bd00 | 2019-06-11 11:38:16 -0400 | [diff] [blame] | 249 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 250 | #define DEF_RESCALE_AND_READ_YUV_IMG_GM(IMAGE_FILE, TAG, SRC_RECT, W, H) \ |
| 251 | DEF_SIMPLE_GM_CAN_FAIL(async_rescale_and_read_yuv420_##TAG, canvas, errorMsg, 3 * W, 2 * H) { \ |
| 252 | ToolUtils::draw_checkerboard(canvas, SK_ColorDKGRAY, SK_ColorLTGRAY, 25); \ |
| 253 | return do_rescale_image_grid(canvas, #IMAGE_FILE, SRC_RECT, {W, H}, true, true, errorMsg); \ |
| 254 | } |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 255 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 256 | DEF_RESCALE_AND_READ_YUV_SURF_GM( |
| 257 | images/yellow_rose.webp, rose, SkIRect::MakeXYWH(50, 5, 200, 150), 410, 376) |
Brian Salomon | 201700f | 2019-05-17 12:05:44 -0400 | [diff] [blame] | 258 | |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 259 | DEF_RESCALE_AND_READ_YUV_IMG_GM( |
| 260 | images/yellow_rose.webp, rose_down, SkIRect::MakeXYWH(50, 5, 200, 150), 106, 60) |
| 261 | |
| 262 | DEF_RESCALE_AND_READ_SURF_GM( |
| 263 | images/yellow_rose.webp, rose, SkIRect::MakeXYWH(100, 20, 100, 100), 410, 410) |
| 264 | |
| 265 | DEF_RESCALE_AND_READ_SURF_GM(images/dog.jpg, dog_down, SkIRect::MakeXYWH(0, 10, 180, 150), 45, 45) |
| 266 | DEF_RESCALE_AND_READ_IMG_GM(images/dog.jpg, dog_up, SkIRect::MakeWH(180, 180), 800, 400) |
| 267 | |
| 268 | DEF_RESCALE_AND_READ_IMG_GM( |
| 269 | images/text.png, text_down, SkIRect::MakeWH(637, 105), (int)(0.7 * 637), (int)(0.7 * 105)) |
| 270 | DEF_RESCALE_AND_READ_SURF_GM( |
| 271 | images/text.png, text_up, SkIRect::MakeWH(637, 105), (int)(1.2 * 637), (int)(1.2 * 105)) |
| 272 | DEF_RESCALE_AND_READ_IMG_GM(images/text.png, |
| 273 | text_up_large, |
| 274 | SkIRect::MakeXYWH(300, 0, 300, 105), |
| 275 | (int)(2.4 * 300), |
| 276 | (int)(2.4 * 105)) |
Brian Salomon | 286b557 | 2019-05-20 10:25:50 -0400 | [diff] [blame] | 277 | |
Brian Salomon | a7e5c7c | 2020-01-27 15:41:40 -0500 | [diff] [blame] | 278 | // Exercises non-scaling YUV420. Reads from the original canvas's surface in order to |
| 279 | // exercise case where source surface is not a texture (in glbert config). |
| 280 | DEF_SIMPLE_GM_CAN_FAIL(async_yuv_no_scale, canvas, errorMsg, 400, 300) { |
| 281 | auto surface = canvas->getSurface(); |
| 282 | if (!surface) { |
| 283 | *errorMsg = "Not supported on recording/vector backends."; |
| 284 | return skiagm::DrawResult::kSkip; |
| 285 | } |
| 286 | |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 287 | auto dContext = GrAsDirectContext(surface->recordingContext()); |
Robert Phillips | 41fc174 | 2020-10-15 10:04:03 -0400 | [diff] [blame] | 288 | if (!dContext && surface->recordingContext()) { |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 289 | *errorMsg = "Not supported in DDL mode"; |
| 290 | return skiagm::DrawResult::kSkip; |
| 291 | } |
| 292 | |
Brian Salomon | a7e5c7c | 2020-01-27 15:41:40 -0500 | [diff] [blame] | 293 | auto image = GetResourceAsImage("images/yellow_rose.webp"); |
| 294 | if (!image) { |
| 295 | return skiagm::DrawResult::kFail; |
| 296 | } |
| 297 | SkPaint paint; |
| 298 | canvas->drawImage(image.get(), 0, 0); |
| 299 | |
| 300 | SkScopeExit scopeExit; |
| 301 | auto yuvImage = do_read_and_scale_yuv( |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 302 | surface, dContext, kRec601_SkYUVColorSpace, SkIRect::MakeWH(400, 300), |
Mike Reed | 1efa14d | 2021-01-02 21:44:59 -0500 | [diff] [blame] | 303 | {400, 300}, SkImage::RescaleGamma::kSrc, SkImage::RescaleMode::kNearest, &scopeExit); |
Brian Salomon | a7e5c7c | 2020-01-27 15:41:40 -0500 | [diff] [blame] | 304 | |
| 305 | canvas->clear(SK_ColorWHITE); |
| 306 | canvas->drawImage(yuvImage.get(), 0, 0); |
| 307 | |
| 308 | return skiagm::DrawResult::kOk; |
| 309 | } |
| 310 | |
Brian Salomon | 286b557 | 2019-05-20 10:25:50 -0400 | [diff] [blame] | 311 | DEF_SIMPLE_GM_CAN_FAIL(async_rescale_and_read_no_bleed, canvas, errorMsg, 60, 60) { |
| 312 | if (canvas->imageInfo().colorType() == kUnknown_SkColorType) { |
| 313 | *errorMsg = "Not supported on recording/vector backends."; |
| 314 | return skiagm::DrawResult::kSkip; |
| 315 | } |
| 316 | |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 317 | auto dContext = GrAsDirectContext(canvas->recordingContext()); |
Robert Phillips | 41fc174 | 2020-10-15 10:04:03 -0400 | [diff] [blame] | 318 | if (!dContext && canvas->recordingContext()) { |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 319 | *errorMsg = "Not supported in DDL mode"; |
| 320 | return skiagm::DrawResult::kSkip; |
| 321 | } |
| 322 | |
Brian Salomon | 286b557 | 2019-05-20 10:25:50 -0400 | [diff] [blame] | 323 | static constexpr int kBorder = 5; |
| 324 | static constexpr int kInner = 5; |
| 325 | const auto srcRect = SkIRect::MakeXYWH(kBorder, kBorder, kInner, kInner); |
| 326 | auto surfaceII = |
| 327 | SkImageInfo::Make(kInner + 2 * kBorder, kInner + 2 * kBorder, kRGBA_8888_SkColorType, |
| 328 | kPremul_SkAlphaType, SkColorSpace::MakeSRGB()); |
| 329 | auto surface = canvas->makeSurface(surfaceII); |
Brian Salomon | af9b7b9 | 2019-05-20 12:19:37 -0400 | [diff] [blame] | 330 | if (!surface) { |
| 331 | *errorMsg = "Could not create surface for image."; |
Brian Salomon | 9be911c | 2019-05-20 14:01:21 -0400 | [diff] [blame] | 332 | // When testing abandoned GrContext we expect surface creation to fail. |
Robert Phillips | d436b78 | 2020-07-10 11:49:06 -0400 | [diff] [blame] | 333 | if (canvas->recordingContext() && canvas->recordingContext()->abandoned()) { |
Brian Salomon | 9be911c | 2019-05-20 14:01:21 -0400 | [diff] [blame] | 334 | return skiagm::DrawResult::kSkip; |
| 335 | } |
Brian Salomon | af9b7b9 | 2019-05-20 12:19:37 -0400 | [diff] [blame] | 336 | return skiagm::DrawResult::kFail; |
| 337 | } |
Brian Salomon | 286b557 | 2019-05-20 10:25:50 -0400 | [diff] [blame] | 338 | surface->getCanvas()->clear(SK_ColorRED); |
| 339 | surface->getCanvas()->save(); |
| 340 | surface->getCanvas()->clipRect(SkRect::Make(srcRect), SkClipOp::kIntersect, false); |
| 341 | surface->getCanvas()->clear(SK_ColorBLUE); |
| 342 | surface->getCanvas()->restore(); |
| 343 | static constexpr int kPad = 2; |
| 344 | canvas->translate(kPad, kPad); |
| 345 | skiagm::DrawResult result; |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 346 | SkISize downSize = {static_cast<int>(kInner/2), static_cast<int>(kInner / 2)}; |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 347 | result = do_rescale_grid(canvas, surface.get(), dContext, srcRect, downSize, false, errorMsg, |
Brian Salomon | 63a0a75 | 2020-06-26 13:32:09 -0400 | [diff] [blame] | 348 | kPad); |
| 349 | |
Brian Salomon | 286b557 | 2019-05-20 10:25:50 -0400 | [diff] [blame] | 350 | if (result != skiagm::DrawResult::kOk) { |
| 351 | return result; |
| 352 | } |
Mike Klein | f68213f | 2020-01-27 12:16:22 -0600 | [diff] [blame] | 353 | canvas->translate(0, 4 * downSize.height()); |
Brian Salomon | 9241a6d | 2019-10-03 13:26:54 -0400 | [diff] [blame] | 354 | SkISize upSize = {static_cast<int>(kInner * 3.5), static_cast<int>(kInner * 4.6)}; |
Robert Phillips | 27f283f | 2020-10-14 11:46:51 -0400 | [diff] [blame] | 355 | result = do_rescale_grid(canvas, surface.get(), dContext, srcRect, upSize, false, errorMsg, |
| 356 | kPad); |
Brian Salomon | 286b557 | 2019-05-20 10:25:50 -0400 | [diff] [blame] | 357 | if (result != skiagm::DrawResult::kOk) { |
| 358 | return result; |
| 359 | } |
| 360 | return skiagm::DrawResult::kOk; |
| 361 | } |
Brian Salomon | bacbb92 | 2021-01-21 19:48:00 -0500 | [diff] [blame] | 362 | |
| 363 | DEF_SIMPLE_GM_CAN_FAIL(async_rescale_and_read_alpha_type, canvas, errorMsg, 512, 512) { |
| 364 | auto dContext = GrAsDirectContext(canvas->recordingContext()); |
| 365 | if (!dContext && canvas->recordingContext()) { |
| 366 | *errorMsg = "Not supported in DDL mode"; |
| 367 | return skiagm::DrawResult::kSkip; |
| 368 | } |
| 369 | if (dContext && dContext->abandoned()) { |
| 370 | return skiagm::DrawResult::kSkip; |
| 371 | } |
| 372 | |
| 373 | auto upmII = SkImageInfo::Make(200, 200, kRGBA_8888_SkColorType, kUnpremul_SkAlphaType); |
| 374 | |
| 375 | auto pmII = upmII.makeAlphaType(kPremul_SkAlphaType); |
| 376 | |
| 377 | auto upmSurf = SkSurface::MakeRaster(upmII); |
| 378 | auto pmSurf = SkSurface::MakeRaster( pmII); |
| 379 | |
| 380 | SkColor4f colors[] = { |
| 381 | {.3f, .3f, .3f, .3f}, |
| 382 | {1.f, .2f, .6f, .9f}, |
| 383 | {0.f, .1f, 1.f, .1f}, |
| 384 | {.7f, .8f, .2f, .7f}, |
| 385 | }; |
| 386 | auto shader = SkGradientShader::MakeRadial({100, 100}, |
| 387 | 230, |
| 388 | colors, |
| 389 | nullptr, |
| 390 | nullptr, |
| 391 | SK_ARRAY_COUNT(colors), |
| 392 | SkTileMode::kRepeat); |
| 393 | SkPaint paint; |
| 394 | paint.setShader(std::move(shader)); |
| 395 | |
| 396 | upmSurf->getCanvas()->drawPaint(paint); |
| 397 | pmSurf ->getCanvas()->drawPaint(paint); |
| 398 | |
| 399 | auto pmImg = pmSurf->makeImageSnapshot(); |
| 400 | auto upmImg = upmSurf->makeImageSnapshot(); |
| 401 | |
| 402 | if (dContext) { |
| 403 | pmImg = pmImg->makeTextureImage(dContext); |
| 404 | upmImg = upmImg->makeTextureImage(dContext); |
| 405 | if (!pmImg || !upmImg) { |
| 406 | *errorMsg = "could not make texture images"; |
| 407 | return skiagm::DrawResult::kFail; |
| 408 | } |
| 409 | } |
| 410 | int size = 256; |
| 411 | |
| 412 | ToolUtils::draw_checkerboard(canvas, SK_ColorWHITE, SK_ColorBLACK, 32); |
| 413 | |
| 414 | for (const auto& img : {pmImg, upmImg}) { |
| 415 | canvas->save(); |
| 416 | for (auto readAT : {kPremul_SkAlphaType, kUnpremul_SkAlphaType}) { |
| 417 | auto readInfo = img->imageInfo().makeAlphaType(readAT).makeWH(size, size); |
| 418 | |
| 419 | auto* asyncContext = new AsyncContext(); |
| 420 | img->asyncRescaleAndReadPixels(readInfo, |
| 421 | SkIRect::MakeSize(img->dimensions()), |
| 422 | SkImage::RescaleGamma::kSrc, |
| 423 | SkImage::RescaleMode::kRepeatedCubic, |
| 424 | async_callback, |
| 425 | asyncContext); |
| 426 | if (dContext) { |
| 427 | dContext->submit(); |
| 428 | } |
| 429 | while (!asyncContext->fCalled) { |
| 430 | // Only GPU should actually be asynchronous. |
| 431 | SkASSERT(dContext); |
| 432 | dContext->checkAsyncWorkCompletion(); |
| 433 | } |
| 434 | if (asyncContext->fResult) { |
| 435 | SkPixmap pixmap(readInfo, |
| 436 | asyncContext->fResult->data(0), |
| 437 | asyncContext->fResult->rowBytes(0)); |
| 438 | auto releasePixels = [](const void*, void* c) { |
| 439 | delete static_cast<AsyncContext*>(c); |
| 440 | }; |
| 441 | auto result = SkImage::MakeFromRaster(pixmap, releasePixels, asyncContext); |
| 442 | |
| 443 | canvas->drawImage(result, 0, 0); |
| 444 | } else { |
| 445 | delete asyncContext; |
| 446 | *errorMsg = "async readback failed"; |
| 447 | return skiagm::DrawResult::kFail; |
| 448 | } |
| 449 | canvas->translate(size, 0); |
| 450 | } |
| 451 | canvas->restore(); |
| 452 | canvas->translate(0, size); |
| 453 | } |
| 454 | return skiagm::DrawResult::kOk; |
| 455 | } |