*/
#include "SkCanvas.h"
-#include "SkDevice.h"
+#include "SkCommandLineFlags.h"
+#include "SkDocument.h"
#include "SkForceLinking.h"
#include "SkGraphics.h"
#include "SkImageEncoder.h"
#include "SkOSFile.h"
#include "SkPicture.h"
-#include "SkPixelRef.h"
#include "SkStream.h"
#include "SkTArray.h"
-#include "PdfRenderer.h"
-#include "picture_utils.h"
+#include "SkTSort.h"
+#include "ProcStats.h"
__SK_FORCE_IMAGE_DECODER_LINKING;
static const char PDF_FILE_EXTENSION[] = "pdf";
static const char SKP_FILE_EXTENSION[] = "skp";
-static void usage(const char* argv0) {
- SkDebugf("SKP to PDF rendering tool\n");
- SkDebugf("\n"
-"Usage: \n"
-" %s <input>... [-w <outputDir>] [--jpegQuality N] \n"
-, argv0);
- SkDebugf("\n\n");
- SkDebugf(
-" input: A list of directories and files to use as input. Files are\n"
-" expected to have the .skp extension.\n\n");
- SkDebugf(
-" outputDir: directory to write the rendered pdfs.\n\n");
- SkDebugf("\n");
- SkDebugf(
-" jpegQuality N: encodes images in JPEG at quality level N, which can\n"
-" be in range 0-100).\n"
-" N = -1 will disable JPEG compression.\n"
-" Default is N = 100, maximum quality.\n\n");
- SkDebugf("\n");
-}
+
+DEFINE_string2(inputPaths, r, "",
+ "A list of directories and files to use as input. "
+ "Files are expected to have the .skp extension.");
+
+DEFINE_string2(outputDir, w, "",
+ "Directory to write the rendered pdfs.");
+
+DEFINE_string2(match, m, "",
+ "[~][^]substring[$] [...] of filenames to run.\n"
+ "Multiple matches may be separated by spaces.\n"
+ "~ causes a matching file to always be skipped\n"
+ "^ requires the start of the file to match\n"
+ "$ requires the end of the file to match\n"
+ "^ and $ requires an exact match\n"
+ "If a file does not match any list entry,\n"
+ "it is skipped unless some list entry starts with ~");
+
+DEFINE_int32(jpegQuality, 100,
+ "Encodes images in JPEG at quality level N, which can be in "
+ "range 0-100). N = -1 will disable JPEG compression. "
+ "Default is N = 100, maximum quality.");
/** Replaces the extension of a file.
* @param path File name whose extension will be changed.
return false;
}
-int gJpegQuality = 100;
// the size_t* parameter is deprecated, so we ignore it
static SkData* encode_to_dct_data(size_t*, const SkBitmap& bitmap) {
- if (gJpegQuality == -1) {
+ if (FLAGS_jpegQuality == -1) {
return NULL;
}
bm = copy;
#endif
- return SkImageEncoder::EncodeData(bm,
- SkImageEncoder::kJPEG_Type,
- gJpegQuality);
+ return SkImageEncoder::EncodeData(
+ bm, SkImageEncoder::kJPEG_Type, FLAGS_jpegQuality);
}
/** Builds the output filename. path = dir/name, and it replaces expected
*/
static bool make_output_filepath(SkString* path, const SkString& dir,
const SkString& name) {
- *path = SkOSPath::SkPathJoin(dir.c_str(), name.c_str());
+ *path = SkOSPath::Join(dir.c_str(), name.c_str());
return replace_filename_extension(path,
SKP_FILE_EXTENSION,
PDF_FILE_EXTENSION);
}
+namespace {
+// This is a write-only stream.
+class NullWStream : public SkWStream {
+public:
+ NullWStream() : fBytesWritten(0) { }
+ virtual bool write(const void*, size_t size) SK_OVERRIDE {
+ fBytesWritten += size;
+ return true;
+ }
+ virtual size_t bytesWritten() const SK_OVERRIDE { return fBytesWritten; }
+ size_t fBytesWritten;
+};
+} // namespace
+
/** Write the output of pdf renderer to a file.
* @param outputDir Output dir.
* @param inputFilename The skp file that was read.
static SkWStream* open_stream(const SkString& outputDir,
const SkString& inputFilename) {
if (outputDir.isEmpty()) {
- return SkNEW(SkDynamicMemoryWStream);
+ return SkNEW(NullWStream);
}
SkString outputPath;
return NULL;
}
- SkFILEWStream* stream = SkNEW_ARGS(SkFILEWStream, (outputPath.c_str()));
- if (!stream->isValid()) {
+ SkAutoTDelete<SkFILEWStream> stream(
+ SkNEW_ARGS(SkFILEWStream, (outputPath.c_str())));
+ if (!stream.get() || !stream->isValid()) {
SkDebugf("Could not write to file %s\n", outputPath.c_str());
return NULL;
}
- return stream;
+ return stream.detach();
}
-/** Reads an skp file, renders it to pdf and writes the output to a pdf file
- * @param inputPath The skp file to be read.
- * @param outputDir Output dir.
- * @param renderer The object responsible to render the skp object into pdf.
+/**
+ * Given a SkPicture, write a one-page PDF document to the given
+ * output, using the provided encoder.
*/
-static bool render_pdf(const SkString& inputPath, const SkString& outputDir,
- sk_tools::PdfRenderer& renderer) {
- SkString inputFilename = SkOSPath::SkBasename(inputPath.c_str());
-
- SkFILEStream inputStream;
- inputStream.setPath(inputPath.c_str());
- if (!inputStream.isValid()) {
- SkDebugf("Could not open file %s\n", inputPath.c_str());
- return false;
- }
-
- SkAutoTUnref<SkPicture> picture(SkPicture::CreateFromStream(&inputStream));
-
- if (NULL == picture.get()) {
- SkDebugf("Could not read an SkPicture from %s\n", inputPath.c_str());
- return false;
- }
-
- SkDebugf("exporting... [%i %i] %s\n", picture->width(), picture->height(),
- inputPath.c_str());
-
- SkWStream* stream(open_stream(outputDir, inputFilename));
-
- if (!stream) {
- return false;
- }
-
- renderer.init(picture, stream);
-
- bool success = renderer.render();
- SkDELETE(stream);
-
- renderer.end();
-
- return success;
+static bool pdf_to_stream(SkPicture* picture,
+ SkWStream* output,
+ SkPicture::EncodeBitmap encoder) {
+ SkAutoTUnref<SkDocument> pdfDocument(
+ SkDocument::CreatePDF(output, NULL, encoder));
+ SkCanvas* canvas = pdfDocument->beginPage(
+ SkIntToScalar(picture->width()),
+ SkIntToScalar(picture->height()));
+ canvas->drawPicture(picture);
+ canvas->flush();
+ return pdfDocument->close();
}
-/** For each file in the directory or for the file passed in input, call
- * render_pdf.
- * @param input A directory or an skp file.
- * @param outputDir Output dir.
- * @param renderer The object responsible to render the skp object into pdf.
- */
-static int process_input(const SkString& input, const SkString& outputDir,
- sk_tools::PdfRenderer& renderer) {
- int failures = 0;
- if (sk_isdir(input.c_str())) {
- SkOSFile::Iter iter(input.c_str(), SKP_FILE_EXTENSION);
- SkString inputFilename;
- while (iter.next(&inputFilename)) {
- SkString inputPath = SkOSPath::SkPathJoin(input.c_str(), inputFilename.c_str());
- if (!render_pdf(inputPath, outputDir, renderer)) {
- ++failures;
- }
- }
- } else {
- SkString inputPath(input);
- if (!render_pdf(inputPath, outputDir, renderer)) {
- ++failures;
- }
- }
- return failures;
+static bool operator<(const SkString& a, const SkString& b) {
+ return strcmp(a.c_str(), b.c_str()) < 0;
}
-static void parse_commandline(int argc, char* const argv[],
- SkTArray<SkString>* inputs,
- SkString* outputDir) {
- const char* argv0 = argv[0];
- char* const* stop = argv + argc;
-
- for (++argv; argv < stop; ++argv) {
- if ((0 == strcmp(*argv, "-h")) || (0 == strcmp(*argv, "--help"))) {
- usage(argv0);
- exit(-1);
- } else if (0 == strcmp(*argv, "-w")) {
- ++argv;
- if (argv >= stop) {
- SkDebugf("Missing outputDir for -w\n");
- usage(argv0);
- exit(-1);
- }
- *outputDir = SkString(*argv);
- } else if (0 == strcmp(*argv, "--jpegQuality")) {
- ++argv;
- if (argv >= stop) {
- SkDebugf("Missing argument for --jpegQuality\n");
- usage(argv0);
- exit(-1);
- }
- gJpegQuality = atoi(*argv);
- if (gJpegQuality < -1 || gJpegQuality > 100) {
- SkDebugf("Invalid argument for --jpegQuality\n");
- usage(argv0);
- exit(-1);
+/**
+ * @param A list of directories or a skp files.
+ * @returns an alphabetical list of skp files.
+ */
+static void process_input_files(
+ const SkCommandLineFlags::StringArray& inputs,
+ SkTArray<SkString>* files) {
+ for (int i = 0; i < inputs.count(); i ++) {
+ const char* input = inputs[i];
+ if (sk_isdir(input)) {
+ SkOSFile::Iter iter(input, SKP_FILE_EXTENSION);
+ SkString inputFilename;
+ while (iter.next(&inputFilename)) {
+ if (!SkCommandLineFlags::ShouldSkip(
+ FLAGS_match, inputFilename.c_str())) {
+ files->push_back(
+ SkOSPath::Join(input, inputFilename.c_str()));
+ }
}
} else {
- inputs->push_back(SkString(*argv));
+ if (!SkCommandLineFlags::ShouldSkip(FLAGS_match, input)) {
+ files->push_back(SkString(input));
+ }
}
}
-
- if (inputs->count() < 1) {
- usage(argv0);
- exit(-1);
+ if (files->count() > 0) {
+ SkTQSort<SkString>(files->begin(), files->end() - 1);
}
}
+/** For each input skp file, read it, render it to pdf and write. the
+ * output to a pdf file
+ */
int tool_main_core(int argc, char** argv);
int tool_main_core(int argc, char** argv) {
- SkAutoGraphics ag;
- SkTArray<SkString> inputs;
+ SkCommandLineFlags::Parse(argc, argv);
- SkAutoTUnref<sk_tools::PdfRenderer>
- renderer(SkNEW_ARGS(sk_tools::SimplePdfRenderer, (encode_to_dct_data)));
- SkASSERT(renderer.get());
+ SkAutoGraphics ag;
SkString outputDir;
- parse_commandline(argc, argv, &inputs, &outputDir);
+ if (FLAGS_outputDir.count() > 0) {
+ outputDir = FLAGS_outputDir[0];
+ if (!sk_mkdir(outputDir.c_str())) {
+ SkDebugf("Unable to mkdir '%s'\n", outputDir.c_str());
+ return 1;
+ }
+ }
- int failures = 0;
- for (int i = 0; i < inputs.count(); i ++) {
- failures += process_input(inputs[i], outputDir, *renderer);
+ SkTArray<SkString> files;
+ process_input_files(FLAGS_inputPaths, &files);
+
+ size_t maximumPathLength = 0;
+ for (int i = 0; i < files.count(); i ++) {
+ SkString basename = SkOSPath::Basename(files[i].c_str());
+ maximumPathLength = SkTMax(maximumPathLength, basename.size());
}
+ int failures = 0;
+ for (int i = 0; i < files.count(); i ++) {
+ SkString basename = SkOSPath::Basename(files[i].c_str());
+
+ SkFILEStream inputStream;
+ inputStream.setPath(files[i].c_str());
+ if (!inputStream.isValid()) {
+ SkDebugf("Could not open file %s\n", files[i].c_str());
+ ++failures;
+ continue;
+ }
+
+ SkAutoTUnref<SkPicture> picture(
+ SkPicture::CreateFromStream(&inputStream));
+ if (NULL == picture.get()) {
+ SkDebugf("Could not read an SkPicture from %s\n",
+ files[i].c_str());
+ ++failures;
+ continue;
+ }
+ SkDebugf("[%-4i %6i] %-*s", picture->width(), picture->height(),
+ maximumPathLength, basename.c_str());
+
+ SkAutoTDelete<SkWStream> stream(open_stream(outputDir, files[i]));
+ if (!stream.get()) {
+ ++failures;
+ continue;
+ }
+ if (!pdf_to_stream(picture, stream.get(), encode_to_dct_data)) {
+ SkDebugf("Error in PDF Serialization.");
+ ++failures;
+ }
+
+ int max_rss_kb = sk_tools::getMaxResidentSetSizeKB();
+ if (max_rss_kb >= 0) {
+ SkDebugf(" %4dM peak rss", max_rss_kb / 1024);
+ }
+
+ SkDebugf("\n");
+ }
if (failures != 0) {
- SkDebugf("Failed to render %i PDFs.\n", failures);
+ SkDebugf("Failed to render %i of %i PDFs.\n", failures, files.count());
return 1;
}
#endif
return 0;
}
-
#if !defined SK_BUILD_FOR_IOS
int main(int argc, char * const argv[]) {
return tool_main(argc, (char**) argv);