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 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 8 | #include "include/core/SkCanvas.h" |
| 9 | #include "include/effects/SkGradientShader.h" |
| 10 | #include "samplecode/Sample.h" |
| 11 | #include "src/core/SkMakeUnique.h" |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 12 | |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 13 | static sk_sp<SkShader> make_grad(SkScalar w, SkScalar h) { |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 14 | SkColor colors[] = { 0xFF000000, 0xFF333333 }; |
| 15 | SkPoint pts[] = { { 0, 0 }, { w, h } }; |
Mike Reed | fae8fce | 2019-04-03 10:27:45 -0400 | [diff] [blame] | 16 | return SkGradientShader::MakeLinear(pts, colors, nullptr, 2, SkTileMode::kClamp); |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 17 | } |
| 18 | |
Ben Wagner | b2c4ea6 | 2018-08-08 11:36:17 -0400 | [diff] [blame] | 19 | class BigGradientView : public Sample { |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 20 | public: |
rmistry@google.com | ae933ce | 2012-08-23 18:19:56 +0000 | [diff] [blame] | 21 | BigGradientView() {} |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 22 | |
| 23 | protected: |
Ben Wagner | b2c4ea6 | 2018-08-08 11:36:17 -0400 | [diff] [blame] | 24 | bool onQuery(Sample::Event* evt) override { |
| 25 | if (Sample::TitleQ(*evt)) { |
| 26 | Sample::TitleR(evt, "BigGradient"); |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 27 | return true; |
| 28 | } |
| 29 | return this->INHERITED::onQuery(evt); |
| 30 | } |
| 31 | |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 32 | void onDrawContent(SkCanvas* canvas) override { |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 33 | SkRect r; |
| 34 | r.set(0, 0, this->width(), this->height()); |
| 35 | SkPaint p; |
reed | 8a21c9f | 2016-03-08 18:50:00 -0800 | [diff] [blame] | 36 | p.setShader(make_grad(this->width(), this->height())); |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 37 | canvas->drawRect(r, p); |
| 38 | } |
| 39 | |
| 40 | private: |
Ben Wagner | b2c4ea6 | 2018-08-08 11:36:17 -0400 | [diff] [blame] | 41 | typedef Sample INHERITED; |
reed@google.com | 55b8e8c | 2011-01-13 16:22:35 +0000 | [diff] [blame] | 42 | }; |
| 43 | |
| 44 | /////////////////////////////////////////////////////////////////////////////// |
| 45 | |
Ben Wagner | b2c4ea6 | 2018-08-08 11:36:17 -0400 | [diff] [blame] | 46 | DEF_SAMPLE( return new BigGradientView(); ) |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 47 | |
| 48 | /////////////////////////////////////////////////////////////////////////////// |
| 49 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 50 | #include "include/core/SkRasterHandleAllocator.h" |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 51 | |
| 52 | class GraphicsPort { |
| 53 | protected: |
| 54 | SkCanvas* fCanvas; |
| 55 | |
| 56 | public: |
| 57 | GraphicsPort(SkCanvas* canvas) : fCanvas(canvas) {} |
| 58 | virtual ~GraphicsPort() {} |
| 59 | |
| 60 | void save() { fCanvas->save(); } |
| 61 | void saveLayer(const SkRect& bounds, SkAlpha alpha) { |
| 62 | fCanvas->saveLayerAlpha(&bounds, alpha); |
| 63 | } |
| 64 | void restore() { fCanvas->restore(); } |
| 65 | |
| 66 | void translate(float x, float y) { fCanvas->translate(x, y); } |
| 67 | void scale(float s) { fCanvas->scale(s, s); } |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 68 | void clip(const SkRect& r) { fCanvas->clipRect(r); } |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 69 | |
| 70 | void drawOval(const SkRect& r, SkColor c) { |
| 71 | SkPaint p; |
| 72 | p.setColor(c); |
| 73 | fCanvas->drawOval(r, p); |
| 74 | } |
| 75 | |
| 76 | virtual void drawRect(const SkRect& r, SkColor c) { |
| 77 | SkPaint p; |
| 78 | p.setColor(c); |
| 79 | fCanvas->drawRect(r, p); |
| 80 | } |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 81 | |
| 82 | SkCanvas* peekCanvas() const { return fCanvas; } |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 83 | }; |
| 84 | |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 85 | #ifdef SK_BUILD_FOR_MAC |
| 86 | |
Mike Klein | c0bd9f9 | 2019-04-23 12:05:21 -0500 | [diff] [blame] | 87 | #include "include/utils/mac/SkCGUtils.h" |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 88 | class CGGraphicsPort : public GraphicsPort { |
| 89 | public: |
| 90 | CGGraphicsPort(SkCanvas* canvas) : GraphicsPort(canvas) {} |
| 91 | |
| 92 | void drawRect(const SkRect& r, SkColor c) override { |
| 93 | CGContextRef cg = (CGContextRef)fCanvas->accessTopRasterHandle(); |
Ben Wagner | 63fd760 | 2017-10-09 15:45:33 -0400 | [diff] [blame] | 94 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 95 | CGColorRef color = CGColorCreateGenericRGB(SkColorGetR(c)/255.f, |
| 96 | SkColorGetG(c)/255.f, |
| 97 | SkColorGetB(c)/255.f, |
| 98 | SkColorGetA(c)/255.f); |
| 99 | |
| 100 | CGContextSetFillColorWithColor(cg, color); |
| 101 | CGContextFillRect(cg, CGRectMake(r.x(), r.y(), r.width(), r.height())); |
| 102 | } |
| 103 | }; |
| 104 | |
| 105 | static CGAffineTransform matrix_to_transform(CGContextRef cg, const SkMatrix& ctm) { |
| 106 | SkMatrix matrix; |
| 107 | matrix.setScale(1, -1); |
| 108 | matrix.postTranslate(0, SkIntToScalar(CGBitmapContextGetHeight(cg))); |
| 109 | matrix.preConcat(ctm); |
| 110 | |
| 111 | return CGAffineTransformMake(matrix[SkMatrix::kMScaleX], |
| 112 | matrix[SkMatrix::kMSkewY], |
| 113 | matrix[SkMatrix::kMSkewX], |
| 114 | matrix[SkMatrix::kMScaleY], |
| 115 | matrix[SkMatrix::kMTransX], |
| 116 | matrix[SkMatrix::kMTransY]); |
| 117 | } |
| 118 | |
| 119 | class Allocator_CG : public SkRasterHandleAllocator { |
| 120 | public: |
| 121 | Allocator_CG() {} |
Ben Wagner | 63fd760 | 2017-10-09 15:45:33 -0400 | [diff] [blame] | 122 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 123 | bool allocHandle(const SkImageInfo& info, Rec* rec) override { |
| 124 | // let CG allocate the pixels |
| 125 | CGContextRef cg = SkCreateCGContext(SkPixmap(info, nullptr, 0)); |
| 126 | if (!cg) { |
| 127 | return false; |
| 128 | } |
| 129 | rec->fReleaseProc = [](void* pixels, void* ctx){ CGContextRelease((CGContextRef)ctx); }; |
| 130 | rec->fReleaseCtx = cg; |
| 131 | rec->fPixels = CGBitmapContextGetData(cg); |
| 132 | rec->fRowBytes = CGBitmapContextGetBytesPerRow(cg); |
| 133 | rec->fHandle = cg; |
| 134 | CGContextSaveGState(cg); // balanced each time updateContext is called |
| 135 | return true; |
| 136 | } |
| 137 | |
| 138 | void updateHandle(Handle hndl, const SkMatrix& ctm, const SkIRect& clip) override { |
| 139 | CGContextRef cg = (CGContextRef)hndl; |
Ben Wagner | 63fd760 | 2017-10-09 15:45:33 -0400 | [diff] [blame] | 140 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 141 | CGContextRestoreGState(cg); |
| 142 | CGContextSaveGState(cg); |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 143 | CGContextClipToRect(cg, CGRectMake(clip.x(), clip.y(), clip.width(), clip.height())); |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 144 | CGContextConcatCTM(cg, matrix_to_transform(cg, ctm)); |
| 145 | } |
| 146 | }; |
| 147 | |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 148 | #define MyPort CGGraphicsPort |
| 149 | #define MyAllocator Allocator_CG |
| 150 | |
| 151 | #elif defined(WIN32) |
| 152 | |
Mike Reed | 9e937af | 2017-01-12 14:18:39 -0500 | [diff] [blame] | 153 | static RECT toRECT(const SkIRect& r) { |
| 154 | return { r.left(), r.top(), r.right(), r.bottom() }; |
| 155 | } |
| 156 | |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 157 | class GDIGraphicsPort : public GraphicsPort { |
| 158 | public: |
| 159 | GDIGraphicsPort(SkCanvas* canvas) : GraphicsPort(canvas) {} |
| 160 | |
| 161 | void drawRect(const SkRect& r, SkColor c) override { |
| 162 | HDC hdc = (HDC)fCanvas->accessTopRasterHandle(); |
| 163 | |
| 164 | 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] | 165 | RECT rounded = toRECT(r.round()); |
| 166 | FillRect(hdc, &rounded, CreateSolidBrush(cr)); |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 167 | |
| 168 | // Assuming GDI wrote zeros for alpha, this will or-in 0xFF for alpha |
| 169 | SkPaint paint; |
| 170 | paint.setBlendMode(SkBlendMode::kDstATop); |
| 171 | fCanvas->drawRect(r, paint); |
| 172 | } |
| 173 | }; |
| 174 | |
| 175 | static void DeleteHDCCallback(void*, void* context) { |
| 176 | HDC hdc = static_cast<HDC>(context); |
| 177 | HBITMAP hbitmap = static_cast<HBITMAP>(SelectObject(hdc, nullptr)); |
| 178 | DeleteObject(hbitmap); |
| 179 | DeleteDC(hdc); |
| 180 | } |
| 181 | |
| 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); |
| 213 | SelectObject(hdc, hbitmap); |
| 214 | |
| 215 | rec->fReleaseProc = DeleteHDCCallback; |
| 216 | rec->fReleaseCtx = hdc; |
| 217 | rec->fPixels = pixels; |
| 218 | rec->fRowBytes = row_bytes; |
| 219 | rec->fHandle = hdc; |
| 220 | return true; |
| 221 | } |
| 222 | |
| 223 | /** |
| 224 | * Subclass of SkRasterHandleAllocator that returns an HDC as its "handle". |
| 225 | */ |
| 226 | class GDIAllocator : public SkRasterHandleAllocator { |
| 227 | public: |
| 228 | GDIAllocator() {} |
| 229 | |
| 230 | bool allocHandle(const SkImageInfo& info, Rec* rec) override { |
| 231 | SkASSERT(info.colorType() == kN32_SkColorType); |
| 232 | return Create(info.width(), info.height(), info.isOpaque(), rec); |
| 233 | } |
| 234 | |
| 235 | void updateHandle(Handle handle, const SkMatrix& ctm, const SkIRect& clip_bounds) override { |
| 236 | HDC hdc = static_cast<HDC>(handle); |
| 237 | |
| 238 | XFORM xf; |
| 239 | xf.eM11 = ctm[SkMatrix::kMScaleX]; |
| 240 | xf.eM21 = ctm[SkMatrix::kMSkewX]; |
| 241 | xf.eDx = ctm[SkMatrix::kMTransX]; |
| 242 | xf.eM12 = ctm[SkMatrix::kMSkewY]; |
| 243 | xf.eM22 = ctm[SkMatrix::kMScaleY]; |
| 244 | xf.eDy = ctm[SkMatrix::kMTransY]; |
| 245 | SetWorldTransform(hdc, &xf); |
| 246 | |
Mike Klein | c722f79 | 2017-07-31 11:57:21 -0400 | [diff] [blame] | 247 | RECT clip_bounds_RECT = toRECT(clip_bounds); |
| 248 | HRGN hrgn = CreateRectRgnIndirect(&clip_bounds_RECT); |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 249 | int result = SelectClipRgn(hdc, hrgn); |
| 250 | SkASSERT(result != ERROR); |
| 251 | result = DeleteObject(hrgn); |
| 252 | SkASSERT(result != 0); |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 253 | } |
| 254 | }; |
| 255 | |
| 256 | #define MyPort GDIGraphicsPort |
| 257 | #define MyAllocator GDIAllocator |
| 258 | |
| 259 | #endif |
| 260 | |
| 261 | #ifdef MyAllocator |
Ben Wagner | b2c4ea6 | 2018-08-08 11:36:17 -0400 | [diff] [blame] | 262 | class RasterAllocatorSample : public Sample { |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 263 | public: |
| 264 | RasterAllocatorSample() {} |
| 265 | |
| 266 | protected: |
Ben Wagner | b2c4ea6 | 2018-08-08 11:36:17 -0400 | [diff] [blame] | 267 | bool onQuery(Sample::Event* evt) override { |
| 268 | if (Sample::TitleQ(*evt)) { |
| 269 | Sample::TitleR(evt, "raster-allocator"); |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 270 | return true; |
| 271 | } |
| 272 | return this->INHERITED::onQuery(evt); |
| 273 | } |
| 274 | |
| 275 | void doDraw(GraphicsPort* port) { |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 276 | SkAutoCanvasRestore acr(port->peekCanvas(), true); |
| 277 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 278 | port->drawRect({0, 0, 256, 256}, SK_ColorRED); |
| 279 | port->save(); |
| 280 | port->translate(30, 30); |
| 281 | port->drawRect({0, 0, 30, 30}, SK_ColorBLUE); |
| 282 | port->drawOval({10, 10, 20, 20}, SK_ColorWHITE); |
| 283 | port->restore(); |
Ben Wagner | 63fd760 | 2017-10-09 15:45:33 -0400 | [diff] [blame] | 284 | |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 285 | port->saveLayer({50, 50, 100, 100}, 0x80); |
| 286 | port->drawRect({55, 55, 95, 95}, SK_ColorGREEN); |
| 287 | port->restore(); |
Mike Reed | 108f55e | 2017-01-12 11:28:01 -0500 | [diff] [blame] | 288 | |
| 289 | port->clip({150, 50, 200, 200}); |
| 290 | port->drawRect({0, 0, 256, 256}, 0xFFCCCCCC); |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 291 | } |
| 292 | |
| 293 | void onDrawContent(SkCanvas* canvas) override { |
| 294 | GraphicsPort skp(canvas); |
| 295 | doDraw(&skp); |
| 296 | |
| 297 | const SkImageInfo info = SkImageInfo::MakeN32Premul(256, 256); |
| 298 | std::unique_ptr<SkCanvas> c2 = |
Mike Reed | e4ca362 | 2017-01-11 21:18:25 -0500 | [diff] [blame] | 299 | SkRasterHandleAllocator::MakeCanvas(skstd::make_unique<MyAllocator>(), info); |
| 300 | MyPort cgp(c2.get()); |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 301 | doDraw(&cgp); |
| 302 | |
| 303 | SkPixmap pm; |
| 304 | c2->peekPixels(&pm); |
| 305 | SkBitmap bm; |
| 306 | bm.installPixels(pm); |
| 307 | canvas->drawBitmap(bm, 280, 0, nullptr); |
| 308 | } |
| 309 | |
| 310 | private: |
Ben Wagner | b2c4ea6 | 2018-08-08 11:36:17 -0400 | [diff] [blame] | 311 | typedef Sample INHERITED; |
Mike Reed | 356f7c2 | 2017-01-10 11:58:39 -0500 | [diff] [blame] | 312 | }; |
| 313 | DEF_SAMPLE( return new RasterAllocatorSample; ) |
| 314 | #endif |