SkDocument::setDCTEncoder() for old versions of webkit

tests/PDFDocumentTest.cpp

 /*
  * Copyright 2013 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 #include "Test.h"
 
+#include "Resources.h"
 #include "SkCanvas.h"
 #include "SkDocument.h"
 #include "SkOSFile.h"
 #include "SkStream.h"
+#include "SkPixelSerializer.h"
 
 static void test_empty(skiatest::Reporter* reporter) {
     SkDynamicMemoryWStream stream;
 
     SkAutoTUnref<SkDocument> doc(SkDocument::CreatePDF(&stream));
 
     doc->close();
 
     REPORTER_ASSERT(reporter, stream.bytesWritten() == 0);
 }
@@ skipping 80 matching lines @@
 }
 
 DEF_TEST(document_tests, reporter) {
     REQUIRE_PDF_DOCUMENT(document_tests, reporter);
     test_empty(reporter);
     test_abort(reporter);
     test_abortWithFile(reporter);
     test_file(reporter);
     test_close(reporter);
 }
+
+namespace {
+class JPEGSerializer final : public SkPixelSerializer {
+    bool onUseEncodedData(const void*, size_t) override { return true; }
+    SkData* onEncode(const SkPixmap& pixmap) override {
+        SkBitmap bm;
+        return bm.installPixels(pixmap.info(),
+                                pixmap.writable_addr(),
+                                pixmap.rowBytes(),
+                                pixmap.ctable(),
+                                nullptr, nullptr)
+            ? SkImageEncoder::EncodeData(bm, SkImageEncoder::kJPEG_Type, 85)
+            : nullptr;
+    }
+};
+}  // namespace
+
+size_t count_bytes(const SkBitmap& bm, bool useDCT) {
+    SkDynamicMemoryWStream stream;
+    SkAutoTUnref<SkDocument> doc;
+    if (useDCT) {
+        SkAutoTUnref<SkPixelSerializer> serializer(new JPEGSerializer);
+        doc.reset(SkDocument::CreatePDF(
+                          &stream, SK_ScalarDefaultRasterDPI, serializer));
+    } else {
+        doc.reset(SkDocument::CreatePDF(&stream));
+    }
+    SkCanvas* canvas = doc->beginPage(64, 64);
+    canvas->drawBitmap(bm, 0, 0);
+    doc->endPage();
+    doc->close();
+    return stream.bytesWritten();
+}
+
+DEF_TEST(document_dct_encoder, r) {
+    REQUIRE_PDF_DOCUMENT(document_dct_encoder, r);
+    SkBitmap bm;
+    if (GetResourceAsBitmap("mandrill_64.png", &bm)) {
+        // Lossy encoding works better on photographs.
+        REPORTER_ASSERT(r, count_bytes(bm, true) < count_bytes(bm, false));
+    }
+}