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 "GrColorSpaceXform.h"
9 #include "SkColorSpace.h"
10 #include "SkColorSpacePriv.h"
11 #include "glsl/GrGLSLColorSpaceXformHelper.h"
12 #include "glsl/GrGLSLFragmentProcessor.h"
13 #include "glsl/GrGLSLFragmentShaderBuilder.h"
14
Make(SkColorSpace * src,SkAlphaType srcAT,SkColorSpace * dst,SkAlphaType dstAT)15 sk_sp<GrColorSpaceXform> GrColorSpaceXform::Make(SkColorSpace* src, SkAlphaType srcAT,
16 SkColorSpace* dst, SkAlphaType dstAT) {
17 SkColorSpaceXformSteps steps(src, srcAT, dst, dstAT);
18 return steps.flags.mask() == 0 ? nullptr /* Noop transform */
19 : sk_make_sp<GrColorSpaceXform>(steps);
20 }
21
Equals(const GrColorSpaceXform * a,const GrColorSpaceXform * b)22 bool GrColorSpaceXform::Equals(const GrColorSpaceXform* a, const GrColorSpaceXform* b) {
23 if (a == b) {
24 return true;
25 }
26
27 if (!a || !b || a->fSteps.flags.mask() != b->fSteps.flags.mask()) {
28 return false;
29 }
30
31 if (a->fSteps.flags.linearize &&
32 0 != memcmp(&a->fSteps.srcTF, &b->fSteps.srcTF, sizeof(a->fSteps.srcTF))) {
33 return false;
34 }
35
36 if (a->fSteps.flags.gamut_transform &&
37 0 != memcmp(&a->fSteps.src_to_dst_matrix, &b->fSteps.src_to_dst_matrix,
38 sizeof(a->fSteps.src_to_dst_matrix))) {
39 return false;
40 }
41
42 if (a->fSteps.flags.encode &&
43 0 != memcmp(&a->fSteps.dstTFInv, &b->fSteps.dstTFInv, sizeof(a->fSteps.dstTFInv))) {
44 return false;
45 }
46
47 return true;
48 }
49
apply(const SkColor4f & srcColor)50 SkColor4f GrColorSpaceXform::apply(const SkColor4f& srcColor) {
51 SkColor4f result = srcColor;
52 fSteps.apply(result.vec());
53 return result;
54 }
55
56 //////////////////////////////////////////////////////////////////////////////
57
58 class GrGLColorSpaceXformEffect : public GrGLSLFragmentProcessor {
59 public:
emitCode(EmitArgs & args)60 void emitCode(EmitArgs& args) override {
61 const GrColorSpaceXformEffect& csxe = args.fFp.cast<GrColorSpaceXformEffect>();
62 GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder;
63 GrGLSLUniformHandler* uniformHandler = args.fUniformHandler;
64
65 fColorSpaceHelper.emitCode(uniformHandler, csxe.colorXform());
66
67 if (this->numChildProcessors()) {
68 SkString childColor("src_color");
69 this->emitChild(0, &childColor, args);
70
71 SkString xformedColor;
72 fragBuilder->appendColorGamutXform(&xformedColor, childColor.c_str(), &fColorSpaceHelper);
73 fragBuilder->codeAppendf("%s = %s * %s;", args.fOutputColor, xformedColor.c_str(),
74 args.fInputColor);
75 } else {
76 SkString xformedColor;
77 fragBuilder->appendColorGamutXform(&xformedColor, args.fInputColor, &fColorSpaceHelper);
78 fragBuilder->codeAppendf("%s = %s;", args.fOutputColor, xformedColor.c_str());
79 }
80 }
81
82 private:
onSetData(const GrGLSLProgramDataManager & pdman,const GrFragmentProcessor & processor)83 void onSetData(const GrGLSLProgramDataManager& pdman,
84 const GrFragmentProcessor& processor) override {
85 const GrColorSpaceXformEffect& csxe = processor.cast<GrColorSpaceXformEffect>();
86 fColorSpaceHelper.setData(pdman, csxe.colorXform());
87 }
88
89 GrGLSLColorSpaceXformHelper fColorSpaceHelper;
90
91 typedef GrGLSLFragmentProcessor INHERITED;
92 };
93
94 //////////////////////////////////////////////////////////////////////////////
95
GrColorSpaceXformEffect(std::unique_ptr<GrFragmentProcessor> child,sk_sp<GrColorSpaceXform> colorXform)96 GrColorSpaceXformEffect::GrColorSpaceXformEffect(std::unique_ptr<GrFragmentProcessor> child,
97 sk_sp<GrColorSpaceXform> colorXform)
98 : INHERITED(kGrColorSpaceXformEffect_ClassID, OptFlags(child.get()))
99 , fColorXform(std::move(colorXform)) {
100 if (child) {
101 this->registerChildProcessor(std::move(child));
102 }
103 }
104
clone() const105 std::unique_ptr<GrFragmentProcessor> GrColorSpaceXformEffect::clone() const {
106 std::unique_ptr<GrFragmentProcessor> child =
107 this->numChildProcessors() ? this->childProcessor(0).clone() : nullptr;
108 return std::unique_ptr<GrFragmentProcessor>(
109 new GrColorSpaceXformEffect(std::move(child), fColorXform));
110 }
111
onIsEqual(const GrFragmentProcessor & s) const112 bool GrColorSpaceXformEffect::onIsEqual(const GrFragmentProcessor& s) const {
113 const GrColorSpaceXformEffect& other = s.cast<GrColorSpaceXformEffect>();
114 return GrColorSpaceXform::Equals(fColorXform.get(), other.fColorXform.get());
115 }
116
onGetGLSLProcessorKey(const GrShaderCaps & caps,GrProcessorKeyBuilder * b) const117 void GrColorSpaceXformEffect::onGetGLSLProcessorKey(const GrShaderCaps& caps,
118 GrProcessorKeyBuilder* b) const {
119 b->add32(GrColorSpaceXform::XformKey(fColorXform.get()));
120 }
121
onCreateGLSLInstance() const122 GrGLSLFragmentProcessor* GrColorSpaceXformEffect::onCreateGLSLInstance() const {
123 return new GrGLColorSpaceXformEffect();
124 }
125
OptFlags(const GrFragmentProcessor * child)126 GrFragmentProcessor::OptimizationFlags GrColorSpaceXformEffect::OptFlags(
127 const GrFragmentProcessor* child) {
128 // TODO: Implement constant output for constant input
129 if (child) {
130 OptimizationFlags flags = kNone_OptimizationFlags;
131 if (child->compatibleWithCoverageAsAlpha()) {
132 flags |= kCompatibleWithCoverageAsAlpha_OptimizationFlag;
133 }
134 if (child->preservesOpaqueInput()) {
135 flags |= kPreservesOpaqueInput_OptimizationFlag;
136 }
137 return flags;
138 } else {
139 return kCompatibleWithCoverageAsAlpha_OptimizationFlag |
140 kPreservesOpaqueInput_OptimizationFlag;
141 }
142 }
143
Make(SkColorSpace * src,SkAlphaType srcAT,SkColorSpace * dst,SkAlphaType dstAT)144 std::unique_ptr<GrFragmentProcessor> GrColorSpaceXformEffect::Make(SkColorSpace* src,
145 SkAlphaType srcAT,
146 SkColorSpace* dst,
147 SkAlphaType dstAT) {
148 auto xform = GrColorSpaceXform::Make(src, srcAT,
149 dst, dstAT);
150 if (!xform) {
151 return nullptr;
152 }
153
154 return std::unique_ptr<GrFragmentProcessor>(new GrColorSpaceXformEffect(nullptr,
155 std::move(xform)));
156 }
157
Make(std::unique_ptr<GrFragmentProcessor> child,SkColorSpace * src,SkAlphaType srcAT,SkColorSpace * dst)158 std::unique_ptr<GrFragmentProcessor> GrColorSpaceXformEffect::Make(
159 std::unique_ptr<GrFragmentProcessor> child,
160 SkColorSpace* src, SkAlphaType srcAT, SkColorSpace* dst) {
161 if (!child) {
162 return nullptr;
163 }
164
165 auto xform = GrColorSpaceXform::Make(src, srcAT,
166 dst, kPremul_SkAlphaType);
167 if (!xform) {
168 return child;
169 }
170
171 return std::unique_ptr<GrFragmentProcessor>(new GrColorSpaceXformEffect(std::move(child),
172 std::move(xform)));
173 }
174
Make(std::unique_ptr<GrFragmentProcessor> child,sk_sp<GrColorSpaceXform> colorXform)175 std::unique_ptr<GrFragmentProcessor> GrColorSpaceXformEffect::Make(
176 std::unique_ptr<GrFragmentProcessor> child, sk_sp<GrColorSpaceXform> colorXform) {
177 if (!child) {
178 return nullptr;
179 }
180 if (!colorXform) {
181 return child;
182 }
183
184 return std::unique_ptr<GrFragmentProcessor>(new GrColorSpaceXformEffect(std::move(child),
185 std::move(colorXform)));
186 }
187