halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 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 | |
halcanary | 7a14b31 | 2015-10-01 07:28:13 -0700 | [diff] [blame] | 8 | #include "SkImage.h" |
halcanary | 1b5c604 | 2015-02-18 11:29:56 -0800 | [diff] [blame] | 9 | #include "SkPDFBitmap.h" |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 10 | #include "SkPDFCanon.h" |
| 11 | #include "SkPDFFont.h" |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 12 | |
| 13 | //////////////////////////////////////////////////////////////////////////////// |
| 14 | |
halcanary | cee1342 | 2016-08-18 09:52:48 -0700 | [diff] [blame] | 15 | SkPDFCanon::~SkPDFCanon() { |
halcanary | be27a11 | 2015-04-01 13:31:19 -0700 | [diff] [blame] | 16 | fGraphicStateRecords.foreach ([](WrapGS w) { w.fPtr->unref(); }); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 17 | } |
| 18 | |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 19 | //////////////////////////////////////////////////////////////////////////////// |
| 20 | |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 21 | template <typename T> |
| 22 | sk_sp<SkPDFObject> find_shader(const SkTArray<T>& records, |
| 23 | const SkPDFShader::State& state) { |
| 24 | for (const T& record : records) { |
| 25 | if (record.fShaderState == state) { |
| 26 | return record.fShaderObject; |
| 27 | } |
| 28 | } |
| 29 | return nullptr; |
| 30 | } |
| 31 | |
| 32 | sk_sp<SkPDFObject> SkPDFCanon::findFunctionShader( |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 33 | const SkPDFShader::State& state) const { |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 34 | return find_shader(fFunctionShaderRecords, state); |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 35 | } |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 36 | void SkPDFCanon::addFunctionShader(sk_sp<SkPDFObject> pdfShader, |
| 37 | SkPDFShader::State state) { |
Hal Canary | 5c1b360 | 2017-04-17 16:30:06 -0400 | [diff] [blame] | 38 | fFunctionShaderRecords.emplace_back(ShaderRec{std::move(state), std::move(pdfShader)}); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 39 | } |
| 40 | |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 41 | sk_sp<SkPDFObject> SkPDFCanon::findAlphaShader( |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 42 | const SkPDFShader::State& state) const { |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 43 | return find_shader(fAlphaShaderRecords, state); |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 44 | } |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 45 | void SkPDFCanon::addAlphaShader(sk_sp<SkPDFObject> pdfShader, |
| 46 | SkPDFShader::State state) { |
Hal Canary | 5c1b360 | 2017-04-17 16:30:06 -0400 | [diff] [blame] | 47 | fAlphaShaderRecords.emplace_back(ShaderRec{std::move(state), std::move(pdfShader)}); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 48 | } |
| 49 | |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 50 | sk_sp<SkPDFObject> SkPDFCanon::findImageShader( |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 51 | const SkPDFShader::State& state) const { |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 52 | return find_shader(fImageShaderRecords, state); |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 53 | } |
| 54 | |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 55 | void SkPDFCanon::addImageShader(sk_sp<SkPDFObject> pdfShader, |
| 56 | SkPDFShader::State state) { |
Hal Canary | 5c1b360 | 2017-04-17 16:30:06 -0400 | [diff] [blame] | 57 | fImageShaderRecords.emplace_back(ShaderRec{std::move(state), std::move(pdfShader)}); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 58 | } |
| 59 | |
| 60 | //////////////////////////////////////////////////////////////////////////////// |
| 61 | |
halcanary | be27a11 | 2015-04-01 13:31:19 -0700 | [diff] [blame] | 62 | const SkPDFGraphicState* SkPDFCanon::findGraphicState( |
| 63 | const SkPDFGraphicState& key) const { |
| 64 | const WrapGS* ptr = fGraphicStateRecords.find(WrapGS(&key)); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 65 | return ptr ? ptr->fPtr : nullptr; |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 66 | } |
| 67 | |
halcanary | be27a11 | 2015-04-01 13:31:19 -0700 | [diff] [blame] | 68 | void SkPDFCanon::addGraphicState(const SkPDFGraphicState* state) { |
| 69 | SkASSERT(state); |
| 70 | WrapGS w(SkRef(state)); |
| 71 | SkASSERT(!fGraphicStateRecords.contains(w)); |
| 72 | fGraphicStateRecords.add(w); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 73 | } |
halcanary | 1b5c604 | 2015-02-18 11:29:56 -0800 | [diff] [blame] | 74 | |