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" |
| 11 | |
| 12 | class ColorCubeBench : public Benchmark { |
| 13 | SkISize fSize; |
| 14 | int fCubeDimension; |
| 15 | SkData* fCubeData; |
| 16 | SkBitmap fBitmap; |
| 17 | |
| 18 | public: |
| 19 | ColorCubeBench() |
| 20 | : fCubeDimension(0) |
| 21 | , fCubeData(NULL) { |
| 22 | fSize = SkISize::Make(2880, 1800); // 2014 Macbook Pro resolution |
| 23 | } |
| 24 | |
| 25 | ~ColorCubeBench() { |
| 26 | SkSafeUnref(fCubeData); |
| 27 | } |
| 28 | |
| 29 | protected: |
| 30 | virtual const char* onGetName() SK_OVERRIDE { |
| 31 | return "colorcube"; |
| 32 | } |
| 33 | |
| 34 | virtual void onPreDraw() SK_OVERRIDE { |
| 35 | if (!SkToBool(fCubeData)) { |
| 36 | this->makeCubeData(); |
| 37 | this->make_bitmap(); |
| 38 | } |
| 39 | } |
| 40 | |
| 41 | virtual void onDraw(const int loops, SkCanvas* canvas) SK_OVERRIDE { |
| 42 | this->test(loops, canvas); |
| 43 | } |
| 44 | |
| 45 | virtual SkIPoint onGetSize() SK_OVERRIDE { |
| 46 | return SkIPoint::Make(fSize.width(), fSize.height()); |
| 47 | } |
| 48 | |
| 49 | private: |
| 50 | static SkShader* MakeLinear(const SkISize& size) { |
| 51 | const SkPoint pts[2] = { |
| 52 | { 0, 0 }, |
| 53 | { SkIntToScalar(size.width()), SkIntToScalar(size.height()) } |
| 54 | }; |
| 55 | static const SkColor colors[] = { SK_ColorYELLOW, SK_ColorBLUE }; |
| 56 | return SkGradientShader::CreateLinear( |
| 57 | pts, colors, NULL, 2, SkShader::kRepeat_TileMode, 0, &SkMatrix::I()); |
| 58 | } |
| 59 | |
| 60 | void make_bitmap() { |
| 61 | fBitmap.allocN32Pixels(fSize.width(), fSize.height()); |
| 62 | SkCanvas canvas(fBitmap); |
| 63 | canvas.clear(0x00000000); |
| 64 | SkPaint paint; |
| 65 | paint.setAntiAlias(true); |
| 66 | SkShader* shader = MakeLinear(fSize); |
| 67 | paint.setShader(shader); |
| 68 | SkRect r = { 0, 0, SkIntToScalar(fSize.width()), SkIntToScalar(fSize.height()) }; |
| 69 | canvas.drawRect(r, paint); |
| 70 | shader->unref(); |
| 71 | } |
| 72 | |
| 73 | void makeCubeData() { |
| 74 | fCubeDimension = 32; |
| 75 | fCubeData = SkData::NewUninitialized(sizeof(SkColor) * |
| 76 | fCubeDimension * fCubeDimension * fCubeDimension); |
| 77 | SkColor* pixels = (SkColor*)(fCubeData->writable_data()); |
| 78 | SkAutoMalloc lutMemory(fCubeDimension); |
| 79 | uint8_t* lut = (uint8_t*)lutMemory.get(); |
| 80 | const int maxIndex = fCubeDimension - 1; |
| 81 | for (int i = 0; i < fCubeDimension; ++i) { |
| 82 | // Make an invert lut, but the content of |
| 83 | // the lut shouldn't affect performance. |
| 84 | lut[i] = ((maxIndex - i) * 255) / maxIndex; |
| 85 | } |
| 86 | for (int r = 0; r < fCubeDimension; ++r) { |
| 87 | for (int g = 0; g < fCubeDimension; ++g) { |
| 88 | for (int b = 0; b < fCubeDimension; ++b) { |
| 89 | pixels[(fCubeDimension * ((fCubeDimension * b) + g)) + r] = |
| 90 | SkColorSetARGB(0xFF, lut[r], lut[g], lut[b]); |
| 91 | } |
| 92 | } |
| 93 | } |
| 94 | } |
| 95 | |
| 96 | void test(const int loops, SkCanvas* canvas) { |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 97 | SkPaint paint; |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 98 | for (int i = 0; i < loops; i++) { |
sugoi | 12b1831 | 2014-10-09 11:35:09 -0700 | [diff] [blame] | 99 | SkAutoTUnref<SkColorFilter> colorCube( |
| 100 | SkColorCubeFilter::Create(fCubeData, fCubeDimension)); |
| 101 | paint.setColorFilter(colorCube); |
sugoi | ce68627 | 2014-10-09 05:27:23 -0700 | [diff] [blame] | 102 | canvas->drawBitmap(fBitmap, 0, 0, &paint); |
| 103 | } |
| 104 | } |
| 105 | |
| 106 | typedef Benchmark INHERITED; |
| 107 | }; |
| 108 | |
| 109 | /////////////////////////////////////////////////////////////////////////////// |
| 110 | |
| 111 | DEF_BENCH( return new ColorCubeBench(); ) |