robertphillips | f7142e7 | 2016-04-18 07:20:05 -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 "GrAlphaThresholdFragmentProcessor.h" |
| 9 | |
| 10 | #if SK_SUPPORT_GPU |
| 11 | |
| 12 | #include "GrInvariantOutput.h" |
bungeman | 06ca8ec | 2016-06-09 08:01:03 -0700 | [diff] [blame] | 13 | #include "SkRefCnt.h" |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 14 | |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 15 | #include "glsl/GrGLSLColorSpaceXformHelper.h" |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 16 | #include "glsl/GrGLSLFragmentProcessor.h" |
| 17 | #include "glsl/GrGLSLFragmentShaderBuilder.h" |
| 18 | #include "glsl/GrGLSLUniformHandler.h" |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame^] | 19 | #include "../private/GrGLSL.h" |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 20 | |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 21 | sk_sp<GrFragmentProcessor> GrAlphaThresholdFragmentProcessor::Make( |
| 22 | GrTexture* texture, |
| 23 | sk_sp<GrColorSpaceXform> colorSpaceXform, |
| 24 | GrTexture* maskTexture, |
| 25 | float innerThreshold, |
| 26 | float outerThreshold, |
| 27 | const SkIRect& bounds) { |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 28 | return sk_sp<GrFragmentProcessor>(new GrAlphaThresholdFragmentProcessor( |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 29 | texture, std::move(colorSpaceXform), |
| 30 | maskTexture, |
| 31 | innerThreshold, outerThreshold, |
| 32 | bounds)); |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 33 | } |
| 34 | |
| 35 | static SkMatrix make_div_and_translate_matrix(GrTexture* texture, int x, int y) { |
| 36 | SkMatrix matrix = GrCoordTransform::MakeDivByTextureWHMatrix(texture); |
| 37 | matrix.preTranslate(SkIntToScalar(x), SkIntToScalar(y)); |
| 38 | return matrix; |
| 39 | } |
| 40 | |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 41 | GrAlphaThresholdFragmentProcessor::GrAlphaThresholdFragmentProcessor( |
| 42 | GrTexture* texture, |
| 43 | sk_sp<GrColorSpaceXform> colorSpaceXform, |
| 44 | GrTexture* maskTexture, |
| 45 | float innerThreshold, |
| 46 | float outerThreshold, |
| 47 | const SkIRect& bounds) |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 48 | : fInnerThreshold(innerThreshold) |
| 49 | , fOuterThreshold(outerThreshold) |
Brian Salomon | 2ebd0c8 | 2016-10-03 17:15:28 -0400 | [diff] [blame] | 50 | , fImageCoordTransform(GrCoordTransform::MakeDivByTextureWHMatrix(texture), texture, |
Brian Salomon | 514baff | 2016-11-17 15:17:07 -0500 | [diff] [blame] | 51 | GrSamplerParams::kNone_FilterMode) |
Brian Salomon | 0bbecb2 | 2016-11-17 11:38:22 -0500 | [diff] [blame] | 52 | , fImageTextureSampler(texture) |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 53 | , fColorSpaceXform(std::move(colorSpaceXform)) |
Brian Salomon | 2ebd0c8 | 2016-10-03 17:15:28 -0400 | [diff] [blame] | 54 | , fMaskCoordTransform(make_div_and_translate_matrix(maskTexture, -bounds.x(), -bounds.y()), |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 55 | maskTexture, |
Brian Salomon | 514baff | 2016-11-17 15:17:07 -0500 | [diff] [blame] | 56 | GrSamplerParams::kNone_FilterMode) |
Brian Salomon | 0bbecb2 | 2016-11-17 11:38:22 -0500 | [diff] [blame] | 57 | , fMaskTextureSampler(maskTexture) { |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 58 | this->initClassID<GrAlphaThresholdFragmentProcessor>(); |
| 59 | this->addCoordTransform(&fImageCoordTransform); |
Brian Salomon | 0bbecb2 | 2016-11-17 11:38:22 -0500 | [diff] [blame] | 60 | this->addTextureSampler(&fImageTextureSampler); |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 61 | this->addCoordTransform(&fMaskCoordTransform); |
Brian Salomon | 0bbecb2 | 2016-11-17 11:38:22 -0500 | [diff] [blame] | 62 | this->addTextureSampler(&fMaskTextureSampler); |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 63 | } |
| 64 | |
| 65 | bool GrAlphaThresholdFragmentProcessor::onIsEqual(const GrFragmentProcessor& sBase) const { |
| 66 | const GrAlphaThresholdFragmentProcessor& s = sBase.cast<GrAlphaThresholdFragmentProcessor>(); |
| 67 | return (this->fInnerThreshold == s.fInnerThreshold && |
| 68 | this->fOuterThreshold == s.fOuterThreshold); |
| 69 | } |
| 70 | |
| 71 | void GrAlphaThresholdFragmentProcessor::onComputeInvariantOutput(GrInvariantOutput* inout) const { |
Brian Salomon | db4183d | 2016-11-17 12:48:40 -0500 | [diff] [blame] | 72 | GrPixelConfig config = this->textureSampler(0).texture()->config(); |
Brian Salomon | 0bbecb2 | 2016-11-17 11:38:22 -0500 | [diff] [blame] | 73 | if (GrPixelConfigIsAlphaOnly(config)) { |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 74 | inout->mulByUnknownSingleComponent(); |
Brian Salomon | 0bbecb2 | 2016-11-17 11:38:22 -0500 | [diff] [blame] | 75 | } else if (GrPixelConfigIsOpaque(config) && fOuterThreshold >= 1.f) { |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 76 | inout->mulByUnknownOpaqueFourComponents(); |
| 77 | } else { |
| 78 | inout->mulByUnknownFourComponents(); |
| 79 | } |
| 80 | } |
| 81 | |
| 82 | /////////////////////////////////////////////////////////////////////////////// |
| 83 | |
| 84 | class GrGLAlphaThresholdFragmentProcessor : public GrGLSLFragmentProcessor { |
| 85 | public: |
| 86 | void emitCode(EmitArgs&) override; |
| 87 | |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame^] | 88 | static inline void GenKey(const GrProcessor& effect, const GrShaderCaps&, |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 89 | GrProcessorKeyBuilder* b) { |
| 90 | const GrAlphaThresholdFragmentProcessor& atfp = |
| 91 | effect.cast<GrAlphaThresholdFragmentProcessor>(); |
| 92 | b->add32(GrColorSpaceXform::XformKey(atfp.colorSpaceXform())); |
| 93 | } |
| 94 | |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 95 | protected: |
| 96 | void onSetData(const GrGLSLProgramDataManager&, const GrProcessor&) override; |
| 97 | |
| 98 | private: |
| 99 | GrGLSLProgramDataManager::UniformHandle fInnerThresholdVar; |
| 100 | GrGLSLProgramDataManager::UniformHandle fOuterThresholdVar; |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 101 | GrGLSLProgramDataManager::UniformHandle fColorSpaceXformVar; |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 102 | typedef GrGLSLFragmentProcessor INHERITED; |
| 103 | }; |
| 104 | |
| 105 | void GrGLAlphaThresholdFragmentProcessor::emitCode(EmitArgs& args) { |
| 106 | GrGLSLUniformHandler* uniformHandler = args.fUniformHandler; |
| 107 | fInnerThresholdVar = uniformHandler->addUniform(kFragment_GrShaderFlag, |
| 108 | kFloat_GrSLType, kDefault_GrSLPrecision, |
| 109 | "inner_threshold"); |
| 110 | fOuterThresholdVar = uniformHandler->addUniform(kFragment_GrShaderFlag, |
| 111 | kFloat_GrSLType, kDefault_GrSLPrecision, |
| 112 | "outer_threshold"); |
| 113 | |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 114 | const GrAlphaThresholdFragmentProcessor& atfp = |
| 115 | args.fFp.cast<GrAlphaThresholdFragmentProcessor>(); |
| 116 | GrGLSLColorSpaceXformHelper colorSpaceHelper(uniformHandler, atfp.colorSpaceXform(), |
| 117 | &fColorSpaceXformVar); |
| 118 | |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 119 | GrGLSLFPFragmentBuilder* fragBuilder = args.fFragBuilder; |
bsalomon | 1a1aa93 | 2016-09-12 09:30:36 -0700 | [diff] [blame] | 120 | SkString coords2D = fragBuilder->ensureCoords2D(args.fTransformedCoords[0]); |
| 121 | SkString maskCoords2D = fragBuilder->ensureCoords2D(args.fTransformedCoords[1]); |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 122 | |
| 123 | fragBuilder->codeAppendf("vec2 coord = %s;", coords2D.c_str()); |
| 124 | fragBuilder->codeAppendf("vec2 mask_coord = %s;", maskCoords2D.c_str()); |
| 125 | fragBuilder->codeAppend("vec4 input_color = "); |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 126 | fragBuilder->appendTextureLookup(args.fTexSamplers[0], "coord", kVec2f_GrSLType, |
| 127 | &colorSpaceHelper); |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 128 | fragBuilder->codeAppend(";"); |
| 129 | fragBuilder->codeAppend("vec4 mask_color = "); |
| 130 | fragBuilder->appendTextureLookup(args.fTexSamplers[1], "mask_coord"); |
| 131 | fragBuilder->codeAppend(";"); |
| 132 | |
| 133 | fragBuilder->codeAppendf("float inner_thresh = %s;", |
| 134 | uniformHandler->getUniformCStr(fInnerThresholdVar)); |
| 135 | fragBuilder->codeAppendf("float outer_thresh = %s;", |
| 136 | uniformHandler->getUniformCStr(fOuterThresholdVar)); |
| 137 | fragBuilder->codeAppend("float mask = mask_color.a;"); |
| 138 | |
| 139 | fragBuilder->codeAppend("vec4 color = input_color;"); |
| 140 | fragBuilder->codeAppend("if (mask < 0.5) {" |
| 141 | "if (color.a > outer_thresh) {" |
| 142 | "float scale = outer_thresh / color.a;" |
| 143 | "color.rgb *= scale;" |
| 144 | "color.a = outer_thresh;" |
| 145 | "}" |
| 146 | "} else if (color.a < inner_thresh) {" |
| 147 | "float scale = inner_thresh / max(0.001, color.a);" |
| 148 | "color.rgb *= scale;" |
| 149 | "color.a = inner_thresh;" |
| 150 | "}"); |
| 151 | |
| 152 | fragBuilder->codeAppendf("%s = %s;", args.fOutputColor, |
| 153 | (GrGLSLExpr4(args.fInputColor) * GrGLSLExpr4("color")).c_str()); |
| 154 | } |
| 155 | |
| 156 | void GrGLAlphaThresholdFragmentProcessor::onSetData(const GrGLSLProgramDataManager& pdman, |
| 157 | const GrProcessor& proc) { |
| 158 | const GrAlphaThresholdFragmentProcessor& atfp = proc.cast<GrAlphaThresholdFragmentProcessor>(); |
| 159 | pdman.set1f(fInnerThresholdVar, atfp.innerThreshold()); |
| 160 | pdman.set1f(fOuterThresholdVar, atfp.outerThreshold()); |
Brian Osman | 6fb592e | 2016-10-03 13:15:12 -0400 | [diff] [blame] | 161 | if (SkToBool(atfp.colorSpaceXform())) { |
| 162 | pdman.setSkMatrix44(fColorSpaceXformVar, atfp.colorSpaceXform()->srcToDst()); |
| 163 | } |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 164 | } |
| 165 | |
| 166 | ///////////////////////////////////////////////////////////////////// |
| 167 | |
| 168 | GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrAlphaThresholdFragmentProcessor); |
| 169 | |
bungeman | 06ca8ec | 2016-06-09 08:01:03 -0700 | [diff] [blame] | 170 | sk_sp<GrFragmentProcessor> GrAlphaThresholdFragmentProcessor::TestCreate(GrProcessorTestData* d) { |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 171 | GrTexture* bmpTex = d->fTextures[GrProcessorUnitTest::kSkiaPMTextureIdx]; |
| 172 | GrTexture* maskTex = d->fTextures[GrProcessorUnitTest::kAlphaTextureIdx]; |
| 173 | float innerThresh = d->fRandom->nextUScalar1(); |
| 174 | float outerThresh = d->fRandom->nextUScalar1(); |
| 175 | const int kMaxWidth = 1000; |
| 176 | const int kMaxHeight = 1000; |
| 177 | uint32_t width = d->fRandom->nextULessThan(kMaxWidth); |
| 178 | uint32_t height = d->fRandom->nextULessThan(kMaxHeight); |
| 179 | uint32_t x = d->fRandom->nextULessThan(kMaxWidth - width); |
| 180 | uint32_t y = d->fRandom->nextULessThan(kMaxHeight - height); |
| 181 | SkIRect bounds = SkIRect::MakeXYWH(x, y, width, height); |
Brian Osman | e2f732f | 2016-10-03 14:23:50 -0400 | [diff] [blame] | 182 | auto colorSpaceXform = GrTest::TestColorXform(d->fRandom); |
| 183 | return GrAlphaThresholdFragmentProcessor::Make(bmpTex, colorSpaceXform, maskTex, |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 184 | innerThresh, outerThresh, |
bungeman | 06ca8ec | 2016-06-09 08:01:03 -0700 | [diff] [blame] | 185 | bounds); |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 186 | } |
| 187 | |
| 188 | /////////////////////////////////////////////////////////////////////////////// |
| 189 | |
Brian Salomon | 94efbf5 | 2016-11-29 13:43:05 -0500 | [diff] [blame^] | 190 | void GrAlphaThresholdFragmentProcessor::onGetGLSLProcessorKey(const GrShaderCaps& caps, |
robertphillips | f7142e7 | 2016-04-18 07:20:05 -0700 | [diff] [blame] | 191 | GrProcessorKeyBuilder* b) const { |
| 192 | GrGLAlphaThresholdFragmentProcessor::GenKey(*this, caps, b); |
| 193 | } |
| 194 | |
| 195 | GrGLSLFragmentProcessor* GrAlphaThresholdFragmentProcessor::onCreateGLSLInstance() const { |
| 196 | return new GrGLAlphaThresholdFragmentProcessor; |
| 197 | } |
| 198 | |
| 199 | #endif |