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" |
robertphillips | 391395d | 2016-03-02 09:26:36 -0800 | [diff] [blame] | 15 | #include "GrDrawContextPriv.h" |
joshualitt | 04194f3 | 2016-01-13 10:08:27 -0800 | [diff] [blame] | 16 | #include "GrPipelineBuilder.h" |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 17 | #include "SkBitmap.h" |
| 18 | #include "SkGr.h" |
| 19 | #include "SkGradientShader.h" |
joshualitt | 04194f3 | 2016-01-13 10:08:27 -0800 | [diff] [blame] | 20 | #include "batches/GrDrawBatch.h" |
| 21 | #include "batches/GrRectBatchFactory.h" |
bsalomon | f267c1e | 2016-02-01 13:16:14 -0800 | [diff] [blame] | 22 | #include "effects/GrYUVEffect.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 { |
robertphillips | 175dd9b | 2016-04-28 14:32:04 -0700 | [diff] [blame] | 72 | GrDrawContext* drawContext = canvas->internal_private_accessTopLayerDrawContext(); |
| 73 | if (!drawContext) { |
halcanary | 2a24338 | 2015-09-09 08:16:41 -0700 | [diff] [blame] | 74 | skiagm::GM::DrawGpuOnlyMessage(canvas); |
robertphillips | 175dd9b | 2016-04-28 14:32:04 -0700 | [diff] [blame] | 75 | return; |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 76 | } |
| 77 | |
robertphillips | 175dd9b | 2016-04-28 14:32:04 -0700 | [diff] [blame] | 78 | GrContext* context = canvas->getGrContext(); |
| 79 | if (!context) { |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 80 | return; |
| 81 | } |
| 82 | |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 83 | SkAutoTUnref<GrTexture> texture[3]; |
bsalomon | afa95e2 | 2015-10-12 10:39:46 -0700 | [diff] [blame] | 84 | texture[0].reset(GrRefCachedBitmapTexture(context, fBmp[0], |
| 85 | GrTextureParams::ClampBilerp())); |
| 86 | texture[1].reset(GrRefCachedBitmapTexture(context, fBmp[1], |
| 87 | GrTextureParams::ClampBilerp())); |
| 88 | texture[2].reset(GrRefCachedBitmapTexture(context, fBmp[2], |
| 89 | GrTextureParams::ClampBilerp())); |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 90 | |
| 91 | if (!texture[0] || !texture[1] || !texture[2]) { |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 92 | return; |
| 93 | } |
| 94 | |
| 95 | static const SkScalar kDrawPad = 10.f; |
| 96 | static const SkScalar kTestPad = 10.f; |
sugoi | 4ccce7e | 2015-02-13 13:57:09 -0800 | [diff] [blame] | 97 | static const SkScalar kColorSpaceOffset = 36.f; |
| 98 | SkISize sizes[3] = {{YSIZE, YSIZE}, {USIZE, USIZE}, {VSIZE, VSIZE}}; |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 99 | |
rileya | abaef86 | 2014-09-12 17:45:58 -0700 | [diff] [blame] | 100 | for (int space = kJPEG_SkYUVColorSpace; space <= kLastEnum_SkYUVColorSpace; |
| 101 | ++space) { |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 102 | SkRect renderRect = SkRect::MakeWH(SkIntToScalar(fBmp[0].width()), |
| 103 | SkIntToScalar(fBmp[0].height())); |
| 104 | renderRect.outset(kDrawPad, kDrawPad); |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 105 | |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 106 | SkScalar y = kDrawPad + kTestPad + space * kColorSpaceOffset; |
| 107 | SkScalar x = kDrawPad + kTestPad; |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 108 | |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 109 | const int indices[6][3] = {{0, 1, 2}, {0, 2, 1}, {1, 0, 2}, |
| 110 | {1, 2, 0}, {2, 0, 1}, {2, 1, 0}}; |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 111 | |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 112 | for (int i = 0; i < 6; ++i) { |
joshualitt | 2cdec31 | 2015-07-09 07:31:31 -0700 | [diff] [blame] | 113 | GrPipelineBuilder pipelineBuilder; |
egdaniel | c4b7272 | 2015-11-23 13:20:41 -0800 | [diff] [blame] | 114 | pipelineBuilder.setXPFactory( |
| 115 | GrPorterDuffXPFactory::Create(SkXfermode::kSrc_Mode))->unref(); |
bsalomon | f267c1e | 2016-02-01 13:16:14 -0800 | [diff] [blame] | 116 | SkAutoTUnref<const GrFragmentProcessor> fp( |
| 117 | GrYUVEffect::CreateYUVToRGB(texture[indices[i][0]], |
| 118 | texture[indices[i][1]], |
| 119 | texture[indices[i][2]], |
| 120 | sizes, |
| 121 | static_cast<SkYUVColorSpace>(space))); |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 122 | if (fp) { |
| 123 | SkMatrix viewMatrix; |
| 124 | viewMatrix.setTranslate(x, y); |
robertphillips | 175dd9b | 2016-04-28 14:32:04 -0700 | [diff] [blame] | 125 | pipelineBuilder.setRenderTarget(drawContext->accessRenderTarget()); |
bsalomon | ac856c9 | 2015-08-27 06:30:17 -0700 | [diff] [blame] | 126 | pipelineBuilder.addColorFragmentProcessor(fp); |
joshualitt | 04194f3 | 2016-01-13 10:08:27 -0800 | [diff] [blame] | 127 | SkAutoTUnref<GrDrawBatch> batch( |
| 128 | GrRectBatchFactory::CreateNonAAFill(GrColor_WHITE, viewMatrix, |
| 129 | renderRect, nullptr, nullptr)); |
robertphillips | 391395d | 2016-03-02 09:26:36 -0800 | [diff] [blame] | 130 | drawContext->drawContextPriv().testingOnly_drawBatch(pipelineBuilder, batch); |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 131 | } |
| 132 | x += renderRect.width() + kTestPad; |
| 133 | } |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 134 | } |
bsalomon | bcf0a52 | 2014-10-08 08:40:09 -0700 | [diff] [blame] | 135 | } |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 136 | |
| 137 | private: |
| 138 | SkBitmap fBmp[3]; |
| 139 | |
| 140 | typedef GM INHERITED; |
| 141 | }; |
| 142 | |
halcanary | 385fe4d | 2015-08-26 13:07:48 -0700 | [diff] [blame] | 143 | DEF_GM(return new YUVtoRGBEffect;) |
sugoi | 24dcac2 | 2014-07-07 15:09:48 -0700 | [diff] [blame] | 144 | } |
| 145 | |
| 146 | #endif |