Chromium Code Reviews| Index: ui/base/dragdrop/drag_utils_win.cc |
| diff --git a/ui/base/dragdrop/drag_utils_win.cc b/ui/base/dragdrop/drag_utils_win.cc |
| index fa482e2e026284b56e1edc4761cc0e5781c482b5..53c864a62432602a9b652e3ddc9400dfbf3ac2a1 100644 |
| --- a/ui/base/dragdrop/drag_utils_win.cc |
| +++ b/ui/base/dragdrop/drag_utils_win.cc |
| @@ -23,65 +23,55 @@ |
| namespace drag_utils { |
| -static void SetDragImageOnDataObject(HBITMAP hbitmap, |
|
danakj
2017/04/03 22:34:01
Since these have one caller, I collapsed them all
|
| - const gfx::Size& size_in_pixels, |
| - const gfx::Vector2d& cursor_offset, |
| - IDataObject* data_object) { |
| - base::win::ScopedComPtr<IDragSourceHelper> helper; |
| - HRESULT rv = CoCreateInstance(CLSID_DragDropHelper, 0, CLSCTX_INPROC_SERVER, |
| - IID_IDragSourceHelper, helper.ReceiveVoid()); |
| - if (SUCCEEDED(rv)) { |
| - SHDRAGIMAGE sdi; |
| - sdi.sizeDragImage = size_in_pixels.ToSIZE(); |
| - sdi.crColorKey = 0xFFFFFFFF; |
| - sdi.hbmpDragImage = hbitmap; |
| - sdi.ptOffset = gfx::PointAtOffsetFromOrigin(cursor_offset).ToPOINT(); |
| - helper->InitializeFromBitmap(&sdi, data_object); |
| - } |
| -} |
| - |
| -// Blit the contents of the canvas to a new HBITMAP. It is the caller's |
|
danakj
2017/04/03 22:34:01
I read MSDN for a while and I think this comment i
|
| -// responsibility to release the |bits| buffer. |
| -static HBITMAP CreateHBITMAPFromSkBitmap(const SkBitmap& sk_bitmap) { |
| - base::win::ScopedGetDC screen_dc(NULL); |
| - BITMAPINFOHEADER header; |
| - skia::CreateBitmapHeader(sk_bitmap.width(), sk_bitmap.height(), &header); |
| - void* bits; |
| - HBITMAP bitmap = |
| - CreateDIBSection(screen_dc, reinterpret_cast<BITMAPINFO*>(&header), |
| - DIB_RGB_COLORS, &bits, NULL, 0); |
| - if (!bitmap || !bits) |
| - return NULL; |
| - DCHECK_EQ(sk_bitmap.rowBytes(), static_cast<size_t>(sk_bitmap.width() * 4)); |
| - SkAutoLockPixels lock(sk_bitmap); |
| - memcpy( |
| - bits, sk_bitmap.getPixels(), sk_bitmap.height() * sk_bitmap.rowBytes()); |
| - return bitmap; |
| -} |
| - |
| void SetDragImageOnDataObject(const gfx::ImageSkia& image_skia, |
| const gfx::Vector2d& cursor_offset, |
| ui::OSExchangeData* data_object) { |
| DCHECK(data_object && !image_skia.size().IsEmpty()); |
| + |
| // InitializeFromBitmap() doesn't expect an alpha channel and is confused |
| // by premultiplied colors, so unpremultiply the bitmap. |
| // SetDragImageOnDataObject(HBITMAP) takes ownership of the bitmap. |
| - HBITMAP bitmap = CreateHBITMAPFromSkBitmap( |
| - SkBitmapOperations::UnPreMultiply(*image_skia.bitmap())); |
| - if (bitmap) { |
| - // Attach 'bitmap' to the data_object. |
| - SetDragImageOnDataObject( |
| - bitmap, |
| - gfx::Size(image_skia.bitmap()->width(), image_skia.bitmap()->height()), |
| - cursor_offset, |
| - ui::OSExchangeDataProviderWin::GetIDataObject(*data_object)); |
| + SkBitmap unpremul_bitmap = |
| + SkBitmapOperations::UnPreMultiply(*image_skia.bitmap()); |
| + int width = unpremul_bitmap.width(); |
| + int height = unpremul_bitmap.height(); |
| + size_t rowbytes = unpremul_bitmap.rowBytes(); |
| + DCHECK_EQ(rowbytes, static_cast<size_t>(width) * 4u); |
| + |
| + void* bits; |
| + HBITMAP hbitmap; |
| + { |
| + BITMAPINFOHEADER header; |
| + skia::CreateBitmapHeader(width, height, &header); |
| + |
| + base::win::ScopedGetDC screen_dc(NULL); |
|
danakj
2017/04/03 22:34:01
I scoped this DC to just be around CreateDIBSectio
Peter Kasting
2017/04/04 00:31:14
It makes me a little leery. I suggest just not pu
danakj
2017/04/04 15:25:33
I like to scope things as tightly as possible, if
|
| + // 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, NULL, 0); |
| + } |
| + if (!hbitmap) |
| + return; |
| + |
| + { |
| + SkAutoLockPixels lock(unpremul_bitmap); |
| + memcpy(bits, unpremul_bitmap.getPixels(), height * rowbytes); |
| } |
| - // TODO: the above code is used in non-Ash, while below is used in Ash. If we |
| - // could figure this context out then we wouldn't do unnecessary work. However |
| - // as it stands getting this information in ui/base would be a layering |
| - // violation. |
| - data_object->provider().SetDragImage(image_skia, cursor_offset); |
| + base::win::ScopedComPtr<IDragSourceHelper> helper; |
| + HRESULT rv = CoCreateInstance(CLSID_DragDropHelper, 0, CLSCTX_INPROC_SERVER, |
| + IID_IDragSourceHelper, helper.ReceiveVoid()); |
| + if (!SUCCEEDED(rv)) |
| + return; |
| + |
| + SHDRAGIMAGE sdi; |
| + sdi.sizeDragImage.cx = width; |
| + sdi.sizeDragImage.cy = height; |
| + sdi.crColorKey = 0xFFFFFFFF; |
| + sdi.hbmpDragImage = hbitmap; |
| + sdi.ptOffset = gfx::PointAtOffsetFromOrigin(cursor_offset).ToPOINT(); |
| + helper->InitializeFromBitmap(&sdi, data_object); |
| } |
| } // namespace drag_utils |