| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "chrome/browser/thumbnails/thumbnail_tab_helper.h" | 5 #include "chrome/browser/thumbnails/thumbnail_tab_helper.h" |
| 6 | 6 |
| 7 #include "chrome/browser/browser_process.h" | 7 #include "chrome/browser/browser_process.h" |
| 8 #include "chrome/browser/profiles/profile.h" | 8 #include "chrome/browser/profiles/profile.h" |
| 9 #include "chrome/browser/thumbnails/thumbnail_service.h" | 9 #include "chrome/browser/thumbnails/thumbnail_service.h" |
| 10 #include "chrome/browser/thumbnails/thumbnail_service_factory.h" | 10 #include "chrome/browser/thumbnails/thumbnail_service_factory.h" |
| (...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 51 using thumbnails::ClipResult; | 51 using thumbnails::ClipResult; |
| 52 using thumbnails::ThumbnailingContext; | 52 using thumbnails::ThumbnailingContext; |
| 53 using thumbnails::ThumbnailingAlgorithm; | 53 using thumbnails::ThumbnailingAlgorithm; |
| 54 | 54 |
| 55 namespace { | 55 namespace { |
| 56 | 56 |
| 57 // Feed the constructed thumbnail to the thumbnail service. | 57 // Feed the constructed thumbnail to the thumbnail service. |
| 58 void UpdateThumbnail(const ThumbnailingContext& context, | 58 void UpdateThumbnail(const ThumbnailingContext& context, |
| 59 const SkBitmap& thumbnail) { | 59 const SkBitmap& thumbnail) { |
| 60 gfx::Image image = gfx::Image::CreateFrom1xBitmap(thumbnail); | 60 gfx::Image image = gfx::Image::CreateFrom1xBitmap(thumbnail); |
| 61 context.service()->SetPageThumbnail(context, image); | 61 context.service->SetPageThumbnail(context, image); |
| 62 DVLOG(1) << "Thumbnail taken for " << context.GetURL() << ": " | 62 DVLOG(1) << "Thumbnail taken for " << context.url << ": " |
| 63 << context.score().ToString(); | 63 << context.score.ToString(); |
| 64 } | 64 } |
| 65 | 65 |
| 66 void ProcessCapturedBitmap(scoped_refptr<ThumbnailingContext> context, | 66 void ProcessCapturedBitmap(scoped_refptr<ThumbnailingContext> context, |
| 67 scoped_refptr<ThumbnailingAlgorithm> algorithm, | 67 scoped_refptr<ThumbnailingAlgorithm> algorithm, |
| 68 const SkBitmap& bitmap, | 68 const SkBitmap& bitmap, |
| 69 content::ReadbackResponse response) { | 69 content::ReadbackResponse response) { |
| 70 // Was the web contents destroyed before the thumbnail could be generated? | |
| 71 if (!context->web_contents()) | |
| 72 return; | |
| 73 | |
| 74 // Balance the IncrementCapturerCount() from UpdateThumbnailIfNecessary(). | |
| 75 context->web_contents()->DecrementCapturerCount(); | |
| 76 | |
| 77 if (response != content::READBACK_SUCCESS) | 70 if (response != content::READBACK_SUCCESS) |
| 78 return; | 71 return; |
| 79 | 72 |
| 80 // On success, we must be on the UI thread (on failure because of shutdown we | 73 // On success, we must be on the UI thread (on failure because of shutdown we |
| 81 // are not on the UI thread). | 74 // are not on the UI thread). |
| 82 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); | 75 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 83 | 76 |
| 84 algorithm->ProcessBitmap(context, base::Bind(&UpdateThumbnail), bitmap); | 77 algorithm->ProcessBitmap(context, base::Bind(&UpdateThumbnail), bitmap); |
| 85 } | 78 } |
| 86 | 79 |
| 87 void AsyncProcessThumbnail(scoped_refptr<ThumbnailingContext> context, | 80 void AsyncProcessThumbnail(content::WebContents* web_contents, |
| 81 scoped_refptr<ThumbnailingContext> context, |
| 88 scoped_refptr<ThumbnailingAlgorithm> algorithm) { | 82 scoped_refptr<ThumbnailingAlgorithm> algorithm) { |
| 89 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); | 83 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); |
| 90 RenderWidgetHost* render_widget_host = | 84 RenderWidgetHost* render_widget_host = |
| 91 context->web_contents()->GetRenderViewHost()->GetWidget(); | 85 web_contents->GetRenderViewHost()->GetWidget(); |
| 92 content::RenderWidgetHostView* view = render_widget_host->GetView(); | 86 content::RenderWidgetHostView* view = render_widget_host->GetView(); |
| 93 if (!view) | 87 if (!view) |
| 94 return; | 88 return; |
| 95 if (!view->IsSurfaceAvailableForCopy()) | 89 if (!view->IsSurfaceAvailableForCopy()) |
| 96 return; | 90 return; |
| 97 | 91 |
| 98 gfx::Rect copy_rect = gfx::Rect(view->GetViewBounds().size()); | 92 gfx::Rect copy_rect = gfx::Rect(view->GetViewBounds().size()); |
| 99 // Clip the pixels that will commonly hold a scrollbar, which looks bad in | 93 // Clip the pixels that will commonly hold a scrollbar, which looks bad in |
| 100 // thumbnails. | 94 // thumbnails. |
| 101 int scrollbar_size = gfx::scrollbar_size(); | 95 int scrollbar_size = gfx::scrollbar_size(); |
| 102 gfx::Size copy_size; | 96 gfx::Size copy_size; |
| 103 copy_rect.Inset(0, 0, scrollbar_size, scrollbar_size); | 97 copy_rect.Inset(0, 0, scrollbar_size, scrollbar_size); |
| 104 | 98 |
| 105 if (copy_rect.IsEmpty()) | 99 if (copy_rect.IsEmpty()) |
| 106 return; | 100 return; |
| 107 | 101 |
| 108 ui::ScaleFactor scale_factor = | 102 ui::ScaleFactor scale_factor = |
| 109 ui::GetSupportedScaleFactor( | 103 ui::GetSupportedScaleFactor( |
| 110 ui::GetScaleFactorForNativeView(view->GetNativeView())); | 104 ui::GetScaleFactorForNativeView(view->GetNativeView())); |
| 111 gfx::Size requested_copy_size; | 105 context->clip_result = algorithm->GetCanvasCopyInfo( |
| 112 context->set_clip_result(algorithm->GetCanvasCopyInfo( | |
| 113 copy_rect.size(), | 106 copy_rect.size(), |
| 114 scale_factor, | 107 scale_factor, |
| 115 ©_rect, | 108 ©_rect, |
| 116 &requested_copy_size)); | 109 &context->requested_copy_size); |
| 117 context->set_requested_copy_size(requested_copy_size); | |
| 118 render_widget_host->CopyFromBackingStore( | 110 render_widget_host->CopyFromBackingStore( |
| 119 copy_rect, | 111 copy_rect, |
| 120 requested_copy_size, | 112 context->requested_copy_size, |
| 121 base::Bind(&ProcessCapturedBitmap, context, algorithm), | 113 base::Bind(&ProcessCapturedBitmap, context, algorithm), |
| 122 kN32_SkColorType); | 114 kN32_SkColorType); |
| 123 } | 115 } |
| 124 | 116 |
| 125 } // namespace | 117 } // namespace |
| 126 | 118 |
| 127 ThumbnailTabHelper::ThumbnailTabHelper(content::WebContents* contents) | 119 ThumbnailTabHelper::ThumbnailTabHelper(content::WebContents* contents) |
| 128 : content::WebContentsObserver(contents), | 120 : content::WebContentsObserver(contents), |
| 129 enabled_(true), | 121 enabled_(true), |
| 130 load_interrupted_(false) { | 122 load_interrupted_(false) { |
| (...skipping 66 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 197 | 189 |
| 198 scoped_refptr<thumbnails::ThumbnailService> thumbnail_service = | 190 scoped_refptr<thumbnails::ThumbnailService> thumbnail_service = |
| 199 ThumbnailServiceFactory::GetForProfile(profile); | 191 ThumbnailServiceFactory::GetForProfile(profile); |
| 200 | 192 |
| 201 // Skip if we don't need to update the thumbnail. | 193 // Skip if we don't need to update the thumbnail. |
| 202 if (thumbnail_service.get() == NULL || | 194 if (thumbnail_service.get() == NULL || |
| 203 !thumbnail_service->ShouldAcquirePageThumbnail(url)) { | 195 !thumbnail_service->ShouldAcquirePageThumbnail(url)) { |
| 204 return; | 196 return; |
| 205 } | 197 } |
| 206 | 198 |
| 207 web_contents->IncrementCapturerCount(gfx::Size()); | |
| 208 | |
| 209 scoped_refptr<thumbnails::ThumbnailingAlgorithm> algorithm( | 199 scoped_refptr<thumbnails::ThumbnailingAlgorithm> algorithm( |
| 210 thumbnail_service->GetThumbnailingAlgorithm()); | 200 thumbnail_service->GetThumbnailingAlgorithm()); |
| 211 | 201 |
| 212 scoped_refptr<ThumbnailingContext> context(new ThumbnailingContext( | 202 scoped_refptr<ThumbnailingContext> context(new ThumbnailingContext( |
| 213 web_contents, thumbnail_service.get(), load_interrupted_)); | 203 web_contents, thumbnail_service.get(), load_interrupted_)); |
| 214 AsyncProcessThumbnail(context, algorithm); | 204 AsyncProcessThumbnail(web_contents, context, algorithm); |
| 215 } | 205 } |
| 216 | 206 |
| 217 void ThumbnailTabHelper::RenderViewHostCreated( | 207 void ThumbnailTabHelper::RenderViewHostCreated( |
| 218 content::RenderViewHost* renderer) { | 208 content::RenderViewHost* renderer) { |
| 219 // NOTIFICATION_WEB_CONTENTS_RENDER_VIEW_HOST_CREATED is really a new | 209 // NOTIFICATION_WEB_CONTENTS_RENDER_VIEW_HOST_CREATED is really a new |
| 220 // RenderView, not RenderViewHost, and there is no good way to get | 210 // RenderView, not RenderViewHost, and there is no good way to get |
| 221 // notifications of RenderViewHosts. So just be tolerant of re-registrations. | 211 // notifications of RenderViewHosts. So just be tolerant of re-registrations. |
| 222 bool registered = registrar_.IsRegistered( | 212 bool registered = registrar_.IsRegistered( |
| 223 this, content::NOTIFICATION_RENDER_WIDGET_VISIBILITY_CHANGED, | 213 this, content::NOTIFICATION_RENDER_WIDGET_VISIBILITY_CHANGED, |
| 224 content::Source<RenderWidgetHost>(renderer->GetWidget())); | 214 content::Source<RenderWidgetHost>(renderer->GetWidget())); |
| 225 if (!registered) { | 215 if (!registered) { |
| 226 registrar_.Add(this, content::NOTIFICATION_RENDER_WIDGET_VISIBILITY_CHANGED, | 216 registrar_.Add(this, content::NOTIFICATION_RENDER_WIDGET_VISIBILITY_CHANGED, |
| 227 content::Source<RenderWidgetHost>(renderer->GetWidget())); | 217 content::Source<RenderWidgetHost>(renderer->GetWidget())); |
| 228 } | 218 } |
| 229 } | 219 } |
| 230 | 220 |
| 231 void ThumbnailTabHelper::WidgetHidden(RenderWidgetHost* widget) { | 221 void ThumbnailTabHelper::WidgetHidden(RenderWidgetHost* widget) { |
| 232 if (!enabled_) | 222 if (!enabled_) |
| 233 return; | 223 return; |
| 234 UpdateThumbnailIfNecessary(web_contents()); | 224 UpdateThumbnailIfNecessary(web_contents()); |
| 235 } | 225 } |
| OLD | NEW |