/* * Copyright 2012 Google Inc. * * Use of this source code is governed by a BSD-style license that can be * found in the LICENSE file. */ #include "include/core/SkCanvas.h" #include "include/core/SkMallocPixelRef.h" #include "include/private/SkImageInfoPriv.h" #include "src/core/SkDevice.h" #include "src/core/SkImagePriv.h" #include "src/image/SkSurface_Base.h" class SkSurface_Raster : public SkSurface_Base { public: SkSurface_Raster(const SkImageInfo&, void*, size_t rb, void (*releaseProc)(void* pixels, void* context), void* context, const SkSurfaceProps*); SkSurface_Raster(const SkImageInfo& info, sk_sp, const SkSurfaceProps*); SkCanvas* onNewCanvas() override; sk_sp onNewSurface(const SkImageInfo&) override; sk_sp onNewImageSnapshot(const SkIRect* subset) override; void onWritePixels(const SkPixmap&, int x, int y) override; void onDraw(SkCanvas*, SkScalar x, SkScalar y, const SkPaint*) override; void onCopyOnWrite(ContentChangeMode) override; void onRestoreBackingMutability() override; private: SkBitmap fBitmap; size_t fRowBytes; bool fWeOwnThePixels; typedef SkSurface_Base INHERITED; }; /////////////////////////////////////////////////////////////////////////////// bool SkSurfaceValidateRasterInfo(const SkImageInfo& info, size_t rowBytes) { if (!SkImageInfoIsValid(info)) { return false; } if (kIgnoreRowBytesValue == rowBytes) { return true; } int shift = info.shiftPerPixel(); uint64_t minRB = (uint64_t)info.width() << shift; if (minRB > rowBytes) { return false; } size_t alignedRowBytes = rowBytes >> shift << shift; if (alignedRowBytes != rowBytes) { return false; } uint64_t size = sk_64_mul(info.height(), rowBytes); static const size_t kMaxTotalSize = SK_MaxS32; if (size > kMaxTotalSize) { return false; } return true; } SkSurface_Raster::SkSurface_Raster(const SkImageInfo& info, void* pixels, size_t rb, void (*releaseProc)(void* pixels, void* context), void* context, const SkSurfaceProps* props) : INHERITED(info, props) { fBitmap.installPixels(info, pixels, rb, releaseProc, context); fRowBytes = 0; // don't need to track the rowbytes fWeOwnThePixels = false; // We are "Direct" } SkSurface_Raster::SkSurface_Raster(const SkImageInfo& info, sk_sp pr, const SkSurfaceProps* props) : INHERITED(pr->width(), pr->height(), props) { fBitmap.setInfo(info, pr->rowBytes()); fRowBytes = pr->rowBytes(); // we track this, so that subsequent re-allocs will match fBitmap.setPixelRef(std::move(pr), 0, 0); fWeOwnThePixels = true; } SkCanvas* SkSurface_Raster::onNewCanvas() { return new SkCanvas(fBitmap, this->props()); } sk_sp SkSurface_Raster::onNewSurface(const SkImageInfo& info) { return SkSurface::MakeRaster(info, &this->props()); } void SkSurface_Raster::onDraw(SkCanvas* canvas, SkScalar x, SkScalar y, const SkPaint* paint) { canvas->drawBitmap(fBitmap, x, y, paint); } sk_sp SkSurface_Raster::onNewImageSnapshot(const SkIRect* subset) { if (subset) { SkASSERT(SkIRect::MakeWH(fBitmap.width(), fBitmap.height()).contains(*subset)); SkBitmap dst; dst.allocPixels(fBitmap.info().makeWH(subset->width(), subset->height())); SkAssertResult(fBitmap.readPixels(dst.pixmap(), subset->left(), subset->top())); dst.setImmutable(); // key, so MakeFromBitmap doesn't make a copy of the buffer return SkImage::MakeFromBitmap(dst); } SkCopyPixelsMode cpm = kIfMutable_SkCopyPixelsMode; if (fWeOwnThePixels) { // SkImage_raster requires these pixels are immutable for its full lifetime. // We'll undo this via onRestoreBackingMutability() if we can avoid the COW. if (SkPixelRef* pr = fBitmap.pixelRef()) { pr->setTemporarilyImmutable(); } } else { cpm = kAlways_SkCopyPixelsMode; } // Our pixels are in memory, so read access on the snapshot SkImage could be cheap. // Lock the shared pixel ref to ensure peekPixels() is usable. return SkMakeImageFromRasterBitmap(fBitmap, cpm); } void SkSurface_Raster::onWritePixels(const SkPixmap& src, int x, int y) { fBitmap.writePixels(src, x, y); } void SkSurface_Raster::onRestoreBackingMutability() { SkASSERT(!this->hasCachedImage()); // Shouldn't be any snapshots out there. if (SkPixelRef* pr = fBitmap.pixelRef()) { pr->restoreMutability(); } } void SkSurface_Raster::onCopyOnWrite(ContentChangeMode mode) { // are we sharing pixelrefs with the image? sk_sp cached(this->refCachedImage()); SkASSERT(cached); if (SkBitmapImageGetPixelRef(cached.get()) == fBitmap.pixelRef()) { SkASSERT(fWeOwnThePixels); if (kDiscard_ContentChangeMode == mode) { fBitmap.allocPixels(); } else { SkBitmap prev(fBitmap); fBitmap.allocPixels(); SkASSERT(prev.info() == fBitmap.info()); SkASSERT(prev.rowBytes() == fBitmap.rowBytes()); memcpy(fBitmap.getPixels(), prev.getPixels(), fBitmap.computeByteSize()); } SkASSERT(fBitmap.rowBytes() == fRowBytes); // be sure we always use the same value // Now fBitmap is a deep copy of itself (and therefore different from // what is being used by the image. Next we update the canvas to use // this as its backend, so we can't modify the image's pixels anymore. SkASSERT(this->getCachedCanvas()); this->getCachedCanvas()->getDevice()->replaceBitmapBackendForRasterSurface(fBitmap); } } /////////////////////////////////////////////////////////////////////////////// sk_sp SkSurface::MakeRasterDirectReleaseProc(const SkImageInfo& info, void* pixels, size_t rb, void (*releaseProc)(void* pixels, void* context), void* context, const SkSurfaceProps* props) { if (nullptr == releaseProc) { context = nullptr; } if (!SkSurfaceValidateRasterInfo(info, rb)) { return nullptr; } if (nullptr == pixels) { return nullptr; } return sk_make_sp(info, pixels, rb, releaseProc, context, props); } sk_sp SkSurface::MakeRasterDirect(const SkImageInfo& info, void* pixels, size_t rowBytes, const SkSurfaceProps* props) { return MakeRasterDirectReleaseProc(info, pixels, rowBytes, nullptr, nullptr, props); } sk_sp SkSurface::MakeRaster(const SkImageInfo& info, size_t rowBytes, const SkSurfaceProps* props) { if (!SkSurfaceValidateRasterInfo(info)) { return nullptr; } sk_sp pr = SkMallocPixelRef::MakeAllocate(info, rowBytes); if (!pr) { return nullptr; } if (rowBytes) { SkASSERT(pr->rowBytes() == rowBytes); } return sk_make_sp(info, std::move(pr), props); } sk_sp SkSurface::MakeRasterN32Premul(int width, int height, const SkSurfaceProps* surfaceProps) { return MakeRaster(SkImageInfo::MakeN32Premul(width, height), surfaceProps); }