| /* |
| * 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 "src/base/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, SkSamplingOptions(), &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 bitmap.asImage(); |
| } |
| |
| 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); |
| } |