mac: pdf2png

experimental/tools/coreGraphicsPdf2png.cpp

+/*
+ * Copyright 2015 Google Inc.
+ *
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ */
+
+#include <stdio.h>
+
+#include "SkBitmap.h"
+#include "SkCGUtils.h"
+#include "SkImageEncoder.h"
+#include "SkStream.h"
+
+class StdOutWStream : public SkWStream {
+public:
+    StdOutWStream() : fBytesWritten(0) {}
+    bool write(const void* buffer, size_t size) final {
+        fBytesWritten += size;
+        return size == fwrite(buffer, 1, size, stdout);
+    }
+    size_t bytesWritten() const final { return fBytesWritten; }
+
+private:
+    size_t fBytesWritten;
+};
+
+static SkStreamAsset* open_for_reading(const char* path) {
+    if (!path || !path[0] || 0 == strcmp(path, "-")) {
+        return new SkFILEStream(stdin, SkFILEStream::kCallerRetains_Ownership);
+    }
+    return SkStream::NewFromFile(path);
+}
+
+static SkWStream* open_for_writing(const char* path) {
+    if (!path || !path[0] || 0 == strcmp(path, "-")) {
+        return new StdOutWStream;
+    }
+    return new SkFILEWStream(path);
+}
+
+static bool to_png(SkWStream* o, const SkBitmap& bm) {
+    return SkImageEncoder::EncodeStream(o, bm, SkImageEncoder::kPNG_Type, 100);
+}
+
+// Note: I could implement this using only MacOS|CG API calls, but
+// since most of this is already done in Skia, here it is.
+int main(int argc, char** argv) {
+    SkBitmap bm;
+    SkAutoTDelete<SkStream> in(open_for_reading(argc > 1 ? argv[1] : NULL));
+    SkAutoTDelete<SkWStream> out(open_for_writing(argc > 2 ? argv[2] : NULL));
+    if (SkPDFDocumentToBitmap(in.detach(), &bm) && to_png(out, bm)) {
+        return 0;
+    } else {
+        return 1;
+    }
+}