Transfer canvas state to the next frame with noticable restrictions.

Source/platform/graphics/RecordingImageBufferSurface.cpp

 // Copyright 2014 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 "config.h"
 
 #include "platform/graphics/RecordingImageBufferSurface.h"
 
 #include "platform/graphics/GraphicsContext.h"
 #include "platform/graphics/ImageBuffer.h"
@@ skipping 100 matching lines @@
 
 void RecordingImageBufferSurface::didClearCanvas()
 {
     m_frameWasCleared = true;
 }
 
 bool RecordingImageBufferSurface::finalizeFrameInternal()
 {
     if (!m_imageBuffer->isDirty()) {
         if (m_currentFrame && !m_previousFrame) {
-            // Create an initial blank frame
             m_previousFrame = adoptRef(m_currentFrame->endRecording());
             initializeCurrentFrame();
         }
         return m_currentFrame;
     }
 
     if (!m_currentFrame) {
         return false;
     }
 
     IntRect canvasRect(IntPoint(0, 0), size());
     if (!m_frameWasCleared && !m_imageBuffer->context()->opaqueRegion().asRect().contains(canvasRect)) {
         return false;
     }
 
-    SkCanvas* oldCanvas = m_currentFrame->getRecordingCanvas(); // Could be raster or picture
+    StateStack stateStack;
 
-    // FIXME(crbug.com/392614): handle transferring complex state from the current picture to the new one.
-    if (oldCanvas->getSaveCount() > m_initialSaveCount)
+    if (!saveState(m_currentFrame->getRecordingCanvas(), &stateStack))
         return false;
 
-    if (!oldCanvas->isClipRect())
-        return false;
-
-    SkMatrix ctm = oldCanvas->getTotalMatrix();
-    SkRect clip;
-    oldCanvas->getClipBounds(&clip);
-
     m_previousFrame = adoptRef(m_currentFrame->endRecording());
     initializeCurrentFrame();
 
-    SkCanvas* newCanvas = m_currentFrame->getRecordingCanvas();
-    newCanvas->concat(ctm);
-    newCanvas->clipRect(clip);
+    setCurrentState(m_currentFrame->getRecordingCanvas(), &stateStack);
 
     m_frameWasCleared = false;
     return true;
 }
 
+bool RecordingImageBufferSurface::saveState(SkCanvas* srcCanvas, StateStack* stateStack)
+{
+    // FIXME(crbug.com/392614): handle transferring complex state from the current picture to the new one:
+    // - non-rectangular clip
+    // - non-invertible clip
+    if (!srcCanvas->isClipRect())
+        return false;
+
+    StateRec state;
+
+    // we will remove all the saved state stack in endRecording anyway, so it makes no difference
+    while (srcCanvas->getSaveCount() > m_initialSaveCount) {
+        state.m_ctm = srcCanvas->getTotalMatrix();
+        srcCanvas->getClipBounds(&state.m_clip);
+        stateStack->append(state);
+        srcCanvas->restore();
+        // FIXME(crbug.com/408392): don't mess up the state in case we will have to fallback
+        if (!srcCanvas->isClipRect())
+            return false;
+    }
+
+    state.m_ctm = srcCanvas->getTotalMatrix();
+    srcCanvas->getClipBounds(&state.m_clip);
+    stateStack->append(state);
+
+    return true;
+}
+
+void RecordingImageBufferSurface::setCurrentState(SkCanvas* dstCanvas, StateStack* stateStack)
+{
+    while (stateStack->size() > 1) {
+        dstCanvas->concat(stateStack->last().m_ctm);

[Justin Novosad, 2014/08/28 14:53:00]

I think this should be setMatrix rather than concat.  A good layout test would
confirm this.

[Sergey, 2014/09/03 01:50:51]

Done.

+        dstCanvas->clipRect(stateStack->last().m_clip);
+        dstCanvas->save();
+        stateStack->removeLast();
+    }
+
+    dstCanvas->concat(stateStack->last().m_ctm);

[Justin Novosad, 2014/08/28 14:53:00]

here too.

[Sergey, 2014/09/03 01:50:51]

Done.

+    dstCanvas->clipRect(stateStack->last().m_clip);
+}
+
 } // namespace blink