src/ui/views/examples/tree_view_example.h

   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 #ifndef UI_VIEWS_EXAMPLES_TREE_VIEW_EXAMPLE_H_
   6 #define UI_VIEWS_EXAMPLES_TREE_VIEW_EXAMPLE_H_
   7
   8 #include "base/macros.h"
   9 #include "ui/base/models/simple_menu_model.h"
  10 #include "ui/base/models/tree_node_model.h"
  11 #include "ui/views/context_menu_controller.h"
  12 #include "ui/views/controls/button/button.h"
  13 #include "ui/views/controls/tree/tree_view_controller.h"
  14 #include "ui/views/examples/example_base.h"
  15
  16 namespace views {
  17
  18 class MenuRunner;
  19 class TreeView;
  20
  21 namespace examples {
  22
  23 class VIEWS_EXAMPLES_EXPORT TreeViewExample
  24     : public ExampleBase,
  25       public ButtonListener,
  26       public TreeViewController,
  27       public ContextMenuController,
  28       public ui::SimpleMenuModel::Delegate {
  29  public:
  30   TreeViewExample();
  31   ~TreeViewExample() override;
  32
  33   // ExampleBase:
  34   void CreateExampleView(View* container) override;
  35
  36  private:
  37   // IDs used by the context menu.
  38   enum MenuIDs {
  39     ID_EDIT,
  40     ID_REMOVE,
  41     ID_ADD
  42   };
  43
  44   // Adds a new node.
  45   void AddNewNode();
  46
  47   // Non-const version of IsCommandIdEnabled.
  48   bool IsCommandIdEnabled(int command_id);
  49
  50   // ButtonListener:
  51   void ButtonPressed(Button* sender, const ui::Event& event) override;
  52
  53   // TreeViewController:
  54   void OnTreeViewSelectionChanged(TreeView* tree_view) override;
  55   bool CanEdit(TreeView* tree_view, ui::TreeModelNode* node) override;
  56
  57   // ContextMenuController:
  58   void ShowContextMenuForView(View* source,
  59                               const gfx::Point& point,
  60                               ui::MenuSourceType source_type) override;
  61
  62   // SimpleMenuModel::Delegate:
  63   bool IsCommandIdChecked(int command_id) const override;
  64   bool IsCommandIdEnabled(int command_id) const override;
  65   bool GetAcceleratorForCommandId(int command_id,
  66                                   ui::Accelerator* accelerator) override;
  67   void ExecuteCommand(int command_id, int event_flags) override;
  68
  69   // The tree view to be tested.
  70   TreeView* tree_view_;
  71
  72   // Control buttons to modify the model.
  73   Button* add_;
  74   Button* remove_;
  75   Button* change_title_;
  76
  77   typedef ui::TreeNodeWithValue<int> NodeType;
  78
  79   ui::TreeNodeModel<NodeType> model_;
  80
  81   scoped_ptr<MenuRunner> context_menu_runner_;
  82
  83   DISALLOW_COPY_AND_ASSIGN(TreeViewExample);
  84 };
  85
  86 }  // namespace examples
  87 }  // namespace views
  88
  89 #endif  // UI_VIEWS_EXAMPLES_TREE_VIEW_EXAMPLE_H_