sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2014 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 | // This test only works with the GPU backend. |
| 9 | |
| 10 | #include "gm.h" |
| 11 | |
| 12 | #if SK_SUPPORT_GPU |
| 13 | |
| 14 | #include "GrContext.h" |
Brian Osman | 1105224 | 2016-10-27 14:47:55 -0400 | [diff] [blame] | 15 | #include "GrRenderTargetContextPriv.h" |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 16 | #include "GrTextureProxy.h" |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 17 | #include "SkBitmap.h" |
| 18 | #include "SkGr.h" |
| 19 | #include "SkGradientShader.h" |
bsalomon | f267c1e | 2016-02-01 13:16:14 -0800 | [diff] [blame] | 20 | #include "effects/GrYUVEffect.h" |
Brian Salomon | 8952743 | 2016-12-16 09:52:16 -0500 | [diff] [blame] | 21 | #include "ops/GrDrawOp.h" |
Brian Salomon | baaf439 | 2017-06-15 09:59:23 -0400 | [diff] [blame] | 22 | #include "ops/GrRectOpFactory.h" |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 23 | |
sugoi | 4ccce7e | 2015-02-13 13:57:09 -0800 | [diff] [blame] | 24 | #define YSIZE 8 |
| 25 | #define USIZE 4 |
| 26 | #define VSIZE 4 |
| 27 | |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 28 | namespace skiagm { |
| 29 | /** |
| 30 | * This GM directly exercises GrYUVtoRGBEffect. |
| 31 | */ |
| 32 | class YUVtoRGBEffect : public GM { |
| 33 | public: |
| 34 | YUVtoRGBEffect() { |
| 35 | this->setBGColor(0xFFFFFFFF); |
| 36 | } |
| 37 | |
| 38 | protected: |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 39 | SkString onShortName() override { |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 40 | return SkString("yuv_to_rgb_effect"); |
| 41 | } |
| 42 | |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 43 | SkISize onISize() override { |
rileya | 1340039 | 2015-07-20 15:00:03 -0700 | [diff] [blame] | 44 | return SkISize::Make(238, 120); |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 45 | } |
| 46 | |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 47 | void onOnceBeforeDraw() override { |
sugoi | 4ccce7e | 2015-02-13 13:57:09 -0800 | [diff] [blame] | 48 | SkImageInfo yinfo = SkImageInfo::MakeA8(YSIZE, YSIZE); |
| 49 | fBmp[0].allocPixels(yinfo); |
| 50 | SkImageInfo uinfo = SkImageInfo::MakeA8(USIZE, USIZE); |
| 51 | fBmp[1].allocPixels(uinfo); |
| 52 | SkImageInfo vinfo = SkImageInfo::MakeA8(VSIZE, VSIZE); |
| 53 | fBmp[2].allocPixels(vinfo); |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 54 | unsigned char* pixels[3]; |
| 55 | for (int i = 0; i < 3; ++i) { |
| 56 | pixels[i] = (unsigned char*)fBmp[i].getPixels(); |
| 57 | } |
| 58 | int color[] = {0, 85, 170}; |
| 59 | const int limit[] = {255, 0, 255}; |
| 60 | const int invl[] = {0, 255, 0}; |
| 61 | const int inc[] = {1, -1, 1}; |
sugoi | 4ccce7e | 2015-02-13 13:57:09 -0800 | [diff] [blame] | 62 | for (int i = 0; i < 3; ++i) { |
| 63 | const size_t nbBytes = fBmp[i].rowBytes() * fBmp[i].height(); |
| 64 | for (size_t j = 0; j < nbBytes; ++j) { |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 65 | pixels[i][j] = (unsigned char)color[i]; |
| 66 | color[i] = (color[i] == limit[i]) ? invl[i] : color[i] + inc[i]; |
| 67 | } |
| 68 | } |
| 69 | } |
| 70 | |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 71 | void onDraw(SkCanvas* canvas) override { |
Brian Osman | 1105224 | 2016-10-27 14:47:55 -0400 | [diff] [blame] | 72 | GrRenderTargetContext* renderTargetContext = |
| 73 | canvas->internal_private_accessTopLayerRenderTargetContext(); |
| 74 | if (!renderTargetContext) { |
halcanary | 2a24338 | 2015-09-09 08:16:41 -0700 | [diff] [blame] | 75 | skiagm::GM::DrawGpuOnlyMessage(canvas); |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 76 | return; |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 77 | } |
| 78 | |
robertphillips | 175dd9b | 2016-04-28 14:32:04 -0700 | [diff] [blame] | 79 | GrContext* context = canvas->getGrContext(); |
| 80 | if (!context) { |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 81 | return; |
| 82 | } |
| 83 | |
Robert Phillips | 2f49314 | 2017-03-02 18:18:38 -0500 | [diff] [blame] | 84 | sk_sp<GrTextureProxy> proxy[3]; |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 85 | |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 86 | { |
| 87 | GrSurfaceDesc desc; |
| 88 | desc.fOrigin = kTopLeft_GrSurfaceOrigin; |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 89 | |
| 90 | for (int i = 0; i < 3; ++i) { |
| 91 | desc.fWidth = fBmp[i].width(); |
| 92 | desc.fHeight = fBmp[i].height(); |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 93 | desc.fConfig = SkImageInfo2GrPixelConfig(fBmp[i].info(), *context->caps()); |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 94 | |
Robert Phillips | 26c90e0 | 2017-03-14 14:39:29 -0400 | [diff] [blame] | 95 | proxy[i] = GrSurfaceProxy::MakeDeferred(context->resourceProvider(), |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 96 | desc, SkBudgeted::kYes, |
| 97 | fBmp[i].getPixels(), fBmp[i].rowBytes()); |
| 98 | if (!proxy[i]) { |
| 99 | return; |
| 100 | } |
| 101 | } |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 102 | } |
| 103 | |
mtklein | dbfd7ab | 2016-09-01 11:24:54 -0700 | [diff] [blame] | 104 | constexpr SkScalar kDrawPad = 10.f; |
| 105 | constexpr SkScalar kTestPad = 10.f; |
| 106 | constexpr SkScalar kColorSpaceOffset = 36.f; |
sugoi | 4ccce7e | 2015-02-13 13:57:09 -0800 | [diff] [blame] | 107 | SkISize sizes[3] = {{YSIZE, YSIZE}, {USIZE, USIZE}, {VSIZE, VSIZE}}; |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 108 | |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 109 | for (int space = kJPEG_SkYUVColorSpace; space <= kLastEnum_SkYUVColorSpace; ++space) { |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 110 | SkRect renderRect = SkRect::MakeWH(SkIntToScalar(fBmp[0].width()), |
| 111 | SkIntToScalar(fBmp[0].height())); |
| 112 | renderRect.outset(kDrawPad, kDrawPad); |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 113 | |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 114 | SkScalar y = kDrawPad + kTestPad + space * kColorSpaceOffset; |
| 115 | SkScalar x = kDrawPad + kTestPad; |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 116 | |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 117 | const int indices[6][3] = {{0, 1, 2}, {0, 2, 1}, {1, 0, 2}, |
| 118 | {1, 2, 0}, {2, 0, 1}, {2, 1, 0}}; |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 119 | |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 120 | for (int i = 0; i < 6; ++i) { |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 121 | std::unique_ptr<GrFragmentProcessor> fp( |
Robert Phillips | fbcef6e | 2017-06-15 12:07:18 -0400 | [diff] [blame] | 122 | GrYUVEffect::MakeYUVToRGB(proxy[indices[i][0]], |
Robert Phillips | 2f49314 | 2017-03-02 18:18:38 -0500 | [diff] [blame] | 123 | proxy[indices[i][1]], |
| 124 | proxy[indices[i][2]], |
Hal Canary | cefc431 | 2016-11-04 16:26:16 -0400 | [diff] [blame] | 125 | sizes, |
| 126 | static_cast<SkYUVColorSpace>(space), |
| 127 | false)); |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 128 | if (fp) { |
Brian Salomon | 82f4431 | 2017-01-11 13:42:54 -0500 | [diff] [blame] | 129 | GrPaint grPaint; |
| 130 | grPaint.setXPFactory(GrPorterDuffXPFactory::Get(SkBlendMode::kSrc)); |
| 131 | grPaint.addColorFragmentProcessor(std::move(fp)); |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 132 | SkMatrix viewMatrix; |
| 133 | viewMatrix.setTranslate(x, y); |
Brian Salomon | ac70f84 | 2017-05-08 10:43:33 -0400 | [diff] [blame] | 134 | renderTargetContext->priv().testingOnly_addDrawOp( |
Brian Salomon | baaf439 | 2017-06-15 09:59:23 -0400 | [diff] [blame] | 135 | GrRectOpFactory::MakeNonAAFill(std::move(grPaint), viewMatrix, |
| 136 | renderRect, GrAAType::kNone)); |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 137 | } |
| 138 | x += renderRect.width() + kTestPad; |
| 139 | } |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 140 | } |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 141 | } |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 142 | |
| 143 | private: |
| 144 | SkBitmap fBmp[3]; |
| 145 | |
| 146 | typedef GM INHERITED; |
| 147 | }; |
| 148 | |
halcanary | 385fe4d | 2015-08-26 13:07:48 -0700 | [diff] [blame] | 149 | DEF_GM(return new YUVtoRGBEffect;) |
jbauman | b445a57 | 2016-06-09 13:24:48 -0700 | [diff] [blame] | 150 | |
| 151 | ////////////////////////////////////////////////////////////////////////////// |
| 152 | |
| 153 | class YUVNV12toRGBEffect : public GM { |
| 154 | public: |
| 155 | YUVNV12toRGBEffect() { |
| 156 | this->setBGColor(0xFFFFFFFF); |
| 157 | } |
| 158 | |
| 159 | protected: |
| 160 | SkString onShortName() override { |
| 161 | return SkString("yuv_nv12_to_rgb_effect"); |
| 162 | } |
| 163 | |
| 164 | SkISize onISize() override { |
| 165 | return SkISize::Make(48, 120); |
| 166 | } |
| 167 | |
| 168 | void onOnceBeforeDraw() override { |
| 169 | SkImageInfo yinfo = SkImageInfo::MakeA8(YSIZE, YSIZE); |
| 170 | fBmp[0].allocPixels(yinfo); |
| 171 | SkImageInfo uvinfo = SkImageInfo::MakeN32Premul(USIZE, USIZE); |
| 172 | fBmp[1].allocPixels(uvinfo); |
| 173 | int color[] = {0, 85, 170}; |
| 174 | const int limit[] = {255, 0, 255}; |
| 175 | const int invl[] = {0, 255, 0}; |
| 176 | const int inc[] = {1, -1, 1}; |
| 177 | |
| 178 | { |
| 179 | unsigned char* pixels = (unsigned char*)fBmp[0].getPixels(); |
| 180 | const size_t nbBytes = fBmp[0].rowBytes() * fBmp[0].height(); |
| 181 | for (size_t j = 0; j < nbBytes; ++j) { |
| 182 | pixels[j] = (unsigned char)color[0]; |
| 183 | color[0] = (color[0] == limit[0]) ? invl[0] : color[0] + inc[0]; |
| 184 | } |
| 185 | } |
| 186 | |
| 187 | { |
| 188 | for (int y = 0; y < fBmp[1].height(); ++y) { |
| 189 | uint32_t* pixels = fBmp[1].getAddr32(0, y); |
| 190 | for (int j = 0; j < fBmp[1].width(); ++j) { |
| 191 | pixels[j] = SkColorSetARGB(0, color[1], color[2], 0); |
| 192 | color[1] = (color[1] == limit[1]) ? invl[1] : color[1] + inc[1]; |
| 193 | color[2] = (color[2] == limit[2]) ? invl[2] : color[2] + inc[2]; |
| 194 | } |
| 195 | } |
| 196 | } |
| 197 | } |
| 198 | |
| 199 | void onDraw(SkCanvas* canvas) override { |
Brian Osman | 1105224 | 2016-10-27 14:47:55 -0400 | [diff] [blame] | 200 | GrRenderTargetContext* renderTargetContext = |
| 201 | canvas->internal_private_accessTopLayerRenderTargetContext(); |
| 202 | if (!renderTargetContext) { |
jbauman | b445a57 | 2016-06-09 13:24:48 -0700 | [diff] [blame] | 203 | skiagm::GM::DrawGpuOnlyMessage(canvas); |
| 204 | return; |
| 205 | } |
| 206 | |
| 207 | GrContext* context = canvas->getGrContext(); |
| 208 | if (!context) { |
| 209 | return; |
| 210 | } |
| 211 | |
Robert Phillips | 2f49314 | 2017-03-02 18:18:38 -0500 | [diff] [blame] | 212 | sk_sp<GrTextureProxy> proxy[3]; |
jbauman | b445a57 | 2016-06-09 13:24:48 -0700 | [diff] [blame] | 213 | |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 214 | { |
| 215 | GrSurfaceDesc desc; |
| 216 | desc.fOrigin = kTopLeft_GrSurfaceOrigin; |
| 217 | |
| 218 | for (int i = 0; i < 3; ++i) { |
| 219 | int index = (0 == i) ? 0 : 1; |
| 220 | |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 221 | desc.fWidth = fBmp[index].width(); |
| 222 | desc.fHeight = fBmp[index].height(); |
Robert Phillips | 40fd7c9 | 2017-01-30 08:06:27 -0500 | [diff] [blame] | 223 | desc.fConfig = SkImageInfo2GrPixelConfig(fBmp[index].info(), *context->caps()); |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 224 | |
Robert Phillips | 26c90e0 | 2017-03-14 14:39:29 -0400 | [diff] [blame] | 225 | proxy[i] = GrSurfaceProxy::MakeDeferred(context->resourceProvider(), |
Robert Phillips | bc7a4fb | 2017-01-23 15:30:35 -0500 | [diff] [blame] | 226 | desc, SkBudgeted::kYes, |
| 227 | fBmp[index].getPixels(), |
| 228 | fBmp[index].rowBytes()); |
| 229 | if (!proxy[i]) { |
| 230 | return; |
| 231 | } |
| 232 | } |
jbauman | b445a57 | 2016-06-09 13:24:48 -0700 | [diff] [blame] | 233 | } |
| 234 | |
mtklein | dbfd7ab | 2016-09-01 11:24:54 -0700 | [diff] [blame] | 235 | constexpr SkScalar kDrawPad = 10.f; |
| 236 | constexpr SkScalar kTestPad = 10.f; |
| 237 | constexpr SkScalar kColorSpaceOffset = 36.f; |
jbauman | b445a57 | 2016-06-09 13:24:48 -0700 | [diff] [blame] | 238 | SkISize sizes[3] = {{YSIZE, YSIZE}, {USIZE, USIZE}, {VSIZE, VSIZE}}; |
| 239 | |
| 240 | for (int space = kJPEG_SkYUVColorSpace; space <= kLastEnum_SkYUVColorSpace; ++space) { |
| 241 | SkRect renderRect = |
| 242 | SkRect::MakeWH(SkIntToScalar(fBmp[0].width()), SkIntToScalar(fBmp[0].height())); |
| 243 | renderRect.outset(kDrawPad, kDrawPad); |
| 244 | |
| 245 | SkScalar y = kDrawPad + kTestPad + space * kColorSpaceOffset; |
| 246 | SkScalar x = kDrawPad + kTestPad; |
| 247 | |
robertphillips | 28a838e | 2016-06-23 14:07:00 -0700 | [diff] [blame] | 248 | GrPaint grPaint; |
Brian Salomon | a163392 | 2017-01-09 11:46:10 -0500 | [diff] [blame] | 249 | grPaint.setXPFactory(GrPorterDuffXPFactory::Get(SkBlendMode::kSrc)); |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 250 | auto fp = GrYUVEffect::MakeYUVToRGB(proxy[0], proxy[1], proxy[2], sizes, |
| 251 | static_cast<SkYUVColorSpace>(space), true); |
jbauman | b445a57 | 2016-06-09 13:24:48 -0700 | [diff] [blame] | 252 | if (fp) { |
| 253 | SkMatrix viewMatrix; |
| 254 | viewMatrix.setTranslate(x, y); |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 255 | grPaint.addColorFragmentProcessor(std::move(fp)); |
Brian Salomon | baaf439 | 2017-06-15 09:59:23 -0400 | [diff] [blame] | 256 | renderTargetContext->priv().testingOnly_addDrawOp(GrRectOpFactory::MakeNonAAFill( |
| 257 | std::move(grPaint), viewMatrix, renderRect, GrAAType::kNone)); |
jbauman | b445a57 | 2016-06-09 13:24:48 -0700 | [diff] [blame] | 258 | } |
| 259 | } |
| 260 | } |
| 261 | |
| 262 | private: |
| 263 | SkBitmap fBmp[2]; |
| 264 | |
| 265 | typedef GM INHERITED; |
| 266 | }; |
| 267 | |
| 268 | DEF_GM(return new YUVNV12toRGBEffect;) |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 269 | } |
| 270 | |
| 271 | #endif |