Implement on demand quad rasterization for PicturePiles.

cc/trees/layer_tree_host_pixeltest_on_demand_raster.cc

+// Copyright 2013 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/layers/append_quads_data.h"
+#include "cc/layers/content_layer_client.h"
+#include "cc/layers/picture_layer.h"
+#include "cc/layers/picture_layer_impl.h"
+#include "cc/quads/draw_quad.h"
+#include "cc/test/layer_tree_pixel_test.h"
+#include "cc/test/mock_quad_culler.h"
+#include "cc/trees/layer_tree_impl.h"
+#include "third_party/skia/include/core/SkCanvas.h"
+#include "third_party/skia/include/core/SkColor.h"
+#include "ui/gfx/rect.h"
+#include "ui/gfx/rect_f.h"
+
+#if !defined(OS_ANDROID)
+
+namespace cc {
+namespace {
+
+class LayerTreeHostOnDemandRasterPixelTest : public LayerTreePixelTest {
+ public:
+  virtual void InitializeSettings(LayerTreeSettings* settings) OVERRIDE {
+    settings->impl_side_painting = true;
+  }
+
+  virtual void BeginCommitOnThread(LayerTreeHostImpl* impl) OVERRIDE {
+    // Not enough memory available. Enforce on-demand rasterization.
+    impl->SetManagedMemoryPolicy(
+        ManagedMemoryPolicy(1, ManagedMemoryPolicy::CUTOFF_ALLOW_EVERYTHING,
+                            1, ManagedMemoryPolicy::CUTOFF_ALLOW_NOTHING));
+  }
+
+  virtual void SwapBuffersOnThread(LayerTreeHostImpl* host_impl,
+                                   bool result) OVERRIDE {
+    // Find the PictureLayerImpl ask it to append quads to check their material.
+    // The PictureLayerImpl is assumed to be the first child of the root layer
+    // in the active tree.
+    PictureLayerImpl* picture_layer = static_cast<PictureLayerImpl*>(
+        host_impl->active_tree()->root_layer()->child_at(0));
+
+    QuadList quads;
+    SharedQuadStateList shared_states;
+    MockQuadCuller quad_culler(&quads, &shared_states);
+
+    AppendQuadsData data;
+    picture_layer->AppendQuads(&quad_culler, &data);
+
+    for (size_t i = 0; i < quads.size(); ++i)
+      EXPECT_EQ(quads[i]->material, DrawQuad::PICTURE_CONTENT);
+
+    // Triggers pixel readback and ends the test.
+    LayerTreePixelTest::SwapBuffersOnThread(host_impl, result);
+  }
+};
+
+class BlueYellowLayerClient : public ContentLayerClient {
+ public:
+  BlueYellowLayerClient(gfx::Rect layer_rect) : layer_rect_(layer_rect) { }
+
+  virtual void PaintContents(SkCanvas* canvas,
+                             gfx::Rect clip,
+                             gfx::RectF* opaque) OVERRIDE {
+    *opaque = gfx::RectF(layer_rect_.width(), layer_rect_.height());
+
+    // Skia uses BGRA ordering rather than RGBA. As the results will be later
+    // compared with a RGBA image, swap the red and blue components here.
+    SkPaint paint;
+    paint.setColor(SwapRB(SK_ColorBLUE));
+    canvas->drawRect(SkRect::MakeWH(layer_rect_.width() / 2,
+                                    layer_rect_.height()),
+                     paint);
+
+    paint.setColor(SwapRB(SK_ColorYELLOW));
+    canvas->drawRect(
+        SkRect::MakeXYWH(layer_rect_.width() / 2, 0,

[danakj, 2013/03/22 18:02:41]

nit: 1 argument per line if they don't all fit on one line

[Leandro Graciá Gil, 2013/03/22 20:40:58]

Done.

[Leandro Graciá Gil, 2013/03/22 20:40:58]

Done.

+                         layer_rect_.width() / 2, layer_rect_.height()),
+        paint);
+  }
+
+  static SkColor SwapRB(SkColor color) {
+    return SkColorSetARGB(
+        SkColorGetA(color),
+        SkColorGetB(color),
+        SkColorGetG(color),
+        SkColorGetR(color));
+  }
+
+ private:
+  gfx::Rect layer_rect_;
+};
+
+TEST_F(LayerTreeHostOnDemandRasterPixelTest, RasterPictureLayer) {
+  // Use multiple colors in a single layer to prevent bypassing on-demand
+  // rasterization if a single solid color is detected in picture analysis.
+  gfx::Rect layer_rect(200, 200);
+  BlueYellowLayerClient client(layer_rect);
+  scoped_refptr<PictureLayer> layer = PictureLayer::Create(&client);
+
+  layer->SetIsDrawable(true);
+  layer->SetAnchorPoint(gfx::PointF());
+  layer->SetBounds(layer_rect.size());
+  layer->SetPosition(layer_rect.origin());
+
+  RunPixelTest(layer, base::FilePath(FILE_PATH_LITERAL("blue_yellow.png")));
+}
+
+}  // namespace
+}  // namespace cc
+
+#endif  // OS_ANDROID