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 using base::ASCIIToUTF16;
  15
  16 namespace views {
  17 namespace examples {
  18
  19 TreeViewExample::TreeViewExample()
  20     : ExampleBase("Tree View"),
  21       tree_view_(NULL),
  22       model_(new NodeType(ASCIIToUTF16("root"), 1)) {
  23 }
  24
  25 TreeViewExample::~TreeViewExample() {
  26   // Delete the view before the model.
  27   delete tree_view_;
  28   tree_view_ = NULL;
  29 }
  30
  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
  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
  90 bool TreeViewExample::IsCommandIdEnabled(int command_id) {
  91   return command_id != ID_REMOVE ||
  92       tree_view_->GetSelectedNode() != model_.GetRoot();
  93 }
  94
  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
 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
 122 bool TreeViewExample::CanEdit(TreeView* tree_view,
 123                               ui::TreeModelNode* node) {
 124   return true;
 125 }
 126
 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
 145 bool TreeViewExample::IsCommandIdChecked(int command_id) const {
 146   return false;
 147 }
 148
 149 bool TreeViewExample::IsCommandIdEnabled(int command_id) const {
 150   return const_cast<TreeViewExample*>(this)->IsCommandIdEnabled(command_id);
 151 }
 152
 153 bool TreeViewExample::GetAcceleratorForCommandId(
 154     int command_id,
 155     ui::Accelerator* accelerator) {
 156   return false;
 157 }
 158
 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