/* * Copyright 2013 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/SkColorSpace.h" #include "include/core/SkImage.h" #include "include/core/SkImageFilter.h" #include "include/core/SkImageInfo.h" #include "include/core/SkMatrix.h" #include "include/core/SkPaint.h" #include "include/core/SkRect.h" #include "include/core/SkRefCnt.h" #include "include/core/SkRegion.h" #include "include/core/SkSize.h" #include "include/core/SkString.h" #include "include/core/SkSurface.h" #include "include/effects/SkImageFilters.h" #include "include/utils/SkRandom.h" #include "tools/ToolUtils.h" #include <utility> #define WIDTH 500 #define HEIGHT 500 static void draw_rects(SkCanvas* canvas) { SkPaint rectPaint; rectPaint.setColor(SK_ColorBLUE); canvas->drawRect(SkRect::MakeXYWH(0, 0, WIDTH / 2, HEIGHT / 2), rectPaint); rectPaint.setColor(0xBFFF0000); canvas->drawRect(SkRect::MakeXYWH(WIDTH / 2, 0, WIDTH / 2, HEIGHT / 2), rectPaint); rectPaint.setColor(0x3F00FF00); canvas->drawRect(SkRect::MakeXYWH(0, HEIGHT / 2, WIDTH / 2, HEIGHT / 2), rectPaint); rectPaint.setColor(SK_ColorTRANSPARENT); canvas->drawRect(SkRect::MakeXYWH(WIDTH / 2, HEIGHT / 2, WIDTH / 2, HEIGHT / 2), rectPaint); } static SkPaint create_filter_paint(SkIRect* cropRect = nullptr) { SkIRect rects[2]; rects[0] = SkIRect::MakeXYWH(0, 150, WIDTH, HEIGHT - 300); rects[1] = SkIRect::MakeXYWH(150, 0, WIDTH - 300, HEIGHT); SkRegion region; region.setRects(rects, 2); SkPaint paint; sk_sp<SkImageFilter> offset(SkImageFilters::Offset(25, 25, nullptr)); paint.setImageFilter( SkImageFilters::AlphaThreshold(region, 0.2f, 0.7f, std::move(offset), cropRect)); return paint; } class ImageAlphaThresholdGM : public skiagm::GM { public: ImageAlphaThresholdGM(bool useCropRect) : fUseCropRect(useCropRect) { this->setBGColor(SK_ColorWHITE); } protected: SkString onShortName() override { if (fUseCropRect) { return SkString("imagealphathreshold_crop"); } return SkString("imagealphathreshold"); } SkISize onISize() override { return SkISize::Make(WIDTH, HEIGHT); } void onDraw(SkCanvas* canvas) override { SkMatrix matrix; matrix.reset(); matrix.setTranslate(WIDTH * .1f, HEIGHT * .1f); matrix.postScale(.8f, .8f); canvas->concat(matrix); SkIRect cropRect = SkIRect::MakeLTRB(100, 100, WIDTH - 100, HEIGHT - 100); SkPaint paint = create_filter_paint(fUseCropRect ? &cropRect : nullptr); canvas->saveLayer(nullptr, &paint); draw_rects(canvas); canvas->restore(); } private: bool fUseCropRect; using INHERITED = GM; }; // Create a 'width' x 'height' SkSurface that matches the colorType of 'canvas' as // best we can static sk_sp<SkSurface> make_color_matching_surface(SkCanvas* canvas, int width, int height, SkAlphaType at) { SkColorType ct = canvas->imageInfo().colorType(); sk_sp<SkColorSpace> cs(canvas->imageInfo().refColorSpace()); if (kUnknown_SkColorType == ct) { // For backends that aren't yet color-space aware we just fallback to N32. ct = kN32_SkColorType; cs = nullptr; } else if (SkColorTypeIsAlwaysOpaque(ct)) { at = kOpaque_SkAlphaType; } SkImageInfo info = SkImageInfo::Make(width, height, ct, at, std::move(cs)); return ToolUtils::makeSurface(canvas, info); } class ImageAlphaThresholdSurfaceGM : public skiagm::GM { public: ImageAlphaThresholdSurfaceGM() { this->setBGColor(0xFFFFFFFF); } protected: SkString onShortName() override { return SkString("imagealphathreshold_surface"); } SkISize onISize() override { return SkISize::Make(WIDTH, HEIGHT); } DrawResult onDraw(SkCanvas* canvas, SkString* errorMsg) override { SkMatrix matrix; matrix.reset(); matrix.setTranslate(WIDTH * .1f, HEIGHT * .1f); matrix.postScale(.8f, .8f); canvas->concat(matrix); sk_sp<SkSurface> surface(make_color_matching_surface(canvas, WIDTH, HEIGHT, kPremul_SkAlphaType)); if (!surface) { *errorMsg = "make_color_matching_surface failed"; return DrawResult::kFail; } surface->getCanvas()->clear(SK_ColorTRANSPARENT); draw_rects(surface->getCanvas()); SkPaint paint = create_filter_paint(); canvas->clipRect(SkRect::MakeLTRB(100, 100, WIDTH - 100, HEIGHT - 100)); canvas->drawImage(surface->makeImageSnapshot().get(), 0, 0, &paint); return DrawResult::kOk; } private: using INHERITED = skiagm::GM; }; ////////////////////////////////////////////////////////////////////////////// DEF_GM(return new ImageAlphaThresholdGM(true);) DEF_GM(return new ImageAlphaThresholdGM(false);) DEF_GM(return new ImageAlphaThresholdSurfaceGM();) ////////////////////////////////////////////////////////////////////////////// static sk_sp<SkImage> make_img() { SkBitmap bitmap; bitmap.allocPixels(SkImageInfo::MakeS32(WIDTH, HEIGHT, kPremul_SkAlphaType)); SkCanvas canvas(bitmap); SkPaint paint; SkRect rect = SkRect::MakeWH(WIDTH, HEIGHT); SkRandom rnd; while (!rect.isEmpty()) { paint.setColor(rnd.nextU() | (0xFF << 24)); canvas.drawRect(rect, paint); rect.inset(25, 25); } return SkImage::MakeFromBitmap(bitmap); } DEF_SIMPLE_GM_BG(imagealphathreshold_image, canvas, WIDTH * 2, HEIGHT, SK_ColorBLACK) { sk_sp<SkImage> image(make_img()); SkIRect rects[2]; rects[0] = SkIRect::MakeXYWH(0, 150, WIDTH, HEIGHT - 300); rects[1] = SkIRect::MakeXYWH(150, 0, WIDTH - 300, HEIGHT); SkRegion region; region.setRects(rects, 2); SkPaint filterPaint; sk_sp<SkImageFilter> imageSource(SkImageFilters::Image(image)); filterPaint.setImageFilter(SkImageFilters::AlphaThreshold(region, 0.2f, 0.7f, std::move(imageSource))); canvas->saveLayer(nullptr, &filterPaint); canvas->restore(); canvas->translate(WIDTH, 0); canvas->drawImage(image, 0, 0); }