skia2/experimental/tools/multipage_pdf_profiler.cpp
2014-12-02 14:01:46 -08:00

136 lines
4.1 KiB
C++

/*
* Copyright 2014 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 "SkDocument.h"
#include "SkForceLinking.h"
#include "SkGraphics.h"
#include "SkNullCanvas.h"
#include "SkPicture.h"
#include "SkStream.h"
#include "SkTemplates.h"
#include "PageCachingDocument.h"
#include "ProcStats.h"
#include "flags/SkCommandLineFlags.h"
DEFINE_string2(readPath,
r,
"",
"(Required) The path to a .skp Skia Picture file.");
DEFINE_string2(writePath, w, "", "If set, write PDF output to this file.");
DEFINE_bool(cachePages, false, "Use a PageCachingDocument.");
DEFINE_bool(nullCanvas, true, "Render to a SkNullCanvas as a control.");
__SK_FORCE_IMAGE_DECODER_LINKING;
namespace {
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;
};
SkDocument* CreatePDFDocument(SkWStream* out) {
if (FLAGS_cachePages) {
return CreatePageCachingDocument(out);
} else {
return SkDocument::CreatePDF(out);
}
}
} // namespace
int main(int argc, char** argv) {
SkCommandLineFlags::Parse(argc, argv);
if (FLAGS_readPath.isEmpty()) {
SkDebugf("Error: missing requires --readPath option\n");
return 1;
}
const char* path = FLAGS_readPath[0];
SkFILEStream inputStream(path);
if (!inputStream.isValid()) {
SkDebugf("Could not open file %s\n", path);
return 2;
}
SkAutoGraphics ag;
SkAutoTUnref<SkPicture> picture(SkPicture::CreateFromStream(&inputStream));
if (NULL == picture.get()) {
SkDebugf("Could not read an SkPicture from %s\n", path);
return 3;
}
int width = picture->cullRect().width();
int height = picture->cullRect().height();
const int kLetterWidth = 612; // 8.5 * 72
const int kLetterHeight = 792; // 11 * 72
SkRect letterRect = SkRect::MakeWH(SkIntToScalar(kLetterWidth),
SkIntToScalar(kLetterHeight));
int xPages = ((width - 1) / kLetterWidth) + 1;
int yPages = ((height - 1) / kLetterHeight) + 1;
SkAutoTDelete<SkWStream> out(SkNEW(NullWStream));
if (!FLAGS_writePath.isEmpty()) {
SkAutoTDelete<SkFILEWStream> fileStream(
SkNEW_ARGS(SkFILEWStream, (FLAGS_writePath[0])));
if (!fileStream->isValid()) {
SkDebugf("Can't open file \"%s\" for writing.", FLAGS_writePath[0]);
return 1;
}
out.reset(fileStream.detach());
}
SkCanvas* nullCanvas = SkCreateNullCanvas();
SkAutoTUnref<SkDocument> pdfDocument;
if (!FLAGS_nullCanvas) {
pdfDocument.reset(CreatePDFDocument(out.get()));
}
for (int y = 0; y < yPages; ++y) {
for (int x = 0; x < xPages; ++x) {
SkCanvas* canvas;
if (FLAGS_nullCanvas) {
canvas = nullCanvas;
} else {
int w = SkTMin(kLetterWidth, width - (x * kLetterWidth));
int h = SkTMin(kLetterHeight, height - (y * kLetterHeight));
canvas = pdfDocument->beginPage(w, h);
}
{
SkAutoCanvasRestore autoCanvasRestore(canvas, true);
canvas->clipRect(letterRect);
canvas->translate(SkIntToScalar(-kLetterWidth * x),
SkIntToScalar(-kLetterHeight * y));
picture->playback(canvas);
//canvas->drawPicture(picture);
}
canvas->flush();
if (!FLAGS_nullCanvas) {
pdfDocument->endPage();
}
}
}
if (!FLAGS_nullCanvas) {
pdfDocument->close();
pdfDocument.reset(NULL);
}
printf(SK_SIZE_T_SPECIFIER "\t%4d\n",
inputStream.getLength(),
sk_tools::getMaxResidentSetSizeMB());
return 0;
}