Invalidate custom pseudo elements for RuleSet invalidations.

third_party/WebKit/Source/core/css/RuleFeature.cpp

 /*
  * Copyright (C) 1999 Lars Knoll (knoll@kde.org)
  *           (C) 2004-2005 Allan Sandfeld Jensen (kde@carewolf.com)
  * Copyright (C) 2006, 2007 Nicholas Shanks (webkit@nickshanks.com)
  * Copyright (C) 2005, 2006, 2007, 2008, 2009, 2010, 2011, 2012 Apple Inc. All
  * rights reserved.
  * Copyright (C) 2007 Alexey Proskuryakov <ap@webkit.org>
  * Copyright (C) 2007, 2008 Eric Seidel <eric@webkit.org>
  * Copyright (C) 2008, 2009 Torch Mobile Inc. All rights reserved.
  * (http://www.torchmobile.com/)
@@ skipping 462 matching lines @@
   if (lastInCompound)
     updateFeaturesFromCombinator(*lastInCompound, nullptr, features,
                                  siblingFeatures, features);
 
   addFeaturesToInvalidationSets(*nextCompound, siblingFeatures, features);
   updateRuleSetInvalidation(features);
 }
 
 void RuleFeatureSet::updateRuleSetInvalidation(
     const InvalidationSetFeatures& features) {
-  if (!features.hasFeaturesForRuleSetInvalidation) {
-    if (features.forceSubtree || features.tagNames.isEmpty())
-      m_metadata.needsFullRecalcForRuleSetInvalidation = true;
-    else
-      addTagNamesToTypeRuleInvalidationSet(features.tagNames);
+  if (features.hasFeaturesForRuleSetInvalidation)
+    return;
+  if (features.forceSubtree ||
+      (!features.customPseudoElement && features.tagNames.isEmpty())) {
+    m_metadata.needsFullRecalcForRuleSetInvalidation = true;
+    return;
   }
+
+  ensureTypeRuleInvalidationSet();
+
+  if (features.customPseudoElement) {
+    m_typeRuleInvalidationSet->setCustomPseudoInvalid();
+    m_typeRuleInvalidationSet->setTreeBoundaryCrossing();
+  }
+
+  for (auto tagName : features.tagNames)
+    m_typeRuleInvalidationSet->addTagName(tagName);
 }
 
 void RuleFeatureSet::updateInvalidationSetsForContentAttribute(
     const RuleData& ruleData) {
   // If any ::before and ::after rules specify 'content: attr(...)', we
   // need to create invalidation sets for those attributes to have content
   // changes applied through style recalc.
 
   const StylePropertySet& propertySet = ruleData.rule()->properties();
 
@@ skipping 661 matching lines @@
   universalSet.updateMaxDirectAdjacentSelectors(
       siblingFeatures.maxDirectAdjacentSelectors);
 
   if (&siblingFeatures == &descendantFeatures)
     universalSet.setInvalidatesSelf();
   else
     addFeaturesToInvalidationSet(universalSet.ensureSiblingDescendants(),
                                  descendantFeatures);
 }
 
-void RuleFeatureSet::addTagNamesToTypeRuleInvalidationSet(
-    const Vector<AtomicString>& tagNames) {
-  DCHECK(!tagNames.isEmpty());
-  ensureTypeRuleInvalidationSet();
-  for (auto tagName : tagNames)
-    m_typeRuleInvalidationSet->addTagName(tagName);
-}
-
 DEFINE_TRACE(RuleFeatureSet) {
   visitor->trace(m_siblingRules);
   visitor->trace(m_uncommonAttributeRules);
   visitor->trace(m_viewportDependentMediaQueryResults);
   visitor->trace(m_deviceDependentMediaQueryResults);
 }
 
 void RuleFeatureSet::InvalidationSetFeatures::add(
     const InvalidationSetFeatures& other) {
   classes.appendVector(other.classes);
@@ skipping 15 matching lines @@
 bool RuleFeatureSet::InvalidationSetFeatures::hasFeatures() const {
   return !classes.isEmpty() || !attributes.isEmpty() || !ids.isEmpty() ||
          !tagNames.isEmpty() || customPseudoElement;
 }
 
 bool RuleFeatureSet::InvalidationSetFeatures::hasIdClassOrAttribute() const {
   return !classes.isEmpty() || !attributes.isEmpty() || !ids.isEmpty();
 }
 
 }  // namespace blink