cairo_lang_lowering/scc/
mod.rs

1use cairo_lang_utils::graph_algos::graph_node::GraphNode;
2use cairo_lang_utils::graph_algos::strongly_connected_components::compute_scc;
3use salsa::Database;
4
5use crate::DependencyType;
6use crate::db::LoweringGroup;
7use crate::ids::FunctionWithBodyId;
8
9/// Query implementation of [crate::db::LoweringGroup::function_with_body_scc].
10#[salsa::tracked(returns(ref))]
11pub fn function_with_body_scc<'db>(
12    db: &'db dyn Database,
13    function_id: FunctionWithBodyId<'db>,
14    dependency_type: DependencyType,
15) -> Vec<FunctionWithBodyId<'db>> {
16    compute_scc(&FunctionWithBodyNode { function_with_body_id: function_id, dependency_type, db })
17}
18
19/// A node to use in the SCC computation.
20#[derive(Clone)]
21struct FunctionWithBodyNode<'db> {
22    function_with_body_id: FunctionWithBodyId<'db>,
23    dependency_type: DependencyType,
24    db: &'db dyn Database,
25}
26impl<'db> GraphNode for FunctionWithBodyNode<'db> {
27    type NodeId = FunctionWithBodyId<'db>;
28
29    fn get_neighbors(&self) -> Vec<Self> {
30        self.db
31            .function_with_body_direct_function_with_body_callees(
32                self.function_with_body_id,
33                self.dependency_type,
34            )
35            .into_iter()
36            .flatten()
37            .map(|function_with_body_id| FunctionWithBodyNode {
38                function_with_body_id: *function_with_body_id,
39                dependency_type: self.dependency_type,
40                db: self.db,
41            })
42            .collect()
43    }
44
45    fn get_id(&self) -> Self::NodeId {
46        self.function_with_body_id
47    }
48}
cairo_lang_lowering/scc/mod.rs

cairo_lang_lowering/scc/
mod.rs
