Ethan Nicholas | a70693b | 2019-03-04 13:07:36 -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 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 8 | #include "gm/gm.h" |
| 9 | #include "include/core/SkCanvas.h" |
Ben Wagner | 7fde8e1 | 2019-05-01 17:28:53 -0400 | [diff] [blame] | 10 | #include "include/core/SkColorFilter.h" |
| 11 | #include "include/core/SkData.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 12 | #include "include/core/SkImage.h" |
Ben Wagner | 7fde8e1 | 2019-05-01 17:28:53 -0400 | [diff] [blame] | 13 | #include "include/core/SkPaint.h" |
| 14 | #include "include/core/SkRefCnt.h" |
| 15 | #include "include/core/SkSize.h" |
| 16 | #include "include/core/SkString.h" |
Brian Osman | ee426f2 | 2020-01-02 11:55:24 -0500 | [diff] [blame] | 17 | #include "include/effects/SkRuntimeEffect.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 18 | #include "tools/Resources.h" |
Ben Wagner | 7fde8e1 | 2019-05-01 17:28:53 -0400 | [diff] [blame] | 19 | |
| 20 | #include <stddef.h> |
| 21 | #include <utility> |
| 22 | |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 23 | const char* gNoop = R"( |
Brian Osman | 91292e9 | 2020-11-04 15:40:50 -0500 | [diff] [blame] | 24 | uniform shader input; |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 25 | half4 main() { |
| 26 | return sample(input); |
| 27 | } |
| 28 | )"; |
| 29 | |
Brian Osman | 5240e18 | 2020-08-12 10:22:34 -0400 | [diff] [blame] | 30 | const char* gLumaSrc = R"( |
Brian Osman | 91292e9 | 2020-11-04 15:40:50 -0500 | [diff] [blame] | 31 | uniform shader input; |
Brian Osman | 767f444 | 2020-08-13 16:59:48 -0400 | [diff] [blame] | 32 | half4 main() { |
| 33 | return dot(sample(input).rgb, half3(0.3, 0.6, 0.1)).000r; |
Ethan Nicholas | a70693b | 2019-03-04 13:07:36 -0500 | [diff] [blame] | 34 | } |
| 35 | )"; |
| 36 | |
Brian Osman | 5240e18 | 2020-08-12 10:22:34 -0400 | [diff] [blame] | 37 | const char* gLumaSrcWithCoords = R"( |
Brian Osman | 91292e9 | 2020-11-04 15:40:50 -0500 | [diff] [blame] | 38 | uniform shader input; |
Brian Osman | 767f444 | 2020-08-13 16:59:48 -0400 | [diff] [blame] | 39 | half4 main(float2 p) { |
| 40 | return dot(sample(input).rgb, half3(0.3, 0.6, 0.1)).000r; |
Brian Osman | 5240e18 | 2020-08-12 10:22:34 -0400 | [diff] [blame] | 41 | } |
| 42 | )"; |
| 43 | |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 44 | // Build up the same effect with increasingly complex control flow syntax. |
| 45 | // All of these are semantically equivalent and can be reduced in principle to one basic block. |
| 46 | |
| 47 | // Simplest to run; hardest to write? |
| 48 | const char* gTernary = R"( |
Brian Osman | 91292e9 | 2020-11-04 15:40:50 -0500 | [diff] [blame] | 49 | uniform shader input; |
Mike Klein | 7485ffc | 2020-10-14 14:05:50 -0500 | [diff] [blame] | 50 | half4 main() { |
| 51 | half4 color = sample(input); |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 52 | half luma = dot(color.rgb, half3(0.3, 0.6, 0.1)); |
Mike Klein | 7485ffc | 2020-10-14 14:05:50 -0500 | [diff] [blame] | 53 | |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 54 | half scale = luma < 0.33333 ? 0.5 |
| 55 | : luma < 0.66666 ? (0.166666 + 2.0 * (luma - 0.33333)) / luma |
| 56 | : /* else */ (0.833333 + 0.5 * (luma - 0.66666)) / luma; |
| 57 | return half4(color.rgb * scale, color.a); |
| 58 | } |
| 59 | )"; |
| 60 | |
| 61 | // Uses conditional if statements but no early return. |
| 62 | const char* gIfs = R"( |
Brian Osman | 91292e9 | 2020-11-04 15:40:50 -0500 | [diff] [blame] | 63 | uniform shader input; |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 64 | half4 main() { |
| 65 | half4 color = sample(input); |
Mike Klein | 7485ffc | 2020-10-14 14:05:50 -0500 | [diff] [blame] | 66 | half luma = dot(color.rgb, half3(0.3, 0.6, 0.1)); |
| 67 | |
| 68 | half scale = 0; |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 69 | if (luma < 0.33333) { |
| 70 | scale = 0.5; |
| 71 | } else if (luma < 0.66666) { |
| 72 | scale = (0.166666 + 2.0 * (luma - 0.33333)) / luma; |
| 73 | } else { |
| 74 | scale = (0.833333 + 0.5 * (luma - 0.66666)) / luma; |
| 75 | } |
| 76 | return half4(color.rgb * scale, color.a); |
| 77 | } |
| 78 | )"; |
Mike Klein | 7485ffc | 2020-10-14 14:05:50 -0500 | [diff] [blame] | 79 | |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 80 | // Distilled from AOSP tone mapping shaders, more like what people tend to write. |
| 81 | const char* gEarlyReturn = R"( |
Brian Osman | 91292e9 | 2020-11-04 15:40:50 -0500 | [diff] [blame] | 82 | uniform shader input; |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 83 | half4 main() { |
| 84 | half4 color = sample(input); |
| 85 | half luma = dot(color.rgb, half3(0.3, 0.6, 0.1)); |
| 86 | |
| 87 | half scale = 0; |
Mike Klein | 7485ffc | 2020-10-14 14:05:50 -0500 | [diff] [blame] | 88 | if (luma < 0.33333) { |
| 89 | return half4(color.rgb * 0.5, color.a); |
| 90 | } else if (luma < 0.66666) { |
| 91 | scale = 0.166666 + 2.0 * (luma - 0.33333); |
| 92 | } else { |
| 93 | scale = 0.833333 + 0.5 * (luma - 0.66666); |
| 94 | } |
Mike Klein | 7485ffc | 2020-10-14 14:05:50 -0500 | [diff] [blame] | 95 | return half4(color.rgb * (scale/luma), color.a); |
| 96 | } |
| 97 | )"; |
| 98 | |
| 99 | |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 100 | DEF_SIMPLE_GM(runtimecolorfilter, canvas, 256 * 3, 256 * 2) { |
| 101 | sk_sp<SkImage> img = GetResourceAsImage("images/mandrill_256.png"); |
Ethan Nicholas | a70693b | 2019-03-04 13:07:36 -0500 | [diff] [blame] | 102 | |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 103 | auto draw_filter = [&](const char* src) { |
| 104 | auto [effect, err] = SkRuntimeEffect::Make(SkString(src)); |
| 105 | if (!effect) { |
| 106 | SkDebugf("%s\n%s\n", src, err.c_str()); |
| 107 | } |
Ethan Nicholas | 63d7ee3 | 2020-08-17 10:57:12 -0400 | [diff] [blame] | 108 | SkASSERT(effect); |
Brian Osman | 5240e18 | 2020-08-12 10:22:34 -0400 | [diff] [blame] | 109 | SkPaint p; |
Brian Osman | 767f444 | 2020-08-13 16:59:48 -0400 | [diff] [blame] | 110 | sk_sp<SkColorFilter> input = nullptr; |
| 111 | p.setColorFilter(effect->makeColorFilter(nullptr, &input, 1)); |
Brian Osman | 5240e18 | 2020-08-12 10:22:34 -0400 | [diff] [blame] | 112 | canvas->drawImage(img, 0, 0, &p); |
Mike Klein | a0f5452 | 2020-10-14 16:31:36 -0500 | [diff] [blame] | 113 | canvas->translate(256, 0); |
| 114 | }; |
| 115 | |
| 116 | for (const char* src : { gNoop, gLumaSrc, gLumaSrcWithCoords}) { |
| 117 | draw_filter(src); |
| 118 | } |
| 119 | canvas->translate(-256*3, 256); |
| 120 | for (const char* src : { gTernary, gIfs, gEarlyReturn}) { |
| 121 | draw_filter(src); |
Brian Osman | 5240e18 | 2020-08-12 10:22:34 -0400 | [diff] [blame] | 122 | } |
Ethan Nicholas | a70693b | 2019-03-04 13:07:36 -0500 | [diff] [blame] | 123 | } |