blob: 3aed94ca27b0a1ba6e522faea5b53f376b53f706 [file] [log] [blame]
/*
* Copyright 2016 Google Inc.
*
* Use of this source code is governed by a BSD-style license that can be
* found in the LICENSE file
*/
#include "src/core/SkSpecialSurface.h"
#include <memory>
#include "include/core/SkCanvas.h"
#include "src/core/SkSpecialImage.h"
#include "src/core/SkSurfacePriv.h"
///////////////////////////////////////////////////////////////////////////////
class SkSpecialSurface_Base : public SkSpecialSurface {
public:
SkSpecialSurface_Base(const SkIRect& subset, const SkSurfaceProps& props)
: INHERITED(subset, props)
, fCanvas(nullptr) {
}
// reset is called after an SkSpecialImage has been snapped
void reset() { fCanvas.reset(); }
// This can return nullptr if reset has already been called or something when wrong in the ctor
SkCanvas* onGetCanvas() { return fCanvas.get(); }
virtual sk_sp<SkSpecialImage> onMakeImageSnapshot() = 0;
protected:
std::unique_ptr<SkCanvas> fCanvas; // initialized by derived classes in ctors
private:
using INHERITED = SkSpecialSurface;
};
///////////////////////////////////////////////////////////////////////////////
static SkSpecialSurface_Base* as_SB(SkSpecialSurface* surface) {
return static_cast<SkSpecialSurface_Base*>(surface);
}
SkSpecialSurface::SkSpecialSurface(const SkIRect& subset,
const SkSurfaceProps& props)
: fProps(props.flags(), kUnknown_SkPixelGeometry)
, fSubset(subset) {
SkASSERT(fSubset.width() > 0);
SkASSERT(fSubset.height() > 0);
}
SkCanvas* SkSpecialSurface::getCanvas() {
return as_SB(this)->onGetCanvas();
}
sk_sp<SkSpecialImage> SkSpecialSurface::makeImageSnapshot() {
sk_sp<SkSpecialImage> image(as_SB(this)->onMakeImageSnapshot());
as_SB(this)->reset();
return image; // the caller gets the creation ref
}
///////////////////////////////////////////////////////////////////////////////
#include "include/core/SkMallocPixelRef.h"
class SkSpecialSurface_Raster : public SkSpecialSurface_Base {
public:
SkSpecialSurface_Raster(const SkImageInfo& info,
sk_sp<SkPixelRef> pr,
const SkIRect& subset,
const SkSurfaceProps& props)
: INHERITED(subset, props) {
SkASSERT(info.width() == pr->width() && info.height() == pr->height());
fBitmap.setInfo(info, info.minRowBytes());
fBitmap.setPixelRef(std::move(pr), 0, 0);
fCanvas = std::make_unique<SkCanvas>(fBitmap, this->props());
fCanvas->clipRect(SkRect::Make(subset));
#ifdef SK_IS_BOT
fCanvas->clear(SK_ColorRED); // catch any imageFilter sloppiness
#endif
}
~SkSpecialSurface_Raster() override { }
sk_sp<SkSpecialImage> onMakeImageSnapshot() override {
return SkSpecialImage::MakeFromRaster(this->subset(), fBitmap, this->props());
}
private:
SkBitmap fBitmap;
using INHERITED = SkSpecialSurface_Base;
};
sk_sp<SkSpecialSurface> SkSpecialSurface::MakeFromBitmap(const SkIRect& subset, SkBitmap& bm,
const SkSurfaceProps& props) {
if (subset.isEmpty() || !SkSurfaceValidateRasterInfo(bm.info(), bm.rowBytes())) {
return nullptr;
}
return sk_make_sp<SkSpecialSurface_Raster>(bm.info(), sk_ref_sp(bm.pixelRef()), subset, props);
}
sk_sp<SkSpecialSurface> SkSpecialSurface::MakeRaster(const SkImageInfo& info,
const SkSurfaceProps& props) {
if (!SkSurfaceValidateRasterInfo(info)) {
return nullptr;
}
sk_sp<SkPixelRef> pr = SkMallocPixelRef::MakeAllocate(info, 0);
if (!pr) {
return nullptr;
}
const SkIRect subset = SkIRect::MakeWH(info.width(), info.height());
return sk_make_sp<SkSpecialSurface_Raster>(info, std::move(pr), subset, props);
}
#if SK_SUPPORT_GPU
///////////////////////////////////////////////////////////////////////////////
#include "include/gpu/GrRecordingContext.h"
#include "src/gpu/GrRecordingContextPriv.h"
class SkSpecialSurface_Gpu : public SkSpecialSurface_Base {
public:
SkSpecialSurface_Gpu(sk_sp<skgpu::BaseDevice> device, SkIRect subset)
: INHERITED(subset, device->surfaceProps())
, fReadView(device->readSurfaceView()) {
fCanvas = std::make_unique<SkCanvas>(std::move(device));
fCanvas->clipRect(SkRect::Make(subset));
#ifdef SK_IS_BOT
fCanvas->clear(SK_ColorRED); // catch any imageFilter sloppiness
#endif
}
sk_sp<SkSpecialImage> onMakeImageSnapshot() override {
if (!fReadView.asTextureProxy()) {
return nullptr;
}
GrColorType ct = SkColorTypeToGrColorType(fCanvas->imageInfo().colorType());
// Note: SkSpecialImages can only be snapShotted once, so this call is destructive and we
// move fReadMove.
return SkSpecialImage::MakeDeferredFromGpu(fCanvas->recordingContext(),
this->subset(),
kNeedNewImageUniqueID_SpecialImage,
std::move(fReadView), ct,
fCanvas->imageInfo().refColorSpace(),
this->props());
}
private:
GrSurfaceProxyView fReadView;
using INHERITED = SkSpecialSurface_Base;
};
sk_sp<SkSpecialSurface> SkSpecialSurface::MakeRenderTarget(GrRecordingContext* rContext,
const SkImageInfo& ii,
const SkSurfaceProps& props,
GrSurfaceOrigin surfaceOrigin) {
if (!rContext) {
return nullptr;
}
auto device = rContext->priv().createDevice(SkBudgeted::kYes, ii, SkBackingFit::kApprox, 1,
GrMipmapped::kNo, GrProtected::kNo,
surfaceOrigin, props,
skgpu::BaseDevice::InitContents::kUninit);
if (!device) {
return nullptr;
}
const SkIRect subset = SkIRect::MakeSize(ii.dimensions());
return sk_make_sp<SkSpecialSurface_Gpu>(std::move(device), subset);
}
#endif