Fix mus_demo

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_tree_host_factory.h"
+#include "components/mus/public/cpp/window_tree_connection.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() {}
+MusDemo::MusDemo() : window_manager_factory_binding_(this) {}
 
 MusDemo::~MusDemo() {}
 
 void MusDemo::Initialize(shell::Connector* connector,
                          const shell::Identity& identity,
                          uint32_t id) {
   connector_ = connector;
-
-  CreateWindowTreeHost(connector_, this, &window_tree_host_, this);
-  window_tree_host_->SetTitle("MUS Demo");
+  mus::mojom::WindowManagerFactoryServicePtr wm_factory_service;
+  connector->ConnectToInterface("mojo:mus", &wm_factory_service);
+  wm_factory_service->SetWindowManagerFactory(
+      window_manager_factory_binding_.CreateInterfacePtrAndBind());
 }
 
 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)));
 }
 
 void MusDemo::OnUnembed(mus::Window* root) {}
 
 void MusDemo::OnConnectionLost(mus::WindowTreeConnection* connection) {
   timer_.Stop();
 }
 
 void MusDemo::OnEventObserved(const ui::Event& event, mus::Window* target) {}
 
-void MusDemo::SetWindowManagerClient(mus::WindowManagerClient* client) {}
+// mus::mojom::WindowManagerFactory:
+void MusDemo::CreateWindowManager(mus::mojom::DisplayPtr display,
+                                  mus::mojom::WindowTreeClientRequest request) {
+  mus::WindowTreeConnection::CreateForWindowManager(
+      this, std::move(request),
+      mus::WindowTreeConnection::CreateType::DONT_WAIT_FOR_EMBED, this);
+}
+
+// mus::WindowManagerDelegate:
+void MusDemo::SetWindowManagerClient(mus::WindowManagerClient* client) {
+  window_manager_client_ = 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) {
-  // TODO(kylechar): Check if this should return something useful.
   return nullptr;
 }
 
-void MusDemo::OnAccelerator(uint32_t id, const ui::Event& event) {}
+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(),
                                                 kPremul_SkAlphaType);
   bitmap_.allocPixels(image_info);
 }
@@ skipping 28 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