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/SkImageInfo.h"
12 #include "include/core/SkMatrix.h"
13 #include "include/core/SkPaint.h"
14 #include "include/core/SkRect.h"
15 #include "include/core/SkShader.h"
16 #include "include/core/SkSize.h"
17 #include "include/core/SkString.h"
18 #include "include/core/SkTileMode.h"
19 #include "include/core/SkTypes.h"
20 #include "include/gpu/GrContext.h"
21 #include "src/gpu/GrBitmapTextureMaker.h"
22 #include "src/gpu/GrCaps.h"
23 #include "src/gpu/GrContextPriv.h"
24 #include "src/gpu/GrCoordTransform.h"
25 #include "src/gpu/GrFragmentProcessor.h"
26 #include "src/gpu/GrRenderTargetContext.h"
27 #include "src/gpu/GrRenderTargetContextPriv.h"
28 #include "src/gpu/effects/GrRRectEffect.h"
29 #include "src/gpu/effects/GrSkSLFP.h"
30 #include "src/gpu/glsl/GrGLSLFragmentShaderBuilder.h"
31 #include "src/gpu/ops/GrFillRectOp.h"
32 #include "tools/Resources.h"
33 #include "tools/ToolUtils.h"
34
35 class SampleCoordEffect : public GrFragmentProcessor {
36 public:
37 static constexpr GrProcessor::ClassID CLASS_ID = (GrProcessor::ClassID) 0;
38
SampleCoordEffect(std::unique_ptr<GrFragmentProcessor> child)39 SampleCoordEffect(std::unique_ptr<GrFragmentProcessor> child)
40 : INHERITED(CLASS_ID, kNone_OptimizationFlags) {
41 child->setSampledWithExplicitCoords(true);
42 this->registerChildProcessor(std::move(child));
43 }
44
name() const45 const char* name() const override { return "SampleCoordEffect"; }
46
clone() const47 std::unique_ptr<GrFragmentProcessor> clone() const override {
48 SkASSERT(false);
49 return nullptr;
50 }
51
onGetGLSLProcessorKey(const GrShaderCaps &,GrProcessorKeyBuilder *) const52 void onGetGLSLProcessorKey(const GrShaderCaps&, GrProcessorKeyBuilder*) const override {
53 }
54
onIsEqual(const GrFragmentProcessor &) const55 bool onIsEqual(const GrFragmentProcessor&) const override {
56 SkASSERT(false);
57 return true;
58 }
59
60 private:
61 GrGLSLFragmentProcessor* onCreateGLSLInstance() const override;
62 typedef GrFragmentProcessor INHERITED;
63 };
64
65 class GLSLSampleCoordEffect : public GrGLSLFragmentProcessor {
emitCode(EmitArgs & args)66 void emitCode(EmitArgs& args) override {
67 GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
68 SkString sample1 = this->invokeChild(0, args, "float2(sk_FragCoord.x, sk_FragCoord.y)");
69 SkString sample2 = this->invokeChild(0, args, "float2(sk_FragCoord.x, 512-sk_FragCoord.y)");
70 fragBuilder->codeAppendf("%s = (%s + %s) / 2;\n", args.fOutputColor, sample1.c_str(),
71 sample2.c_str());
72 }
73 };
74
onCreateGLSLInstance() const75 GrGLSLFragmentProcessor* SampleCoordEffect::onCreateGLSLInstance() const {
76 return new GLSLSampleCoordEffect();
77 }
78
79 DEF_SIMPLE_GPU_GM_BG(fpcoordinateoverride, ctx, rtCtx, canvas, 512, 512,
80 ToolUtils::color_to_565(0xFF66AA99)) {
81 SkRect bounds = SkRect::MakeIWH(512, 512);
82
83 SkBitmap bmp;
84 GetResourceAsBitmap("images/mandrill_512_q075.jpg", &bmp);
85 GrBitmapTextureMaker maker(ctx, bmp);
86 auto[view, grCT] = maker.view(GrMipMapped::kNo);
87 std::unique_ptr<GrFragmentProcessor> imgFP =
88 GrTextureEffect::Make(std::move(view), bmp.alphaType(), SkMatrix());
89 auto fp = std::unique_ptr<GrFragmentProcessor>(new SampleCoordEffect(std::move(imgFP)));
90
91 GrPaint grPaint;
92 grPaint.addCoverageFragmentProcessor(std::move(fp));
93
94 rtCtx->priv().testingOnly_addDrawOp(GrFillRectOp::MakeNonAARect(ctx,
95 std::move(grPaint),
96 SkMatrix::I(),
97 bounds));
98 }
99