SPv2: Add support for tracking raster paint invalidations in testing.

third_party/WebKit/Source/platform/graphics/compositing/PaintArtifactCompositor.h

 // 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.
 
 #ifndef PaintArtifactCompositor_h
 #define PaintArtifactCompositor_h
 
 #include "base/memory/ref_counted.h"
 #include "platform/PlatformExport.h"
+#include "platform/RuntimeEnabledFeatures.h"
+#include "platform/graphics/GraphicsLayerClient.h"
+#include "platform/graphics/paint/PaintController.h"
 #include "wtf/Noncopyable.h"
 #include "wtf/PtrUtil.h"
 #include "wtf/Vector.h"
 #include <memory>
 
 namespace cc {
 class Layer;
 }
 
 namespace gfx {
 class Transform;
 class Vector2dF;
 }
 
 namespace blink {
 
+class DisplayItemClient;
+class IntRect;
+class JSONObject;
 class PaintArtifact;
 class WebLayer;
 struct PaintChunk;
 
 // Responsible for managing compositing in terms of a PaintArtifact.
 //
 // Owns a subtree of the compositor layer tree, and updates it in response to
 // changes in the paint artifact.
 //
 // PaintArtifactCompositor is the successor to PaintLayerCompositor, reflecting
 // the new home of compositing decisions after paint in Slimming Paint v2.
 class PLATFORM_EXPORT PaintArtifactCompositor {
     WTF_MAKE_NONCOPYABLE(PaintArtifactCompositor);
 public:
     ~PaintArtifactCompositor();
 
     static std::unique_ptr<PaintArtifactCompositor> create()
     {
         return wrapUnique(new PaintArtifactCompositor());
     }
 
     // Updates the layer tree to match the provided paint artifact.
-    void update(const PaintArtifact&);
+    void update(const PaintArtifact&, RasterInvalidationTrackingMap<const PaintChunk>* paintChunkInvalidations);
 
     // The root layer of the tree managed by this object.
     cc::Layer* rootLayer() const { return m_rootLayer.get(); }
 
     // Wraps rootLayer(), so that it can be attached as a child of another
     // WebLayer.
     WebLayer* getWebLayer() const { return m_webLayer.get(); }
 
     // Returns extra information recorded during unit tests.
     // While not part of the normal output of this class, this provides a simple
     // way of locating the layers of interest, since there are still a slew of
     // placeholder layers required.
     struct ExtraDataForTesting {
         Vector<scoped_refptr<cc::Layer>> contentLayers;
     };
     void enableExtraDataForTesting() { m_extraDataForTestingEnabled = true; }
     ExtraDataForTesting* getExtraDataForTesting() const { return m_extraDataForTesting.get(); }
 
+    void setTracksRasterInvalidations(bool);
+    void resetTrackedRasterInvalidations();
+    bool hasTrackedRasterInvalidations() const;
+
+    std::unique_ptr<JSONObject> layersAsJSON(LayerTreeFlags) const;
+
 private:
     PaintArtifactCompositor();
 
     class ContentLayerClientImpl;
 
     // Builds a leaf layer that represents a single paint chunk.
     // Note: cc::Layer API assumes the layer bounds to start at (0, 0) but the bounding box of
     // a paint chunk does not necessarily start at (0, 0) and could even be negative. Internally
     // the generated layer translates the paint chunk to align the bounding box to (0, 0) and
     // return the actual origin of the paint chunk in output parameter layerOffset.
     scoped_refptr<cc::Layer> layerForPaintChunk(const PaintArtifact&, const PaintChunk&, gfx::Vector2dF& layerOffset,
-        Vector<std::unique_ptr<ContentLayerClientImpl>>& newContentLayerClients);
+        Vector<std::unique_ptr<ContentLayerClientImpl>>& newContentLayerClients, RasterInvalidationTracking*);
 
     // Finds a client among the current vector of clients that matches the paint chunk's id,
     // or otherwise allocates a new one.
-    std::unique_ptr<ContentLayerClientImpl> clientForPaintChunk(const PaintChunk&);
+    std::unique_ptr<ContentLayerClientImpl> clientForPaintChunk(const PaintChunk&, const PaintArtifact&);
 
     scoped_refptr<cc::Layer> m_rootLayer;
     std::unique_ptr<WebLayer> m_webLayer;
     Vector<std::unique_ptr<ContentLayerClientImpl>> m_contentLayerClients;
 
     bool m_extraDataForTestingEnabled = false;
     std::unique_ptr<ExtraDataForTesting> m_extraDataForTesting;
     friend class StubChromeClientForSPv2;
+
+    bool m_isTrackingRasterInvalidations;
 };
 
 } // namespace blink
 
 #endif // PaintArtifactCompositor_h