Skip to main content

i_slint_compiler/passes/
collect_globals.rs

1// Copyright © SixtyFPS GmbH <info@slint.dev>
2// SPDX-License-Identifier: GPL-3.0-only OR LicenseRef-Slint-Royalty-free-2.0 OR LicenseRef-Slint-Software-3.0
3
4//! This pass fills the root component used_types.globals
5
6#![allow(clippy::mutable_key_type)] // NamedReference/Component keys intentionally use interior mutability
7
8use by_address::ByAddress;
9use smol_str::format_smolstr;
10
11use crate::diagnostics::BuildDiagnostics;
12use crate::expression_tree::NamedReference;
13use crate::object_tree::*;
14use std::collections::HashSet;
15use std::rc::Rc;
16
17/// Fill the root_component's used_types.globals
18pub fn collect_globals(doc: &Document, _diag: &mut BuildDiagnostics) {
19    doc.used_types.borrow_mut().globals.clear();
20    let mut set = HashSet::new();
21    let mut sorted_globals = Vec::new();
22    for (_, ty) in &*doc.exports {
23        if let Some(c) = ty.as_ref().left()
24            && c.is_global()
25            && set.insert(ByAddress(c.clone()))
26        {
27            collect_in_component(c, &mut set, &mut sorted_globals);
28            sorted_globals.push(c.clone());
29        }
30    }
31    doc.visit_all_used_components(|component| {
32        collect_in_component(component, &mut set, &mut sorted_globals)
33    });
34
35    doc.used_types.borrow_mut().globals = sorted_globals;
36}
37
38pub fn mark_library_globals(doc: &Document) {
39    let mut used_types = doc.used_types.borrow_mut();
40    used_types.globals.clone().iter().for_each(|component| {
41        if let Some(library_info) = doc.library_exports.get(component.id.as_str()) {
42            component.from_library.set(true);
43            used_types.library_types_imports.push((component.id.clone(), library_info.clone()));
44            used_types
45                .library_types_imports
46                .push((format_smolstr!("Inner{}", component.id.clone()), library_info.clone()));
47        }
48    });
49}
50
51fn collect_in_component(
52    component: &Rc<Component>,
53    global_set: &mut HashSet<ByAddress<Rc<Component>>>,
54    sorted_globals: &mut Vec<Rc<Component>>,
55) {
56    let mut maybe_collect_global = |nr: &mut NamedReference| {
57        let element = nr.element();
58        let global_component = element.borrow().enclosing_component.upgrade().unwrap();
59        if global_component.is_global() && global_set.insert(ByAddress(global_component.clone())) {
60            collect_in_component(&global_component, global_set, sorted_globals);
61            sorted_globals.push(global_component);
62        }
63    };
64    visit_all_named_references(component, &mut maybe_collect_global);
65}