src/chrome/browser/thumbnails/thumbnail_service_impl.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/thumbnails/thumbnail_service_impl.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/memory/ref_counted_memory.h"
   9 #include "base/time/time.h"
  10 #include "chrome/browser/history/history_service.h"
  11 #include "chrome/browser/search/search.h"
  12 #include "chrome/browser/thumbnails/content_based_thumbnailing_algorithm.h"
  13 #include "chrome/browser/thumbnails/simple_thumbnail_crop.h"
  14 #include "chrome/browser/thumbnails/thumbnailing_context.h"
  15 #include "chrome/common/chrome_switches.h"
  16 #include "url/gurl.h"
  17
  18 using content::BrowserThread;
  19
  20 namespace {
  21
  22 // The thumbnail size in DIP.
  23 const int kThumbnailWidth = 212;
  24 const int kThumbnailHeight = 132;
  25
  26 // True if thumbnail retargeting feature is enabled (Finch/flags).
  27 bool IsThumbnailRetargetingEnabled() {
  28   if (!chrome::IsInstantExtendedAPIEnabled())
  29     return false;
  30
  31   return CommandLine::ForCurrentProcess()->HasSwitch(
  32       switches::kEnableThumbnailRetargeting);
  33 }
  34
  35 void AddForcedURLOnUIThread(scoped_refptr<history::TopSites> top_sites,
  36                             const GURL& url) {
  37   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
  38
  39   if (top_sites.get() != NULL)
  40     top_sites->AddForcedURL(url, base::Time::Now());
  41 }
  42
  43 }  // namespace
  44
  45 namespace thumbnails {
  46
  47 ThumbnailServiceImpl::ThumbnailServiceImpl(Profile* profile)
  48     : top_sites_(profile->GetTopSites()),
  49       use_thumbnail_retargeting_(IsThumbnailRetargetingEnabled()) {
  50 }
  51
  52 ThumbnailServiceImpl::~ThumbnailServiceImpl() {
  53 }
  54
  55 bool ThumbnailServiceImpl::SetPageThumbnail(const ThumbnailingContext& context,
  56                                             const gfx::Image& thumbnail) {
  57   scoped_refptr<history::TopSites> local_ptr(top_sites_);
  58   if (local_ptr.get() == NULL)
  59     return false;
  60
  61   return local_ptr->SetPageThumbnail(context.url, thumbnail, context.score);
  62 }
  63
  64 bool ThumbnailServiceImpl::GetPageThumbnail(
  65     const GURL& url,
  66     bool prefix_match,
  67     scoped_refptr<base::RefCountedMemory>* bytes) {
  68   scoped_refptr<history::TopSites> local_ptr(top_sites_);
  69   if (local_ptr.get() == NULL)
  70     return false;
  71
  72   return local_ptr->GetPageThumbnail(url, prefix_match, bytes);
  73 }
  74
  75 void ThumbnailServiceImpl::AddForcedURL(const GURL& url) {
  76   scoped_refptr<history::TopSites> local_ptr(top_sites_);
  77   if (local_ptr.get() == NULL)
  78     return;
  79
  80   // Adding
  81   BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
  82                           base::Bind(AddForcedURLOnUIThread, local_ptr, url));
  83 }
  84
  85 ThumbnailingAlgorithm* ThumbnailServiceImpl::GetThumbnailingAlgorithm()
  86     const {
  87   const gfx::Size thumbnail_size(kThumbnailWidth, kThumbnailHeight);
  88   if (use_thumbnail_retargeting_)
  89     return new ContentBasedThumbnailingAlgorithm(thumbnail_size);
  90   return new SimpleThumbnailCrop(thumbnail_size);
  91 }
  92
  93 bool ThumbnailServiceImpl::ShouldAcquirePageThumbnail(const GURL& url) {
  94   scoped_refptr<history::TopSites> local_ptr(top_sites_);
  95
  96   if (local_ptr.get() == NULL)
  97     return false;
  98
  99   // Skip if the given URL is not appropriate for history.
 100   if (!HistoryService::CanAddURL(url))
 101     return false;
 102   // Skip if the top sites list is full, and the URL is not known.
 103   if (local_ptr->IsNonForcedFull() && !local_ptr->IsKnownURL(url))
 104     return false;
 105   // Skip if we don't have to udpate the existing thumbnail.
 106   ThumbnailScore current_score;
 107   if (local_ptr->GetPageThumbnailScore(url, &current_score) &&
 108       !current_score.ShouldConsiderUpdating())
 109     return false;
 110   // Skip if we don't have to udpate the temporary thumbnail (i.e. the one
 111   // not yet saved).
 112   ThumbnailScore temporary_score;
 113   if (local_ptr->GetTemporaryPageThumbnailScore(url, &temporary_score) &&
 114       !temporary_score.ShouldConsiderUpdating())
 115     return false;
 116
 117   return true;
 118 }
 119
 120 void ThumbnailServiceImpl::ShutdownOnUIThread() {
 121   // Since each call uses its own scoped_refptr, we can just clear the reference
 122   // here by assigning null. If another call is completed, it added its own
 123   // reference.
 124   top_sites_ = NULL;
 125 }
 126
 127 }  // namespace thumbnails