1 /*
2 * Copyright 2016 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 #include "SkNormalFlatSource.h"
9
10 #include "SkArenaAlloc.h"
11 #include "SkNormalSource.h"
12 #include "SkNormalSourcePriv.h"
13 #include "SkPoint3.h"
14 #include "SkReadBuffer.h"
15 #include "SkWriteBuffer.h"
16
17 #if SK_SUPPORT_GPU
18 #include "glsl/GrGLSLFragmentProcessor.h"
19 #include "glsl/GrGLSLFragmentShaderBuilder.h"
20
21 class NormalFlatFP : public GrFragmentProcessor {
22 public:
NormalFlatFP()23 NormalFlatFP() : INHERITED(kConstantOutputForConstantInput_OptimizationFlag) {
24 this->initClassID<NormalFlatFP>();
25 }
26
27 class GLSLNormalFlatFP : public GLSLNormalFP {
28 public:
GLSLNormalFlatFP()29 GLSLNormalFlatFP() {}
30
onEmitCode(EmitArgs & args)31 void onEmitCode(EmitArgs& args) override {
32 GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
33
34 fragBuilder->codeAppendf("%s = vec4(0, 0, 1, 0);", args.fOutputColor);
35 }
36
GenKey(const GrProcessor & proc,const GrShaderCaps &,GrProcessorKeyBuilder * b)37 static void GenKey(const GrProcessor& proc, const GrShaderCaps&, GrProcessorKeyBuilder* b) {
38 b->add32(0x0);
39 }
40
41 protected:
setNormalData(const GrGLSLProgramDataManager & pdman,const GrProcessor & proc)42 void setNormalData(const GrGLSLProgramDataManager& pdman,
43 const GrProcessor& proc) override {}
44 };
45
name() const46 const char* name() const override { return "NormalFlatFP"; }
47
48 private:
onGetGLSLProcessorKey(const GrShaderCaps & caps,GrProcessorKeyBuilder * b) const49 void onGetGLSLProcessorKey(const GrShaderCaps& caps, GrProcessorKeyBuilder* b) const override {
50 GLSLNormalFlatFP::GenKey(*this, caps, b);
51 }
52
constantOutputForConstantInput(GrColor4f) const53 GrColor4f constantOutputForConstantInput(GrColor4f) const override {
54 return GrColor4f(0, 0, 1, 0);
55 }
onCreateGLSLInstance() const56 GrGLSLFragmentProcessor* onCreateGLSLInstance() const override { return new GLSLNormalFlatFP; }
57
onIsEqual(const GrFragmentProcessor &) const58 bool onIsEqual(const GrFragmentProcessor&) const override { return true; }
59
60 typedef GrFragmentProcessor INHERITED;
61 };
62
asFragmentProcessor(const SkShader::AsFPArgs &) const63 sk_sp<GrFragmentProcessor> SkNormalFlatSourceImpl::asFragmentProcessor(
64 const SkShader::AsFPArgs&) const {
65
66 return sk_make_sp<NormalFlatFP>();
67 }
68
69 #endif // SK_SUPPORT_GPU
70
71 ////////////////////////////////////////////////////////////////////////////
72
Provider()73 SkNormalFlatSourceImpl::Provider::Provider() {}
74
~Provider()75 SkNormalFlatSourceImpl::Provider::~Provider() {}
76
asProvider(const SkShader::ContextRec & rec,SkArenaAlloc * alloc) const77 SkNormalSource::Provider* SkNormalFlatSourceImpl::asProvider(const SkShader::ContextRec &rec,
78 SkArenaAlloc *alloc) const {
79 return alloc->make<Provider>();
80 }
81
fillScanLine(int x,int y,SkPoint3 output[],int count) const82 void SkNormalFlatSourceImpl::Provider::fillScanLine(int x, int y, SkPoint3 output[],
83 int count) const {
84 for (int i = 0; i < count; i++) {
85 output[i] = {0.0f, 0.0f, 1.0f};
86 }
87 }
88
89 ////////////////////////////////////////////////////////////////////////////////
90
CreateProc(SkReadBuffer & buf)91 sk_sp<SkFlattenable> SkNormalFlatSourceImpl::CreateProc(SkReadBuffer& buf) {
92 return sk_make_sp<SkNormalFlatSourceImpl>();
93 }
94
flatten(SkWriteBuffer & buf) const95 void SkNormalFlatSourceImpl::flatten(SkWriteBuffer& buf) const {
96 this->INHERITED::flatten(buf);
97 }
98
99 ////////////////////////////////////////////////////////////////////////////
100
MakeFlat()101 sk_sp<SkNormalSource> SkNormalSource::MakeFlat() {
102 return sk_make_sp<SkNormalFlatSourceImpl>();
103 }
104