cc: Do GatherPixelRefs from skia at record time

cc/resources/picture.h

 // Copyright 2012 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_RESOURCES_PICTURE_H_
 #define CC_RESOURCES_PICTURE_H_
 
 #include <list>
+#include <utility>
 #include <vector>
 
 #include "base/basictypes.h"
+#include "base/hash_tables.h"
 #include "base/memory/ref_counted.h"
 #include "cc/base/cc_export.h"
+#include "cc/base/hash_pair.h"
 #include "skia/ext/lazy_pixel_ref.h"
 #include "skia/ext/refptr.h"
 #include "third_party/skia/include/core/SkPixelRef.h"
 #include "third_party/skia/include/core/SkTileGridPicture.h"
 #include "ui/gfx/rect.h"
 
 namespace skia {
 class AnalysisCanvas;
 }
 
 namespace cc {
 
 class ContentLayerClient;
 struct RenderingStats;
 
 class CC_EXPORT Picture
     : public base::RefCountedThreadSafe<Picture> {
  public:
+  typedef base::hash_map<
+      std::pair<int, int>,
+      std::list<skia::LazyPixelRef*> > PixelRefsMap;
+
   static scoped_refptr<Picture> Create(gfx::Rect layer_rect);
 
   const gfx::Rect& LayerRect() const { return layer_rect_; }
   const gfx::Rect& OpaqueRect() const { return opaque_rect_; }
 
   // Get thread-safe clone for rasterizing with on a specific thread.
   scoped_refptr<Picture> GetCloneForDrawingOnThread(
       unsigned thread_index) const;
 
   // Make thread-safe clones for rasterizing with.
   void CloneForDrawing(int num_threads);
 
   // Record a paint operation. To be able to safely use this SkPicture for
   // playback on a different thread this can only be called once.
   void Record(ContentLayerClient*, RenderingStats*,
       const SkTileGridPicture::TileGridInfo& tile_grid_info);
 
   // Has Record() been called yet?
   bool HasRecording() const { return picture_.get() != NULL; }
 
   // Apply this contents scale and raster the content rect into the canvas.
   void Raster(SkCanvas* canvas,
               gfx::Rect content_rect,
               float contents_scale,
               bool enable_lcd_text);
 
-  void GatherPixelRefs(
-      const gfx::Rect& layer_rect,
-      std::list<skia::LazyPixelRef*>& pixel_ref_list);
+  class CC_EXPORT LazyPixelRefsIterator {
+   public:
+    LazyPixelRefsIterator();
+    LazyPixelRefsIterator(gfx::Rect layer_rect, const Picture* picture);
+    ~LazyPixelRefsIterator();
+
+    skia::LazyPixelRef* operator->() const { return current_pixel_ref_; }
+    skia::LazyPixelRef* operator*() const { return current_pixel_ref_; }
+    LazyPixelRefsIterator& operator++();
+    operator bool() const { return !!current_pixel_ref_; }
+
+   private:
+    const Picture* picture_;
+    std::list<skia::LazyPixelRef*> current_list_;
+    skia::LazyPixelRef* current_pixel_ref_;
+
+    gfx::Point min_point_;
+    gfx::Point max_point_;
+    int current_x_;
+    int current_y_;
+  };
 
  private:
   explicit Picture(gfx::Rect layer_rect);
   // This constructor assumes SkPicture is already ref'd and transfers
   // ownership to this picture.
   Picture(const skia::RefPtr<SkPicture>&,
           gfx::Rect layer_rect,
-          gfx::Rect opaque_rect);
+          gfx::Rect opaque_rect,
+          const PixelRefsMap& lazy_pixel_refs);
   ~Picture();
 
+  void GatherPixelRefsFromSkia(
+      const gfx::Rect& layer_rect,
+      std::list<skia::LazyPixelRef*>& pixel_ref_list);
+  void GatherAllPixelRefs();
+
   gfx::Rect layer_rect_;
   gfx::Rect opaque_rect_;
   skia::RefPtr<SkPicture> picture_;
 
   typedef std::vector<scoped_refptr<Picture> > PictureVector;
   PictureVector clones_;
 
+  PixelRefsMap lazy_pixel_refs_;
+
   friend class base::RefCountedThreadSafe<Picture>;
+  friend class LazyPixelRefsIterator;
   DISALLOW_COPY_AND_ASSIGN(Picture);
 };
 
 }  // namespace cc
 
 #endif  // CC_RESOURCES_PICTURE_H_