15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/dragdrop/drag_utils.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <objidl.h> 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <shlobj.h> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <shobjidl.h> 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/win/scoped_comptr.h" 12ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch#include "base/win/scoped_hdc.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "third_party/skia/include/core/SkBitmap.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/dragdrop/os_exchange_data.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/dragdrop/os_exchange_data_provider_win.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/canvas.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/gdi_util.h" 18f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles)#include "ui/gfx/geometry/size.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/image/image_skia.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/skbitmap_operations.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace drag_utils { 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static void SetDragImageOnDataObject(HBITMAP hbitmap, 25f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) const gfx::Size& size_in_pixels, 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const gfx::Vector2d& cursor_offset, 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IDataObject* data_object) { 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::win::ScopedComPtr<IDragSourceHelper> helper; 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HRESULT rv = CoCreateInstance(CLSID_DragDropHelper, 0, CLSCTX_INPROC_SERVER, 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) IID_IDragSourceHelper, helper.ReceiveVoid()); 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (SUCCEEDED(rv)) { 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SHDRAGIMAGE sdi; 33f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) sdi.sizeDragImage = size_in_pixels.ToSIZE(); 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sdi.crColorKey = 0xFFFFFFFF; 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sdi.hbmpDragImage = hbitmap; 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) sdi.ptOffset = gfx::PointAtOffsetFromOrigin(cursor_offset).ToPOINT(); 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) helper->InitializeFromBitmap(&sdi, data_object); 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Blit the contents of the canvas to a new HBITMAP. It is the caller's 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// responsibility to release the |bits| buffer. 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)static HBITMAP CreateHBITMAPFromSkBitmap(const SkBitmap& sk_bitmap) { 44ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch base::win::ScopedGetDC screen_dc(NULL); 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BITMAPINFOHEADER header; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) gfx::CreateBitmapHeader(sk_bitmap.width(), sk_bitmap.height(), &header); 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void* bits; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HBITMAP bitmap = 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CreateDIBSection(screen_dc, reinterpret_cast<BITMAPINFO*>(&header), 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DIB_RGB_COLORS, &bits, NULL, 0); 51ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch if (!bitmap || !bits) 52ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch return NULL; 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(sk_bitmap.rowBytes(), static_cast<size_t>(sk_bitmap.width() * 4)); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkAutoLockPixels lock(sk_bitmap); 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) memcpy( 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bits, sk_bitmap.getPixels(), sk_bitmap.height() * sk_bitmap.rowBytes()); 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return bitmap; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void SetDragImageOnDataObject(const gfx::ImageSkia& image_skia, 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const gfx::Vector2d& cursor_offset, 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::OSExchangeData* data_object) { 63f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) DCHECK(data_object && !image_skia.size().IsEmpty()); 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // InitializeFromBitmap() doesn't expect an alpha channel and is confused 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // by premultiplied colors, so unpremultiply the bitmap. 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // SetDragImageOnDataObject(HBITMAP) takes ownership of the bitmap. 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) HBITMAP bitmap = CreateHBITMAPFromSkBitmap( 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkBitmapOperations::UnPreMultiply(*image_skia.bitmap())); 69ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch if (bitmap) { 70ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch // Attach 'bitmap' to the data_object. 71f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) SetDragImageOnDataObject( 72f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) bitmap, 73f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) gfx::Size(image_skia.bitmap()->width(), image_skia.bitmap()->height()), 74f8ee788a64d60abd8f2d742a5fdedde054ecd910Torne (Richard Coles) cursor_offset, 75ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch ui::OSExchangeDataProviderWin::GetIDataObject(*data_object)); 76ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch } 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // TODO: the above code is used in non-Ash, while below is used in Ash. If we 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // could figure this context out then we wouldn't do unnecessary work. However 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // as it stands getting this information in ui/base would be a layering 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // violation. 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) data_object->provider().SetDragImage(image_skia, cursor_offset); 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace drag_utils 86