OLD | NEW |
(Empty) | |
| 1 /* |
| 2 * Copyright 2015 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 #ifndef WrappedBenchmark_DEFINED |
| 9 #define WrappedBenchmark_DEFINED |
| 10 |
| 11 #include "Benchmark.h" |
| 12 #include "SkSurface.h" |
| 13 |
| 14 // Wrap some other benchmark to allow specialization to either |
| 15 // cpu or gpu backends. The derived class will override 'setupOffScreen' |
| 16 // to create an offscreen surface in which the actual rendering will occur. |
| 17 class WrappedBenchmark : public Benchmark { |
| 18 public: |
| 19 // Takes ownership of caller's ref on `bench`. |
| 20 explicit WrappedBenchmark(Benchmark* bench) : fBench(bench) {} |
| 21 |
| 22 const char* onGetName() override { return fBench->getName(); } |
| 23 const char* onGetUniqueName() override { return fBench->getUniqueName(); } |
| 24 |
| 25 void onDelayedSetup() override { fBench->delayedSetup(); } |
| 26 void onPerCanvasPreDraw(SkCanvas* canvas) override { |
| 27 fOffScreen.reset(this->setupOffScreen(canvas)); |
| 28 fBench->perCanvasPreDraw(fOffScreen->getCanvas()); |
| 29 } |
| 30 void onPreDraw(SkCanvas* canvas) override { |
| 31 SkASSERT(fOffScreen.get()); |
| 32 fBench->preDraw(fOffScreen->getCanvas()); |
| 33 } |
| 34 void onPostDraw(SkCanvas* canvas) override { |
| 35 SkASSERT(fOffScreen.get()); |
| 36 fBench->postDraw(fOffScreen->getCanvas()); |
| 37 } |
| 38 void onPerCanvasPostDraw(SkCanvas* canvas) override { |
| 39 SkASSERT(fOffScreen.get()); |
| 40 fBench->perCanvasPostDraw(fOffScreen->getCanvas()); |
| 41 } |
| 42 |
| 43 void onDraw(int loops, SkCanvas* canvas) override { |
| 44 SkASSERT(fOffScreen.get()); |
| 45 fBench->draw(loops, fOffScreen->getCanvas()); |
| 46 SkAutoTUnref<SkImage> image(fOffScreen->newImageSnapshot()); |
| 47 canvas->drawImage(image, 0,0); |
| 48 } |
| 49 |
| 50 virtual SkIPoint onGetSize() override { return fBench->getSize(); } |
| 51 |
| 52 private: |
| 53 virtual SkSurface* setupOffScreen(SkCanvas*)=0; |
| 54 |
| 55 SkAutoTUnref<SkSurface> fOffScreen; |
| 56 SkAutoTUnref<Benchmark> fBench; |
| 57 }; |
| 58 |
| 59 // Create a raster surface for off screen rendering |
| 60 class CpuWrappedBenchmark : public WrappedBenchmark { |
| 61 public: |
| 62 explicit CpuWrappedBenchmark(Benchmark* bench) : INHERITED(bench) {} |
| 63 |
| 64 private: |
| 65 SkSurface* setupOffScreen(SkCanvas* canvas) override { |
| 66 return SkSurface::NewRaster(canvas->imageInfo()); |
| 67 } |
| 68 |
| 69 typedef WrappedBenchmark INHERITED; |
| 70 }; |
| 71 |
| 72 // Create an MSAA & NVPR-enabled GPU backend |
| 73 class NvprWrappedBenchmark : public WrappedBenchmark { |
| 74 public: |
| 75 explicit NvprWrappedBenchmark(Benchmark* bench, int numSamples) |
| 76 : INHERITED(bench) |
| 77 , fNumSamples(numSamples) {} |
| 78 |
| 79 private: |
| 80 SkSurface* setupOffScreen(SkCanvas* canvas) override { |
| 81 return SkSurface::NewRenderTarget(canvas->getGrContext(), |
| 82 SkSurface::kNo_Budgeted, |
| 83 canvas->imageInfo(), |
| 84 fNumSamples); |
| 85 } |
| 86 |
| 87 int fNumSamples; |
| 88 typedef WrappedBenchmark INHERITED; |
| 89 }; |
| 90 |
| 91 #endif //WrappedBenchmark_DEFINED |
OLD | NEW |