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 "chrome/browser/ui/gtk/tabs/drag_data.h"
6
7 #include "chrome/browser/ui/gtk/tabs/tab_gtk.h"
8 #include "chrome/browser/ui/tabs/tab_strip_model.h"
9 #include "content/public/browser/web_contents.h"
10 #include "content/public/browser/web_contents_delegate.h"
11
12 using content::WebContents;
13
DraggedTabData()14 DraggedTabData::DraggedTabData()
15 : tab_(NULL),
16 contents_(NULL),
17 original_delegate_(NULL),
18 source_model_index_(-1),
19 pinned_(false),
20 mini_(false) {
21 }
22
DraggedTabData(TabGtk * tab,WebContents * contents,content::WebContentsDelegate * original_delegate,int source_model_index,bool pinned,bool mini)23 DraggedTabData::DraggedTabData(TabGtk* tab,
24 WebContents* contents,
25 content::WebContentsDelegate* original_delegate,
26 int source_model_index,
27 bool pinned,
28 bool mini)
29 : tab_(tab),
30 contents_(contents),
31 original_delegate_(original_delegate),
32 source_model_index_(source_model_index),
33 pinned_(pinned),
34 mini_(mini) {
35 }
36
~DraggedTabData()37 DraggedTabData::~DraggedTabData() {
38 }
39
ResetDelegate()40 void DraggedTabData::ResetDelegate() {
41 contents_->SetDelegate(original_delegate_);
42 }
43
DragData(std::vector<DraggedTabData> drag_data,int source_tab_index)44 DragData::DragData(std::vector<DraggedTabData> drag_data, int source_tab_index)
45 : drag_data_(drag_data),
46 source_tab_index_(source_tab_index),
47 non_mini_tab_count_(0),
48 mini_tab_count_(0) {
49 GetNumberOfMiniNonMiniTabs(0, drag_data_.size(),
50 &mini_tab_count_, &non_mini_tab_count_);
51 }
52
~DragData()53 DragData::~DragData() {
54 }
55
GetDraggedTabs() const56 std::vector<TabGtk*> DragData::GetDraggedTabs() const {
57 std::vector<TabGtk*> tabs;
58 for (size_t i = 0; i < drag_data_.size(); ++i) {
59 if (drag_data_[i].tab_)
60 tabs.push_back(drag_data_[i].tab_);
61 }
62 return tabs;
63 }
64
GetDraggedTabsContents() const65 std::vector<WebContents*> DragData::GetDraggedTabsContents() const {
66 std::vector<WebContents*> web_contentses;
67 for (size_t i = 0; i < drag_data_.size(); ++i) {
68 if (drag_data_[i].contents_)
69 web_contentses.push_back(drag_data_[i].contents_);
70 }
71 return web_contentses;
72 }
73
GetNumberOfMiniNonMiniTabs(int from,int to,int * mini,int * non_mini) const74 void DragData::GetNumberOfMiniNonMiniTabs(
75 int from, int to, int* mini, int* non_mini) const {
76 DCHECK(to <= static_cast<int>(drag_data_.size()));
77
78 *mini = 0;
79 *non_mini = 0;
80 for (int i = from; i < to; ++i) {
81 if (drag_data_[i].mini_)
82 (*mini)++;
83 else
84 (*non_mini)++;
85 }
86 }
87
GetAddTypesForDraggedTabAt(size_t index)88 int DragData::GetAddTypesForDraggedTabAt(size_t index) {
89 int add_types = TabStripModel::ADD_NONE;
90 if (get(index)->pinned_)
91 add_types |= TabStripModel::ADD_PINNED;
92 if (static_cast<int>(index) == source_tab_index_)
93 add_types |= TabStripModel::ADD_ACTIVE;
94 return add_types;
95 }
96
GetSourceWebContents()97 WebContents* DragData::GetSourceWebContents() {
98 return GetSourceTabData()->contents_;
99 }
100
GetSourceTabData()101 DraggedTabData* DragData::GetSourceTabData() {
102 return get(source_tab_index_);
103 }
104