123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393 |
- // Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
- // Use of this source code is governed by a BSD-style license that can be
- // found in the LICENSE file.
- #include "skia/ext/skia_utils_win.h"
- #include <stddef.h>
- #include <windows.h>
- #include "base/check_op.h"
- #include "base/debug/gdi_debug_util_win.h"
- #include "base/numerics/checked_math.h"
- #include "base/win/scoped_hdc.h"
- #include "base/win/scoped_hglobal.h"
- #include "skia/ext/legacy_display_globals.h"
- #include "skia/ext/skia_utils_base.h"
- #include "third_party/skia/include/core/SkColorSpace.h"
- #include "third_party/skia/include/core/SkRect.h"
- #include "third_party/skia/include/core/SkSurface.h"
- #include "third_party/skia/include/core/SkTypes.h"
- namespace {
- static_assert(offsetof(RECT, left) == offsetof(SkIRect, fLeft), "o1");
- static_assert(offsetof(RECT, top) == offsetof(SkIRect, fTop), "o2");
- static_assert(offsetof(RECT, right) == offsetof(SkIRect, fRight), "o3");
- static_assert(offsetof(RECT, bottom) == offsetof(SkIRect, fBottom), "o4");
- static_assert(sizeof(RECT().left) == sizeof(SkIRect().fLeft), "o5");
- static_assert(sizeof(RECT().top) == sizeof(SkIRect().fTop), "o6");
- static_assert(sizeof(RECT().right) == sizeof(SkIRect().fRight), "o7");
- static_assert(sizeof(RECT().bottom) == sizeof(SkIRect().fBottom), "o8");
- static_assert(sizeof(RECT) == sizeof(SkIRect), "o9");
- void CreateBitmapHeaderWithColorDepth(LONG width,
- LONG height,
- WORD color_depth,
- BITMAPINFOHEADER* hdr) {
- // These values are shared with gfx::PlatformDevice.
- hdr->biSize = sizeof(BITMAPINFOHEADER);
- hdr->biWidth = width;
- hdr->biHeight = -height; // Minus means top-down bitmap.
- hdr->biPlanes = 1;
- hdr->biBitCount = color_depth;
- hdr->biCompression = BI_RGB; // No compression.
- hdr->biSizeImage = 0;
- hdr->biXPelsPerMeter = 1;
- hdr->biYPelsPerMeter = 1;
- hdr->biClrUsed = 0;
- hdr->biClrImportant = 0;
- }
- // Fills in a BITMAPV5HEADER structure. This is to be used for images that have
- // an alpha channel and are in the ARGB8888 format. This is because DIBV5 has an
- // explicit mask for each component which default to XRGB and we manually set
- // flag so the alpha channel is the first byte. This is not supported by the
- // older-style BITMAPINFOHEADER.
- void CreateBitmapV5HeaderForARGB8888(LONG width,
- LONG height,
- LONG image_size,
- BITMAPV5HEADER* hdr) {
- memset(hdr, 0, sizeof(BITMAPV5HEADER));
- hdr->bV5Size = sizeof(BITMAPV5HEADER);
- hdr->bV5Width = width;
- // If height is positive this means that the image will be bottom-up.
- hdr->bV5Height = height;
- hdr->bV5Planes = 1;
- hdr->bV5BitCount = 32;
- hdr->bV5Compression = BI_RGB;
- hdr->bV5AlphaMask = 0xff000000;
- hdr->bV5CSType = LCS_WINDOWS_COLOR_SPACE;
- hdr->bV5Intent = LCS_GM_IMAGES;
- hdr->bV5ClrUsed = 0;
- hdr->bV5ClrImportant = 0;
- hdr->bV5ProfileData = 0;
- }
- } // namespace
- namespace skia {
- POINT SkPointToPOINT(const SkPoint& point) {
- POINT win_point = {
- SkScalarRoundToInt(point.fX), SkScalarRoundToInt(point.fY)
- };
- return win_point;
- }
- SkRect RECTToSkRect(const RECT& rect) {
- SkRect sk_rect = { SkIntToScalar(rect.left), SkIntToScalar(rect.top),
- SkIntToScalar(rect.right), SkIntToScalar(rect.bottom) };
- return sk_rect;
- }
- SkColor COLORREFToSkColor(COLORREF color) {
- #ifndef _MSC_VER
- return SkColorSetRGB(GetRValue(color), GetGValue(color), GetBValue(color));
- #else
- // ARGB = 0xFF000000 | ((0BGR -> RGB0) >> 8)
- return 0xFF000000u | (_byteswap_ulong(color) >> 8);
- #endif
- }
- COLORREF SkColorToCOLORREF(SkColor color) {
- #ifndef _MSC_VER
- return RGB(SkColorGetR(color), SkColorGetG(color), SkColorGetB(color));
- #else
- // 0BGR = ((ARGB -> BGRA) >> 8)
- return (_byteswap_ulong(color) >> 8);
- #endif
- }
- void InitializeDC(HDC context) {
- // Enables world transformation.
- // If the GM_ADVANCED graphics mode is set, GDI always draws arcs in the
- // counterclockwise direction in logical space. This is equivalent to the
- // statement that, in the GM_ADVANCED graphics mode, both arc control points
- // and arcs themselves fully respect the device context's world-to-device
- // transformation.
- BOOL res = SetGraphicsMode(context, GM_ADVANCED);
- SkASSERT(res != 0);
- // Enables dithering.
- res = SetStretchBltMode(context, HALFTONE);
- SkASSERT(res != 0);
- // As per SetStretchBltMode() documentation, SetBrushOrgEx() must be called
- // right after.
- res = SetBrushOrgEx(context, 0, 0, NULL);
- SkASSERT(res != 0);
- // Sets up default orientation.
- res = SetArcDirection(context, AD_CLOCKWISE);
- SkASSERT(res != 0);
- // Sets up default colors.
- res = SetBkColor(context, RGB(255, 255, 255));
- SkASSERT(res != CLR_INVALID);
- res = SetTextColor(context, RGB(0, 0, 0));
- SkASSERT(res != CLR_INVALID);
- res = SetDCBrushColor(context, RGB(255, 255, 255));
- SkASSERT(res != CLR_INVALID);
- res = SetDCPenColor(context, RGB(0, 0, 0));
- SkASSERT(res != CLR_INVALID);
- // Sets up default transparency.
- res = SetBkMode(context, OPAQUE);
- SkASSERT(res != 0);
- res = SetROP2(context, R2_COPYPEN);
- SkASSERT(res != 0);
- }
- void LoadTransformToDC(HDC dc, const SkMatrix& matrix) {
- XFORM xf;
- xf.eM11 = matrix[SkMatrix::kMScaleX];
- xf.eM21 = matrix[SkMatrix::kMSkewX];
- xf.eDx = matrix[SkMatrix::kMTransX];
- xf.eM12 = matrix[SkMatrix::kMSkewY];
- xf.eM22 = matrix[SkMatrix::kMScaleY];
- xf.eDy = matrix[SkMatrix::kMTransY];
- SetWorldTransform(dc, &xf);
- }
- void CopyHDC(HDC source, HDC destination, int x, int y, bool is_opaque,
- const RECT& src_rect, const SkMatrix& transform) {
- int copy_width = src_rect.right - src_rect.left;
- int copy_height = src_rect.bottom - src_rect.top;
- // We need to reset the translation for our bitmap or (0,0) won't be in the
- // upper left anymore
- SkMatrix identity;
- identity.reset();
- LoadTransformToDC(source, identity);
- if (is_opaque) {
- BitBlt(destination,
- x,
- y,
- copy_width,
- copy_height,
- source,
- src_rect.left,
- src_rect.top,
- SRCCOPY);
- } else {
- SkASSERT(copy_width != 0 && copy_height != 0);
- BLENDFUNCTION blend_function = {AC_SRC_OVER, 0, 255, AC_SRC_ALPHA};
- GdiAlphaBlend(destination,
- x,
- y,
- copy_width,
- copy_height,
- source,
- src_rect.left,
- src_rect.top,
- copy_width,
- copy_height,
- blend_function);
- }
- LoadTransformToDC(source, transform);
- }
- SkImageInfo PrepareAllocation(HDC context, BITMAP* backing) {
- HBITMAP backing_handle =
- static_cast<HBITMAP>(GetCurrentObject(context, OBJ_BITMAP));
- const size_t backing_size = sizeof *backing;
- return (GetObject(backing_handle, backing_size, backing) == backing_size)
- ? SkImageInfo::MakeN32Premul(backing->bmWidth, backing->bmHeight)
- : SkImageInfo();
- }
- sk_sp<SkSurface> MapPlatformSurface(HDC context) {
- BITMAP backing;
- const SkImageInfo size(PrepareAllocation(context, &backing));
- SkSurfaceProps props = skia::LegacyDisplayGlobals::GetSkSurfaceProps();
- return size.isEmpty()
- ? nullptr
- : SkSurface::MakeRasterDirect(size, backing.bmBits,
- backing.bmWidthBytes, &props);
- }
- SkBitmap MapPlatformBitmap(HDC context) {
- BITMAP backing;
- const SkImageInfo size(PrepareAllocation(context, &backing));
- SkBitmap bitmap;
- if (!size.isEmpty())
- bitmap.installPixels(size, backing.bmBits, size.minRowBytes());
- return bitmap;
- }
- void CreateBitmapHeaderForN32SkBitmap(const SkBitmap& bitmap,
- BITMAPINFOHEADER* hdr) {
- // Native HBITMAPs are XRGB-backed, and we expect SkBitmaps that we will use
- // with them to also be of the same format.
- CHECK_EQ(bitmap.colorType(), kN32_SkColorType);
- // The header will be for an RGB bitmap with 32 bits-per-pixel. The SkBitmap
- // data to go into the bitmap should be of the same size. If the SkBitmap
- // SkColorType is for a larger number of bits-per-pixel, copying the SkBitmap
- // into the HBITMAP for this header would cause a write out-of-bounds.
- CHECK_EQ(4, bitmap.info().bytesPerPixel());
- // The HBITMAP's bytes will always be tightly packed so we expect the SkBitmap
- // to be also. Row padding would mean the number of bytes in the SkBitmap and
- // in the HBITMAP for this header would be different, which can cause out-of-
- // bound reads or writes.
- CHECK_EQ(bitmap.rowBytes(), bitmap.width() * static_cast<size_t>(4));
- CreateBitmapHeaderWithColorDepth(bitmap.width(), bitmap.height(), 32, hdr);
- }
- HGLOBAL CreateHGlobalForByteArray(
- const std::vector<unsigned char>& byte_array) {
- HGLOBAL hglobal = ::GlobalAlloc(GHND, byte_array.size());
- if (!hglobal) {
- return nullptr;
- }
- base::win::ScopedHGlobal<uint8_t*> global_mem(hglobal);
- if (!global_mem.get()) {
- ::GlobalFree(hglobal);
- return nullptr;
- }
- memcpy(global_mem.get(), byte_array.data(), byte_array.size());
- return hglobal;
- }
- HGLOBAL CreateDIBV5ImageDataFromN32SkBitmap(const SkBitmap& bitmap) {
- // While DIBV5 support bit flags which would allow us to put channels in a any
- // order, we require an ARGB format because it is more convenient to use.
- CHECK_EQ(bitmap.colorType(), kN32_SkColorType);
- // The header will be for an ARGB bitmap with 32 bits-per-pixel. The SkBitmap
- // data to go into the bitmap should be of the same size. If the SkBitmap
- // SkColorType is for a larger number of bits-per-pixel, copying the SkBitmap
- // into the DIBV5ImageData for this header would cause a write out-of-bounds.
- CHECK_EQ(4, bitmap.info().bytesPerPixel());
- // The DIBV5ImageData bytes will always be tightly packed so we expect the
- // SkBitmap to be also. Row padding would mean the number of bytes in the
- // SkBitmap and in the DIBV5ImageData for this header would be different,
- // which can cause out-of- bound reads or writes.
- CHECK_EQ(bitmap.rowBytes(), bitmap.width() * static_cast<size_t>(4));
- int width = bitmap.width();
- int height = bitmap.height();
- size_t bytes;
- // Native DIBV5 bitmaps store 32-bit ARGB data, and the SkBitmap used with it
- // must also, as verified at the start of this function. A size_t type causes
- // a type change from int when multiplying.
- constexpr size_t bpp = 4;
- if (!base::CheckMul(height, base::CheckMul(width, bpp)).AssignIfValid(&bytes))
- return nullptr;
- HGLOBAL hglobal = ::GlobalAlloc(GHND, sizeof(BITMAPV5HEADER) + bytes);
- if (hglobal == nullptr)
- return nullptr;
- base::win::ScopedHGlobal<BITMAPV5HEADER*> header(hglobal);
- if (!header.get()) {
- ::GlobalFree(hglobal);
- return nullptr;
- }
- CreateBitmapV5HeaderForARGB8888(width, height, bytes, header.get());
- auto* dst_pixels =
- reinterpret_cast<uint8_t*>(header.get()) + sizeof(BITMAPV5HEADER);
- // CreateBitmapV5HeaderForARGB8888 creates a bitmap with a positive height as
- // stated in the image's header. Having a positive value implies that the
- // image is stored bottom-up. As skia uses the opposite, we have to flip
- // vertically so the image's content while copying in the DIBV5 data structure
- // to account for that. In theory, we could use a negative value to avoid the
- // flip, but not all programs treat a negative value properly.
- SkImageInfo infoSRGB = bitmap.info()
- .makeColorSpace(SkColorSpace::MakeSRGB())
- .makeWH(bitmap.width(), 1);
- const size_t row_bytes = bitmap.rowBytes();
- for (size_t line = 0; line < height; line++) {
- size_t flipped_line_index = height - 1 - line;
- auto* current_dst = dst_pixels + (row_bytes * flipped_line_index);
- bool success = bitmap.readPixels(infoSRGB, current_dst, row_bytes, 0, line);
- DCHECK(success);
- }
- return hglobal;
- }
- base::win::ScopedBitmap CreateHBitmapFromN32SkBitmap(const SkBitmap& bitmap) {
- BITMAPINFOHEADER header;
- CreateBitmapHeaderForN32SkBitmap(bitmap, &header);
- int width = bitmap.width();
- int height = bitmap.height();
- size_t bytes;
- // Native HBITMAPs store 32-bit RGB data, and the SkBitmap used with it must
- // also, as verified by CreateBitmapHeaderForN32SkBitmap(). A size_t type
- // causes a type change from int when multiplying.
- const size_t bpp = 4;
- if (!base::CheckMul(height, base::CheckMul(width, bpp)).AssignIfValid(&bytes))
- return {};
- void* bits;
- HBITMAP hbitmap;
- {
- base::win::ScopedGetDC screen_dc(nullptr);
- // By giving a null hSection, the |bits| will be destroyed when the
- // |hbitmap| is destroyed.
- hbitmap =
- CreateDIBSection(screen_dc, reinterpret_cast<BITMAPINFO*>(&header),
- DIB_RGB_COLORS, &bits, nullptr, 0);
- }
- if (hbitmap) {
- memcpy(bits, bitmap.getPixels(), bytes);
- } else {
- // If CreateDIBSection() failed, try to get some useful information out
- // before we crash for post-mortem analysis.
- base::debug::CollectGDIUsageAndDie(&header, nullptr);
- }
- return base::win::ScopedBitmap(hbitmap);
- }
- void CreateBitmapHeaderForXRGB888(int width,
- int height,
- BITMAPINFOHEADER* hdr) {
- CreateBitmapHeaderWithColorDepth(width, height, 32, hdr);
- }
- base::win::ScopedBitmap CreateHBitmapXRGB8888(int width,
- int height,
- HANDLE shared_section,
- void** data) {
- // CreateDIBSection fails to allocate anything if we try to create an empty
- // bitmap, so just create a minimal bitmap.
- if ((width == 0) || (height == 0)) {
- width = 1;
- height = 1;
- }
- BITMAPINFOHEADER hdr = {0};
- CreateBitmapHeaderWithColorDepth(width, height, 32, &hdr);
- HBITMAP hbitmap = CreateDIBSection(NULL, reinterpret_cast<BITMAPINFO*>(&hdr),
- 0, data, shared_section, 0);
- // If CreateDIBSection() failed, try to get some useful information out
- // before we crash for post-mortem analysis.
- if (!hbitmap)
- base::debug::CollectGDIUsageAndDie(&hdr, shared_section);
- return base::win::ScopedBitmap(hbitmap);
- }
- } // namespace skia
|