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
6use by_address::ByAddress;
7use smol_str::format_smolstr;
8
9use crate::diagnostics::BuildDiagnostics;
10use crate::expression_tree::NamedReference;
11use crate::object_tree::*;
12use std::collections::HashSet;
13use std::rc::Rc;
14
15/// Fill the root_component's used_types.globals
16pub fn collect_globals(doc: &Document, _diag: &mut BuildDiagnostics) {
17    doc.used_types.borrow_mut().globals.clear();
18    let mut set = HashSet::new();
19    let mut sorted_globals = vec![];
20    for (_, ty) in &*doc.exports {
21        if let Some(c) = ty.as_ref().left() {
22            if c.is_global() && set.insert(ByAddress(c.clone())) {
23                collect_in_component(c, &mut set, &mut sorted_globals);
24                sorted_globals.push(c.clone());
25            }
26        }
27    }
28    doc.visit_all_used_components(|component| {
29        collect_in_component(component, &mut set, &mut sorted_globals)
30    });
31
32    doc.used_types.borrow_mut().globals = sorted_globals;
33}
34
35pub fn mark_library_globals(doc: &Document) {
36    let mut used_types = doc.used_types.borrow_mut();
37    used_types.globals.clone().iter().for_each(|component| {
38        if let Some(library_info) = doc.library_exports.get(component.id.as_str()) {
39            component.from_library.set(true);
40            used_types.library_types_imports.push((component.id.clone(), library_info.clone()));
41            used_types
42                .library_types_imports
43                .push((format_smolstr!("Inner{}", component.id.clone()), library_info.clone()));
44        }
45    });
46}
47
48fn collect_in_component(
49    component: &Rc<Component>,
50    global_set: &mut HashSet<ByAddress<Rc<Component>>>,
51    sorted_globals: &mut Vec<Rc<Component>>,
52) {
53    let mut maybe_collect_global = |nr: &mut NamedReference| {
54        let element = nr.element();
55        let global_component = element.borrow().enclosing_component.upgrade().unwrap();
56        if global_component.is_global() && global_set.insert(ByAddress(global_component.clone())) {
57            collect_in_component(&global_component, global_set, sorted_globals);
58            sorted_globals.push(global_component);
59        }
60    };
61    visit_all_named_references(component, &mut maybe_collect_global);
62}