Add display-resolution caching to GPU IDC

cc/tiles/gpu_image_decode_controller.cc

 // 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.
 
 #include "cc/tiles/gpu_image_decode_controller.h"
 
 #include <inttypes.h>
 
 #include "base/memory/discardable_memory_allocator.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/numerics/safe_math.h"
 #include "base/strings/stringprintf.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "cc/debug/devtools_instrumentation.h"
 #include "cc/output/context_provider.h"
 #include "cc/raster/tile_task.h"
 #include "cc/resources/resource_format_utils.h"
+#include "cc/tiles/mipmap_util.h"
 #include "gpu/command_buffer/client/context_support.h"
 #include "gpu/command_buffer/client/gles2_interface.h"
 #include "gpu_image_decode_controller.h"
 #include "skia/ext/texture_handle.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 #include "third_party/skia/include/core/SkRefCnt.h"
 #include "third_party/skia/include/core/SkSurface.h"
 #include "third_party/skia/include/gpu/GrContext.h"
 #include "third_party/skia/include/gpu/GrTexture.h"
 #include "ui/gfx/skia_util.h"
@@ skipping 12 matching lines @@
   if (std::abs(draw_image.scale().width()) <
           std::numeric_limits<float>::epsilon() ||
       std::abs(draw_image.scale().height()) <
           std::numeric_limits<float>::epsilon()) {
     return true;
   }
   return false;
 }
 
 SkImage::DeferredTextureImageUsageParams ParamsFromDrawImage(
-    const DrawImage& draw_image) {
+    const DrawImage& draw_image,
+    int upload_scale_mip_level) {
   SkImage::DeferredTextureImageUsageParams params;
   params.fMatrix = draw_image.matrix();
   params.fQuality = draw_image.filter_quality();
+  params.fPreScaleMipLevel = upload_scale_mip_level;
 
   return params;
 }
 
+// Calculate the mip level to upload-scale the image to before uploading. We use
+// mip levels rather than exact scales to increase re-use of scaled images.
+int CalculateUploadScaleMipLevel(const DrawImage& draw_image) {
+  // Images which are being clipped will have color-bleeding if scaled.
+  // TODO(ericrk): Investigate uploading clipped images to handle this case and
+  // provide further optimization. crbug.com/620899
+  if (draw_image.src_rect() != draw_image.image()->bounds()) {

[vmpstr, 2016/06/22 21:33:34]

nit: no braces

[ericrk, 2016/06/23 18:16:55]

Done.

+    return 0;
+  }
+
+  gfx::Size base_size(draw_image.image()->width(),
+                      draw_image.image()->height());
+  // Ceil our scaled size so that the mip map generated is guaranteed to be
+  // larger.
+  gfx::Size scaled_size = gfx::ScaleToCeiledSize(
+      base_size, draw_image.scale().width(), draw_image.scale().height());
+
+  return MipMapUtil::GetLevelForSize(base_size, scaled_size);
+}
+
+// Calculates the scale factor which can be used to scale an image to a given
+// mip level.
+SkSize CalculateScaleFactorForMipLevel(const DrawImage& draw_image,
+                                       int mip_level) {
+  gfx::Size base_size(draw_image.image()->width(),
+                      draw_image.image()->height());
+  return MipMapUtil::GetScaleAdjustmentForLevel(base_size, mip_level);
+}
+
+// Generates a uint64_t which uniquely identifies a DrawImage for the purposes
+// of the |in_use_cache_|. The key is generated as follows:
+//   ╔══════════════════════╤═══════════╤═══════════╗
+//   ║       image_id       │ mip_level │  quality  ║
+//   ╚════════32═bits═══════╧══16═bits══╧══16═bits══╝
+uint64_t GenerateInUseCacheKey(const DrawImage& draw_image) {
+  static_assert(
+      kLast_SkFilterQuality <= std::numeric_limits<uint16_t>::max(),
+      "InUseCacheKey depends on SkFilterQuality fitting in a uint16_t.");
+
+  SkFilterQuality filter_quality = draw_image.filter_quality();
+  DCHECK_LE(filter_quality, kLast_SkFilterQuality);
+
+  // An image has at most log_2(max(width, height)) mip levels, so given our
+  // usage of 32-bit sizes for images, key.mip_level is at most 31.
+  int32_t mip_level = CalculateUploadScaleMipLevel(draw_image);
+  DCHECK_LT(mip_level, 32);
+
+  return (static_cast<uint64_t>(draw_image.image()->uniqueID()) << 32) |
+         (mip_level << 16) | draw_image.filter_quality();
+}
+
 }  // namespace
 
