123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233 |
- /*
- * Copyright 2019 Google LLC
- *
- * Use of this source code is governed by a BSD-style license that can be
- * found in the LICENSE file.
- */
- #include "bench/Benchmark.h"
- #include "include/utils/SkRandom.h"
- #include "src/sksl/SkSLByteCode.h"
- #include "src/sksl/SkSLCompiler.h"
- // Benchmarks the interpreter with a function that has a color-filter style signature
- class SkSLInterpreterCFBench : public Benchmark {
- public:
- SkSLInterpreterCFBench(SkSL::String name, int pixels, bool striped, const char* src)
- : fName(SkStringPrintf("sksl_interp_cf_%d_%d_%s", pixels, striped ? 1 : 0, name.c_str()))
- , fSrc(src)
- , fCount(pixels)
- , fStriped(striped) {}
- protected:
- const char* onGetName() override {
- return fName.c_str();
- }
- bool isSuitableFor(Backend backend) override {
- return backend == kNonRendering_Backend;
- }
- void onDelayedSetup() override {
- SkSL::Compiler compiler;
- SkSL::Program::Settings settings;
- auto program = compiler.convertProgram(SkSL::Program::kGeneric_Kind, fSrc, settings);
- SkASSERT(compiler.errorCount() == 0);
- fByteCode = compiler.toByteCode(*program);
- SkASSERT(compiler.errorCount() == 0);
- fMain = fByteCode->getFunction("main");
- SkRandom rnd;
- fPixels.resize(fCount * 4);
- for (float& c : fPixels) {
- c = rnd.nextF();
- }
- }
- void onDraw(int loops, SkCanvas*) override {
- for (int i = 0; i < loops; i++) {
- if (fStriped) {
- float* args[] = {
- fPixels.data() + 0 * fCount,
- fPixels.data() + 1 * fCount,
- fPixels.data() + 2 * fCount,
- fPixels.data() + 3 * fCount,
- };
- SkAssertResult(fByteCode->runStriped(fMain, args, 4, fCount,
- nullptr, 0, nullptr, 0));
- } else {
- SkAssertResult(fByteCode->run(fMain, fPixels.data(), nullptr, fCount, nullptr, 0));
- }
- }
- }
- private:
- SkString fName;
- SkSL::String fSrc;
- std::unique_ptr<SkSL::ByteCode> fByteCode;
- const SkSL::ByteCodeFunction* fMain;
- int fCount;
- bool fStriped;
- std::vector<float> fPixels;
- typedef Benchmark INHERITED;
- };
- ///////////////////////////////////////////////////////////////////////////////
- const char* kLumaToAlphaSrc = R"(
- void main(inout float4 color) {
- color.a = color.r*0.3 + color.g*0.6 + color.b*0.1;
- color.r = 0;
- color.g = 0;
- color.b = 0;
- }
- )";
- const char* kHighContrastFilterSrc = R"(
- half ucontrast_Stage2;
- half hue2rgb_Stage2(half p, half q, half t) {
- if (t < 0) t += 1;
- if (t > 1) t -= 1;
- return (t < 1 / 6.) ? p + (q - p) * 6 * t
- : (t < 1 / 2.) ? q
- : (t < 2 / 3.) ? p + (q - p) * (2 / 3. - t) * 6
- : p;
- }
- half max(half a, half b) { return a > b ? a : b; }
- half min(half a, half b) { return a < b ? a : b; }
- void main(inout half4 color) {
- ucontrast_Stage2 = 0.2;
- // HighContrastFilter
- half nonZeroAlpha = max(color.a, 0.0001);
- color = half4(color.rgb / nonZeroAlpha, nonZeroAlpha);
- color.rgb = color.rgb * color.rgb;
- half fmax = max(color.r, max(color.g, color.b));
- half fmin = min(color.r, min(color.g, color.b));
- half l = (fmax + fmin) / 2;
- half h;
- half s;
- if (fmax == fmin) {
- h = 0;
- s = 0;
- } else {
- half d = fmax - fmin;
- s = l > 0.5 ? d / (2 - fmax - fmin) : d / (fmax + fmin);
- if (color.r >= color.g && color.r >= color.b) {
- h = (color.g - color.b) / d + (color.g < color.b ? 6 : 0);
- } else if (color.g >= color.b) {
- h = (color.b - color.r) / d + 2;
- } else {
- h = (color.r - color.g) / d + 4;
- }
- }
- h /= 6;
- l = 1.0 - l;
- if (s == 0) {
- color = half4(l, l, l, 0);
- } else {
- half q = l < 0.5 ? l * (1 + s) : l + s - l * s;
- half p = 2 * l - q;
- color.r = hue2rgb_Stage2(p, q, h + 1 / 3.);
- color.g = hue2rgb_Stage2(p, q, h);
- color.b = hue2rgb_Stage2(p, q, h - 1 / 3.);
- }
- if (ucontrast_Stage2 != 0) {
- half m = (1 + ucontrast_Stage2) / (1 - ucontrast_Stage2);
- half off = (-0.5 * m + 0.5);
- color = m * color + off;
- }
- // color = saturate(color);
- color.rgb = sqrt(color.rgb);
- color.rgb *= color.a;
- }
- )";
- DEF_BENCH(return new SkSLInterpreterCFBench("lumaToAlpha", 256, false, kLumaToAlphaSrc));
- DEF_BENCH(return new SkSLInterpreterCFBench("lumaToAlpha", 256, true, kLumaToAlphaSrc));
- DEF_BENCH(return new SkSLInterpreterCFBench("hcf", 256, false, kHighContrastFilterSrc));
- DEF_BENCH(return new SkSLInterpreterCFBench("hcf", 256, true, kHighContrastFilterSrc));
- class SkSLInterpreterSortBench : public Benchmark {
- public:
- SkSLInterpreterSortBench(int groups, int values, const char* src)
- : fName(SkStringPrintf("sksl_interp_sort_%dx%d", groups, values))
- , fCode(src)
- , fGroups(groups)
- , fValues(values) {
- }
- protected:
- const char* onGetName() override {
- return fName.c_str();
- }
- bool isSuitableFor(Backend backend) override {
- return backend == kNonRendering_Backend;
- }
- void onDelayedSetup() override {
- SkSL::Compiler compiler;
- SkSL::Program::Settings settings;
- auto program = compiler.convertProgram(SkSL::Program::kGeneric_Kind, fCode, settings);
- SkASSERT(compiler.errorCount() == 0);
- fByteCode = compiler.toByteCode(*program);
- SkASSERT(compiler.errorCount() == 0);
- fMain = fByteCode->getFunction("main");
- fSrc.resize(fGroups * fValues);
- fDst.resize(fGroups * fValues);
- SkRandom rnd;
- for (float& x : fSrc) {
- x = rnd.nextF();
- }
- // Trigger one run now to check correctness
- SkAssertResult(fByteCode->run(fMain, fSrc.data(), fDst.data(), fGroups, nullptr, 0));
- for (int i = 0; i < fGroups; ++i) {
- for (int j = 1; j < fValues; ++j) {
- SkASSERT(fDst[i * fValues + j] >= fDst[i * fValues + j - 1]);
- }
- }
- }
- void onDraw(int loops, SkCanvas*) override {
- for (int i = 0; i < loops; i++) {
- SkAssertResult(fByteCode->run(fMain, fSrc.data(), fDst.data(), fGroups, nullptr, 0));
- }
- }
- private:
- SkString fName;
- SkSL::String fCode;
- std::unique_ptr<SkSL::ByteCode> fByteCode;
- const SkSL::ByteCodeFunction* fMain;
- int fGroups;
- int fValues;
- std::vector<float> fSrc;
- std::vector<float> fDst;
- typedef Benchmark INHERITED;
- };
- // Currently, this exceeds the interpreter's stack. Consider it a test case for some eventual
- // bounds checking.
- #if 0
- DEF_BENCH(return new SkSLInterpreterSortBench(1024, 32, R"(
- float[32] main(float v[32]) {
- for (int i = 1; i < 32; ++i) {
- for (int j = i; j > 0 && v[j-1] > v[j]; --j) {
- float t = v[j];
- v[j] = v[j-1];
- v[j-1] = t;
- }
- }
- return v;
- }
- )"));
- #endif
|