Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2017 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 | |
| 8 | #include "../src/jumper/SkJumper.h" |
| 9 | #include "Benchmark.h" |
| 10 | #include "SkColor.h" |
Mike Klein | c857956 | 2018-08-07 13:21:21 -0400 | [diff] [blame] | 11 | #include "SkColorSpaceXformer.h" |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 12 | #include "SkColorSpaceXformSteps.h" |
| 13 | #include "SkMakeUnique.h" |
Brian Osman | 87311c6 | 2018-09-28 10:51:13 -0400 | [diff] [blame] | 14 | #include "SkPM4f.h" |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 15 | #include "SkRandom.h" |
| 16 | #include "SkRasterPipeline.h" |
| 17 | |
Brian Osman | 5ea41fc | 2018-09-26 18:49:27 +0000 | [diff] [blame] | 18 | enum class Mode { steps, pipeA, pipeB, xformer }; |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 19 | |
| 20 | struct ColorSpaceXformBench : public Benchmark { |
| 21 | ColorSpaceXformBench(Mode mode) : fMode(mode) {} |
| 22 | |
| 23 | const Mode fMode; |
| 24 | |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 25 | std::unique_ptr<SkColorSpaceXformSteps> fSteps; |
| 26 | std::function<void(size_t, size_t, size_t, size_t)> fPipeA; |
| 27 | |
| 28 | SkJumper_MemoryCtx fPipeSrc = {nullptr,0}, |
| 29 | fPipeDst = {nullptr,0}; |
| 30 | SkSTArenaAlloc<1024> fAlloc; |
| 31 | |
Mike Klein | c857956 | 2018-08-07 13:21:21 -0400 | [diff] [blame] | 32 | std::unique_ptr<SkColorSpaceXformer> fXformer; |
| 33 | |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 34 | const char* onGetName() override { |
| 35 | switch (fMode) { |
Mike Klein | c857956 | 2018-08-07 13:21:21 -0400 | [diff] [blame] | 36 | case Mode::steps : return "ColorSpaceXformBench_steps"; |
| 37 | case Mode::pipeA : return "ColorSpaceXformBench_pipeA"; |
| 38 | case Mode::pipeB : return "ColorSpaceXformBench_pipeB"; |
| 39 | case Mode::xformer: return "ColorSpaceXformBench_xformer"; |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 40 | } |
| 41 | return ""; |
| 42 | } |
| 43 | |
| 44 | bool isSuitableFor(Backend backend) override { return kNonRendering_Backend == backend; } |
| 45 | |
| 46 | void onDelayedSetup() override { |
| 47 | sk_sp<SkColorSpace> src = SkColorSpace::MakeSRGB(), |
| 48 | dst = SkColorSpace::MakeRGB(SkColorSpace::kSRGB_RenderTargetGamma, |
| 49 | SkColorSpace::kDCIP3_D65_Gamut); |
| 50 | |
Mike Klein | 8f3d36c | 2018-08-14 10:28:05 -0400 | [diff] [blame] | 51 | fSteps = skstd::make_unique<SkColorSpaceXformSteps>(src.get(), kOpaque_SkAlphaType, |
| 52 | dst.get(), kPremul_SkAlphaType); |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 53 | |
| 54 | SkRasterPipeline p(&fAlloc); |
| 55 | p.append(SkRasterPipeline::load_bgra, &fPipeSrc); |
| 56 | fSteps->apply(&p); |
| 57 | p.append(SkRasterPipeline::store_bgra, &fPipeDst); |
| 58 | |
| 59 | fPipeA = p.compile(); |
Mike Klein | c857956 | 2018-08-07 13:21:21 -0400 | [diff] [blame] | 60 | |
| 61 | fXformer = SkColorSpaceXformer::Make(dst); // src is implicitly sRGB, what we want anyway |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 62 | } |
| 63 | |
| 64 | void onDraw(int n, SkCanvas* canvas) override { |
| 65 | volatile SkColor junk = 0; |
| 66 | SkRandom rand; |
| 67 | |
| 68 | for (int i = 0; i < n; i++) { |
| 69 | SkColor src = rand.nextU(), |
| 70 | dst; |
| 71 | fPipeSrc.pixels = &src; |
| 72 | fPipeDst.pixels = &dst; |
| 73 | |
| 74 | switch (fMode) { |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 75 | case Mode::steps: { |
| 76 | float rgba[4]; |
| 77 | swizzle_rb(Sk4f_fromL32(src)).store(rgba); |
| 78 | fSteps->apply(rgba); |
| 79 | dst = Sk4f_toL32(swizzle_rb(Sk4f::Load(rgba))); |
| 80 | } break; |
| 81 | |
| 82 | case Mode::pipeA: { |
| 83 | fPipeA(0,0,1,1); |
| 84 | } break; |
| 85 | |
| 86 | case Mode::pipeB: { |
| 87 | SkSTArenaAlloc<1024> alloc; |
| 88 | SkRasterPipeline p(&alloc); |
| 89 | p.append(SkRasterPipeline::load_bgra, &fPipeSrc); |
| 90 | fSteps->apply(&p); |
| 91 | p.append(SkRasterPipeline::store_bgra, &fPipeDst); |
| 92 | p.run(0,0,1,1); |
| 93 | } break; |
Mike Klein | c857956 | 2018-08-07 13:21:21 -0400 | [diff] [blame] | 94 | |
| 95 | case Mode::xformer: { |
| 96 | dst = fXformer->apply(src); |
| 97 | } break; |
Mike Klein | 4ed73e0 | 2018-08-06 17:03:33 -0400 | [diff] [blame] | 98 | } |
| 99 | |
| 100 | if (false && i == 0) { |
| 101 | SkDebugf("%x ~~> %x\n", src, dst); |
| 102 | } |
| 103 | |
| 104 | junk ^= dst; |
| 105 | } |
| 106 | } |
| 107 | }; |
| 108 | |
Mike Klein | c857956 | 2018-08-07 13:21:21 -0400 | [diff] [blame] | 109 | DEF_BENCH(return new ColorSpaceXformBench{Mode::steps };) |
| 110 | DEF_BENCH(return new ColorSpaceXformBench{Mode::pipeA };) |
| 111 | DEF_BENCH(return new ColorSpaceXformBench{Mode::pipeB };) |
| 112 | DEF_BENCH(return new ColorSpaceXformBench{Mode::xformer};) |