skia2/modules/svg/utils/SvgTool.cpp
Florin Malita dd29e20904 [svg] Plumb a ResourceProvider
... for image loading.

Update the SVG tools to pass local/FS resource providers.

Change-Id: I2c0e446047da87f177fde0f23b7ea1f0a856e808
Reviewed-on: https://skia-review.googlesource.com/c/skia/+/359996
Commit-Queue: Florin Malita <fmalita@google.com>
Reviewed-by: Tyler Denniston <tdenniston@google.com>
2021-01-27 14:10:09 +00:00

82 lines
2.4 KiB
C++

/*
* Copyright 2020 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file.
*/
#include <iostream>
#include "include/core/SkMatrix.h"
#include "include/core/SkStream.h"
#include "include/core/SkSurface.h"
#include "include/encode/SkPngEncoder.h"
#include "modules/skresources/include/SkResources.h"
#include "modules/svg/include/SkSVGDOM.h"
#include "src/utils/SkOSPath.h"
#include "tools/flags/CommandLineFlags.h"
static DEFINE_string2(input , i, nullptr, "Input SVG file.");
static DEFINE_string2(output, o, nullptr, "Output PNG file.");
static DEFINE_int(width , 1024, "Output width.");
static DEFINE_int(height, 1024, "Output height.");
int main(int argc, char** argv) {
CommandLineFlags::Parse(argc, argv);
if (FLAGS_input.isEmpty() || FLAGS_output.isEmpty()) {
std::cerr << "Missing required 'input' and 'output' args.\n";
return 1;
}
if (FLAGS_width <= 0 || FLAGS_height <= 0) {
std::cerr << "Invalid width/height.\n";
return 1;
}
SkFILEStream in(FLAGS_input[0]);
if (!in.isValid()) {
std::cerr << "Could not open " << FLAGS_input[0] << "\n";
return 1;
}
auto rp = skresources::DataURIResourceProviderProxy::Make(
skresources::FileResourceProvider::Make(SkOSPath::Dirname(FLAGS_input[0]),
/*predecode=*/true),
/*predecode=*/true);
auto svg_dom = SkSVGDOM::Builder()
.setFontManager(SkFontMgr::RefDefault())
.setResourceProvider(std::move(rp))
.make(in);
if (!svg_dom) {
std::cerr << "Could not parse " << FLAGS_input[0] << "\n";
return 1;
}
auto surface = SkSurface::MakeRasterN32Premul(FLAGS_width, FLAGS_height);
svg_dom->setContainerSize(SkSize::Make(FLAGS_width, FLAGS_height));
svg_dom->render(surface->getCanvas());
SkPixmap pixmap;
surface->peekPixels(&pixmap);
SkFILEWStream out(FLAGS_output[0]);
if (!out.isValid()) {
std::cerr << "Could not open " << FLAGS_output[0] << " for writing.\n";
return 1;
}
// Use default encoding options.
SkPngEncoder::Options png_options;
if (!SkPngEncoder::Encode(&out, pixmap, png_options)) {
std::cerr << "PNG encoding failed.\n";
return 1;
}
return 0;
}