sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2014 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 | #include "Benchmark.h" |
| 8 | #include "SkCanvas.h" |
| 9 | #include "SkColorCubeFilter.h" |
| 10 | #include "SkGradientShader.h" |
scroggo | 565901d | 2015-12-10 10:44:13 -0800 | [diff] [blame] | 11 | #include "SkTemplates.h" |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 12 | |
| 13 | class ColorCubeBench : public Benchmark { |
| 14 | SkISize fSize; |
| 15 | int fCubeDimension; |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 16 | sk_sp<SkData> fCubeData; |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 17 | SkBitmap fBitmap; |
| 18 | |
| 19 | public: |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 20 | ColorCubeBench() : fCubeDimension(0) { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 21 | fSize = SkISize::Make(2880, 1800); // 2014 Macbook Pro resolution |
| 22 | } |
| 23 | |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 24 | protected: |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 25 | const char* onGetName() override { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 26 | return "colorcube"; |
| 27 | } |
| 28 | |
joshualitt | 8a6697a | 2015-09-30 12:11:07 -0700 | [diff] [blame] | 29 | void onDelayedSetup() override { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 30 | if (!SkToBool(fCubeData)) { |
| 31 | this->makeCubeData(); |
| 32 | this->make_bitmap(); |
| 33 | } |
| 34 | } |
| 35 | |
mtklein | a1ebeb2 | 2015-10-01 09:43:39 -0700 | [diff] [blame] | 36 | void onDraw(int loops, SkCanvas* canvas) override { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 37 | this->test(loops, canvas); |
| 38 | } |
| 39 | |
mtklein | 36352bf | 2015-03-25 18:17:31 -0700 | [diff] [blame] | 40 | SkIPoint onGetSize() override { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 41 | return SkIPoint::Make(fSize.width(), fSize.height()); |
| 42 | } |
| 43 | |
| 44 | private: |
reed | c6f28f7 | 2016-03-14 12:22:10 -0700 | [diff] [blame] | 45 | static sk_sp<SkShader> MakeLinear(const SkISize& size) { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 46 | const SkPoint pts[2] = { |
| 47 | { 0, 0 }, |
| 48 | { SkIntToScalar(size.width()), SkIntToScalar(size.height()) } |
| 49 | }; |
| 50 | static const SkColor colors[] = { SK_ColorYELLOW, SK_ColorBLUE }; |
reed | c6f28f7 | 2016-03-14 12:22:10 -0700 | [diff] [blame] | 51 | return SkGradientShader::MakeLinear( |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 52 | pts, colors, nullptr, 2, SkShader::kRepeat_TileMode, 0, &SkMatrix::I()); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 53 | } |
| 54 | |
| 55 | void make_bitmap() { |
| 56 | fBitmap.allocN32Pixels(fSize.width(), fSize.height()); |
| 57 | SkCanvas canvas(fBitmap); |
| 58 | canvas.clear(0x00000000); |
| 59 | SkPaint paint; |
| 60 | paint.setAntiAlias(true); |
reed | c6f28f7 | 2016-03-14 12:22:10 -0700 | [diff] [blame] | 61 | paint.setShader(MakeLinear(fSize)); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 62 | SkRect r = { 0, 0, SkIntToScalar(fSize.width()), SkIntToScalar(fSize.height()) }; |
| 63 | canvas.drawRect(r, paint); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 64 | } |
| 65 | |
| 66 | void makeCubeData() { |
| 67 | fCubeDimension = 32; |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 68 | fCubeData = SkData::MakeUninitialized(sizeof(SkColor) * |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 69 | fCubeDimension * fCubeDimension * fCubeDimension); |
| 70 | SkColor* pixels = (SkColor*)(fCubeData->writable_data()); |
scroggo | 565901d | 2015-12-10 10:44:13 -0800 | [diff] [blame] | 71 | SkAutoTMalloc<uint8_t> lutMemory(fCubeDimension); |
| 72 | uint8_t* lut = lutMemory.get(); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 73 | const int maxIndex = fCubeDimension - 1; |
| 74 | for (int i = 0; i < fCubeDimension; ++i) { |
| 75 | // Make an invert lut, but the content of |
| 76 | // the lut shouldn't affect performance. |
| 77 | lut[i] = ((maxIndex - i) * 255) / maxIndex; |
| 78 | } |
| 79 | for (int r = 0; r < fCubeDimension; ++r) { |
| 80 | for (int g = 0; g < fCubeDimension; ++g) { |
| 81 | for (int b = 0; b < fCubeDimension; ++b) { |
| 82 | pixels[(fCubeDimension * ((fCubeDimension * b) + g)) + r] = |
| 83 | SkColorSetARGB(0xFF, lut[r], lut[g], lut[b]); |
| 84 | } |
| 85 | } |
| 86 | } |
| 87 | } |
| 88 | |
mtklein | a1ebeb2 | 2015-10-01 09:43:39 -0700 | [diff] [blame] | 89 | void test(int loops, SkCanvas* canvas) { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 90 | SkPaint paint; |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 91 | for (int i = 0; i < loops; i++) { |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 92 | paint.setColorFilter(SkColorCubeFilter::Make(fCubeData, fCubeDimension)); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 93 | canvas->drawBitmap(fBitmap, 0, 0, &paint); |
| 94 | } |
| 95 | } |
| 96 | |
| 97 | typedef Benchmark INHERITED; |
| 98 | }; |
| 99 | |
| 100 | /////////////////////////////////////////////////////////////////////////////// |
| 101 | |
| 102 | DEF_BENCH( return new ColorCubeBench(); ) |