Mike Reed | fae8fce | 2019-04-03 10:27:45 -0400 | [diff] [blame] | 1 | /* |
epoger@google.com | ec3ed6a | 2011-07-28 14:26:00 +0000 | [diff] [blame] | 2 | * Copyright 2011 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 | */ |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 7 | |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 8 | #include "gm/gm.h" |
| 9 | #include "include/core/SkBitmap.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 10 | #include "include/core/SkCanvas.h" |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 11 | #include "include/core/SkPixmap.h" |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 12 | #include "include/core/SkRasterHandleAllocator.h" |
Ben Wagner | 9f02d2d | 2019-08-14 15:38:43 -0400 | [diff] [blame^] | 13 | #include "include/core/SkSurface.h" |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 14 | #include "src/core/SkMakeUnique.h" |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 15 | |
| 16 | class GraphicsPort { |
| 17 | protected: |
| 18 | SkCanvas* fCanvas; |
| 19 | |
| 20 | public: |
| 21 | GraphicsPort(SkCanvas* canvas) : fCanvas(canvas) {} |
| 22 | virtual ~GraphicsPort() {} |
| 23 | |
| 24 | void save() { fCanvas->save(); } |
| 25 | void saveLayer(const SkRect& bounds, SkAlpha alpha) { |
| 26 | fCanvas->saveLayerAlpha(&bounds, alpha); |
| 27 | } |
| 28 | void restore() { fCanvas->restore(); } |
| 29 | |
| 30 | void translate(float x, float y) { fCanvas->translate(x, y); } |
| 31 | void scale(float s) { fCanvas->scale(s, s); } |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 32 | void clip(const SkRect& r) { fCanvas->clipRect(r); } |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 33 | |
| 34 | void drawOval(const SkRect& r, SkColor c) { |
| 35 | SkPaint p; |
| 36 | p.setColor(c); |
| 37 | fCanvas->drawOval(r, p); |
| 38 | } |
| 39 | |
| 40 | virtual void drawRect(const SkRect& r, SkColor c) { |
| 41 | SkPaint p; |
| 42 | p.setColor(c); |
| 43 | fCanvas->drawRect(r, p); |
| 44 | } |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 45 | |
| 46 | SkCanvas* peekCanvas() const { return fCanvas; } |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 47 | }; |
| 48 | |
Ben Wagner | 9f02d2d | 2019-08-14 15:38:43 -0400 | [diff] [blame^] | 49 | class SkiaGraphicsPort : public GraphicsPort { |
| 50 | public: |
| 51 | SkiaGraphicsPort(SkCanvas* canvas) : GraphicsPort(canvas) {} |
| 52 | |
| 53 | void drawRect(const SkRect& r, SkColor c) override { |
| 54 | SkCanvas* canvas = (SkCanvas*)fCanvas->accessTopRasterHandle(); |
| 55 | canvas->drawRect(r, SkPaint(SkColor4f::FromColor(c))); |
| 56 | } |
| 57 | }; |
| 58 | |
| 59 | class SkiaAllocator : public SkRasterHandleAllocator { |
| 60 | public: |
| 61 | SkiaAllocator() {} |
| 62 | |
| 63 | bool allocHandle(const SkImageInfo& info, Rec* rec) override { |
| 64 | sk_sp<SkSurface> surface = SkSurface::MakeRaster(info); |
| 65 | if (!surface) { |
| 66 | return false; |
| 67 | } |
| 68 | SkCanvas* canvas = surface->getCanvas(); |
| 69 | SkPixmap pixmap; |
| 70 | canvas->peekPixels(&pixmap); |
| 71 | |
| 72 | rec->fReleaseProc = [](void* pixels, void* ctx){ SkSafeUnref((SkSurface*)ctx); }; |
| 73 | rec->fReleaseCtx = surface.release(); |
| 74 | rec->fPixels = pixmap.writable_addr(); |
| 75 | rec->fRowBytes = pixmap.rowBytes(); |
| 76 | rec->fHandle = canvas; |
| 77 | canvas->save(); // balanced each time updateHandle is called |
| 78 | return true; |
| 79 | } |
| 80 | |
| 81 | void updateHandle(Handle hndl, const SkMatrix& ctm, const SkIRect& clip) override { |
| 82 | SkCanvas* canvas = (SkCanvas*)hndl; |
| 83 | canvas->restore(); |
| 84 | canvas->save(); |
| 85 | canvas->clipRect(SkRect::Make(clip)); |
| 86 | canvas->concat(ctm); |
| 87 | } |
| 88 | }; |
| 89 | |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 90 | #ifdef SK_BUILD_FOR_MAC |
| 91 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 92 | #include "include/utils/mac/SkCGUtils.h" |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 93 | class CGGraphicsPort : public GraphicsPort { |
| 94 | public: |
| 95 | CGGraphicsPort(SkCanvas* canvas) : GraphicsPort(canvas) {} |
| 96 | |
| 97 | void drawRect(const SkRect& r, SkColor c) override { |
| 98 | CGContextRef cg = (CGContextRef)fCanvas->accessTopRasterHandle(); |
Ben Wagner | 63fd760 | 2017-10-09 15:45:33 -0400 | [diff] [blame] | 99 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 100 | CGColorRef color = CGColorCreateGenericRGB(SkColorGetR(c)/255.f, |
| 101 | SkColorGetG(c)/255.f, |
| 102 | SkColorGetB(c)/255.f, |
| 103 | SkColorGetA(c)/255.f); |
| 104 | |
| 105 | CGContextSetFillColorWithColor(cg, color); |
| 106 | CGContextFillRect(cg, CGRectMake(r.x(), r.y(), r.width(), r.height())); |
| 107 | } |
| 108 | }; |
| 109 | |
| 110 | static CGAffineTransform matrix_to_transform(CGContextRef cg, const SkMatrix& ctm) { |
| 111 | SkMatrix matrix; |
| 112 | matrix.setScale(1, -1); |
| 113 | matrix.postTranslate(0, SkIntToScalar(CGBitmapContextGetHeight(cg))); |
| 114 | matrix.preConcat(ctm); |
| 115 | |
| 116 | return CGAffineTransformMake(matrix[SkMatrix::kMScaleX], |
| 117 | matrix[SkMatrix::kMSkewY], |
| 118 | matrix[SkMatrix::kMSkewX], |
| 119 | matrix[SkMatrix::kMScaleY], |
| 120 | matrix[SkMatrix::kMTransX], |
| 121 | matrix[SkMatrix::kMTransY]); |
| 122 | } |
| 123 | |
Ben Wagner | 9f02d2d | 2019-08-14 15:38:43 -0400 | [diff] [blame^] | 124 | class CGAllocator : public SkRasterHandleAllocator { |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 125 | public: |
Ben Wagner | 9f02d2d | 2019-08-14 15:38:43 -0400 | [diff] [blame^] | 126 | CGAllocator() {} |
Ben Wagner | 63fd760 | 2017-10-09 15:45:33 -0400 | [diff] [blame] | 127 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 128 | bool allocHandle(const SkImageInfo& info, Rec* rec) override { |
| 129 | // let CG allocate the pixels |
| 130 | CGContextRef cg = SkCreateCGContext(SkPixmap(info, nullptr, 0)); |
| 131 | if (!cg) { |
| 132 | return false; |
| 133 | } |
| 134 | rec->fReleaseProc = [](void* pixels, void* ctx){ CGContextRelease((CGContextRef)ctx); }; |
| 135 | rec->fReleaseCtx = cg; |
| 136 | rec->fPixels = CGBitmapContextGetData(cg); |
| 137 | rec->fRowBytes = CGBitmapContextGetBytesPerRow(cg); |
| 138 | rec->fHandle = cg; |
Ben Wagner | 9f02d2d | 2019-08-14 15:38:43 -0400 | [diff] [blame^] | 139 | CGContextSaveGState(cg); // balanced each time updateHandle is called |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 140 | return true; |
| 141 | } |
| 142 | |
| 143 | void updateHandle(Handle hndl, const SkMatrix& ctm, const SkIRect& clip) override { |
| 144 | CGContextRef cg = (CGContextRef)hndl; |
Ben Wagner | 63fd760 | 2017-10-09 15:45:33 -0400 | [diff] [blame] | 145 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 146 | CGContextRestoreGState(cg); |
| 147 | CGContextSaveGState(cg); |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 148 | CGContextClipToRect(cg, CGRectMake(clip.x(), clip.y(), clip.width(), clip.height())); |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 149 | CGContextConcatCTM(cg, matrix_to_transform(cg, ctm)); |
| 150 | } |
| 151 | }; |
| 152 | |
Ben Wagner | 9f02d2d | 2019-08-14 15:38:43 -0400 | [diff] [blame^] | 153 | using MyPort = CGGraphicsPort; |
| 154 | using MyAllocator = CGAllocator; |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 155 | |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 156 | #elif defined(SK_BUILD_FOR_WIN) |
| 157 | |
| 158 | #include "src/core/SkLeanWindows.h" |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 159 | |
Mike Reed | 9e937af | 2017-01-12 14:18:39 -0500 | [diff] [blame] | 160 | static RECT toRECT(const SkIRect& r) { |
| 161 | return { r.left(), r.top(), r.right(), r.bottom() }; |
| 162 | } |
| 163 | |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 164 | class GDIGraphicsPort : public GraphicsPort { |
| 165 | public: |
| 166 | GDIGraphicsPort(SkCanvas* canvas) : GraphicsPort(canvas) {} |
| 167 | |
| 168 | void drawRect(const SkRect& r, SkColor c) override { |
| 169 | HDC hdc = (HDC)fCanvas->accessTopRasterHandle(); |
| 170 | |
| 171 | COLORREF cr = RGB(SkColorGetR(c), SkColorGetG(c), SkColorGetB(c));// SkEndian_Swap32(c) >> 8; |
Mike Klein | c722f79 | 2017-07-31 11:57:21 -0400 | [diff] [blame] | 172 | RECT rounded = toRECT(r.round()); |
| 173 | FillRect(hdc, &rounded, CreateSolidBrush(cr)); |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 174 | |
| 175 | // Assuming GDI wrote zeros for alpha, this will or-in 0xFF for alpha |
| 176 | SkPaint paint; |
| 177 | paint.setBlendMode(SkBlendMode::kDstATop); |
| 178 | fCanvas->drawRect(r, paint); |
| 179 | } |
| 180 | }; |
| 181 | |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 182 | // We use this static factory function instead of the regular constructor so |
| 183 | // that we can create the pixel data before calling the constructor. This is |
| 184 | // required so that we can call the base class' constructor with the pixel |
| 185 | // data. |
| 186 | static bool Create(int width, int height, bool is_opaque, SkRasterHandleAllocator::Rec* rec) { |
Chris Dalton | 1ef8094 | 2017-12-04 12:01:30 -0700 | [diff] [blame] | 187 | BITMAPINFOHEADER hdr; |
| 188 | memset(&hdr, 0, sizeof(hdr)); |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 189 | hdr.biSize = sizeof(BITMAPINFOHEADER); |
| 190 | hdr.biWidth = width; |
| 191 | hdr.biHeight = -height; // Minus means top-down bitmap. |
| 192 | hdr.biPlanes = 1; |
| 193 | hdr.biBitCount = 32; |
| 194 | hdr.biCompression = BI_RGB; // No compression. |
| 195 | hdr.biSizeImage = 0; |
| 196 | hdr.biXPelsPerMeter = 1; |
| 197 | hdr.biYPelsPerMeter = 1; |
| 198 | void* pixels; |
| 199 | HBITMAP hbitmap = CreateDIBSection(nullptr, (const BITMAPINFO*)&hdr, 0, &pixels, 0, 0); |
| 200 | if (!hbitmap) { |
| 201 | return false; |
| 202 | } |
| 203 | |
| 204 | size_t row_bytes = width * sizeof(SkPMColor); |
| 205 | sk_bzero(pixels, row_bytes * height); |
| 206 | |
| 207 | HDC hdc = CreateCompatibleDC(nullptr); |
| 208 | if (!hdc) { |
| 209 | DeleteObject(hbitmap); |
| 210 | return false; |
| 211 | } |
| 212 | SetGraphicsMode(hdc, GM_ADVANCED); |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 213 | HGDIOBJ origBitmap = SelectObject(hdc, hbitmap); |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 214 | |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 215 | struct ReleaseContext { |
| 216 | HDC hdc; |
| 217 | HGDIOBJ hbitmap; |
| 218 | }; |
| 219 | rec->fReleaseProc = [](void*, void* context) { |
| 220 | ReleaseContext* ctx = static_cast<ReleaseContext*>(context); |
| 221 | HBITMAP hbitmap = static_cast<HBITMAP>(SelectObject(ctx->hdc, ctx->hbitmap)); |
| 222 | DeleteObject(hbitmap); |
| 223 | DeleteDC(ctx->hdc); |
| 224 | delete ctx; |
| 225 | }; |
| 226 | rec->fReleaseCtx = new ReleaseContext{hdc, origBitmap}; |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 227 | rec->fPixels = pixels; |
| 228 | rec->fRowBytes = row_bytes; |
| 229 | rec->fHandle = hdc; |
| 230 | return true; |
| 231 | } |
| 232 | |
| 233 | /** |
| 234 | * Subclass of SkRasterHandleAllocator that returns an HDC as its "handle". |
| 235 | */ |
| 236 | class GDIAllocator : public SkRasterHandleAllocator { |
| 237 | public: |
| 238 | GDIAllocator() {} |
| 239 | |
| 240 | bool allocHandle(const SkImageInfo& info, Rec* rec) override { |
| 241 | SkASSERT(info.colorType() == kN32_SkColorType); |
| 242 | return Create(info.width(), info.height(), info.isOpaque(), rec); |
| 243 | } |
| 244 | |
| 245 | void updateHandle(Handle handle, const SkMatrix& ctm, const SkIRect& clip_bounds) override { |
| 246 | HDC hdc = static_cast<HDC>(handle); |
| 247 | |
| 248 | XFORM xf; |
| 249 | xf.eM11 = ctm[SkMatrix::kMScaleX]; |
| 250 | xf.eM21 = ctm[SkMatrix::kMSkewX]; |
| 251 | xf.eDx = ctm[SkMatrix::kMTransX]; |
| 252 | xf.eM12 = ctm[SkMatrix::kMSkewY]; |
| 253 | xf.eM22 = ctm[SkMatrix::kMScaleY]; |
| 254 | xf.eDy = ctm[SkMatrix::kMTransY]; |
| 255 | SetWorldTransform(hdc, &xf); |
| 256 | |
Mike Klein | c722f79 | 2017-07-31 11:57:21 -0400 | [diff] [blame] | 257 | RECT clip_bounds_RECT = toRECT(clip_bounds); |
| 258 | HRGN hrgn = CreateRectRgnIndirect(&clip_bounds_RECT); |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 259 | int result = SelectClipRgn(hdc, hrgn); |
| 260 | SkASSERT(result != ERROR); |
| 261 | result = DeleteObject(hrgn); |
| 262 | SkASSERT(result != 0); |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 263 | } |
| 264 | }; |
| 265 | |
Ben Wagner | 9f02d2d | 2019-08-14 15:38:43 -0400 | [diff] [blame^] | 266 | using MyPort = GDIGraphicsPort; |
| 267 | using MyAllocator = GDIAllocator; |
| 268 | |
| 269 | #else |
| 270 | |
| 271 | using MyPort = SkiaGraphicsPort; |
| 272 | using MyAllocator = SkiaAllocator; |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 273 | |
| 274 | #endif |
| 275 | |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 276 | DEF_SIMPLE_GM(rasterallocator, canvas, 600, 300) { |
| 277 | auto doDraw = [](GraphicsPort* port) { |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 278 | SkAutoCanvasRestore acr(port->peekCanvas(), true); |
| 279 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 280 | port->drawRect({0, 0, 256, 256}, SK_ColorRED); |
| 281 | port->save(); |
| 282 | port->translate(30, 30); |
| 283 | port->drawRect({0, 0, 30, 30}, SK_ColorBLUE); |
| 284 | port->drawOval({10, 10, 20, 20}, SK_ColorWHITE); |
| 285 | port->restore(); |
Ben Wagner | 63fd760 | 2017-10-09 15:45:33 -0400 | [diff] [blame] | 286 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 287 | port->saveLayer({50, 50, 100, 100}, 0x80); |
| 288 | port->drawRect({55, 55, 95, 95}, SK_ColorGREEN); |
| 289 | port->restore(); |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 290 | |
| 291 | port->clip({150, 50, 200, 200}); |
| 292 | port->drawRect({0, 0, 256, 256}, 0xFFCCCCCC); |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 293 | }; |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 294 | |
Ben Wagner | 53ae936 | 2019-06-07 14:16:54 -0400 | [diff] [blame] | 295 | // TODO: this common code fails pic-8888 and serialize-8888 |
Ben Wagner | 53ae936 | 2019-06-07 14:16:54 -0400 | [diff] [blame] | 296 | //GraphicsPort skiaPort(canvas); |
| 297 | //doDraw(&skiaPort); |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 298 | |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 299 | const SkImageInfo info = SkImageInfo::MakeN32Premul(256, 256); |
Ben Wagner | 53ae936 | 2019-06-07 14:16:54 -0400 | [diff] [blame] | 300 | std::unique_ptr<SkCanvas> nativeCanvas = |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 301 | SkRasterHandleAllocator::MakeCanvas(skstd::make_unique<MyAllocator>(), info); |
Ben Wagner | 53ae936 | 2019-06-07 14:16:54 -0400 | [diff] [blame] | 302 | MyPort nativePort(nativeCanvas.get()); |
| 303 | doDraw(&nativePort); |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 304 | |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 305 | SkPixmap pm; |
Ben Wagner | 53ae936 | 2019-06-07 14:16:54 -0400 | [diff] [blame] | 306 | nativeCanvas->peekPixels(&pm); |
Ben Wagner | 2b4dcd3 | 2019-06-06 14:39:52 -0400 | [diff] [blame] | 307 | SkBitmap bm; |
| 308 | bm.installPixels(pm); |
| 309 | canvas->drawBitmap(bm, 280, 0, nullptr); |
Ben Wagner | 53ae936 | 2019-06-07 14:16:54 -0400 | [diff] [blame] | 310 | } |