src/ui/views/examples/tree_view_example.cc

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