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 | namespace { |
| 16 | template <typename K, typename V> struct UnrefValue { |
| 17 | void operator()(K, V** v) { SkSafeUnref(*v); } |
| 18 | }; |
| 19 | } |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 20 | |
halcanary | cee1342 | 2016-08-18 09:52:48 -0700 | [diff] [blame] | 21 | SkPDFCanon::~SkPDFCanon() { |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 22 | // TODO(halcanary): make SkTHashSet work nicely with sk_sp<>, |
| 23 | // or use std::unordered_set<> |
halcanary | be27a11 | 2015-04-01 13:31:19 -0700 | [diff] [blame] | 24 | fGraphicStateRecords.foreach ([](WrapGS w) { w.fPtr->unref(); }); |
halcanary | cee1342 | 2016-08-18 09:52:48 -0700 | [diff] [blame] | 25 | fPDFBitmapMap.foreach(UnrefValue<SkBitmapKey, SkPDFObject>()); |
| 26 | fTypefaceMetrics.foreach(UnrefValue<uint32_t, SkAdvancedTypefaceMetrics>()); |
| 27 | fFontDescriptors.foreach(UnrefValue<uint32_t, SkPDFDict>()); |
| 28 | fFontMap.foreach(UnrefValue<uint64_t, SkPDFFont>()); |
halcanary | 2e3f9d8 | 2015-02-27 12:41:03 -0800 | [diff] [blame] | 29 | } |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 30 | |
halcanary | cee1342 | 2016-08-18 09:52:48 -0700 | [diff] [blame] | 31 | void SkPDFCanon::reset() { |
| 32 | this->~SkPDFCanon(); |
| 33 | new (this)SkPDFCanon; |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 34 | } |
| 35 | |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 36 | //////////////////////////////////////////////////////////////////////////////// |
| 37 | |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 38 | template <typename T> |
| 39 | sk_sp<SkPDFObject> find_shader(const SkTArray<T>& records, |
| 40 | const SkPDFShader::State& state) { |
| 41 | for (const T& record : records) { |
| 42 | if (record.fShaderState == state) { |
| 43 | return record.fShaderObject; |
| 44 | } |
| 45 | } |
| 46 | return nullptr; |
| 47 | } |
| 48 | |
| 49 | sk_sp<SkPDFObject> SkPDFCanon::findFunctionShader( |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 50 | const SkPDFShader::State& state) const { |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 51 | return find_shader(fFunctionShaderRecords, state); |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 52 | } |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 53 | void SkPDFCanon::addFunctionShader(sk_sp<SkPDFObject> pdfShader, |
| 54 | SkPDFShader::State state) { |
| 55 | fFunctionShaderRecords.emplace_back(std::move(state), std::move(pdfShader)); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 56 | } |
| 57 | |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 58 | sk_sp<SkPDFObject> SkPDFCanon::findAlphaShader( |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 59 | const SkPDFShader::State& state) const { |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 60 | return find_shader(fAlphaShaderRecords, state); |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 61 | } |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 62 | void SkPDFCanon::addAlphaShader(sk_sp<SkPDFObject> pdfShader, |
| 63 | SkPDFShader::State state) { |
| 64 | fAlphaShaderRecords.emplace_back(std::move(state), std::move(pdfShader)); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 65 | } |
| 66 | |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 67 | sk_sp<SkPDFObject> SkPDFCanon::findImageShader( |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 68 | const SkPDFShader::State& state) const { |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 69 | return find_shader(fImageShaderRecords, state); |
halcanary | 530ea8e | 2015-01-23 06:17:35 -0800 | [diff] [blame] | 70 | } |
| 71 | |
halcanary | dabd4f0 | 2016-08-03 11:16:56 -0700 | [diff] [blame] | 72 | void SkPDFCanon::addImageShader(sk_sp<SkPDFObject> pdfShader, |
| 73 | SkPDFShader::State state) { |
| 74 | fImageShaderRecords.emplace_back(std::move(state), std::move(pdfShader)); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 75 | } |
| 76 | |
| 77 | //////////////////////////////////////////////////////////////////////////////// |
| 78 | |
halcanary | be27a11 | 2015-04-01 13:31:19 -0700 | [diff] [blame] | 79 | const SkPDFGraphicState* SkPDFCanon::findGraphicState( |
| 80 | const SkPDFGraphicState& key) const { |
| 81 | const WrapGS* ptr = fGraphicStateRecords.find(WrapGS(&key)); |
halcanary | 96fcdcc | 2015-08-27 07:41:13 -0700 | [diff] [blame] | 82 | return ptr ? ptr->fPtr : nullptr; |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 83 | } |
| 84 | |
halcanary | be27a11 | 2015-04-01 13:31:19 -0700 | [diff] [blame] | 85 | void SkPDFCanon::addGraphicState(const SkPDFGraphicState* state) { |
| 86 | SkASSERT(state); |
| 87 | WrapGS w(SkRef(state)); |
| 88 | SkASSERT(!fGraphicStateRecords.contains(w)); |
| 89 | fGraphicStateRecords.add(w); |
halcanary | fb62b3d | 2015-01-21 09:59:14 -0800 | [diff] [blame] | 90 | } |
halcanary | 1b5c604 | 2015-02-18 11:29:56 -0800 | [diff] [blame] | 91 | |
| 92 | //////////////////////////////////////////////////////////////////////////////// |
| 93 | |
halcanary | a50151d | 2016-03-25 11:57:49 -0700 | [diff] [blame] | 94 | sk_sp<SkPDFObject> SkPDFCanon::findPDFBitmap(SkBitmapKey key) const { |
| 95 | SkPDFObject** ptr = fPDFBitmapMap.find(key); |
| 96 | return ptr ? sk_ref_sp(*ptr) : sk_sp<SkPDFObject>(); |
halcanary | 1b5c604 | 2015-02-18 11:29:56 -0800 | [diff] [blame] | 97 | } |
| 98 | |
halcanary | a50151d | 2016-03-25 11:57:49 -0700 | [diff] [blame] | 99 | void SkPDFCanon::addPDFBitmap(SkBitmapKey key, sk_sp<SkPDFObject> pdfBitmap) { |
| 100 | fPDFBitmapMap.set(key, pdfBitmap.release()); |
halcanary | 7a14b31 | 2015-10-01 07:28:13 -0700 | [diff] [blame] | 101 | } |
| 102 | |
halcanary | a50151d | 2016-03-25 11:57:49 -0700 | [diff] [blame] | 103 | //////////////////////////////////////////////////////////////////////////////// |
halcanary | 1437c1e | 2016-03-13 18:30:24 -0700 | [diff] [blame] | 104 | |
| 105 | sk_sp<SkPDFStream> SkPDFCanon::makeInvertFunction() { |
| 106 | if (fInvertFunction) { |
| 107 | return fInvertFunction; |
| 108 | } |
| 109 | fInvertFunction = SkPDFGraphicState::MakeInvertFunction(); |
| 110 | return fInvertFunction; |
| 111 | } |
| 112 | sk_sp<SkPDFDict> SkPDFCanon::makeNoSmaskGraphicState() { |
| 113 | if (fNoSmaskGraphicState) { |
| 114 | return fNoSmaskGraphicState; |
| 115 | } |
| 116 | fNoSmaskGraphicState = SkPDFGraphicState::MakeNoSmaskGraphicState(); |
| 117 | return fNoSmaskGraphicState; |
| 118 | } |
| 119 | sk_sp<SkPDFArray> SkPDFCanon::makeRangeObject() { |
| 120 | if (fRangeObject) { |
| 121 | return fRangeObject; |
| 122 | } |
| 123 | fRangeObject = SkPDFShader::MakeRangeObject(); |
| 124 | return fRangeObject; |
| 125 | } |