brianosman | 54f30c1 | 2016-07-18 10:53:52 -0700 | [diff] [blame] | 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" |
Brian Osman | f06ead9 | 2017-10-30 13:47:41 -0400 | [diff] [blame] | 10 | #include "SkColorSpacePriv.h" |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 11 | #include "glsl/GrGLSLColorSpaceXformHelper.h" |
| 12 | #include "glsl/GrGLSLFragmentProcessor.h" |
| 13 | #include "glsl/GrGLSLFragmentShaderBuilder.h" |
Brian Osman | 7c2114f | 2016-10-20 15:34:06 -0400 | [diff] [blame] | 14 | |
Mike Klein | e03a176 | 2018-08-22 11:52:16 -0400 | [diff] [blame] | 15 | sk_sp<GrColorSpaceXform> GrColorSpaceXform::Make(SkColorSpace* src, SkAlphaType srcAT, |
| 16 | SkColorSpace* dst, SkAlphaType dstAT) { |
Mike Klein | e03a176 | 2018-08-22 11:52:16 -0400 | [diff] [blame] | 17 | SkColorSpaceXformSteps steps(src, srcAT, dst, dstAT); |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 18 | return steps.flags.mask() == 0 ? nullptr /* Noop transform */ |
| 19 | : sk_make_sp<GrColorSpaceXform>(steps); |
brianosman | 54f30c1 | 2016-07-18 10:53:52 -0700 | [diff] [blame] | 20 | } |
brianosman | 5a7ae7e | 2016-09-12 12:07:25 -0700 | [diff] [blame] | 21 | |
brianosman | b9c5137 | 2016-09-15 11:09:45 -0700 | [diff] [blame] | 22 | bool GrColorSpaceXform::Equals(const GrColorSpaceXform* a, const GrColorSpaceXform* b) { |
| 23 | if (a == b) { |
| 24 | return true; |
| 25 | } |
| 26 | |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 27 | if (!a || !b || a->fSteps.flags.mask() != b->fSteps.flags.mask()) { |
brianosman | b9c5137 | 2016-09-15 11:09:45 -0700 | [diff] [blame] | 28 | return false; |
| 29 | } |
| 30 | |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 31 | if (a->fSteps.flags.linearize && |
| 32 | 0 != memcmp(&a->fSteps.srcTF, &b->fSteps.srcTF, sizeof(a->fSteps.srcTF))) { |
Brian Osman | f06ead9 | 2017-10-30 13:47:41 -0400 | [diff] [blame] | 33 | return false; |
| 34 | } |
| 35 | |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 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))) { |
Brian Osman | f06ead9 | 2017-10-30 13:47:41 -0400 | [diff] [blame] | 44 | return false; |
| 45 | } |
| 46 | |
| 47 | return true; |
brianosman | b9c5137 | 2016-09-15 11:09:45 -0700 | [diff] [blame] | 48 | } |
| 49 | |
Brian Osman | f28e55d | 2018-10-03 16:35:54 -0400 | [diff] [blame] | 50 | SkColor4f GrColorSpaceXform::apply(const SkColor4f& srcColor) { |
| 51 | SkColor4f result = srcColor; |
| 52 | fSteps.apply(result.vec()); |
| 53 | return result; |
| 54 | } |
| 55 | |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 56 | ////////////////////////////////////////////////////////////////////////////// |
| 57 | |
| 58 | class GrGLColorSpaceXformEffect : public GrGLSLFragmentProcessor { |
| 59 | public: |
| 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 | |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 67 | if (this->numChildProcessors()) { |
| 68 | SkString childColor("src_color"); |
| 69 | this->emitChild(0, &childColor, args); |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 70 | |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 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 { |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 76 | SkString xformedColor; |
| 77 | fragBuilder->appendColorGamutXform(&xformedColor, args.fInputColor, &fColorSpaceHelper); |
| 78 | fragBuilder->codeAppendf("%s = %s;", args.fOutputColor, xformedColor.c_str()); |
| 79 | } |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 80 | } |
| 81 | |
| 82 | private: |
| 83 | void onSetData(const GrGLSLProgramDataManager& pdman, |
| 84 | const GrFragmentProcessor& processor) override { |
| 85 | const GrColorSpaceXformEffect& csxe = processor.cast<GrColorSpaceXformEffect>(); |
Brian Osman | c891b10 | 2018-06-14 14:50:17 -0400 | [diff] [blame] | 86 | fColorSpaceHelper.setData(pdman, csxe.colorXform()); |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 87 | } |
| 88 | |
| 89 | GrGLSLColorSpaceXformHelper fColorSpaceHelper; |
| 90 | |
| 91 | typedef GrGLSLFragmentProcessor INHERITED; |
| 92 | }; |
| 93 | |
| 94 | ////////////////////////////////////////////////////////////////////////////// |
| 95 | |
| 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)) { |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 100 | if (child) { |
| 101 | this->registerChildProcessor(std::move(child)); |
| 102 | } |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 103 | } |
| 104 | |
| 105 | std::unique_ptr<GrFragmentProcessor> GrColorSpaceXformEffect::clone() const { |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 106 | std::unique_ptr<GrFragmentProcessor> child = |
| 107 | this->numChildProcessors() ? this->childProcessor(0).clone() : nullptr; |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 108 | return std::unique_ptr<GrFragmentProcessor>( |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 109 | new GrColorSpaceXformEffect(std::move(child), fColorXform)); |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 110 | } |
| 111 | |
| 112 | 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 | |
| 117 | void GrColorSpaceXformEffect::onGetGLSLProcessorKey(const GrShaderCaps& caps, |
| 118 | GrProcessorKeyBuilder* b) const { |
| 119 | b->add32(GrColorSpaceXform::XformKey(fColorXform.get())); |
| 120 | } |
| 121 | |
| 122 | GrGLSLFragmentProcessor* GrColorSpaceXformEffect::onCreateGLSLInstance() const { |
| 123 | return new GrGLColorSpaceXformEffect(); |
| 124 | } |
| 125 | |
| 126 | GrFragmentProcessor::OptimizationFlags GrColorSpaceXformEffect::OptFlags( |
| 127 | const GrFragmentProcessor* child) { |
| 128 | // TODO: Implement constant output for constant input |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 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; |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 141 | } |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 142 | } |
| 143 | |
| 144 | std::unique_ptr<GrFragmentProcessor> GrColorSpaceXformEffect::Make(SkColorSpace* src, |
Brian Osman | 21fc5ce | 2018-08-27 20:36:19 +0000 | [diff] [blame] | 145 | SkAlphaType srcAT, |
Brian Osman | 15f0f29 | 2018-10-01 14:14:46 -0400 | [diff] [blame] | 146 | SkColorSpace* dst, |
| 147 | SkAlphaType dstAT) { |
Brian Osman | 21fc5ce | 2018-08-27 20:36:19 +0000 | [diff] [blame] | 148 | auto xform = GrColorSpaceXform::Make(src, srcAT, |
Brian Osman | 15f0f29 | 2018-10-01 14:14:46 -0400 | [diff] [blame] | 149 | dst, dstAT); |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 150 | if (!xform) { |
| 151 | return nullptr; |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 152 | } |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 153 | |
| 154 | return std::unique_ptr<GrFragmentProcessor>(new GrColorSpaceXformEffect(nullptr, |
| 155 | std::move(xform))); |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 156 | } |
| 157 | |
| 158 | std::unique_ptr<GrFragmentProcessor> GrColorSpaceXformEffect::Make( |
| 159 | std::unique_ptr<GrFragmentProcessor> child, |
Brian Osman | 21fc5ce | 2018-08-27 20:36:19 +0000 | [diff] [blame] | 160 | SkColorSpace* src, SkAlphaType srcAT, SkColorSpace* dst) { |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 161 | if (!child) { |
| 162 | return nullptr; |
| 163 | } |
| 164 | |
Brian Osman | 21fc5ce | 2018-08-27 20:36:19 +0000 | [diff] [blame] | 165 | auto xform = GrColorSpaceXform::Make(src, srcAT, |
Mike Klein | e03a176 | 2018-08-22 11:52:16 -0400 | [diff] [blame] | 166 | dst, kPremul_SkAlphaType); |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 167 | if (!xform) { |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 168 | return child; |
| 169 | } |
Brian Osman | 3567c14 | 2018-06-18 10:20:32 -0400 | [diff] [blame] | 170 | |
| 171 | return std::unique_ptr<GrFragmentProcessor>(new GrColorSpaceXformEffect(std::move(child), |
| 172 | std::move(xform))); |
Brian Osman | c4f93ca | 2017-10-17 17:15:52 -0400 | [diff] [blame] | 173 | } |