1 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/GrTextureDomain.h" 18 #include "SkBitmap.h" 19 #include "SkGr.h" 20 #include "SkGradientShader.h" 21 22 namespace skiagm { 23 /** 24 * This GM directly exercises GrTextureDomainEffect. 25 */ 26 class TextureDomainEffect : public GM { 27 public: TextureDomainEffect()28 TextureDomainEffect() { 29 this->setBGColor(0xFFFFFFFF); 30 } 31 32 protected: onShortName()33 virtual SkString onShortName() SK_OVERRIDE { 34 return SkString("texture_domain_effect"); 35 } 36 onISize()37 virtual SkISize onISize() SK_OVERRIDE { 38 const SkScalar canvasWidth = kDrawPad + 39 (kTargetWidth + 2 * kDrawPad) * GrTextureDomain::kModeCount + 40 kTestPad * GrTextureDomain::kModeCount; 41 return SkISize::Make(SkScalarCeilToInt(canvasWidth), 800); 42 } 43 onGetFlags() const44 virtual uint32_t onGetFlags() const SK_OVERRIDE { 45 // This is a GPU-specific GM. 46 return kGPUOnly_Flag; 47 } 48 onOnceBeforeDraw()49 virtual void onOnceBeforeDraw() SK_OVERRIDE { 50 fBmp.allocN32Pixels(kTargetWidth, kTargetHeight); 51 SkCanvas canvas(fBmp); 52 canvas.clear(0x00000000); 53 SkPaint paint; 54 55 SkColor colors1[] = { SK_ColorCYAN, SK_ColorLTGRAY, SK_ColorGRAY }; 56 paint.setShader(SkGradientShader::CreateSweep(65.f, 75.f, colors1, 57 NULL, SK_ARRAY_COUNT(colors1)))->unref(); 58 canvas.drawOval(SkRect::MakeXYWH(-5.f, -5.f, 59 fBmp.width() + 10.f, fBmp.height() + 10.f), paint); 60 61 SkColor colors2[] = { SK_ColorMAGENTA, SK_ColorLTGRAY, SK_ColorYELLOW }; 62 paint.setShader(SkGradientShader::CreateSweep(45.f, 55.f, colors2, NULL, 63 SK_ARRAY_COUNT(colors2)))->unref(); 64 paint.setXfermodeMode(SkXfermode::kDarken_Mode); 65 canvas.drawOval(SkRect::MakeXYWH(-5.f, -5.f, 66 fBmp.width() + 10.f, fBmp.height() + 10.f), paint); 67 68 SkColor colors3[] = { SK_ColorBLUE, SK_ColorLTGRAY, SK_ColorGREEN }; 69 paint.setShader(SkGradientShader::CreateSweep(25.f, 35.f, colors3, NULL, 70 SK_ARRAY_COUNT(colors3)))->unref(); 71 paint.setXfermodeMode(SkXfermode::kLighten_Mode); 72 canvas.drawOval(SkRect::MakeXYWH(-5.f, -5.f, 73 fBmp.width() + 10.f, fBmp.height() + 10.f), paint); 74 } 75 onDraw(SkCanvas * canvas)76 virtual void onDraw(SkCanvas* canvas) SK_OVERRIDE { 77 GrRenderTarget* rt = canvas->internal_private_accessTopLayerRenderTarget(); 78 if (NULL == rt) { 79 return; 80 } 81 GrContext* context = rt->getContext(); 82 if (NULL == context) { 83 return; 84 } 85 86 GrTestTarget tt; 87 context->getTestTarget(&tt); 88 if (NULL == tt.target()) { 89 SkDEBUGFAIL("Couldn't get Gr test target."); 90 return; 91 } 92 93 GrDrawState* drawState = tt.target()->drawState(); 94 95 GrTexture* texture = GrLockAndRefCachedBitmapTexture(context, fBmp, NULL); 96 if (NULL == texture) { 97 return; 98 } 99 100 SkTArray<SkMatrix> textureMatrices; 101 textureMatrices.push_back().setIDiv(texture->width(), texture->height()); 102 textureMatrices.push_back() = textureMatrices[0]; 103 textureMatrices.back().postScale(1.5f, 0.85f); 104 textureMatrices.push_back() = textureMatrices[0]; 105 textureMatrices.back().preRotate(45.f, texture->width() / 2.f, texture->height() / 2.f); 106 107 const SkIRect texelDomains[] = { 108 SkIRect::MakeWH(fBmp.width(), fBmp.height()), 109 SkIRect::MakeXYWH(fBmp.width() / 4, 110 fBmp.height() / 4, 111 fBmp.width() / 2, 112 fBmp.height() / 2), 113 }; 114 115 SkRect renderRect = SkRect::MakeWH(SkIntToScalar(fBmp.width()), 116 SkIntToScalar(fBmp.height())); 117 renderRect.outset(kDrawPad, kDrawPad); 118 119 SkScalar y = kDrawPad + kTestPad; 120 for (int tm = 0; tm < textureMatrices.count(); ++tm) { 121 for (size_t d = 0; d < SK_ARRAY_COUNT(texelDomains); ++d) { 122 SkScalar x = kDrawPad + kTestPad; 123 for (int m = 0; m < GrTextureDomain::kModeCount; ++m) { 124 GrTextureDomain::Mode mode = (GrTextureDomain::Mode) m; 125 SkAutoTUnref<GrFragmentProcessor> fp( 126 GrTextureDomainEffect::Create(texture, textureMatrices[tm], 127 GrTextureDomain::MakeTexelDomain(texture, 128 texelDomains[d]), 129 mode, GrTextureParams::kNone_FilterMode)); 130 131 if (!fp) { 132 continue; 133 } 134 SkMatrix viewMatrix; 135 viewMatrix.setTranslate(x, y); 136 drawState->reset(viewMatrix); 137 drawState->setRenderTarget(rt); 138 drawState->setColor(0xffffffff); 139 drawState->addColorProcessor(fp); 140 141 tt.target()->drawSimpleRect(renderRect); 142 x += renderRect.width() + kTestPad; 143 } 144 y += renderRect.height() + kTestPad; 145 } 146 } 147 GrUnlockAndUnrefCachedBitmapTexture(texture); 148 } 149 150 private: 151 static const SkScalar kDrawPad; 152 static const SkScalar kTestPad; 153 static const int kTargetWidth = 100; 154 static const int kTargetHeight = 100; 155 SkBitmap fBmp; 156 157 typedef GM INHERITED; 158 }; 159 160 // Windows builds did not like SkScalar initialization in class :( 161 const SkScalar TextureDomainEffect::kDrawPad = 10.f; 162 const SkScalar TextureDomainEffect::kTestPad = 10.f; 163 164 DEF_GM( return SkNEW(TextureDomainEffect); ) 165 } 166 167 #endif 168