| /* | |
| * Copyright 2012 Google Inc. | |
| * | |
| * Use of this source code is governed by a BSD-style license that can be | |
| * found in the LICENSE file. | |
| */ | |
| #include "SkCanvas.h" | |
| #include "SkDevice.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" | |
| __SK_FORCE_IMAGE_DECODER_LINKING; | |
| #ifdef SK_USE_CDB | |
| #include "win_dbghelp.h" | |
| #endif | |
| /** | |
| * render_pdfs | |
| * | |
| * Given list of directories and files to use as input, expects to find .skp | |
| * files and it will convert them to .pdf files writing them in the output | |
| * directory. | |
| * | |
| * Returns zero exit code if all .skp files were converted successfully, | |
| * otherwise returns error code 1. | |
| */ | |
| 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"); | |
| } | |
| /** Replaces the extension of a file. | |
| * @param path File name whose extension will be changed. | |
| * @param old_extension The old extension. | |
| * @param new_extension The new extension. | |
| * @returns false if the file did not has the expected extension. | |
| * if false is returned, contents of path are undefined. | |
| */ | |
| static bool replace_filename_extension(SkString* path, | |
| const char old_extension[], | |
| const char new_extension[]) { | |
| if (path->endsWith(old_extension)) { | |
| path->remove(path->size() - strlen(old_extension), | |
| strlen(old_extension)); | |
| if (!path->endsWith(".")) { | |
| return false; | |
| } | |
| path->append(new_extension); | |
| return true; | |
| } | |
| 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) { | |
| return NULL; | |
| } | |
| SkBitmap bm = bitmap; | |
| #if defined(SK_BUILD_FOR_MAC) | |
| // Workaround bug #1043 where bitmaps with referenced pixels cause | |
| // CGImageDestinationFinalize to crash | |
| SkBitmap copy; | |
| bitmap.deepCopyTo(©); | |
| bm = copy; | |
| #endif | |
| return SkImageEncoder::EncodeData(bm, | |
| SkImageEncoder::kJPEG_Type, | |
| gJpegQuality); | |
| } | |
| /** Builds the output filename. path = dir/name, and it replaces expected | |
| * .skp extension with .pdf extention. | |
| * @param path Output filename. | |
| * @param name The name of the file. | |
| * @returns false if the file did not has the expected extension. | |
| * if false is returned, contents of path are undefined. | |
| */ | |
| static bool make_output_filepath(SkString* path, const SkString& dir, | |
| const SkString& name) { | |
| *path = SkOSPath::SkPathJoin(dir.c_str(), name.c_str()); | |
| return replace_filename_extension(path, | |
| SKP_FILE_EXTENSION, | |
| PDF_FILE_EXTENSION); | |
| } | |
| /** Write the output of pdf renderer to a file. | |
| * @param outputDir Output dir. | |
| * @param inputFilename The skp file that was read. | |
| * @param renderer The object responsible to write the pdf file. | |
| */ | |
| static SkWStream* open_stream(const SkString& outputDir, | |
| const SkString& inputFilename) { | |
| if (outputDir.isEmpty()) { | |
| return SkNEW(SkDynamicMemoryWStream); | |
| } | |
| SkString outputPath; | |
| if (!make_output_filepath(&outputPath, outputDir, inputFilename)) { | |
| return NULL; | |
| } | |
| SkFILEWStream* stream = SkNEW_ARGS(SkFILEWStream, (outputPath.c_str())); | |
| if (!stream->isValid()) { | |
| SkDebugf("Could not write to file %s\n", outputPath.c_str()); | |
| return NULL; | |
| } | |
| return stream; | |
| } | |
| /** 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. | |
| */ | |
| 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; | |
| } | |
| /** 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 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); | |
| } | |
| } else { | |
| inputs->push_back(SkString(*argv)); | |
| } | |
| } | |
| if (inputs->count() < 1) { | |
| usage(argv0); | |
| exit(-1); | |
| } | |
| } | |
| int tool_main_core(int argc, char** argv); | |
| int tool_main_core(int argc, char** argv) { | |
| SkAutoGraphics ag; | |
| SkTArray<SkString> inputs; | |
| SkAutoTUnref<sk_tools::PdfRenderer> | |
| renderer(SkNEW_ARGS(sk_tools::SimplePdfRenderer, (encode_to_dct_data))); | |
| SkASSERT(renderer.get()); | |
| SkString outputDir; | |
| parse_commandline(argc, argv, &inputs, &outputDir); | |
| int failures = 0; | |
| for (int i = 0; i < inputs.count(); i ++) { | |
| failures += process_input(inputs[i], outputDir, *renderer); | |
| } | |
| if (failures != 0) { | |
| SkDebugf("Failed to render %i PDFs.\n", failures); | |
| return 1; | |
| } | |
| return 0; | |
| } | |
| int tool_main(int argc, char** argv); | |
| int tool_main(int argc, char** argv) { | |
| #ifdef SK_USE_CDB | |
| setUpDebuggingFromArgs(argv[0]); | |
| __try { | |
| #endif | |
| return tool_main_core(argc, argv); | |
| #ifdef SK_USE_CDB | |
| } | |
| __except(GenerateDumpAndPrintCallstack(GetExceptionInformation())) | |
| { | |
| return -1; | |
| } | |
| #endif | |
| return 0; | |
| } | |
| #if !defined SK_BUILD_FOR_IOS | |
| int main(int argc, char * const argv[]) { | |
| return tool_main(argc, (char**) argv); | |
| } | |
| #endif |