hugr_model/v0/ast/
view.rs

1use std::sync::Arc;
2
3use super::{LinkName, Node, Operation, Param, Region, SeqPart, Symbol, SymbolName, Term, VarName};
4use crate::v0::table::{self, NodeId, TermId, VarId, View};
5
6impl<'a> View<'a, TermId> for Term {
7    fn view(module: &'a table::Module<'a>, id: TermId) -> Option<Self> {
8        let term = module.get_term(id)?;
9        Some(match term {
10            table::Term::Wildcard => Term::Wildcard,
11            table::Term::Var(var) => Term::Var(module.view(*var)?),
12            table::Term::Apply(symbol, terms) => {
13                let symbol = module.view(*symbol)?;
14                let terms = module.view(*terms)?;
15                Term::Apply(symbol, terms)
16            }
17            table::Term::Func(region_id) => Term::Func(Arc::new(module.view(*region_id)?)),
18            table::Term::List(list_parts) => Term::List(module.view(*list_parts)?),
19            table::Term::Literal(literal) => Term::Literal(literal.clone()),
20            table::Term::Tuple(tuple_parts) => Term::List(module.view(*tuple_parts)?),
21        })
22    }
23}
24
25impl<'a> View<'a, NodeId> for Node {
26    fn view(module: &'a table::Module<'a>, id: NodeId) -> Option<Self> {
27        let node = module.get_node(id)?;
28
29        let operation = match node.operation {
30            table::Operation::Invalid => Operation::Invalid,
31            table::Operation::Dfg => Operation::Dfg,
32            table::Operation::Cfg => Operation::Cfg,
33            table::Operation::Block => Operation::Block,
34            table::Operation::DefineFunc(symbol) => {
35                Operation::DefineFunc(Box::new(module.view(*symbol)?))
36            }
37            table::Operation::DeclareFunc(symbol) => {
38                Operation::DeclareFunc(Box::new(module.view(*symbol)?))
39            }
40            table::Operation::Custom(operation) => Operation::Custom(module.view(operation)?),
41            table::Operation::DefineAlias(symbol, value) => {
42                let symbol = Box::new(module.view(*symbol)?);
43                let value = module.view(value)?;
44                Operation::DefineAlias(symbol, value)
45            }
46            table::Operation::DeclareAlias(symbol) => {
47                Operation::DeclareAlias(Box::new(module.view(*symbol)?))
48            }
49            table::Operation::DeclareConstructor(symbol) => {
50                Operation::DeclareConstructor(Box::new(module.view(*symbol)?))
51            }
52            table::Operation::DeclareOperation(symbol) => {
53                Operation::DeclareOperation(Box::new(module.view(*symbol)?))
54            }
55            table::Operation::TailLoop => Operation::TailLoop,
56            table::Operation::Conditional => Operation::Conditional,
57            table::Operation::Import { name } => Operation::Import(SymbolName::new(name)),
58        };
59
60        let meta = module.view(node.meta)?;
61        let signature = module.view(node.signature)?;
62        let inputs = module.view(node.inputs)?;
63        let outputs = module.view(node.outputs)?;
64        let regions = module.view(node.regions)?;
65
66        Some(Node {
67            operation,
68            inputs,
69            outputs,
70            regions,
71            meta,
72            signature,
73        })
74    }
75}
76
77impl<'a> View<'a, table::LinkIndex> for LinkName {
78    fn view(_module: &'a table::Module<'a>, index: table::LinkIndex) -> Option<Self> {
79        Some(LinkName::new_index(index))
80    }
81}
82
83impl<'a> View<'a, table::SeqPart> for SeqPart {
84    fn view(module: &'a table::Module<'a>, part: table::SeqPart) -> Option<Self> {
85        Some(match part {
86            table::SeqPart::Item(term_id) => SeqPart::Item(module.view(term_id)?),
87            table::SeqPart::Splice(term_id) => SeqPart::Splice(module.view(term_id)?),
88        })
89    }
90}
91
92impl<'a> View<'a, table::Symbol<'a>> for Symbol {
93    fn view(module: &'a table::Module<'a>, id: table::Symbol<'a>) -> Option<Self> {
94        let visibility = id.visibility.clone();
95        let name = SymbolName::new(id.name);
96        let params = module.view(id.params)?;
97        let constraints = module.view(id.constraints)?;
98        let signature = module.view(id.signature)?;
99        Some(Symbol {
100            visibility,
101            name,
102            params,
103            constraints,
104            signature,
105        })
106    }
107}
108
109impl<'a> View<'a, table::Param<'a>> for Param {
110    fn view(module: &'a table::Module<'a>, param: table::Param<'a>) -> Option<Self> {
111        let name = VarName::new(param.name);
112        let r#type = module.view(param.r#type)?;
113        Some(Param { name, r#type })
114    }
115}
116
117impl<'a> View<'a, table::RegionId> for Region {
118    fn view(module: &'a table::Module<'a>, id: table::RegionId) -> Option<Self> {
119        let region = module.get_region(id)?;
120        let sources = module.view(region.sources)?;
121        let targets = module.view(region.targets)?;
122        let meta = module.view(region.meta)?;
123        let children = module.view(region.children)?;
124        let signature = module.view(region.signature)?;
125
126        Some(Region {
127            kind: region.kind,
128            sources,
129            targets,
130            children,
131            meta,
132            signature,
133        })
134    }
135}
136
137impl<'a> View<'a, VarId> for VarName {
138    fn view(module: &'a table::Module<'a>, id: VarId) -> Option<Self> {
139        let node = module.get_node(id.0)?;
140
141        let symbol = match node.operation {
142            table::Operation::DefineFunc(symbol) => symbol,
143            table::Operation::DeclareFunc(symbol) => symbol,
144            table::Operation::DefineAlias(symbol, _) => symbol,
145            table::Operation::DeclareAlias(symbol) => symbol,
146            table::Operation::DeclareConstructor(symbol) => symbol,
147            table::Operation::DeclareOperation(symbol) => symbol,
148            _ => return None,
149        };
150
151        let param = &symbol.params[id.1 as usize];
152        Some(Self(param.name.into()))
153    }
154}
155
156impl<'a> View<'a, NodeId> for SymbolName {
157    fn view(module: &'a table::Module<'a>, id: NodeId) -> Option<Self> {
158        let node = module.get_node(id)?;
159        let name = node.operation.symbol()?;
160        Some(Self(name.into()))
161    }
162}
hugr_model/v0/ast/view.rs

hugr_model/v0/ast/
view.rs
