|
1 | 1 | #pragma once |
2 | 2 |
|
3 | 3 | #include <memory> |
| 4 | +#include <unordered_map> |
4 | 5 | #include <utility> |
5 | 6 |
|
6 | 7 | #include <jsi/jsi.h> |
|
14 | 15 | #pragma clang diagnostic ignored "-Wdocumentation" |
15 | 16 |
|
16 | 17 | #include "include/core/SkStream.h" |
| 18 | +#include "modules/skresources/include/SkResources.h" |
17 | 19 |
|
18 | 20 | #pragma clang diagnostic pop |
19 | 21 |
|
20 | 22 | namespace RNSkia { |
21 | 23 |
|
22 | 24 | namespace jsi = facebook::jsi; |
23 | 25 |
|
| 26 | +class SVGAssetProvider : public skresources::ResourceProvider { |
| 27 | +public: |
| 28 | + using AssetMap = std::unordered_map<std::string, sk_sp<SkData>>; |
| 29 | + |
| 30 | + static sk_sp<SVGAssetProvider> |
| 31 | + Make(AssetMap assets, skresources::ImageDecodeStrategy strategy = |
| 32 | + skresources::ImageDecodeStrategy::kPreDecode) { |
| 33 | + return sk_sp<SVGAssetProvider>( |
| 34 | + new SVGAssetProvider(std::move(assets), strategy)); |
| 35 | + } |
| 36 | + |
| 37 | + // Override loadImageAsset() to handle image loading |
| 38 | + sk_sp<skresources::ImageAsset> |
| 39 | + loadImageAsset(const char[] /*path*/, const char name[], |
| 40 | + const char[] /*id*/) const override { |
| 41 | + // Identify resources by name only |
| 42 | + auto it = fAssets.find(name); |
| 43 | + if (it != fAssets.end()) { |
| 44 | + // Create ImageAsset from SkData |
| 45 | + return skresources::MultiFrameImageAsset::Make(it->second, fStrategy); |
| 46 | + } |
| 47 | + return nullptr; |
| 48 | + } |
| 49 | + |
| 50 | +private: |
| 51 | + explicit SVGAssetProvider(AssetMap assets, |
| 52 | + skresources::ImageDecodeStrategy strategy) |
| 53 | + : fAssets(std::move(assets)), fStrategy(strategy) { |
| 54 | + } |
| 55 | + const AssetMap fAssets; |
| 56 | + const skresources::ImageDecodeStrategy fStrategy; |
| 57 | +}; |
| 58 | + |
24 | 59 | class JsiSkSVGFactory : public JsiSkHostObject { |
25 | 60 | public: |
26 | | - JSI_HOST_FUNCTION(MakeFromData) { |
27 | | - auto data = JsiSkData::fromValue(runtime, arguments[0]); |
28 | | - auto stream = SkMemoryStream::Make(data); |
| 61 | + // Helper function to parse asset map from JS object |
| 62 | + static SVGAssetProvider::AssetMap parseAssetMap(jsi::Runtime &runtime, |
| 63 | + const jsi::Value &jsValue) { |
| 64 | + SVGAssetProvider::AssetMap assets; |
29 | 65 |
|
30 | | - auto fontMgr = count > 1 && arguments[1].isObject() |
31 | | - ? JsiSkFontMgr::fromValue(runtime, arguments[1]) |
32 | | - : nullptr; |
| 66 | + if (!jsValue.isObject()) { |
| 67 | + return assets; |
| 68 | + } |
33 | 69 |
|
| 70 | + auto jsAssetMap = jsValue.asObject(runtime); |
| 71 | + |
| 72 | + // Convert JS object to C++ AssetMap |
| 73 | + auto propertyNames = jsAssetMap.getPropertyNames(runtime); |
| 74 | + size_t propertyCount = propertyNames.size(runtime); |
| 75 | + |
| 76 | + for (size_t i = 0; i < propertyCount; i++) { |
| 77 | + auto propertyName = |
| 78 | + propertyNames.getValueAtIndex(runtime, i).asString(runtime); |
| 79 | + auto key = propertyName.utf8(runtime); |
| 80 | + auto jsValue = jsAssetMap.getProperty(runtime, propertyName); |
| 81 | + |
| 82 | + // Skip null or undefined values |
| 83 | + if (jsValue.isNull() || jsValue.isUndefined()) { |
| 84 | + continue; |
| 85 | + } |
| 86 | + |
| 87 | + if (jsValue.isObject()) { |
| 88 | + auto jsObject = jsValue.asObject(runtime); |
| 89 | + if (jsObject.isHostObject(runtime)) { |
| 90 | + auto hostObject = jsObject.getHostObject(runtime); |
| 91 | + auto skData = std::dynamic_pointer_cast<JsiSkData>(hostObject); |
| 92 | + if (skData) { |
| 93 | + assets[key] = skData->getObject(); |
| 94 | + } |
| 95 | + } |
| 96 | + } |
| 97 | + } |
| 98 | + |
| 99 | + return assets; |
| 100 | + } |
| 101 | + |
| 102 | +private: |
| 103 | + jsi::Value makeSVGFromStream(jsi::Runtime &runtime, |
| 104 | + std::unique_ptr<SkMemoryStream> stream, |
| 105 | + sk_sp<SkFontMgr> fontMgr, |
| 106 | + SVGAssetProvider::AssetMap assets) { |
34 | 107 | auto builder = SkSVGDOM::Builder(); |
| 108 | + |
35 | 109 | if (fontMgr) { |
36 | 110 | builder.setFontManager(fontMgr); |
37 | 111 | } |
| 112 | + |
| 113 | + auto baseProvider = SVGAssetProvider::Make( |
| 114 | + std::move(assets), skresources::ImageDecodeStrategy::kPreDecode); |
| 115 | + auto provider = skresources::DataURIResourceProviderProxy::Make( |
| 116 | + std::move(baseProvider), skresources::ImageDecodeStrategy::kPreDecode, fontMgr); |
| 117 | + |
| 118 | + // TODO: this sk_sp subclassing issue needs to be fixed. |
| 119 | + provider->ref(); |
| 120 | + builder.setResourceProvider(provider); |
| 121 | + |
38 | 122 | auto svg_dom = builder.make(*stream); |
39 | | - auto svg = std::make_shared<JsiSkSVG>(getContext(), std::move(svg_dom)); |
| 123 | + auto svg = |
| 124 | + std::make_shared<JsiSkSVG>(getContext(), std::move(svg_dom)); |
40 | 125 | return JSI_CREATE_HOST_OBJECT_WITH_MEMORY_PRESSURE(runtime, svg, |
41 | 126 | getContext()); |
42 | 127 | } |
43 | 128 |
|
| 129 | +public: |
| 130 | + JSI_HOST_FUNCTION(MakeFromData) { |
| 131 | + auto data = JsiSkData::fromValue(runtime, arguments[0]); |
| 132 | + auto stream = SkMemoryStream::Make(data); |
| 133 | + |
| 134 | + // Parse fontMgr (second parameter) |
| 135 | + auto fontMgr = count > 1 && arguments[1].isObject() |
| 136 | + ? JsiSkFontMgr::fromValue(runtime, arguments[1]) |
| 137 | + : nullptr; |
| 138 | + |
| 139 | + // Parse assets map (third parameter) |
| 140 | + auto assets = count > 2 ? parseAssetMap(runtime, arguments[2]) |
| 141 | + : SVGAssetProvider::AssetMap(); |
| 142 | + |
| 143 | + return makeSVGFromStream(runtime, std::move(stream), fontMgr, std::move(assets)); |
| 144 | + } |
| 145 | + |
44 | 146 | JSI_HOST_FUNCTION(MakeFromString) { |
45 | 147 | auto svgText = arguments[0].asString(runtime).utf8(runtime); |
46 | 148 | auto stream = SkMemoryStream::MakeDirect(svgText.c_str(), svgText.size()); |
47 | 149 |
|
| 150 | + // Parse fontMgr (second parameter) |
48 | 151 | auto fontMgr = count > 1 && arguments[1].isObject() |
49 | 152 | ? JsiSkFontMgr::fromValue(runtime, arguments[1]) |
50 | 153 | : nullptr; |
51 | 154 |
|
52 | | - auto builder = SkSVGDOM::Builder(); |
53 | | - if (fontMgr) { |
54 | | - builder.setFontManager(fontMgr); |
55 | | - } |
56 | | - auto svg_dom = builder.make(*stream); |
57 | | - auto svg = std::make_shared<JsiSkSVG>(getContext(), std::move(svg_dom)); |
58 | | - return JSI_CREATE_HOST_OBJECT_WITH_MEMORY_PRESSURE(runtime, svg, |
59 | | - getContext()); |
| 155 | + // Parse assets map (third parameter) |
| 156 | + auto assets = count > 2 ? parseAssetMap(runtime, arguments[2]) |
| 157 | + : SVGAssetProvider::AssetMap(); |
| 158 | + |
| 159 | + return makeSVGFromStream(runtime, std::move(stream), fontMgr, std::move(assets)); |
60 | 160 | } |
61 | 161 |
|
62 | 162 | size_t getMemoryPressure() const override { return 512; } |
|
0 commit comments