| /* |
| * 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.release(), &bm) && to_png(out, bm)) { |
| return 0; |
| } else { |
| return 1; |
| } |
| } |