layout/inspector/ServoStyleRuleMap.cpp

   1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
   2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */
   3 /* This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "mozilla/ServoStyleRuleMap.h"
   8
   9 #include "mozilla/css/GroupRule.h"
  10 #include "mozilla/dom/CSSImportRule.h"
  11 #include "mozilla/dom/CSSRuleBinding.h"
  12 #include "mozilla/dom/CSSStyleRule.h"
  13 #include "mozilla/dom/Document.h"
  14 #include "mozilla/dom/Element.h"
  15 #include "mozilla/dom/ShadowRoot.h"
  16 #include "mozilla/IntegerRange.h"
  17 #include "mozilla/ServoStyleSet.h"
  18 #include "mozilla/StyleSheetInlines.h"
  19 #include "nsStyleSheetService.h"
  20
  21 using namespace mozilla::dom;
  22
  23 namespace mozilla {
  24
  25 void ServoStyleRuleMap::EnsureTable(ServoStyleSet& aStyleSet) {
  26   if (!IsEmpty()) {
  27     return;
  28   }
  29   aStyleSet.EnumerateStyleSheets(
  30       [&](StyleSheet& aSheet) { FillTableFromStyleSheet(aSheet); });
  31 }
  32
  33 void ServoStyleRuleMap::EnsureTable(ShadowRoot& aShadowRoot) {
  34   if (!IsEmpty()) {
  35     return;
  36   }
  37   for (auto index : IntegerRange(aShadowRoot.SheetCount())) {
  38     FillTableFromStyleSheet(*aShadowRoot.SheetAt(index));
  39   }
  40   for (const auto& sheet : aShadowRoot.AdoptedStyleSheets()) {
  41     FillTableFromStyleSheet(*sheet);
  42   }
  43 }
  44
  45 void ServoStyleRuleMap::SheetAdded(StyleSheet& aStyleSheet) {
  46   if (!IsEmpty()) {
  47     FillTableFromStyleSheet(aStyleSheet);
  48   }
  49 }
  50
  51 void ServoStyleRuleMap::SheetCloned(StyleSheet& aStyleSheet) {
  52   // Invalidate all data inside. We could probably track down all the individual
  53   // rules that changed etc, but it doesn't seem worth it.
  54   //
  55   // TODO: We can't do this until GetCssRulesInternal stops cloning.
  56   // mTable.Clear();
  57 }
  58
  59 void ServoStyleRuleMap::SheetRemoved(StyleSheet& aStyleSheet) {
  60   // Invalidate all data inside. This isn't strictly necessary since
  61   // we should always get update from document before new queries come.
  62   // But it is probably still safer if we try to avoid having invalid
  63   // pointers inside. Also if the document keep adding and removing
  64   // stylesheets, this would also prevent us from infinitely growing
  65   // memory usage.
  66   mTable.Clear();
  67 }
  68
  69 void ServoStyleRuleMap::RuleAdded(StyleSheet& aStyleSheet,
  70                                   css::Rule& aStyleRule) {
  71   if (!IsEmpty()) {
  72     FillTableFromRule(aStyleRule);
  73   }
  74 }
  75
  76 void ServoStyleRuleMap::RuleRemoved(StyleSheet& aStyleSheet,
  77                                     css::Rule& aStyleRule) {
  78   if (IsEmpty()) {
  79     return;
  80   }
  81
  82   switch (aStyleRule.Type()) {
  83     case StyleCssRuleType::Style:
  84     case StyleCssRuleType::Import:
  85     case StyleCssRuleType::Media:
  86     case StyleCssRuleType::Supports:
  87     case StyleCssRuleType::LayerBlock:
  88     case StyleCssRuleType::Container:
  89     case StyleCssRuleType::Document:
  90     case StyleCssRuleType::Scope: {
  91       // See the comment in SheetRemoved.
  92       mTable.Clear();
  93       break;
  94     }
  95     case StyleCssRuleType::LayerStatement:
  96     case StyleCssRuleType::FontFace:
  97     case StyleCssRuleType::Page:
  98     case StyleCssRuleType::Property:
  99     case StyleCssRuleType::Keyframes:
 100     case StyleCssRuleType::Keyframe:
 101     case StyleCssRuleType::Margin:
 102     case StyleCssRuleType::Namespace:
 103     case StyleCssRuleType::CounterStyle:
 104     case StyleCssRuleType::FontFeatureValues:
 105     case StyleCssRuleType::FontPaletteValues:
 106       break;
 107   }
 108 }
 109
 110 size_t ServoStyleRuleMap::SizeOfIncludingThis(
 111     MallocSizeOf aMallocSizeOf) const {
 112   size_t n = aMallocSizeOf(this);
 113   n += mTable.ShallowSizeOfExcludingThis(aMallocSizeOf);
 114   return n;
 115 }
 116
 117 void ServoStyleRuleMap::FillTableFromRule(css::Rule& aRule) {
 118   switch (aRule.Type()) {
 119     case StyleCssRuleType::Style: {
 120       auto& rule = static_cast<CSSStyleRule&>(aRule);
 121       mTable.InsertOrUpdate(rule.Raw(), &rule);
 122       [[fallthrough]];
 123     }
 124     case StyleCssRuleType::LayerBlock:
 125     case StyleCssRuleType::Media:
 126     case StyleCssRuleType::Supports:
 127     case StyleCssRuleType::Container:
 128     case StyleCssRuleType::Document:
 129     case StyleCssRuleType::Scope: {
 130       auto& rule = static_cast<css::GroupRule&>(aRule);
 131       FillTableFromRuleList(*rule.CssRules());
 132       break;
 133     }
 134     case StyleCssRuleType::Import: {
 135       auto& rule = static_cast<CSSImportRule&>(aRule);
 136       if (auto* sheet = rule.GetStyleSheet()) {
 137         FillTableFromStyleSheet(*sheet);
 138       }
 139       break;
 140     }
 141     case StyleCssRuleType::LayerStatement:
 142     case StyleCssRuleType::FontFace:
 143     case StyleCssRuleType::Page:
 144     case StyleCssRuleType::Property:
 145     case StyleCssRuleType::Keyframes:
 146     case StyleCssRuleType::Keyframe:
 147     case StyleCssRuleType::Margin:
 148     case StyleCssRuleType::Namespace:
 149     case StyleCssRuleType::CounterStyle:
 150     case StyleCssRuleType::FontFeatureValues:
 151     case StyleCssRuleType::FontPaletteValues:
 152       break;
 153   }
 154 }
 155
 156 void ServoStyleRuleMap::FillTableFromRuleList(ServoCSSRuleList& aRuleList) {
 157   for (uint32_t i : IntegerRange(aRuleList.Length())) {
 158     FillTableFromRule(*aRuleList.GetRule(i));
 159   }
 160 }
 161
 162 void ServoStyleRuleMap::FillTableFromStyleSheet(StyleSheet& aSheet) {
 163   if (aSheet.IsComplete()) {
 164     FillTableFromRuleList(*aSheet.GetCssRulesInternal());
 165   }
 166 }
 167
 168 }  // namespace mozilla