skia2/gm/imagegeneratorexternal.cpp
Florin Malita 58cda8fa49 External image generator API: return an SkPixmap instead of SkImage
R=reed@google.com
BUG=skia:5806

GOLD_TRYBOT_URL= https://gold.skia.org/search?issue=5134

Change-Id: Ib0215f63b67f4d5b619ebe1f1b70c7d95e604871
Reviewed-on: https://skia-review.googlesource.com/5134
Reviewed-by: Mike Reed <reed@google.com>
Commit-Queue: Florin Malita <fmalita@chromium.org>
2016-11-25 00:12:32 +00:00

195 lines
6.6 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.
*/
#include "gm.h"
#include "SkImage.h"
#include "SkImageGenerator.h"
#include "SkMutex.h"
#include "SkSurface.h"
#include "SkTArray.h"
namespace {
void release_proc(void*, void* releaseCtx) {
reinterpret_cast<SkImage*>(releaseCtx)->unref();
}
class ExternalGenerator : public SkImageGenerator {
public:
ExternalGenerator(const SkISize size)
: INHERITED(SkImageInfo::MakeN32Premul(size.width(), size.height())) {
int level = 0;
for (int size = kMaxSize; size; size /= 2) {
sk_sp<SkSurface> surface = SkSurface::MakeRasterN32Premul(size, size);
DrawRings(surface->getCanvas(), 0xff008000, level++);
fMips.emplace_back(surface->makeImageSnapshot());
}
}
virtual ~ExternalGenerator() {}
protected:
bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes,
SkPMColor[], int*) override {
sk_sp<SkSurface> s = SkSurface::MakeRasterDirect(info, pixels, rowBytes);
s->getCanvas()->clear(SK_ColorTRANSPARENT);
DrawRings(s->getCanvas(), SK_ColorRED);
return true;
}
bool onAccessScaledImage(const SkRect& src, const SkMatrix& matrix, SkFilterQuality,
ScaledImageRec* rec) override {
// Not strictly needed for this immutable class.
SkAutoExclusive lock(fMutex);
SkSize scaleSize;
if (!matrix.decomposeScale(&scaleSize, nullptr)) {
return false;
}
scaleSize.set(scaleSize.width() * this->getInfo().width() / kMaxSize,
scaleSize.height() * this->getInfo().height() / kMaxSize);
const SkScalar scale = SkTMin(scaleSize.width(), scaleSize.height());
const int lvl = SkScalarFloorToInt(-SkScalarLog2(scale));
const sk_sp<SkImage>& img = fMips[SkTPin(lvl, 0, fMips.count())];
SkAssertResult(img->peekPixels(&rec->fPixmap));
const SkRect origBounds = SkRect::Make(this->getInfo().bounds());
const SkRect newBounds = SkRect::Make(img->bounds());
SkMatrix srcMap = SkMatrix::MakeScale(newBounds.width() / origBounds.width(),
newBounds.height() / origBounds.height());
srcMap.preTranslate(src.x(), src.y());
srcMap.mapRect(&rec->fSrcRect, SkRect::MakeWH(src.width(), src.height()));
rec->fQuality = kLow_SkFilterQuality;
rec->fReleaseProc = release_proc;
rec->fReleaseCtx = SkRef(img.get());
return true;
}
private:
static void DrawRings(SkCanvas* c, SkColor color, int lvl = 0) {
static constexpr SkScalar kStep = 0.2f;
SkRect rect = SkRect::MakeWH(1, 1);
SkPaint p;
p.setStyle(SkPaint::kStroke_Style);
p.setStrokeWidth(0.02f);
p.setAntiAlias(true);
p.setColor(color);
c->concat(SkMatrix::MakeRectToRect(SkRect::MakeWH(1, 1),
SkRect::MakeIWH(c->imageInfo().width(),
c->imageInfo().height()),
SkMatrix::kFill_ScaleToFit));
while (!rect.isEmpty()) {
c->drawRect(rect, p);
rect.inset(kStep, kStep);
}
static constexpr SkScalar kTxtSize = 0.2f;
SkASSERT(lvl >= 0 && lvl <= 9);
const char label = '0' + lvl;
p.setTextSize(kTxtSize);
p.setLinearText(true);
p.setStyle(SkPaint::kFill_Style);
SkRect labelBounds;
p.measureText(&label, 1, &labelBounds);
c->drawText(&label, 1, 0.5f - labelBounds.width() / 2, 0.5f + labelBounds.height() / 2, p);
}
SkMutex fMutex;
static constexpr int kMaxSize = 512;
SkTArray<sk_sp<SkImage>> fMips;
typedef SkImageGenerator INHERITED;
};
} // anonymous ns
class ImageGenExternalGM : public skiagm::GM {
public:
explicit ImageGenExternalGM(bool useShader) : fUseShader(useShader) {}
protected:
SkString onShortName() override {
return SkStringPrintf("ImageGeneratorExternal%s", fUseShader ? "_shader" : "_rect");
}
SkISize onISize() override {
return SkISize::Make(800, 800);
}
void onOnceBeforeDraw() override {
fImage = SkImage::MakeFromGenerator(new ExternalGenerator(SkISize::Make(kGeneratorSize,
kGeneratorSize)));
}
void onDraw(SkCanvas* canvas) override {
static const SkRect gSubsets[] = {
SkRect::MakeLTRB(0 , 0 , 1 , 1 ),
SkRect::MakeLTRB(0 , 0 , 0.5f , 0.5f ),
SkRect::MakeLTRB(0.5f , 0 , 1 , 0.5f ),
SkRect::MakeLTRB(0.5f , 0.5f , 1 , 1 ),
SkRect::MakeLTRB(0 , 0.5f , 0.5f , 1 ),
SkRect::MakeLTRB(0.25f, 0.25f, 0.75f, 0.75f),
};
SkPaint p;
p.setFilterQuality(kLow_SkFilterQuality);
for (int i = 1; i <= 4; ++i) {
const SkRect dst = SkRect::MakeIWH(kGeneratorSize / i, kGeneratorSize / i);
canvas->save();
for (size_t j = 0; j < SK_ARRAY_COUNT(gSubsets); ++j) {
SkRect subset = gSubsets[j];
subset.set(kGeneratorSize * subset.left(),
kGeneratorSize * subset.top(),
kGeneratorSize * subset.right(),
kGeneratorSize * subset.bottom());
this->drawSubset(canvas, subset, dst, p);
canvas->translate(kGeneratorSize * 1.1f, 0);
}
canvas->restore();
canvas->translate(0, dst.height() * 1.2f);
}
}
private:
void drawSubset(SkCanvas* canvas, const SkRect& src, const SkRect& dst,
const SkPaint& paint) const {
if (fUseShader) {
SkPaint p(paint);
SkMatrix localMatrix = SkMatrix::MakeRectToRect(src, dst, SkMatrix::kFill_ScaleToFit);
p.setShader(fImage->makeShader(SkShader::kClamp_TileMode,
SkShader::kClamp_TileMode,
&localMatrix));
canvas->drawRect(dst, p);
} else {
canvas->drawImageRect(fImage, src, dst, &paint);
}
}
static constexpr int kGeneratorSize = 200;
sk_sp<SkImage> fImage;
bool fUseShader;
typedef skiagm::GM INHERITED;
};
DEF_GM( return new ImageGenExternalGM(false); )
DEF_GM( return new ImageGenExternalGM(true); )