1 files changed, 83 insertions, 0 deletions
diff --git a/layout/base/PresShellInlines.h b/layout/base/PresShellInlines.h
new file mode 100644
index 0000000000..ec3df6d267
--- /dev/null
+++ b/layout/base/PresShellInlines.h
@@ -0,0 +1,83 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef mozilla_PresShellInlines_h
+#define mozilla_PresShellInlines_h
+
+#include "nsDocShell.h"
+#include "GeckoProfiler.h"
+#include "mozilla/PresShell.h"
+#include "mozilla/dom/Document.h"
+#include "mozilla/dom/Element.h"
+
+namespace mozilla {
+
+void PresShell::SetNeedLayoutFlush() {
+  mNeedLayoutFlush = true;
+  if (dom::Document* doc = mDocument->GetDisplayDocument()) {
+    if (PresShell* presShell = doc->GetPresShell()) {
+      presShell->mNeedLayoutFlush = true;
+    }
+  }
+
+  if (!mReflowCause) {
+    mReflowCause = profiler_capture_backtrace();
+  }
+
+  mLayoutTelemetry.IncReqsPerFlush(FlushType::Layout);
+}
+
+void PresShell::SetNeedStyleFlush() {
+  mNeedStyleFlush = true;
+  PROFILER_MARKER_UNTYPED(
+      "SetNeedStyleFlush", LAYOUT,
+      MarkerOptions(MarkerStack::Capture(StackCaptureOptions::NonNative),
+                    mPresContext ? MarkerInnerWindowIdFromDocShell(
+                                       mPresContext->GetDocShell())
+                                 : MarkerInnerWindowId::NoId()));
+
+  if (dom::Document* doc = mDocument->GetDisplayDocument()) {
+    if (PresShell* presShell = doc->GetPresShell()) {
+      presShell->mNeedStyleFlush = true;
+    }
+  }
+
+  if (!mStyleCause) {
+    mStyleCause = profiler_capture_backtrace();
+  }
+
+  mLayoutTelemetry.IncReqsPerFlush(FlushType::Layout);
+}
+
+void PresShell::EnsureStyleFlush() {
+  SetNeedStyleFlush();
+  ObserveStyleFlushes();
+}
+
+void PresShell::SetNeedThrottledAnimationFlush() {
+  mNeedThrottledAnimationFlush = true;
+  if (dom::Document* doc = mDocument->GetDisplayDocument()) {
+    if (PresShell* presShell = doc->GetPresShell()) {
+      presShell->mNeedThrottledAnimationFlush = true;
+    }
+  }
+}
+
+ServoStyleSet* PresShell::StyleSet() const {
+  return mDocument->StyleSetForPresShellOrMediaQueryEvaluation();
+}
+
+/* static */
+inline void PresShell::EventHandler::OnPresShellDestroy(Document* aDocument) {
+  if (sLastKeyDownEventTargetElement &&
+      sLastKeyDownEventTargetElement->OwnerDoc() == aDocument) {
+    sLastKeyDownEventTargetElement = nullptr;
+  }
+}
+
+}  // namespace mozilla
+
+#endif  // mozilla_PresShellInlines_h