1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "ui/views/button_drag_utils.h"
6
7#include "base/strings/utf_string_conversions.h"
8#include "grit/ui_resources.h"
9#include "ui/base/dragdrop/drag_utils.h"
10#include "ui/base/dragdrop/os_exchange_data.h"
11#include "ui/base/resource/resource_bundle.h"
12#include "ui/gfx/canvas.h"
13#include "ui/gfx/geometry/point.h"
14#include "ui/gfx/image/image.h"
15#include "ui/views/controls/button/text_button.h"
16#include "ui/views/drag_utils.h"
17#include "url/gurl.h"
18
19namespace button_drag_utils {
20
21// Maximum width of the link drag image in pixels.
22static const int kLinkDragImageMaxWidth = 150;
23
24void SetURLAndDragImage(const GURL& url,
25                        const base::string16& title,
26                        const gfx::ImageSkia& icon,
27                        const gfx::Point* press_pt,
28                        ui::OSExchangeData* data,
29                        views::Widget* widget) {
30  DCHECK(url.is_valid() && data);
31  data->SetURL(url, title);
32  SetDragImage(url, title, icon, press_pt, data, widget);
33}
34
35void SetDragImage(const GURL& url,
36                  const base::string16& title,
37                  const gfx::ImageSkia& icon,
38                  const gfx::Point* press_pt,
39                  ui::OSExchangeData* data,
40                  views::Widget* widget) {
41  // Create a button to render the drag image for us.
42  views::TextButton button(NULL,
43                           title.empty() ? base::UTF8ToUTF16(url.spec())
44                                         : title);
45  button.set_max_width(kLinkDragImageMaxWidth);
46  if (icon.isNull()) {
47    button.SetIcon(*ui::ResourceBundle::GetSharedInstance().GetImageNamed(
48                   IDR_DEFAULT_FAVICON).ToImageSkia());
49  } else {
50    button.SetIcon(icon);
51  }
52  gfx::Size prefsize = button.GetPreferredSize();
53  button.SetBounds(0, 0, prefsize.width(), prefsize.height());
54
55  gfx::Vector2d press_point;
56  if (press_pt)
57    press_point = press_pt->OffsetFromOrigin();
58  else
59    press_point = gfx::Vector2d(prefsize.width() / 2, prefsize.height() / 2);
60
61  // Render the image.
62  scoped_ptr<gfx::Canvas> canvas(
63      views::GetCanvasForDragImage(widget, prefsize));
64  button.PaintButton(canvas.get(), views::TextButton::PB_FOR_DRAG);
65  drag_utils::SetDragImageOnDataObject(*canvas, press_point, data);
66}
67
68}  // namespace button_drag_utils
69