cc: Add checker-imaging support to TileManager.

cc/playback/image_hijack_canvas.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/playback/image_hijack_canvas.h"
 
 #include "base/optional.h"
 #include "cc/playback/discardable_image_map.h"
 #include "cc/tiles/image_decode_cache.h"
 #include "third_party/skia/include/core/SkPath.h"
@@ skipping 101 matching lines @@
   const SkPaint& paint() { return paint_; }
 
  private:
   ScopedImagePaint(ScopedDecodedImageLock lock, SkPaint paint)
       : lock_(std::move(lock)), paint_(std::move(paint)) {}
 
   ScopedDecodedImageLock lock_;
   SkPaint paint_;
 };
 
+const SkImage* GetImageInPaint(const SkPaint& paint) {
+  SkShader* shader = paint.getShader();
+  return shader ? shader->isAImage(nullptr, nullptr) : nullptr;
+}
+
 }  // namespace
 
 ImageHijackCanvas::ImageHijackCanvas(int width,
                                      int height,
-                                     ImageDecodeCache* image_decode_cache)
-    : SkNWayCanvas(width, height), image_decode_cache_(image_decode_cache) {}
+                                     ImageDecodeCache* image_decode_cache,
+                                     const ImageIdFlatSet* images_to_skip)
+    : SkNWayCanvas(width, height),
+      image_decode_cache_(image_decode_cache),
+      images_to_skip_(images_to_skip) {}
 
 void ImageHijackCanvas::onDrawPicture(const SkPicture* picture,
                                       const SkMatrix* matrix,
                                       const SkPaint* paint) {
   // Ensure that pictures are unpacked by this canvas, instead of being
   // forwarded to the raster canvas.
   SkCanvas::onDrawPicture(picture, matrix, paint);
 }
 
 void ImageHijackCanvas::onDrawImage(const SkImage* image,
                                     SkScalar x,
                                     SkScalar y,
                                     const SkPaint* paint) {
   if (!image->isLazyGenerated()) {
+    DCHECK(!ShouldSkipImage(image));
     SkNWayCanvas::onDrawImage(image, x, y, paint);
     return;
   }
 
+  if (ShouldSkipImage(image))
+    return;
+
   SkMatrix ctm = getTotalMatrix();
 
   ScopedDecodedImageLock scoped_lock(
       image_decode_cache_, sk_ref_sp(image),
       SkRect::MakeIWH(image->width(), image->height()), ctm, paint);
   const DecodedDrawImage& decoded_image = scoped_lock.decoded_image();
   if (!decoded_image.image())
     return;
 
   DCHECK_EQ(0, static_cast<int>(decoded_image.src_rect_offset().width()));
@@ skipping 10 matching lines @@
   if (need_scale)
     SkNWayCanvas::restore();
 }
 
 void ImageHijackCanvas::onDrawImageRect(const SkImage* image,
                                         const SkRect* src,
                                         const SkRect& dst,
                                         const SkPaint* paint,
                                         SrcRectConstraint constraint) {
   if (!image->isLazyGenerated()) {
+    DCHECK(!ShouldSkipImage(image));
     SkNWayCanvas::onDrawImageRect(image, src, dst, paint, constraint);
     return;
   }
 
+  if (ShouldSkipImage(image))
+    return;
+
   SkRect src_storage;
   if (!src) {
     src_storage = SkRect::MakeIWH(image->width(), image->height());
     src = &src_storage;
   }
   SkMatrix matrix;
   matrix.setRectToRect(*src, dst, SkMatrix::kFill_ScaleToFit);
   matrix.postConcat(getTotalMatrix());
 
   ScopedDecodedImageLock scoped_lock(image_decode_cache_, sk_ref_sp(image),
@@ skipping 12 matching lines @@
     float y_scale = decoded_image.scale_adjustment().height();
     adjusted_src = SkRect::MakeXYWH(
         adjusted_src.x() * x_scale, adjusted_src.y() * y_scale,
         adjusted_src.width() * x_scale, adjusted_src.height() * y_scale);
   }
   SkNWayCanvas::onDrawImageRect(decoded_image.image().get(), &adjusted_src, dst,
                                 decoded_paint, constraint);
 }
 
 void ImageHijackCanvas::onDrawRect(const SkRect& r, const SkPaint& paint) {
+  if (ShouldSkipImageInPaint(paint))
+    return;
+
   base::Optional<ScopedImagePaint> image_paint =
       ScopedImagePaint::TryCreate(image_decode_cache_, getTotalMatrix(), paint);
   if (!image_paint.has_value()) {
     SkNWayCanvas::onDrawRect(r, paint);
     return;
   }
   SkNWayCanvas::onDrawRect(r, image_paint.value().paint());
 }
 
 void ImageHijackCanvas::onDrawPath(const SkPath& path, const SkPaint& paint) {
+  if (ShouldSkipImageInPaint(paint))
+    return;
+
   base::Optional<ScopedImagePaint> image_paint =
       ScopedImagePaint::TryCreate(image_decode_cache_, getTotalMatrix(), paint);
   if (!image_paint.has_value()) {
     SkNWayCanvas::onDrawPath(path, paint);
     return;
   }
   SkNWayCanvas::onDrawPath(path, image_paint.value().paint());
 }
 
 void ImageHijackCanvas::onDrawOval(const SkRect& r, const SkPaint& paint) {
+  if (ShouldSkipImageInPaint(paint))
+    return;
+
   base::Optional<ScopedImagePaint> image_paint =
       ScopedImagePaint::TryCreate(image_decode_cache_, getTotalMatrix(), paint);
   if (!image_paint.has_value()) {
     SkNWayCanvas::onDrawOval(r, paint);
     return;
   }
   SkNWayCanvas::onDrawOval(r, image_paint.value().paint());
 }
 
 void ImageHijackCanvas::onDrawArc(const SkRect& r,
                                   SkScalar start_angle,
                                   SkScalar sweep_angle,
                                   bool use_center,
                                   const SkPaint& paint) {
+  if (ShouldSkipImageInPaint(paint))
+    return;
+
   base::Optional<ScopedImagePaint> image_paint =
       ScopedImagePaint::TryCreate(image_decode_cache_, getTotalMatrix(), paint);
   if (!image_paint.has_value()) {
     SkNWayCanvas::onDrawArc(r, start_angle, sweep_angle, use_center, paint);
     return;
   }
   SkNWayCanvas::onDrawArc(r, start_angle, sweep_angle, use_center,
                           image_paint.value().paint());
 }
 
 void ImageHijackCanvas::onDrawRRect(const SkRRect& rr, const SkPaint& paint) {
+  if (ShouldSkipImageInPaint(paint))
+    return;
+
   base::Optional<ScopedImagePaint> image_paint =
       ScopedImagePaint::TryCreate(image_decode_cache_, getTotalMatrix(), paint);
   if (!image_paint.has_value()) {
     SkNWayCanvas::onDrawRRect(rr, paint);
     return;
   }
   SkNWayCanvas::onDrawRRect(rr, image_paint.value().paint());
 }
 
 void ImageHijackCanvas::onDrawImageNine(const SkImage* image,
                                         const SkIRect& center,
                                         const SkRect& dst,
                                         const SkPaint* paint) {
   // No cc embedder issues image nine calls.
   NOTREACHED();
 }
 
+bool ImageHijackCanvas::ShouldSkipImage(const SkImage* image) const {
+  return images_to_skip_->find(image->uniqueID()) != images_to_skip_->end();
+}
+
+bool ImageHijackCanvas::ShouldSkipImageInPaint(const SkPaint& paint) const {
+  const SkImage* image = GetImageInPaint(paint);
+  return image ? ShouldSkipImage(image) : false;
+}
+
 }  // namespace cc