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