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/examples/tree_view_example.h"
6
7 #include "base/strings/utf_string_conversions.h"
8 #include "ui/views/controls/button/label_button.h"
9 #include "ui/views/controls/menu/menu_model_adapter.h"
10 #include "ui/views/controls/menu/menu_runner.h"
11 #include "ui/views/controls/tree/tree_view.h"
12 #include "ui/views/layout/grid_layout.h"
13
14 using base::ASCIIToUTF16;
15
16 namespace views {
17 namespace examples {
18
TreeViewExample()19 TreeViewExample::TreeViewExample()
20 : ExampleBase("Tree View"),
21 tree_view_(NULL),
22 model_(new NodeType(ASCIIToUTF16("root"), 1)) {
23 }
24
~TreeViewExample()25 TreeViewExample::~TreeViewExample() {
26 // Delete the view before the model.
27 delete tree_view_;
28 tree_view_ = NULL;
29 }
30
CreateExampleView(View * container)31 void TreeViewExample::CreateExampleView(View* container) {
32 // Add some sample data.
33 NodeType* colors_node = new NodeType(ASCIIToUTF16("colors"), 1);
34 model_.GetRoot()->Add(colors_node, 0);
35 colors_node->Add(new NodeType(ASCIIToUTF16("red"), 1), 0);
36 colors_node->Add(new NodeType(ASCIIToUTF16("green"), 1), 1);
37 colors_node->Add(new NodeType(ASCIIToUTF16("blue"), 1), 2);
38
39 NodeType* sheep_node = new NodeType(ASCIIToUTF16("sheep"), 1);
40 model_.GetRoot()->Add(sheep_node, 0);
41 sheep_node->Add(new NodeType(ASCIIToUTF16("Sheep 1"), 1), 0);
42 sheep_node->Add(new NodeType(ASCIIToUTF16("Sheep 2"), 1), 1);
43
44 tree_view_ = new TreeView();
45 tree_view_->set_context_menu_controller(this);
46 tree_view_->SetRootShown(false);
47 tree_view_->SetModel(&model_);
48 tree_view_->SetController(this);
49 add_ = new LabelButton(this, ASCIIToUTF16("Add"));
50 add_->SetFocusable(true);
51 remove_ = new LabelButton(this, ASCIIToUTF16("Remove"));
52 remove_->SetFocusable(true);
53 change_title_ = new LabelButton(this, ASCIIToUTF16("Change Title"));
54 change_title_->SetFocusable(true);
55
56 GridLayout* layout = new GridLayout(container);
57 container->SetLayoutManager(layout);
58
59 const int tree_view_column = 0;
60 ColumnSet* column_set = layout->AddColumnSet(tree_view_column);
61 column_set->AddColumn(GridLayout::FILL, GridLayout::FILL,
62 1.0f, GridLayout::USE_PREF, 0, 0);
63 layout->StartRow(1 /* expand */, tree_view_column);
64 layout->AddView(tree_view_->CreateParentIfNecessary());
65
66 // Add control buttons horizontally.
67 const int button_column = 1;
68 column_set = layout->AddColumnSet(button_column);
69 for (int i = 0; i < 3; i++) {
70 column_set->AddColumn(GridLayout::FILL, GridLayout::FILL,
71 1.0f, GridLayout::USE_PREF, 0, 0);
72 }
73
74 layout->StartRow(0 /* no expand */, button_column);
75 layout->AddView(add_);
76 layout->AddView(remove_);
77 layout->AddView(change_title_);
78 }
79
AddNewNode()80 void TreeViewExample::AddNewNode() {
81 NodeType* selected_node =
82 static_cast<NodeType*>(tree_view_->GetSelectedNode());
83 if (!selected_node)
84 selected_node = model_.GetRoot();
85 NodeType* new_node = new NodeType(selected_node->GetTitle(), 1);
86 model_.Add(selected_node, new_node, selected_node->child_count());
87 tree_view_->SetSelectedNode(new_node);
88 }
89
IsCommandIdEnabled(int command_id)90 bool TreeViewExample::IsCommandIdEnabled(int command_id) {
91 return command_id != ID_REMOVE ||
92 tree_view_->GetSelectedNode() != model_.GetRoot();
93 }
94
ButtonPressed(Button * sender,const ui::Event & event)95 void TreeViewExample::ButtonPressed(Button* sender, const ui::Event& event) {
96 NodeType* selected_node =
97 static_cast<NodeType*>(tree_view_->GetSelectedNode());
98 if (sender == add_) {
99 AddNewNode();
100 } else if (sender == remove_) {
101 DCHECK(selected_node);
102 DCHECK_NE(model_.GetRoot(), selected_node);
103 model_.Remove(selected_node->parent(), selected_node);
104 } else if (sender == change_title_) {
105 DCHECK(selected_node);
106 model_.SetTitle(selected_node,
107 selected_node->GetTitle() + ASCIIToUTF16("new"));
108 }
109 }
110
OnTreeViewSelectionChanged(TreeView * tree_view)111 void TreeViewExample::OnTreeViewSelectionChanged(TreeView* tree_view) {
112 ui::TreeModelNode* node = tree_view_->GetSelectedNode();
113 if (node) {
114 change_title_->SetEnabled(true);
115 remove_->SetEnabled(node != model_.GetRoot());
116 } else {
117 change_title_->SetEnabled(false);
118 remove_->SetEnabled(false);
119 }
120 }
121
CanEdit(TreeView * tree_view,ui::TreeModelNode * node)122 bool TreeViewExample::CanEdit(TreeView* tree_view,
123 ui::TreeModelNode* node) {
124 return true;
125 }
126
ShowContextMenuForView(View * source,const gfx::Point & point,ui::MenuSourceType source_type)127 void TreeViewExample::ShowContextMenuForView(View* source,
128 const gfx::Point& point,
129 ui::MenuSourceType source_type) {
130 ui::SimpleMenuModel context_menu_model(this);
131 context_menu_model.AddItem(ID_EDIT, ASCIIToUTF16("Edit"));
132 context_menu_model.AddItem(ID_REMOVE, ASCIIToUTF16("Remove"));
133 context_menu_model.AddItem(ID_ADD, ASCIIToUTF16("Add"));
134 context_menu_runner_.reset(new MenuRunner(&context_menu_model, 0));
135 if (context_menu_runner_->RunMenuAt(source->GetWidget(),
136 NULL,
137 gfx::Rect(point, gfx::Size()),
138 MENU_ANCHOR_TOPLEFT,
139 source_type) ==
140 MenuRunner::MENU_DELETED) {
141 return;
142 }
143 }
144
IsCommandIdChecked(int command_id) const145 bool TreeViewExample::IsCommandIdChecked(int command_id) const {
146 return false;
147 }
148
IsCommandIdEnabled(int command_id) const149 bool TreeViewExample::IsCommandIdEnabled(int command_id) const {
150 return const_cast<TreeViewExample*>(this)->IsCommandIdEnabled(command_id);
151 }
152
GetAcceleratorForCommandId(int command_id,ui::Accelerator * accelerator)153 bool TreeViewExample::GetAcceleratorForCommandId(
154 int command_id,
155 ui::Accelerator* accelerator) {
156 return false;
157 }
158
ExecuteCommand(int command_id,int event_flags)159 void TreeViewExample::ExecuteCommand(int command_id, int event_flags) {
160 NodeType* selected_node =
161 static_cast<NodeType*>(tree_view_->GetSelectedNode());
162 switch (command_id) {
163 case ID_EDIT:
164 tree_view_->StartEditing(selected_node);
165 break;
166 case ID_REMOVE:
167 model_.Remove(selected_node->parent(), selected_node);
168 break;
169 case ID_ADD:
170 AddNewNode();
171 break;
172 default:
173 NOTREACHED();
174 }
175 }
176
177 } // namespace examples
178 } // namespace views
179