123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434 |
- /*
- * Copyright 2011 Google Inc.
- *
- * Use of this source code is governed by a BSD-style license that can be
- * found in the LICENSE file.
- */
- #include "gm/gm.h"
- #include "include/core/SkBitmap.h"
- #include "include/core/SkCanvas.h"
- #include "include/core/SkColor.h"
- #include "include/core/SkColorPriv.h"
- #include "include/core/SkColorSpace.h"
- #include "include/core/SkData.h"
- #include "include/core/SkEncodedImageFormat.h"
- #include "include/core/SkFilterQuality.h"
- #include "include/core/SkFont.h"
- #include "include/core/SkImage.h"
- #include "include/core/SkImageEncoder.h"
- #include "include/core/SkImageInfo.h"
- #include "include/core/SkPaint.h"
- #include "include/core/SkPicture.h"
- #include "include/core/SkPictureRecorder.h"
- #include "include/core/SkPixmap.h"
- #include "include/core/SkRect.h"
- #include "include/core/SkRefCnt.h"
- #include "include/core/SkScalar.h"
- #include "include/core/SkSize.h"
- #include "include/core/SkString.h"
- #include "include/core/SkSurface.h"
- #include "include/core/SkTypeface.h"
- #include "include/core/SkTypes.h"
- #include "include/private/SkMalloc.h"
- #include "src/core/SkAutoPixmapStorage.h"
- #include "src/core/SkReadBuffer.h"
- #include "src/core/SkWriteBuffer.h"
- #include "tools/ToolUtils.h"
- #include <functional>
- #include <utility>
- class GrContext;
- class GrRenderTargetContext;
- static void drawContents(SkSurface* surface, SkColor fillC) {
- SkSize size = SkSize::Make(SkIntToScalar(surface->width()),
- SkIntToScalar(surface->height()));
- SkCanvas* canvas = surface->getCanvas();
- SkScalar stroke = size.fWidth / 10;
- SkScalar radius = (size.fWidth - stroke) / 2;
- SkPaint paint;
- paint.setAntiAlias(true);
- paint.setColor(fillC);
- canvas->drawCircle(size.fWidth/2, size.fHeight/2, radius, paint);
- paint.setStyle(SkPaint::kStroke_Style);
- paint.setStrokeWidth(stroke);
- paint.setColor(SK_ColorBLACK);
- canvas->drawCircle(size.fWidth/2, size.fHeight/2, radius, paint);
- }
- static void test_surface(SkCanvas* canvas, SkSurface* surf, bool usePaint) {
- drawContents(surf, SK_ColorRED);
- sk_sp<SkImage> imgR = surf->makeImageSnapshot();
- if (true) {
- sk_sp<SkImage> imgR2 = surf->makeImageSnapshot();
- SkASSERT(imgR == imgR2);
- }
- drawContents(surf, SK_ColorGREEN);
- sk_sp<SkImage> imgG = surf->makeImageSnapshot();
- // since we've drawn after we snapped imgR, imgG will be a different obj
- SkASSERT(imgR != imgG);
- drawContents(surf, SK_ColorBLUE);
- SkPaint paint;
- // paint.setFilterBitmap(true);
- // paint.setAlpha(0x80);
- canvas->drawImage(imgR, 0, 0, usePaint ? &paint : nullptr);
- canvas->drawImage(imgG, 0, 80, usePaint ? &paint : nullptr);
- surf->draw(canvas, 0, 160, usePaint ? &paint : nullptr);
- SkRect src1, src2, src3;
- src1.iset(0, 0, surf->width(), surf->height());
- src2.iset(-surf->width() / 2, -surf->height() / 2,
- surf->width(), surf->height());
- src3.iset(0, 0, surf->width() / 2, surf->height() / 2);
- SkRect dst1, dst2, dst3, dst4;
- dst1.set(0, 240, 65, 305);
- dst2.set(0, 320, 65, 385);
- dst3.set(0, 400, 65, 465);
- dst4.set(0, 480, 65, 545);
- canvas->drawImageRect(imgR, src1, dst1, usePaint ? &paint : nullptr);
- canvas->drawImageRect(imgG, src2, dst2, usePaint ? &paint : nullptr);
- canvas->drawImageRect(imgR, src3, dst3, usePaint ? &paint : nullptr);
- canvas->drawImageRect(imgG, dst4, usePaint ? &paint : nullptr);
- }
- class ImageGM : public skiagm::GM {
- void* fBuffer;
- size_t fBufferSize;
- SkSize fSize;
- enum {
- W = 64,
- H = 64,
- RB = W * 4 + 8,
- };
- public:
- ImageGM() {
- fBufferSize = RB * H;
- fBuffer = sk_malloc_throw(fBufferSize);
- fSize.set(SkIntToScalar(W), SkIntToScalar(H));
- }
- ~ImageGM() override {
- sk_free(fBuffer);
- }
- protected:
- SkString onShortName() override {
- return SkString("image-surface");
- }
- SkISize onISize() override {
- return SkISize::Make(960, 1200);
- }
- void onDraw(SkCanvas* canvas) override {
- canvas->scale(2, 2);
- SkFont font(ToolUtils::create_portable_typeface(), 8);
- canvas->drawString("Original Img", 10, 60, font, SkPaint());
- canvas->drawString("Modified Img", 10, 140, font, SkPaint());
- canvas->drawString("Cur Surface", 10, 220, font, SkPaint());
- canvas->drawString("Full Crop", 10, 300, font, SkPaint());
- canvas->drawString("Over-crop", 10, 380, font, SkPaint());
- canvas->drawString("Upper-left", 10, 460, font, SkPaint());
- canvas->drawString("No Crop", 10, 540, font, SkPaint());
- canvas->drawString("Pre-Alloc Img", 80, 10, font, SkPaint());
- canvas->drawString("New Alloc Img", 160, 10, font, SkPaint());
- canvas->drawString( "GPU", 265, 10, font, SkPaint());
- canvas->translate(80, 20);
- // since we draw into this directly, we need to start fresh
- sk_bzero(fBuffer, fBufferSize);
- SkImageInfo info = SkImageInfo::MakeN32Premul(W, H);
- sk_sp<SkSurface> surf0(SkSurface::MakeRasterDirect(info, fBuffer, RB));
- sk_sp<SkSurface> surf1(SkSurface::MakeRaster(info));
- sk_sp<SkSurface> surf2(SkSurface::MakeRenderTarget(canvas->getGrContext(),
- SkBudgeted::kNo, info));
- test_surface(canvas, surf0.get(), true);
- canvas->translate(80, 0);
- test_surface(canvas, surf1.get(), true);
- if (surf2) {
- canvas->translate(80, 0);
- test_surface(canvas, surf2.get(), true);
- }
- }
- private:
- typedef skiagm::GM INHERITED;
- };
- DEF_GM( return new ImageGM; )
- ///////////////////////////////////////////////////////////////////////////////////////////////////
- static void draw_pixmap(SkCanvas* canvas, const SkPixmap& pmap) {
- SkBitmap bitmap;
- bitmap.installPixels(pmap);
- canvas->drawBitmap(bitmap, 0, 0, nullptr);
- }
- static void show_scaled_pixels(SkCanvas* canvas, SkImage* image) {
- SkAutoCanvasRestore acr(canvas, true);
- canvas->drawImage(image, 0, 0, nullptr);
- canvas->translate(110, 10);
- const SkImageInfo info = SkImageInfo::MakeN32Premul(40, 40);
- SkAutoPixmapStorage storage;
- storage.alloc(info);
- const SkImage::CachingHint chints[] = {
- SkImage::kAllow_CachingHint, SkImage::kDisallow_CachingHint,
- };
- const SkFilterQuality qualities[] = {
- kNone_SkFilterQuality, kLow_SkFilterQuality, kMedium_SkFilterQuality, kHigh_SkFilterQuality,
- };
- for (auto ch : chints) {
- canvas->save();
- for (auto q : qualities) {
- if (image->scalePixels(storage, q, ch)) {
- draw_pixmap(canvas, storage);
- }
- canvas->translate(70, 0);
- }
- canvas->restore();
- canvas->translate(0, 45);
- }
- }
- static void draw_contents(SkCanvas* canvas) {
- SkPaint paint;
- paint.setStyle(SkPaint::kStroke_Style);
- paint.setStrokeWidth(20);
- canvas->drawCircle(50, 50, 35, paint);
- }
- static sk_sp<SkImage> make_raster(const SkImageInfo& info, GrContext*, void (*draw)(SkCanvas*)) {
- auto surface(SkSurface::MakeRaster(info));
- draw(surface->getCanvas());
- return surface->makeImageSnapshot();
- }
- static sk_sp<SkImage> make_picture(const SkImageInfo& info, GrContext*, void (*draw)(SkCanvas*)) {
- SkPictureRecorder recorder;
- draw(recorder.beginRecording(SkRect::MakeIWH(info.width(), info.height())));
- return SkImage::MakeFromPicture(recorder.finishRecordingAsPicture(),
- info.dimensions(), nullptr, nullptr, SkImage::BitDepth::kU8,
- SkColorSpace::MakeSRGB());
- }
- static sk_sp<SkImage> make_codec(const SkImageInfo& info, GrContext*, void (*draw)(SkCanvas*)) {
- sk_sp<SkImage> image(make_raster(info, nullptr, draw));
- return SkImage::MakeFromEncoded(image->encodeToData());
- }
- static sk_sp<SkImage> make_gpu(const SkImageInfo& info, GrContext* ctx, void (*draw)(SkCanvas*)) {
- if (!ctx) { return nullptr; }
- auto surface(SkSurface::MakeRenderTarget(ctx, SkBudgeted::kNo, info));
- if (!surface) { return nullptr; }
- draw(surface->getCanvas());
- return surface->makeImageSnapshot();
- }
- typedef sk_sp<SkImage> (*ImageMakerProc)(const SkImageInfo&, GrContext*, void (*)(SkCanvas*));
- class ScalePixelsGM : public skiagm::GM {
- public:
- ScalePixelsGM() {}
- protected:
- SkString onShortName() override {
- return SkString("scale-pixels");
- }
- SkISize onISize() override {
- return SkISize::Make(960, 1200);
- }
- void onDraw(SkCanvas* canvas) override {
- const SkImageInfo info = SkImageInfo::MakeN32Premul(100, 100);
- const ImageMakerProc procs[] = {
- make_codec, make_raster, make_picture, make_codec, make_gpu,
- };
- for (auto& proc : procs) {
- sk_sp<SkImage> image(proc(info, canvas->getGrContext(), draw_contents));
- if (image) {
- show_scaled_pixels(canvas, image.get());
- }
- canvas->translate(0, 120);
- }
- }
- private:
- typedef skiagm::GM INHERITED;
- };
- DEF_GM( return new ScalePixelsGM; )
- ///////////////////////////////////////////////////////////////////////////////////////////////////
- DEF_SIMPLE_GPU_GM(new_texture_image, context, rtc, canvas, 280, 60) {
- auto render_image = [](SkCanvas* canvas) {
- canvas->clear(SK_ColorBLUE);
- SkPaint paint;
- paint.setColor(SK_ColorRED);
- canvas->drawRect(SkRect::MakeXYWH(10.f,10.f,10.f,10.f), paint);
- paint.setColor(SK_ColorGREEN);
- canvas->drawRect(SkRect::MakeXYWH(30.f,10.f,10.f,10.f), paint);
- paint.setColor(SK_ColorYELLOW);
- canvas->drawRect(SkRect::MakeXYWH(10.f,30.f,10.f,10.f), paint);
- paint.setColor(SK_ColorCYAN);
- canvas->drawRect(SkRect::MakeXYWH(30.f,30.f,10.f,10.f), paint);
- };
- static constexpr int kSize = 50;
- SkBitmap bmp;
- bmp.allocPixels(SkImageInfo::MakeS32(kSize, kSize, kPremul_SkAlphaType));
- SkCanvas bmpCanvas(bmp);
- render_image(&bmpCanvas);
- std::function<sk_sp<SkImage>()> imageFactories[] = {
- // Create sw raster image.
- [bmp] {
- return SkImage::MakeFromBitmap(bmp);
- },
- // Create encoded image.
- [bmp] {
- auto src = SkEncodeBitmap(bmp, SkEncodedImageFormat::kPNG, 100);
- return SkImage::MakeFromEncoded(std::move(src));
- },
- // Create YUV encoded image.
- [bmp] {
- auto src = SkEncodeBitmap(bmp, SkEncodedImageFormat::kJPEG, 100);
- return SkImage::MakeFromEncoded(std::move(src));
- },
- // Create a picture image.
- [render_image] {
- SkPictureRecorder recorder;
- SkCanvas* canvas = recorder.beginRecording(SkIntToScalar(kSize), SkIntToScalar(kSize));
- render_image(canvas);
- sk_sp<SkColorSpace> srgbColorSpace = SkColorSpace::MakeSRGB();
- return SkImage::MakeFromPicture(recorder.finishRecordingAsPicture(),
- SkISize::Make(kSize, kSize), nullptr, nullptr,
- SkImage::BitDepth::kU8, srgbColorSpace);
- },
- // Create a texture image
- [context, render_image]() -> sk_sp<SkImage> {
- auto surface(SkSurface::MakeRenderTarget(context, SkBudgeted::kYes,
- SkImageInfo::MakeS32(kSize, kSize,
- kPremul_SkAlphaType)));
- if (!surface) {
- return nullptr;
- }
- render_image(surface->getCanvas());
- return surface->makeImageSnapshot();
- }
- };
- constexpr SkScalar kPad = 5.f;
- canvas->translate(kPad, kPad);
- for (auto factory : imageFactories) {
- auto image(factory());
- if (image) {
- sk_sp<SkImage> texImage(image->makeTextureImage(context,
- canvas->imageInfo().colorSpace()));
- if (texImage) {
- canvas->drawImage(texImage, 0, 0);
- }
- }
- canvas->translate(kSize + kPad, 0);
- }
- }
- static void draw_pixmap(SkCanvas* canvas, const SkPixmap& pm, SkScalar x, SkScalar y) {
- canvas->drawImage(SkImage::MakeRasterCopy(pm), x, y, nullptr);
- }
- static void slam_ff(const SkPixmap& pm) {
- for (int y = 0; y < pm.height(); ++y) {
- for (int x = 0; x < pm.width(); ++x) {
- *pm.writable_addr32(x, y) = *pm.addr32(x, y) | SkPackARGB32(0xFF, 0, 0, 0);
- }
- }
- }
- DEF_SIMPLE_GM(scalepixels_unpremul, canvas, 1080, 280) {
- SkImageInfo info = SkImageInfo::MakeN32(16, 16, kUnpremul_SkAlphaType);
- SkAutoPixmapStorage pm;
- pm.alloc(info);
- for (int y = 0; y < 16; ++y) {
- for (int x = 0; x < 16; ++x) {
- *pm.writable_addr32(x, y) = SkPackARGB32NoCheck(0, (y << 4) | y, (x << 4) | x, 0xFF);
- }
- }
- SkAutoPixmapStorage pm2;
- pm2.alloc(SkImageInfo::MakeN32(256, 256, kUnpremul_SkAlphaType));
- const SkFilterQuality qualities[] = {
- kNone_SkFilterQuality, kLow_SkFilterQuality, kMedium_SkFilterQuality, kHigh_SkFilterQuality
- };
- for (auto fq : qualities) {
- pm.scalePixels(pm2, fq);
- slam_ff(pm2);
- draw_pixmap(canvas, pm2, 10, 10);
- canvas->translate(pm2.width() + 10.0f, 0);
- }
- }
- ///////////////////////////////////////////////////////////////////////////////////////////////////
- static sk_sp<SkImage> make_lazy_image(SkSurface* surf) {
- surf->getCanvas()->drawCircle(100, 100, 100, SkPaint());
- sk_sp<SkData> data = surf->makeImageSnapshot()->encodeToData();
- if (!data) {
- return nullptr;
- }
- return SkImage::MakeFromEncoded(std::move(data));
- }
- static sk_sp<SkImage> serial_deserial(SkImage* img) {
- SkBinaryWriteBuffer writer;
- writer.writeImage(img);
- size_t length = writer.bytesWritten();
- auto data = SkData::MakeUninitialized(length);
- writer.writeToMemory(data->writable_data());
- SkReadBuffer reader(data->data(), length);
- return reader.readImage();
- }
- DEF_SIMPLE_GM_CAN_FAIL(image_subset, canvas, errorMsg, 440, 220) {
- SkImageInfo info = SkImageInfo::MakeN32Premul(200, 200, nullptr);
- auto surf = ToolUtils::makeSurface(canvas, info, nullptr);
- auto img = make_lazy_image(surf.get());
- if (!img) {
- *errorMsg = "Failed to make lazy image.";
- return skiagm::DrawResult::kFail;
- }
- canvas->drawImage(img, 10, 10, nullptr);
- auto sub = img->makeSubset({100, 100, 200, 200});
- canvas->drawImage(sub, 220, 10);
- sub = serial_deserial(sub.get());
- canvas->drawImage(sub, 220+110, 10);
- return skiagm::DrawResult::kOk;
- }
|