cc: Add image decode control in the compositor.

cc/tiles/image_decode_controller_unittest.cc

+// Copyright 2015 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/image_decode_controller.h"
+
+#include "cc/playback/draw_image.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace cc {
+namespace {
+
+skia::RefPtr<SkImage> CreateImage(int width, int height) {
+  SkImageInfo info = SkImageInfo::MakeN32Premul(width, height);
+  SkBitmap bitmap;
+  bitmap.allocPixels(info);
+  return skia::AdoptRef(SkImage::NewFromBitmap(bitmap));
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyLowQuality) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality qualities[] = {kNone_SkFilterQuality, kLow_SkFilterQuality};
+  for (auto quality : qualities) {
+    DrawImage draw_image(
+        image.get(), SkIRect::MakeWH(image->width(), image->height()),
+        SkSize::Make(0.5f, 1.5f), quality, has_perspective, is_decomposable);
+
+    auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+    EXPECT_EQ(image->uniqueID(), key.image_id());
+    EXPECT_EQ(quality, key.filter_quality());
+    EXPECT_EQ(50, key.target_size().width());
+    EXPECT_EQ(150, key.target_size().height());
+    EXPECT_EQ(50u * 150u * 4u, key.target_bytes());
+  }
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyMediumQuality) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kMedium_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 1.5f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(quality, key.filter_quality());
+  EXPECT_EQ(50, key.target_size().width());
+  EXPECT_EQ(150, key.target_size().height());
+  EXPECT_EQ(50u * 150u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyMediumQualityEvenWithPerspective) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = true;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kMedium_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 1.5f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(quality, key.filter_quality());
+  EXPECT_EQ(50, key.target_size().width());
+  EXPECT_EQ(150, key.target_size().height());
+  EXPECT_EQ(50u * 150u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyMediumQualityDropToLowIfEnlarging) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kMedium_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(1.5f, 1.5f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(150, key.target_size().width());
+  EXPECT_EQ(150, key.target_size().height());
+  EXPECT_EQ(150u * 150u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyMediumQualityDropToLowIfIdentity) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kMedium_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(1.f, 1.f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(100, key.target_size().width());
+  EXPECT_EQ(100, key.target_size().height());
+  EXPECT_EQ(100u * 100u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest,
+     ImageKeyMediumQualityDropToLowIfNearlyIdentity) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kMedium_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(1.001f, 1.001f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(100, key.target_size().width());
+  EXPECT_EQ(100, key.target_size().height());
+  EXPECT_EQ(100u * 100u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest,
+     ImageKeyMediumQualityDropToLowIfNearlyIdentity2) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kMedium_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.999f, 0.999f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(100, key.target_size().width());
+  EXPECT_EQ(100, key.target_size().height());
+  EXPECT_EQ(100u * 100u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest,
+     ImageKeyMediumQualityDropToLowIfNotDecomposable) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = false;
+  SkFilterQuality quality = kMedium_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 1.5f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(50, key.target_size().width());
+  EXPECT_EQ(150, key.target_size().height());
+  EXPECT_EQ(50u * 150u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyHighQuality) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 1.5f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(quality, key.filter_quality());
+  EXPECT_EQ(50, key.target_size().width());
+  EXPECT_EQ(150, key.target_size().height());
+  EXPECT_EQ(50u * 150u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest,
+     ImageKeyHighQualityDropToMediumWithPerspective) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = true;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 1.5f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kMedium_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(50, key.target_size().width());
+  EXPECT_EQ(150, key.target_size().height());
+  EXPECT_EQ(50u * 150u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyHighQualityDropToMediumIfTooLarge) {
+  // Just over 64MB when scaled.
+  skia::RefPtr<SkImage> image = CreateImage(4555, 2048);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  // At least one dimension should scale down, so that medium quality doesn't
+  // become low.
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.9f, 2.f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kMedium_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(4100, key.target_size().width());
+  EXPECT_EQ(4096, key.target_size().height());
+  EXPECT_EQ(4100u * 4096u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyHighQualityDropToLowIfNotDecomposable) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = false;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 1.5f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(50, key.target_size().width());
+  EXPECT_EQ(150, key.target_size().height());
+  EXPECT_EQ(50u * 150u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyHighQualityDropToLowIfIdentity) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(1.f, 1.f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(100, key.target_size().width());
+  EXPECT_EQ(100, key.target_size().height());
+  EXPECT_EQ(100u * 100u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyHighQualityDropToLowIfNearlyIdentity) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(1.001f, 1.001f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(100, key.target_size().width());
+  EXPECT_EQ(100, key.target_size().height());
+  EXPECT_EQ(100u * 100u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, ImageKeyHighQualityDropToLowIfNearlyIdentity2) {
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.999f, 0.999f), quality, has_perspective, is_decomposable);
+
+  auto key = ImageDecodeControllerKey::FromDrawImage(draw_image);
+  EXPECT_EQ(image->uniqueID(), key.image_id());
+  EXPECT_EQ(kLow_SkFilterQuality, key.filter_quality());
+  EXPECT_EQ(100, key.target_size().width());
+  EXPECT_EQ(100, key.target_size().height());
+  EXPECT_EQ(100u * 100u * 4u, key.target_bytes());
+}
+
+TEST(ImageDecodeControllerTest, GetTaskForImageSameImage) {
+  ImageDecodeController controller;
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+  uint64_t prepare_tiles_id = 1;
+
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> task;
+  bool need_unref =
+      controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id, &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  DrawImage another_draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> another_task;
+  need_unref = controller.GetTaskForImageAndRef(
+      another_draw_image, prepare_tiles_id, &another_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task.get() == another_task.get());
+
+  controller.UnrefImage(draw_image);
+  controller.UnrefImage(draw_image);
+}
+
+TEST(ImageDecodeControllerTest, GetTaskForImageSameImageDifferentQuality) {
+  ImageDecodeController controller;
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+
+  DrawImage high_quality_draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), kHigh_SkFilterQuality, has_perspective,
+      is_decomposable);
+  scoped_refptr<ImageDecodeTask> high_quality_task;
+  bool need_unref = controller.GetTaskForImageAndRef(
+      high_quality_draw_image, prepare_tiles_id, &high_quality_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(high_quality_task);
+
+  DrawImage medium_quality_draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), kMedium_SkFilterQuality, has_perspective,
+      is_decomposable);
+  scoped_refptr<ImageDecodeTask> medium_quality_task;
+  need_unref = controller.GetTaskForImageAndRef(
+      medium_quality_draw_image, prepare_tiles_id, &medium_quality_task);
+  // Medium quality isn't handled by the controller, so it won't ref it. Note
+  // that this will change when medium quality is handled and will need to be
+  // updated.
+  EXPECT_FALSE(need_unref);
+  EXPECT_TRUE(medium_quality_task);
+  EXPECT_TRUE(high_quality_task.get() != medium_quality_task.get());
+
+  DrawImage low_quality_draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), kLow_SkFilterQuality, has_perspective,
+      is_decomposable);
+  scoped_refptr<ImageDecodeTask> low_quality_task;
+  need_unref = controller.GetTaskForImageAndRef(
+      low_quality_draw_image, prepare_tiles_id, &low_quality_task);
+  // Note that when we pin the original decode, we might ref low quality images
+  // too, but we don't support that right now.
+  EXPECT_FALSE(need_unref);
+  EXPECT_TRUE(low_quality_task);
+  EXPECT_TRUE(high_quality_task.get() != low_quality_task.get());
+  EXPECT_TRUE(medium_quality_task.get() != low_quality_task.get());
+
+  controller.UnrefImage(high_quality_draw_image);
+}
+
+TEST(ImageDecodeControllerTest, GetTaskForImageSameImageDifferentSize) {
+  ImageDecodeController controller;
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  DrawImage half_size_draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> half_size_task;
+  bool need_unref = controller.GetTaskForImageAndRef(
+      half_size_draw_image, prepare_tiles_id, &half_size_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(half_size_task);
+
+  DrawImage quarter_size_draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.25f, 0.25f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> quarter_size_task;
+  need_unref = controller.GetTaskForImageAndRef(
+      quarter_size_draw_image, prepare_tiles_id, &quarter_size_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(quarter_size_task);
+  EXPECT_TRUE(half_size_task.get() != quarter_size_task.get());
+
+  controller.UnrefImage(half_size_draw_image);
+  controller.UnrefImage(quarter_size_draw_image);
+}
+
+TEST(ImageDecodeControllerTest, GetTaskForImageDifferentImage) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> first_image = CreateImage(100, 100);
+  DrawImage first_draw_image(
+      first_image.get(),
+      SkIRect::MakeWH(first_image->width(), first_image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> first_task;
+  bool need_unref = controller.GetTaskForImageAndRef(
+      first_draw_image, prepare_tiles_id, &first_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(first_task);
+
+  skia::RefPtr<SkImage> second_image = CreateImage(100, 100);
+  DrawImage second_draw_image(
+      second_image.get(),
+      SkIRect::MakeWH(second_image->width(), second_image->height()),
+      SkSize::Make(0.25f, 0.25f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> second_task;
+  need_unref = controller.GetTaskForImageAndRef(second_draw_image,
+                                                prepare_tiles_id, &second_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(second_task);
+  EXPECT_TRUE(first_task.get() != second_task.get());
+
+  controller.UnrefImage(first_draw_image);
+  controller.UnrefImage(second_draw_image);
+}
+
+TEST(ImageDecodeControllerTest, GetTaskForImageAlreadyDecoded) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> task;
+  bool need_unref =
+      controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id, &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  task->WillSchedule();
+  task->ScheduleOnOriginThread(nullptr);
+  task->DidSchedule();
+  task->RunOnWorkerThread();
+
+  scoped_refptr<ImageDecodeTask> another_task;
+  need_unref = controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id,
+                                                &another_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_FALSE(another_task);
+
+  task->WillComplete();
+  task->CompleteOnOriginThread(nullptr);
+  task->DidComplete();
+
+  controller.UnrefImage(draw_image);
+  controller.UnrefImage(draw_image);
+}
+
+TEST(ImageDecodeControllerTest, GetTaskForImageAlreadyPrerolled) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kLow_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(1.f, 1.f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> task;
+  bool need_unref =
+      controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id, &task);
+  // We're not currently locking original scale or low quality images.
+  EXPECT_FALSE(need_unref);
+  EXPECT_TRUE(task);
+
+  task->WillSchedule();
+  task->ScheduleOnOriginThread(nullptr);
+  task->DidSchedule();
+  task->RunOnWorkerThread();
+
+  scoped_refptr<ImageDecodeTask> another_task;
+  need_unref = controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id,
+                                                &another_task);
+  EXPECT_FALSE(need_unref);
+  EXPECT_FALSE(another_task);
+
+  task->WillComplete();
+  task->CompleteOnOriginThread(nullptr);
+  task->DidComplete();
+
+  scoped_refptr<ImageDecodeTask> third_task;
+  need_unref = controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id,
+                                                &third_task);
+  EXPECT_FALSE(need_unref);
+  EXPECT_FALSE(third_task);
+}
+
+TEST(ImageDecodeControllerTest, GetTaskForImageCanceledGetsNewTask) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> task;
+  bool need_unref =
+      controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id, &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  task->WillSchedule();
+  task->ScheduleOnOriginThread(nullptr);
+  task->DidSchedule();
+
+  scoped_refptr<ImageDecodeTask> another_task;
+  need_unref = controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id,
+                                                &another_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(another_task.get() == task.get());
+
+  // Didn't run the task, complete it (it was canceled).
+  task->WillComplete();
+  task->CompleteOnOriginThread(nullptr);
+  task->DidComplete();
+
+  // Fully cancel everything (so the raster would unref things).
+  controller.UnrefImage(draw_image);
+  controller.UnrefImage(draw_image);
+
+  // Here a new task is created.
+  scoped_refptr<ImageDecodeTask> third_task;
+  need_unref = controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id,
+                                                &third_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(third_task);
+  EXPECT_FALSE(third_task.get() == task.get());
+
+  controller.UnrefImage(draw_image);
+}
+
+TEST(ImageDecodeControllerTest, GetTaskForImageCanceledWhileReffedGetsNewTask) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> task;
+  bool need_unref =
+      controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id, &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  task->WillSchedule();
+  task->ScheduleOnOriginThread(nullptr);
+  task->DidSchedule();
+
+  scoped_refptr<ImageDecodeTask> another_task;
+  need_unref = controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id,
+                                                &another_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(another_task.get() == task.get());
+
+  // Didn't run the task, complete it (it was canceled).
+  task->WillComplete();
+  task->CompleteOnOriginThread(nullptr);
+  task->DidComplete();
+
+  // Note that here, everything is reffed, but a new task is created. This is
+  // possible with repeated schedule/cancel operations.
+  scoped_refptr<ImageDecodeTask> third_task;
+  need_unref = controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id,
+                                                &third_task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(third_task);
+  EXPECT_FALSE(third_task.get() == task.get());
+
+  // 3 Unrefs!
+  controller.UnrefImage(draw_image);
+  controller.UnrefImage(draw_image);
+  controller.UnrefImage(draw_image);
+}
+
+TEST(ImageDecodeControllerTest, GetDecodedImageForDraw) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+  scoped_refptr<ImageDecodeTask> task;
+  bool need_unref =
+      controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id, &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  task->WillSchedule();
+  task->ScheduleOnOriginThread(nullptr);
+  task->DidSchedule();
+
+  task->RunOnWorkerThread();
+
+  task->WillComplete();
+  task->CompleteOnOriginThread(nullptr);
+  task->DidComplete();
+
+  DecodedDrawImage decoded_draw_image =
+      controller.GetDecodedImageForDraw(draw_image);
+  EXPECT_TRUE(decoded_draw_image.image());
+  EXPECT_EQ(50, decoded_draw_image.image()->width());
+  EXPECT_EQ(50, decoded_draw_image.image()->height());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().width());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().height());
+  EXPECT_EQ(kLow_SkFilterQuality, decoded_draw_image.filter_quality());
+  EXPECT_FALSE(decoded_draw_image.is_scale_adjustment_identity());
+  EXPECT_FALSE(decoded_draw_image.is_at_raster_decode());
+
+  controller.DrawWithImageFinished(draw_image, decoded_draw_image);
+  controller.UnrefImage(draw_image);
+}
+
+TEST(ImageDecodeControllerTest, GetDecodedImageForDrawAtRasterDecode) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+
+  DecodedDrawImage decoded_draw_image =
+      controller.GetDecodedImageForDraw(draw_image);
+  EXPECT_TRUE(decoded_draw_image.image());
+  EXPECT_EQ(50, decoded_draw_image.image()->width());
+  EXPECT_EQ(50, decoded_draw_image.image()->height());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().width());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().height());
+  EXPECT_EQ(kLow_SkFilterQuality, decoded_draw_image.filter_quality());
+  EXPECT_FALSE(decoded_draw_image.is_scale_adjustment_identity());
+  EXPECT_TRUE(decoded_draw_image.is_at_raster_decode());
+
+  controller.DrawWithImageFinished(draw_image, decoded_draw_image);
+}
+
+TEST(ImageDecodeControllerTest,
+     GetDecodedImageForDrawAtRasterDecodeMultipleTimes) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+
+  DecodedDrawImage decoded_draw_image =
+      controller.GetDecodedImageForDraw(draw_image);
+  EXPECT_TRUE(decoded_draw_image.image());
+  EXPECT_EQ(50, decoded_draw_image.image()->width());
+  EXPECT_EQ(50, decoded_draw_image.image()->height());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().width());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().height());
+  EXPECT_EQ(kLow_SkFilterQuality, decoded_draw_image.filter_quality());
+  EXPECT_FALSE(decoded_draw_image.is_scale_adjustment_identity());
+  EXPECT_TRUE(decoded_draw_image.is_at_raster_decode());
+
+  DecodedDrawImage another_decoded_draw_image =
+      controller.GetDecodedImageForDraw(draw_image);
+  EXPECT_EQ(decoded_draw_image.image()->uniqueID(),
+            another_decoded_draw_image.image()->uniqueID());
+
+  controller.DrawWithImageFinished(draw_image, decoded_draw_image);
+  controller.DrawWithImageFinished(draw_image, another_decoded_draw_image);
+}
+
+TEST(ImageDecodeControllerTest,
+     GetDecodedImageForDrawAtRasterDecodeDoesNotPreventTasks) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+
+  DecodedDrawImage decoded_draw_image =
+      controller.GetDecodedImageForDraw(draw_image);
+  EXPECT_TRUE(decoded_draw_image.image());
+  EXPECT_EQ(50, decoded_draw_image.image()->width());
+  EXPECT_EQ(50, decoded_draw_image.image()->height());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().width());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().height());
+  EXPECT_EQ(kLow_SkFilterQuality, decoded_draw_image.filter_quality());
+  EXPECT_FALSE(decoded_draw_image.is_scale_adjustment_identity());
+  EXPECT_TRUE(decoded_draw_image.is_at_raster_decode());
+
+  scoped_refptr<ImageDecodeTask> task;
+  bool need_unref =
+      controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id, &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  task->WillSchedule();
+  task->ScheduleOnOriginThread(nullptr);
+  task->DidSchedule();
+
+  task->RunOnWorkerThread();
+
+  task->WillComplete();
+  task->CompleteOnOriginThread(nullptr);
+  task->DidComplete();
+
+  DecodedDrawImage another_decoded_draw_image =
+      controller.GetDecodedImageForDraw(draw_image);
+  // This should get the new decoded/locked image, not the one we're using at
+  // raster.
+  // TODO(vmpstr): We can possibly optimize this so that the decode simply moves
+  // the image to the right spot.
+  EXPECT_NE(decoded_draw_image.image()->uniqueID(),
+            another_decoded_draw_image.image()->uniqueID());
+  EXPECT_FALSE(another_decoded_draw_image.is_at_raster_decode());
+
+  controller.DrawWithImageFinished(draw_image, decoded_draw_image);
+  controller.DrawWithImageFinished(draw_image, another_decoded_draw_image);
+  controller.UnrefImage(draw_image);
+}
+
+TEST(ImageDecodeControllerTest,
+     GetDecodedImageForDrawAtRasterDecodeIsUsedForLockedCache) {
+  ImageDecodeController controller;
+  bool has_perspective = false;
+  bool is_decomposable = true;
+  uint64_t prepare_tiles_id = 1;
+  SkFilterQuality quality = kHigh_SkFilterQuality;
+
+  skia::RefPtr<SkImage> image = CreateImage(100, 100);
+  DrawImage draw_image(
+      image.get(), SkIRect::MakeWH(image->width(), image->height()),
+      SkSize::Make(0.5f, 0.5f), quality, has_perspective, is_decomposable);
+
+  DecodedDrawImage decoded_draw_image =
+      controller.GetDecodedImageForDraw(draw_image);
+  EXPECT_TRUE(decoded_draw_image.image());
+  EXPECT_EQ(50, decoded_draw_image.image()->width());
+  EXPECT_EQ(50, decoded_draw_image.image()->height());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().width());
+  EXPECT_FLOAT_EQ(0.5f, decoded_draw_image.scale_adjustment().height());
+  EXPECT_EQ(kLow_SkFilterQuality, decoded_draw_image.filter_quality());
+  EXPECT_FALSE(decoded_draw_image.is_scale_adjustment_identity());
+  EXPECT_TRUE(decoded_draw_image.is_at_raster_decode());
+
+  scoped_refptr<ImageDecodeTask> task;
+  bool need_unref =
+      controller.GetTaskForImageAndRef(draw_image, prepare_tiles_id, &task);
+  EXPECT_TRUE(need_unref);
+  EXPECT_TRUE(task);
+
+  task->WillSchedule();
+  task->ScheduleOnOriginThread(nullptr);
+  task->DidSchedule();
+
+  // If we finish the draw here, then we will use it for the locked decode
+  // instead of decoding again.
+  controller.DrawWithImageFinished(draw_image, decoded_draw_image);
+
+  task->RunOnWorkerThread();
+
+  task->WillComplete();
+  task->CompleteOnOriginThread(nullptr);
+  task->DidComplete();
+
+  DecodedDrawImage another_decoded_draw_image =
+      controller.GetDecodedImageForDraw(draw_image);
+  // This should get the decoded/locked image which we originally decoded at
+  // raster time, since it's now in the locked cache.
+  EXPECT_EQ(decoded_draw_image.image()->uniqueID(),
+            another_decoded_draw_image.image()->uniqueID());
+  EXPECT_FALSE(another_decoded_draw_image.is_at_raster_decode());
+
+  controller.DrawWithImageFinished(draw_image, another_decoded_draw_image);
+  controller.UnrefImage(draw_image);
+}
+
+}  // namespace
+}  // namespace cc