| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "ui/snapshot/screenshot_grabber.h" | 5 #include "ui/snapshot/screenshot_grabber.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 | 8 |
| 9 #include <climits> | 9 #include <climits> |
| 10 #include <string> | 10 #include <string> |
| (...skipping 25 matching lines...) Expand all Loading... |
| 36 // more than 1000 to prevent the conflict of filenames. | 36 // more than 1000 to prevent the conflict of filenames. |
| 37 const int kScreenshotMinimumIntervalInMS = 1000; | 37 const int kScreenshotMinimumIntervalInMS = 1000; |
| 38 | 38 |
| 39 using ShowNotificationCallback = | 39 using ShowNotificationCallback = |
| 40 base::Callback<void(ScreenshotGrabberObserver::Result screenshot_result, | 40 base::Callback<void(ScreenshotGrabberObserver::Result screenshot_result, |
| 41 const base::FilePath& screenshot_path)>; | 41 const base::FilePath& screenshot_path)>; |
| 42 | 42 |
| 43 void SaveScreenshot(scoped_refptr<base::TaskRunner> ui_task_runner, | 43 void SaveScreenshot(scoped_refptr<base::TaskRunner> ui_task_runner, |
| 44 const ShowNotificationCallback& callback, | 44 const ShowNotificationCallback& callback, |
| 45 const base::FilePath& screenshot_path, | 45 const base::FilePath& screenshot_path, |
| 46 scoped_refptr<base::RefCountedMemory> png_data, | 46 scoped_refptr<base::RefCountedBytes> png_data, |
| 47 ScreenshotGrabberDelegate::FileResult result, | 47 ScreenshotGrabberDelegate::FileResult result, |
| 48 const base::FilePath& local_path) { | 48 const base::FilePath& local_path) { |
| 49 DCHECK(!base::MessageLoopForUI::IsCurrent()); | 49 DCHECK(!base::MessageLoopForUI::IsCurrent()); |
| 50 DCHECK(!screenshot_path.empty()); | 50 DCHECK(!screenshot_path.empty()); |
| 51 | 51 |
| 52 // Convert FileResult into ScreenshotGrabberObserver::Result. | 52 // Convert FileResult into ScreenshotGrabberObserver::Result. |
| 53 ScreenshotGrabberObserver::Result screenshot_result = | 53 ScreenshotGrabberObserver::Result screenshot_result = |
| 54 ScreenshotGrabberObserver::SCREENSHOT_SUCCESS; | 54 ScreenshotGrabberObserver::SCREENSHOT_SUCCESS; |
| 55 switch (result) { | 55 switch (result) { |
| 56 case ScreenshotGrabberDelegate::FILE_SUCCESS: | 56 case ScreenshotGrabberDelegate::FILE_SUCCESS: |
| 57 // Successfully got a local file to write to, write png data. | 57 // Successfully got a local file to write to, write png data. |
| 58 DCHECK_GT(static_cast<int>(png_data->size()), 0); | 58 DCHECK_GT(static_cast<int>(png_data->size()), 0); |
| 59 if (static_cast<size_t>(base::WriteFile( | 59 if (static_cast<size_t>(base::WriteFile( |
| 60 local_path, reinterpret_cast<const char*>(png_data->front()), | 60 local_path, reinterpret_cast<char*>(&(png_data->data()[0])), |
| 61 static_cast<int>(png_data->size()))) != png_data->size()) { | 61 static_cast<int>(png_data->size()))) != png_data->size()) { |
| 62 LOG(ERROR) << "Failed to save to " << local_path.value(); | 62 LOG(ERROR) << "Failed to save to " << local_path.value(); |
| 63 screenshot_result = | 63 screenshot_result = |
| 64 ScreenshotGrabberObserver::SCREENSHOT_WRITE_FILE_FAILED; | 64 ScreenshotGrabberObserver::SCREENSHOT_WRITE_FILE_FAILED; |
| 65 } | 65 } |
| 66 break; | 66 break; |
| 67 case ScreenshotGrabberDelegate::FILE_CHECK_DIR_FAILED: | 67 case ScreenshotGrabberDelegate::FILE_CHECK_DIR_FAILED: |
| 68 screenshot_result = | 68 screenshot_result = |
| 69 ScreenshotGrabberObserver::SCREENSHOT_CHECK_DIR_FAILED; | 69 ScreenshotGrabberObserver::SCREENSHOT_CHECK_DIR_FAILED; |
| 70 break; | 70 break; |
| (...skipping 90 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 161 // screen (i.e. non partial) screenshot. The only time is_partial can be | 161 // screen (i.e. non partial) screenshot. The only time is_partial can be |
| 162 // false, we will also have an identification string for the window. | 162 // false, we will also have an identification string for the window. |
| 163 std::string window_identifier; | 163 std::string window_identifier; |
| 164 #if defined(USE_AURA) | 164 #if defined(USE_AURA) |
| 165 aura::Window* aura_window = static_cast<aura::Window*>(window); | 165 aura::Window* aura_window = static_cast<aura::Window*>(window); |
| 166 is_partial = rect.size() != aura_window->bounds().size(); | 166 is_partial = rect.size() != aura_window->bounds().size(); |
| 167 window_identifier = aura_window->GetBoundsInScreen().ToString(); | 167 window_identifier = aura_window->GetBoundsInScreen().ToString(); |
| 168 | 168 |
| 169 cursor_hider_ = ScopedCursorHider::Create(aura_window->GetRootWindow()); | 169 cursor_hider_ = ScopedCursorHider::Create(aura_window->GetRootWindow()); |
| 170 #endif | 170 #endif |
| 171 ui::GrabWindowSnapshotAsyncPNG( | 171 ui::GrabWindowSnapshotAsync( |
| 172 window, rect, blocking_task_runner_, | 172 window, rect, blocking_task_runner_, |
| 173 base::Bind(&ScreenshotGrabber::GrabWindowSnapshotAsyncCallback, | 173 base::Bind(&ScreenshotGrabber::GrabWindowSnapshotAsyncCallback, |
| 174 factory_.GetWeakPtr(), window_identifier, screenshot_path, | 174 factory_.GetWeakPtr(), window_identifier, screenshot_path, |
| 175 is_partial)); | 175 is_partial)); |
| 176 } | 176 } |
| 177 | 177 |
| 178 bool ScreenshotGrabber::CanTakeScreenshot() { | 178 bool ScreenshotGrabber::CanTakeScreenshot() { |
| 179 return last_screenshot_timestamp_.is_null() || | 179 return last_screenshot_timestamp_.is_null() || |
| 180 base::TimeTicks::Now() - last_screenshot_timestamp_ > | 180 base::TimeTicks::Now() - last_screenshot_timestamp_ > |
| 181 base::TimeDelta::FromMilliseconds(kScreenshotMinimumIntervalInMS); | 181 base::TimeDelta::FromMilliseconds(kScreenshotMinimumIntervalInMS); |
| (...skipping 20 matching lines...) Expand all Loading... |
| 202 | 202 |
| 203 bool ScreenshotGrabber::HasObserver( | 203 bool ScreenshotGrabber::HasObserver( |
| 204 const ScreenshotGrabberObserver* observer) const { | 204 const ScreenshotGrabberObserver* observer) const { |
| 205 return observers_.HasObserver(observer); | 205 return observers_.HasObserver(observer); |
| 206 } | 206 } |
| 207 | 207 |
| 208 void ScreenshotGrabber::GrabWindowSnapshotAsyncCallback( | 208 void ScreenshotGrabber::GrabWindowSnapshotAsyncCallback( |
| 209 const std::string& window_identifier, | 209 const std::string& window_identifier, |
| 210 base::FilePath screenshot_path, | 210 base::FilePath screenshot_path, |
| 211 bool is_partial, | 211 bool is_partial, |
| 212 scoped_refptr<base::RefCountedMemory> png_data) { | 212 scoped_refptr<base::RefCountedBytes> png_data) { |
| 213 DCHECK(base::MessageLoopForUI::IsCurrent()); | 213 DCHECK(base::MessageLoopForUI::IsCurrent()); |
| 214 if (!png_data.get()) { | 214 if (!png_data.get()) { |
| 215 if (is_partial) { | 215 if (is_partial) { |
| 216 LOG(ERROR) << "Failed to grab the window screenshot"; | 216 LOG(ERROR) << "Failed to grab the window screenshot"; |
| 217 NotifyScreenshotCompleted( | 217 NotifyScreenshotCompleted( |
| 218 ScreenshotGrabberObserver::SCREENSHOT_GRABWINDOW_PARTIAL_FAILED, | 218 ScreenshotGrabberObserver::SCREENSHOT_GRABWINDOW_PARTIAL_FAILED, |
| 219 screenshot_path); | 219 screenshot_path); |
| 220 } else { | 220 } else { |
| 221 LOG(ERROR) << "Failed to grab the window screenshot for " | 221 LOG(ERROR) << "Failed to grab the window screenshot for " |
| 222 << window_identifier; | 222 << window_identifier; |
| 223 NotifyScreenshotCompleted( | 223 NotifyScreenshotCompleted( |
| 224 ScreenshotGrabberObserver::SCREENSHOT_GRABWINDOW_FULL_FAILED, | 224 ScreenshotGrabberObserver::SCREENSHOT_GRABWINDOW_FULL_FAILED, |
| 225 screenshot_path); | 225 screenshot_path); |
| 226 } | 226 } |
| 227 return; | 227 return; |
| 228 } | 228 } |
| 229 | 229 |
| 230 ShowNotificationCallback notification_callback(base::Bind( | 230 ShowNotificationCallback notification_callback(base::Bind( |
| 231 &ScreenshotGrabber::NotifyScreenshotCompleted, factory_.GetWeakPtr())); | 231 &ScreenshotGrabber::NotifyScreenshotCompleted, factory_.GetWeakPtr())); |
| 232 client_->PrepareFileAndRunOnBlockingPool( | 232 client_->PrepareFileAndRunOnBlockingPool( |
| 233 screenshot_path, blocking_task_runner_, | 233 screenshot_path, blocking_task_runner_, |
| 234 base::Bind(&SaveScreenshot, base::ThreadTaskRunnerHandle::Get(), | 234 base::Bind(&SaveScreenshot, base::ThreadTaskRunnerHandle::Get(), |
| 235 notification_callback, screenshot_path, png_data)); | 235 notification_callback, screenshot_path, png_data)); |
| 236 } | 236 } |
| 237 | 237 |
| 238 } // namespace ui | 238 } // namespace ui |
| OLD | NEW |