Selects last item in a menu when pressing 'up' initially

ui/views/controls/menu/menu_controller.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "ui/views/controls/menu/menu_controller.h"
 
 #include "base/i18n/case_conversion.h"
 #include "base/i18n/rtl.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/time/time.h"
@@ skipping 71 matching lines @@
 // Returns true if |menu| doesn't have a mnemonic and first character of the its
 // title is |key|.
 bool TitleMatchesMnemonic(MenuItemView* menu, base::char16 key) {
   if (menu->GetMnemonic())
     return false;
 
   base::string16 lower_title = base::i18n::ToLower(menu->title());
   return !lower_title.empty() && lower_title[0] == key;
 }
 
-}  // namespace
-
 // Returns the first descendant of |view| that is hot tracked.
 static CustomButton* GetFirstHotTrackedView(View* view) {
   if (!view)
     return NULL;
   CustomButton* button = CustomButton::AsCustomButton(view);
   if (button) {
     if (button->IsHotTracked())
       return button;
   }
 
@@ skipping 46 matching lines @@
     if (forward || index != -1) {
       View* next = GetFirstFocusableView(new_parent, index, forward);
       if (next)
         return next;
     }
     parent = new_parent;
   } while (parent != ancestor);
   return NULL;
 }
 
+}  // namespace
+
 // MenuScrollTask --------------------------------------------------------------
 
 // MenuScrollTask is used when the SubmenuView does not all fit on screen and
 // the mouse is over the scroll up/down buttons. MenuScrollTask schedules
 // itself with a RepeatingTimer. When Run is invoked MenuScrollTask scrolls
 // appropriately.
 
 class MenuController::MenuScrollTask {
  public:
   MenuScrollTask() : submenu_(NULL), is_scrolling_up_(false), start_y_(0) {
@@ skipping 851 matching lines @@
   item->GetWidget()->RunShellDrag(NULL, data, widget_loc, drag_ops,
       ui::DragDropTypes::DRAG_EVENT_SOURCE_MOUSE);
   did_initiate_drag_ = false;
 }
 
 bool MenuController::OnKeyDown(ui::KeyboardCode key_code) {
   DCHECK(blocking_run_);
 
   switch (key_code) {
     case ui::VKEY_UP:
-      IncrementSelection(-1);
+      IncrementSelection(INCREMENT_SELECTION_UP);
       break;
 
     case ui::VKEY_DOWN:
-      IncrementSelection(1);
+      IncrementSelection(INCREMENT_SELECTION_DOWN);
       break;
 
     // Handling of VK_RIGHT and VK_LEFT is different depending on the UI
     // layout.
     case ui::VKEY_RIGHT:
       if (base::i18n::IsRTL())
         CloseSubmenu();
       else
         OpenSubmenuChangeSelectionIfCan();
       break;
@@ skipping 843 matching lines @@
         pref.height() / 2 - y + y_old);
   }
   return gfx::Rect(x, y, pref.width(), pref.height());
 }
 
 // static
 int MenuController::MenuDepth(MenuItemView* item) {
   return item ? (MenuDepth(item->GetParentMenuItem()) + 1) : 0;
 }
 
