SkPictureImageGenerator

src/core/SkPictureImageGenerator.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 "SkImageGenerator.h"
+
+#include "SkCanvas.h"
+#include "SkMatrix.h"
+#include "SkPaint.h"
+#include "SkPicture.h"
+
+class SkPictureImageGenerator : SkImageGenerator {
+public:
+    static SkImageGenerator* Create(const SkISize&, const SkPicture*, const SkMatrix*,
+                                    const SkPaint*);
+
+protected:
+    bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes, SkPMColor ctable[],
+                     int* ctableCount) override;
+
+private:
+    SkPictureImageGenerator(const SkISize&, const SkPicture*, const SkMatrix*, const SkPaint*);
+
+    SkAutoTUnref<const SkPicture> fPicture;
+    SkMatrix                      fMatrix;
+    SkTLazy<SkPaint>              fPaint;
+
+    typedef SkImageGenerator INHERITED;
+};
+
+SkImageGenerator* SkPictureImageGenerator::Create(const SkISize& size, const SkPicture* picture,
+                                const SkMatrix* matrix, const SkPaint* paint) {
+    if (!picture || size.isEmpty()) {
+        return nullptr;
+    }
+
+    return SkNEW_ARGS(SkPictureImageGenerator, (size, picture, matrix, paint));
+}
+
+SkPictureImageGenerator::SkPictureImageGenerator(const SkISize& size, const SkPicture* picture,
+                                                 const SkMatrix* matrix, const SkPaint* paint)
+    : INHERITED(SkImageInfo::MakeN32Premul(size))
+    , fPicture(SkRef(picture)) {
+
+    if (matrix) {
+        fMatrix = *matrix;
+    } else {
+        fMatrix.reset();
+    }
+
+    if (paint) {
+        fPaint.set(*paint);
+    }
+}
+
+bool SkPictureImageGenerator::onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes,
+                                          SkPMColor ctable[], int* ctableCount) {
+    if (info != getInfo() || ctable || ctableCount) {
+        return false;
+    }
+
+    SkBitmap bitmap;
+    if (!bitmap.installPixels(info, pixels, rowBytes)) {
+        return false;
+    }
+
+    bitmap.eraseColor(SK_ColorTRANSPARENT);
+    SkCanvas canvas(bitmap, SkSurfaceProps(0, kUnknown_SkPixelGeometry));
+    canvas.drawPicture(fPicture, &fMatrix, fPaint.getMaybeNull());
+
+    return true;
+}
+
+SkImageGenerator* SkImageGenerator::NewFromPicture(const SkISize& size, const SkPicture* picture,
+                                                   const SkMatrix* matrix, const SkPaint* paint) {
+    return SkPictureImageGenerator::Create(size, picture, matrix, paint);
+}