100 lines
3.3 KiB
C++
100 lines
3.3 KiB
C++
/*
|
|
* Copyright 2015 Google Inc.
|
|
*
|
|
* Use of this source code is governed by a BSD-style license that can be
|
|
* found in the LICENSE file.
|
|
*/
|
|
|
|
#ifndef WrappedBenchmark_DEFINED
|
|
#define WrappedBenchmark_DEFINED
|
|
|
|
#include "Benchmark.h"
|
|
#include "SkSurface.h"
|
|
|
|
// Wrap some other benchmark to allow specialization to either
|
|
// cpu or gpu backends. The derived class will override 'setupOffScreen'
|
|
// to create an offscreen surface in which the actual rendering will occur.
|
|
class WrappedBenchmark : public Benchmark {
|
|
public:
|
|
// Takes ownership of caller's ref on `bench`.
|
|
explicit WrappedBenchmark(const SkSurfaceProps& surfaceProps, Benchmark* bench)
|
|
: fSurfaceProps(surfaceProps)
|
|
, fBench(bench) {}
|
|
|
|
const SkSurfaceProps& surfaceProps() const { return fSurfaceProps; }
|
|
|
|
const char* onGetName() override { return fBench->getName(); }
|
|
const char* onGetUniqueName() override { return fBench->getUniqueName(); }
|
|
|
|
void onDelayedSetup() override { fBench->delayedSetup(); }
|
|
void onPerCanvasPreDraw(SkCanvas* canvas) override {
|
|
fOffScreen.reset(this->setupOffScreen(canvas));
|
|
fBench->perCanvasPreDraw(fOffScreen->getCanvas());
|
|
}
|
|
void onPreDraw(SkCanvas* canvas) override {
|
|
SkASSERT(fOffScreen.get());
|
|
fBench->preDraw(fOffScreen->getCanvas());
|
|
}
|
|
void onPostDraw(SkCanvas* canvas) override {
|
|
SkASSERT(fOffScreen.get());
|
|
fBench->postDraw(fOffScreen->getCanvas());
|
|
}
|
|
void onPerCanvasPostDraw(SkCanvas* canvas) override {
|
|
SkASSERT(fOffScreen.get());
|
|
fBench->perCanvasPostDraw(fOffScreen->getCanvas());
|
|
}
|
|
|
|
void onDraw(int loops, SkCanvas* canvas) override {
|
|
SkASSERT(fOffScreen.get());
|
|
fBench->draw(loops, fOffScreen->getCanvas());
|
|
SkAutoTUnref<SkImage> image(fOffScreen->newImageSnapshot());
|
|
canvas->drawImage(image, 0,0);
|
|
}
|
|
|
|
virtual SkIPoint onGetSize() override { return fBench->getSize(); }
|
|
|
|
private:
|
|
virtual SkSurface* setupOffScreen(SkCanvas*)=0;
|
|
|
|
SkSurfaceProps fSurfaceProps;
|
|
SkAutoTUnref<SkSurface> fOffScreen;
|
|
SkAutoTUnref<Benchmark> fBench;
|
|
};
|
|
|
|
// Create a raster surface for off screen rendering
|
|
class CpuWrappedBenchmark : public WrappedBenchmark {
|
|
public:
|
|
explicit CpuWrappedBenchmark(const SkSurfaceProps& surfaceProps, Benchmark* bench)
|
|
: INHERITED(surfaceProps, bench) {}
|
|
|
|
private:
|
|
SkSurface* setupOffScreen(SkCanvas* canvas) override {
|
|
return SkSurface::NewRaster(canvas->imageInfo(), &this->surfaceProps());
|
|
}
|
|
|
|
typedef WrappedBenchmark INHERITED;
|
|
};
|
|
|
|
// Create an MSAA & NVPR-enabled GPU backend
|
|
class NvprWrappedBenchmark : public WrappedBenchmark {
|
|
public:
|
|
explicit NvprWrappedBenchmark(const SkSurfaceProps& surfaceProps, Benchmark* bench,
|
|
int numSamples)
|
|
: INHERITED(surfaceProps, bench)
|
|
, fNumSamples(numSamples) {}
|
|
|
|
private:
|
|
SkSurface* setupOffScreen(SkCanvas* canvas) override {
|
|
return SkSurface::NewRenderTarget(canvas->getGrContext(),
|
|
SkSurface::kNo_Budgeted,
|
|
canvas->imageInfo(),
|
|
fNumSamples,
|
|
&this->surfaceProps());
|
|
}
|
|
|
|
int fNumSamples;
|
|
typedef WrappedBenchmark INHERITED;
|
|
};
|
|
|
|
#endif //WrappedBenchmark_DEFINED
|