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 |
egdaniel | 16a04b8 | 2015-01-14 10:49:18 -0800 | [diff] [blame] | 15 | #include "SkArithmeticMode_gpu.h" |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 16 | #endif |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 17 | |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 18 | static const bool gUseUnpremul = false; |
| 19 | |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 20 | class SkArithmeticMode_scalar : public SkXfermode { |
| 21 | public: |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 22 | static SkArithmeticMode_scalar* Create(SkScalar k1, SkScalar k2, SkScalar k3, SkScalar k4, |
| 23 | bool enforcePMColor) { |
| 24 | return SkNEW_ARGS(SkArithmeticMode_scalar, (k1, k2, k3, k4, enforcePMColor)); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 25 | } |
| 26 | |
| 27 | virtual void xfer32(SkPMColor dst[], const SkPMColor src[], int count, |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 28 | const SkAlpha aa[]) const override; |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 29 | |
commit-bot@chromium.org | 0f10f7b | 2014-03-13 18:02:17 +0000 | [diff] [blame] | 30 | SK_TO_STRING_OVERRIDE() |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 31 | SK_DECLARE_PUBLIC_FLATTENABLE_DESERIALIZATION_PROCS(SkArithmeticMode_scalar) |
| 32 | |
| 33 | #if SK_SUPPORT_GPU |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 34 | bool asFragmentProcessor(GrFragmentProcessor**, GrTexture* background) const override; |
egdaniel | f351aa3 | 2015-01-14 12:53:01 -0800 | [diff] [blame] | 35 | |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 36 | bool asXPFactory(GrXPFactory**) const override; |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 37 | #endif |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 38 | |
| 39 | private: |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 40 | 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] | 41 | fK[0] = k1; |
| 42 | fK[1] = k2; |
| 43 | fK[2] = k3; |
| 44 | fK[3] = k4; |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 45 | fEnforcePMColor = enforcePMColor; |
commit-bot@chromium.org | 0a2bf90 | 2014-02-20 20:40:19 +0000 | [diff] [blame] | 46 | } |
| 47 | |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 48 | void flatten(SkWriteBuffer& buffer) const override { |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 49 | buffer.writeScalar(fK[0]); |
| 50 | buffer.writeScalar(fK[1]); |
| 51 | buffer.writeScalar(fK[2]); |
| 52 | buffer.writeScalar(fK[3]); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 53 | buffer.writeBool(fEnforcePMColor); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 54 | } |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 55 | SkScalar fK[4]; |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 56 | bool fEnforcePMColor; |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 57 | |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 58 | friend class SkArithmeticMode; |
| 59 | |
skia.committer@gmail.com | 98ded84 | 2013-01-23 07:06:17 +0000 | [diff] [blame] | 60 | typedef SkXfermode INHERITED; |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 61 | }; |
| 62 | |
reed | 9fa60da | 2014-08-21 07:59:51 -0700 | [diff] [blame] | 63 | SkFlattenable* SkArithmeticMode_scalar::CreateProc(SkReadBuffer& buffer) { |
| 64 | const SkScalar k1 = buffer.readScalar(); |
| 65 | const SkScalar k2 = buffer.readScalar(); |
| 66 | const SkScalar k3 = buffer.readScalar(); |
| 67 | const SkScalar k4 = buffer.readScalar(); |
| 68 | const bool enforcePMColor = buffer.readBool(); |
| 69 | return Create(k1, k2, k3, k4, enforcePMColor); |
| 70 | } |
| 71 | |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 72 | static int pinToByte(int value) { |
| 73 | if (value < 0) { |
| 74 | value = 0; |
| 75 | } else if (value > 255) { |
| 76 | value = 255; |
| 77 | } |
| 78 | return value; |
| 79 | } |
| 80 | |
| 81 | static int arith(SkScalar k1, SkScalar k2, SkScalar k3, SkScalar k4, |
| 82 | int src, int dst) { |
| 83 | SkScalar result = SkScalarMul(k1, src * dst) + |
| 84 | SkScalarMul(k2, src) + |
| 85 | SkScalarMul(k3, dst) + |
| 86 | k4; |
| 87 | int res = SkScalarRoundToInt(result); |
| 88 | return pinToByte(res); |
| 89 | } |
| 90 | |
| 91 | static int blend(int src, int dst, int scale) { |
| 92 | return dst + ((src - dst) * scale >> 8); |
| 93 | } |
| 94 | |
| 95 | static bool needsUnpremul(int alpha) { |
| 96 | return 0 != alpha && 0xFF != alpha; |
| 97 | } |
| 98 | |
| 99 | void SkArithmeticMode_scalar::xfer32(SkPMColor dst[], const SkPMColor src[], |
reed@google.com | 30da745 | 2012-12-17 19:55:24 +0000 | [diff] [blame] | 100 | int count, const SkAlpha aaCoverage[]) const { |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 101 | SkScalar k1 = fK[0] / 255; |
| 102 | SkScalar k2 = fK[1]; |
| 103 | SkScalar k3 = fK[2]; |
| 104 | SkScalar k4 = fK[3] * 255; |
| 105 | |
| 106 | for (int i = 0; i < count; ++i) { |
| 107 | if ((NULL == aaCoverage) || aaCoverage[i]) { |
| 108 | SkPMColor sc = src[i]; |
| 109 | SkPMColor dc = dst[i]; |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 110 | |
| 111 | int a, r, g, b; |
| 112 | |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 113 | if (gUseUnpremul) { |
| 114 | int sa = SkGetPackedA32(sc); |
| 115 | int da = SkGetPackedA32(dc); |
| 116 | |
| 117 | int srcNeedsUnpremul = needsUnpremul(sa); |
| 118 | int dstNeedsUnpremul = needsUnpremul(da); |
| 119 | |
| 120 | if (!srcNeedsUnpremul && !dstNeedsUnpremul) { |
| 121 | a = arith(k1, k2, k3, k4, sa, da); |
| 122 | r = arith(k1, k2, k3, k4, SkGetPackedR32(sc), SkGetPackedR32(dc)); |
| 123 | g = arith(k1, k2, k3, k4, SkGetPackedG32(sc), SkGetPackedG32(dc)); |
| 124 | b = arith(k1, k2, k3, k4, SkGetPackedB32(sc), SkGetPackedB32(dc)); |
| 125 | } else { |
| 126 | int sr = SkGetPackedR32(sc); |
| 127 | int sg = SkGetPackedG32(sc); |
| 128 | int sb = SkGetPackedB32(sc); |
| 129 | if (srcNeedsUnpremul) { |
| 130 | SkUnPreMultiply::Scale scale = SkUnPreMultiply::GetScale(sa); |
| 131 | sr = SkUnPreMultiply::ApplyScale(scale, sr); |
| 132 | sg = SkUnPreMultiply::ApplyScale(scale, sg); |
| 133 | sb = SkUnPreMultiply::ApplyScale(scale, sb); |
| 134 | } |
| 135 | |
| 136 | int dr = SkGetPackedR32(dc); |
| 137 | int dg = SkGetPackedG32(dc); |
| 138 | int db = SkGetPackedB32(dc); |
| 139 | if (dstNeedsUnpremul) { |
| 140 | SkUnPreMultiply::Scale scale = SkUnPreMultiply::GetScale(da); |
| 141 | dr = SkUnPreMultiply::ApplyScale(scale, dr); |
| 142 | dg = SkUnPreMultiply::ApplyScale(scale, dg); |
| 143 | db = SkUnPreMultiply::ApplyScale(scale, db); |
| 144 | } |
| 145 | |
| 146 | a = arith(k1, k2, k3, k4, sa, da); |
| 147 | r = arith(k1, k2, k3, k4, sr, dr); |
| 148 | g = arith(k1, k2, k3, k4, sg, dg); |
| 149 | b = arith(k1, k2, k3, k4, sb, db); |
| 150 | } |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 151 | } else { |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 152 | a = arith(k1, k2, k3, k4, SkGetPackedA32(sc), SkGetPackedA32(dc)); |
| 153 | r = arith(k1, k2, k3, k4, SkGetPackedR32(sc), SkGetPackedR32(dc)); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 154 | g = arith(k1, k2, k3, k4, SkGetPackedG32(sc), SkGetPackedG32(dc)); |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 155 | b = arith(k1, k2, k3, k4, SkGetPackedB32(sc), SkGetPackedB32(dc)); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 156 | if (fEnforcePMColor) { |
| 157 | r = SkMin32(r, a); |
| 158 | g = SkMin32(g, a); |
| 159 | b = SkMin32(b, a); |
| 160 | } |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 161 | } |
| 162 | |
| 163 | // apply antialias coverage if necessary |
| 164 | if (aaCoverage && 0xFF != aaCoverage[i]) { |
| 165 | int scale = aaCoverage[i] + (aaCoverage[i] >> 7); |
| 166 | a = blend(a, SkGetPackedA32(sc), scale); |
| 167 | r = blend(r, SkGetPackedR32(sc), scale); |
| 168 | g = blend(g, SkGetPackedG32(sc), scale); |
| 169 | b = blend(b, SkGetPackedB32(sc), scale); |
| 170 | } |
| 171 | |
mike@reedtribe.org | 0e330ae | 2012-08-12 19:08:41 +0000 | [diff] [blame] | 172 | // turn the result back into premul |
senorblanco@chromium.org | b1b3d51 | 2013-06-12 22:15:43 +0000 | [diff] [blame] | 173 | if (gUseUnpremul && (0xFF != a)) { |
mike@reedtribe.org | 0e330ae | 2012-08-12 19:08:41 +0000 | [diff] [blame] | 174 | int scale = a + (a >> 7); |
| 175 | r = SkAlphaMul(r, scale); |
| 176 | g = SkAlphaMul(g, scale); |
| 177 | b = SkAlphaMul(b, scale); |
| 178 | } |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 179 | 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] | 180 | } |
| 181 | } |
| 182 | } |
| 183 | |
commit-bot@chromium.org | 0f10f7b | 2014-03-13 18:02:17 +0000 | [diff] [blame] | 184 | #ifndef SK_IGNORE_TO_STRING |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 185 | void SkArithmeticMode_scalar::toString(SkString* str) const { |
| 186 | str->append("SkArithmeticMode_scalar: "); |
| 187 | for (int i = 0; i < 4; ++i) { |
| 188 | str->appendScalar(fK[i]); |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 189 | str->append(" "); |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 190 | } |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 191 | str->appendS32(fEnforcePMColor ? 1 : 0); |
robertphillips@google.com | b83b6b4 | 2013-01-22 14:32:09 +0000 | [diff] [blame] | 192 | } |
| 193 | #endif |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 194 | |
| 195 | /////////////////////////////////////////////////////////////////////////////// |
| 196 | |
| 197 | static bool fitsInBits(SkScalar x, int bits) { |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 198 | return SkScalarAbs(x) < (1 << (bits - 1)); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 199 | } |
| 200 | |
caryclark@google.com | 383d5d4 | 2012-06-06 12:09:18 +0000 | [diff] [blame] | 201 | #if 0 // UNUSED |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 202 | static int32_t toDot8(SkScalar x) { |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 203 | return (int32_t)(x * 256); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 204 | } |
caryclark@google.com | 383d5d4 | 2012-06-06 12:09:18 +0000 | [diff] [blame] | 205 | #endif |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 206 | |
| 207 | SkXfermode* SkArithmeticMode::Create(SkScalar k1, SkScalar k2, |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 208 | SkScalar k3, SkScalar k4, |
| 209 | bool enforcePMColor) { |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 210 | if (fitsInBits(k1, 8) && fitsInBits(k2, 16) && |
| 211 | fitsInBits(k2, 16) && fitsInBits(k2, 24)) { |
| 212 | |
caryclark@google.com | 383d5d4 | 2012-06-06 12:09:18 +0000 | [diff] [blame] | 213 | #if 0 // UNUSED |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 214 | int32_t i1 = toDot8(k1); |
| 215 | int32_t i2 = toDot8(k2); |
| 216 | int32_t i3 = toDot8(k3); |
| 217 | int32_t i4 = toDot8(k4); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 218 | if (i1) { |
| 219 | return SkNEW_ARGS(SkArithmeticMode_quad, (i1, i2, i3, i4)); |
| 220 | } |
| 221 | if (0 == i2) { |
| 222 | return SkNEW_ARGS(SkArithmeticMode_dst, (i3, i4)); |
| 223 | } |
| 224 | if (0 == i3) { |
| 225 | return SkNEW_ARGS(SkArithmeticMode_src, (i2, i4)); |
| 226 | } |
| 227 | return SkNEW_ARGS(SkArithmeticMode_linear, (i2, i3, i4)); |
| 228 | #endif |
| 229 | } |
commit-bot@chromium.org | 51a11b7 | 2014-04-02 19:32:15 +0000 | [diff] [blame] | 230 | return SkArithmeticMode_scalar::Create(k1, k2, k3, k4, enforcePMColor); |
mike@reedtribe.org | e51755f | 2011-12-10 19:36:56 +0000 | [diff] [blame] | 231 | } |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 232 | |
| 233 | |
| 234 | ////////////////////////////////////////////////////////////////////////////// |
| 235 | |
| 236 | #if SK_SUPPORT_GPU |
joshualitt | b0a8a37 | 2014-09-23 09:50:21 -0700 | [diff] [blame] | 237 | bool SkArithmeticMode_scalar::asFragmentProcessor(GrFragmentProcessor** fp, |
| 238 | GrTexture* background) const { |
| 239 | if (fp) { |
egdaniel | 16a04b8 | 2015-01-14 10:49:18 -0800 | [diff] [blame] | 240 | *fp = GrArithmeticFP::Create(SkScalarToFloat(fK[0]), |
| 241 | SkScalarToFloat(fK[1]), |
| 242 | SkScalarToFloat(fK[2]), |
| 243 | SkScalarToFloat(fK[3]), |
| 244 | fEnforcePMColor, |
| 245 | background); |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 246 | } |
| 247 | return true; |
| 248 | } |
| 249 | |
egdaniel | f351aa3 | 2015-01-14 12:53:01 -0800 | [diff] [blame] | 250 | bool SkArithmeticMode_scalar::asXPFactory(GrXPFactory** xpf) const { |
| 251 | if (xpf) { |
| 252 | *xpf = GrArithmeticXPFactory::Create(SkScalarToFloat(fK[0]), |
| 253 | SkScalarToFloat(fK[1]), |
| 254 | SkScalarToFloat(fK[2]), |
| 255 | SkScalarToFloat(fK[3]), |
| 256 | fEnforcePMColor); |
| 257 | } |
| 258 | return true; |
| 259 | } |
| 260 | |
senorblanco@chromium.org | 9a6eb0e | 2013-05-29 20:55:09 +0000 | [diff] [blame] | 261 | #endif |
| 262 | |
| 263 | SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_START(SkArithmeticMode) |
| 264 | SK_DEFINE_FLATTENABLE_REGISTRAR_ENTRY(SkArithmeticMode_scalar) |
| 265 | SK_DEFINE_FLATTENABLE_REGISTRAR_GROUP_END |