-void MenuController::IncrementSelection(int delta) {
+void MenuController::IncrementSelection(
+    SelectionIncrementDirectionType direction) {
   MenuItemView* item = pending_state_.item;
   DCHECK(item);
   if (pending_state_.submenu_open && item->HasSubmenu() &&
       item->GetSubmenu()->IsShowing()) {
     // A menu is selected and open, but none of its children are selected,
     // select the first menu item that is visible and enabled.
     if (item->GetSubmenu()->GetMenuItemCount()) {
-      MenuItemView* to_select = FindFirstSelectableMenuItem(item);
+      MenuItemView* to_select = FindInitialSelectableMenuItem(item, direction);
       if (to_select)
         SetSelection(to_select, SELECTION_DEFAULT);
       return;
     }
   }
 
   if (item->has_children()) {
     CustomButton* button = GetFirstHotTrackedView(item);
     if (button) {
       button->SetHotTracked(false);
-      View* to_make_hot = GetNextFocusableView(item, button, delta == 1);
+      View* to_make_hot = GetNextFocusableView(
+          item, button, direction == INCREMENT_SELECTION_DOWN);
       CustomButton* button_hot = CustomButton::AsCustomButton(to_make_hot);
       if (button_hot) {
         button_hot->SetHotTracked(true);
         return;
       }
     } else {
-      View* to_make_hot = GetInitialFocusableView(item, delta == 1);
+      View* to_make_hot =
+          GetInitialFocusableView(item, direction == INCREMENT_SELECTION_DOWN);
       CustomButton* button_hot = CustomButton::AsCustomButton(to_make_hot);
       if (button_hot) {
         button_hot->SetHotTracked(true);
         return;
       }
     }
   }
 
   MenuItemView* parent = item->GetParentMenuItem();
   if (parent) {
     int parent_count = parent->GetSubmenu()->GetMenuItemCount();
     if (parent_count > 1) {
       for (int i = 0; i < parent_count; ++i) {
         if (parent->GetSubmenu()->GetMenuItemAt(i) == item) {
           MenuItemView* to_select =
-              FindNextSelectableMenuItem(parent, i, delta);
+              FindNextSelectableMenuItem(parent, i, direction);
           if (!to_select)
             break;
           SetSelection(to_select, SELECTION_DEFAULT);
-          View* to_make_hot = GetInitialFocusableView(to_select, delta == 1);
+          View* to_make_hot = GetInitialFocusableView(
+              to_select, direction == INCREMENT_SELECTION_DOWN);
           CustomButton* button_hot = CustomButton::AsCustomButton(to_make_hot);
           if (button_hot)
             button_hot->SetHotTracked(true);
           break;
         }
       }
     }
   }
 }
 
-MenuItemView* MenuController::FindFirstSelectableMenuItem(
-    MenuItemView* parent) {
-  MenuItemView* child = parent->GetSubmenu()->GetMenuItemAt(0);
-  if (!child->visible() || !child->enabled())
-    child = FindNextSelectableMenuItem(parent, 0, 1);
-  return child;
+MenuItemView* MenuController::FindInitialSelectableMenuItem(
+    MenuItemView* parent,
+    SelectionIncrementDirectionType direction) {
+  return FindNextSelectableMenuItem(
+      parent, direction == INCREMENT_SELECTION_DOWN ? -1 : 0, direction);
 }
 
-MenuItemView* MenuController::FindNextSelectableMenuItem(MenuItemView* parent,
-                                                         int index,
-                                                         int delta) {
-  int start_index = index;
+MenuItemView* MenuController::FindNextSelectableMenuItem(
+    MenuItemView* parent,
+    int index,
+    SelectionIncrementDirectionType direction) {
   int parent_count = parent->GetSubmenu()->GetMenuItemCount();
+  int stop_index = (index + parent_count) % parent_count;
+  bool include_all_items =
+      (index == -1 && direction == INCREMENT_SELECTION_DOWN) ||
+      (index == 0 && direction == INCREMENT_SELECTION_UP);
+  int delta = direction == INCREMENT_SELECTION_UP ? -1 : 1;
   // Loop through the menu items skipping any invisible menus. The loop stops
   // when we wrap or find a visible and enabled child.
   do {
     index = (index + delta + parent_count) % parent_count;
-    if (index == start_index)
+    if (index == stop_index && !include_all_items)
       return NULL;
     MenuItemView* child = parent->GetSubmenu()->GetMenuItemAt(index);
     if (child->visible() && child->enabled())
       return child;
-  } while (index != start_index);
+  } while (index != stop_index);
   return NULL;
 }
 
 void MenuController::OpenSubmenuChangeSelectionIfCan() {
   MenuItemView* item = pending_state_.item;
   if (!item->HasSubmenu() || !item->enabled())
     return;
   MenuItemView* to_select = NULL;
   if (item->GetSubmenu()->GetMenuItemCount() > 0)
-    to_select = FindFirstSelectableMenuItem(item);
+    to_select = FindInitialSelectableMenuItem(item, INCREMENT_SELECTION_DOWN);
   if (to_select) {
     SetSelection(to_select, SELECTION_UPDATE_IMMEDIATELY);
     return;
   }
   // No menu items, just show the sub-menu.
   SetSelection(item, SELECTION_OPEN_SUBMENU | SELECTION_UPDATE_IMMEDIATELY);
 }
 
 void MenuController::CloseSubmenu() {
   MenuItemView* item = state_.item;
@@ skipping 358 matching lines @@
   }
 }
 
 gfx::Screen* MenuController::GetScreen() {
   Widget* root = owner_ ? owner_->GetTopLevelWidget() : NULL;
   return root ? gfx::Screen::GetScreenFor(root->GetNativeView())
               : gfx::Screen::GetNativeScreen();
 }
 
 }  // namespace views