robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2013 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 | |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 8 | #include "SkArithmeticMode.h" |
| 9 | #include "SkColorPriv.h" |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 10 | #include "SkReadBuffer.h" |
| 11 | #include "SkWriteBuffer.h" |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 12 | #include "SkString.h" |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 13 | #include "SkUnPreMultiply.h" |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 14 | #if SK_SUPPORT_GPU |
| 15 | #include "GrContext.h" |
bsalomon@google.com | 77af680 | 2013-10-02 13:04:56 +0000 | [diff] [blame] | 16 | #include "GrCoordTransform.h" |
egdaniel | 605dd0f | 2014-11-12 08:35:25 -0800 | [diff] [blame] | 17 | #include "GrInvariantOutput.h" |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 18 | #include "gl/GrGLProcessor.h" |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 19 | #include "gl/builders/GrGLProgramBuilder.h" |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 20 | #include "GrTBackendProcessorFactory.h" |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 21 | #endif |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 22 | |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 23 | static const bool gUseUnpremul = false; |
| 24 | |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 25 | class SkArithmeticMode_scalar : public SkXfermode { |
| 26 | public: |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 27 | static SkArithmeticMode_scalar* Create(SkScalar k1, SkScalar k2, SkScalar k3, SkScalar k4, |
| 28 | bool enforcePMColor) { |
| 29 | return SkNEW_ARGS(SkArithmeticMode_scalar, (k1, k2, k3, k4, enforcePMColor)); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 30 | } |
| 31 | |
| 32 | virtual void xfer32(SkPMColor dst[], const SkPMColor src[], int count, |
reed@google.com | 30da745 | 2012-12-17 19:55:24 +0000 | [diff] [blame] | 33 | const SkAlpha aa[]) const SK_OVERRIDE; |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 34 | |
commit-bot@chromium.org | 0f10f7b | 2014-03-13 18:02:17 +0000 | [diff] [blame] | 35 | SK_TO_STRING_OVERRIDE() |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 36 | SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkArithmeticMode_scalar) |
| 37 | |
| 38 | #if SK_SUPPORT_GPU |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 39 | virtual bool asFragmentProcessor(GrFragmentProcessor**, |
| 40 | GrTexture* background) const SK_OVERRIDE; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 41 | #endif |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 42 | |
| 43 | private: |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 44 | SkArithmeticMode_scalar(SkScalar k1, SkScalar k2, SkScalar k3, SkScalar k4, bool enforcePMColor) { |
commit-bot@chromium.org | 0a2bf90 | 2014-02-20 20:40:19 +0000 | [diff] [blame] | 45 | fK[0] = k1; |
| 46 | fK[1] = k2; |
| 47 | fK[2] = k3; |
| 48 | fK[3] = k4; |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 49 | fEnforcePMColor = enforcePMColor; |
commit-bot@chromium.org | 0a2bf90 | 2014-02-20 20:40:19 +0000 | [diff] [blame] | 50 | } |
| 51 | |
commit-bot@chromium.org | 8b0e8ac | 2014-01-30 18:58:24 +0000 | [diff] [blame] | 52 | virtual void flatten(SkWriteBuffer& buffer) const SK_OVERRIDE { |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 53 | buffer.writeScalar(fK[0]); |
| 54 | buffer.writeScalar(fK[1]); |
| 55 | buffer.writeScalar(fK[2]); |
| 56 | buffer.writeScalar(fK[3]); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 57 | buffer.writeBool(fEnforcePMColor); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 58 | } |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 59 | SkScalar fK[4]; |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 60 | bool fEnforcePMColor; |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 61 | |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 62 | friend class SkArithmeticMode; |
| 63 | |
skia.committer@gmail.com | 98ded84 | 2013-01-23 07:06:17 +0000 | [diff] [blame] | 64 | typedef SkXfermode INHERITED; |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 65 | }; |
| 66 | |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 67 | SkFlattenable* SkArithmeticMode_scalar::CreateProc(SkReadBuffer& buffer) { |
| 68 | const SkScalar k1 = buffer.readScalar(); |
| 69 | const SkScalar k2 = buffer.readScalar(); |
| 70 | const SkScalar k3 = buffer.readScalar(); |
| 71 | const SkScalar k4 = buffer.readScalar(); |
| 72 | const bool enforcePMColor = buffer.readBool(); |
| 73 | return Create(k1, k2, k3, k4, enforcePMColor); |
| 74 | } |
| 75 | |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 76 | static int pinToByte(int value) { |
| 77 | if (value < 0) { |
| 78 | value = 0; |
| 79 | } else if (value > 255) { |
| 80 | value = 255; |
| 81 | } |
| 82 | return value; |
| 83 | } |
| 84 | |
| 85 | static int arith(SkScalar k1, SkScalar k2, SkScalar k3, SkScalar k4, |
| 86 | int src, int dst) { |
| 87 | SkScalar result = SkScalarMul(k1, src * dst) + |
| 88 | SkScalarMul(k2, src) + |
| 89 | SkScalarMul(k3, dst) + |
| 90 | k4; |
| 91 | int res = SkScalarRoundToInt(result); |
| 92 | return pinToByte(res); |
| 93 | } |
| 94 | |
| 95 | static int blend(int src, int dst, int scale) { |
| 96 | return dst + ((src - dst) * scale >> 8); |
| 97 | } |
| 98 | |
| 99 | static bool needsUnpremul(int alpha) { |
| 100 | return 0 != alpha && 0xFF != alpha; |
| 101 | } |
| 102 | |
| 103 | void SkArithmeticMode_scalar::xfer32(SkPMColor dst[], const SkPMColor src[], |
reed@google.com | 30da745 | 2012-12-17 19:55:24 +0000 | [diff] [blame] | 104 | int count, const SkAlpha aaCoverage[]) const { |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 105 | SkScalar k1 = fK[0] / 255; |
| 106 | SkScalar k2 = fK[1]; |
| 107 | SkScalar k3 = fK[2]; |
| 108 | SkScalar k4 = fK[3] * 255; |
| 109 | |
| 110 | for (int i = 0; i < count; ++i) { |
| 111 | if ((NULL == aaCoverage) || aaCoverage[i]) { |
| 112 | SkPMColor sc = src[i]; |
| 113 | SkPMColor dc = dst[i]; |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 114 | |
| 115 | int a, r, g, b; |
| 116 | |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 117 | if (gUseUnpremul) { |
| 118 | int sa = SkGetPackedA32(sc); |
| 119 | int da = SkGetPackedA32(dc); |
| 120 | |
| 121 | int srcNeedsUnpremul = needsUnpremul(sa); |
| 122 | int dstNeedsUnpremul = needsUnpremul(da); |
| 123 | |
| 124 | if (!srcNeedsUnpremul && !dstNeedsUnpremul) { |
| 125 | a = arith(k1, k2, k3, k4, sa, da); |
| 126 | r = arith(k1, k2, k3, k4, SkGetPackedR32(sc), SkGetPackedR32(dc)); |
| 127 | g = arith(k1, k2, k3, k4, SkGetPackedG32(sc), SkGetPackedG32(dc)); |
| 128 | b = arith(k1, k2, k3, k4, SkGetPackedB32(sc), SkGetPackedB32(dc)); |
| 129 | } else { |
| 130 | int sr = SkGetPackedR32(sc); |
| 131 | int sg = SkGetPackedG32(sc); |
| 132 | int sb = SkGetPackedB32(sc); |
| 133 | if (srcNeedsUnpremul) { |
| 134 | SkUnPreMultiply::Scale scale = SkUnPreMultiply::GetScale(sa); |
| 135 | sr = SkUnPreMultiply::ApplyScale(scale, sr); |
| 136 | sg = SkUnPreMultiply::ApplyScale(scale, sg); |
| 137 | sb = SkUnPreMultiply::ApplyScale(scale, sb); |
| 138 | } |
| 139 | |
| 140 | int dr = SkGetPackedR32(dc); |
| 141 | int dg = SkGetPackedG32(dc); |
| 142 | int db = SkGetPackedB32(dc); |
| 143 | if (dstNeedsUnpremul) { |
| 144 | SkUnPreMultiply::Scale scale = SkUnPreMultiply::GetScale(da); |
| 145 | dr = SkUnPreMultiply::ApplyScale(scale, dr); |
| 146 | dg = SkUnPreMultiply::ApplyScale(scale, dg); |
| 147 | db = SkUnPreMultiply::ApplyScale(scale, db); |
| 148 | } |
| 149 | |
| 150 | a = arith(k1, k2, k3, k4, sa, da); |
| 151 | r = arith(k1, k2, k3, k4, sr, dr); |
| 152 | g = arith(k1, k2, k3, k4, sg, dg); |
| 153 | b = arith(k1, k2, k3, k4, sb, db); |
| 154 | } |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 155 | } else { |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 156 | a = arith(k1, k2, k3, k4, SkGetPackedA32(sc), SkGetPackedA32(dc)); |
| 157 | r = arith(k1, k2, k3, k4, SkGetPackedR32(sc), SkGetPackedR32(dc)); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 158 | g = arith(k1, k2, k3, k4, SkGetPackedG32(sc), SkGetPackedG32(dc)); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 159 | b = arith(k1, k2, k3, k4, SkGetPackedB32(sc), SkGetPackedB32(dc)); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 160 | if (fEnforcePMColor) { |
| 161 | r = SkMin32(r, a); |
| 162 | g = SkMin32(g, a); |
| 163 | b = SkMin32(b, a); |
| 164 | } |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 165 | } |
| 166 | |
| 167 | // apply antialias coverage if necessary |
| 168 | if (aaCoverage && 0xFF != aaCoverage[i]) { |
| 169 | int scale = aaCoverage[i] + (aaCoverage[i] >> 7); |
| 170 | a = blend(a, SkGetPackedA32(sc), scale); |
| 171 | r = blend(r, SkGetPackedR32(sc), scale); |
| 172 | g = blend(g, SkGetPackedG32(sc), scale); |
| 173 | b = blend(b, SkGetPackedB32(sc), scale); |
| 174 | } |
| 175 | |
mike@reedtribe.org | 0e330ae | 2012-08-12 19:08:41 +0000 | [diff] [blame] | 176 | // turn the result back into premul |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 177 | if (gUseUnpremul && (0xFF != a)) { |
mike@reedtribe.org | 0e330ae | 2012-08-12 19:08:41 +0000 | [diff] [blame] | 178 | int scale = a + (a >> 7); |
| 179 | r = SkAlphaMul(r, scale); |
| 180 | g = SkAlphaMul(g, scale); |
| 181 | b = SkAlphaMul(b, scale); |
| 182 | } |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 183 | dst[i] = fEnforcePMColor ? SkPackARGB32(a, r, g, b) : SkPackARGB32NoCheck(a, r, g, b); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 184 | } |
| 185 | } |
| 186 | } |
| 187 | |
commit-bot@chromium.org | 0f10f7b | 2014-03-13 18:02:17 +0000 | [diff] [blame] | 188 | #ifndef SK_IGNORE_TO_STRING |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 189 | void SkArithmeticMode_scalar::toString(SkString* str) const { |
| 190 | str->append("SkArithmeticMode_scalar: "); |
| 191 | for (int i = 0; i < 4; ++i) { |
| 192 | str->appendScalar(fK[i]); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 193 | str->append(" "); |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 194 | } |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 195 | str->appendS32(fEnforcePMColor ? 1 : 0); |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 196 | } |
| 197 | #endif |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 198 | |
| 199 | /////////////////////////////////////////////////////////////////////////////// |
| 200 | |
| 201 | static bool fitsInBits(SkScalar x, int bits) { |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 202 | return SkScalarAbs(x) < (1 << (bits - 1)); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 203 | } |
| 204 | |
caryclark@google.com | 383d5d4 | 2012-06-06 12:09:18 +0000 | [diff] [blame] | 205 | #if 0 // UNUSED |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 206 | static int32_t toDot8(SkScalar x) { |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 207 | return (int32_t)(x * 256); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 208 | } |
caryclark@google.com | 383d5d4 | 2012-06-06 12:09:18 +0000 | [diff] [blame] | 209 | #endif |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 210 | |
| 211 | SkXfermode* SkArithmeticMode::Create(SkScalar k1, SkScalar k2, |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 212 | SkScalar k3, SkScalar k4, |
| 213 | bool enforcePMColor) { |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 214 | if (fitsInBits(k1, 8) && fitsInBits(k2, 16) && |
| 215 | fitsInBits(k2, 16) && fitsInBits(k2, 24)) { |
| 216 | |
caryclark@google.com | 383d5d4 | 2012-06-06 12:09:18 +0000 | [diff] [blame] | 217 | #if 0 // UNUSED |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 218 | int32_t i1 = toDot8(k1); |
| 219 | int32_t i2 = toDot8(k2); |
| 220 | int32_t i3 = toDot8(k3); |
| 221 | int32_t i4 = toDot8(k4); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 222 | if (i1) { |
| 223 | return SkNEW_ARGS(SkArithmeticMode_quad, (i1, i2, i3, i4)); |
| 224 | } |
| 225 | if (0 == i2) { |
| 226 | return SkNEW_ARGS(SkArithmeticMode_dst, (i3, i4)); |
| 227 | } |
| 228 | if (0 == i3) { |
| 229 | return SkNEW_ARGS(SkArithmeticMode_src, (i2, i4)); |
| 230 | } |
| 231 | return SkNEW_ARGS(SkArithmeticMode_linear, (i2, i3, i4)); |
| 232 | #endif |
| 233 | } |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 234 | return SkArithmeticMode_scalar::Create(k1, k2, k3, k4, enforcePMColor); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 235 | } |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 236 | |
| 237 | |
| 238 | ////////////////////////////////////////////////////////////////////////////// |
| 239 | |
| 240 | #if SK_SUPPORT_GPU |
| 241 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 242 | class GrGLArithmeticEffect : public GrGLFragmentProcessor { |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 243 | public: |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 244 | GrGLArithmeticEffect(const GrBackendProcessorFactory&, const GrProcessor&); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 245 | virtual ~GrGLArithmeticEffect(); |
| 246 | |
joshualitt | 1598899 | 2014-10-09 15:04:05 -0700 | [diff] [blame] | 247 | virtual void emitCode(GrGLFPBuilder*, |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 248 | const GrFragmentProcessor&, |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 249 | const char* outputColor, |
| 250 | const char* inputColor, |
bsalomon@google.com | 77af680 | 2013-10-02 13:04:56 +0000 | [diff] [blame] | 251 | const TransformedCoordsArray&, |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 252 | const TextureSamplerArray&) SK_OVERRIDE; |
| 253 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 254 | virtual void setData(const GrGLProgramDataManager&, const GrProcessor&) SK_OVERRIDE; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 255 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 256 | static void GenKey(const GrProcessor&, const GrGLCaps& caps, GrProcessorKeyBuilder* b); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 257 | |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 258 | private: |
kkinnunen | 7510b22 | 2014-07-30 00:04:16 -0700 | [diff] [blame] | 259 | GrGLProgramDataManager::UniformHandle fKUni; |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 260 | bool fEnforcePMColor; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 261 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 262 | typedef GrGLFragmentProcessor INHERITED; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 263 | }; |
| 264 | |
| 265 | /////////////////////////////////////////////////////////////////////////////// |
| 266 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 267 | class GrArithmeticEffect : public GrFragmentProcessor { |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 268 | public: |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 269 | static GrFragmentProcessor* Create(float k1, float k2, float k3, float k4, bool enforcePMColor, |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 270 | GrTexture* background) { |
bsalomon | 55fad7a | 2014-07-08 07:34:20 -0700 | [diff] [blame] | 271 | return SkNEW_ARGS(GrArithmeticEffect, (k1, k2, k3, k4, enforcePMColor, background)); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 272 | } |
| 273 | |
| 274 | virtual ~GrArithmeticEffect(); |
| 275 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 276 | virtual const GrBackendFragmentProcessorFactory& getFactory() const SK_OVERRIDE; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 277 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 278 | typedef GrGLArithmeticEffect GLProcessor; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 279 | static const char* Name() { return "Arithmetic"; } |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 280 | GrTexture* backgroundTexture() const { return fBackgroundAccess.getTexture(); } |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 281 | |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 282 | float k1() const { return fK1; } |
| 283 | float k2() const { return fK2; } |
| 284 | float k3() const { return fK3; } |
| 285 | float k4() const { return fK4; } |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 286 | bool enforcePMColor() const { return fEnforcePMColor; } |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 287 | |
| 288 | private: |
bsalomon | 0e08fc1 | 2014-10-15 08:19:04 -0700 | [diff] [blame] | 289 | virtual bool onIsEqual(const GrFragmentProcessor&) const SK_OVERRIDE; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 290 | |
egdaniel | 605dd0f | 2014-11-12 08:35:25 -0800 | [diff] [blame] | 291 | virtual void onComputeInvariantOutput(GrInvariantOutput* inout) const SK_OVERRIDE; |
egdaniel | 1a8ecdf | 2014-10-03 06:24:12 -0700 | [diff] [blame] | 292 | |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 293 | GrArithmeticEffect(float k1, float k2, float k3, float k4, bool enforcePMColor, |
| 294 | GrTexture* background); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 295 | float fK1, fK2, fK3, fK4; |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 296 | bool fEnforcePMColor; |
bsalomon@google.com | 77af680 | 2013-10-02 13:04:56 +0000 | [diff] [blame] | 297 | GrCoordTransform fBackgroundTransform; |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 298 | GrTextureAccess fBackgroundAccess; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 299 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 300 | GR_DECLARE_FRAGMENT_PROCESSOR_TEST; |
| 301 | typedef GrFragmentProcessor INHERITED; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 302 | |
| 303 | }; |
| 304 | |
| 305 | /////////////////////////////////////////////////////////////////////////////// |
| 306 | |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 307 | GrArithmeticEffect::GrArithmeticEffect(float k1, float k2, float k3, float k4, |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 308 | bool enforcePMColor, GrTexture* background) |
| 309 | : fK1(k1), fK2(k2), fK3(k3), fK4(k4), fEnforcePMColor(enforcePMColor) { |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 310 | if (background) { |
bsalomon@google.com | 77af680 | 2013-10-02 13:04:56 +0000 | [diff] [blame] | 311 | fBackgroundTransform.reset(kLocal_GrCoordSet, background); |
| 312 | this->addCoordTransform(&fBackgroundTransform); |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 313 | fBackgroundAccess.reset(background); |
| 314 | this->addTextureAccess(&fBackgroundAccess); |
| 315 | } else { |
| 316 | this->setWillReadDstColor(); |
| 317 | } |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 318 | } |
| 319 | |
| 320 | GrArithmeticEffect::~GrArithmeticEffect() { |
| 321 | } |
| 322 | |
bsalomon | 0e08fc1 | 2014-10-15 08:19:04 -0700 | [diff] [blame] | 323 | bool GrArithmeticEffect::onIsEqual(const GrFragmentProcessor& sBase) const { |
joshualitt | 49586be | 2014-09-16 08:21:41 -0700 | [diff] [blame] | 324 | const GrArithmeticEffect& s = sBase.cast<GrArithmeticEffect>(); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 325 | return fK1 == s.fK1 && |
| 326 | fK2 == s.fK2 && |
| 327 | fK3 == s.fK3 && |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 328 | fK4 == s.fK4 && |
bsalomon | 420d7e9 | 2014-10-16 09:18:09 -0700 | [diff] [blame] | 329 | fEnforcePMColor == s.fEnforcePMColor; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 330 | } |
| 331 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 332 | const GrBackendFragmentProcessorFactory& GrArithmeticEffect::getFactory() const { |
| 333 | return GrTBackendFragmentProcessorFactory<GrArithmeticEffect>::getInstance(); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 334 | } |
| 335 | |
egdaniel | 605dd0f | 2014-11-12 08:35:25 -0800 | [diff] [blame] | 336 | void GrArithmeticEffect::onComputeInvariantOutput(GrInvariantOutput* inout) const { |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 337 | // TODO: optimize this |
egdaniel | 605dd0f | 2014-11-12 08:35:25 -0800 | [diff] [blame] | 338 | inout->setToUnknown(GrInvariantOutput::kWill_ReadInput); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 339 | } |
| 340 | |
| 341 | /////////////////////////////////////////////////////////////////////////////// |
| 342 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 343 | GrGLArithmeticEffect::GrGLArithmeticEffect(const GrBackendProcessorFactory& factory, |
| 344 | const GrProcessor&) |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 345 | : INHERITED(factory), |
| 346 | fEnforcePMColor(true) { |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 347 | } |
| 348 | |
| 349 | GrGLArithmeticEffect::~GrGLArithmeticEffect() { |
| 350 | } |
| 351 | |
joshualitt | 1598899 | 2014-10-09 15:04:05 -0700 | [diff] [blame] | 352 | void GrGLArithmeticEffect::emitCode(GrGLFPBuilder* builder, |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 353 | const GrFragmentProcessor& fp, |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 354 | const char* outputColor, |
| 355 | const char* inputColor, |
bsalomon@google.com | 77af680 | 2013-10-02 13:04:56 +0000 | [diff] [blame] | 356 | const TransformedCoordsArray& coords, |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 357 | const TextureSamplerArray& samplers) { |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 358 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 359 | GrTexture* backgroundTex = fp.cast<GrArithmeticEffect>().backgroundTexture(); |
joshualitt | 1598899 | 2014-10-09 15:04:05 -0700 | [diff] [blame] | 360 | GrGLFPFragmentBuilder* fsBuilder = builder->getFragmentShaderBuilder(); |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 361 | const char* dstColor; |
| 362 | if (backgroundTex) { |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 363 | fsBuilder->codeAppend("\t\tvec4 bgColor = "); |
joshualitt | 23e280d | 2014-09-18 12:26:38 -0700 | [diff] [blame] | 364 | fsBuilder->appendTextureLookup(samplers[0], coords[0].c_str(), coords[0].getType()); |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 365 | fsBuilder->codeAppendf(";\n"); |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 366 | dstColor = "bgColor"; |
| 367 | } else { |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 368 | dstColor = fsBuilder->dstColor(); |
senorblanco@chromium.org | 86fc266 | 2013-05-31 17:49:12 +0000 | [diff] [blame] | 369 | } |
| 370 | |
bsalomon | 49f085d | 2014-09-05 13:34:00 -0700 | [diff] [blame] | 371 | SkASSERT(dstColor); |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 372 | fKUni = builder->addUniform(GrGLProgramBuilder::kFragment_Visibility, |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 373 | kVec4f_GrSLType, "k"); |
| 374 | const char* kUni = builder->getUniformCStr(fKUni); |
| 375 | |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 376 | // We don't try to optimize for this case at all |
| 377 | if (NULL == inputColor) { |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 378 | fsBuilder->codeAppendf("\t\tconst vec4 src = vec4(1);\n"); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 379 | } else { |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 380 | fsBuilder->codeAppendf("\t\tvec4 src = %s;\n", inputColor); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 381 | if (gUseUnpremul) { |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 382 | fsBuilder->codeAppendf("\t\tsrc.rgb = clamp(src.rgb / src.a, 0.0, 1.0);\n"); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 383 | } |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 384 | } |
| 385 | |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 386 | fsBuilder->codeAppendf("\t\tvec4 dst = %s;\n", dstColor); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 387 | if (gUseUnpremul) { |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 388 | fsBuilder->codeAppendf("\t\tdst.rgb = clamp(dst.rgb / dst.a, 0.0, 1.0);\n"); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 389 | } |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 390 | |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 391 | fsBuilder->codeAppendf("\t\t%s = %s.x * src * dst + %s.y * src + %s.z * dst + %s.w;\n", outputColor, kUni, kUni, kUni, kUni); |
| 392 | fsBuilder->codeAppendf("\t\t%s = clamp(%s, 0.0, 1.0);\n", outputColor, outputColor); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 393 | if (gUseUnpremul) { |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 394 | fsBuilder->codeAppendf("\t\t%s.rgb *= %s.a;\n", outputColor, outputColor); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 395 | } else if (fEnforcePMColor) { |
joshualitt | 30ba436 | 2014-08-21 20:18:45 -0700 | [diff] [blame] | 396 | fsBuilder->codeAppendf("\t\t%s.rgb = min(%s.rgb, %s.a);\n", outputColor, outputColor, outputColor); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 397 | } |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 398 | } |
| 399 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 400 | void GrGLArithmeticEffect::setData(const GrGLProgramDataManager& pdman, |
| 401 | const GrProcessor& processor) { |
| 402 | const GrArithmeticEffect& arith = processor.cast<GrArithmeticEffect>(); |
kkinnunen | 7510b22 | 2014-07-30 00:04:16 -0700 | [diff] [blame] | 403 | pdman.set4f(fKUni, arith.k1(), arith.k2(), arith.k3(), arith.k4()); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 404 | fEnforcePMColor = arith.enforcePMColor(); |
| 405 | } |
| 406 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 407 | void GrGLArithmeticEffect::GenKey(const GrProcessor& processor, |
| 408 | const GrGLCaps&, GrProcessorKeyBuilder* b) { |
| 409 | const GrArithmeticEffect& arith = processor.cast<GrArithmeticEffect>(); |
bsalomon | 63e99f7 | 2014-07-21 08:03:14 -0700 | [diff] [blame] | 410 | uint32_t key = arith.enforcePMColor() ? 1 : 0; |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 411 | if (arith.backgroundTexture()) { |
| 412 | key |= 2; |
| 413 | } |
bsalomon | 63e99f7 | 2014-07-21 08:03:14 -0700 | [diff] [blame] | 414 | b->add32(key); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 415 | } |
| 416 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 417 | GrFragmentProcessor* GrArithmeticEffect::TestCreate(SkRandom* rand, |
| 418 | GrContext*, |
| 419 | const GrDrawTargetCaps&, |
| 420 | GrTexture*[]) { |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 421 | float k1 = rand->nextF(); |
| 422 | float k2 = rand->nextF(); |
| 423 | float k3 = rand->nextF(); |
| 424 | float k4 = rand->nextF(); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 425 | bool enforcePMColor = rand->nextBool(); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 426 | |
bsalomon | 55fad7a | 2014-07-08 07:34:20 -0700 | [diff] [blame] | 427 | return SkNEW_ARGS(GrArithmeticEffect, (k1, k2, k3, k4, enforcePMColor, NULL)); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 428 | } |
| 429 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 430 | GR_DEFINE_FRAGMENT_PROCESSOR_TEST(GrArithmeticEffect); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 431 | |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 432 | bool SkArithmeticMode_scalar::asFragmentProcessor(GrFragmentProcessor** fp, |
| 433 | GrTexture* background) const { |
| 434 | if (fp) { |
| 435 | *fp = GrArithmeticEffect::Create(SkScalarToFloat(fK[0]), |
| 436 | SkScalarToFloat(fK[1]), |
| 437 | SkScalarToFloat(fK[2]), |
| 438 | SkScalarToFloat(fK[3]), |
| 439 | fEnforcePMColor, |
| 440 | background); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 441 | } |
| 442 | return true; |
| 443 | } |
| 444 | |
| 445 | #endif |
| 446 | |
| 447 | SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_START(SkArithmeticMode) |
| 448 | SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkArithmeticMode_scalar) |
| 449 | SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_END |