skia2/tools/RuntimeBlendUtils.cpp

371 lines
15 KiB
C++
Raw Normal View History

/*
* Copyright 2021 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include "include/core/SkBlendMode.h"
#include "include/core/SkRefCnt.h"
#include "include/core/SkString.h"
#include "include/effects/SkRuntimeEffect.h"
static sk_sp<SkBlender> blender_func(std::initializer_list<const char*> code) {
SkString concat;
for (const char* part : code) {
concat.append(part);
}
SkRuntimeEffect::Result result = SkRuntimeEffect::MakeForBlender(std::move(concat));
SkASSERTF(result.effect, "%s", result.errorText.c_str());
return result.effect->makeBlender(/*uniforms=*/nullptr);
}
static sk_sp<SkBlender> blender_expr(const char* returnExpr) {
return blender_func({"half4 main(half4 src, half4 dst) {\n"
" return ", returnExpr, ";\n"
"}"});
}
sk_sp<SkBlender> GetRuntimeBlendForBlendMode(SkBlendMode mode) {
static constexpr char kGuardedDivide[] = R"(
half _guarded_divide(half n, half d) {
return n/(d + 0.00000001);
}
half3 _guarded_divide(half3 n, half d) {
return n/(d + 0.00000001);
}
)";
static constexpr char kHSLCSharedCode[] = R"(
half _blend_color_luminance(half3 color) { return dot(half3(0.3, 0.59, 0.11), color); }
half3 _blend_set_color_luminance(half3 hueSatColor, half alpha, half3 lumColor) {
half lum = _blend_color_luminance(lumColor);
half3 result = lum - _blend_color_luminance(hueSatColor) + hueSatColor;
half minComp = min(min(result.r, result.g), result.b);
half maxComp = max(max(result.r, result.g), result.b);
if (minComp < 0 && lum != minComp) {
result = lum + (result - lum) * _guarded_divide(lum, (lum - minComp));
}
if (maxComp > alpha && maxComp != lum) {
return lum + _guarded_divide((result - lum) * (alpha - lum), (maxComp - lum));
} else {
return result;
}
}
half _blend_color_saturation(half3 color) {
return max(max(color.r, color.g), color.b) - min(min(color.r, color.g), color.b);
}
half3 _blend_set_color_saturation_helper(half3 minMidMax, half sat) {
if (minMidMax.r < minMidMax.b) {
return half3(0,
_guarded_divide(sat*(minMidMax.g - minMidMax.r),
(minMidMax.b - minMidMax.r)),
sat);
} else {
return half3(0);
}
}
half3 _blend_set_color_saturation(half3 hueLumColor, half3 satColor) {
half sat = _blend_color_saturation(satColor);
if (hueLumColor.r <= hueLumColor.g) {
if (hueLumColor.g <= hueLumColor.b) {
return _blend_set_color_saturation_helper(hueLumColor.rgb, sat);
} else if (hueLumColor.r <= hueLumColor.b) {
return _blend_set_color_saturation_helper(hueLumColor.rbg, sat).rbg;
} else {
return _blend_set_color_saturation_helper(hueLumColor.brg, sat).gbr;
}
} else if (hueLumColor.r <= hueLumColor.b) {
return _blend_set_color_saturation_helper(hueLumColor.grb, sat).grb;
} else if (hueLumColor.g <= hueLumColor.b) {
return _blend_set_color_saturation_helper(hueLumColor.gbr, sat).brg;
} else {
return _blend_set_color_saturation_helper(hueLumColor.bgr, sat).bgr;
}
}
)";
switch (mode) {
case SkBlendMode::kClear: {
static auto blender = blender_expr("half4(0)");
return blender;
}
case SkBlendMode::kSrc: {
static auto blender = blender_expr("src");
return blender;
}
case SkBlendMode::kDst: {
static auto blender = blender_expr("dst");
return blender;
}
case SkBlendMode::kSrcOver: {
static auto blender = blender_expr("src + (1 - src.a)*dst");
return blender;
}
case SkBlendMode::kDstOver: {
static auto blender = blender_expr("(1 - dst.a)*src + dst");
return blender;
}
case SkBlendMode::kSrcIn: {
static auto blender = blender_expr("(src == half4(0) ? half4(0) : src*dst.a)");
return blender;
}
case SkBlendMode::kDstIn: {
static auto blender = blender_expr("(dst == half4(0) ? half4(0) : dst*src.a)");
return blender;
}
case SkBlendMode::kSrcOut: {
static auto blender = blender_expr("(1 - dst.a)*src");
return blender;
}
case SkBlendMode::kDstOut: {
static auto blender = blender_expr("(1 - src.a)*dst");
return blender;
}
case SkBlendMode::kSrcATop: {
static auto blender = blender_expr("dst.a*src + (1 - src.a)*dst");
return blender;
}
case SkBlendMode::kDstATop: {
static auto blender = blender_expr("(1 - dst.a) * src + src.a*dst");
return blender;
}
case SkBlendMode::kXor: {
static auto blender = blender_expr("(1 - dst.a)*src + (1 - src.a)*dst");
return blender;
}
case SkBlendMode::kPlus: {
static auto blender = blender_expr("min(src + dst, 1)");
return blender;
}
case SkBlendMode::kModulate: {
static auto blender = blender_expr("src*dst");
return blender;
}
case SkBlendMode::kScreen: {
static auto blender = blender_expr("src + (1 - src)*dst");
return blender;
}
case SkBlendMode::kOverlay: {
static auto blender = blender_func({R"(
half _blend_overlay_component(half2 s, half2 d) {
return (2*d.x <= d.y)
? 2*s.x*d.x
: s.y*d.y - 2*(d.y - d.x)*(s.y - s.x);
}
half4 main(half4 src, half4 dst) {
half4 result = half4(_blend_overlay_component(src.ra, dst.ra),
_blend_overlay_component(src.ga, dst.ga),
_blend_overlay_component(src.ba, dst.ba),
src.a + (1 - src.a)*dst.a);
result.rgb += dst.rgb*(1 - src.a) + src.rgb*(1 - dst.a);
return result;
}
)"});
return blender;
}
case SkBlendMode::kDarken: {
static auto blender = blender_func({R"(
half4 main(half4 src, half4 dst) {
half4 result = src + (1 - src.a)*dst;
result.rgb = min(result.rgb, (1 - dst.a)*src.rgb + dst.rgb);
return result;
}
)"});
return blender;
}
case SkBlendMode::kLighten: {
static auto blender = blender_func({R"(
half4 main(half4 src, half4 dst) {
half4 result = src + (1 - src.a)*dst;
result.rgb = max(result.rgb, (1 - dst.a)*src.rgb + dst.rgb);
return result;
}
)"});
return blender;
}
case SkBlendMode::kColorDodge: {
static auto blender = blender_func({kGuardedDivide, R"(
half _color_dodge_component(half2 s, half2 d) {
if (d.x == 0) {
return s.x*(1 - d.y);
} else {
half delta = s.y - s.x;
if (delta == 0) {
return s.y*d.y + s.x*(1 - d.y) + d.x*(1 - s.y);
} else {
delta = min(d.y, _guarded_divide(d.x*s.y, delta));
return delta*s.y + s.x*(1 - d.y) + d.x*(1 - s.y);
}
}
}
half4 main(half4 src, half4 dst) {
return half4(_color_dodge_component(src.ra, dst.ra),
_color_dodge_component(src.ga, dst.ga),
_color_dodge_component(src.ba, dst.ba),
src.a + (1 - src.a)*dst.a);
}
)"});
return blender;
}
case SkBlendMode::kColorBurn: {
static auto blender = blender_func({kGuardedDivide, R"(
half _color_burn_component(half2 s, half2 d) {
if (d.y == d.x) {
return s.y*d.y + s.x*(1 - d.y) + d.x*(1 - s.y);
} else if (s.x == 0) {
return d.x*(1 - s.y);
} else {
half delta = max(0, d.y - _guarded_divide((d.y - d.x)*s.y, s.x));
return delta*s.y + s.x*(1 - d.y) + d.x*(1 - s.y);
}
}
half4 main(half4 src, half4 dst) {
return half4(_color_burn_component(src.ra, dst.ra),
_color_burn_component(src.ga, dst.ga),
_color_burn_component(src.ba, dst.ba),
src.a + (1 - src.a)*dst.a);
}
)"});
return blender;
}
case SkBlendMode::kHardLight: {
static auto blender = blender_func({R"(
half _blend_overlay_component(half2 s, half2 d) {
return (2*d.x <= d.y)
? 2*s.x*d.x
: s.y*d.y - 2*(d.y - d.x)*(s.y - s.x);
}
half4 main(half4 dst, half4 src) {
half4 result = half4(_blend_overlay_component(src.ra, dst.ra),
_blend_overlay_component(src.ga, dst.ga),
_blend_overlay_component(src.ba, dst.ba),
src.a + (1 - src.a)*dst.a);
result.rgb += dst.rgb*(1 - src.a) + src.rgb*(1 - dst.a);
return result;
}
)"});
return blender;
}
case SkBlendMode::kSoftLight: {
static auto blender = blender_func({kGuardedDivide, R"(
half _soft_light_component(half2 s, half2 d) {
if (2*s.x <= s.y) {
return _guarded_divide(d.x*d.x*(s.y - 2*s.x), d.y) +
(1 - d.y)*s.x + d.x*(-s.y + 2*s.x + 1);
} else if (4.0 * d.x <= d.y) {
half DSqd = d.x*d.x;
half DCub = DSqd*d.x;
half DaSqd = d.y*d.y;
half DaCub = DaSqd*d.y;
return _guarded_divide(DaSqd*(s.x - d.x*(3*s.y - 6*s.x - 1))
+ 12*d.y*DSqd*(s.y - 2*s.x)
- 16*DCub * (s.y - 2*s.x) - DaCub*s.x, DaSqd);
} else {
return d.x*(s.y - 2*s.x + 1) + s.x - sqrt(d.y*d.x)*(s.y - 2*s.x) - d.y*s.x;
}
}
half4 main(half4 src, half4 dst) {
return (dst.a == 0) ? src : half4(_soft_light_component(src.ra, dst.ra),
_soft_light_component(src.ga, dst.ga),
_soft_light_component(src.ba, dst.ba),
src.a + (1 - src.a)*dst.a);
}
)"});
return blender;
}
case SkBlendMode::kDifference: {
static auto blender = blender_func({R"(
half4 main(half4 src, half4 dst) {
return half4(src.rgb + dst.rgb - 2*min(src.rgb*dst.a, dst.rgb*src.a),
src.a + (1 - src.a)*dst.a);
}
)"});
return blender;
}
case SkBlendMode::kExclusion: {
static auto blender = blender_func({R"(
half4 main(half4 src, half4 dst) {
return half4(dst.rgb + src.rgb - 2*dst.rgb*src.rgb, src.a + (1 - src.a)*dst.a);
}
)"});
return blender;
}
case SkBlendMode::kMultiply: {
static auto blender = blender_func({R"(
half4 main(half4 src, half4 dst) {
return half4((1 - src.a)*dst.rgb + (1 - dst.a)*src.rgb + src.rgb*dst.rgb,
src.a + (1 - src.a)*dst.a);
}
)"});
return blender;
}
case SkBlendMode::kHue: {
static auto blender = blender_func({kGuardedDivide, kHSLCSharedCode, R"(
half4 main(half4 src, half4 dst) {
half alpha = dst.a*src.a;
half3 sda = src.rgb*dst.a;
half3 dsa = dst.rgb*src.a;
return half4(_blend_set_color_luminance(_blend_set_color_saturation(sda, dsa),
alpha, dsa) +
dst.rgb - dsa + src.rgb - sda,
src.a + dst.a - alpha);
}
)"});
return blender;
}
case SkBlendMode::kSaturation: {
static auto blender = blender_func({kGuardedDivide, kHSLCSharedCode, R"(
half4 main(half4 src, half4 dst) {
half alpha = dst.a*src.a;
half3 sda = src.rgb*dst.a;
half3 dsa = dst.rgb*src.a;
return half4(_blend_set_color_luminance(_blend_set_color_saturation(dsa, sda),
alpha, dsa) +
dst.rgb - dsa + src.rgb - sda,
src.a + dst.a - alpha);
}
)"});
return blender;
}
case SkBlendMode::kColor: {
static auto blender = blender_func({kGuardedDivide, kHSLCSharedCode, R"(
half4 main(half4 src, half4 dst) {
half alpha = dst.a*src.a;
half3 sda = src.rgb*dst.a;
half3 dsa = dst.rgb*src.a;
return half4(_blend_set_color_luminance(sda, alpha, dsa) + dst.rgb
- dsa + src.rgb - sda, src.a + dst.a - alpha);
}
)"});
return blender;
}
case SkBlendMode::kLuminosity: {
static auto blender = blender_func({kGuardedDivide, kHSLCSharedCode, R"(
half4 main(half4 src, half4 dst) {
half alpha = dst.a*src.a;
half3 sda = src.rgb*dst.a;
half3 dsa = dst.rgb*src.a;
return half4(_blend_set_color_luminance(dsa, alpha, sda) + dst.rgb - dsa
+ src.rgb - sda, src.a + dst.a - alpha);
}
)"});
return blender;
}
default:
SkDEBUGFAILF("unrecognized blend mode %d", (int)mode);
return nullptr;
}
}