SkPDF/Bench: add bench for SkPDFSharedStream (deflate)

bench/PDFBench.cpp

 /*
  * Copyright 2016 Google Inc.
  *
  * Use of this source code is governed by a BSD-style license that can be
  * found in the LICENSE file.
  */
 
 #include "Benchmark.h"
 #include "Resources.h"
+#include "SkData.h"
 #include "SkImage.h"
+#include "SkPDFBitmap.h"
 #include "SkPixmap.h"
-#include "SkData.h"
-
-#if SK_SUPPORT_PDF
-
-#include "SkPDFBitmap.h"
+#include "SkRandom.h"
 
 namespace {
 struct NullWStream : public SkWStream {
     NullWStream() : fN(0) {}
     bool write(const void*, size_t n) override { fN += n; return true; }
     size_t bytesWritten() const override { return fN; }
     size_t fN;
 };
 
-static void test_pdf_image_serialization(SkImage* img) {
-    SkAutoTUnref<SkPDFObject> object(
-            SkPDFCreateBitmapObject(img, nullptr));
-    if (!object) {
-        SkDEBUGFAIL("");
-        return;
-    }
+static void test_pdf_object_serialization(SkPDFObject* object) {
     // SkDebugWStream wStream;
     NullWStream wStream;
     SkPDFSubstituteMap substitutes;
     SkPDFObjNumMap objNumMap;
     objNumMap.addObjectRecursively(object, substitutes);
     for (int i = 0; i < objNumMap.objects().count(); ++i) {
         SkPDFObject* object = objNumMap.objects()[i];
         wStream.writeDecAsText(i + 1);
         wStream.writeText(" 0 obj\n");
         object->emitObject(&wStream, objNumMap, substitutes);
         wStream.writeText("\nendobj\n");
     }
 }
 
+static void test_pdf_image_serialization(SkImage* img) {
+    SkAutoTUnref<SkPDFObject> object(
+            SkPDFCreateBitmapObject(img, nullptr));
+    if (!object) {
+        SkDEBUGFAIL("");
+        return;
+    }
+    test_pdf_object_serialization(object);
+}
+
 class PDFImageBench : public Benchmark {
 public:
     PDFImageBench() {}
     virtual ~PDFImageBench() {}
 
 protected:
     const char* onGetName() override { return "PDFImage"; }
     bool isSuitableFor(Backend backend) override {
         return backend == kNonRendering_Backend;
     }
@@ skipping 49 matching lines @@
         }
         while (loops-- > 0) {
             test_pdf_image_serialization(fImage);
         }
     }
 
 private:
     SkAutoTUnref<SkImage> fImage;
 };
 
+static SkAutoTDelete<SkStreamAsset> generate_random_text() {
+    SkRandom random(0x5EED);
+    size_t N = 1 << 16;
+    SkAutoTDelete<SkStreamAsset> buffer(new SkMemoryStream(N));
+    uint8_t* ptr = (uint8_t*)(buffer->getMemoryBase());
+    const uint8_t* const end = &ptr[N];
+    while (ptr != end) {
+        *ptr++ = SkToU8(random.nextRangeU(' ', '~'));
+    }
+    return buffer;
+}
+
+/** Test calling DEFLATE on 64k of random text. Used for measuring
+    alternate zlib settings, usage, and library versions. */
+class PDFCompressionBench : public Benchmark {
+public:
+    PDFCompressionBench() {}
+    virtual ~PDFCompressionBench() {}
+
+protected:
+    const char* onGetName() override { return "PDFCompression"; }
+    bool isSuitableFor(Backend backend) override {
+        return backend == kNonRendering_Backend;
+    }
+    void onDelayedSetup() override {
+        fRandomText = generate_random_text();
+    }
+    void onDraw(int loops, SkCanvas*) override {
+        SkASSERT(fRandomText);
+        while (loops-- > 0) {
+            SkAutoTUnref<SkPDFObject> object(
+                    new SkPDFSharedStream(fRandomText->duplicate()));
+            test_pdf_object_serialization(object);
+        }
+    }
+
+private:
+    SkAutoTDelete<SkStreamAsset> fRandomText;
+};
+
 }  // namespace
 DEF_BENCH(return new PDFImageBench;)
 DEF_BENCH(return new PDFJpegImageBench;)
-
-#endif  // SK_SUPPORT_PDF
+DEF_BENCH(return new PDFCompressionBench;)