Index: components/suggestions/image_manager.cc |
diff --git a/components/suggestions/image_manager.cc b/components/suggestions/image_manager.cc |
deleted file mode 100644 |
index 197c11fc6bd3753d1943c50eca4dafb8198de8e8..0000000000000000000000000000000000000000 |
--- a/components/suggestions/image_manager.cc |
+++ /dev/null |
@@ -1,228 +0,0 @@ |
-// Copyright 2014 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 "components/suggestions/image_manager.h" |
- |
-#include "base/bind.h" |
-#include "components/suggestions/image_fetcher.h" |
-#include "ui/gfx/codec/jpeg_codec.h" |
- |
-using leveldb_proto::ProtoDatabase; |
- |
-namespace { |
- |
-// From JPEG-encoded bytes to SkBitmap. |
-SkBitmap* DecodeImage(const std::vector<unsigned char>& encoded_data) { |
- return gfx::JPEGCodec::Decode(&encoded_data[0], encoded_data.size()); |
-} |
- |
-} // namespace |
- |
-namespace suggestions { |
- |
-ImageManager::ImageManager() : weak_ptr_factory_(this) {} |
- |
-ImageManager::ImageManager(scoped_ptr<ImageFetcher> image_fetcher, |
- scoped_ptr<ProtoDatabase<ImageData> > database, |
- const base::FilePath& database_dir) |
- : image_fetcher_(image_fetcher.Pass()), |
- database_(database.Pass()), |
- database_ready_(false), |
- weak_ptr_factory_(this) { |
- image_fetcher_->SetImageFetcherDelegate(this); |
- database_->Init(database_dir, base::Bind(&ImageManager::OnDatabaseInit, |
- weak_ptr_factory_.GetWeakPtr())); |
-} |
- |
-ImageManager::~ImageManager() {} |
- |
-ImageManager::ImageCacheRequest::ImageCacheRequest() {} |
- |
-ImageManager::ImageCacheRequest::~ImageCacheRequest() {} |
- |
-void ImageManager::Initialize(const SuggestionsProfile& suggestions) { |
- image_url_map_.clear(); |
- for (int i = 0; i < suggestions.suggestions_size(); ++i) { |
- const ChromeSuggestion& suggestion = suggestions.suggestions(i); |
- if (suggestion.has_thumbnail()) { |
- image_url_map_[GURL(suggestion.url())] = GURL(suggestion.thumbnail()); |
- } |
- } |
-} |
- |
-void ImageManager::GetImageForURL( |
- const GURL& url, |
- base::Callback<void(const GURL&, const SkBitmap*)> callback) { |
- DCHECK(thread_checker_.CalledOnValidThread()); |
- // If |url| is not found in |image_url_map_|, then invoke |callback| with |
- // NULL since there is no associated image for this |url|. |
- GURL image_url; |
- if (!GetImageURL(url, &image_url)) { |
- callback.Run(url, NULL); |
- return; |
- } |
- |
- // |database_| can be NULL if something went wrong in initialization. |
- if (database_.get() && !database_ready_) { |
- // Once database is initialized, it will serve pending requests from either |
- // cache or network. |
- QueueCacheRequest(url, image_url, callback); |
- return; |
- } |
- |
- ServeFromCacheOrNetwork(url, image_url, callback); |
-} |
- |
-void ImageManager::OnImageFetched(const GURL& url, const SkBitmap* bitmap) { |
- SaveImage(url, *bitmap); |
-} |
- |
-bool ImageManager::GetImageURL(const GURL& url, GURL* image_url) { |
- std::map<GURL, GURL>::iterator it = image_url_map_.find(url); |
- if (it == image_url_map_.end()) return false; // Not found. |
- *image_url = it->second; |
- return true; |
-} |
- |
-void ImageManager::QueueCacheRequest( |
- const GURL& url, const GURL& image_url, |
- base::Callback<void(const GURL&, const SkBitmap*)> callback) { |
- // To be served when the database has loaded. |
- ImageCacheRequestMap::iterator it = pending_cache_requests_.find(url); |
- if (it == pending_cache_requests_.end()) { |
- ImageCacheRequest request; |
- request.url = url; |
- request.image_url = image_url; |
- request.callbacks.push_back(callback); |
- pending_cache_requests_[url] = request; |
- } else { |
- // Request already queued for this url. |
- it->second.callbacks.push_back(callback); |
- } |
-} |
- |
-void ImageManager::ServeFromCacheOrNetwork( |
- const GURL& url, const GURL& image_url, |
- base::Callback<void(const GURL&, const SkBitmap*)> callback) { |
- // If there is a image available in memory, return it. |
- if (!ServeFromCache(url, callback)) { |
- image_fetcher_->StartOrQueueNetworkRequest(url, image_url, callback); |
- } |
-} |
- |
-bool ImageManager::ServeFromCache( |
- const GURL& url, |
- base::Callback<void(const GURL&, const SkBitmap*)> callback) { |
- SkBitmap* bitmap = GetBitmapFromCache(url); |
- if (bitmap) { |
- callback.Run(url, bitmap); |
- return true; |
- } |
- return false; |
-} |
- |
-SkBitmap* ImageManager::GetBitmapFromCache(const GURL& url) { |
- ImageMap::iterator image_iter = image_map_.find(url.spec()); |
- if (image_iter != image_map_.end()) { |
- return &image_iter->second; |
- } |
- return NULL; |
-} |
- |
-void ImageManager::SaveImage(const GURL& url, const SkBitmap& bitmap) { |
- // Update the image map. |
- image_map_.insert(std::make_pair(url.spec(), bitmap)); |
- |
- if (!database_ready_) return; |
- |
- // Attempt to save a JPEG representation to the database. If not successful, |
- // the fetched bitmap will still be inserted in the cache, above. |
- std::vector<unsigned char> encoded_data; |
- if (EncodeImage(bitmap, &encoded_data)) { |
- // Save the resulting bitmap to the database. |
- ImageData data; |
- data.set_url(url.spec()); |
- data.set_data(std::string(encoded_data.begin(), encoded_data.end())); |
- scoped_ptr<ProtoDatabase<ImageData>::KeyEntryVector> entries_to_save( |
- new ProtoDatabase<ImageData>::KeyEntryVector()); |
- scoped_ptr<std::vector<std::string> > keys_to_remove( |
- new std::vector<std::string>()); |
- entries_to_save->push_back(std::make_pair(data.url(), data)); |
- database_->UpdateEntries(entries_to_save.Pass(), keys_to_remove.Pass(), |
- base::Bind(&ImageManager::OnDatabaseSave, |
- weak_ptr_factory_.GetWeakPtr())); |
- } |
-} |
- |
-void ImageManager::OnDatabaseInit(bool success) { |
- if (!success) { |
- DVLOG(1) << "Image database init failed."; |
- database_.reset(); |
- ServePendingCacheRequests(); |
- return; |
- } |
- database_->LoadEntries(base::Bind(&ImageManager::OnDatabaseLoad, |
- weak_ptr_factory_.GetWeakPtr())); |
-} |
- |
-void ImageManager::OnDatabaseLoad(bool success, |
- scoped_ptr<ImageDataVector> entries) { |
- if (!success) { |
- DVLOG(1) << "Image database load failed."; |
- database_.reset(); |
- ServePendingCacheRequests(); |
- return; |
- } |
- database_ready_ = true; |
- |
- LoadEntriesInCache(entries.Pass()); |
- ServePendingCacheRequests(); |
-} |
- |
-void ImageManager::OnDatabaseSave(bool success) { |
- if (!success) { |
- DVLOG(1) << "Image database save failed."; |
- database_.reset(); |
- database_ready_ = false; |
- } |
-} |
- |
-void ImageManager::LoadEntriesInCache(scoped_ptr<ImageDataVector> entries) { |
- for (ImageDataVector::iterator it = entries->begin(); it != entries->end(); |
- ++it) { |
- std::vector<unsigned char> encoded_data(it->data().begin(), |
- it->data().end()); |
- |
- scoped_ptr<SkBitmap> bitmap(DecodeImage(encoded_data)); |
- if (bitmap.get()) { |
- image_map_.insert(std::make_pair(it->url(), *bitmap)); |
- } |
- } |
-} |
- |
-void ImageManager::ServePendingCacheRequests() { |
- for (ImageCacheRequestMap::iterator it = pending_cache_requests_.begin(); |
- it != pending_cache_requests_.end(); ++it) { |
- const ImageCacheRequest& request = it->second; |
- for (CallbackVector::const_iterator callback_it = request.callbacks.begin(); |
- callback_it != request.callbacks.end(); ++callback_it) { |
- ServeFromCacheOrNetwork(request.url, request.image_url, *callback_it); |
- } |
- } |
-} |
- |
-// static |
-bool ImageManager::EncodeImage(const SkBitmap& bitmap, |
- std::vector<unsigned char>* dest) { |
- SkAutoLockPixels bitmap_lock(bitmap); |
- if (!bitmap.readyToDraw() || bitmap.isNull()) { |
- return false; |
- } |
- return gfx::JPEGCodec::Encode( |
- reinterpret_cast<unsigned char*>(bitmap.getAddr32(0, 0)), |
- gfx::JPEGCodec::FORMAT_SkBitmap, bitmap.width(), bitmap.height(), |
- bitmap.width() * bitmap.bytesPerPixel(), 100, dest); |
-} |
- |
-} // namespace suggestions |