| /* |
| * Copyright 2017 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #include "SkColorFilter.h" |
| #include "SkColorSpaceXformer.h" |
| #include "SkColorSpaceXform_Base.h" |
| #include "SkDrawLooper.h" |
| #include "SkGradientShader.h" |
| #include "SkImage_Base.h" |
| #include "SkImagePriv.h" |
| #include "SkMakeUnique.h" |
| |
| std::unique_ptr<SkColorSpaceXformer> SkColorSpaceXformer::Make(sk_sp<SkColorSpace> dst) { |
| std::unique_ptr<SkColorSpaceXform> fromSRGB = SkColorSpaceXform_Base::New( |
| SkColorSpace::MakeSRGB().get(), dst.get(), SkTransferFunctionBehavior::kIgnore); |
| if (!fromSRGB) { |
| return nullptr; |
| } |
| |
| auto xformer = std::unique_ptr<SkColorSpaceXformer>(new SkColorSpaceXformer()); |
| xformer->fDst = std::move(dst); |
| xformer->fFromSRGB = std::move(fromSRGB); |
| return xformer; |
| } |
| |
| sk_sp<SkImage> SkColorSpaceXformer::apply(const SkImage* src) { |
| return as_IB(src)->makeColorSpace(fDst); |
| } |
| |
| sk_sp<SkImage> SkColorSpaceXformer::apply(const SkBitmap& src) { |
| sk_sp<SkImage> image = SkMakeImageFromRasterBitmap(src, kNever_SkCopyPixelsMode); |
| if (!image) { |
| return nullptr; |
| } |
| |
| sk_sp<SkImage> xformed = as_IB(image)->makeColorSpace(fDst); |
| // We want to be sure we don't let the kNever_SkCopyPixelsMode image escape this stack frame. |
| SkASSERT(xformed != image); |
| return xformed; |
| } |
| |
| void SkColorSpaceXformer::apply(SkColor* xformed, const SkColor* srgb, int n) { |
| SkAssertResult(fFromSRGB->apply(SkColorSpaceXform::kBGRA_8888_ColorFormat, xformed, |
| SkColorSpaceXform::kBGRA_8888_ColorFormat, srgb, |
| n, kUnpremul_SkAlphaType)); |
| } |
| |
| SkColor SkColorSpaceXformer::apply(SkColor srgb) { |
| SkColor xformed; |
| this->apply(&xformed, &srgb, 1); |
| return xformed; |
| } |
| |
| // TODO: Is this introspection going to be enough, or do we need a new SkShader method? |
| sk_sp<SkShader> SkColorSpaceXformer::apply(const SkShader* shader) { |
| SkColor color; |
| if (shader->isConstant() && shader->asLuminanceColor(&color)) { |
| return SkShader::MakeColorShader(this->apply(color)) |
| ->makeWithLocalMatrix(shader->getLocalMatrix()); |
| } |
| |
| SkShader::TileMode xy[2]; |
| SkMatrix local; |
| if (auto img = shader->isAImage(&local, xy)) { |
| return this->apply(img)->makeShader(xy[0], xy[1], &local); |
| } |
| |
| SkShader::ComposeRec compose; |
| if (shader->asACompose(&compose)) { |
| auto A = this->apply(compose.fShaderA), |
| B = this->apply(compose.fShaderB); |
| if (A && B) { |
| return SkShader::MakeComposeShader(std::move(A), std::move(B), compose.fBlendMode) |
| ->makeWithLocalMatrix(shader->getLocalMatrix()); |
| } |
| } |
| |
| SkShader::GradientInfo gradient; |
| sk_bzero(&gradient, sizeof(gradient)); |
| if (auto type = shader->asAGradient(&gradient)) { |
| SkSTArray<8, SkColor> colors(gradient.fColorCount); |
| SkSTArray<8, SkScalar> pos(gradient.fColorCount); |
| |
| gradient.fColors = colors.begin(); |
| gradient.fColorOffsets = pos.begin(); |
| shader->asAGradient(&gradient); |
| |
| SkSTArray<8, SkColor> xformed(gradient.fColorCount); |
| this->apply(xformed.begin(), gradient.fColors, gradient.fColorCount); |
| |
| switch (type) { |
| case SkShader::kNone_GradientType: |
| case SkShader::kColor_GradientType: |
| SkASSERT(false); // Should be unreachable. |
| break; |
| |
| case SkShader::kLinear_GradientType: |
| return SkGradientShader::MakeLinear(gradient.fPoint, |
| xformed.begin(), |
| gradient.fColorOffsets, |
| gradient.fColorCount, |
| gradient.fTileMode, |
| gradient.fGradientFlags, |
| &shader->getLocalMatrix()); |
| case SkShader::kRadial_GradientType: |
| return SkGradientShader::MakeRadial(gradient.fPoint[0], |
| gradient.fRadius[0], |
| xformed.begin(), |
| gradient.fColorOffsets, |
| gradient.fColorCount, |
| gradient.fTileMode, |
| gradient.fGradientFlags, |
| &shader->getLocalMatrix()); |
| case SkShader::kSweep_GradientType: |
| return SkGradientShader::MakeSweep(gradient.fPoint[0].fX, |
| gradient.fPoint[0].fY, |
| xformed.begin(), |
| gradient.fColorOffsets, |
| gradient.fColorCount, |
| gradient.fGradientFlags, |
| &shader->getLocalMatrix()); |
| case SkShader::kConical_GradientType: |
| return SkGradientShader::MakeTwoPointConical(gradient.fPoint[0], |
| gradient.fRadius[0], |
| gradient.fPoint[1], |
| gradient.fRadius[1], |
| xformed.begin(), |
| gradient.fColorOffsets, |
| gradient.fColorCount, |
| gradient.fTileMode, |
| gradient.fGradientFlags, |
| &shader->getLocalMatrix()); |
| } |
| } |
| |
| return sk_ref_sp(const_cast<SkShader*>(shader)); |
| } |
| |
| SkPaint SkColorSpaceXformer::apply(const SkPaint& src) { |
| SkPaint dst = src; |
| |
| // All SkColorSpaces have the same black point. |
| if (src.getColor() & 0xffffff) { |
| dst.setColor(this->apply(src.getColor())); |
| } |
| |
| if (auto shader = src.getShader()) { |
| dst.setShader(this->apply(shader)); |
| } |
| |
| // As far as I know, SkModeColorFilter is the only color filter that holds a color. |
| if (auto cf = src.getColorFilter()) { |
| SkColor color; |
| SkBlendMode mode; |
| if (cf->asColorMode(&color, &mode)) { |
| dst.setColorFilter(SkColorFilter::MakeModeFilter(this->apply(color), mode)); |
| } |
| } |
| |
| if (auto looper = src.getDrawLooper()) { |
| dst.setDrawLooper(looper->makeColorSpace(this)); |
| } |
| |
| // TODO: |
| // - image filters? |
| return dst; |
| } |