reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2013 Google Inc. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license that can be |
| 5 | * found in the LICENSE file. |
| 6 | */ |
| 7 | |
Herb Derby | 83e939b | 2017-02-07 14:25:11 -0500 | [diff] [blame] | 8 | #include "SkArenaAlloc.h" |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 9 | #include "SkColorFilterShader.h" |
Matt Sarett | 72266ee | 2017-04-20 10:09:23 -0400 | [diff] [blame] | 10 | #include "SkColorSpaceXformer.h" |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 11 | #include "SkReadBuffer.h" |
| 12 | #include "SkWriteBuffer.h" |
| 13 | #include "SkShader.h" |
| 14 | #include "SkString.h" |
| 15 | |
| 16 | #if SK_SUPPORT_GPU |
| 17 | #include "GrFragmentProcessor.h" |
| 18 | #endif |
| 19 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 20 | SkColorFilterShader::SkColorFilterShader(sk_sp<SkShader> shader, sk_sp<SkColorFilter> filter) |
| 21 | : fShader(std::move(shader)) |
| 22 | , fFilter(std::move(filter)) |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 23 | { |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 24 | SkASSERT(fShader); |
| 25 | SkASSERT(fFilter); |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 26 | } |
| 27 | |
reed | 60c9b58 | 2016-04-03 09:11:13 -0700 | [diff] [blame] | 28 | sk_sp<SkFlattenable> SkColorFilterShader::CreateProc(SkReadBuffer& buffer) { |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 29 | auto shader = buffer.readShader(); |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 30 | auto filter = buffer.readColorFilter(); |
| 31 | if (!shader || !filter) { |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 32 | return nullptr; |
| 33 | } |
reed | 60c9b58 | 2016-04-03 09:11:13 -0700 | [diff] [blame] | 34 | return sk_make_sp<SkColorFilterShader>(shader, filter); |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 35 | } |
| 36 | |
| 37 | void SkColorFilterShader::flatten(SkWriteBuffer& buffer) const { |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 38 | buffer.writeFlattenable(fShader.get()); |
| 39 | buffer.writeFlattenable(fFilter.get()); |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 40 | } |
| 41 | |
Mike Reed | 1d8c42e | 2017-08-29 14:58:19 -0400 | [diff] [blame] | 42 | bool SkColorFilterShader::onAppendStages(const StageRec& rec) const { |
| 43 | if (!as_SB(fShader)->appendStages(rec)) { |
Mike Reed | 2aab802 | 2017-05-31 12:14:56 -0400 | [diff] [blame] | 44 | return false; |
| 45 | } |
Mike Reed | 1d8c42e | 2017-08-29 14:58:19 -0400 | [diff] [blame] | 46 | fFilter->appendStages(rec.fPipeline, rec.fDstCS, rec.fAlloc, fShader->isOpaque()); |
Mike Reed | 2aab802 | 2017-05-31 12:14:56 -0400 | [diff] [blame] | 47 | return true; |
| 48 | } |
| 49 | |
Matt Sarett | 72266ee | 2017-04-20 10:09:23 -0400 | [diff] [blame] | 50 | sk_sp<SkShader> SkColorFilterShader::onMakeColorSpace(SkColorSpaceXformer* xformer) const { |
Mike Klein | 2814d91 | 2017-05-10 12:35:51 -0400 | [diff] [blame] | 51 | return xformer->apply(fShader.get())->makeWithColorFilter(xformer->apply(fFilter.get())); |
Matt Sarett | 72266ee | 2017-04-20 10:09:23 -0400 | [diff] [blame] | 52 | } |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 53 | |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 54 | #if SK_SUPPORT_GPU |
| 55 | ///////////////////////////////////////////////////////////////////// |
| 56 | |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 57 | std::unique_ptr<GrFragmentProcessor> SkColorFilterShader::asFragmentProcessor( |
Mike Reed | e3429e6 | 2018-01-19 11:43:34 -0500 | [diff] [blame] | 58 | const GrFPArgs& args) const { |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 59 | auto fp1 = as_SB(fShader)->asFragmentProcessor(args); |
bungeman | 06ca8ec | 2016-06-09 08:01:03 -0700 | [diff] [blame] | 60 | if (!fp1) { |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 61 | return nullptr; |
| 62 | } |
| 63 | |
Brian Salomon | 4cbb6e6 | 2017-10-25 15:12:19 -0400 | [diff] [blame] | 64 | auto fp2 = fFilter->asFragmentProcessor(args.fContext, *args.fDstColorSpaceInfo); |
bungeman | 06ca8ec | 2016-06-09 08:01:03 -0700 | [diff] [blame] | 65 | if (!fp2) { |
| 66 | return fp1; |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 67 | } |
| 68 | |
Brian Salomon | aff329b | 2017-08-11 09:40:37 -0400 | [diff] [blame] | 69 | std::unique_ptr<GrFragmentProcessor> fpSeries[] = { std::move(fp1), std::move(fp2) }; |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 70 | return GrFragmentProcessor::RunInSeries(fpSeries, 2); |
| 71 | } |
| 72 | #endif |
| 73 | |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 74 | /////////////////////////////////////////////////////////////////////////////////////////////////// |
| 75 | |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 76 | sk_sp<SkShader> SkShader::makeWithColorFilter(sk_sp<SkColorFilter> filter) const { |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 77 | SkShader* base = const_cast<SkShader*>(this); |
| 78 | if (!filter) { |
reed | 150835e | 2016-03-10 06:36:49 -0800 | [diff] [blame] | 79 | return sk_ref_sp(base); |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 80 | } |
reed | d053ce9 | 2016-03-22 10:17:23 -0700 | [diff] [blame] | 81 | return sk_make_sp<SkColorFilterShader>(sk_ref_sp(base), filter); |
reed | 3061af4 | 2016-01-07 15:47:29 -0800 | [diff] [blame] | 82 | } |