diff --git a/src/core/SkPictureImageGenerator.cpp b/src/core/SkPictureImageGenerator.cpp index a960001e13..34046e9ae4 100644 --- a/src/core/SkPictureImageGenerator.cpp +++ b/src/core/SkPictureImageGenerator.cpp @@ -7,6 +7,7 @@ #include "SkImage_Base.h" #include "SkCanvas.h" +#include "SkColorSpaceXformCanvas.h" #include "SkMakeUnique.h" #include "SkMatrix.h" #include "SkPaint.h" @@ -69,6 +70,25 @@ bool SkPictureImageGenerator::onGetPixels(const SkImageInfo& info, void* pixels, return true; } +bool SkPictureImageGenerator::onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes, + const Options& opts) { + // No need to use the xform canvas if we want fully color correct behavior or if we do not + // have a destination color space. + if (SkTransferFunctionBehavior::kRespect == opts.fBehavior || !info.colorSpace()) { + return this->onGetPixels(info, pixels, rowBytes, opts.fColorTable, opts.fColorTableCount); + } + + auto canvas = SkCanvas::MakeRasterDirect(info.makeColorSpace(nullptr), pixels, rowBytes); + if (!canvas) { + return false; + } + canvas->clear(0); + + auto xformCanvas = SkCreateColorSpaceXformCanvas(canvas.get(), info.refColorSpace()); + xformCanvas->drawPicture(fPicture, &fMatrix, fPaint.getMaybeNull()); + return true; +} + /////////////////////////////////////////////////////////////////////////////////////////////////// std::unique_ptr diff --git a/src/core/SkPictureImageGenerator.h b/src/core/SkPictureImageGenerator.h index 83872608a5..ed5e87cfe5 100644 --- a/src/core/SkPictureImageGenerator.h +++ b/src/core/SkPictureImageGenerator.h @@ -19,6 +19,8 @@ public: protected: bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes, SkPMColor ctable[], int* ctableCount) override; + bool onGetPixels(const SkImageInfo& info, void* pixels, size_t rowBytes, const Options& opts) + override; #if SK_SUPPORT_GPU sk_sp onGenerateTexture(GrContext*, const SkImageInfo&,