aboutsummaryrefslogtreecommitdiffstats
path: root/src/main.rs
diff options
context:
space:
mode:
authorJan Tuomi <jans.tuomi@gmail.com>2021-11-21 20:03:08 +0200
committerJan Tuomi <jans.tuomi@gmail.com>2021-11-21 20:03:08 +0200
commit4b845f566fabcab20992aa75dee5b6bf64f39573 (patch)
tree4ac21887af8d18d35bef3d7210959756003cf082 /src/main.rs
parent54e14921df8c5402066979e645e4ca49748238a8 (diff)
Split project into separate files
Diffstat (limited to 'src/main.rs')
-rw-r--r--src/main.rs428
1 files changed, 4 insertions, 424 deletions
diff --git a/src/main.rs b/src/main.rs
index e8d3acd..98be9ea 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -3,429 +3,9 @@ extern crate pest_derive;
#[macro_use]
extern crate pest;
-mod parser {
- #[derive(Parser)]
- #[grammar = "grammar.pest"]
- pub struct Parser;
-}
-
-mod ast {
- use super::parser::Rule;
- use std::rc::Rc;
-
- pub fn from_parse_tree(parse_tree: &mut pest::iterators::Pairs<Rule>) -> Program {
- let root = parse_tree.next().unwrap();
-
- match root.as_rule() {
- Rule::program => {
- let statements_with_eoi: Vec<pest::iterators::Pair<Rule>> =
- root.into_inner().collect();
- let statements_split = statements_with_eoi.split_last().unwrap();
- let statements = statements_split.1.to_vec();
-
- statements
- .iter()
- .map(|s| Statement::from_pair(s.clone()))
- .collect()
- }
- _ => panic!("[ast] first Pair is not a program"),
- }
- }
-
- pub type Program = Vec<Statement>;
-
- #[derive(Debug)]
- pub enum Statement {
- Definition {
- symbol: Symbol,
- expression: Rc<Expression>,
- },
- Expression(Rc<Expression>),
- }
-
- impl Statement {
- fn from_pair(pair: pest::iterators::Pair<Rule>) -> Statement {
- let def_or_expr = pair.into_inner().next().unwrap();
-
- fn definition_from_pair(pair: pest::iterators::Pair<Rule>) -> Statement {
- let mut inner = pair.into_inner();
- let symbol = inner.next().unwrap().as_span().as_str();
- let expression = inner.next().unwrap();
- let expression_inners: Vec<ExpressionInner> = expression
- .into_inner()
- .map(ExpressionInner::from_pair)
- .collect();
-
- Statement::Definition {
- symbol: String::from(symbol),
- expression: expression_vec_to_tuple(&expression_inners),
- }
- }
-
- match def_or_expr.as_rule() {
- Rule::definition => definition_from_pair(def_or_expr),
- Rule::expression => {
- let expression_inners: Vec<ExpressionInner> = def_or_expr
- .into_inner()
- .map(ExpressionInner::from_pair)
- .collect();
- Statement::Expression(expression_vec_to_tuple(&expression_inners))
- }
- rule => panic!("[ast] can't make a statement from {:#?}", rule),
- }
- }
- }
-
- #[derive(Debug, Clone)]
- pub enum ExpressionInner {
- Symbol(Symbol),
- IntegerLiteral(IntegerLiteral),
- StringLiteral(StringLiteral),
- Expression(Rc<Expression>),
- }
-
- impl ExpressionInner {
- fn from_pair(pair: pest::iterators::Pair<Rule>) -> ExpressionInner {
- match pair.as_rule() {
- Rule::symbol => ExpressionInner::Symbol(string_from_pair(pair)),
- Rule::integer_literal => ExpressionInner::IntegerLiteral(integer_from_pair(pair)),
- Rule::string_literal => {
- let s_with_quotes = string_from_pair(pair);
- let mut chars = s_with_quotes.chars();
- chars.next();
- chars.next_back();
- let s = chars.as_str();
- ExpressionInner::StringLiteral(String::from(s))
- }
- Rule::expression => {
- let expression_inners: Vec<ExpressionInner> =
- pair.into_inner().map(ExpressionInner::from_pair).collect();
- ExpressionInner::Expression(expression_vec_to_tuple(&expression_inners))
- }
- rule => panic!("[ast] can't make an expression element from {:#?}", rule),
- }
- }
- }
-
- #[derive(Debug)]
- pub enum Expression {
- Unary(ExpressionInner),
- Binary(ExpressionInner, ExpressionInner),
- }
-
- fn expression_vec_to_tuple(v: &Vec<ExpressionInner>) -> Rc<Expression> {
- match v.len() {
- 1 => Rc::new(Expression::Unary(v[0].clone())),
- 2 => Rc::new(Expression::Binary(v[0].clone(), v[1].clone())),
- _ => {
- let combined = ExpressionInner::Expression(Rc::new(Expression::Binary(
- v[0].clone(),
- v[1].clone(),
- )));
-
- let mut new_v = vec![combined];
- for e in &v[2..] {
- new_v.push(e.clone());
- }
-
- expression_vec_to_tuple(&new_v)
- }
- }
- }
-
- pub type IntegerLiteral = i64;
- fn integer_from_pair(pair: pest::iterators::Pair<Rule>) -> i64 {
- let s = pair.as_span().as_str();
- s.parse().unwrap()
- }
-
- pub type StringLiteral = String;
- fn string_from_pair(pair: pest::iterators::Pair<Rule>) -> String {
- String::from(pair.as_span().as_str())
- }
-
- pub type Symbol = String;
-}
-
-mod runtime {
- use super::ast;
- use std::collections::HashMap;
- use std::fmt;
- use std::rc::Rc;
- use std::sync::atomic::{AtomicUsize, Ordering};
-
- #[derive(Debug, Clone)]
- pub enum BuiltinFunction {
- IntegerIncrement,
- IntegerDecrement,
- IntegerAdd,
- IntegerAdd1(i64),
- IntegerMultiply,
- IntegerMultiply1(i64),
- IntegerEq,
- IntegerEq1(i64),
- }
-
- #[derive(Debug, Clone)]
- pub enum Value {
- Integer(i64),
- String(String),
- Var(usize),
- Function(usize, Rc<Value>),
- BuiltinFunction(BuiltinFunction),
- }
-
- impl fmt::Display for Value {
- fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
- match self {
- Value::Integer(value) => write!(f, "{} :: Integer", value),
- Value::String(value) => write!(f, "{} :: String", value),
- Value::Var(v) => write!(f, "{} :: Unbound variable", v),
- Value::Function(_, _) => {
- write!(f, "Function :: Function")
- }
- Value::BuiltinFunction(_) => write!(f, "Built-in function :: Function"),
- }
- }
- }
-
- static VAR_ID_INC: AtomicUsize = AtomicUsize::new(0);
- const B_INTEGER_INCREMENT: &str = "int.increment";
- const B_INTEGER_DECREMENT: &str = "int.decrement";
- const B_INTEGER_ADD: &str = "int.add";
- const B_INTEGER_MULTIPLY: &str = "int.multiply";
- const B_INTEGER_EQ: &str = "int.eq?";
-
- fn advance_v() -> usize {
- let v = VAR_ID_INC.load(Ordering::Relaxed);
- VAR_ID_INC.store(v + 1, Ordering::Relaxed);
- v
- }
-
- fn make_boolean_true_function() -> Value {
- let x = advance_v();
- let y = advance_v();
- Value::Function(x, Rc::new(Value::Function(y, Rc::new(Value::Var(x)))))
- }
-
- fn make_boolean_false_function() -> Value {
- let x = advance_v();
- let y = advance_v();
- Value::Function(x, Rc::new(Value::Function(y, Rc::new(Value::Var(y)))))
- }
-
- fn try_builtin_symbol_to_value(symbol: &ast::Symbol) -> Option<Value> {
- match symbol.as_str() {
- "true" => Some(make_boolean_true_function()),
- "false" => Some(make_boolean_false_function()),
- "id" => {
- let v = advance_v();
- Some(Value::Function(v, Rc::new(Value::Var(v))))
- }
- B_INTEGER_INCREMENT => Some(Value::BuiltinFunction(BuiltinFunction::IntegerIncrement)),
- B_INTEGER_DECREMENT => Some(Value::BuiltinFunction(BuiltinFunction::IntegerDecrement)),
- B_INTEGER_ADD => Some(Value::BuiltinFunction(BuiltinFunction::IntegerAdd)),
- B_INTEGER_EQ => Some(Value::BuiltinFunction(BuiltinFunction::IntegerEq)),
- _ => None,
- }
- }
-
- fn try_apply_function(
- func_rc: Rc<Value>,
- arg_rc: Rc<Value>,
- bound_v: Option<usize>,
- ) -> Rc<Value> {
- let func = &*func_rc;
- let arg = &*arg_rc;
- match func {
- Value::Function(func_v, body_rc) => {
- let v1 = bound_v.unwrap_or(*func_v);
- let body = &**body_rc;
- match body {
- Value::Var(v2) => {
- if v1 == *v2 {
- arg_rc
- } else {
- Rc::clone(body_rc)
- }
- }
- Value::Function(body_v, _) => {
- let new_body = try_apply_function(Rc::clone(body_rc), arg_rc, Some(v1));
- Rc::new(Value::Function(*body_v, new_body))
- }
- _ => Rc::clone(body_rc),
- }
- }
- Value::BuiltinFunction(builtin) => match builtin {
- BuiltinFunction::IntegerIncrement => match arg {
- Value::Integer(value) => Rc::new(Value::Integer(value + 1)),
- _ => panic!(
- "[runtime] tried to apply non-integer value to {}",
- B_INTEGER_INCREMENT
- ),
- },
- BuiltinFunction::IntegerDecrement => match arg {
- Value::Integer(value) => Rc::new(Value::Integer(value - 1)),
- _ => panic!(
- "[runtime] tried to apply non-integer value to {}",
- B_INTEGER_DECREMENT
- ),
- },
- BuiltinFunction::IntegerAdd => match arg {
- Value::Integer(value) => {
- Rc::new(Value::BuiltinFunction(BuiltinFunction::IntegerAdd1(*value)))
- }
- _ => panic!(
- "[runtime] tried to apply non-integer value to {}",
- B_INTEGER_ADD
- ),
- },
- BuiltinFunction::IntegerAdd1(other) => match arg {
- Value::Integer(value) => Rc::new(Value::Integer(other + value)),
- _ => panic!(
- "[runtime] tried to apply non-integer value to {}",
- B_INTEGER_ADD
- ),
- },
- BuiltinFunction::IntegerMultiply => match arg {
- Value::Integer(value) => Rc::new(Value::BuiltinFunction(
- BuiltinFunction::IntegerMultiply1(*value),
- )),
- _ => panic!(
- "[runtime] tried to apply non-integer value to {}",
- B_INTEGER_MULTIPLY
- ),
- },
- BuiltinFunction::IntegerMultiply1(other) => match arg {
- Value::Integer(value) => Rc::new(Value::Integer(other * value)),
- _ => panic!(
- "[runtime] tried to apply non-integer value to {}",
- B_INTEGER_MULTIPLY
- ),
- },
- BuiltinFunction::IntegerEq => match arg {
- Value::Integer(value) => {
- Rc::new(Value::BuiltinFunction(BuiltinFunction::IntegerEq1(*value)))
- }
- _ => panic!(
- "[runtime] tried to apply non-integer value to {}",
- B_INTEGER_EQ
- ),
- },
- BuiltinFunction::IntegerEq1(other) => match arg {
- Value::Integer(value) => Rc::new(if other == value {
- make_boolean_true_function()
- } else {
- make_boolean_false_function()
- }),
- _ => panic!(
- "[runtime] tried to apply non-integer value to {}",
- B_INTEGER_EQ
- ),
- },
- },
- _ => func_rc,
- }
- }
-
- fn evaluate_expr_inner_unary(
- symbol_table: &HashMap<String, Rc<Value>>,
- inner: &ast::ExpressionInner,
- ) -> Rc<Value> {
- match inner {
- ast::ExpressionInner::IntegerLiteral(value) => Rc::new(Value::Integer(*value)),
- ast::ExpressionInner::StringLiteral(value) => Rc::new(Value::String(value.clone())),
- ast::ExpressionInner::Symbol(value) => {
- let builtin_value = try_builtin_symbol_to_value(value);
- if builtin_value.is_some() {
- return Rc::new(builtin_value.unwrap());
- }
-
- let table_lookup_value = symbol_table.get(value);
- if table_lookup_value.is_some() {
- let lookup_rc = table_lookup_value.unwrap();
- return Rc::clone(lookup_rc);
- }
-
- panic!("[runtime] symbol not defined: {:#?}", value);
- }
- ast::ExpressionInner::Expression(value_rc) => {
- let sub_expr = &**value_rc;
- evaluate_expr(symbol_table, sub_expr)
- }
- }
- }
-
- fn evaluate_expr_inner_binary(
- symbol_table: &HashMap<String, Rc<Value>>,
- lhs: &ast::ExpressionInner,
- rhs: &ast::ExpressionInner,
- ) -> Rc<Value> {
- match lhs {
- ast::ExpressionInner::Expression(value_rc) => {
- let sub_expr = &**value_rc;
-
- let evaled_lhs = evaluate_expr(symbol_table, sub_expr);
- let evaled_rhs = evaluate_expr_inner_unary(symbol_table, rhs);
- try_apply_function(evaled_lhs, evaled_rhs, None)
- }
- ast::ExpressionInner::Symbol(value) => {
- let lhs_value_rc: Rc<Value>;
-
- if let Some(builtin) = try_builtin_symbol_to_value(value) {
- lhs_value_rc = Rc::new(builtin);
- } else if let Some(lookup) = symbol_table.get(value) {
- lhs_value_rc = Rc::clone(lookup);
- } else {
- panic!("[runtime] symbol not defined: {:#?}", value);
- }
-
- let evaled_rhs = evaluate_expr_inner_unary(symbol_table, rhs);
- try_apply_function(lhs_value_rc, evaled_rhs, None)
- }
- other => unreachable!(
- "[runtime] this should not be on the left side of a binary expression: {:#?}",
- other
- ),
- }
- }
-
- fn evaluate_expr(
- symbol_table: &HashMap<String, Rc<Value>>,
- expression: &ast::Expression,
- ) -> Rc<Value> {
- match expression {
- ast::Expression::Unary(inner) => evaluate_expr_inner_unary(symbol_table, inner),
- ast::Expression::Binary(inner1, inner2) => {
- evaluate_expr_inner_binary(symbol_table, inner1, inner2)
- }
- }
- }
-
- pub fn evaluate(program: &ast::Program) {
- let mut symbol_table: HashMap<String, Rc<Value>> = HashMap::new();
-
- for statement in program {
- match statement {
- ast::Statement::Definition { symbol, expression } => {
- println!("[runtime] defining symbol: {:#?}", symbol);
- let value = evaluate_expr(&symbol_table, expression);
- symbol_table.insert(symbol.clone(), value);
- }
- ast::Statement::Expression(expression) => {
- println!("[runtime] evaluating free-standing expression");
- let value = evaluate_expr(&symbol_table, expression);
- println!("Result: {}", value);
- }
- }
- }
-
- println!(
- "[runtime] evaluation done, symbol_table state dump: {:#?}",
- symbol_table
- );
- }
-}
+mod ast;
+mod lparser;
+mod runtime;
fn main() {
use ast::Program;
@@ -436,7 +16,7 @@ fn main() {
let script_path = env::args().nth(1).expect("no script file specified");
let unparsed_file = fs::read_to_string(script_path).expect("cannot read file");
- let parse_tree_result = parser::Parser::parse(parser::Rule::program, &unparsed_file);
+ let parse_tree_result = lparser::LParser::parse(lparser::Rule::program, &unparsed_file);
if parse_tree_result.is_err() {
println!("{}", parse_tree_result.unwrap_err());