Line data Source code
1 : /*
2 : * Copyright 2016 Google Inc.
3 : *
4 : * Use of this source code is governed by a BSD-style license that can be
5 : * found in the LICENSE file
6 : */
7 :
8 : #include "SkCanvas.h"
9 : #include "SkSpecialImage.h"
10 : #include "SkSpecialSurface.h"
11 : #include "SkSurfacePriv.h"
12 :
13 : ///////////////////////////////////////////////////////////////////////////////
14 : class SkSpecialSurface_Base : public SkSpecialSurface {
15 : public:
16 0 : SkSpecialSurface_Base(const SkIRect& subset, const SkSurfaceProps* props)
17 0 : : INHERITED(subset, props)
18 0 : , fCanvas(nullptr) {
19 0 : }
20 :
21 0 : virtual ~SkSpecialSurface_Base() { }
22 :
23 : // reset is called after an SkSpecialImage has been snapped
24 0 : void reset() { fCanvas.reset(); }
25 :
26 : // This can return nullptr if reset has already been called or something when wrong in the ctor
27 0 : SkCanvas* onGetCanvas() { return fCanvas.get(); }
28 :
29 : virtual sk_sp<SkSpecialImage> onMakeImageSnapshot() = 0;
30 :
31 : protected:
32 : std::unique_ptr<SkCanvas> fCanvas; // initialized by derived classes in ctors
33 :
34 : private:
35 : typedef SkSpecialSurface INHERITED;
36 : };
37 :
38 : ///////////////////////////////////////////////////////////////////////////////
39 0 : static SkSpecialSurface_Base* as_SB(SkSpecialSurface* surface) {
40 0 : return static_cast<SkSpecialSurface_Base*>(surface);
41 : }
42 :
43 0 : SkSpecialSurface::SkSpecialSurface(const SkIRect& subset,
44 0 : const SkSurfaceProps* props)
45 0 : : fProps(SkSurfacePropsCopyOrDefault(props).flags(), kUnknown_SkPixelGeometry)
46 0 : , fSubset(subset) {
47 0 : SkASSERT(fSubset.width() > 0);
48 0 : SkASSERT(fSubset.height() > 0);
49 0 : }
50 :
51 0 : SkCanvas* SkSpecialSurface::getCanvas() {
52 0 : return as_SB(this)->onGetCanvas();
53 : }
54 :
55 0 : sk_sp<SkSpecialImage> SkSpecialSurface::makeImageSnapshot() {
56 0 : sk_sp<SkSpecialImage> image(as_SB(this)->onMakeImageSnapshot());
57 0 : as_SB(this)->reset();
58 0 : return image; // the caller gets the creation ref
59 : }
60 :
61 : ///////////////////////////////////////////////////////////////////////////////
62 : #include "SkMallocPixelRef.h"
63 :
64 : class SkSpecialSurface_Raster : public SkSpecialSurface_Base {
65 : public:
66 0 : SkSpecialSurface_Raster(sk_sp<SkPixelRef> pr,
67 : const SkIRect& subset,
68 : const SkSurfaceProps* props)
69 0 : : INHERITED(subset, props) {
70 0 : const SkImageInfo& info = pr->info();
71 :
72 0 : fBitmap.setInfo(info, info.minRowBytes());
73 0 : fBitmap.setPixelRef(std::move(pr), 0, 0);
74 :
75 0 : fCanvas.reset(new SkCanvas(fBitmap, this->props()));
76 0 : fCanvas->clipRect(SkRect::Make(subset));
77 : #ifdef SK_IS_BOT
78 : fCanvas->clear(SK_ColorRED); // catch any imageFilter sloppiness
79 : #endif
80 0 : }
81 :
82 0 : ~SkSpecialSurface_Raster() override { }
83 :
84 0 : sk_sp<SkSpecialImage> onMakeImageSnapshot() override {
85 0 : return SkSpecialImage::MakeFromRaster(this->subset(), fBitmap, &this->props());
86 : }
87 :
88 : private:
89 : SkBitmap fBitmap;
90 :
91 : typedef SkSpecialSurface_Base INHERITED;
92 : };
93 :
94 0 : sk_sp<SkSpecialSurface> SkSpecialSurface::MakeFromBitmap(const SkIRect& subset, SkBitmap& bm,
95 : const SkSurfaceProps* props) {
96 0 : return sk_make_sp<SkSpecialSurface_Raster>(sk_ref_sp(bm.pixelRef()), subset, props);
97 : }
98 :
99 0 : sk_sp<SkSpecialSurface> SkSpecialSurface::MakeRaster(const SkImageInfo& info,
100 : const SkSurfaceProps* props) {
101 0 : sk_sp<SkPixelRef> pr = SkMallocPixelRef::MakeZeroed(info, 0, nullptr);
102 0 : if (!pr) {
103 0 : return nullptr;
104 : }
105 :
106 0 : const SkIRect subset = SkIRect::MakeWH(pr->info().width(), pr->info().height());
107 :
108 0 : return sk_make_sp<SkSpecialSurface_Raster>(std::move(pr), subset, props);
109 : }
110 :
111 : #if SK_SUPPORT_GPU
112 : ///////////////////////////////////////////////////////////////////////////////
113 : #include "GrContext.h"
114 : #include "SkGpuDevice.h"
115 :
116 : class SkSpecialSurface_Gpu : public SkSpecialSurface_Base {
117 : public:
118 0 : SkSpecialSurface_Gpu(GrContext* context, sk_sp<GrRenderTargetContext> renderTargetContext,
119 : int width, int height, const SkIRect& subset)
120 0 : : INHERITED(subset, &renderTargetContext->surfaceProps())
121 0 : , fRenderTargetContext(std::move(renderTargetContext)) {
122 :
123 0 : sk_sp<SkBaseDevice> device(SkGpuDevice::Make(context, fRenderTargetContext, width, height,
124 0 : SkGpuDevice::kUninit_InitContents));
125 0 : if (!device) {
126 0 : return;
127 : }
128 :
129 0 : fCanvas.reset(new SkCanvas(device.get()));
130 0 : fCanvas->clipRect(SkRect::Make(subset));
131 : #ifdef SK_IS_BOT
132 : fCanvas->clear(SK_ColorRED); // catch any imageFilter sloppiness
133 : #endif
134 : }
135 :
136 0 : ~SkSpecialSurface_Gpu() override { }
137 :
138 0 : sk_sp<SkSpecialImage> onMakeImageSnapshot() override {
139 0 : if (!fRenderTargetContext->asTextureProxy()) {
140 0 : return nullptr;
141 : }
142 : sk_sp<SkSpecialImage> tmp(SkSpecialImage::MakeDeferredFromGpu(
143 : fCanvas->getGrContext(),
144 : this->subset(),
145 : kNeedNewImageUniqueID_SpecialImage,
146 0 : fRenderTargetContext->asTextureProxyRef(),
147 0 : fRenderTargetContext->refColorSpace(),
148 0 : &this->props()));
149 0 : fRenderTargetContext = nullptr;
150 0 : return tmp;
151 : }
152 :
153 : private:
154 : sk_sp<GrRenderTargetContext> fRenderTargetContext;
155 :
156 : typedef SkSpecialSurface_Base INHERITED;
157 : };
158 :
159 0 : sk_sp<SkSpecialSurface> SkSpecialSurface::MakeRenderTarget(GrContext* context,
160 : int width, int height,
161 : GrPixelConfig config,
162 : sk_sp<SkColorSpace> colorSpace) {
163 0 : if (!context) {
164 0 : return nullptr;
165 : }
166 :
167 : sk_sp<GrRenderTargetContext> renderTargetContext(context->makeRenderTargetContext(
168 0 : SkBackingFit::kApprox, width, height, config, std::move(colorSpace)));
169 0 : if (!renderTargetContext) {
170 0 : return nullptr;
171 : }
172 :
173 0 : const SkIRect subset = SkIRect::MakeWH(width, height);
174 :
175 0 : return sk_make_sp<SkSpecialSurface_Gpu>(context, std::move(renderTargetContext),
176 0 : width, height, subset);
177 : }
178 :
179 : #endif
|