cc: Add color space to image decode caches

cc/tiles/gpu_image_decode_cache.h

 // Copyright 2016 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.
 
 #ifndef CC_TILES_GPU_IMAGE_DECODE_CACHE_H_
 #define CC_TILES_GPU_IMAGE_DECODE_CACHE_H_
 
 #include <memory>
 #include <unordered_map>
 #include <vector>
@@ skipping 214 matching lines @@
     void ReportUsageStats() const;
 
     // May be null if image not yet uploaded / prepared.
     sk_sp<SkImage> image_;
     UsageStats usage_stats_;
   };
 
   struct ImageData : public base::RefCounted<ImageData> {
     ImageData(DecodedDataMode mode,
               size_t size,
+              const gfx::ColorSpace& target_color_space,
               const SkImage::DeferredTextureImageUsageParams& upload_params);
 
     const DecodedDataMode mode;
     const size_t size;
+    gfx::ColorSpace target_color_space;
     bool is_at_raster = false;
     SkImage::DeferredTextureImageUsageParams upload_params;
 
     // If true, this image is no longer in our |persistent_cache_| and will be
     // deleted as soon as its ref count reaches zero.
     bool is_orphaned = false;
 
     DecodedImageData decode;
     UploadedImageData upload;
 
    private:
     friend class base::RefCounted<ImageData>;
     ~ImageData();
   };
 
   // A ref-count and ImageData, used to associate the ImageData with a specific
   // DrawImage in the |in_use_cache_|.
   struct InUseCacheEntry {
     explicit InUseCacheEntry(scoped_refptr<ImageData> image_data);
     InUseCacheEntry(const InUseCacheEntry& other);
     InUseCacheEntry(InUseCacheEntry&& other);
     ~InUseCacheEntry();
 
     uint32_t ref_count = 0;
     scoped_refptr<ImageData> image_data;
   };
 
   // Uniquely identifies (without collisions) a specific DrawImage for use in
   // the |in_use_cache_|.
-  using InUseCacheKey = uint64_t;
+  struct InUseCacheKeyHash;
+  struct InUseCacheKey {
+    static InUseCacheKey FromDrawImage(const DrawImage& draw_image);
+    bool operator==(const InUseCacheKey& other) const;
+
+   private:
+    friend struct GpuImageDecodeCache::InUseCacheKeyHash;
+    explicit InUseCacheKey(const DrawImage& draw_image);
+
+    uint32_t image_id;
+    int mip_level;
+    SkFilterQuality filter_quality;
+    gfx::ColorSpace target_color_space;
+  };
+  struct InUseCacheKeyHash {
+    size_t operator()(const InUseCacheKey&) const;
+  };
+  InUseCacheKey FromDrawImage(const DrawImage& draw_image);
 
   // All private functions should only be called while holding |lock_|. Some
   // functions also require the |context_| lock. These are indicated by
   // additional comments.
 
   // Similar to GetTaskForImageAndRef, but gets the dependent decode task
   // rather than the upload task, if necessary.
   scoped_refptr<TileTask> GetImageDecodeTaskAndRef(
       const DrawImage& image,
       const TracingInfo& tracing_info,
@@ skipping 51 matching lines @@
   // All members below this point must only be accessed while holding |lock_|.
   base::Lock lock_;
 
   // |persistent_cache_| represents the long-lived cache, keeping a certain
   // budget of ImageDatas alive even when their ref count reaches zero.
   using PersistentCache = base::MRUCache<uint32_t, scoped_refptr<ImageData>>;
   PersistentCache persistent_cache_;
 
   // |in_use_cache_| represents the in-use (short-lived) cache. Entries are
   // cleaned up as soon as their ref count reaches zero.
-  using InUseCache = std::unordered_map<InUseCacheKey, InUseCacheEntry>;
+  using InUseCache =
+      std::unordered_map<InUseCacheKey, InUseCacheEntry, InUseCacheKeyHash>;
   InUseCache in_use_cache_;
 
   size_t max_working_set_bytes_;
   const size_t normal_max_cache_bytes_;
   size_t cached_bytes_limit_ = normal_max_cache_bytes_;
   size_t bytes_used_ = 0;
   base::MemoryState memory_state_ = base::MemoryState::NORMAL;
 
   // We can't release GPU backed SkImages without holding the context lock,
   // so we add them to this list and defer deletion until the next time the lock
   // is held.
   std::vector<sk_sp<SkImage>> images_pending_deletion_;
 };
 
 }  // namespace cc
 
 #endif  // CC_TILES_GPU_IMAGE_DECODE_CACHE_H_