Schedule sibling invalidation sets for sibling insert/remove.

third_party/WebKit/Source/core/dom/StyleEngine.cpp

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 1999 Antti Koivisto (koivisto@kde.org)
  *           (C) 2001 Dirk Mueller (mueller@kde.org)
  *           (C) 2006 Alexey Proskuryakov (ap@webkit.org)
  * Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2011, 2012 Apple Inc. All rights reserved.
  * Copyright (C) 2008, 2009 Torch Mobile Inc. All rights reserved. (http://www.torchmobile.com/)
  * Copyright (C) 2008, 2009, 2011, 2012 Google Inc. All rights reserved.
  * Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies)
  * Copyright (C) Research In Motion Limited 2010-2011. All rights reserved.
@@ skipping 19 matching lines @@
 #include "core/HTMLNames.h"
 #include "core/css/CSSDefaultStyleSheets.h"
 #include "core/css/CSSFontSelector.h"
 #include "core/css/CSSStyleSheet.h"
 #include "core/css/FontFaceCache.h"
 #include "core/css/StyleSheetContents.h"
 #include "core/css/invalidation/InvalidationSet.h"
 #include "core/css/resolver/ScopedStyleResolver.h"
 #include "core/dom/DocumentStyleSheetCollector.h"
 #include "core/dom/Element.h"
+#include "core/dom/ElementTraversal.h"
 #include "core/dom/ProcessingInstruction.h"
 #include "core/dom/ShadowTreeStyleSheetCollection.h"
 #include "core/dom/StyleChangeReason.h"
 #include "core/dom/shadow/ShadowRoot.h"
 #include "core/frame/Settings.h"
 #include "core/html/HTMLIFrameElement.h"
 #include "core/html/HTMLLinkElement.h"
 #include "core/html/imports/HTMLImportsController.h"
 #include "core/inspector/InspectorInstrumentation.h"
 #include "core/page/Page.h"
@@ skipping 633 matching lines @@
 void StyleEngine::pseudoStateChangedForElement(CSSSelector::PseudoType pseudoType, Element& element)
 {
     if (shouldSkipInvalidationFor(element))
         return;
 
     InvalidationLists invalidationLists;
     ensureResolver().ensureUpdatedRuleFeatureSet().collectInvalidationSetsForPseudoClass(invalidationLists, element, pseudoType);
     m_styleInvalidator.scheduleInvalidationSetsForElement(invalidationLists, element);
 }
 
+void StyleEngine::scheduleSiblingInvalidationsForElement(Element& element, Element& schedulingElement, unsigned minDirectAdjacent, bool invalidateSchedulingElement)
+{
+    InvalidationLists invalidationLists;
+
+    RuleFeatureSet& ruleFeatureSet = ensureResolver().ensureUpdatedRuleFeatureSet();
+
+    if (element.hasID())
+        ruleFeatureSet.collectSiblingInvalidationSetForId(invalidationLists, element, element.idForStyleResolution(), minDirectAdjacent);
+
+    if (element.hasClass()) {
+        const SpaceSplitString& classNames = element.classNames();
+        for (size_t i = 0; i < classNames.size(); i++)
+            ruleFeatureSet.collectSiblingInvalidationSetForClass(invalidationLists, element, classNames[i], minDirectAdjacent);
+    }
+
+    for (const Attribute& attribute : element.attributes())
+        ruleFeatureSet.collectSiblingInvalidationSetForAttribute(invalidationLists, element, attribute.name(), minDirectAdjacent);
+
+    ruleFeatureSet.collectUniversalSiblingInvalidationSet(invalidationLists, minDirectAdjacent);
+
+    m_styleInvalidator.scheduleInvalidationSetsForElement(invalidationLists, schedulingElement);
+    if (invalidateSchedulingElement)
+        m_styleInvalidator.scheduleDescendantsFromSiblingSets(schedulingElement, invalidationLists.siblings);
+}
+
+void StyleEngine::scheduleInvalidationsForInsertedSibling(Element* beforeElement, Element& insertedElement)
+{
+    unsigned affectedSiblings = insertedElement.parentNode()->childrenAffectedByIndirectAdjacentRules() ? UINT_MAX : m_maxDirectAdjacentSelectors;
+
+    scheduleSiblingInvalidationsForElement(insertedElement, insertedElement, 1, false);
+
+    Element* schedulingElement = &insertedElement;
+    for (unsigned i = 1; beforeElement && i <= affectedSiblings; i++, beforeElement = ElementTraversal::previousSibling(*(schedulingElement = beforeElement)))
+        scheduleSiblingInvalidationsForElement(*beforeElement, *schedulingElement, i, false);
+}
+
+void StyleEngine::scheduleInvalidationsForRemovedSibling(Element* beforeElement, Element& removedElement, Element& afterElement)
+{
+    unsigned affectedSiblings = afterElement.parentNode()->childrenAffectedByIndirectAdjacentRules() ? UINT_MAX : m_maxDirectAdjacentSelectors;
+
+    Element* schedulingElement = beforeElement;
+    if (!schedulingElement)
+        schedulingElement = &afterElement;
+
+    scheduleSiblingInvalidationsForElement(removedElement, *schedulingElement, 1, schedulingElement == &afterElement);
+
+    for (unsigned i = 2; beforeElement && i <= affectedSiblings; i++, beforeElement = ElementTraversal::previousSibling(*beforeElement))
+        scheduleSiblingInvalidationsForElement(*beforeElement, *beforeElement, i, false);
+}
+
 void StyleEngine::setStatsEnabled(bool enabled)
 {
     if (!enabled) {
         m_styleResolverStats = nullptr;
         return;
     }
     if (!m_styleResolverStats)
         m_styleResolverStats = StyleResolverStats::create();
     else
         m_styleResolverStats->reset();
 }
 
+void StyleEngine::willRemoveChild(Node& child)
+{
+    if (!child.needsStyleInvalidation())
+        return;
+    if (!child.isElementNode())
+        return;
+    if (!child.inShadowIncludingDocument())
+        return;
+
+    Element* elementTo = ElementTraversal::previousSibling(child);
+    bool invalidateElementTo = !elementTo;
+    if (!elementTo)
+        elementTo = ElementTraversal::nextSibling(child);
+    if (!elementTo)
+        return;
+    styleInvalidator().rescheduleSiblingInvalidationSetsForRemovedElement(toElement(child), *elementTo, invalidateElementTo);
+}
+
 void StyleEngine::setPreferredStylesheetSetNameIfNotSet(const String& name)
 {
     if (!m_preferredStylesheetSetName.isEmpty())
         return;
     m_preferredStylesheetSetName = name;
     // TODO(rune@opera.com): Setting the selected set here is wrong if the set
     // has been previously set by through Document.selectedStylesheetSet. Our
     // current implementation ignores the effect of Document.selectedStylesheetSet
     // and either only collects persistent style, or additionally preferred
     // style when present. We are currently not marking the document scope dirty
@@ skipping 46 matching lines @@
 
 DEFINE_TRACE_WRAPPERS(StyleEngine)
 {
     for (auto sheet : m_injectedAuthorStyleSheets) {
         visitor->traceWrappers(sheet);
     }
     visitor->traceWrappers(m_documentStyleSheetCollection);
 }
 
 } // namespace blink