Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2019 Google LLC |
| 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 "gm/gm.h" |
| 9 | #include "include/core/SkCanvas.h" |
| 10 | #include "include/core/SkData.h" |
| 11 | #include "include/core/SkFont.h" |
| 12 | #include "include/core/SkPaint.h" |
| 13 | #include "include/core/SkSize.h" |
| 14 | #include "include/core/SkString.h" |
| 15 | #include "include/core/SkSurface.h" |
| 16 | #include "include/effects/SkGradientShader.h" |
| 17 | #include "include/effects/SkImageFilters.h" |
| 18 | #include "include/effects/SkRuntimeEffect.h" |
| 19 | #include "include/utils/SkRandom.h" |
| 20 | #include "tools/Resources.h" |
| 21 | #include "tools/ToolUtils.h" |
| 22 | |
Brian Osman | 24000785 | 2020-11-11 17:04:57 -0500 | [diff] [blame] | 23 | static constexpr int kBoxSize = 100; |
| 24 | static constexpr int kPadding = 5; |
| 25 | static constexpr int kLabelHeight = 15; |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 26 | |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 27 | /* |
| 28 | Test cases are inserted into the middle of this shader. The pasted expression is expected to |
| 29 | produce a single float. It can reference: |
| 30 | |
| 31 | 'x' : float in [xMin, xMax] |
Brian Osman | 9ba7a24 | 2020-11-13 16:31:37 -0500 | [diff] [blame^] | 32 | 'p' : float2 in [xMin, xMax] Lerps from (xMax, xMin) to (xMin, xMax) |
| 33 | (helpful for intrinsics with a mix of scalar/vector params) |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 34 | 'v1' : float2(1) |
| 35 | 'v2' : float2(2) |
| 36 | */ |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 37 | static SkString make_unary_sksl_1d(const char* fn) { |
| 38 | return SkStringPrintf( |
| 39 | "uniform float xScale; uniform float xBias;" |
| 40 | "uniform float yScale; uniform float yBias;" |
| 41 | "half4 main(float2 p) {" |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 42 | " float2 v1 = float2(1);" |
| 43 | " float2 v2 = float2(2);" |
Brian Osman | 9ba7a24 | 2020-11-13 16:31:37 -0500 | [diff] [blame^] | 44 | " p = float2(p.x, 1 - p.x) * xScale + xBias;" |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 45 | " float x = p.x;" |
Brian Osman | 24000785 | 2020-11-11 17:04:57 -0500 | [diff] [blame] | 46 | " float y = %s * yScale + yBias;" |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 47 | " return y.xxx1;" |
| 48 | "}", |
| 49 | fn); |
| 50 | } |
| 51 | |
| 52 | // Draws one row of boxes, then advances the canvas translation vertically |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 53 | static void plot(SkCanvas* canvas, |
| 54 | const char* fn, |
| 55 | float xMin, |
| 56 | float xMax, |
| 57 | float yMin, |
| 58 | float yMax, |
| 59 | const char* label = nullptr) { |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 60 | canvas->save(); |
| 61 | |
| 62 | SkFont font(ToolUtils::create_portable_typeface()); |
| 63 | SkPaint p(SkColors::kBlack); |
| 64 | SkRect bounds; |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 65 | if (!label) { |
| 66 | label = fn; |
| 67 | } |
| 68 | font.measureText(label, strlen(label), SkTextEncoding::kUTF8, &bounds); |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 69 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 70 | canvas->drawSimpleText(label, strlen(label), SkTextEncoding::kUTF8, |
Brian Osman | 24000785 | 2020-11-11 17:04:57 -0500 | [diff] [blame] | 71 | (kBoxSize - bounds.width()) * 0.5f, |
| 72 | (kLabelHeight + bounds.height()) * 0.5f, font, p); |
| 73 | canvas->translate(0, kLabelHeight); |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 74 | |
| 75 | const float xScale = xMax - xMin, |
| 76 | xBias = xMin, |
| 77 | yScale = 1.0f / (yMax - yMin), |
| 78 | yBias = -yMin / (yMax - yMin); |
| 79 | |
| 80 | { |
| 81 | auto [effect, error] = SkRuntimeEffect::Make(make_unary_sksl_1d(fn)); |
| 82 | if (!effect) { |
| 83 | SkDebugf("Error: %s\n", error.c_str()); |
| 84 | return; |
| 85 | } |
| 86 | |
| 87 | SkRuntimeShaderBuilder builder(effect); |
| 88 | builder.uniform("xScale") = xScale; |
| 89 | builder.uniform("xBias") = xBias; |
| 90 | builder.uniform("yScale") = yScale; |
| 91 | builder.uniform("yBias") = yBias; |
| 92 | |
| 93 | SkPaint paint; |
| 94 | paint.setShader(builder.makeShader(nullptr, false)); |
| 95 | |
| 96 | SkImageInfo info = SkImageInfo::MakeN32Premul({ kBoxSize, kBoxSize}); |
| 97 | auto surface = canvas->makeSurface(info); |
| 98 | if (!surface) { |
| 99 | surface = SkSurface::MakeRaster(info); |
| 100 | } |
| 101 | |
| 102 | surface->getCanvas()->clear(SK_ColorWHITE); |
| 103 | surface->getCanvas()->scale(kBoxSize, kBoxSize); |
| 104 | surface->getCanvas()->drawRect({0, 0, 1, 1}, paint); |
| 105 | |
| 106 | SkBitmap bitmap; |
| 107 | bitmap.allocPixels(info); |
| 108 | surface->readPixels(bitmap, 0, 0); |
| 109 | |
| 110 | canvas->drawBitmap(bitmap, 0, 0); |
| 111 | |
| 112 | // Plot... |
| 113 | SkPaint plotPaint({ 0.0f, 0.5f, 0.0f, 1.0f }); |
| 114 | SkPoint pts[kBoxSize]; |
| 115 | for (int x = 0; x < kBoxSize; ++x) { |
| 116 | SkColor c = bitmap.getColor(x, 0); |
| 117 | SkScalar y = (1 - (SkColorGetR(c) / 255.0f)) * kBoxSize; |
| 118 | pts[x].set(x + 0.5f, y); |
| 119 | } |
| 120 | canvas->drawPoints(SkCanvas::kPoints_PointMode, kBoxSize, pts, plotPaint); |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 121 | } |
| 122 | |
| 123 | canvas->restore(); |
Brian Osman | 24000785 | 2020-11-11 17:04:57 -0500 | [diff] [blame] | 124 | } |
| 125 | |
| 126 | static void col(SkCanvas* canvas) { |
| 127 | canvas->translate(kBoxSize + kPadding, 0); |
| 128 | } |
| 129 | |
| 130 | static void row(SkCanvas* canvas) { |
| 131 | canvas->restore(); |
| 132 | canvas->translate(0, kBoxSize + kPadding + kLabelHeight); |
| 133 | canvas->save(); |
| 134 | } |
| 135 | |
| 136 | static constexpr int columns_to_width(int columns) { |
| 137 | return (kPadding + kBoxSize) * columns + kPadding; |
| 138 | } |
| 139 | |
| 140 | static constexpr int rows_to_height(int rows) { |
| 141 | return (kPadding + kLabelHeight + kBoxSize) * rows + kPadding; |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 142 | } |
| 143 | |
| 144 | // The OpenGL ES Shading Language, Version 1.00, Section 8.1 |
Brian Osman | 24000785 | 2020-11-11 17:04:57 -0500 | [diff] [blame] | 145 | DEF_SIMPLE_GM_BG( |
| 146 | runtime_intrinsics_trig, canvas, columns_to_width(3), rows_to_height(5), SK_ColorWHITE) { |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 147 | const float kPI = SK_FloatPI, kTwoPI = 2 * SK_FloatPI, kPIOverTwo = SK_FloatPI / 2; |
| 148 | |
| 149 | canvas->translate(kPadding, kPadding); |
Brian Osman | 24000785 | 2020-11-11 17:04:57 -0500 | [diff] [blame] | 150 | canvas->save(); |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 151 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 152 | plot(canvas, "radians(x)", 0.0f, 360.0f, 0.0f, kTwoPI); col(canvas); |
| 153 | plot(canvas, "degrees(x)", 0.0f, kTwoPI, 0.0f, 360.0f); row(canvas); |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 154 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 155 | plot(canvas, "sin(x)", 0.0f, kTwoPI, -1.0f, 1.0f); col(canvas); |
| 156 | plot(canvas, "cos(x)", 0.0f, kTwoPI, -1.0f, 1.0f); col(canvas); |
| 157 | plot(canvas, "tan(x)", 0.0f, kPI, -10.0f, 10.0f); row(canvas); |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 158 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 159 | plot(canvas, "asin(x)", -1.0f, 1.0f, -kPIOverTwo, kPIOverTwo); col(canvas); |
| 160 | plot(canvas, "acos(x)", -1.0f, 1.0f, 0.0f, kPI); col(canvas); |
| 161 | plot(canvas, "atan(x)", -10.0f, 10.0f, -kPIOverTwo, kPIOverTwo); row(canvas); |
Brian Osman | 24000785 | 2020-11-11 17:04:57 -0500 | [diff] [blame] | 162 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 163 | plot(canvas, "atan(0.1, x)", -1.0f, 1.0f, 0.0f, kPI); col(canvas); |
| 164 | plot(canvas, "atan(-0.1, x)", -1.0f, 1.0f, -kPI, 0.0f); row(canvas); |
Brian Osman | 24000785 | 2020-11-11 17:04:57 -0500 | [diff] [blame] | 165 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 166 | plot(canvas, "atan(x, 0.1)", -1.0f, 1.0f, -kPIOverTwo, kPIOverTwo); col(canvas); |
| 167 | plot(canvas, "atan(x, -0.1)", -1.0f, 1.0f, -kPI, kPI); row(canvas); |
Brian Osman | 3a004df | 2020-11-09 15:14:49 -0500 | [diff] [blame] | 168 | } |
Brian Osman | 8f6d4d3 | 2020-11-11 14:44:54 -0500 | [diff] [blame] | 169 | |
| 170 | // The OpenGL ES Shading Language, Version 1.00, Section 8.2 |
| 171 | DEF_SIMPLE_GM_BG(runtime_intrinsics_exponential, |
| 172 | canvas, |
| 173 | columns_to_width(2), |
| 174 | rows_to_height(5), |
| 175 | SK_ColorWHITE) { |
| 176 | canvas->translate(kPadding, kPadding); |
| 177 | canvas->save(); |
| 178 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 179 | plot(canvas, "pow(x, 3)", 0.0f, 8.0f, 0.0f, 500.0f); col(canvas); |
| 180 | plot(canvas, "pow(x, -3)", 0.0f, 4.0f, 0.0f, 10.0f); row(canvas); |
Brian Osman | 8f6d4d3 | 2020-11-11 14:44:54 -0500 | [diff] [blame] | 181 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 182 | plot(canvas, "pow(0.9, x)", -10.0f, 10.0f, 0.0f, 3.0f); col(canvas); |
| 183 | plot(canvas, "pow(1.1, x)", -10.0f, 10.0f, 0.0f, 3.0f); row(canvas); |
Brian Osman | 8f6d4d3 | 2020-11-11 14:44:54 -0500 | [diff] [blame] | 184 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 185 | plot(canvas, "exp(x)", -1.0f, 7.0f, 0.0f, 1000.0f); col(canvas); |
| 186 | plot(canvas, "log(x)", 0.0f, 2.5f, -4.0f, 1.0f); row(canvas); |
Brian Osman | 8f6d4d3 | 2020-11-11 14:44:54 -0500 | [diff] [blame] | 187 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 188 | plot(canvas, "exp2(x)", -1.0f, 7.0f, 0.0f, 130.0f); col(canvas); |
| 189 | plot(canvas, "log2(x)", 0.0f, 4.0f, -4.0f, 2.0f); row(canvas); |
Brian Osman | 8f6d4d3 | 2020-11-11 14:44:54 -0500 | [diff] [blame] | 190 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 191 | plot(canvas, "sqrt(x)", 0.0f, 25.0f, 0.0f, 5.0f); col(canvas); |
| 192 | plot(canvas, "inversesqrt(x)", 0.0f, 25.0f, 0.2f, 4.0f); row(canvas); |
Brian Osman | 8f6d4d3 | 2020-11-11 14:44:54 -0500 | [diff] [blame] | 193 | } |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 194 | |
| 195 | // The OpenGL ES Shading Language, Version 1.00, Section 8.3 |
| 196 | DEF_SIMPLE_GM_BG(runtime_intrinsics_common, |
| 197 | canvas, |
| 198 | columns_to_width(6), |
| 199 | rows_to_height(6), |
| 200 | SK_ColorWHITE) { |
| 201 | canvas->translate(kPadding, kPadding); |
| 202 | canvas->save(); |
| 203 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 204 | plot(canvas, "abs(x)", -10.0f, 10.0f, 0.0f, 10.0f); col(canvas); |
| 205 | plot(canvas, "sign(x)", -1.0f, 1.0f, -1.5f, 1.5f); row(canvas); |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 206 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 207 | plot(canvas, "floor(x)", -3.0f, 3.0f, -4.0f, 4.0f); col(canvas); |
| 208 | plot(canvas, "ceil(x)", -3.0f, 3.0f, -4.0f, 4.0f); col(canvas); |
| 209 | plot(canvas, "fract(x)", -3.0f, 3.0f, 0.0f, 1.0f); col(canvas); |
| 210 | plot(canvas, "mod(x, 2)", -4.0f, 4.0f, -2.0f, 2.0f, "mod(scalar)"); col(canvas); |
| 211 | plot(canvas, "mod(p, -2).x", -4.0f, 4.0f, -2.0f, 2.0f, "mod(mixed)" ); col(canvas); |
| 212 | plot(canvas, "mod(p, v2).x", -4.0f, 4.0f, -2.0f, 2.0f, "mod(vector)"); row(canvas); |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 213 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 214 | plot(canvas, "min(x, 1)", 0.0f, 2.0f, 0.0f, 2.0f, "min(scalar)"); col(canvas); |
| 215 | plot(canvas, "min(p, 1).x", 0.0f, 2.0f, 0.0f, 2.0f, "min(mixed)" ); col(canvas); |
| 216 | plot(canvas, "min(p, v1).x", 0.0f, 2.0f, 0.0f, 2.0f, "min(vector)"); col(canvas); |
| 217 | plot(canvas, "max(x, 1)", 0.0f, 2.0f, 0.0f, 2.0f, "max(scalar)"); col(canvas); |
| 218 | plot(canvas, "max(p, 1).x", 0.0f, 2.0f, 0.0f, 2.0f, "max(mixed)" ); col(canvas); |
| 219 | plot(canvas, "max(p, v1).x", 0.0f, 2.0f, 0.0f, 2.0f, "max(vector)"); row(canvas); |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 220 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 221 | plot(canvas, "clamp(x, 1, 2)", 0.0f, 3.0f, 0.0f, 3.0f, "clamp(scalar)"); col(canvas); |
| 222 | plot(canvas, "clamp(p, 1, 2).x", 0.0f, 3.0f, 0.0f, 3.0f, "clamp(mixed)" ); col(canvas); |
| 223 | plot(canvas, "clamp(p, v1, v2).x", 0.0f, 3.0f, 0.0f, 3.0f, "clamp(vector)"); col(canvas); |
| 224 | plot(canvas, "saturate(x)", -1.0f, 2.0f, -0.5f, 1.5f); row(canvas); |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 225 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 226 | plot(canvas, "mix(1, 2, x)", -1.0f, 2.0f, 0.0f, 3.0f, "mix(scalar)"); col(canvas); |
| 227 | plot(canvas, "mix(v1, v2, x).x", -1.0f, 2.0f, 0.0f, 3.0f, "mix(mixed)" ); col(canvas); |
| 228 | plot(canvas, "mix(v1, v2, p).x", -1.0f, 2.0f, 0.0f, 3.0f, "mix(vector)"); row(canvas); |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 229 | |
Brian Osman | 5035b8f | 2020-11-17 09:53:54 -0500 | [diff] [blame] | 230 | plot(canvas, "step(1, x)", 0.0f, 2.0f, -0.5f, 1.5f, "step(scalar)"); col(canvas); |
| 231 | plot(canvas, "step(1, p).x", 0.0f, 2.0f, -0.5f, 1.5f, "step(mixed)" ); col(canvas); |
| 232 | plot(canvas, "step(v1, p).x", 0.0f, 2.0f, -0.5f, 1.5f, "step(vector)"); col(canvas); |
| 233 | plot(canvas, "smoothstep(1, 2, x)", 0.5f, 2.5f, -0.5f, 1.5f, "smooth(scalar)"); col(canvas); |
| 234 | plot(canvas, "smoothstep(1, 2, p).x", 0.5f, 2.5f, -0.5f, 1.5f, "smooth(mixed)" ); col(canvas); |
| 235 | plot(canvas, "smoothstep(v1, v2, p).x", 0.5f, 2.5f, -0.5f, 1.5f, "smooth(vector)"); row(canvas); |
Brian Osman | 5160aad | 2020-11-11 16:49:39 -0500 | [diff] [blame] | 236 | } |
Brian Osman | 9ba7a24 | 2020-11-13 16:31:37 -0500 | [diff] [blame^] | 237 | |
| 238 | // The OpenGL ES Shading Language, Version 1.00, Section 8.4 |
| 239 | DEF_SIMPLE_GM_BG(runtime_intrinsics_geometric, |
| 240 | canvas, |
| 241 | columns_to_width(4), |
| 242 | rows_to_height(5), |
| 243 | SK_ColorWHITE) { |
| 244 | canvas->translate(kPadding, kPadding); |
| 245 | canvas->save(); |
| 246 | |
| 247 | plot(canvas, "length(x)", -1.0f, 1.0f, -0.5f, 1.5f); col(canvas); |
| 248 | plot(canvas, "length(p)", 0.0f, 1.0f, 0.5f, 1.5f); col(canvas); |
| 249 | |
| 250 | plot(canvas, "distance(x, 0)", -1.0f, 1.0f, -0.5f, 1.5f); col(canvas); |
| 251 | plot(canvas, "distance(p, v1)", 0.0f, 1.0f, 0.5f, 1.5f); row(canvas); |
| 252 | |
| 253 | plot(canvas, "dot(x, 2)", -1.0f, 1.0f, -2.5f, 2.5f); col(canvas); |
| 254 | plot(canvas, "dot(p, p.y1)", -1.0f, 1.0f, -2.5f, 0.5f); row(canvas); |
| 255 | |
| 256 | plot(canvas, "cross(p.xy1, p.y1x).x", 0.0f, 1.0f, -1.0f, 1.0f); col(canvas); |
| 257 | plot(canvas, "cross(p.xy1, p.y1x).y", 0.0f, 1.0f, -1.0f, 1.0f); col(canvas); |
| 258 | plot(canvas, "cross(p.xy1, p.y1x).z", 0.0f, 1.0f, -1.0f, 1.0f); row(canvas); |
| 259 | |
| 260 | plot(canvas, "normalize(x)", -2.0f, 2.0f, -1.5f, 1.5f); col(canvas); |
| 261 | plot(canvas, "normalize(p).x", 0.0f, 2.0f, 0.0f, 1.0f); col(canvas); |
| 262 | plot(canvas, "normalize(p).y", 0.0f, 2.0f, 0.0f, 1.0f); col(canvas); |
| 263 | |
| 264 | plot(canvas, "faceforward(v1, p.x0, v1.x0).x", -1.0f, 1.0f, -1.5f, 1.5f, "faceforward"); row(canvas); |
| 265 | |
| 266 | plot(canvas, "reflect(p.x1, v1.0x).x", -1.0f, 1.0f, -1.0f, 1.0f, "reflect(horiz)"); col(canvas); |
| 267 | plot(canvas, "reflect(p.x1, normalize(v1)).y", -1.0f, 1.0f, -1.0f, 1.0f, "reflect(diag)" ); col(canvas); |
| 268 | |
| 269 | plot(canvas, "refract(v1.x0, v1.0x, x).x", 0.0f, 1.0f, -1.0f, 1.0f, "refract().x"); col(canvas); |
| 270 | plot(canvas, "refract(v1.x0, v1.0x, x).y", 0.0f, 1.0f, -1.0f, 1.0f, "refract().y"); row(canvas); |
| 271 | } |