Mus Demo: Refactor code to prepare multiple windows in external mode

services/ui/demo/window_tree_data.h

 // Copyright 2017 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 SERVICES_UI_DEMO_WINDOW_TREE_DATA_H_
 #define SERVICES_UI_DEMO_WINDOW_TREE_DATA_H_
 
 #include "base/memory/ptr_util.h"
 #include "base/timer/timer.h"
 
@@ skipping 11 matching lines @@
 
 class WindowTreeData {
  public:
   explicit WindowTreeData(int square_size);
   ~WindowTreeData();
 
   // Initializes the window tree host and start drawing frames.
   void Init(std::unique_ptr<aura::WindowTreeHostMus> window_tree_host);
   bool IsInitialized() const { return !!window_tree_host_; }
 
+  const aura::WindowTreeHostMus* WindowTreeHost() const {
+    return window_tree_host_.get();
+  }
+
  private:
   // Draws one frame, incrementing the rotation angle.
   void DrawFrame();
 
   // Helper function to retrieve the window to which we draw the bitmap.
   aura::Window* bitmap_window();
 
   // The Window tree host corresponding to this data.
   std::unique_ptr<aura::WindowTreeHostMus> window_tree_host_;
 
   // Destroys itself when the window gets destroyed.
   aura_extra::ImageWindowDelegate* window_delegate_ = nullptr;
 
   // Timer for calling DrawFrame().
   base::RepeatingTimer timer_;
 
   // Current rotation angle for drawing.
   double angle_ = 0.0;
 
   // Size in pixels of the square to draw.
   const int square_size_;
 
   DISALLOW_COPY_AND_ASSIGN(WindowTreeData);
 };
 
 }  // namespace demo
 }  // namespace ui
 
 #endif  // SERVICES_UI_DEMO_WINDOW_TREE_DATA_H_