src/chrome/browser/android/thumbnail/thumbnail_store.h

   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
   3 // found in the LICENSE file.
   4
   5 #ifndef CHROME_BROWSER_ANDROID_THUMBNAIL_THUMBNAIL_STORE_H_
   6 #define CHROME_BROWSER_ANDROID_THUMBNAIL_THUMBNAIL_STORE_H_
   7
   8 #include <list>
   9 #include <set>
  10 #include <string>
  11
  12 #include "base/bind.h"
  13 #include "base/containers/hash_tables.h"
  14 #include "base/files/file_path.h"
  15 #include "base/memory/ref_counted.h"
  16 #include "base/observer_list.h"
  17 #include "base/time/time.h"
  18 #include "chrome/browser/android/thumbnail/scoped_ptr_expiring_cache.h"
  19 #include "chrome/browser/android/thumbnail/thumbnail.h"
  20 #include "content/public/browser/android/ui_resource_client_android.h"
  21 #include "content/public/browser/android/ui_resource_provider.h"
  22 #include "third_party/skia/include/core/SkBitmap.h"
  23 #include "ui/gfx/geometry/point.h"
  24 #include "ui/gfx/geometry/size.h"
  25 #include "ui/gfx/geometry/size_f.h"
  26 #include "url/gurl.h"
  27
  28 namespace base {
  29 class File;
  30 class Time;
  31 }
  32
  33 namespace content {
  34 class ContentViewCore;
  35 }
  36
  37 typedef std::list<TabId> TabIdList;
  38
  39 class ThumbnailStoreObserver {
  40  public:
  41   virtual void OnFinishedThumbnailRead(TabId tab_id) = 0;
  42 };
  43
  44 class ThumbnailStore : ThumbnailDelegate {
  45  public:
  46   ThumbnailStore(const std::string& disk_cache_path_str,
  47                  size_t default_cache_size,
  48                  size_t approximation_cache_size,
  49                  size_t compression_queue_max_size,
  50                  size_t write_queue_max_size,
  51                  bool use_approximation_thumbnail);
  52
  53   virtual ~ThumbnailStore();
  54
  55   void SetUIResourceProvider(content::UIResourceProvider* ui_resource_provider);
  56
  57   void AddThumbnailStoreObserver(ThumbnailStoreObserver* observer);
  58   void RemoveThumbnailStoreObserver(
  59       ThumbnailStoreObserver* observer);
  60
  61   void Put(TabId tab_id, const SkBitmap& bitmap, float thumbnail_scale);
  62   void Remove(TabId tab_id);
  63   Thumbnail* Get(TabId tab_id, bool force_disk_read);
  64
  65   void RemoveFromDiskAtAndAboveId(TabId min_id);
  66   void InvalidateThumbnailIfChanged(TabId tab_id, const GURL& url);
  67   bool CheckAndUpdateThumbnailMetaData(TabId tab_id, const GURL& url);
  68   void UpdateVisibleIds(const TabIdList& priority);
  69
  70   // ThumbnailDelegate implementation
  71   virtual void InvalidateCachedThumbnail(Thumbnail* thumbnail) OVERRIDE;
  72
  73  private:
  74   class ThumbnailMetaData {
  75    public:
  76     ThumbnailMetaData();
  77     ThumbnailMetaData(const base::Time& current_time, const GURL& url);
  78     const GURL& url() const { return url_; }
  79     base::Time capture_time() const { return capture_time_; }
  80
  81    private:
  82     base::Time capture_time_;
  83     GURL url_;
  84   };
  85
  86   typedef ScopedPtrExpiringCache<TabId, Thumbnail> ExpiringThumbnailCache;
  87   typedef base::hash_map<TabId, ThumbnailMetaData> ThumbnailMetaDataMap;
  88
  89   void RemoveFromDisk(TabId tab_id);
  90   static void RemoveFromDiskTask(const base::FilePath& file_path);
  91   static void RemoveFromDiskAtAndAboveIdTask(const base::FilePath& dir_path,
  92                                              TabId min_id);
  93   void WriteThumbnailIfNecessary(TabId tab_id,
  94                                  skia::RefPtr<SkPixelRef> compressed_data,
  95                                  float scale,
  96                                  const gfx::Size& content_size);
  97   void CompressThumbnailIfNecessary(TabId tab_id,
  98                                     const base::Time& time_stamp,
  99                                     const SkBitmap& bitmap,
 100                                     float scale);
 101   void ReadNextThumbnail();
 102   void MakeSpaceForNewItemIfNecessary(TabId tab_id);
 103   void RemoveFromReadQueue(TabId tab_id);
 104   base::FilePath GetFilePath(TabId tab_id) const;
 105   static void WriteTask(const base::FilePath& file_path,
 106                         skia::RefPtr<SkPixelRef> compressed_data,
 107                         float scale,
 108                         const gfx::Size& content_size,
 109                         const base::Callback<void()>& post_write_task);
 110   void PostWriteTask();
 111   static void CompressionTask(
 112       SkBitmap raw_data,
 113       const base::Callback<void(skia::RefPtr<SkPixelRef>, const gfx::Size&)>&
 114           post_compression_task);
 115   void PostCompressionTask(TabId tab_id,
 116                            const base::Time& time_stamp,
 117                            float scale,
 118                            skia::RefPtr<SkPixelRef> compressed_data,
 119                            const gfx::Size& content_size);
 120   static void ReadTask(
 121       const base::FilePath& file_path,
 122       const base::Callback<
 123           void(skia::RefPtr<SkPixelRef>, float, const gfx::Size&)>&
 124           post_read_task);
 125   void PostReadTask(TabId tab_id,
 126                     skia::RefPtr<SkPixelRef> compressed_data,
 127                     float scale,
 128                     const gfx::Size& content_size);
 129   void NotifyObserversOfThumbnailRead(TabId tab_id);
 130   void RemoveOnMatchedTimeStamp(TabId tab_id, const base::Time& time_stamp);
 131   static std::pair<SkBitmap, float> CreateApproximation(const SkBitmap& bitmap,
 132                                                         float scale);
 133
 134   const base::FilePath disk_cache_path_;
 135   const size_t compression_queue_max_size_;
 136   const size_t write_queue_max_size_;
 137   const bool use_approximation_thumbnail_;
 138
 139   size_t compression_tasks_count_;
 140   size_t write_tasks_count_;
 141   bool read_in_progress_;
 142
 143   ExpiringThumbnailCache cache_;
 144   ExpiringThumbnailCache approximation_cache_;
 145   ObserverList<ThumbnailStoreObserver> observers_;
 146   ThumbnailMetaDataMap thumbnail_meta_data_;
 147   TabIdList read_queue_;
 148   TabIdList visible_ids_;
 149
 150   content::UIResourceProvider* ui_resource_provider_;
 151
 152   base::WeakPtrFactory<ThumbnailStore> weak_factory_;
 153
 154   DISALLOW_COPY_AND_ASSIGN(ThumbnailStore);
 155 };
 156
 157 #endif  // CHROME_BROWSER_ANDROID_THUMBNAIL_THUMBNAIL_STORE_H_