src/ui/views/controls/menu/menu_controller_aura.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/controls/menu/menu_controller.h"
   6
   7 #include "base/run_loop.h"
   8 #include "ui/aura/client/activation_change_observer.h"
   9 #include "ui/aura/client/activation_client.h"
  10 #include "ui/aura/client/dispatcher_client.h"
  11 #include "ui/aura/client/drag_drop_client.h"
  12 #include "ui/aura/window.h"
  13 #include "ui/aura/window_observer.h"
  14 #include "ui/gfx/screen.h"
  15 #include "ui/views/widget/widget.h"
  16
  17 namespace views {
  18
  19 namespace {
  20
  21 // ActivationChangeObserverImpl is used to observe activation changes and close
  22 // the menu. Additionally it listens for the root window to be destroyed and
  23 // cancel the menu as well.
  24 class ActivationChangeObserverImpl
  25     : public aura::client::ActivationChangeObserver,
  26       public aura::WindowObserver,
  27       public ui::EventHandler {
  28  public:
  29   ActivationChangeObserverImpl(MenuController* controller,
  30                                aura::Window* root)
  31       : controller_(controller),
  32         root_(root) {
  33     aura::client::GetActivationClient(root_)->AddObserver(this);
  34     root_->AddObserver(this);
  35     root_->AddPreTargetHandler(this);
  36   }
  37
  38   virtual ~ActivationChangeObserverImpl() {
  39     Cleanup();
  40   }
  41
  42   // aura::client::ActivationChangeObserver overrides:
  43   virtual void OnWindowActivated(aura::Window* gained_active,
  44                                  aura::Window* lost_active) OVERRIDE {
  45     if (!controller_->drag_in_progress())
  46       controller_->CancelAll();
  47   }
  48
  49   // aura::WindowObserver overrides:
  50   virtual void OnWindowDestroying(aura::Window* window) OVERRIDE {
  51     Cleanup();
  52   }
  53
  54   // ui::EventHandler overrides:
  55   virtual void OnCancelMode(ui::CancelModeEvent* event) OVERRIDE {
  56     controller_->CancelAll();
  57   }
  58
  59  private:
  60   void Cleanup() {
  61     if (!root_)
  62       return;
  63     // The ActivationClient may have been destroyed by the time we get here.
  64     aura::client::ActivationClient* client =
  65         aura::client::GetActivationClient(root_);
  66     if (client)
  67       client->RemoveObserver(this);
  68     root_->RemovePreTargetHandler(this);
  69     root_->RemoveObserver(this);
  70     root_ = NULL;
  71   }
  72
  73   MenuController* controller_;
  74   aura::Window* root_;
  75
  76   DISALLOW_COPY_AND_ASSIGN(ActivationChangeObserverImpl);
  77 };
  78
  79 aura::Window* GetOwnerRootWindow(views::Widget* owner) {
  80   return owner ? owner->GetNativeWindow()->GetRootWindow() : NULL;
  81 }
  82
  83 }  // namespace
  84
  85 void MenuController::RunMessageLoop(bool nested_menu) {
  86   // |owner_| may be NULL.
  87   aura::Window* root = GetOwnerRootWindow(owner_);
  88   if (root) {
  89     scoped_ptr<ActivationChangeObserverImpl> observer;
  90     if (!nested_menu)
  91       observer.reset(new ActivationChangeObserverImpl(this, root));
  92     aura::client::GetDispatcherClient(root)->
  93         RunWithDispatcher(this, owner_->GetNativeWindow());
  94   } else {
  95     base::MessageLoopForUI* loop = base::MessageLoopForUI::current();
  96     base::MessageLoop::ScopedNestableTaskAllower allow(loop);
  97     base::RunLoop run_loop(this);
  98     run_loop.Run();
  99   }
 100 }
 101
 102 bool MenuController::ShouldQuitNow() const {
 103   aura::Window* root = GetOwnerRootWindow(owner_);
 104   return !aura::client::GetDragDropClient(root) ||
 105       !aura::client::GetDragDropClient(root)->IsDragDropInProgress();
 106 }
 107
 108 gfx::Screen* MenuController::GetScreen() {
 109   aura::Window* root = GetOwnerRootWindow(owner_);
 110   return root ?
 111       gfx::Screen::GetScreenFor(root) : gfx::Screen::GetNativeScreen();
 112 }
 113
 114
 115 }  // namespace views