Changes how window manager obtains WindowTree from mus

components/mus/demo/mus_demo.cc

 // Copyright 2016 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 "components/mus/demo/mus_demo.h"
 
 #include "base/time/time.h"
 #include "components/bitmap_uploader/bitmap_uploader.h"
+#include "components/mus/public/cpp/window.h"
 #include "components/mus/public/cpp/window_tree_client.h"
+#include "services/shell/public/cpp/connector.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 #include "third_party/skia/include/core/SkColor.h"
 #include "third_party/skia/include/core/SkImageInfo.h"
 #include "third_party/skia/include/core/SkPaint.h"
 #include "third_party/skia/include/core/SkRect.h"
 #include "ui/gfx/geometry/rect.h"
 
 namespace mus_demo {
 
 namespace {
@@ skipping 27 matching lines @@
     canvas->rotate(angle);
     canvas->translate(-SkFloatToScalar(canvas_size.width() * 0.5f),
                       -SkFloatToScalar(canvas_size.height() * 0.5f));
   }
 
   canvas->drawRect(rect, paint);
 }
 
 }  // namespace
 
-MusDemo::MusDemo() : window_manager_factory_binding_(this) {}
+MusDemo::MusDemo() {}
 
-MusDemo::~MusDemo() {}
+MusDemo::~MusDemo() {
+  delete window_tree_client_;
+}
 
 void MusDemo::Initialize(shell::Connector* connector,
                          const shell::Identity& identity,
                          uint32_t id) {
   connector_ = connector;
-  mus::mojom::WindowManagerFactoryServicePtr wm_factory_service;
-  connector->ConnectToInterface("mojo:mus", &wm_factory_service);
-  wm_factory_service->SetWindowManagerFactory(
-      window_manager_factory_binding_.CreateInterfacePtrAndBind());
+  window_tree_client_ = new mus::WindowTreeClient(this, this, nullptr);
+  window_tree_client_->ConnectAsWindowManager(connector);
 }
 
 bool MusDemo::AcceptConnection(shell::Connection* connection) {
   return true;
 }
 
 void MusDemo::OnEmbed(mus::Window* window) {
-  window_ = window;
-
-  // Initialize bitmap uploader for sending frames to MUS.
-  uploader_.reset(new bitmap_uploader::BitmapUploader(window_));
-  uploader_->Init(connector_);
-
-  // Draw initial frame and start the timer to regularly draw frames.
-  DrawFrame();
-  timer_.Start(FROM_HERE, base::TimeDelta::FromMilliseconds(kFrameDelay),
-               base::Bind(&MusDemo::DrawFrame, base::Unretained(this)));
+  // Not called for the WindowManager.
+  NOTREACHED();
 }
 
-void MusDemo::OnUnembed(mus::Window* root) {}
-
 void MusDemo::OnWindowTreeClientDestroyed(mus::WindowTreeClient* client) {
+  window_tree_client_ = nullptr;
   timer_.Stop();
 }
 
 void MusDemo::OnEventObserved(const ui::Event& event, mus::Window* target) {}
 
-// mus::mojom::WindowManagerFactory:
-void MusDemo::CreateWindowManager(mus::mojom::DisplayPtr display,
-                                  mus::mojom::WindowTreeClientRequest request) {
-  new mus::WindowTreeClient(this, this, std::move(request));
-}
-
-// mus::WindowManagerDelegate:
-void MusDemo::SetWindowManagerClient(mus::WindowManagerClient* client) {
-  window_manager_client_ = client;
-}
+void MusDemo::SetWindowManagerClient(mus::WindowManagerClient* client) {}
 
 bool MusDemo::OnWmSetBounds(mus::Window* window, gfx::Rect* bounds) {
   return true;
 }
 
 bool MusDemo::OnWmSetProperty(mus::Window* window,
                               const std::string& name,
                               std::unique_ptr<std::vector<uint8_t>>* new_data) {
   return true;
 }
 
 mus::Window* MusDemo::OnWmCreateTopLevelWindow(
     std::map<std::string, std::vector<uint8_t>>* properties) {
   return nullptr;
 }
 
 void MusDemo::OnWmClientJankinessChanged(
     const std::set<mus::Window*>& client_windows,
     bool janky) {
   // Don't care
 }
 
+void MusDemo::OnWmNewDisplay(mus::Window* window,
+                             const display::Display& display) {
+  DCHECK(!window_);  // Only support one display.
+  window_ = window;
+
+  // Initialize bitmap uploader for sending frames to MUS.
+  uploader_.reset(new bitmap_uploader::BitmapUploader(window_));
+  uploader_->Init(connector_);
+
+  // Draw initial frame and start the timer to regularly draw frames.
+  DrawFrame();
+  timer_.Start(FROM_HERE, base::TimeDelta::FromMilliseconds(kFrameDelay),
+               base::Bind(&MusDemo::DrawFrame, base::Unretained(this)));
+}
+
 void MusDemo::OnAccelerator(uint32_t id, const ui::Event& event) {
   // Don't care
 }
 
 void MusDemo::AllocBitmap() {
   const gfx::Rect bounds = window_->GetBoundsInRoot();
 
   // Allocate bitmap the same size as the window for drawing.
   bitmap_.reset();
   SkImageInfo image_info = SkImageInfo::MakeN32(bounds.width(), bounds.height(),
@@ skipping 31 matching lines @@
   std::unique_ptr<std::vector<unsigned char>> data(
       new std::vector<unsigned char>(addr, addr + bytes));
   bitmap_.unlockPixels();
 
   // Send frame to MUS via BitmapUploader.
   uploader_->SetBitmap(bounds.width(), bounds.height(), std::move(data),
                        bitmap_uploader::BitmapUploader::BGRA);
 }
 
 }  // namespace mus_demo