123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205 |
- /*
- * 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<SkPixelRef>, const SkSurfaceProps*);
- SkCanvas* onNewCanvas() override;
- sk_sp<SkSurface> onNewSurface(const SkImageInfo&) override;
- sk_sp<SkImage> 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<SkPixelRef> 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> 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<SkImage> 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<SkImage> 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> 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<SkSurface_Raster>(info, pixels, rb, releaseProc, context, props);
- }
- sk_sp<SkSurface> SkSurface::MakeRasterDirect(const SkImageInfo& info, void* pixels, size_t rowBytes,
- const SkSurfaceProps* props) {
- return MakeRasterDirectReleaseProc(info, pixels, rowBytes, nullptr, nullptr, props);
- }
- sk_sp<SkSurface> SkSurface::MakeRaster(const SkImageInfo& info, size_t rowBytes,
- const SkSurfaceProps* props) {
- if (!SkSurfaceValidateRasterInfo(info)) {
- return nullptr;
- }
- sk_sp<SkPixelRef> pr = SkMallocPixelRef::MakeAllocate(info, rowBytes);
- if (!pr) {
- return nullptr;
- }
- if (rowBytes) {
- SkASSERT(pr->rowBytes() == rowBytes);
- }
- return sk_make_sp<SkSurface_Raster>(info, std::move(pr), props);
- }
- sk_sp<SkSurface> SkSurface::MakeRasterN32Premul(int width, int height,
- const SkSurfaceProps* surfaceProps) {
- return MakeRaster(SkImageInfo::MakeN32Premul(width, height), surfaceProps);
- }
|