| /* |
| * Copyright 2016 Google Inc. |
| * |
| * Use of this source code is governed by a BSD-style license that can be |
| * found in the LICENSE file. |
| */ |
| |
| #ifndef GrColorSpaceXform_DEFINED |
| #define GrColorSpaceXform_DEFINED |
| |
| #include "GrColor.h" |
| #include "GrFragmentProcessor.h" |
| #include "SkColorSpace.h" |
| #include "SkMatrix44.h" |
| #include "SkRefCnt.h" |
| |
| /** |
| * Represents a color space transformation |
| */ |
| class GrColorSpaceXform : public SkRefCnt { |
| public: |
| GrColorSpaceXform(const SkColorSpaceTransferFn&, const SkMatrix44&, uint32_t); |
| |
| static sk_sp<GrColorSpaceXform> Make(SkColorSpace* src, GrPixelConfig srcConfig, |
| SkColorSpace* dst); |
| static sk_sp<GrColorSpaceXform> MakeGamutXform(SkColorSpace* src, SkColorSpace* dst) { |
| auto result = Make(src, kUnknown_GrPixelConfig, dst); |
| SkASSERT(!result || 0 == (result->fFlags & ~kApplyGamutXform_Flag)); |
| return result; |
| } |
| |
| const SkColorSpaceTransferFn& transferFn() const { return fSrcTransferFn; } |
| const float* transferFnCoeffs() const { |
| static_assert(0 == offsetof(SkColorSpaceTransferFn, fG), "TransferFn layout"); |
| return &fSrcTransferFn.fG; |
| } |
| |
| const SkMatrix44& gamutXform() const { return fGamutXform; } |
| |
| /** |
| * GrGLSLFragmentProcessor::GenKey() must call this and include the returned value in its |
| * computed key. |
| */ |
| static uint32_t XformKey(const GrColorSpaceXform* xform) { |
| // Code generation depends on which steps we apply (as encoded by fFlags) |
| return SkToBool(xform) ? xform->fFlags : 0; |
| } |
| |
| static bool Equals(const GrColorSpaceXform* a, const GrColorSpaceXform* b); |
| |
| GrColor4f unclampedXform(const GrColor4f& srcColor); |
| GrColor4f clampedXform(const GrColor4f& srcColor); |
| |
| private: |
| friend class GrGLSLColorSpaceXformHelper; |
| |
| enum Flags { |
| kApplyTransferFn_Flag = 0x1, |
| kApplyGamutXform_Flag = 0x2, |
| |
| // Almost never used. This handles the case where the src data is sRGB pixel config, |
| // but the color space has a different transfer function. In that case, we first undo |
| // the HW sRGB -> Linear conversion, before applying any other steps. |
| kApplyInverseSRGB_Flag = 0x4, |
| }; |
| |
| SkColorSpaceTransferFn fSrcTransferFn; |
| SkMatrix44 fGamutXform; |
| uint32_t fFlags; |
| }; |
| |
| class GrColorSpaceXformEffect : public GrFragmentProcessor { |
| public: |
| /** |
| * Returns a fragment processor that calls the passed in fragment processor, and then converts |
| * the color space of the output from src to dst. |
| */ |
| static std::unique_ptr<GrFragmentProcessor> Make(std::unique_ptr<GrFragmentProcessor> child, |
| SkColorSpace* src, GrPixelConfig srcConfig, |
| SkColorSpace* dst); |
| |
| const char* name() const override { return "ColorSpaceXform"; } |
| std::unique_ptr<GrFragmentProcessor> clone() const override; |
| |
| const GrColorSpaceXform* colorXform() const { return fColorXform.get(); } |
| |
| private: |
| GrColorSpaceXformEffect(std::unique_ptr<GrFragmentProcessor> child, |
| sk_sp<GrColorSpaceXform> colorXform); |
| |
| static OptimizationFlags OptFlags(const GrFragmentProcessor* child); |
| |
| GrGLSLFragmentProcessor* onCreateGLSLInstance() const override; |
| void onGetGLSLProcessorKey(const GrShaderCaps&, GrProcessorKeyBuilder*) const override; |
| bool onIsEqual(const GrFragmentProcessor&) const override; |
| |
| sk_sp<GrColorSpaceXform> fColorXform; |
| |
| typedef GrFragmentProcessor INHERITED; |
| }; |
| |
| #endif |