110b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman/* 210b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman * Copyright 2017 Google Inc. 310b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman * 410b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman * Use of this source code is governed by a BSD-style license that can be 510b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman * found in the LICENSE file. 610b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman */ 710b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 892db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein#include "SkPM4fPriv.h" 910b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman#include "SkRasterPipeline.h" 10205c69d4895ac115c2e456707169321b257dc6aeMike Reed#include "SkReadBuffer.h" 1110b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman#include "SkString.h" 1292db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein#include "SkToSRGBColorFilter.h" 13205c69d4895ac115c2e456707169321b257dc6aeMike Reed#include "SkWriteBuffer.h" 1410b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 1510b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman#if SK_SUPPORT_GPU 1692db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein #include "effects/GrNonlinearColorSpaceXformEffect.h" 1710b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman#endif 1810b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 1910b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osmanvoid SkToSRGBColorFilter::onAppendStages(SkRasterPipeline* p, 2092db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein SkColorSpace* /*dst color space*/, 2192db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein SkArenaAlloc* alloc, 2210b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman bool shaderIsOpaque) const { 2392db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein // Step 1: Linearize by undoing the src transfer function. 2492db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein // Linear and sRGB will return true to isNumericalTransferFn(), so we check them first. 2592db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein SkColorSpaceTransferFn srcFn; 2692db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein if (fSrcColorSpace->gammaIsLinear()) { 2792db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein // Nothing to do. 2892db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein } else if (fSrcColorSpace->gammaCloseToSRGB()) { 29f1f1162273b382db99f8609e5bbfff24f5594821Mike Klein p->append(SkRasterPipeline::from_srgb); 3092db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein } else if (fSrcColorSpace->isNumericalTransferFn(&srcFn)) { 31dbd43481f1da3c9c87f5d913660f216ec531870aMike Klein auto copy = alloc->make<SkColorSpaceTransferFn>(srcFn); 32dbd43481f1da3c9c87f5d913660f216ec531870aMike Klein p->append(SkRasterPipeline::parametric_r, copy); 33dbd43481f1da3c9c87f5d913660f216ec531870aMike Klein p->append(SkRasterPipeline::parametric_g, copy); 34dbd43481f1da3c9c87f5d913660f216ec531870aMike Klein p->append(SkRasterPipeline::parametric_b, copy); 3592db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein } else { 3692db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein SkDEBUGFAIL("Looks like we got a table transfer function here, quite unexpectedly."); 3792db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein // TODO: If we really need to handle this, we can, but I don't think Ganesh does. 3892db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein } 3992db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein 4092db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein // Step 2: Transform to sRGB gamut, without clamping. 4137155d476cd727329d985b75ecedbefe380a0f23Mike Klein // TODO: because... 42159db0a6a1001b220b42fcae46ed324e0986b14aMike Klein float* gamut_transform = alloc->makeArrayDefault<float>(12); 4337155d476cd727329d985b75ecedbefe380a0f23Mike Klein (void)append_gamut_transform_noclamp(p, 4437155d476cd727329d985b75ecedbefe380a0f23Mike Klein gamut_transform, 4537155d476cd727329d985b75ecedbefe380a0f23Mike Klein fSrcColorSpace.get(), 4637155d476cd727329d985b75ecedbefe380a0f23Mike Klein SkColorSpace::MakeSRGB().get()); 47159db0a6a1001b220b42fcae46ed324e0986b14aMike Klein 4892db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein // Step 3: Back to sRGB encoding. 4992db5a39d4f810dbef96b2b966e4615f7af0ce51Mike Klein p->append(SkRasterPipeline::to_srgb); 5010b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman} 5110b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 5210b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osmansk_sp<SkColorFilter> SkToSRGBColorFilter::Make(sk_sp<SkColorSpace> srcColorSpace) { 53c9bc81434adc0c0ea6012167fbda0e131548e683Mike Klein if (!srcColorSpace || srcColorSpace->isSRGB()) { 54fa78ece52e8e30f522eea77df7ab67edcd17bddcMike Reed return nullptr; 55fa78ece52e8e30f522eea77df7ab67edcd17bddcMike Reed } else { 56fa78ece52e8e30f522eea77df7ab67edcd17bddcMike Reed return sk_sp<SkColorFilter>(new SkToSRGBColorFilter(std::move(srcColorSpace))); 57fa78ece52e8e30f522eea77df7ab67edcd17bddcMike Reed } 5810b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman} 5910b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 6010b6ad13f996d2f522bc057d17acea58e43a7f0bBrian OsmanSkToSRGBColorFilter::SkToSRGBColorFilter(sk_sp<SkColorSpace> srcColorSpace) 61c9bc81434adc0c0ea6012167fbda0e131548e683Mike Klein : fSrcColorSpace(std::move(srcColorSpace)) { 62c9bc81434adc0c0ea6012167fbda0e131548e683Mike Klein SkASSERT(fSrcColorSpace); 63c9bc81434adc0c0ea6012167fbda0e131548e683Mike Klein} 6410b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 65205c69d4895ac115c2e456707169321b257dc6aeMike Reedsk_sp<SkFlattenable> SkToSRGBColorFilter::CreateProc(SkReadBuffer& buffer) { 66205c69d4895ac115c2e456707169321b257dc6aeMike Reed auto data = buffer.readByteArrayAsData(); 67205c69d4895ac115c2e456707169321b257dc6aeMike Reed if (data) { 68205c69d4895ac115c2e456707169321b257dc6aeMike Reed return Make(SkColorSpace::Deserialize(data->data(), data->size())); 69205c69d4895ac115c2e456707169321b257dc6aeMike Reed } 7010b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman return nullptr; 7110b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman} 7210b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 73205c69d4895ac115c2e456707169321b257dc6aeMike Reedvoid SkToSRGBColorFilter::flatten(SkWriteBuffer& buffer) const { 74205c69d4895ac115c2e456707169321b257dc6aeMike Reed buffer.writeDataAsByteArray(fSrcColorSpace->serialize().get()); 7510b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman} 7610b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 7710b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman#ifndef SK_IGNORE_TO_STRING 7810b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osmanvoid SkToSRGBColorFilter::toString(SkString* str) const { 7910b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman // TODO 8010b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman str->append("SkToSRGBColorFilter "); 8110b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman} 8210b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman#endif 8310b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman 8410b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman#if SK_SUPPORT_GPU 854cbb6e6d551e8bea2c040b1aa9dce253cffb9af0Brian Salomonstd::unique_ptr<GrFragmentProcessor> SkToSRGBColorFilter::asFragmentProcessor( 864cbb6e6d551e8bea2c040b1aa9dce253cffb9af0Brian Salomon GrContext*, const GrColorSpaceInfo&) const { 8710b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman return GrNonlinearColorSpaceXformEffect::Make(fSrcColorSpace.get(), 8810b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman SkColorSpace::MakeSRGB().get()); 8910b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman} 9010b6ad13f996d2f522bc057d17acea58e43a7f0bBrian Osman#endif 91