Set CaptureClient instance for aura_demo and aura_bench

ui/aura/demo/demo_main.cc

 // Copyright (c) 2012 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 "base/at_exit.h"
 #include "base/command_line.h"
 #include "base/i18n/icu_util.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "third_party/skia/include/core/SkXfermode.h"
 #include "ui/aura/client/stacking_client.h"
 #include "ui/aura/env.h"
 #include "ui/aura/event.h"
 #include "ui/aura/root_window.h"
 #include "ui/aura/single_display_manager.h"
+#include "ui/aura/shared/root_window_capture_client.h"
 #include "ui/aura/window.h"
 #include "ui/aura/window_delegate.h"
 #include "ui/base/hit_test.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/base/ui_base_paths.h"
 #include "ui/compositor/test/compositor_test_support.h"
 #include "ui/gfx/canvas.h"
 #include "ui/gfx/rect.h"
 
 #if defined(USE_X11)
@@ skipping 63 matching lines @@
     aura::client::SetStackingClient(this);
   }
 
   virtual ~DemoStackingClient() {
     aura::client::SetStackingClient(NULL);
   }
 
   // Overridden from aura::client::StackingClient:
   virtual aura::Window* GetDefaultParent(aura::Window* window,
                                          const gfx::Rect& bounds) OVERRIDE {
+
+    if (!capture_client_.get()) {
+      capture_client_.reset(
+          new aura::shared::RootWindowCaptureClient(root_window_));
+    }
     return root_window_;
   }
 
  private:
   aura::RootWindow* root_window_;
 
+  scoped_ptr<aura::shared::RootWindowCaptureClient> capture_client_;
+
   DISALLOW_COPY_AND_ASSIGN(DemoStackingClient);
 };
 
 }  // namespace
 
 int main(int argc, char** argv) {
   CommandLine::Init(argc, argv);
 
   // The exit manager is in charge of calling the dtors of singleton objects.
   base::AtExitManager exit_manager;
@@ skipping 36 matching lines @@
   window3.Show();
   window3.SetParent(&window2);
 
   root_window->ShowRootWindow();
   MessageLoopForUI::current()->Run();
 
   ui::CompositorTestSupport::Terminate();
 
   return 0;
 }