layout/

layout_impl.rs

/* 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 https://mozilla.org/MPL/2.0/. */

#![expect(unsafe_code)]

use std::cell::{Cell, OnceCell, RefCell};
use std::collections::HashMap;
use std::fmt::Debug;
use std::rc::Rc;
use std::sync::{Arc, LazyLock};

use app_units::Au;
use bitflags::bitflags;
use embedder_traits::{
    EmbedderMsg, ScriptToEmbedderChan, Theme, UntrustedNodeAddress, ViewportDetails,
};
use euclid::{Point2D, Rect, Scale, Size2D};
use fonts::{FontContext, FontContextWebFontMethods, WebFontDocumentContext};
use fonts_traits::StylesheetWebFontLoadFinishedCallback;
use icu_locid::subtags::Language;
use layout_api::{
    AxesOverflow, BoxAreaType, CSSPixelRectVec, DangerousStyleNode, IFrameSizes, Layout,
    LayoutConfig, LayoutDamage, LayoutElement, LayoutFactory, LayoutNode, NodeRenderingType,
    OffsetParentResponse, PhysicalSides, QueryMsg, ReflowGoal, ReflowPhasesRun, ReflowRequest,
    ReflowRequestRestyle, ReflowResult, ReflowStatistics, ScrollContainerQueryFlags,
    ScrollContainerResponse, TrustedNodeAddress, with_layout_state,
};
use log::{debug, warn};
use malloc_size_of::{MallocConditionalSizeOf, MallocSizeOf, MallocSizeOfOps};
use net_traits::image_cache::ImageCache;
use paint_api::CrossProcessPaintApi;
use paint_api::display_list::{AxesScrollSensitivity, PaintDisplayListInfo, ScrollType};
use parking_lot::{Mutex, RwLock};
use profile_traits::mem::{Report, ReportKind};
use profile_traits::time::{
    self as profile_time, TimerMetadata, TimerMetadataFrameType, TimerMetadataReflowType,
};
use profile_traits::{path, time_profile};
use rustc_hash::FxHashMap;
use script::layout_dom::{
    ServoDangerousStyleDocument, ServoDangerousStyleElement, ServoLayoutElement, ServoLayoutNode,
};
use script_traits::{DrawAPaintImageResult, PaintWorkletError, Painter, ScriptThreadMessage};
use servo_arc::Arc as ServoArc;
use servo_base::Epoch;
use servo_base::generic_channel::GenericSender;
use servo_base::id::{PipelineId, WebViewId};
use servo_config::opts::{self, DiagnosticsLogging, DiagnosticsLoggingOption};
use servo_config::pref;
use servo_url::ServoUrl;
use style::animation::DocumentAnimationSet;
use style::context::{
    QuirksMode, RegisteredSpeculativePainter, RegisteredSpeculativePainters, SharedStyleContext,
};
use style::device::Device;
use style::device::servo::FontMetricsProvider;
use style::dom::{OpaqueNode, ShowSubtreeDataAndPrimaryValues, TDocument, TElement, TNode};
use style::font_metrics::FontMetrics;
use style::global_style_data::GLOBAL_STYLE_DATA;
use style::invalidation::element::restyle_hints::RestyleHint;
use style::invalidation::stylesheets::StylesheetInvalidationSet;
use style::media_queries::{MediaList, MediaType};
use style::properties::style_structs::Font;
use style::properties::{ComputedValues, PropertyId};
use style::queries::values::PrefersColorScheme;
use style::selector_parser::{PseudoElement, SnapshotMap};
use style::servo::media_features::PointerCapabilities;
use style::shared_lock::{SharedRwLock, SharedRwLockReadGuard, StylesheetGuards};
use style::stylesheets::{
    CustomMediaMap, DocumentStyleSheet, Origin, Stylesheet, StylesheetInDocument,
};
use style::stylist::Stylist;
use style::traversal::DomTraversal;
use style::traversal_flags::TraversalFlags;
use style::values::computed::font::GenericFontFamily;
use style::values::computed::{CSSPixelLength, FontSize, Length, NonNegativeLength};
use style::values::specified::font::{KeywordInfo, QueryFontMetricsFlags};
use style::{Zero, driver};
use style_traits::{CSSPixel, SpeculativePainter};
use stylo_atoms::Atom;
use url::Url;
use webrender_api::ExternalScrollId;
use webrender_api::units::{DevicePixel, LayoutVector2D};

use crate::accessibility_tree::AccessibilityTree;
use crate::context::{CachedImageOrError, ImageResolver, LayoutContext};
use crate::display_list::{DisplayListBuilder, HitTest, PaintTimingHandler, StackingContextTree};
use crate::dom::NodeExt;
use crate::query::{
    find_character_offset_in_fragment_descendants, get_the_text_steps, process_box_area_request,
    process_box_areas_request, process_client_rect_request,
    process_containing_block_descendant_query, process_containing_block_query,
    process_current_css_zoom_query, process_effective_overflow_query,
    process_node_scroll_area_request, process_offset_parent_query, process_padding_request,
    process_resolved_font_style_query, process_resolved_style_request,
    process_scroll_container_query,
};
use crate::traversal::{RecalcStyle, compute_damage_and_rebuild_box_tree};
use crate::{BoxTree, FragmentTree};

// This mutex is necessary due to syncronisation issues between two different types of thread-local storage
// which manifest themselves when the layout thread tries to layout iframes in parallel with the main page
//
// See: https://github.com/servo/servo/pull/29792
// And: https://gist.github.com/mukilan/ed57eb61b83237a05fbf6360ec5e33b0
static STYLE_THREAD_POOL: Mutex<&LazyLock<style::global_style_data::StyleThreadPool>> =
    Mutex::new(&style::global_style_data::STYLE_THREAD_POOL);

/// A CSS file to style the user agent stylesheet.
static USER_AGENT_CSS: &[u8] = include_bytes!("./stylesheets/user-agent.css");

/// A CSS file to style the user agent stylesheet in HTML documents.
static HTML_MODE_CSS: &[u8] = include_bytes!("./stylesheets/html-mode.css");

/// A CSS file to style the Servo browser.
static SERVO_CSS: &[u8] = include_bytes!("./stylesheets/servo.css");

/// A CSS file to style the presentational hints.
static PRESENTATIONAL_HINTS_CSS: &[u8] = include_bytes!("./stylesheets/presentational-hints.css");

/// A CSS file to style the quirks mode.
static QUIRKS_MODE_CSS: &[u8] = include_bytes!("./stylesheets/quirks-mode.css");

/// Information needed by layout.
pub struct LayoutThread {
    /// The ID of the pipeline that we belong to.
    id: PipelineId,

    /// The webview that contains the pipeline we belong to.
    webview_id: WebViewId,

    /// The URL of the pipeline that we belong to.
    url: ServoUrl,

    /// Performs CSS selector matching and style resolution.
    stylist: Stylist,

    /// Is the current reflow of an iframe, as opposed to a root window?
    is_iframe: bool,

    /// The channel on which messages can be sent to the script thread.
    script_chan: GenericSender<ScriptThreadMessage>,

    /// The channel on which messages can be sent to the time profiler.
    time_profiler_chan: profile_time::ProfilerChan,

    /// The channel to send messages to the Embedder.
    embedder_chan: ScriptToEmbedderChan,

    /// Reference to the script thread image cache.
    image_cache: Arc<dyn ImageCache>,

    /// A FontContext to be used during layout.
    font_context: Arc<FontContext>,

    /// Whether or not user agent stylesheets have been added to the Stylist or not.
    have_added_user_agent_stylesheets: bool,

    // A vector of parsed `DocumentStyleSheet`s representing the corresponding `UserStyleSheet`s
    // associated with the `WebView` to which this `Layout` belongs. The `DocumentStylesheet`s might
    // be shared with `Layout`s in the same `ScriptThread`.
    user_stylesheets: Rc<Vec<DocumentStyleSheet>>,

    /// Whether or not this [`LayoutImpl`]'s [`Device`] has changed since the last restyle.
    /// If it has, a restyle is pending.
    device_has_changed: bool,

    /// Is this the first reflow in this LayoutThread?
    have_ever_generated_display_list: Cell<bool>,

    /// Whether the last display list we sent was effectively empty.
    last_display_list_was_empty: Cell<bool>,

