123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511 |
- /*
- * 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 "src/core/SkArenaAlloc.h"
- #include "src/core/SkBitmapController.h"
- #include "src/core/SkBitmapProvider.h"
- #include "src/core/SkColorSpacePriv.h"
- #include "src/core/SkColorSpaceXformSteps.h"
- #include "src/core/SkRasterPipeline.h"
- #include "src/core/SkReadBuffer.h"
- #include "src/core/SkWriteBuffer.h"
- #include "src/image/SkImage_Base.h"
- #include "src/shaders/SkBitmapProcShader.h"
- #include "src/shaders/SkEmptyShader.h"
- #include "src/shaders/SkImageShader.h"
- /**
- * We are faster in clamp, so always use that tiling when we can.
- */
- static SkTileMode optimize(SkTileMode tm, int dimension) {
- SkASSERT(dimension > 0);
- #ifdef SK_BUILD_FOR_ANDROID_FRAMEWORK
- // need to update frameworks/base/libs/hwui/tests/unit/SkiaBehaviorTests.cpp:55 to allow
- // for transforming to clamp.
- return tm;
- #else
- return dimension == 1 ? SkTileMode::kClamp : tm;
- #endif
- }
- SkImageShader::SkImageShader(sk_sp<SkImage> img,
- SkTileMode tmx, SkTileMode tmy,
- const SkMatrix* localMatrix,
- bool clampAsIfUnpremul)
- : INHERITED(localMatrix)
- , fImage(std::move(img))
- , fTileModeX(optimize(tmx, fImage->width()))
- , fTileModeY(optimize(tmy, fImage->height()))
- , fClampAsIfUnpremul(clampAsIfUnpremul)
- {}
- // fClampAsIfUnpremul is always false when constructed through public APIs,
- // so there's no need to read or write it here.
- sk_sp<SkFlattenable> SkImageShader::CreateProc(SkReadBuffer& buffer) {
- auto tmx = buffer.read32LE<SkTileMode>(SkTileMode::kLastTileMode);
- auto tmy = buffer.read32LE<SkTileMode>(SkTileMode::kLastTileMode);
- SkMatrix localMatrix;
- buffer.readMatrix(&localMatrix);
- sk_sp<SkImage> img = buffer.readImage();
- if (!img) {
- return nullptr;
- }
- return SkImageShader::Make(std::move(img), tmx, tmy, &localMatrix);
- }
- void SkImageShader::flatten(SkWriteBuffer& buffer) const {
- buffer.writeUInt((unsigned)fTileModeX);
- buffer.writeUInt((unsigned)fTileModeY);
- buffer.writeMatrix(this->getLocalMatrix());
- buffer.writeImage(fImage.get());
- SkASSERT(fClampAsIfUnpremul == false);
- }
- bool SkImageShader::isOpaque() const {
- return fImage->isOpaque() &&
- fTileModeX != SkTileMode::kDecal && fTileModeY != SkTileMode::kDecal;
- }
- #ifdef SK_ENABLE_LEGACY_SHADERCONTEXT
- static bool legacy_shader_can_handle(const SkMatrix& inv) {
- if (!inv.isScaleTranslate()) {
- return false;
- }
- // legacy code uses SkFixed 32.32, so ensure the inverse doesn't map device coordinates
- // out of range.
- const SkScalar max_dev_coord = 32767.0f;
- SkRect src;
- SkAssertResult(inv.mapRect(&src, SkRect::MakeWH(max_dev_coord, max_dev_coord)));
- // take 1/4 of max signed 32bits so we have room to subtract local values
- const SkScalar max_fixed32dot32 = SK_MaxS32 * 0.25f;
- if (!SkRect::MakeLTRB(-max_fixed32dot32, -max_fixed32dot32,
- max_fixed32dot32, max_fixed32dot32).contains(src)) {
- return false;
- }
- // legacy shader impl should be able to handle these matrices
- return true;
- }
- SkShaderBase::Context* SkImageShader::onMakeContext(const ContextRec& rec,
- SkArenaAlloc* alloc) const {
- if (fImage->alphaType() == kUnpremul_SkAlphaType) {
- return nullptr;
- }
- if (fImage->colorType() != kN32_SkColorType) {
- return nullptr;
- }
- if (fTileModeX != fTileModeY) {
- return nullptr;
- }
- if (fTileModeX == SkTileMode::kDecal || fTileModeY == SkTileMode::kDecal) {
- return nullptr;
- }
- // SkBitmapProcShader stores bitmap coordinates in a 16bit buffer,
- // so it can't handle bitmaps larger than 65535.
- //
- // We back off another bit to 32767 to make small amounts of
- // intermediate math safe, e.g. in
- //
- // SkFixed fx = ...;
- // fx = tile(fx + SK_Fixed1);
- //
- // we want to make sure (fx + SK_Fixed1) never overflows.
- if (fImage-> width() > 32767 ||
- fImage->height() > 32767) {
- return nullptr;
- }
- SkMatrix inv;
- if (!this->computeTotalInverse(*rec.fMatrix, rec.fLocalMatrix, &inv) ||
- !legacy_shader_can_handle(inv)) {
- return nullptr;
- }
- if (!rec.isLegacyCompatible(fImage->colorSpace())) {
- return nullptr;
- }
- return SkBitmapProcLegacyShader::MakeContext(*this, fTileModeX, fTileModeY,
- SkBitmapProvider(fImage.get()), rec, alloc);
- }
- #endif
- SkImage* SkImageShader::onIsAImage(SkMatrix* texM, SkTileMode xy[]) const {
- if (texM) {
- *texM = this->getLocalMatrix();
- }
- if (xy) {
- xy[0] = fTileModeX;
- xy[1] = fTileModeY;
- }
- return const_cast<SkImage*>(fImage.get());
- }
- sk_sp<SkShader> SkImageShader::Make(sk_sp<SkImage> image,
- SkTileMode tmx, SkTileMode tmy,
- const SkMatrix* localMatrix,
- bool clampAsIfUnpremul) {
- if (!image) {
- return sk_make_sp<SkEmptyShader>();
- }
- return sk_sp<SkShader>{ new SkImageShader(image, tmx, tmy, localMatrix, clampAsIfUnpremul) };
- }
- ///////////////////////////////////////////////////////////////////////////////////////////////////
- #if SK_SUPPORT_GPU
- #include "include/private/GrRecordingContext.h"
- #include "src/gpu/GrCaps.h"
- #include "src/gpu/GrColorSpaceInfo.h"
- #include "src/gpu/GrRecordingContextPriv.h"
- #include "src/gpu/SkGr.h"
- #include "src/gpu/effects/GrBicubicEffect.h"
- #include "src/gpu/effects/generated/GrSimpleTextureEffect.h"
- static GrSamplerState::WrapMode tile_mode_to_wrap_mode(const SkTileMode tileMode) {
- switch (tileMode) {
- case SkTileMode::kClamp:
- return GrSamplerState::WrapMode::kClamp;
- case SkTileMode::kRepeat:
- return GrSamplerState::WrapMode::kRepeat;
- case SkTileMode::kMirror:
- return GrSamplerState::WrapMode::kMirrorRepeat;
- case SkTileMode::kDecal:
- return GrSamplerState::WrapMode::kClampToBorder;
- }
- SK_ABORT("Unknown tile mode.");
- return GrSamplerState::WrapMode::kClamp;
- }
- std::unique_ptr<GrFragmentProcessor> SkImageShader::asFragmentProcessor(
- const GrFPArgs& args) const {
- const auto lm = this->totalLocalMatrix(args.fPreLocalMatrix, args.fPostLocalMatrix);
- SkMatrix lmInverse;
- if (!lm->invert(&lmInverse)) {
- return nullptr;
- }
- GrSamplerState::WrapMode wrapModes[] = {tile_mode_to_wrap_mode(fTileModeX),
- tile_mode_to_wrap_mode(fTileModeY)};
- // If either domainX or domainY are un-ignored, a texture domain effect has to be used to
- // implement the decal mode (while leaving non-decal axes alone). The wrap mode originally
- // clamp-to-border is reset to clamp since the hw cannot implement it directly.
- GrTextureDomain::Mode domainX = GrTextureDomain::kIgnore_Mode;
- GrTextureDomain::Mode domainY = GrTextureDomain::kIgnore_Mode;
- if (!args.fContext->priv().caps()->clampToBorderSupport()) {
- if (wrapModes[0] == GrSamplerState::WrapMode::kClampToBorder) {
- domainX = GrTextureDomain::kDecal_Mode;
- wrapModes[0] = GrSamplerState::WrapMode::kClamp;
- }
- if (wrapModes[1] == GrSamplerState::WrapMode::kClampToBorder) {
- domainY = GrTextureDomain::kDecal_Mode;
- wrapModes[1] = GrSamplerState::WrapMode::kClamp;
- }
- }
- // Must set wrap and filter on the sampler before requesting a texture. In two places below
- // we check the matrix scale factors to determine how to interpret the filter quality setting.
- // This completely ignores the complexity of the drawVertices case where explicit local coords
- // are provided by the caller.
- bool doBicubic;
- GrSamplerState::Filter textureFilterMode = GrSkFilterQualityToGrFilterMode(
- args.fFilterQuality, *args.fViewMatrix, *lm,
- args.fContext->priv().options().fSharpenMipmappedTextures, &doBicubic);
- GrSamplerState samplerState(wrapModes, textureFilterMode);
- SkScalar scaleAdjust[2] = { 1.0f, 1.0f };
- sk_sp<GrTextureProxy> proxy(as_IB(fImage)->asTextureProxyRef(args.fContext, samplerState,
- scaleAdjust));
- if (!proxy) {
- return nullptr;
- }
- GrPixelConfig config = proxy->config();
- bool isAlphaOnly = GrPixelConfigIsAlphaOnly(config);
- lmInverse.postScale(scaleAdjust[0], scaleAdjust[1]);
- std::unique_ptr<GrFragmentProcessor> inner;
- if (doBicubic) {
- // domainX and domainY will properly apply the decal effect with the texture domain used in
- // the bicubic filter if clamp to border was unsupported in hardware
- static constexpr auto kDir = GrBicubicEffect::Direction::kXY;
- inner = GrBicubicEffect::Make(std::move(proxy), lmInverse, wrapModes, domainX, domainY,
- kDir, fImage->alphaType());
- } else {
- if (domainX != GrTextureDomain::kIgnore_Mode || domainY != GrTextureDomain::kIgnore_Mode) {
- SkRect domain = GrTextureDomain::MakeTexelDomain(
- SkIRect::MakeWH(proxy->width(), proxy->height()),
- domainX, domainY);
- inner = GrTextureDomainEffect::Make(std::move(proxy), lmInverse, domain,
- domainX, domainY, samplerState);
- } else {
- inner = GrSimpleTextureEffect::Make(std::move(proxy), lmInverse, samplerState);
- }
- }
- inner = GrColorSpaceXformEffect::Make(std::move(inner), fImage->colorSpace(),
- fImage->alphaType(),
- args.fDstColorSpaceInfo->colorSpace());
- if (isAlphaOnly) {
- return inner;
- } else if (args.fInputColorIsOpaque) {
- return GrFragmentProcessor::OverrideInput(std::move(inner), SK_PMColor4fWHITE, false);
- }
- return GrFragmentProcessor::MulChildByInputAlpha(std::move(inner));
- }
- #endif
- ///////////////////////////////////////////////////////////////////////////////////////////////////
- #include "src/core/SkImagePriv.h"
- sk_sp<SkShader> SkMakeBitmapShader(const SkBitmap& src, SkTileMode tmx, SkTileMode tmy,
- const SkMatrix* localMatrix, SkCopyPixelsMode cpm) {
- return SkImageShader::Make(SkMakeImageFromRasterBitmap(src, cpm),
- tmx, tmy, localMatrix);
- }
- sk_sp<SkShader> SkMakeBitmapShaderForPaint(const SkPaint& paint, const SkBitmap& src,
- SkTileMode tmx, SkTileMode tmy,
- const SkMatrix* localMatrix, SkCopyPixelsMode mode) {
- auto s = SkMakeBitmapShader(src, tmx, tmy, localMatrix, mode);
- if (!s) {
- return nullptr;
- }
- if (src.colorType() == kAlpha_8_SkColorType && paint.getShader()) {
- // Compose the image shader with the paint's shader. Alpha images+shaders should output the
- // texture's alpha multiplied by the shader's color. DstIn (d*sa) will achieve this with
- // the source image and dst shader (MakeBlend takes dst first, src second).
- s = SkShaders::Blend(SkBlendMode::kDstIn, paint.refShader(), std::move(s));
- }
- return s;
- }
- void SkShaderBase::RegisterFlattenables() { SK_REGISTER_FLATTENABLE(SkImageShader); }
- bool SkImageShader::onAppendStages(const SkStageRec& rec) const {
- SkRasterPipeline* p = rec.fPipeline;
- SkArenaAlloc* alloc = rec.fAlloc;
- SkMatrix matrix;
- if (!this->computeTotalInverse(rec.fCTM, rec.fLocalM, &matrix)) {
- return false;
- }
- auto quality = rec.fPaint.getFilterQuality();
- SkBitmapProvider provider(fImage.get());
- const auto* state = SkBitmapController::RequestBitmap(provider, matrix, quality, alloc);
- if (!state) {
- return false;
- }
- const SkPixmap& pm = state->pixmap();
- matrix = state->invMatrix();
- quality = state->quality();
- auto info = pm.info();
- // When the matrix is just an integer translate, bilerp == nearest neighbor.
- if (quality == kLow_SkFilterQuality &&
- matrix.getType() <= SkMatrix::kTranslate_Mask &&
- matrix.getTranslateX() == (int)matrix.getTranslateX() &&
- matrix.getTranslateY() == (int)matrix.getTranslateY()) {
- quality = kNone_SkFilterQuality;
- }
- // See skia:4649 and the GM image_scale_aligned.
- if (quality == kNone_SkFilterQuality) {
- if (matrix.getScaleX() >= 0) {
- matrix.setTranslateX(nextafterf(matrix.getTranslateX(),
- floorf(matrix.getTranslateX())));
- }
- if (matrix.getScaleY() >= 0) {
- matrix.setTranslateY(nextafterf(matrix.getTranslateY(),
- floorf(matrix.getTranslateY())));
- }
- }
- p->append(SkRasterPipeline::seed_shader);
- p->append_matrix(alloc, matrix);
- auto gather = alloc->make<SkRasterPipeline_GatherCtx>();
- gather->pixels = pm.addr();
- gather->stride = pm.rowBytesAsPixels();
- gather->width = pm.width();
- gather->height = pm.height();
- auto limit_x = alloc->make<SkRasterPipeline_TileCtx>(),
- limit_y = alloc->make<SkRasterPipeline_TileCtx>();
- limit_x->scale = pm.width();
- limit_x->invScale = 1.0f / pm.width();
- limit_y->scale = pm.height();
- limit_y->invScale = 1.0f / pm.height();
- SkRasterPipeline_DecalTileCtx* decal_ctx = nullptr;
- bool decal_x_and_y = fTileModeX == SkTileMode::kDecal && fTileModeY == SkTileMode::kDecal;
- if (fTileModeX == SkTileMode::kDecal || fTileModeY == SkTileMode::kDecal) {
- decal_ctx = alloc->make<SkRasterPipeline_DecalTileCtx>();
- decal_ctx->limit_x = limit_x->scale;
- decal_ctx->limit_y = limit_y->scale;
- }
- auto append_tiling_and_gather = [&] {
- if (decal_x_and_y) {
- p->append(SkRasterPipeline::decal_x_and_y, decal_ctx);
- } else {
- switch (fTileModeX) {
- case SkTileMode::kClamp: /* The gather_xxx stage will clamp for us. */ break;
- case SkTileMode::kMirror: p->append(SkRasterPipeline::mirror_x, limit_x); break;
- case SkTileMode::kRepeat: p->append(SkRasterPipeline::repeat_x, limit_x); break;
- case SkTileMode::kDecal: p->append(SkRasterPipeline::decal_x, decal_ctx); break;
- }
- switch (fTileModeY) {
- case SkTileMode::kClamp: /* The gather_xxx stage will clamp for us. */ break;
- case SkTileMode::kMirror: p->append(SkRasterPipeline::mirror_y, limit_y); break;
- case SkTileMode::kRepeat: p->append(SkRasterPipeline::repeat_y, limit_y); break;
- case SkTileMode::kDecal: p->append(SkRasterPipeline::decal_y, decal_ctx); break;
- }
- }
- void* ctx = gather;
- switch (info.colorType()) {
- case kAlpha_8_SkColorType: p->append(SkRasterPipeline::gather_a8, ctx); break;
- case kRGB_565_SkColorType: p->append(SkRasterPipeline::gather_565, ctx); break;
- case kARGB_4444_SkColorType: p->append(SkRasterPipeline::gather_4444, ctx); break;
- case kRGBA_8888_SkColorType: p->append(SkRasterPipeline::gather_8888, ctx); break;
- case kRGBA_1010102_SkColorType: p->append(SkRasterPipeline::gather_1010102, ctx); break;
- case kRGBA_F16Norm_SkColorType:
- case kRGBA_F16_SkColorType: p->append(SkRasterPipeline::gather_f16, ctx); break;
- case kRGBA_F32_SkColorType: p->append(SkRasterPipeline::gather_f32, ctx); break;
- case kGray_8_SkColorType: p->append(SkRasterPipeline::gather_a8, ctx);
- p->append(SkRasterPipeline::alpha_to_gray ); break;
- case kRGB_888x_SkColorType: p->append(SkRasterPipeline::gather_8888, ctx);
- p->append(SkRasterPipeline::force_opaque ); break;
- case kRGB_101010x_SkColorType: p->append(SkRasterPipeline::gather_1010102, ctx);
- p->append(SkRasterPipeline::force_opaque ); break;
- case kBGRA_8888_SkColorType: p->append(SkRasterPipeline::gather_8888, ctx);
- p->append(SkRasterPipeline::swap_rb ); break;
- case kUnknown_SkColorType: SkASSERT(false);
- }
- if (decal_ctx) {
- p->append(SkRasterPipeline::check_decal_mask, decal_ctx);
- }
- };
- auto append_misc = [&] {
- // TODO: if ref.fDstCS isn't null, we'll premul here then immediately unpremul
- // to do the color space transformation. Might be possible to streamline.
- if (info.colorType() == kAlpha_8_SkColorType) {
- // The color for A8 images comes from the (sRGB) paint color.
- p->append_set_rgb(alloc, rec.fPaint.getColor4f());
- p->append(SkRasterPipeline::premul);
- } else if (info.alphaType() == kUnpremul_SkAlphaType) {
- // Convert unpremul images to premul before we carry on with the rest of the pipeline.
- p->append(SkRasterPipeline::premul);
- }
- if (quality > kLow_SkFilterQuality) {
- // Bicubic filtering naturally produces out of range values on both sides.
- p->append(SkRasterPipeline::clamp_0);
- p->append(fClampAsIfUnpremul ? SkRasterPipeline::clamp_1
- : SkRasterPipeline::clamp_a);
- }
- if (rec.fDstCS) {
- // If color managed, convert from premul source all the way to premul dst color space.
- auto srcCS = info.colorSpace();
- if (!srcCS || info.colorType() == kAlpha_8_SkColorType) {
- // We treat untagged images as sRGB.
- // A8 images get their r,g,b from the paint color, so they're also sRGB.
- srcCS = sk_srgb_singleton();
- }
- alloc->make<SkColorSpaceXformSteps>(srcCS , kPremul_SkAlphaType,
- rec.fDstCS, kPremul_SkAlphaType)
- ->apply(p, info.colorType());
- }
- return true;
- };
- // We've got a fast path for 8888 bilinear clamp/clamp sampling.
- auto ct = info.colorType();
- if (true
- && (ct == kRGBA_8888_SkColorType || ct == kBGRA_8888_SkColorType)
- && quality == kLow_SkFilterQuality
- && fTileModeX == SkTileMode::kClamp && fTileModeY == SkTileMode::kClamp) {
- p->append(SkRasterPipeline::bilerp_clamp_8888, gather);
- if (ct == kBGRA_8888_SkColorType) {
- p->append(SkRasterPipeline::swap_rb);
- }
- return append_misc();
- }
- SkRasterPipeline_SamplerCtx* sampler = nullptr;
- if (quality != kNone_SkFilterQuality) {
- sampler = alloc->make<SkRasterPipeline_SamplerCtx>();
- }
- auto sample = [&](SkRasterPipeline::StockStage setup_x,
- SkRasterPipeline::StockStage setup_y) {
- p->append(setup_x, sampler);
- p->append(setup_y, sampler);
- append_tiling_and_gather();
- p->append(SkRasterPipeline::accumulate, sampler);
- };
- if (quality == kNone_SkFilterQuality) {
- append_tiling_and_gather();
- } else if (quality == kLow_SkFilterQuality) {
- p->append(SkRasterPipeline::save_xy, sampler);
- sample(SkRasterPipeline::bilinear_nx, SkRasterPipeline::bilinear_ny);
- sample(SkRasterPipeline::bilinear_px, SkRasterPipeline::bilinear_ny);
- sample(SkRasterPipeline::bilinear_nx, SkRasterPipeline::bilinear_py);
- sample(SkRasterPipeline::bilinear_px, SkRasterPipeline::bilinear_py);
- p->append(SkRasterPipeline::move_dst_src);
- } else {
- p->append(SkRasterPipeline::save_xy, sampler);
- sample(SkRasterPipeline::bicubic_n3x, SkRasterPipeline::bicubic_n3y);
- sample(SkRasterPipeline::bicubic_n1x, SkRasterPipeline::bicubic_n3y);
- sample(SkRasterPipeline::bicubic_p1x, SkRasterPipeline::bicubic_n3y);
- sample(SkRasterPipeline::bicubic_p3x, SkRasterPipeline::bicubic_n3y);
- sample(SkRasterPipeline::bicubic_n3x, SkRasterPipeline::bicubic_n1y);
- sample(SkRasterPipeline::bicubic_n1x, SkRasterPipeline::bicubic_n1y);
- sample(SkRasterPipeline::bicubic_p1x, SkRasterPipeline::bicubic_n1y);
- sample(SkRasterPipeline::bicubic_p3x, SkRasterPipeline::bicubic_n1y);
- sample(SkRasterPipeline::bicubic_n3x, SkRasterPipeline::bicubic_p1y);
- sample(SkRasterPipeline::bicubic_n1x, SkRasterPipeline::bicubic_p1y);
- sample(SkRasterPipeline::bicubic_p1x, SkRasterPipeline::bicubic_p1y);
- sample(SkRasterPipeline::bicubic_p3x, SkRasterPipeline::bicubic_p1y);
- sample(SkRasterPipeline::bicubic_n3x, SkRasterPipeline::bicubic_p3y);
- sample(SkRasterPipeline::bicubic_n1x, SkRasterPipeline::bicubic_p3y);
- sample(SkRasterPipeline::bicubic_p1x, SkRasterPipeline::bicubic_p3y);
- sample(SkRasterPipeline::bicubic_p3x, SkRasterPipeline::bicubic_p3y);
- p->append(SkRasterPipeline::move_dst_src);
- }
- return append_misc();
- }
|