/*
|
* 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 "SkSurface_Base.h"
|
#include "SkImageInfoPriv.h"
|
#include "SkImagePriv.h"
|
#include "SkCanvas.h"
|
#include "SkDevice.h"
|
#include "SkMallocPixelRef.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::MakeZeroed(info, rowBytes);
|
if (!pr) {
|
return nullptr;
|
}
|
if (rowBytes) {
|
SkASSERT(pr->rowBytes() == rowBytes);
|
}
|
return sk_make_sp<SkSurface_Raster>(info, std::move(pr), props);
|
}
|
