temp work

This commit is contained in:
greg 2019-12-17 02:27:50 -08:00
parent afcb10bb72
commit 9ec1d78f1b
1 changed files with 124 additions and 0 deletions

View File

@ -1,5 +1,6 @@
use std::rc::Rc;
use crate::ast::*;
use crate::util::deref_optional_box;
//TODO maybe these functions should take closures that return a KeepRecursing | StopHere type,
//or a tuple of (T, <that type>)
@ -39,3 +40,126 @@ pub trait ASTVisitor: Sized {
fn prefix_exp(&mut self, _op: &PrefixOp, _arg: &Expression) {}
fn pattern(&mut self, _pat: &Pattern) {}
}
pub trait ASTVisitorNew: Sized {
fn expression(&mut self, _expression: &Expression) -> VisitResult { VisitResult::default() }
fn expression_kind(&mut self, _kind: &ExpressionKind) -> VisitResult { VisitResult::default() }
fn type_annotation(&mut self, _id: ItemId, _type_anno: Option<&TypeIdentifier>) -> VisitResult { VisitResult::default() }
fn named_struct(&mut self, _name: &QualifiedName, _fields: &Vec<(Rc<String>, Expression)>) -> VisitResult { VisitResult::default() }
fn nat_literal(&mut self, _id: ItemId, _n: u64)-> VisitResult { VisitResult::default() }
fn float_literal(&mut self, _id: ItemId, _f: f64) -> VisitResult { VisitResult::default() }
fn string_literal(&mut self, _id: ItemId, _s: &Rc<String>) -> VisitResult { VisitResult::default() }
fn bool_literal(&mut self, _id: ItemId, _b: bool) -> VisitResult { VisitResult::default() }
fn qualified_name(&mut self, _id: ItemId, _name: &QualifiedName) {}
}
enum VisitResult {
KeepRecursing,
Stop
}
impl std::default::Default for VisitResult {
fn default() -> VisitResult { VisitResult::KeepRecursing }
}
pub fn traverse<V: ASTVisitorNew>(visitor: &mut V, expression: &Expression) {
expresion(visitor, expresion)
}
fn expression<V: ASTVisitorNew>(visitor: &mut V, expression: &Expression) {
use ExpressionKind::*;
let Expression { id, kind, type_anno } = expression;
let output = match kind {
NatLiteral(n) => v.nat_literal(id, *n),
FloatLiteral(f) => v.float_literal(id, *f),
StringLiteral(s) => v.string_literal(id, s),
BoolLiteral(b) => v.bool_literal(d, *b),
BinExp(op, lhs, rhs) => {
expression(v, lhs);
expression(v, rhs);
},
PrefixExp(op, arg) => {
expression(v, arg);
}
TupleLiteral(exprs) => {
for expr in exprs {
expression(v, expr);
}
},
Value(name) => v.qualified_name(id, name),
NamedStruct { name, fields } => {
for (_, expr) in fields.iter() {
v.expression(expr);
}
}
Call { f, arguments } => {
expression(v, f);
for arg in args.iter() {
match arg {
InvocationArgument::Positional(expr) => {
expression(v, expr);
},
InvocationArgument::Keyword { expr, .. } => {
expression(v, expr);
},
Ignored => (),
}
}
},
Index { indexee, indexers } => {
expression(v, indexee);
for i in indexers.iter() {
expression(v, i);
}
},
IfExpression { discriminator, body } => {
discriminator.as_ref().map(|d| expression(v, d));
match body {
IfExpressionBody::SimpleConditional { then_case, else_case } => {
walk_block(v, then_case);
else_case.as_ref().map(|block| walk_block(v, block));
},
IfExpressionBody::SimplePatternMatch { pattern, then_case, else_case } => {
v.pattern(pattern);
walk_pattern(v, pattern);
walk_block(v, then_case);
else_case.as_ref().map(|block| walk_block(v, block));
},
IfExpressionBody::CondList(arms) => {
for arm in arms {
v.condition_arm(arm);
condition_arm(v, arm);
}
}
}
},
WhileExpression { condition, body } => (),
ForExpression { enumerators, body } => (),
Lambda { params , type_anno, body } => {
for param in params {
formal_param(v, param);
}
v.type_annotation(type_anno);
},
ListLiteral(exprs) => {
for expr in exprs {
expression(v, expr);
}
},
};
output
}
fn formal_param<V: ASTVisitorNew>(v: &mut V, param: &FormalParam) {
param.default.as_ref().map(|p| {
expression(v, p);
});
v.type_annotation(param.anno.as_ref());
}