blob: 8e791d9608dd9b0e1009e9e16aa8ab0cc419f971 [file] [log] [blame]
/*
* 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 "SkCommandLineFlags.h"
#include "SkDocument.h"
#include "SkForceLinking.h"
#include "SkGraphics.h"
#include "SkImageEncoder.h"
#include "SkOSFile.h"
#include "SkPicture.h"
#include "SkStream.h"
#include "SkTArray.h"
#include "SkTSort.h"
#include "ProcStats.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";
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.
* @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;
}
// the size_t* parameter is deprecated, so we ignore it
static SkData* encode_to_dct_data(size_t*, const SkBitmap& bitmap) {
if (FLAGS_jpegQuality == -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(&copy);
bm = copy;
#endif
return SkImageEncoder::EncodeData(
bm, SkImageEncoder::kJPEG_Type, FLAGS_jpegQuality);
}
/** 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::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.
* @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(NullWStream);
}
SkString outputPath;
if (!make_output_filepath(&outputPath, outputDir, inputFilename)) {
return NULL;
}
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.detach();
}
/**
* Given a SkPicture, write a one-page PDF document to the given
* output, using the provided encoder.
*/
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();
}
static bool operator<(const SkString& a, const SkString& b) {
return strcmp(a.c_str(), b.c_str()) < 0;
}
/**
* @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 {
if (!SkCommandLineFlags::ShouldSkip(FLAGS_match, input)) {
files->push_back(SkString(input));
}
}
}
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) {
SkCommandLineFlags::Parse(argc, argv);
SkAutoGraphics ag;
SkString 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;
}
}
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_mb = sk_tools::getMaxResidentSetSizeMB();
if (max_rss_mb >= 0) {
SkDebugf(" %4dM peak rss", max_rss_mb);
}
SkDebugf("\n");
}
if (failures != 0) {
SkDebugf("Failed to render %i of %i PDFs.\n", failures, files.count());
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