    /// Whether a new display list is necessary due to changes to layout or stacking
    /// contexts. This is set to true every time layout changes, even when a display list
    /// isn't requested for this layout, such as for layout queries. The next time a
    /// layout requests a display list, it is produced unconditionally, even when the
    /// layout trees remain the same.
    need_new_display_list: Cell<bool>,

    /// Whether or not cumulative containing blocks offsets have been set into the
    /// [`FragmentTree`]. This typically happens during [`StackingContextTree`]
    /// construction, but if a layout query needs these value beforehand, they are
    /// eagerly calculated.
    need_containing_block_calculation: Cell<bool>,

    /// Whether or not the existing stacking context tree is dirty and needs to be
    /// rebuilt. This happens after a relayout or overflow update. The reason that we
    /// don't simply clear the stacking context tree when it becomes dirty is that we need
    /// to preserve scroll offsets from the old tree to the new one.
    need_new_stacking_context_tree: Cell<bool>,

    /// The box tree.
    box_tree: RefCell<Option<Arc<BoxTree>>>,

    /// The fragment tree.
    fragment_tree: RefCell<Option<Rc<FragmentTree>>>,

    /// The [`StackingContextTree`] cached from previous layouts.
    stacking_context_tree: RefCell<Option<StackingContextTree>>,

    // A cache that maps image resources specified in CSS (e.g as the `url()` value
    // for `background-image` or `content` properties) to either the final resolved
    // image data, or an error if the image cache failed to load/decode the image.
    resolved_images_cache: Arc<RwLock<HashMap<ServoUrl, CachedImageOrError>>>,

    /// The executors for paint worklets.
    registered_painters: RegisteredPaintersImpl,

    /// Cross-process access to the `Paint` API.
    paint_api: CrossProcessPaintApi,

    /// Debug options, copied from configuration to this `LayoutThread` in order
    /// to avoid having to constantly access the thread-safe global options.
    debug: DiagnosticsLogging,

    /// Tracks the node that was highlighted by the devtools during the last reflow.
    ///
    /// If this changed, then we need to create a new display list.
    previously_highlighted_dom_node: Cell<Option<OpaqueNode>>,

    /// Handler for all Paint Timings
    paint_timing_handler: RefCell<Option<PaintTimingHandler>>,

    /// Whether accessibility is active for this Layout.
    accessibility_active: Cell<bool>,

    /// Layout's internal representation of its accessibility tree.
    /// This is `None` if accessibility is not active.
    accessibility_tree: RefCell<Option<AccessibilityTree>>,

    /// See [Layout::needs_accessibility_update()].
    needs_accessibility_update: Cell<bool>,
}

pub struct LayoutFactoryImpl();

impl LayoutFactory for LayoutFactoryImpl {
    fn create(&self, config: LayoutConfig) -> Box<dyn Layout> {
        Box::new(LayoutThread::new(config))
    }
}

impl Drop for LayoutThread {
    fn drop(&mut self) {
        let (keys, instance_keys) = self
            .font_context
            .collect_unused_webrender_resources(true /* all */);
        self.paint_api
            .remove_unused_font_resources(self.webview_id.into(), keys, instance_keys)
    }
}

impl Layout for LayoutThread {
    fn device(&self) -> &Device {
        self.stylist.device()
    }

    fn set_theme(&mut self, theme: Theme) -> bool {
        let theme: PrefersColorScheme = theme.into();
        let device = self.stylist.device_mut();
        if theme == device.color_scheme() {
            return false;
        }

        device.set_color_scheme(theme);
        self.device_has_changed = true;
        true
    }

    fn set_viewport_details(&mut self, viewport_details: ViewportDetails) -> bool {
        let device = self.stylist.device_mut();
        let device_pixel_ratio = Scale::new(viewport_details.hidpi_scale_factor.get());
        if device.viewport_size() == viewport_details.size &&
            device.device_pixel_ratio() == device_pixel_ratio
        {
            return false;
        }

        device.set_viewport_size(viewport_details.size);
        device.set_device_pixel_ratio(device_pixel_ratio);
        self.device_has_changed = true;
        true
    }

    fn load_web_fonts_from_stylesheet(
        &self,
        stylesheet: &ServoArc<Stylesheet>,
        document_context: &WebFontDocumentContext,
    ) {
        let guard = stylesheet.shared_lock.read();
        self.load_all_web_fonts_from_stylesheet_with_guard(
            &DocumentStyleSheet(stylesheet.clone()),
            &guard,
            document_context,
        );
    }

    #[servo_tracing::instrument(skip_all)]
    fn add_stylesheet(
        &mut self,
        stylesheet: ServoArc<Stylesheet>,
        before_stylesheet: Option<ServoArc<Stylesheet>>,
        document_context: &WebFontDocumentContext,
    ) {
        let guard = stylesheet.shared_lock.read();
        let stylesheet = DocumentStyleSheet(stylesheet.clone());
        self.load_all_web_fonts_from_stylesheet_with_guard(&stylesheet, &guard, document_context);

        match before_stylesheet {
            Some(insertion_point) => self.stylist.insert_stylesheet_before(
                stylesheet,
                DocumentStyleSheet(insertion_point),
                &guard,
            ),
            None => self.stylist.append_stylesheet(stylesheet, &guard),
        }
    }

    #[servo_tracing::instrument(skip_all)]
    fn remove_stylesheet(&mut self, stylesheet: ServoArc<Stylesheet>) {
        let guard = stylesheet.shared_lock.read();
        let stylesheet = DocumentStyleSheet(stylesheet.clone());
        self.stylist.remove_stylesheet(stylesheet.clone(), &guard);
        self.font_context
            .remove_all_web_fonts_from_stylesheet(&stylesheet);
    }

    #[servo_tracing::instrument(skip_all)]
    fn remove_cached_image(&mut self, url: &ServoUrl) {
        let mut resolved_images_cache = self.resolved_images_cache.write();
        resolved_images_cache.remove(url);
    }

    fn node_rendering_type(
        &self,
        node: TrustedNodeAddress,
        pseudo: Option<PseudoElement>,
    ) -> NodeRenderingType {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };

            // Nodes that are not currently styled are never being rendered.
            if node
                .as_element()
                .is_none_or(|element| element.style_data().is_none())
            {
                return NodeRenderingType::NotRendered;
            }

