168 lines
6.7 KiB
C++
168 lines
6.7 KiB
C++
/*************************************************************************/
|
|
/* image_loader_svg.cpp */
|
|
/*************************************************************************/
|
|
/* This file is part of: */
|
|
/* GODOT ENGINE */
|
|
/* https://godotengine.org */
|
|
/*************************************************************************/
|
|
/* Copyright (c) 2007-2019 Juan Linietsky, Ariel Manzur. */
|
|
/* Copyright (c) 2014-2019 Godot Engine contributors (cf. AUTHORS.md) */
|
|
/* */
|
|
/* Permission is hereby granted, free of charge, to any person obtaining */
|
|
/* a copy of this software and associated documentation files (the */
|
|
/* "Software"), to deal in the Software without restriction, including */
|
|
/* without limitation the rights to use, copy, modify, merge, publish, */
|
|
/* distribute, sublicense, and/or sell copies of the Software, and to */
|
|
/* permit persons to whom the Software is furnished to do so, subject to */
|
|
/* the following conditions: */
|
|
/* */
|
|
/* The above copyright notice and this permission notice shall be */
|
|
/* included in all copies or substantial portions of the Software. */
|
|
/* */
|
|
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
|
|
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
|
|
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
|
|
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
|
|
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
|
|
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
|
|
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
|
|
/*************************************************************************/
|
|
|
|
#include "image_loader_svg.h"
|
|
|
|
#include "core/os/os.h"
|
|
#include "core/print_string.h"
|
|
#include "core/ustring.h"
|
|
|
|
void SVGRasterizer::rasterize(NSVGimage *p_image, float p_tx, float p_ty, float p_scale, unsigned char *p_dst, int p_w, int p_h, int p_stride) {
|
|
nsvgRasterize(rasterizer, p_image, p_tx, p_ty, p_scale, p_dst, p_w, p_h, p_stride);
|
|
}
|
|
|
|
SVGRasterizer::SVGRasterizer() {
|
|
rasterizer = nsvgCreateRasterizer();
|
|
}
|
|
SVGRasterizer::~SVGRasterizer() {
|
|
nsvgDeleteRasterizer(rasterizer);
|
|
}
|
|
|
|
SVGRasterizer ImageLoaderSVG::rasterizer;
|
|
|
|
inline void change_nsvg_paint_color(NSVGpaint *p_paint, const uint32_t p_old, const uint32_t p_new) {
|
|
|
|
if (p_paint->type == NSVG_PAINT_COLOR) {
|
|
if (p_paint->color << 8 == p_old << 8) {
|
|
p_paint->color = (p_paint->color & 0xFF000000) | (p_new & 0x00FFFFFF);
|
|
}
|
|
}
|
|
|
|
if (p_paint->type == NSVG_PAINT_LINEAR_GRADIENT || p_paint->type == NSVG_PAINT_RADIAL_GRADIENT) {
|
|
for (int stop_index = 0; stop_index < p_paint->gradient->nstops; stop_index++) {
|
|
if (p_paint->gradient->stops[stop_index].color << 8 == p_old << 8) {
|
|
p_paint->gradient->stops[stop_index].color = p_new;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void ImageLoaderSVG::_convert_colors(NSVGimage *p_svg_image) {
|
|
|
|
for (NSVGshape *shape = p_svg_image->shapes; shape != NULL; shape = shape->next) {
|
|
|
|
for (int i = 0; i < replace_colors.old_colors.size(); i++) {
|
|
change_nsvg_paint_color(&(shape->stroke), replace_colors.old_colors[i], replace_colors.new_colors[i]);
|
|
change_nsvg_paint_color(&(shape->fill), replace_colors.old_colors[i], replace_colors.new_colors[i]);
|
|
}
|
|
}
|
|
}
|
|
|
|
void ImageLoaderSVG::set_convert_colors(Dictionary *p_replace_color) {
|
|
|
|
if (p_replace_color) {
|
|
Dictionary replace_color = *p_replace_color;
|
|
for (int i = 0; i < replace_color.keys().size(); i++) {
|
|
Variant o_c = replace_color.keys()[i];
|
|
Variant n_c = replace_color[replace_color.keys()[i]];
|
|
if (o_c.get_type() == Variant::COLOR && n_c.get_type() == Variant::COLOR) {
|
|
Color old_color = o_c;
|
|
Color new_color = n_c;
|
|
replace_colors.old_colors.push_back(old_color.to_abgr32());
|
|
replace_colors.new_colors.push_back(new_color.to_abgr32());
|
|
}
|
|
}
|
|
} else {
|
|
replace_colors.old_colors.clear();
|
|
replace_colors.new_colors.clear();
|
|
}
|
|
}
|
|
|
|
Error ImageLoaderSVG::_create_image(Ref<Image> p_image, const PoolVector<uint8_t> *p_data, float p_scale, bool upsample, bool convert_colors) {
|
|
NSVGimage *svg_image;
|
|
PoolVector<uint8_t>::Read src_r = p_data->read();
|
|
svg_image = nsvgParse((char *)src_r.ptr(), "px", 96);
|
|
if (svg_image == NULL) {
|
|
ERR_PRINT("SVG Corrupted");
|
|
return ERR_FILE_CORRUPT;
|
|
}
|
|
if (convert_colors)
|
|
_convert_colors(svg_image);
|
|
|
|
float upscale = upsample ? 2.0 : 1.0;
|
|
|
|
int w = (int)(svg_image->width * p_scale * upscale);
|
|
ERR_EXPLAIN(vformat("Can't create image from SVG with scale %s, the resulting image size exceeds max width.", rtos(p_scale)));
|
|
ERR_FAIL_COND_V(w > Image::MAX_WIDTH, ERR_PARAMETER_RANGE_ERROR);
|
|
|
|
int h = (int)(svg_image->height * p_scale * upscale);
|
|
ERR_EXPLAIN(vformat("Can't create image from SVG with scale %s, the resulting image size exceeds max height.", rtos(p_scale)));
|
|
ERR_FAIL_COND_V(h > Image::MAX_HEIGHT, ERR_PARAMETER_RANGE_ERROR);
|
|
|
|
PoolVector<uint8_t> dst_image;
|
|
dst_image.resize(w * h * 4);
|
|
|
|
PoolVector<uint8_t>::Write dw = dst_image.write();
|
|
|
|
rasterizer.rasterize(svg_image, 0, 0, p_scale * upscale, (unsigned char *)dw.ptr(), w, h, w * 4);
|
|
|
|
dw = PoolVector<uint8_t>::Write();
|
|
p_image->create(w, h, false, Image::FORMAT_RGBA8, dst_image);
|
|
if (upsample)
|
|
p_image->shrink_x2();
|
|
|
|
nsvgDelete(svg_image);
|
|
|
|
return OK;
|
|
}
|
|
|
|
Error ImageLoaderSVG::create_image_from_string(Ref<Image> p_image, const char *p_svg_str, float p_scale, bool upsample, bool convert_colors) {
|
|
|
|
size_t str_len = strlen(p_svg_str);
|
|
PoolVector<uint8_t> src_data;
|
|
src_data.resize(str_len + 1);
|
|
PoolVector<uint8_t>::Write src_w = src_data.write();
|
|
memcpy(src_w.ptr(), p_svg_str, str_len + 1);
|
|
|
|
return _create_image(p_image, &src_data, p_scale, upsample, convert_colors);
|
|
}
|
|
|
|
Error ImageLoaderSVG::load_image(Ref<Image> p_image, FileAccess *f, bool p_force_linear, float p_scale) {
|
|
|
|
uint32_t size = f->get_len();
|
|
PoolVector<uint8_t> src_image;
|
|
src_image.resize(size + 1);
|
|
PoolVector<uint8_t>::Write src_w = src_image.write();
|
|
f->get_buffer(src_w.ptr(), size);
|
|
src_w.ptr()[size] = '\0';
|
|
|
|
return _create_image(p_image, &src_image, p_scale, 1.0);
|
|
}
|
|
|
|
void ImageLoaderSVG::get_recognized_extensions(List<String> *p_extensions) const {
|
|
|
|
p_extensions->push_back("svg");
|
|
p_extensions->push_back("svgz");
|
|
}
|
|
|
|
ImageLoaderSVG::ImageLoaderSVG() {
|
|
}
|
|
|
|
ImageLoaderSVG::ReplaceColors ImageLoaderSVG::replace_colors;
|