blob: 12676f262e2477f72cd3a19487ea2b7400e3a0fd [file] [log] [blame]
sugoi24dcac22014-07-07 15:09:48 -07001
2/*
3 * Copyright 2014 Google Inc.
4 *
5 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
7 */
8
9// This test only works with the GPU backend.
10
11#include "gm.h"
12
13#if SK_SUPPORT_GPU
14
15#include "GrContext.h"
16#include "GrTest.h"
17#include "effects/GrYUVtoRGBEffect.h"
18#include "SkBitmap.h"
19#include "SkGr.h"
20#include "SkGradientShader.h"
21
22namespace skiagm {
23/**
24 * This GM directly exercises GrYUVtoRGBEffect.
25 */
26class YUVtoRGBEffect : public GM {
27public:
28 YUVtoRGBEffect() {
29 this->setBGColor(0xFFFFFFFF);
30 }
31
32protected:
33 virtual SkString onShortName() SK_OVERRIDE {
34 return SkString("yuv_to_rgb_effect");
35 }
36
37 virtual SkISize onISize() SK_OVERRIDE {
38 return SkISize::Make(334, 64);
39 }
40
41 virtual uint32_t onGetFlags() const SK_OVERRIDE {
42 // This is a GPU-specific GM.
43 return kGPUOnly_Flag;
44 }
45
46 virtual void onOnceBeforeDraw() SK_OVERRIDE {
47 SkImageInfo info = SkImageInfo::MakeA8(24, 24);
48 fBmp[0].allocPixels(info);
49 fBmp[1].allocPixels(info);
50 fBmp[2].allocPixels(info);
51 unsigned char* pixels[3];
52 for (int i = 0; i < 3; ++i) {
53 pixels[i] = (unsigned char*)fBmp[i].getPixels();
54 }
55 int color[] = {0, 85, 170};
56 const int limit[] = {255, 0, 255};
57 const int invl[] = {0, 255, 0};
58 const int inc[] = {1, -1, 1};
59 for (int j = 0; j < 576; ++j) {
60 for (int i = 0; i < 3; ++i) {
61 pixels[i][j] = (unsigned char)color[i];
62 color[i] = (color[i] == limit[i]) ? invl[i] : color[i] + inc[i];
63 }
64 }
65 }
66
67 virtual void onDraw(SkCanvas* canvas) SK_OVERRIDE {
68 GrRenderTarget* rt = canvas->internal_private_accessTopLayerRenderTarget();
69 if (NULL == rt) {
70 return;
71 }
72 GrContext* context = rt->getContext();
73 if (NULL == context) {
74 return;
75 }
76
77 GrTestTarget tt;
78 context->getTestTarget(&tt);
79 if (NULL == tt.target()) {
80 SkDEBUGFAIL("Couldn't get Gr test target.");
81 return;
82 }
83
84 GrDrawState* drawState = tt.target()->drawState();
85
86 GrTexture* texture[3];
87 texture[0] = GrLockAndRefCachedBitmapTexture(context, fBmp[0], NULL);
88 texture[1] = GrLockAndRefCachedBitmapTexture(context, fBmp[1], NULL);
89 texture[2] = GrLockAndRefCachedBitmapTexture(context, fBmp[2], NULL);
90 if ((NULL == texture[0]) || (NULL == texture[1]) || (NULL == texture[2])) {
91 return;
92 }
93
94 static const SkScalar kDrawPad = 10.f;
95 static const SkScalar kTestPad = 10.f;
96
97 SkRect renderRect = SkRect::MakeWH(SkIntToScalar(fBmp[0].width()),
98 SkIntToScalar(fBmp[0].height()));
99 renderRect.outset(kDrawPad, kDrawPad);
100
101 SkScalar y = kDrawPad + kTestPad;
102 SkScalar x = kDrawPad + kTestPad;
103
104 const int indices[6][3] = {{0, 1, 2}, {0, 2, 1}, {1, 0, 2}, {1, 2, 0}, {2, 0, 1}, {2, 1, 0}};
105
106 for (int i = 0; i < 6; ++i) {
bsalomon83d081a2014-07-08 09:56:10 -0700107 SkAutoTUnref<GrEffect> effect(
sugoi24dcac22014-07-07 15:09:48 -0700108 GrYUVtoRGBEffect::Create(texture[indices[i][0]],
109 texture[indices[i][1]],
110 texture[indices[i][2]]));
111 if (effect) {
112 SkMatrix viewMatrix;
113 viewMatrix.setTranslate(x, y);
114 drawState->reset(viewMatrix);
115 drawState->setRenderTarget(rt);
116 drawState->setColor(0xffffffff);
117 drawState->addColorEffect(effect, 1);
118 tt.target()->drawSimpleRect(renderRect);
119 }
120 x += renderRect.width() + kTestPad;
121 }
122
123 GrUnlockAndUnrefCachedBitmapTexture(texture[0]);
124 GrUnlockAndUnrefCachedBitmapTexture(texture[1]);
125 GrUnlockAndUnrefCachedBitmapTexture(texture[2]);
126 }
127
128private:
129 SkBitmap fBmp[3];
130
131 typedef GM INHERITED;
132};
133
134DEF_GM( return SkNEW(YUVtoRGBEffect); )
135}
136
137#endif