            let node = match pseudo {
                Some(pseudo) => node.with_pseudo(pseudo),
                None => Some(node),
            };
            let Some(node) = node else {
                return NodeRenderingType::NotRendered;
            };
            node.rendering_type()
        })
    }

    /// Return the node corresponding to the containing block of the provided node.
    #[servo_tracing::instrument(skip_all)]
    fn query_containing_block(&self, node: TrustedNodeAddress) -> Option<UntrustedNodeAddress> {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            process_containing_block_query(node)
        })
    }

    /// Return the node corresponding to the containing block of the provided node.
    #[servo_tracing::instrument(skip_all)]
    fn query_containing_block_is_descendant(
        &self,
        root: TrustedNodeAddress,
        possible_descendant: TrustedNodeAddress,
    ) -> bool {
        with_layout_state(|| {
            let (root, possible_descendant) = unsafe {
                (
                    ServoLayoutNode::new(&root),
                    ServoLayoutNode::new(&possible_descendant),
                )
            };
            process_containing_block_descendant_query(root, possible_descendant)
        })
    }

    /// Return the resolved values of this node's padding rect.
    #[servo_tracing::instrument(skip_all)]
    fn query_padding(&self, node: TrustedNodeAddress) -> Option<PhysicalSides> {
        with_layout_state(|| {
            // If we have not built a fragment tree yet, there is no way we have layout information for
            // this query, which can be run without forcing a layout (for IntersectionObserver).
            if self.fragment_tree.borrow().is_none() {
                return None;
            }

            let node = unsafe { ServoLayoutNode::new(&node) };
            process_padding_request(node)
        })
    }

    /// Return the union of this node's areas in the coordinate space of the Document. This is used
    /// to implement `getBoundingClientRect()` and support many other API where the such query is
    /// required.
    ///
    /// Part of <https://drafts.csswg.org/cssom-view-1/#element-get-the-bounding-box>.
    #[servo_tracing::instrument(skip_all)]
    fn query_box_area(
        &self,
        node: TrustedNodeAddress,
        area: BoxAreaType,
        exclude_transform_and_inline: bool,
    ) -> Option<Rect<Au, CSSPixel>> {
        with_layout_state(|| {
            // If we have not built a fragment tree yet, there is no way we have layout information for
            // this query, which can be run without forcing a layout (for IntersectionObserver).
            if self.fragment_tree.borrow().is_none() {
                return None;
            }

            let node = unsafe { ServoLayoutNode::new(&node) };
            let stacking_context_tree = self.stacking_context_tree.borrow();
            let stacking_context_tree = stacking_context_tree.as_ref()?;
            process_box_area_request(
                self,
                stacking_context_tree,
                node,
                area,
                exclude_transform_and_inline,
            )
        })
    }

    /// Get a `Vec` of bounding boxes of this node's `Fragment`s specific area in the coordinate space of
    /// the Document. This is used to implement `getClientRects()`.
    ///
    /// See <https://drafts.csswg.org/cssom-view/#dom-element-getclientrects>.
    #[servo_tracing::instrument(skip_all)]
    fn query_box_areas(&self, node: TrustedNodeAddress, area: BoxAreaType) -> CSSPixelRectVec {
        with_layout_state(|| {
            // If we have not built a fragment tree yet, there is no way we have layout information for
            // this query, which can be run without forcing a layout (for IntersectionObserver).
            if self.fragment_tree.borrow().is_none() {
                return None;
            }

            let node = unsafe { ServoLayoutNode::new(&node) };
            let stacking_context_tree = self.stacking_context_tree.borrow();
            let stacking_context_tree = stacking_context_tree.as_ref()?;
            Some(process_box_areas_request(
                self,
                stacking_context_tree,
                node,
                area,
            ))
        })
        .unwrap_or_default()
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_client_rect(&self, node: TrustedNodeAddress) -> Rect<i32, CSSPixel> {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            process_client_rect_request(node)
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_current_css_zoom(&self, node: TrustedNodeAddress) -> f32 {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            process_current_css_zoom_query(node)
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_element_inner_outer_text(&self, node: layout_api::TrustedNodeAddress) -> String {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            get_the_text_steps(node)
        })
    }
    #[servo_tracing::instrument(skip_all)]
    fn query_offset_parent(&self, node: TrustedNodeAddress) -> OffsetParentResponse {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            let stacking_context_tree = self.stacking_context_tree.borrow();
            let stacking_context_tree = stacking_context_tree.as_ref()?;
            process_offset_parent_query(self, &stacking_context_tree.paint_info.scroll_tree, node)
        })
        .unwrap_or_default()
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_scroll_container(
        &self,
        node: Option<TrustedNodeAddress>,
        flags: ScrollContainerQueryFlags,
    ) -> Option<ScrollContainerResponse> {
        with_layout_state(|| {
            let node = unsafe { node.as_ref().map(|node| ServoLayoutNode::new(node)) };
            let viewport_overflow = self.box_tree.borrow().as_ref()?.viewport_overflow;
            process_scroll_container_query(node, flags, viewport_overflow)
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_resolved_style(
        &self,
        node: TrustedNodeAddress,
        pseudo: Option<PseudoElement>,
        property_id: PropertyId,
        animations: DocumentAnimationSet,
        animation_timeline_value: f64,
    ) -> String {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            let document = unsafe { node.dangerous_style_node() }.owner_doc();
            let shared_locks = document.shared_style_locks();
            let guards = StylesheetGuards {
                author: &shared_locks.author.read(),
                ua_or_user: &shared_locks.ua_or_user.read(),
            };
            let snapshot_map = SnapshotMap::new();

            let shared_style_context = self.build_shared_style_context(
                guards,
                &snapshot_map,
                animation_timeline_value,
                &animations,
                TraversalFlags::empty(),
            );

            process_resolved_style_request(self, &shared_style_context, node, &pseudo, &property_id)
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_resolved_font_style(
        &self,
        node: TrustedNodeAddress,
        value: &str,
        animations: DocumentAnimationSet,
        animation_timeline_value: f64,
    ) -> Option<ServoArc<Font>> {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            let document = unsafe { node.dangerous_style_node() }.owner_doc();
            let shared_locks = document.shared_style_locks();
            let shared_author_lock = &shared_locks.author;
            let guards = StylesheetGuards {
                author: &shared_author_lock.read(),
                ua_or_user: &shared_locks.ua_or_user.read(),
            };
            let snapshot_map = SnapshotMap::new();
            let shared_style_context = self.build_shared_style_context(
                guards,
                &snapshot_map,
                animation_timeline_value,
                &animations,
                TraversalFlags::empty(),
            );

            process_resolved_font_style_query(
                &shared_style_context,
                node,
                value,
                self.url.clone(),
                shared_author_lock,
            )
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_scrolling_area(&self, node: Option<TrustedNodeAddress>) -> Rect<i32, CSSPixel> {
        with_layout_state(|| {
            let node = node.map(|node| unsafe { ServoLayoutNode::new(&node) });
            process_node_scroll_area_request(self, node, self.fragment_tree.borrow().clone())
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_text_index(
        &self,
        node: TrustedNodeAddress,
        point_in_node: Point2D<Au, CSSPixel>,
    ) -> Option<usize> {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            let stacking_context_tree = self.stacking_context_tree.borrow_mut();
            let stacking_context_tree = stacking_context_tree.as_ref()?;
            find_character_offset_in_fragment_descendants(
                &node,
                stacking_context_tree,
                point_in_node,
            )
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_elements_from_point(
        &self,
        point: webrender_api::units::LayoutPoint,
    ) -> Vec<layout_api::ElementsFromPointResult> {
        with_layout_state(|| {
            self.stacking_context_tree
                .borrow_mut()
                .as_mut()
                .map(|tree| HitTest::run(tree, point))
                .unwrap_or_default()
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn query_effective_overflow(&self, node: TrustedNodeAddress) -> Option<AxesOverflow> {
        with_layout_state(|| {
            let node = unsafe { ServoLayoutNode::new(&node) };
            process_effective_overflow_query(node)
        })
    }

    fn exit_now(&mut self) {}

    fn collect_reports(&self, reports: &mut Vec<Report>, ops: &mut MallocSizeOfOps) {
        // TODO: Measure more than just display list, stylist, and font context.
        let formatted_url = &format!("url({})", self.url);
        reports.push(Report {
            path: path![formatted_url, "layout-thread", "display-list"],
            kind: ReportKind::ExplicitJemallocHeapSize,
            size: 0,
        });

        reports.push(Report {
            path: path![formatted_url, "layout-thread", "stylist"],
            kind: ReportKind::ExplicitJemallocHeapSize,
            size: self.stylist.size_of(ops),
        });

        reports.push(Report {
            path: path![formatted_url, "layout-thread", "font-context"],
            kind: ReportKind::ExplicitJemallocHeapSize,
            size: self.font_context.conditional_size_of(ops),
        });

        reports.push(Report {
            path: path![formatted_url, "layout-thread", "box-tree"],
            kind: ReportKind::ExplicitJemallocHeapSize,
            size: self
                .box_tree
                .borrow()
                .as_ref()
                .map_or(0, |tree| tree.conditional_size_of(ops)),
        });

        reports.push(Report {
            path: path![formatted_url, "layout-thread", "fragment-tree"],
            kind: ReportKind::ExplicitJemallocHeapSize,
            size: self
                .fragment_tree
                .borrow()
                .as_ref()
                .map(|tree| tree.conditional_size_of(ops))
                .unwrap_or_default(),
        });

        reports.push(Report {
            path: path![formatted_url, "layout-thread", "stacking-context-tree"],
            kind: ReportKind::ExplicitJemallocHeapSize,
            size: self.stacking_context_tree.size_of(ops),
        });

        reports.extend(self.image_cache.memory_reports(formatted_url, ops));
    }

    fn set_quirks_mode(&mut self, quirks_mode: QuirksMode) {
        self.stylist.set_quirks_mode(quirks_mode);
    }

    fn reflow(&mut self, reflow_request: ReflowRequest) -> Option<ReflowResult> {
        time_profile!(
            profile_time::ProfilerCategory::Layout,
            self.profiler_metadata(),
            self.time_profiler_chan.clone(),
            || with_layout_state(|| self.handle_reflow(reflow_request)),
        )
    }

    fn ensure_stacking_context_tree(&self, viewport_details: ViewportDetails) {
        with_layout_state(|| {
            if self.stacking_context_tree.borrow().is_some() &&
                !self.need_new_stacking_context_tree.get()
            {
                return;
            }
            self.build_stacking_context_tree(viewport_details);
        })
    }

    fn register_paint_worklet_modules(
        &mut self,
        _name: Atom,
        _properties: Vec<Atom>,
        _painter: Box<dyn Painter>,
    ) {
    }

    fn set_scroll_offsets_from_renderer(
        &mut self,
        scroll_states: &FxHashMap<ExternalScrollId, LayoutVector2D>,
    ) {
        let mut stacking_context_tree = self.stacking_context_tree.borrow_mut();
        let Some(stacking_context_tree) = stacking_context_tree.as_mut() else {
            warn!("Received scroll offsets before finishing layout.");
            return;
        };

        stacking_context_tree
            .paint_info
            .scroll_tree
            .set_all_scroll_offsets(scroll_states);
    }

    fn scroll_offset(&self, id: ExternalScrollId) -> Option<LayoutVector2D> {
        self.stacking_context_tree
            .borrow_mut()
            .as_mut()
            .and_then(|tree| tree.paint_info.scroll_tree.scroll_offset(id))
    }

    fn needs_new_display_list(&self) -> bool {
        self.need_new_display_list.get()
    }

    fn set_needs_new_display_list(&self) {
        self.need_new_display_list.set(true);
    }

    /// <https://drafts.css-houdini.org/css-properties-values-api-1/#the-registerproperty-function>
    fn stylist_mut(&mut self) -> &mut Stylist {
        &mut self.stylist
    }

    fn set_accessibility_active(&self, active: bool, epoch: Epoch) {
        self.accessibility_active.set(active);
        if !active {
            self.accessibility_tree.replace(None);
            return;
        }

        self.set_needs_accessibility_update();
        let mut accessibility_tree = self.accessibility_tree.borrow_mut();
        if accessibility_tree.is_some() {
            return;
        }
        *accessibility_tree = Some(AccessibilityTree::new(self.id.into(), epoch));
    }

    fn accessibility_active(&self) -> bool {
        self.accessibility_active.get()
    }

    fn needs_accessibility_update(&self) -> bool {
        self.needs_accessibility_update.get()
    }

    fn set_needs_accessibility_update(&self) {
        self.needs_accessibility_update.set(true);
    }
}

impl LayoutThread {
    fn new(config: LayoutConfig) -> LayoutThread {
        // Let webrender know about this pipeline by sending an empty display list.
        config
            .paint_api
            .send_initial_transaction(config.webview_id, config.id.into());

        let mut font = Font::initial_values();
        let default_font_size = pref!(fonts_default_size);
        font.font_size = FontSize {
            computed_size: NonNegativeLength::new(default_font_size as f32),
            used_size: NonNegativeLength::new(default_font_size as f32),
            keyword_info: KeywordInfo::medium(),
        };

        // The device pixel ratio is incorrect (it does not have the hidpi value),
        // but it will be set correctly when the initial reflow takes place.
        let device = Device::new(
            MediaType::screen(),
            QuirksMode::NoQuirks,
            config.viewport_details.size,
            Scale::new(config.viewport_details.hidpi_scale_factor.get()),
            Box::new(LayoutFontMetricsProvider(config.font_context.clone())),
            ComputedValues::initial_values_with_font_override(font),
            config.theme.into(),
            PointerCapabilities::default(),
            PointerCapabilities::default(),
        );

        LayoutThread {
            id: config.id,
            webview_id: config.webview_id,
            url: config.url,
            is_iframe: config.is_iframe,
            script_chan: config.script_chan.clone(),
            time_profiler_chan: config.time_profiler_chan,
            embedder_chan: config.embedder_chan.clone(),
            registered_painters: RegisteredPaintersImpl(Default::default()),
            image_cache: config.image_cache,
            font_context: config.font_context,
            have_added_user_agent_stylesheets: false,
            have_ever_generated_display_list: Cell::new(false),
            last_display_list_was_empty: Cell::new(true),
            device_has_changed: false,
            need_containing_block_calculation: Cell::new(false),
            need_new_display_list: Cell::new(false),
            need_new_stacking_context_tree: Cell::new(false),
            box_tree: Default::default(),
            fragment_tree: Default::default(),
            stacking_context_tree: Default::default(),
            paint_api: config.paint_api,
            stylist: Stylist::new(device, QuirksMode::NoQuirks),
            resolved_images_cache: Default::default(),
            debug: opts::get().debug.clone(),
            previously_highlighted_dom_node: Cell::new(None),
            paint_timing_handler: Default::default(),
            user_stylesheets: config.user_stylesheets,
            accessibility_active: Cell::new(false),
            accessibility_tree: Default::default(),
            needs_accessibility_update: Cell::new(false),
        }
    }

    fn build_shared_style_context<'a>(
        &'a self,
        guards: StylesheetGuards<'a>,
        snapshot_map: &'a SnapshotMap,
        animation_timeline_value: f64,
        animations: &DocumentAnimationSet,
        traversal_flags: TraversalFlags,
    ) -> SharedStyleContext<'a> {
        SharedStyleContext {
            stylist: &self.stylist,
            options: GLOBAL_STYLE_DATA.options.clone(),
            guards,
            visited_styles_enabled: false,
            animations: animations.clone(),
            registered_speculative_painters: &self.registered_painters,
            current_time_for_animations: animation_timeline_value,
            traversal_flags,
            snapshot_map,
        }
    }

    fn load_all_web_fonts_from_stylesheet_with_guard(
        &self,
        stylesheet: &DocumentStyleSheet,
        guard: &SharedRwLockReadGuard,
        document_context: &WebFontDocumentContext,
    ) {
        let custom_media = &CustomMediaMap::default();
        if !stylesheet.is_effective_for_device(self.stylist.device(), custom_media, guard) {
            return;
        }

        let locked_script_channel = Mutex::new(self.script_chan.clone());
        let pipeline_id = self.id;
        let web_font_finished_loading_callback = move |succeeded: bool| {
            if succeeded {
                let _ = locked_script_channel
                    .lock()
                    .send(ScriptThreadMessage::WebFontLoaded(pipeline_id));
            }
        };

        self.font_context.add_all_web_fonts_from_stylesheet(
            self.webview_id,
            stylesheet,
            guard,
            self.stylist.device(),
            Arc::new(web_font_finished_loading_callback) as StylesheetWebFontLoadFinishedCallback,
            document_context,
        );
    }

    /// In some cases, if a restyle isn't necessary we can skip doing any work for layout
    /// entirely. This check allows us to return early from layout without doing any work
    /// at all.
    fn can_skip_reflow_request_entirely(&self, reflow_request: &ReflowRequest) -> bool {
        // If a restyle is necessary, restyle and reflow is a necessity.
        if reflow_request.restyle.is_some() {
            return false;
        }
        // We always need to at least build a fragment tree.
        if self.fragment_tree.borrow().is_none() {
            return false;
        }
        // If accessibility was just activated, we need reflow to build the accessibility tree.
        if self.needs_accessibility_update() {
            return false;
        }

        // If we have a fragment tree and it's up-to-date and this reflow
        // doesn't need more reflow results, we can skip the rest of layout.
        let necessary_phases = ReflowPhases::necessary(&reflow_request.reflow_goal);
        if necessary_phases.is_empty() {
            return true;
        }

        // If only the stacking context tree is required, and it's up-to-date,
        // layout is unnecessary, otherwise a layout is necessary.
        if necessary_phases == ReflowPhases::StackingContextTreeConstruction {
            return self.stacking_context_tree.borrow().is_some() &&
                !self.need_new_stacking_context_tree.get();
        }

        // Otherwise, the only interesting thing is whether the current display
        // list is up-to-date.
        assert_eq!(
            necessary_phases,
            ReflowPhases::StackingContextTreeConstruction | ReflowPhases::DisplayListConstruction
        );
        !self.need_new_display_list.get()
    }

    fn maybe_print_reflow_event(&self, reflow_request: &ReflowRequest) {
        if !self
            .debug
            .is_enabled(DiagnosticsLoggingOption::RelayoutEvent)
        {
            return;
        }

        println!(
            "**** Reflow({}) => {:?}, {:?}",
            self.id,
            reflow_request.reflow_goal,
            reflow_request
                .restyle
                .as_ref()
                .map(|restyle| restyle.reason)
                .unwrap_or_default()
        );
    }

    /// Checks whether we need to update the scroll node, and report whether the
    /// node is scrolled. We need to update the scroll node whenever it is requested.
    fn handle_update_scroll_node_request(&self, reflow_request: &ReflowRequest) -> bool {
        if let ReflowGoal::UpdateScrollNode(external_scroll_id, offset) = reflow_request.reflow_goal
        {
            self.set_scroll_offset_from_script(external_scroll_id, offset)
        } else {
            false
        }
    }

    fn handle_accessibility_tree_update(
        &self,
        root_element: &ServoLayoutNode,
        reflow_request: &mut ReflowRequest,
    ) -> bool {
        if !self.needs_accessibility_update() {
            return false;
        }
        let mut accessibility_tree = self.accessibility_tree.borrow_mut();
        let Some(accessibility_tree) = accessibility_tree.as_mut() else {
            return false;
        };

        let accessibility_tree = &mut *accessibility_tree;
        let rooted_nodes =
            std::mem::take(&mut reflow_request.rooted_nodes_for_accessibility_integrity_check);

        if let Some(tree_update) = accessibility_tree.update_tree(root_element, rooted_nodes) {
            // FIXME: Handle send error. Could have a method on accessibility tree to
            // finalise after sending, removing accessibility damage? On fail, retain damage
            // for next reflow, as well as retaining document.needs_accessibility_update.
            let _ = self
                .embedder_chan
                .send(EmbedderMsg::AccessibilityTreeUpdate(
                    self.webview_id,
                    tree_update,
                    accessibility_tree.embedder_epoch(),
                ));
        }
        self.needs_accessibility_update.set(false);
        true
    }

    /// The high-level routine that performs layout.
    #[servo_tracing::instrument(skip_all)]
    fn handle_reflow(&mut self, mut reflow_request: ReflowRequest) -> Option<ReflowResult> {
        self.maybe_print_reflow_event(&reflow_request);

        if self.can_skip_reflow_request_entirely(&reflow_request) {
            // We can skip layout, but we might need to update a scroll node.
            return self
                .handle_update_scroll_node_request(&reflow_request)
                .then(|| ReflowResult {
                    reflow_phases_run: ReflowPhasesRun::UpdatedScrollNodeOffset,
                    ..Default::default()
                });
        }

        let document = unsafe { ServoLayoutNode::new(&reflow_request.document) };
        let document = unsafe { document.dangerous_style_node() }
            .as_document()
            .unwrap();
        let Some(root_element) = document.root_element() else {
            if !self.last_display_list_was_empty.get() {
                return self.clear_layout_trees_and_send_empty_display_list(&reflow_request);
            }
            debug!("layout: No root node: bailing");
            return None;
        };

        let image_resolver = Arc::new(ImageResolver {
            origin: reflow_request.origin.clone(),
            image_cache: self.image_cache.clone(),
            resolved_images_cache: self.resolved_images_cache.clone(),
            pending_images: Mutex::default(),
            pending_rasterization_images: Mutex::default(),
            pending_svg_elements_for_serialization: Mutex::default(),
            animating_images: reflow_request.animating_images.clone(),
            animation_timeline_value: reflow_request.animation_timeline_value,
        });
        let mut reflow_statistics = Default::default();

        let (mut reflow_phases_run, iframe_sizes) = self.restyle_and_build_trees(
            &mut reflow_request,
            document,
            root_element,
            &image_resolver,
        );
        if self.build_stacking_context_tree_for_reflow(&reflow_request) {
            reflow_phases_run.insert(ReflowPhasesRun::BuiltStackingContextTree);
        }
        if self.build_display_list(&reflow_request, &image_resolver, &mut reflow_statistics) {
            reflow_phases_run.insert(ReflowPhasesRun::BuiltDisplayList);
        }
        if self.handle_update_scroll_node_request(&reflow_request) {
            reflow_phases_run.insert(ReflowPhasesRun::UpdatedScrollNodeOffset);
        }
        if self.handle_accessibility_tree_update(&root_element.as_node(), &mut reflow_request) {
            reflow_phases_run.insert(ReflowPhasesRun::UpdatedAccessibilityTree);
        }

        if self.debug.is_enabled(DiagnosticsLoggingOption::FlowTree) &&
            reflow_phases_run.contains(ReflowPhasesRun::RanLayout) &&
            let Some(fragment_tree) = &*self.fragment_tree.borrow()
        {
            fragment_tree.print();
        }

        let pending_images = std::mem::take(&mut *image_resolver.pending_images.lock());
        let pending_rasterization_images =
            std::mem::take(&mut *image_resolver.pending_rasterization_images.lock());
        let pending_svg_elements_for_serialization =
            std::mem::take(&mut *image_resolver.pending_svg_elements_for_serialization.lock());

        Some(ReflowResult {
            reflow_phases_run,
            pending_images,
            pending_rasterization_images,
            pending_svg_elements_for_serialization,
            iframe_sizes: Some(iframe_sizes),
            reflow_statistics,
        })
    }

    #[servo_tracing::instrument(skip_all)]
    fn prepare_stylist_for_reflow<'dom>(
        &mut self,
        reflow_request: &ReflowRequest,
        document: ServoDangerousStyleDocument<'dom>,
        guards: &StylesheetGuards,
        ua_stylesheets: &UserAgentStylesheets,
    ) -> StylesheetInvalidationSet {
        if !self.have_added_user_agent_stylesheets {
            for stylesheet in &ua_stylesheets.user_agent_stylesheets {
                self.stylist
                    .append_stylesheet(stylesheet.clone(), guards.ua_or_user);
                self.load_all_web_fonts_from_stylesheet_with_guard(
                    stylesheet,
                    guards.ua_or_user,
                    &reflow_request.document_context,
                );
            }

            if document.is_html_document() {
                self.stylist.append_stylesheet(
                    ua_stylesheets.html_mode_stylesheet.clone(),
                    guards.ua_or_user,
                );
                self.load_all_web_fonts_from_stylesheet_with_guard(
                    &ua_stylesheets.html_mode_stylesheet,
                    guards.ua_or_user,
                    &reflow_request.document_context,
                );
            }

            for user_stylesheet in self.user_stylesheets.iter() {
                self.stylist
                    .append_stylesheet(user_stylesheet.clone(), guards.ua_or_user);
                self.load_all_web_fonts_from_stylesheet_with_guard(
                    user_stylesheet,
                    guards.ua_or_user,
                    &reflow_request.document_context,
                );
            }

            if self.stylist.quirks_mode() == QuirksMode::Quirks {
                self.stylist.append_stylesheet(
                    ua_stylesheets.quirks_mode_stylesheet.clone(),
                    guards.ua_or_user,
                );
                self.load_all_web_fonts_from_stylesheet_with_guard(
                    &ua_stylesheets.quirks_mode_stylesheet,
                    guards.ua_or_user,
                    &reflow_request.document_context,
                );
            }
            self.have_added_user_agent_stylesheets = true;
        }

        if reflow_request.stylesheets_changed() {
            self.stylist
                .force_stylesheet_origins_dirty(Origin::Author.into());
        }

        document.flush_shadow_root_stylesheets_if_necessary(&mut self.stylist, guards.author);

        self.stylist.flush(guards)
    }

    #[servo_tracing::instrument(skip_all)]
    fn restyle_and_build_trees(
        &mut self,
        reflow_request: &mut ReflowRequest,
        document: ServoDangerousStyleDocument<'_>,
        root_element: ServoLayoutElement<'_>,
        image_resolver: &Arc<ImageResolver>,
    ) -> (ReflowPhasesRun, IFrameSizes) {
        let mut snapshot_map = SnapshotMap::new();
        let _snapshot_setter = match reflow_request.restyle.as_mut() {
            Some(restyle) => SnapshotSetter::new(restyle, &mut snapshot_map),
            None => return Default::default(),
        };

        let shared_locks = document.shared_style_locks();
        let user_agent_stylesheets = get_ua_stylesheets(&shared_locks.ua_or_user);
        let guards = StylesheetGuards {
            author: &shared_locks.author.read(),
            ua_or_user: &shared_locks.ua_or_user.read(),
        };

        let rayon_pool = STYLE_THREAD_POOL.lock();
        let rayon_pool = rayon_pool.pool();
        let rayon_pool = rayon_pool.as_ref();

        let device_has_changed = std::mem::replace(&mut self.device_has_changed, false);
        let dangerous_root_element = unsafe { root_element.dangerous_style_element() };
        if device_has_changed {
            let sheet_origins_affected_by_device_change = self
                .stylist
                .media_features_change_changed_style(&guards, self.device());
            self.stylist
                .force_stylesheet_origins_dirty(sheet_origins_affected_by_device_change);

            if let Some(mut data) = dangerous_root_element.mutate_data() {
                data.hint.insert(RestyleHint::recascade_subtree());
            }
        }

        self.prepare_stylist_for_reflow(reflow_request, document, &guards, &user_agent_stylesheets)
            .process_style(dangerous_root_element, Some(&snapshot_map));

        if self.previously_highlighted_dom_node.get() != reflow_request.highlighted_dom_node {
            // Need to manually force layout to build a new display list regardless of whether the box tree
            // changed or not.
            self.need_new_display_list.set(true);
        }

        let layout_context = LayoutContext {
            style_context: self.build_shared_style_context(
                guards,
                &snapshot_map,
                reflow_request.animation_timeline_value,
                &reflow_request.animations,
                match reflow_request.stylesheets_changed() {
                    true => TraversalFlags::ForCSSRuleChanges,
                    false => TraversalFlags::empty(),
                },
            ),
            font_context: self.font_context.clone(),
            iframe_sizes: Mutex::default(),
            allow_parallel_layout: rayon_pool.is_some(),
            image_resolver: image_resolver.clone(),
            painter_id: self.webview_id.into(),
            parallelism_job_count_minimum: pref!(layout_parallelism_job_count_minimum) as usize,
            parallelism_job_size_minimum: pref!(layout_parallelism_job_size_minimum) as usize,
        };

        let restyle = reflow_request
            .restyle
            .as_ref()
            .expect("Should not get here if there is not restyle.");

        let recalc_style_traversal;
        let dirty_root;
        {
            let _span = profile_traits::trace_span!("Styling").entered();

            let original_dirty_root = unsafe {
                ServoLayoutNode::new(&restyle.dirty_root.unwrap())
                    .as_element()
                    .unwrap()
                    .dangerous_style_element()
            };

            recalc_style_traversal = RecalcStyle::new(&layout_context);
            let token = {
                let shared = DomTraversal::<ServoDangerousStyleElement>::shared_context(
                    &recalc_style_traversal,
                );
                RecalcStyle::pre_traverse(original_dirty_root, shared)
            };

            if !token.should_traverse() {
                layout_context.style_context.stylist.rule_tree().maybe_gc();
                return Default::default();
            }

            dirty_root = driver::traverse_dom(&recalc_style_traversal, token, rayon_pool).as_node();
        }

        let root_node = root_element.as_node();
        let damage_from_environment = if device_has_changed {
            LayoutDamage::Relayout
        } else {
            LayoutDamage::empty()
        };

        let mut box_tree = self.box_tree.borrow_mut();
        let mut layout_roots = Vec::new();
        let damage = {
            let box_tree = &mut *box_tree;
            let mut compute_damage_and_build_box_tree = || {
                compute_damage_and_rebuild_box_tree(
                    box_tree,
                    &layout_context,
                    dirty_root.layout_node(),
                    root_node,
                    damage_from_environment,
                    &mut layout_roots,
                )
            };

            if let Some(pool) = rayon_pool {
                pool.install(compute_damage_and_build_box_tree)
            } else {
                compute_damage_and_build_box_tree()
            }
        };

        if damage.contains(LayoutDamage::RebuildStackingContextTree) {
            self.need_new_stacking_context_tree.set(true);
        }
        if damage.contains(LayoutDamage::Repaint) {
            self.need_new_display_list.set(true);
        }

        if !damage.contains(LayoutDamage::Relayout) {
            if damage.contains(LayoutDamage::RecalculateOverflow) {
                assert!(self.need_new_display_list.get());
                assert!(self.need_new_stacking_context_tree.get());
                self.fragment_tree
                    .borrow()
                    .as_ref()
                    .expect("Should always have a FragmentTree when layout unnecessary")
                    .clear_scrollable_overflow();
            }

            if !damage.contains(LayoutDamage::DescendantCollectedAsLayoutRoot) {
                layout_context.style_context.stylist.rule_tree().maybe_gc();
                return (ReflowPhasesRun::empty(), IFrameSizes::default());
            }

            debug_assert!(!layout_roots.is_empty());
            if layout_roots
                .into_iter()
                .all(|layout_root| layout_root.try_layout(&layout_context))
            {
                return (
                    ReflowPhasesRun::RanLayout,
                    std::mem::take(&mut *layout_context.iframe_sizes.lock()),
                );
            }
        }

        let box_tree = &*box_tree;
        let viewport_size = self.stylist.device().au_viewport_size();
        let run_layout = || {
            box_tree
                .as_ref()
                .unwrap()
                .layout(recalc_style_traversal.context(), viewport_size)
        };
        let fragment_tree = Rc::new(if let Some(pool) = rayon_pool {
            pool.install(run_layout)
        } else {
            run_layout()
        });

        *self.fragment_tree.borrow_mut() = Some(fragment_tree);

        if self.debug.is_enabled(DiagnosticsLoggingOption::StyleTree) {
            println!(
                "{:?}",
                ShowSubtreeDataAndPrimaryValues(dangerous_root_element.as_node())
            );
        }
        if self.debug.is_enabled(DiagnosticsLoggingOption::RuleTree) {
            recalc_style_traversal
                .context()
                .style_context
                .stylist
                .rule_tree()
                .dump_stdout(&layout_context.style_context.guards);
        }

        // GC the rule tree if some heuristics are met.
        layout_context.style_context.stylist.rule_tree().maybe_gc();

        let mut iframe_sizes = layout_context.iframe_sizes.lock();
        (
            ReflowPhasesRun::RanLayout,
            std::mem::take(&mut *iframe_sizes),
        )
    }

    fn build_stacking_context_tree_for_reflow(&self, reflow_request: &ReflowRequest) -> bool {
        if !ReflowPhases::necessary(&reflow_request.reflow_goal)
            .contains(ReflowPhases::StackingContextTreeConstruction)
        {
            return false;
        }
        if !self.need_new_stacking_context_tree.get() {
            return false;
        }

        self.build_stacking_context_tree(reflow_request.viewport_details)
    }

    #[servo_tracing::instrument(name = "Stacking Context Tree Construction", skip_all)]
    fn build_stacking_context_tree(&self, viewport_details: ViewportDetails) -> bool {
        let Some(fragment_tree) = &*self.fragment_tree.borrow() else {
            return false;
        };

        let mut stacking_context_tree = self.stacking_context_tree.borrow_mut();
        let old_scroll_offsets = stacking_context_tree
            .as_ref()
            .map(|tree| tree.paint_info.scroll_tree.scroll_offsets());

        // This will be done during `StackingContextTree::new` below
        self.need_containing_block_calculation.set(false);

        // Build the StackingContextTree. This turns the `FragmentTree` into a
        // tree of fragments in CSS painting order and also creates all
        // applicable spatial and clip nodes.
        let mut new_stacking_context_tree = StackingContextTree::new(
            fragment_tree,
            viewport_details,
            self.id.into(),
            !self.have_ever_generated_display_list.get(),
            &self.debug,
        );

        // When a new StackingContextTree is built, it contains a freshly built
        // ScrollTree. We want to preserve any existing scroll offsets in that tree,
        // adjusted by any new scroll constraints.
        if let Some(old_scroll_offsets) = old_scroll_offsets {
            new_stacking_context_tree
                .paint_info
                .scroll_tree
                .set_all_scroll_offsets(&old_scroll_offsets);
        }

        if self.debug.is_enabled(DiagnosticsLoggingOption::ScrollTree) {
            new_stacking_context_tree
                .paint_info
                .scroll_tree
                .debug_print();
        }

        *stacking_context_tree = Some(new_stacking_context_tree);

        // The stacking context tree is up-to-date again.
        self.need_new_stacking_context_tree.set(false);
        assert!(self.need_new_display_list.get());

        true
    }

    /// Build the display list for the current layout and send it to the renderer. If no display
    /// list is built, returns false.
    #[servo_tracing::instrument(name = "Display List Construction", skip_all)]
    fn build_display_list(
        &self,
        reflow_request: &ReflowRequest,
        image_resolver: &Arc<ImageResolver>,
        reflow_statistics: &mut ReflowStatistics,
    ) -> bool {
        if !ReflowPhases::necessary(&reflow_request.reflow_goal)
            .contains(ReflowPhases::DisplayListConstruction)
        {
            return false;
        }
        let Some(fragment_tree) = &*self.fragment_tree.borrow() else {
            return false;
        };
        let mut stacking_context_tree = self.stacking_context_tree.borrow_mut();
        let Some(stacking_context_tree) = stacking_context_tree.as_mut() else {
            return false;
        };

        // If a non-display-list-generating reflow updated layout in a previous refow, we
        // cannot skip display list generation here the next time a display list is
        // requested.
        if !self.need_new_display_list.get() {
            return false;
        }

        // TODO: Eventually this should be set when `paint_info` is created, but that requires
        // ensuring that the Epoch is passed to any method that can creates `StackingContextTree`.
        stacking_context_tree.paint_info.epoch = reflow_request.epoch;

        let mut paint_timing_handler = self.paint_timing_handler.borrow_mut();
        // This ensures that we only create the PaintTimingHandler once per layout thread.
        let paint_timing_handler = match paint_timing_handler.as_mut() {
            Some(paint_timing_handler) => paint_timing_handler,
            None => {
                *paint_timing_handler = Some(PaintTimingHandler::new(
                    stacking_context_tree
                        .paint_info
                        .viewport_details
                        .layout_size(),
                ));
                paint_timing_handler.as_mut().unwrap()
            },
        };

        let built_display_list = DisplayListBuilder::build(
            stacking_context_tree,
            fragment_tree,
            image_resolver.clone(),
            self.device().device_pixel_ratio(),
            reflow_request.highlighted_dom_node,
            &self.debug,
            paint_timing_handler,
            reflow_statistics,
        );
        self.paint_api.send_display_list(
            self.webview_id,
            &stacking_context_tree.paint_info,
            built_display_list,
        );

        if paint_timing_handler.did_lcp_candidate_update() &&
            let Some(lcp_candidate) = paint_timing_handler.largest_contentful_paint_candidate()
        {
            self.paint_api.send_lcp_candidate(
                lcp_candidate,
                self.webview_id,
                self.id,
                stacking_context_tree.paint_info.epoch,
            );
            paint_timing_handler.unset_lcp_candidate_updated();
        }

        let (keys, instance_keys) = self
            .font_context
            .collect_unused_webrender_resources(false /* all */);
        self.paint_api
            .remove_unused_font_resources(self.webview_id.into(), keys, instance_keys);
        self.last_display_list_was_empty.set(false);
        self.have_ever_generated_display_list.set(true);
        self.need_new_display_list.set(false);
        self.previously_highlighted_dom_node
            .set(reflow_request.highlighted_dom_node);
        true
    }

    fn set_scroll_offset_from_script(
        &self,
        external_scroll_id: ExternalScrollId,
        offset: LayoutVector2D,
    ) -> bool {
        let mut stacking_context_tree = self.stacking_context_tree.borrow_mut();
        let Some(stacking_context_tree) = stacking_context_tree.as_mut() else {
            return false;
        };

        if let Some(offset) = stacking_context_tree
            .paint_info
            .scroll_tree
            .set_scroll_offset_for_node_with_external_scroll_id(
                external_scroll_id,
                offset,
                ScrollType::Script,
            )
        {
            self.paint_api.scroll_node_by_delta(
                self.webview_id,
                self.id.into(),
                offset,
                external_scroll_id,
            );
            true
        } else {
            false
        }
    }

    /// Returns profiling information which is passed to the time profiler.
    fn profiler_metadata(&self) -> Option<TimerMetadata> {
        Some(TimerMetadata {
            url: self.url.to_string(),
            iframe: if self.is_iframe {
                TimerMetadataFrameType::IFrame
            } else {
                TimerMetadataFrameType::RootWindow
            },
            incremental: if self.have_ever_generated_display_list.get() {
                TimerMetadataReflowType::Incremental
            } else {
                TimerMetadataReflowType::FirstReflow
            },
        })
    }

    /// Clear all cached layout trees and send an empty display list to paint.
    fn clear_layout_trees_and_send_empty_display_list(
        &self,
        reflow_request: &ReflowRequest,
    ) -> Option<ReflowResult> {
        // Clear layout trees.
        self.box_tree.borrow_mut().take();
        self.fragment_tree.borrow_mut().take();
        self.stacking_context_tree.borrow_mut().take();

        // Send empty display list.
        let paint_info = PaintDisplayListInfo::new(
            reflow_request.viewport_details,
            Size2D::zero(),
            self.id.into(),
            reflow_request.epoch,
            AxesScrollSensitivity {
                x: ScrollType::InputEvents | ScrollType::Script,
                y: ScrollType::InputEvents | ScrollType::Script,
            },
            !self.have_ever_generated_display_list.get(),
        );
        let mut builder = webrender_api::DisplayListBuilder::new(paint_info.pipeline_id);
        builder.begin();
        let (_, empty_display_list) = builder.end();

        self.paint_api
            .send_display_list(self.webview_id, &paint_info, empty_display_list);
        self.last_display_list_was_empty.set(true);
        self.have_ever_generated_display_list.set(true);

        Some(ReflowResult {
            reflow_phases_run: ReflowPhasesRun::BuiltDisplayList,
            ..Default::default()
        })
    }

    pub(crate) fn ensure_containing_block_calculation(&self) {
        if !self.need_containing_block_calculation.get() {
            return;
        }
        let fragment_tree = self.fragment_tree.borrow();
        fragment_tree.as_ref().expect("missing fragment tree").find(
            |fragment, _level, containing_block| {
                fragment.set_containing_block(containing_block);
                None::<()>
            },
        );
        self.need_containing_block_calculation.set(false)
    }
}

fn get_ua_stylesheets(shared_lock: &SharedRwLock) -> Rc<UserAgentStylesheets> {
    // There is an assumption here that there is only a single ScriptThread per thread, which
    // is currently the case in Servo. If this were to change, these user agent stylesheets
    // would need to be managed by the ScriptThread instance.
    thread_local! {
        static USER_AGENT_STYLESHEETS: OnceCell<Rc<UserAgentStylesheets>> = const { OnceCell::new() };
    }

    fn parse_ua_stylesheet(
        shared_lock: &SharedRwLock,
        filename: &str,
        content: &[u8],
    ) -> DocumentStyleSheet {
        let url = Url::parse(&format!("chrome://resources/{filename}")).unwrap_or_else(|_| {
            panic!("Could not parse user stylesheet URL: {filename}");
        });
        DocumentStyleSheet(ServoArc::new(Stylesheet::from_bytes(
            content,
            url.into(),
            None,
            None,
            Origin::UserAgent,
            ServoArc::new(shared_lock.wrap(MediaList::empty())),
            shared_lock.clone(),
            None,
            None,
            QuirksMode::NoQuirks,
        )))
    }

    USER_AGENT_STYLESHEETS.with(|user_stylesheets| {
        user_stylesheets
            .get_or_init(|| {
                // FIXME: presentational-hints.css should be at author origin with zero specificity.
                //        (Does it make a difference?)
                let user_agent_stylesheets = vec![
                    parse_ua_stylesheet(shared_lock, "user-agent.css", USER_AGENT_CSS),
                    parse_ua_stylesheet(shared_lock, "servo.css", SERVO_CSS),
                    parse_ua_stylesheet(
                        shared_lock,
                        "presentational-hints.css",
                        PRESENTATIONAL_HINTS_CSS,
                    ),
                ];

                let html_mode_stylesheet =
                    parse_ua_stylesheet(shared_lock, "html-mode.css", HTML_MODE_CSS);

                let quirks_mode_stylesheet =
                    parse_ua_stylesheet(shared_lock, "quirks-mode.css", QUIRKS_MODE_CSS);

                Rc::new(UserAgentStylesheets {
                    user_agent_stylesheets,
                    html_mode_stylesheet,
                    quirks_mode_stylesheet,
                })
            })
            .clone()
    })
}

/// This structure holds the user-agent stylesheets.
pub struct UserAgentStylesheets {
    /// The user agent stylesheets.
    pub user_agent_stylesheets: Vec<DocumentStyleSheet>,
    /// The user agent stylesheet for HTML documents.
    pub html_mode_stylesheet: DocumentStyleSheet,
    /// The quirks mode stylesheet.
    pub quirks_mode_stylesheet: DocumentStyleSheet,
}

struct RegisteredPainterImpl {
    painter: Box<dyn Painter>,
    name: Atom,
    // FIXME: Should be a PrecomputedHashMap.
    properties: FxHashMap<Atom, PropertyId>,
}

impl SpeculativePainter for RegisteredPainterImpl {
    fn speculatively_draw_a_paint_image(
        &self,
        properties: Vec<(Atom, String)>,
        arguments: Vec<String>,
    ) {
        self.painter
            .speculatively_draw_a_paint_image(properties, arguments);
    }
}

impl RegisteredSpeculativePainter for RegisteredPainterImpl {
    fn properties(&self) -> &FxHashMap<Atom, PropertyId> {
        &self.properties
    }
    fn name(&self) -> Atom {
        self.name.clone()
    }
}

impl Painter for RegisteredPainterImpl {
    fn draw_a_paint_image(
        &self,
        size: Size2D<f32, CSSPixel>,
        device_pixel_ratio: Scale<f32, CSSPixel, DevicePixel>,
        properties: Vec<(Atom, String)>,
        arguments: Vec<String>,
    ) -> Result<DrawAPaintImageResult, PaintWorkletError> {
        self.painter
            .draw_a_paint_image(size, device_pixel_ratio, properties, arguments)
    }
}

struct RegisteredPaintersImpl(HashMap<Atom, RegisteredPainterImpl>);

impl RegisteredSpeculativePainters for RegisteredPaintersImpl {
    fn get(&self, name: &Atom) -> Option<&dyn RegisteredSpeculativePainter> {
        self.0
            .get(name)
            .map(|painter| painter as &dyn RegisteredSpeculativePainter)
    }
}

struct LayoutFontMetricsProvider(Arc<FontContext>);

impl FontMetricsProvider for LayoutFontMetricsProvider {
    fn query_font_metrics(
        &self,
        _vertical: bool,
        font: &Font,
        base_size: CSSPixelLength,
        _flags: QueryFontMetricsFlags,
    ) -> FontMetrics {
        let font_context = &self.0;
        let font_group = self
            .0
            .font_group_with_size(ServoArc::new(font.clone()), base_size.into());

        let Some(first_font_metrics) = font_group
            .first(font_context)
            .map(|font| font.metrics.clone())
        else {
            return Default::default();
        };

        // Only use the x-height of this font if it is non-zero. Some fonts return
        // inaccurate metrics, which shouldn't be used.
        let x_height = Some(first_font_metrics.x_height)
            .filter(|x_height| !x_height.is_zero())
            .map(CSSPixelLength::from);

        let zero_advance_measure = first_font_metrics
            .zero_horizontal_advance
            .or_else(|| {
                font_group
                    .find_by_codepoint(font_context, '0', None, Language::UND)?
                    .metrics
                    .zero_horizontal_advance
            })
            .map(CSSPixelLength::from);

        let ic_width = first_font_metrics
            .ic_horizontal_advance
            .or_else(|| {
                font_group
                    .find_by_codepoint(font_context, '\u{6C34}', None, Language::UND)?
                    .metrics
                    .ic_horizontal_advance
            })
            .map(CSSPixelLength::from);

        FontMetrics {
            x_height,
            zero_advance_measure,
            cap_height: None,
            ic_width,
            ascent: first_font_metrics.ascent.into(),
            script_percent_scale_down: None,
            script_script_percent_scale_down: None,
        }
    }

    fn base_size_for_generic(&self, generic: GenericFontFamily) -> Length {
        Length::new(match generic {
            GenericFontFamily::Monospace => pref!(fonts_default_monospace_size),
            _ => pref!(fonts_default_size),
        } as f32)
        .max(Length::new(0.0))
    }
}

impl Debug for LayoutFontMetricsProvider {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_tuple("LayoutFontMetricsProvider").finish()
    }
}

struct SnapshotSetter<'dom> {
    elements_with_snapshot: Vec<ServoLayoutElement<'dom>>,
}

impl SnapshotSetter<'_> {
    fn new(restyle: &mut ReflowRequestRestyle, snapshot_map: &mut SnapshotMap) -> Self {
        debug!("Draining restyles: {}", restyle.pending_restyles.len());
        let restyles = std::mem::take(&mut restyle.pending_restyles);

        let elements_with_snapshot: Vec<_> = restyles
            .iter()
            .filter(|r| r.1.snapshot.is_some())
            .map(|r| unsafe { ServoLayoutNode::new(&r.0).as_element().unwrap() })
            .collect();

        for (element, restyle) in restyles {
            let element = unsafe { ServoLayoutNode::new(&element).as_element().unwrap() };

            // If we haven't styled this node yet, we don't need to track a
            // restyle.
            let Some(mut style_data) = element
                .style_data()
                .map(|data| data.element_data.borrow_mut())
            else {
                element.unset_snapshot_flags();
                continue;
            };

            debug!("Noting restyle for {:?}: {:?}", element, style_data);
            if let Some(s) = restyle.snapshot {
                element.set_has_snapshot();
                snapshot_map.insert(element.as_node().opaque(), s);
            }

            // Stash the data on the element for processing by the style system.
            style_data.hint.insert(restyle.hint);
            style_data.damage = restyle.damage;
        }
        Self {
            elements_with_snapshot,
        }
    }
}

impl Drop for SnapshotSetter<'_> {
    fn drop(&mut self) {
        for element in &self.elements_with_snapshot {
            element.unset_snapshot_flags();
        }
    }
}

bitflags! {
    #[derive(Clone, Copy, Debug, Eq, PartialEq)]
    pub struct ReflowPhases: u8 {
        const StackingContextTreeConstruction = 1 << 0;
        const DisplayListConstruction = 1 << 1;
    }
}

impl ReflowPhases {
    /// Return the necessary phases of layout for the given [`ReflowGoal`]. Note that all
    /// [`ReflowGoals`] need the basic restyle + box tree layout + fragment tree layout,
    /// so [`ReflowPhases::empty()`] implies that.
    fn necessary(reflow_goal: &ReflowGoal) -> Self {
        match reflow_goal {
            ReflowGoal::LayoutQuery(query) => match query {
                QueryMsg::NodesFromPointQuery => {
                    Self::StackingContextTreeConstruction | Self::DisplayListConstruction
                },
                QueryMsg::BoxArea |
                QueryMsg::BoxAreas |
                QueryMsg::ElementsFromPoint |
                QueryMsg::FlushForUpdateTheRenderingQuery |
                QueryMsg::OffsetParentQuery |
                QueryMsg::ResolvedStyleQuery |
                QueryMsg::ScrollingAreaOrOffsetQuery |
                QueryMsg::TextIndexQuery => Self::StackingContextTreeConstruction,
                QueryMsg::ClientRectQuery |
                QueryMsg::CurrentCSSZoomQuery |
                QueryMsg::EffectiveOverflow |
                QueryMsg::ElementInnerOuterTextQuery |
                QueryMsg::InnerWindowDimensionsQuery |
                QueryMsg::PaddingQuery |
                QueryMsg::ResolvedFontStyleQuery |
                QueryMsg::ScrollParentQuery |
                QueryMsg::StyleQuery => Self::empty(),
            },
            ReflowGoal::UpdateScrollNode(..) | ReflowGoal::UpdateTheRendering => {
                Self::StackingContextTreeConstruction | Self::DisplayListConstruction
            },
        }
    }
}