Add layout states to DocumentLifecycle state machine

Source/core/dom/DocumentLifecycle.cpp

 /*
  * Copyright (C) 2013 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 14 matching lines @@
  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
 #include "core/dom/DocumentLifecycle.h"
 
 #include "wtf/Assertions.h"
-#include <stdio.h>
 
 namespace WebCore {
 
+static DocumentLifecycle::DeprecatedTransition* s_deprecatedTransitionStack = 0;
+
+DocumentLifecycle::Scope::Scope(DocumentLifecycle& lifecycle, State finalState)
+    : m_lifecycle(lifecycle)
+    , m_finalState(finalState)
+{
+}
+
+DocumentLifecycle::Scope::~Scope()
+{
+    m_lifecycle.advanceTo(m_finalState);
+}
+
+DocumentLifecycle::DeprecatedTransition::DeprecatedTransition(State from, State to)
+    : m_previous(s_deprecatedTransitionStack)
+    , m_from(from)
+    , m_to(to)
+{
+    s_deprecatedTransitionStack = this;
+}
+
+DocumentLifecycle::DeprecatedTransition::~DeprecatedTransition()
+{
+    s_deprecatedTransitionStack = m_previous;
+}
+
 DocumentLifecycle::DocumentLifecycle()
     : m_state(Uninitialized)
 {
 }
 
 DocumentLifecycle::~DocumentLifecycle()
 {
 }
 
+#if !ASSERT_DISABLED
+
 bool DocumentLifecycle::canAdvanceTo(State state) const
 {
+    // This transition is bogus, but we've whitelisted it anyway.
+    if (s_deprecatedTransitionStack && m_state == s_deprecatedTransitionStack->from() && state == s_deprecatedTransitionStack->to())
+        return true;
     if (state > m_state)
         return true;
-    // FIXME: We can dispose a document multiple times. This seems wrong.
-    // See https://code.google.com/p/chromium/issues/detail?id=301668.
-    if (m_state == Disposed)
+    if (m_state == Disposed) {
+        // FIXME: We can dispose a document multiple times. This seems wrong.
+        // See https://code.google.com/p/chromium/issues/detail?id=301668.
         return state == Disposed;
-    if (m_state == Clean) {
-        // We can synchronously enter recalc style without rewinding to
-        // StyleRecalcPending.
-        return state == InStyleRecalc;
+    }
+    if (m_state == StyleClean) {
+        if (state == StyleRecalcPending)
+            return true;
+        // We can synchronously recalc style.
+        if (state == InStyleRecalc)
+            return true;
+        // We can synchronously perform layout.
+        if (state == InPreLayout)
+            return true;
+        if (state == InPerformLayout)
+            return true;
+        // We can redundant arrive in the style clean state.
+        if (state == StyleClean)
+            return true;
+        return false;
+    }
+    if (m_state == InPreLayout) {
+        if (state == InStyleRecalc)
+            return true;
+        if (state == StyleClean)
+            return true;
+        if (state == InPreLayout)
+            return true;
+        return false;
+    }
+    if (m_state == AfterPerformLayout) {
+        if (state == InPreLayout)
+            return true;
+        // If we're doing a partial layout, we won't actually end up cleaning
+        // out all the layout dirty bits. Instead, we'll return to StyleClean.
+        if (state == StyleClean)
+            return true;
+        return false;
+    }
+    if (m_state == LayoutClean) {
+        if (state == StyleRecalcPending)
+            return true;
+        // We can synchronously recalc style.
+        if (state == InStyleRecalc)
+            return true;
+        // We can synchronously perform layout.
+        if (state == InPreLayout)
+            return true;
+        if (state == InPerformLayout)
+            return true;
+        // We can redundant arrive in the layout clean state. This situation
+        // can happen when we call layout recursively and we unwind the stack.
+        if (state == LayoutClean)
+            return true;
+        if (state == StyleClean)
+            return true;
+        return false;
     }
     return false;
 }
 
+#endif
+
 void DocumentLifecycle::advanceTo(State state)
 {
     ASSERT(canAdvanceTo(state));
     m_state = state;
 }
 
-void DocumentLifecycle::rewindTo(State state)
-{
-    ASSERT(m_state == Clean);
-    ASSERT(state < m_state);
-    m_state = state;
-    ASSERT(isActive());
 }
-
-}