+GpuImageDecodeController::InUseCacheEntry::InUseCacheEntry(
+    scoped_refptr<ImageData> image_data)
+    : image_data(std::move(image_data)) {}
+GpuImageDecodeController::InUseCacheEntry::InUseCacheEntry(
+    const InUseCacheEntry&) = default;
+GpuImageDecodeController::InUseCacheEntry::InUseCacheEntry(InUseCacheEntry&&) =
+    default;
+GpuImageDecodeController::InUseCacheEntry::~InUseCacheEntry() = default;
+
 // Task which decodes an image and stores the result in discardable memory.
 // This task does not use GPU resources and can be run on any thread.
 class ImageDecodeTaskImpl : public TileTask {
  public:
   ImageDecodeTaskImpl(GpuImageDecodeController* controller,
                       const DrawImage& draw_image,
                       const ImageDecodeController::TracingInfo& tracing_info)
       : TileTask(true),
         controller_(controller),
         image_(draw_image),
@@ skipping 99 matching lines @@
 
 void GpuImageDecodeController::DecodedImageData::ResetData() {
   DCHECK(!is_locked_);
   if (data_)
     ReportUsageStats();
   data_ = nullptr;
   usage_stats_ = UsageStats();
 }
 
 void GpuImageDecodeController::DecodedImageData::ReportUsageStats() const {
-  // lock_count | used  | result state
-  // ===========+=======+==================
-  //  1         | false | WASTED_ONCE
-  //  1         | true  | USED_ONCE
-  //  >1        | false | WASTED_RELOCKED
-  //  >1        | true  | USED_RELOCKED
+  // lock_count │ used  │ result state
+  // ═══════════╪═══════╪══════════════════
+  //  1         │ false │ WASTED_ONCE
+  //  1         │ true  │ USED_ONCE
+  //  >1        │ false │ WASTED_RELOCKED
+  //  >1        │ true  │ USED_RELOCKED
   // Note that it's important not to reorder the following enums, since the
   // numerical values are used in the histogram code.
   enum State : int {
     DECODED_IMAGE_STATE_WASTED_ONCE,
     DECODED_IMAGE_STATE_USED_ONCE,
     DECODED_IMAGE_STATE_WASTED_RELOCKED,
     DECODED_IMAGE_STATE_USED_RELOCKED,
     DECODED_IMAGE_STATE_COUNT
   } state = DECODED_IMAGE_STATE_WASTED_ONCE;
 
@@ skipping 30 matching lines @@
   image_ = std::move(image);
 }
 
 void GpuImageDecodeController::UploadedImageData::ReportUsageStats() const {
   UMA_HISTOGRAM_BOOLEAN("Renderer4.GpuImageUploadState.Used",
                         usage_stats_.used);
   UMA_HISTOGRAM_BOOLEAN("Renderer4.GpuImageUploadState.FirstRefWasted",
                         usage_stats_.first_ref_wasted);
 }
 
-GpuImageDecodeController::ImageData::ImageData(DecodedDataMode mode,
-                                               size_t size)
-    : mode(mode), size(size) {}
+GpuImageDecodeController::ImageData::ImageData(
+    DecodedDataMode mode,
+    size_t size,
+    int upload_scale_mip_level,
+    SkFilterQuality upload_scale_filter_quality)
+    : mode(mode),
+      size(size),
+      upload_scale_mip_level(upload_scale_mip_level),
+      upload_scale_filter_quality(upload_scale_filter_quality) {}
 
-GpuImageDecodeController::ImageData::~ImageData() = default;
+GpuImageDecodeController::ImageData::~ImageData() {
+  // We should never delete ImageData while it is in use or before it has been
+  // cleaned up.
+  DCHECK_EQ(0u, upload.ref_count);
+  DCHECK_EQ(0u, decode.ref_count);
+  DCHECK_EQ(false, decode.is_locked());
+  // This should always be cleaned up before deleting the image, as it needs to
+  // be freed with the GL context lock held.
+  DCHECK(!upload.image());
+}
 
 GpuImageDecodeController::GpuImageDecodeController(ContextProvider* context,
                                                    ResourceFormat decode_format,
                                                    size_t max_gpu_image_bytes)
     : format_(decode_format),
       context_(context),
-      image_data_(ImageDataMRUCache::NO_AUTO_EVICT),
+      persistent_cache_(ImageDataMRUCache::NO_AUTO_EVICT),
       cached_items_limit_(kMaxDiscardableItems),
       cached_bytes_limit_(max_gpu_image_bytes),
       bytes_used_(0),
       max_gpu_image_bytes_(max_gpu_image_bytes) {
   // Acquire the context_lock so that we can safely retrieve the
   // GrContextThreadSafeProxy. This proxy can then be used with no lock held.
   {
     ContextProvider::ScopedContextLock context_lock(context_);
     context_threadsafe_proxy_ = sk_sp<GrContextThreadSafeProxy>(
         context->GrContext()->threadSafeProxy());
@@ skipping 22 matching lines @@
     const DrawImage& draw_image,
     const TracingInfo& tracing_info,
     scoped_refptr<TileTask>* task) {
   if (SkipImage(draw_image)) {
     *task = nullptr;
     return false;
   }
 
   base::AutoLock lock(lock_);
   const auto image_id = draw_image.image()->uniqueID();
-
-  auto found = image_data_.Get(image_id);
-  if (found != image_data_.end()) {
-    ImageData* image_data = found->second.get();
+  ImageData* image_data = GetImageDataForDrawImage(draw_image);
+  if (image_data) {
     if (image_data->is_at_raster) {
       // Image is at-raster, just return, this usage will be at-raster as well.
       *task = nullptr;
       return false;
     }
 
     if (image_data->decode.decode_failure) {
       // We have already tried and failed to decode this image, so just return.
       *task = nullptr;
       return false;
     }
 
     if (image_data->upload.image()) {
       // The image is already uploaded, ref and return.
       RefImage(draw_image);
       *task = nullptr;
       return true;
     }
-  }
 
-  // We didn't have a pre-uploaded image, so we need an upload task. Try to find
-  // an existing one.
-  scoped_refptr<TileTask>& existing_task =
-      pending_image_upload_tasks_[image_id];
-  if (existing_task) {
-    // We had an existing upload task, ref the image and return the task.
-    RefImage(draw_image);
-    *task = existing_task;
-    return true;
+    // We didn't have a pre-uploaded image, so we need an upload task. Try to
+    // find an existing one.
+    if (image_data->upload.task) {
+      // We had an existing upload task, ref the image and return the task.
+      RefImage(draw_image);
+      *task = image_data->upload.task;
+      return true;
+    } else {

[vmpstr, 2016/06/22 21:33:35]

nit: remove

[ericrk, 2016/06/23 18:16:55]

Done.

+    }
   }
 
   // We will be creating a new upload task. If necessary, create a placeholder
   // ImageData to hold the result.
-  std::unique_ptr<ImageData> new_data;
-  ImageData* data;
-  if (found == image_data_.end()) {
+  scoped_refptr<ImageData> new_data;
+  if (!image_data) {

[vmpstr, 2016/06/22 21:33:35]

Can you restructure the code from 369 to 405 as:

scoped_refptr<ImageData> new_data;
ImageData* image_data = GetImageDataForDrawImage(draw_image);
if (!image_data) {
 ...
} else if (image_data->is_at_raster) {
 ...
} else if (image_data->decode.decode_failure) {
 ...
} 

etc?

[ericrk, 2016/06/23 18:16:55]

good call.

     new_data = CreateImageData(draw_image);
-    data = new_data.get();
-  } else {
-    data = found->second.get();
+    image_data = new_data.get();
   }
 
   // Ensure that the image we're about to decode/upload will fit in memory.
-  if (!EnsureCapacity(data->size)) {
+  if (!EnsureCapacity(image_data->size)) {
     // Image will not fit, do an at-raster decode.
     *task = nullptr;
     return false;
   }
 
   // If we had to create new image data, add it to our map now that we know it
   // will fit.
   if (new_data)
-    found = image_data_.Put(image_id, std::move(new_data));
+    persistent_cache_.Put(image_id, std::move(new_data));
 
   // Ref image and create a upload and decode tasks. We will release this ref
   // in UploadTaskCompleted.
   RefImage(draw_image);
-  existing_task = make_scoped_refptr(new ImageUploadTaskImpl(
+  *task = make_scoped_refptr(new ImageUploadTaskImpl(
       this, draw_image, GetImageDecodeTaskAndRef(draw_image, tracing_info),
       tracing_info));
+  image_data->upload.task = *task;
 
   // Ref the image again - this ref is owned by the caller, and it is their
   // responsibility to release it by calling UnrefImage.
   RefImage(draw_image);
-  *task = existing_task;
   return true;
 }
 
 void GpuImageDecodeController::UnrefImage(const DrawImage& draw_image) {
   base::AutoLock lock(lock_);
   UnrefImageInternal(draw_image);
 }
 
 DecodedDrawImage GpuImageDecodeController::GetDecodedImageForDraw(
     const DrawImage& draw_image) {
   // We are being called during raster. The context lock must already be
   // acquired by the caller.
   context_->GetLock()->AssertAcquired();
 
   if (SkipImage(draw_image))
     return DecodedDrawImage(nullptr, draw_image.filter_quality());
 
   TRACE_EVENT0("cc", "GpuImageDecodeController::GetDecodedImageForDraw");
 
   base::AutoLock lock(lock_);
-  const uint32_t unique_id = draw_image.image()->uniqueID();
-  auto found = image_data_.Peek(unique_id);
-  if (found == image_data_.end()) {
+  ImageData* image_data = GetImageDataForDrawImage(draw_image);
+  if (!image_data) {
     // We didn't find the image, create a new entry.
     auto data = CreateImageData(draw_image);
-    found = image_data_.Put(unique_id, std::move(data));
+    image_data = data.get();
+    persistent_cache_.Put(draw_image.image()->uniqueID(), std::move(data));
   }
 
-  ImageData* image_data = found->second.get();
-
   if (!image_data->upload.budgeted) {
     // If image data is not budgeted by this point, it is at-raster.
     image_data->is_at_raster = true;
   }
 
   // Ref the image and decode so that they stay alive while we are
   // decoding/uploading.
   RefImage(draw_image);
   RefImageDecode(draw_image);
 
   // We may or may not need to decode and upload the image we've found, the
   // following functions early-out to if we already decoded.
   DecodeImageIfNecessary(draw_image, image_data);
   UploadImageIfNecessary(draw_image, image_data);
   // Unref the image decode, but not the image. The image ref will be released
   // in DrawWithImageFinished.
   UnrefImageDecode(draw_image);
 
   sk_sp<SkImage> image = image_data->upload.image();
   image_data->upload.mark_used();
   DCHECK(image || image_data->decode.decode_failure);
 
-  DecodedDrawImage decoded_draw_image(std::move(image),
+  SkSize scale_factor = CalculateScaleFactorForMipLevel(
+      draw_image, image_data->upload_scale_mip_level);
+  DecodedDrawImage decoded_draw_image(std::move(image), SkSize(), scale_factor,
                                       draw_image.filter_quality());
   decoded_draw_image.set_at_raster_decode(image_data->is_at_raster);
   return decoded_draw_image;
 }
 
 void GpuImageDecodeController::DrawWithImageFinished(
     const DrawImage& draw_image,
     const DecodedDrawImage& decoded_draw_image) {
   // We are being called during raster. The context lock must already be
   // acquired by the caller.
   context_->GetLock()->AssertAcquired();
 
   if (SkipImage(draw_image))
     return;
 
   base::AutoLock lock(lock_);
+  TRACE_EVENT0("cc", "GpuImageDecodeController::DrawWithImageFinished");

[vmpstr, 2016/06/22 21:33:35]

nit: Move this before the early out

[ericrk, 2016/06/23 18:16:55]

I also log "GetDecodedImageForDraw" after the early out, I'll move both of
these.

   UnrefImageInternal(draw_image);
 
   // We are mid-draw and holding the context lock, ensure we clean up any
   // textures (especially at-raster), which may have just been marked for
   // deletion by UnrefImage.
   DeletePendingImages();
 }
 
 void GpuImageDecodeController::ReduceCacheUsage() {
   base::AutoLock lock(lock_);
@@ skipping 15 matching lines @@
     DeletePendingImages();
   } else {
     base::AutoLock lock(lock_);
     cached_bytes_limit_ = max_gpu_image_bytes_;
   }
 }
 
 bool GpuImageDecodeController::OnMemoryDump(
     const base::trace_event::MemoryDumpArgs& args,
     base::trace_event::ProcessMemoryDump* pmd) {
-  for (const auto& image_pair : image_data_) {
+  for (const auto& image_pair : persistent_cache_) {
     const ImageData* image_data = image_pair.second.get();
     const uint32_t image_id = image_pair.first;
 
     // If we have discardable decoded data, dump this here.
     if (image_data->decode.data()) {
       std::string discardable_dump_name = base::StringPrintf(
           "cc/image_memory/controller_0x%" PRIXPTR "/discardable/image_%d",
           reinterpret_cast<uintptr_t>(this), image_id);
       base::trace_event::MemoryAllocatorDump* dump =
           image_data->decode.data()->CreateMemoryAllocatorDump(
               discardable_dump_name.c_str(), pmd);
-
       // If our image is locked, dump the "locked_size" as an additional column.
       // This lets us see the amount of discardable which is contributing to
       // memory pressure.
       if (image_data->decode.is_locked()) {
         dump->AddScalar("locked_size",
                         base::trace_event::MemoryAllocatorDump::kUnitsBytes,
                         image_data->size);
       }
     }
 
@@ skipping 27 matching lines @@
       pmd->CreateSharedGlobalAllocatorDump(guid);
       pmd->AddOwnershipEdge(dump->guid(), guid, kImportance);
     }
   }
 
   return true;
 }
 
 void GpuImageDecodeController::DecodeImage(const DrawImage& draw_image) {
   base::AutoLock lock(lock_);
-  auto found = image_data_.Peek(draw_image.image()->uniqueID());
-  DCHECK(found != image_data_.end());
-  DCHECK(!found->second->is_at_raster);
-  DecodeImageIfNecessary(draw_image, found->second.get());
+  ImageData* image_data = GetImageDataForDrawImage(draw_image);
+  DCHECK(image_data);
+  DCHECK(!image_data->is_at_raster);
+  DecodeImageIfNecessary(draw_image, image_data);
 }
 
 void GpuImageDecodeController::UploadImage(const DrawImage& draw_image) {
   ContextProvider::ScopedContextLock context_lock(context_);
   base::AutoLock lock(lock_);
-  auto found = image_data_.Peek(draw_image.image()->uniqueID());
-  DCHECK(found != image_data_.end());
-  DCHECK(!found->second->is_at_raster);
-  UploadImageIfNecessary(draw_image, found->second.get());
+  ImageData* image_data = GetImageDataForDrawImage(draw_image);
+  DCHECK(image_data);
+  DCHECK(!image_data->is_at_raster);
+  UploadImageIfNecessary(draw_image, image_data);
 }
 
 void GpuImageDecodeController::OnImageDecodeTaskCompleted(
     const DrawImage& draw_image) {
   base::AutoLock lock(lock_);
-  // Decode task is complete, remove it from our list of pending tasks.
-  pending_image_decode_tasks_.erase(draw_image.image()->uniqueID());
+  // Decode task is complete, remove our reference to it.
+  ImageData* image_data = GetImageDataForDrawImage(draw_image);
+  DCHECK(image_data);
+  DCHECK(image_data->decode.task);
+  image_data->decode.task = nullptr;
 
   // While the decode task is active, we keep a ref on the decoded data.
   // Release that ref now.
   UnrefImageDecode(draw_image);
 }
 
 void GpuImageDecodeController::OnImageUploadTaskCompleted(
     const DrawImage& draw_image) {
   base::AutoLock lock(lock_);
-  // Upload task is complete, remove it from our list of pending tasks.
-  pending_image_upload_tasks_.erase(draw_image.image()->uniqueID());
+  // Upload task is complete, remove our reference to it.
+  ImageData* image_data = GetImageDataForDrawImage(draw_image);
+  DCHECK(image_data);
+  DCHECK(image_data->upload.task);
+  image_data->upload.task = nullptr;
 
   // While the upload task is active, we keep a ref on both the image it will be
   // populating, as well as the decode it needs to populate it. Release these
   // refs now.
   UnrefImageDecode(draw_image);
   UnrefImageInternal(draw_image);
 }
 
 // Checks if an existing image decode exists. If not, returns a task to produce
 // the requested decode.
 scoped_refptr<TileTask> GpuImageDecodeController::GetImageDecodeTaskAndRef(
     const DrawImage& draw_image,
     const TracingInfo& tracing_info) {
   lock_.AssertAcquired();
 
-  const uint32_t image_id = draw_image.image()->uniqueID();
-
   // This ref is kept alive while an upload task may need this decode. We
   // release this ref in UploadTaskCompleted.
   RefImageDecode(draw_image);
 
-  auto found = image_data_.Peek(image_id);
-  if (found != image_data_.end() && found->second->decode.is_locked()) {
+  ImageData* image_data = GetImageDataForDrawImage(draw_image);
+  DCHECK(image_data);
+  if (image_data->decode.is_locked()) {
     // We should never be creating a decode task for an at raster image.
-    DCHECK(!found->second->is_at_raster);
+    DCHECK(!image_data->is_at_raster);
     // We should never be creating a decode for an already-uploaded image.
-    DCHECK(!found->second->upload.image());
+    DCHECK(!image_data->upload.image());
     return nullptr;
   }
 
   // We didn't have an existing locked image, create a task to lock or decode.
-  scoped_refptr<TileTask>& existing_task =
-      pending_image_decode_tasks_[image_id];
+  scoped_refptr<TileTask>& existing_task = image_data->decode.task;
   if (!existing_task) {
     // Ref image decode and create a decode task. This ref will be released in
     // DecodeTaskCompleted.
     RefImageDecode(draw_image);
     existing_task = make_scoped_refptr(
         new ImageDecodeTaskImpl(this, draw_image, tracing_info));
   }
   return existing_task;
 }
 
 void GpuImageDecodeController::RefImageDecode(const DrawImage& draw_image) {
   lock_.AssertAcquired();
-  auto found = image_data_.Peek(draw_image.image()->uniqueID());
-  DCHECK(found != image_data_.end());
-  ++found->second->decode.ref_count;
-  RefCountChanged(found->second.get());
+  auto found = in_use_cache_.find(GenerateInUseCacheKey(draw_image));
+  DCHECK(found != in_use_cache_.end());
+  ++found->second.ref_count;
+  ++found->second.image_data->decode.ref_count;
+  OwnershipChanged(found->second.image_data.get());
 }
 
 void GpuImageDecodeController::UnrefImageDecode(const DrawImage& draw_image) {
   lock_.AssertAcquired();
-  auto found = image_data_.Peek(draw_image.image()->uniqueID());
-  DCHECK(found != image_data_.end());
-  DCHECK_GT(found->second->decode.ref_count, 0u);
-  --found->second->decode.ref_count;
-  RefCountChanged(found->second.get());
+  auto found = in_use_cache_.find(GenerateInUseCacheKey(draw_image));
+  DCHECK(found != in_use_cache_.end());
+  DCHECK_GT(found->second.image_data->decode.ref_count, 0u);
+  DCHECK_GT(found->second.ref_count, 0u);
+  --found->second.ref_count;
+  --found->second.image_data->decode.ref_count;
+  OwnershipChanged(found->second.image_data.get());
+  if (found->second.ref_count == 0u) {
+    in_use_cache_.erase(found);
+  }
 }
 
 void GpuImageDecodeController::RefImage(const DrawImage& draw_image) {
   lock_.AssertAcquired();
-  auto found = image_data_.Peek(draw_image.image()->uniqueID());
-  DCHECK(found != image_data_.end());
-  ++found->second->upload.ref_count;
-  RefCountChanged(found->second.get());
+  InUseCacheKey key = GenerateInUseCacheKey(draw_image);
+  auto found = in_use_cache_.find(key);
+
+  // If no secondary cache entry was found for the given |draw_image|, then
+  // the draw_image only exists in the |persistent_cache_|. Create an in-use
+  // cache entry now.
+  if (found == in_use_cache_.end()) {
+    auto found_image = persistent_cache_.Peek(draw_image.image()->uniqueID());
+    DCHECK(found_image != persistent_cache_.end());
+    DCHECK(found_image->second->upload_scale_mip_level <=
+           CalculateUploadScaleMipLevel(draw_image));
+    found =
+        in_use_cache_.emplace(key, InUseCacheEntry(found_image->second)).first;
+  }
+
+  DCHECK(found != in_use_cache_.end());
+  ++found->second.ref_count;
+  ++found->second.image_data->upload.ref_count;
+  OwnershipChanged(found->second.image_data.get());
 }
 
 void GpuImageDecodeController::UnrefImageInternal(const DrawImage& draw_image) {
   lock_.AssertAcquired();
-  auto found = image_data_.Peek(draw_image.image()->uniqueID());
-  DCHECK(found != image_data_.end());
-  DCHECK_GT(found->second->upload.ref_count, 0u);
-  --found->second->upload.ref_count;
-  if (found->second->upload.ref_count == 0)
-    found->second->upload.notify_ref_reached_zero();
-  RefCountChanged(found->second.get());
+  auto found = in_use_cache_.find(GenerateInUseCacheKey(draw_image));
+  DCHECK(found != in_use_cache_.end());
+  DCHECK_GT(found->second.image_data->upload.ref_count, 0u);
+  DCHECK_GT(found->second.ref_count, 0u);
+  --found->second.ref_count;
+  --found->second.image_data->upload.ref_count;
+  OwnershipChanged(found->second.image_data.get());
+  if (found->second.ref_count == 0u) {
+    in_use_cache_.erase(found);
+  }
 }
 
 // Called any time an image or decode ref count changes. Takes care of any
 // necessary memory budget book-keeping and cleanup.
-void GpuImageDecodeController::RefCountChanged(ImageData* image_data) {
+void GpuImageDecodeController::OwnershipChanged(ImageData* image_data) {
   lock_.AssertAcquired();
 
   bool has_any_refs =
       image_data->upload.ref_count > 0 || image_data->decode.ref_count > 0;
 
+  // Don't keep around orphaned images.
+  if (image_data->is_orphaned && !has_any_refs) {
+    images_pending_deletion_.push_back(std::move(image_data->upload.image()));
+    image_data->upload.SetImage(nullptr);
+  }
+
   // Don't keep CPU images if they are unused, these images can be recreated by
   // re-locking discardable (rather than requiring a full upload like GPU
   // images).
   if (image_data->mode == DecodedDataMode::CPU && !has_any_refs) {
     images_pending_deletion_.push_back(image_data->upload.image());
     image_data->upload.SetImage(nullptr);
   }
 
   if (image_data->is_at_raster && !has_any_refs) {
     // We have an at-raster image which has reached zero refs. If it won't fit
@@ skipping 65 matching lines @@
   lock_.AssertAcquired();
 
   if (CanFitSize(required_size) && !ExceedsPreferredCount())
     return true;
 
   // While we are over memory or preferred item capacity, we iterate through
   // our set of cached image data in LRU order. For each image, we can do two
   // things: 1) We can free the uploaded image, reducing the memory usage of
   // the cache and 2) we can remove the entry entirely, reducing the count of
   // elements in the cache.
-  for (auto it = image_data_.rbegin(); it != image_data_.rend();) {
+  for (auto it = persistent_cache_.rbegin(); it != persistent_cache_.rend();) {
     if (it->second->decode.ref_count != 0 ||
         it->second->upload.ref_count != 0) {
       ++it;
       continue;
     }
 
     // Current entry has no refs. Ensure it is not locked.
     DCHECK(!it->second->decode.is_locked());
 
     // If an image without refs is budgeted, it must have an associated image
     // upload.
     DCHECK(!it->second->upload.budgeted || it->second->upload.image());
 
     // Free the uploaded image if possible.
     if (it->second->upload.image()) {
       DCHECK(it->second->upload.budgeted);
       DCHECK_GE(bytes_used_, it->second->size);
       bytes_used_ -= it->second->size;
       images_pending_deletion_.push_back(it->second->upload.image());
       it->second->upload.SetImage(nullptr);
       it->second->upload.budgeted = false;
     }
 
     // Free the entire entry if necessary.
     if (ExceedsPreferredCount()) {
-      it = image_data_.Erase(it);
+      it = persistent_cache_.Erase(it);
     } else {
       ++it;
     }
 
     if (CanFitSize(required_size) && !ExceedsPreferredCount())
       return true;
   }
 
   // Preferred count is only used as a guideline when triming the cache. Allow
   // new elements to be added as long as we are below our size limit.
   return CanFitSize(required_size);
 }
 
 bool GpuImageDecodeController::CanFitSize(size_t size) const {
   lock_.AssertAcquired();
 
   base::CheckedNumeric<uint32_t> new_size(bytes_used_);
   new_size += size;
   return new_size.IsValid() && new_size.ValueOrDie() <= cached_bytes_limit_;
 }
 
 bool GpuImageDecodeController::ExceedsPreferredCount() const {
   lock_.AssertAcquired();
 
-  return image_data_.size() > cached_items_limit_;
+  return persistent_cache_.size() > cached_items_limit_;
 }
 
 void GpuImageDecodeController::DecodeImageIfNecessary(
     const DrawImage& draw_image,
     ImageData* image_data) {
   lock_.AssertAcquired();
 
   DCHECK_GT(image_data->decode.ref_count, 0u);
 
   if (image_data->decode.decode_failure) {
@@ skipping 16 matching lines @@
 
   image_data->decode.ResetData();
   std::unique_ptr<base::DiscardableMemory> backing_memory;
   {
     base::AutoUnlock unlock(lock_);
     switch (image_data->mode) {
       case DecodedDataMode::CPU: {
         backing_memory =
             base::DiscardableMemoryAllocator::GetInstance()
                 ->AllocateLockedDiscardableMemory(image_data->size);
-        SkImageInfo image_info = CreateImageInfoForDrawImage(draw_image);
-        if (!draw_image.image()->readPixels(image_info, backing_memory->data(),
-                                            image_info.minRowBytes(), 0, 0,
-                                            SkImage::kDisallow_CachingHint)) {
+        SkImageInfo image_info = CreateImageInfoForDrawImage(
+            draw_image, image_data->upload_scale_mip_level);
+        // In order to match GPU scaling quality (which uses mip-maps at high
+        // quality), we want to use at most medium filter quality for the
+        // scale.
+        SkFilterQuality scale_quality =
+            std::max(kMedium_SkFilterQuality, draw_image.filter_quality());

[vmpstr, 2016/06/22 21:33:34]

Based on the comment, this should be std::min

[ericrk, 2016/06/23 18:16:55]

definitely :D - Re-structured the code a bit to put this logic in a common
function that can be used multiple places. Also added a unit test that tests one
of these (others are difficult to unit test).

+        SkPixmap image_pixmap(image_info, backing_memory->data(),
+                              image_info.minRowBytes());
+        // Note that scalePixels falls back to readPixels if the sale is 1x, so
+        // no need to special case that as an optimization.
+        if (!draw_image.image()->scalePixels(image_pixmap, scale_quality,
+                                             SkImage::kDisallow_CachingHint)) {
           backing_memory.reset();
         }
         break;
       }
       case DecodedDataMode::GPU: {
         backing_memory =
             base::DiscardableMemoryAllocator::GetInstance()
                 ->AllocateLockedDiscardableMemory(image_data->size);
-        auto params = ParamsFromDrawImage(draw_image);
+        auto params =
+            ParamsFromDrawImage(draw_image, image_data->upload_scale_mip_level);
         if (!draw_image.image()->getDeferredTextureImageData(
                 *context_threadsafe_proxy_.get(), &params, 1,
                 backing_memory->data())) {
           backing_memory.reset();
         }
         break;
       }
     }
   }
 
@@ skipping 35 matching lines @@
   // We are about to upload a new image and are holding the context lock.
   // Ensure that any images which have been marked for deletion are actually
   // cleaned up so we don't exceed our memory limit during this upload.
   DeletePendingImages();
 
   sk_sp<SkImage> uploaded_image;
   {
     base::AutoUnlock unlock(lock_);
     switch (image_data->mode) {
       case DecodedDataMode::CPU: {
-        SkImageInfo image_info = CreateImageInfoForDrawImage(draw_image);
+        SkImageInfo image_info = CreateImageInfoForDrawImage(
+            draw_image, image_data->upload_scale_mip_level);
         SkPixmap pixmap(image_info, image_data->decode.data()->data(),
                         image_info.minRowBytes());
         uploaded_image =
             SkImage::MakeFromRaster(pixmap, [](const void*, void*) {}, nullptr);
         break;
       }
       case DecodedDataMode::GPU: {
         uploaded_image = SkImage::MakeFromDeferredTextureImageData(
             context_->GrContext(), image_data->decode.data()->data(),
             SkBudgeted::kNo);
         break;
       }
     }
   }
   image_data->decode.mark_used();
   DCHECK(uploaded_image);
 
   // At-raster may have decoded this while we were unlocked. If so, ignore our
   // result.
   if (!image_data->upload.image())
     image_data->upload.SetImage(std::move(uploaded_image));
 }
 
-std::unique_ptr<GpuImageDecodeController::ImageData>
+scoped_refptr<GpuImageDecodeController::ImageData>
 GpuImageDecodeController::CreateImageData(const DrawImage& draw_image) {
   lock_.AssertAcquired();
 
   DecodedDataMode mode;
-  SkImageInfo info = CreateImageInfoForDrawImage(draw_image);
+  int upload_scale_mip_level = CalculateUploadScaleMipLevel(draw_image);
   SkImage::DeferredTextureImageUsageParams params =
-      ParamsFromDrawImage(draw_image);
+      ParamsFromDrawImage(draw_image, upload_scale_mip_level);
   size_t data_size = draw_image.image()->getDeferredTextureImageData(
       *context_threadsafe_proxy_.get(), &params, 1, nullptr);
 
   if (data_size == 0) {
     // Can't upload image, too large or other failure. Try to use SW fallback.
-    data_size = info.getSafeSize(info.minRowBytes());
+    SkImageInfo image_info =
+        CreateImageInfoForDrawImage(draw_image, upload_scale_mip_level);
+    data_size = image_info.getSafeSize(image_info.minRowBytes());
     mode = DecodedDataMode::CPU;
   } else {
     mode = DecodedDataMode::GPU;
   }
 
-  return base::WrapUnique(new ImageData(mode, data_size));
+  return make_scoped_refptr(new ImageData(
+      mode, data_size, upload_scale_mip_level, draw_image.filter_quality()));
 }
 
 void GpuImageDecodeController::DeletePendingImages() {
   context_->GetLock()->AssertAcquired();
   lock_.AssertAcquired();
   images_pending_deletion_.clear();
 }
 
 SkImageInfo GpuImageDecodeController::CreateImageInfoForDrawImage(
+    const DrawImage& draw_image,
+    int upload_scale_mip_level) const {
+  DrawImage scaled_draw_image = draw_image.ApplyScale(
+      CalculateScaleFactorForMipLevel(draw_image, upload_scale_mip_level));
+  return SkImageInfo::Make(
+      scaled_draw_image.image()->width() * scaled_draw_image.scale().width(),
+      scaled_draw_image.image()->height() * scaled_draw_image.scale().height(),
+      ResourceFormatToClosestSkColorType(format_), kPremul_SkAlphaType);
+}
+
+// Tries to find an ImageData that can be used to draw the provided
+// |draw_image|. First looks for an exact entry in our |in_use_cache_|. If one
+// cannot be found, it looks for a compatible entry in our |persistent_cache_|.
+GpuImageDecodeController::ImageData*
+GpuImageDecodeController::GetImageDataForDrawImage(
+    const DrawImage& draw_image) {
+  lock_.AssertAcquired();
+  auto found_in_use = in_use_cache_.find(GenerateInUseCacheKey(draw_image));
+  if (found_in_use != in_use_cache_.end())
+    return found_in_use->second.image_data.get();
+
+  auto found_persistent = persistent_cache_.Get(draw_image.image()->uniqueID());
+  if (found_persistent != persistent_cache_.end()) {
+    ImageData* image_data = found_persistent->second.get();
+    if (IsCompatibleWithDrawImage(image_data, draw_image)) {
+      return image_data;
+    } else {
+      found_persistent->second->is_orphaned = true;
+      // Call OwnershipChanged before erasing the orphaned task from the
+      // persistent cache. This ensures that if the orphaned task has 0
+      // references, it is cleaned up safely before it is deleted.
+      OwnershipChanged(image_data);
+      persistent_cache_.Erase(found_persistent);
+    }
+  }
+
+  return nullptr;
+}
+
+// Determines if we can draw the provided |draw_image| using the provided
+// |image_data|. This is true if the |image_data| is not scaled, or if it
+// is scaled at an equal or larger scale and equal or larger quality to
+// the provided |draw_image|.
+bool GpuImageDecodeController::IsCompatibleWithDrawImage(
+    const ImageData* image_data,
     const DrawImage& draw_image) const {
-  return SkImageInfo::Make(
-      draw_image.image()->width(), draw_image.image()->height(),
-      ResourceFormatToClosestSkColorType(format_), kPremul_SkAlphaType);
+  bool is_scaled = image_data->upload_scale_mip_level != 0;
+  bool scale_is_compatible = CalculateUploadScaleMipLevel(draw_image) >=
+                             image_data->upload_scale_mip_level;
+  bool quality_is_compatible =
+      draw_image.filter_quality() <= image_data->upload_scale_filter_quality;
+  return !is_scaled || (scale_is_compatible && quality_is_compatible);
+}
+
+size_t GpuImageDecodeController::GetDrawImageSizeForTesting(
+    const DrawImage& image) {
+  base::AutoLock lock(lock_);
+  scoped_refptr<ImageData> data = CreateImageData(image);
+  return data->size;
 }
 
 void GpuImageDecodeController::SetImageDecodingFailedForTesting(
     const DrawImage& image) {
   base::AutoLock lock(lock_);
-  auto found = image_data_.Peek(image.image()->uniqueID());
-  DCHECK(found != image_data_.end());
+  auto found = persistent_cache_.Peek(image.image()->uniqueID());
+  DCHECK(found != persistent_cache_.end());
   ImageData* image_data = found->second.get();
   image_data->decode.decode_failure = true;
 }
 
 bool GpuImageDecodeController::DiscardableIsLockedForTesting(
     const DrawImage& image) {
   base::AutoLock lock(lock_);
-  auto found = image_data_.Peek(image.image()->uniqueID());
-  DCHECK(found != image_data_.end());
+  auto found = persistent_cache_.Peek(image.image()->uniqueID());
+  DCHECK(found != persistent_cache_.end());
   ImageData* image_data = found->second.get();
   return image_data->decode.is_locked();
 }
 
 }  // namespace cc