Use capturing with XGetImage() only in screenshot_taker.

chrome/browser/ui/views/ash/screenshot_taker.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 "chrome/browser/ui/views/ash/screenshot_taker.h"
 
 #include <climits>
 #include <string>
 
 #include "ash/shell.h"
 #include "ash/shell_window_ids.h"
 #include "base/bind.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/i18n/time_formatting.h"
 #include "base/logging.h"
 #include "base/memory/ref_counted_memory.h"
 #include "base/stringprintf.h"
 #include "base/time.h"
 #include "chrome/browser/download/download_util.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/ui/window_snapshot/window_snapshot.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/browser_thread.h"
+#include "ui/aura/root_window.h"
 #include "ui/aura/window.h"
 
 #if defined(OS_CHROMEOS)
 #include "chrome/browser/chromeos/login/user_manager.h"
 #endif
 
 namespace {
 bool ShouldUse24HourClock() {
 #if defined(OS_CHROMEOS)
   Profile* profile = ProfileManager::GetDefaultProfileOrOffTheRecord();
@@ skipping 73 matching lines @@
   }
 
   if (static_cast<size_t>(file_util::WriteFile(
           screenshot_path,
           reinterpret_cast<char*>(&(png_data->data()[0])),
           png_data->size())) != png_data->size()) {
     LOG(ERROR) << "Failed to save to " << screenshot_path.value();
   }
 }
 
+bool GrabWindowSnapshot(aura::Window* window,
+                        const gfx::Rect& snapshot_bounds,
+                        std::vector<unsigned char>* png_data) {
+#if defined(OS_LINUX)
+  // We use XGetImage() for Linux/ChromeOS for performance reasons.
+  // See crbug.com/119492
+  // TODO(mukai): remove this when the performance issue has been fixed.
+  if (window->GetRootWindow()->GrabSnapshot(snapshot_bounds, png_data))
+    return true;
+#endif  // OS_LINUX
+
+  return browser::GrabWindowSnapshot(window, png_data, snapshot_bounds);
+}
+
 // How opaque should the layer that we flash onscreen to provide visual
 // feedback after the screenshot is taken be?
 const float kVisualFeedbackLayerOpacity = 0.25f;
 
 // How long should the visual feedback layer be displayed?
 const int64 kVisualFeedbackLayerDisplayTimeMs = 100;
 
 }  // namespace
 
 ScreenshotTaker::ScreenshotTaker() {
 }
 
 ScreenshotTaker::~ScreenshotTaker() {
 }
 
 void ScreenshotTaker::HandleTakePartialScreenshot(
     aura::Window* window, const gfx::Rect& rect) {
   DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
 
   scoped_refptr<base::RefCountedBytes> png_data(new base::RefCountedBytes);
 
   bool is_logged_in = true;
 #if defined(OS_CHROMEOS)
   is_logged_in = chromeos::UserManager::Get()->IsUserLoggedIn();
 #endif
 
   bool use_24hour_clock = ShouldUse24HourClock();
 
-  if (browser::GrabWindowSnapshot(window, &png_data->data(), rect)) {
+  if (GrabWindowSnapshot(window, rect, &png_data->data())) {
     DisplayVisualFeedback(rect);
     content::BrowserThread::PostTask(
         content::BrowserThread::FILE, FROM_HERE,
         base::Bind(&SaveScreenshot, is_logged_in, use_24hour_clock, png_data));
   } else {
     LOG(ERROR) << "Failed to grab the window screenshot";
   }
 }
 
 void ScreenshotTaker::HandleTakeScreenshot(aura::Window* window) {
@@ skipping 15 matching lines @@
       ash::internal::kShellWindowId_OverlayContainer)->layer();
   parent->Add(visual_feedback_layer_.get());
   visual_feedback_layer_->SetVisible(true);
 
   MessageLoopForUI::current()->PostDelayedTask(
       FROM_HERE,
       base::Bind(&ScreenshotTaker::CloseVisualFeedbackLayer,
                  base::Unretained(this)),
       base::TimeDelta::FromMilliseconds(kVisualFeedbackLayerDisplayTimeMs));
 }