1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
use leo_ast::IntegerType;
use crate::{
AsgConvertError,
Expression,
ExpressionNode,
FromAst,
InnerVariable,
Node,
PartialType,
Scope,
Span,
Statement,
Variable,
};
use std::{
cell::RefCell,
sync::{Arc, Weak},
};
pub struct IterationStatement {
pub parent: Option<Weak<Statement>>,
pub span: Option<Span>,
pub variable: Variable,
pub start: Arc<Expression>,
pub stop: Arc<Expression>,
pub body: Arc<Statement>,
}
impl Node for IterationStatement {
fn span(&self) -> Option<&Span> {
self.span.as_ref()
}
}
impl FromAst<leo_ast::IterationStatement> for Arc<Statement> {
fn from_ast(
scope: &Scope,
statement: &leo_ast::IterationStatement,
_expected_type: Option<PartialType>,
) -> Result<Arc<Statement>, AsgConvertError> {
let expected_index_type = Some(PartialType::Integer(None, Some(IntegerType::U32)));
let start = Arc::<Expression>::from_ast(scope, &statement.start, expected_index_type.clone())?;
let stop = Arc::<Expression>::from_ast(scope, &statement.stop, expected_index_type)?;
let variable = Arc::new(RefCell::new(InnerVariable {
id: uuid::Uuid::new_v4(),
name: statement.variable.clone(),
type_: start
.get_type()
.ok_or_else(|| AsgConvertError::unresolved_type(&statement.variable.name, &statement.span))?,
mutable: false,
declaration: crate::VariableDeclaration::IterationDefinition,
references: vec![],
assignments: vec![],
}));
scope
.borrow_mut()
.variables
.insert(statement.variable.name.clone(), variable.clone());
let statement = Arc::new(Statement::Iteration(IterationStatement {
parent: None,
span: Some(statement.span.clone()),
variable: variable.clone(),
stop,
start,
body: Arc::new(Statement::Block(crate::BlockStatement::from_ast(
scope,
&statement.block,
None,
)?)),
}));
variable.borrow_mut().assignments.push(Arc::downgrade(&statement));
Ok(statement)
}
}
impl Into<leo_ast::IterationStatement> for &IterationStatement {
fn into(self) -> leo_ast::IterationStatement {
leo_ast::IterationStatement {
variable: self.variable.borrow().name.clone(),
start: self.start.as_ref().into(),
stop: self.stop.as_ref().into(),
block: match self.body.as_ref() {
Statement::Block(block) => block.into(),
_ => unimplemented!(),
},
span: self.span.clone().unwrap_or_default(),
}
}
}