i_slint_compiler/passes/
collect_globals.rs1#![allow(clippy::mutable_key_type)] use 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
17pub 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}