Adds WindowManagerManager to mus

components/mus/ws/window_tree_host_impl.h

 // Copyright 2015 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.
 
 #ifndef COMPONENTS_MUS_WS_WINDOW_TREE_HOST_IMPL_H_
 #define COMPONENTS_MUS_WS_WINDOW_TREE_HOST_IMPL_H_
 
 #include <stdint.h>
 
+#include <map>
 #include <queue>
 
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "components/mus/common/types.h"
 #include "components/mus/public/interfaces/window_manager_constants.mojom.h"
 #include "components/mus/public/interfaces/window_tree_host.mojom.h"
 #include "components/mus/ws/display_manager.h"
 #include "components/mus/ws/event_dispatcher.h"
 #include "components/mus/ws/event_dispatcher_delegate.h"
 #include "components/mus/ws/focus_controller_delegate.h"
 #include "components/mus/ws/focus_controller_observer.h"
 #include "components/mus/ws/server_window.h"
 #include "components/mus/ws/server_window_observer.h"
 #include "components/mus/ws/server_window_tracker.h"
 
 namespace mus {
 namespace ws {
 
 class ConnectionManager;
 class FocusController;
+class WindowManagerState;
 class WindowTreeHostConnection;
 class WindowTreeImpl;
 
 // WindowTreeHostImpl is an implementation of the WindowTreeHost interface.
 // It serves as a top level root window for a window. Its lifetime is managed by
 // ConnectionManager. If the connection to the client breaks or if the user
 // closes the associated window, then this object and related state will be
 // deleted.
+//
+// WindowTreeHost has a single root window whose children are the roots for
+// a per-user WindowManager. WindowTreeHost is configured in two distinct
+// ways:
+// . with a WindowTreeHostConnection. In this mode there is only ever one
+//   WindowManager for the host, which comes from the client that created
+//   the WindowTreeHost.
+// . without a WindowTreeHostConnection. In this mode a WindowManager is
+//   automatically created per user.
 class WindowTreeHostImpl : public DisplayManagerDelegate,
                            public mojom::WindowTreeHost,
                            public FocusControllerObserver,
                            public FocusControllerDelegate,
                            public EventDispatcherDelegate,
                            public ServerWindowObserver {
  public:
   // TODO(fsamuel): All these parameters are just plumbing for creating
   // DisplayManagers. We should probably just store these common parameters
   // in the DisplayManagerFactory and pass them along on DisplayManager::Create.
   WindowTreeHostImpl(ConnectionManager* connection_manager,
                      mojo::Connector* connector,
                      const scoped_refptr<GpuState>& gpu_state,
                      const scoped_refptr<SurfacesState>& surfaces_state);
   ~WindowTreeHostImpl() override;
 
   // Initializes state that depends on the existence of a WindowTreeHostImpl.
   void Init(scoped_ptr<WindowTreeHostConnection> connection);
 
   uint32_t id() const { return id_; }
 
-  const WindowTreeImpl* GetWindowTree() const;
-  WindowTreeImpl* GetWindowTree();
-
+  // TODO(sky): move to WMM.
   void SetFrameDecorationValues(mojom::FrameDecorationValuesPtr values);
   const mojom::FrameDecorationValues& frame_decoration_values() const {
     return *frame_decoration_values_;
   }
 
-  // Returns whether |window| is a descendant of this root but not itself a
-  // root window.
-  bool IsWindowAttachedToRoot(const ServerWindow* window) const;
-
   // Schedules a paint for the specified region in the coordinates of |window|
-  // if
-  // the |window| is in this viewport. Returns whether |window| is in the
+  // if the |window| is in this viewport. Returns whether |window| is in the
   // viewport.
   bool SchedulePaintIfInViewport(const ServerWindow* window,
                                  const gfx::Rect& bounds);
 
   // Schedules destruction of surfaces in |window|. If a frame has been
   // scheduled but not drawn surface destruction is delayed until the frame is
   // drawn, otherwise destruction is immediate.
   void ScheduleSurfaceDestruction(ServerWindow* window);
 
   // Returns the metrics for this viewport.
   const mojom::ViewportMetrics& GetViewportMetrics() const;
 
   mojom::Rotation GetRotation() const;
 
   ConnectionManager* connection_manager() { return connection_manager_; }
 
   EventDispatcher* event_dispatcher() { return &event_dispatcher_; }
 
-  // Returns the root ServerWindow of this viewport.
+  // Returns the root of the WindowTreeHost. The root's children are the roots
+  // of the corresponding WindowManagers.
   ServerWindow* root_window() { return root_.get(); }
   const ServerWindow* root_window() const { return root_.get(); }
 
+  ServerWindow* GetRootWithId(const WindowId& id);
+
+  WindowManagerState* GetWindowManagerStateWithRoot(const ServerWindow* window);
+  // TODO(sky): this is wrong, plumb through user_id.
+  WindowManagerState* GetFirstWindowManagerState();
+
   void SetCapture(ServerWindow* window, bool in_nonclient_area);
 
   void SetFocusedWindow(ServerWindow* window);
   ServerWindow* GetFocusedWindow();
   void DestroyFocusController();
   FocusController* focus_controller() { return focus_controller_.get(); }
 
   void AddActivationParent(ServerWindow* window);
   void RemoveActivationParent(ServerWindow* window);
 
@@ skipping 24 matching lines @@
 
   // WindowTreeHost:
   void SetSize(mojo::SizePtr size) override;
   void SetTitle(const mojo::String& title) override;
 
   void OnEventAck(mojom::WindowTree* tree);
 
  private:
   class ProcessedEventTarget;
   friend class WindowTreeTest;
+  using WindowManagerStateMap =
+      std::map<uint32_t, scoped_ptr<WindowManagerState>>;
 
   // There are two types of events that may be queued, both occur only when
   // waiting for an ack from a client.
   // . We get an event from the DisplayManager. This results in |event| being
   //   set, but |processed_target| is null.
   // . We get an event from the EventDispatcher. In this case both |event| and
   //   |processed_target| are valid.
   // The second case happens if EventDispatcher generates more than one event
   // at a time.
   struct QueuedEvent {
     QueuedEvent();
     ~QueuedEvent();
 
     scoped_ptr<ui::Event> event;
     scoped_ptr<ProcessedEventTarget> processed_target;
   };
 
   // Inits the necessary state once the display is ready.
-  void CallOnDisplayInitializedIfNecessary();
+  void InitWindowManagersIfNecessary();
 
   void OnEventAckTimeout();
 
   // Schedules an event to be processed later.
   void QueueEvent(const ui::Event& event,
                   scoped_ptr<ProcessedEventTarget> processed_event_target);
 
   // Processes the next valid event in |event_queue_|. If the event has already
   // been processed it is dispatched, otherwise the event is passed to the
   // EventDispatcher for processing.
   void ProcessNextEventFromQueue();
 
   // Dispatches the event to the appropriate client and starts the ack timer.
   void DispatchInputEventToWindowImpl(ServerWindow* target,
                                       bool in_nonclient_area,
                                       const ui::Event& event);
 
+  // Creates the set of WindowManagerStates from the
+  // WindowManagerFactoryRegistry.
+  void CreateWindowManagerStatesFromRegistry();
+
   void UpdateNativeCursor(int32_t cursor_id);
 
   // DisplayManagerDelegate:
   ServerWindow* GetRootWindow() override;
   void OnEvent(const ui::Event& event) override;
   void OnNativeCaptureLost() override;
   void OnDisplayClosed() override;
   void OnViewportMetricsChanged(
       const mojom::ViewportMetrics& old_metrics,
       const mojom::ViewportMetrics& new_metrics) override;
@@ skipping 19 matching lines @@
   void OnServerWindowCaptureLost(ServerWindow* window) override;
   void DispatchInputEventToWindow(ServerWindow* target,
                                   bool in_nonclient_area,
                                   const ui::Event& event) override;
 
   // ServerWindowObserver:
   void OnWindowDestroyed(ServerWindow* window) override;
 
   const uint32_t id_;
   scoped_ptr<WindowTreeHostConnection> window_tree_host_connection_;
-  WindowTreeImpl* window_tree_ = nullptr;
+  // Set once Init() has been called.
+  bool init_called_ = false;
   ConnectionManager* const connection_manager_;
   EventDispatcher event_dispatcher_;
   scoped_ptr<ServerWindow> root_;
   scoped_ptr<DisplayManager> display_manager_;
   scoped_ptr<FocusController> focus_controller_;
   mojom::WindowTree* tree_awaiting_input_ack_;
 
   // The last cursor set. Used to track whether we need to change the cursor.
   int32_t last_cursor_;
 
   ServerWindowTracker activation_parents_;
 
   // Set of windows with surfaces that need to be destroyed once the frame
   // draws.
   std::set<ServerWindow*> windows_needing_frame_destruction_;
 
   std::queue<scoped_ptr<QueuedEvent>> event_queue_;
   base::OneShotTimer event_ack_timer_;
 
+  WindowManagerStateMap window_manager_state_map_;
+
   mojom::FrameDecorationValuesPtr frame_decoration_values_;
 
   DISALLOW_COPY_AND_ASSIGN(WindowTreeHostImpl);
 };
 
 }  // namespace ws
 }  // namespace mus
 
 #endif  // COMPONENTS_MUS_WS_WINDOW_TREE_HOST_IMPL_H_