ProvableHQ/leo
1
1
Fork 0
mirror of https://github.com/ProvableHQ/leo.git synced 2024-11-23 15:15:47 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add expressions to the type map

Browse Source
This commit is contained in:
Pranav Gaddamadugu 2023-10-10 21:06:01 -04:00 committed by Pranav Gaddamadugu
parent 2fc155798e
commit 621a2f2a95
2 changed files with 25 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
compiler/passes/src/common/symbol_table/mod.rs
View File

@ -31,6 +31,7 @@ use serde::{Deserialize, Serialize};
use serde_json;
// TODO (@d0cd) Consider a safe interface for the symbol table.
// TODO (@d0cd) Cleanup API
#[derive(Clone, Debug, Default, Serialize, Deserialize)]
pub struct SymbolTable {
/// The parent scope if it exists.
@ -105,9 +106,8 @@ impl SymbolTable {
}
/// Inserts a type for a node ID into the symbol table.
pub fn insert_type(&mut self, node_id: NodeID, type_: Type) -> Result<()> {
pub fn insert_type(&mut self, node_id: NodeID, type_: Type) {
self.types.insert(node_id, type_);
Ok(())
}
/// Removes a variable from the symbol table.

23
compiler/passes/src/type_checking/check_expressions.rs
View File

@ -42,6 +42,29 @@ impl<'a> ExpressionVisitor<'a> for TypeChecker<'a> {
type AdditionalInput = Option<Type>;
type Output = Option<Type>;
fn visit_expression(&mut self, input: &'a Expression, additional: &Self::AdditionalInput) -> Self::Output {
let output = match input {
Expression::Access(access) => self.visit_access(access, additional),
Expression::Binary(binary) => self.visit_binary(binary, additional),
Expression::Call(call) => self.visit_call(call, additional),
Expression::Cast(cast) => self.visit_cast(cast, additional),
Expression::Struct(struct_) => self.visit_struct_init(struct_, additional),
Expression::Err(err) => self.visit_err(err, additional),
Expression::Identifier(identifier) => self.visit_identifier(identifier, additional),
Expression::Literal(literal) => self.visit_literal(literal, additional),
Expression::Ternary(ternary) => self.visit_ternary(ternary, additional),
Expression::Tuple(tuple) => self.visit_tuple(tuple, additional),
Expression::Unary(unary) => self.visit_unary(unary, additional),
Expression::Unit(unit) => self.visit_unit(unit, additional),
};
// If the output type is known, add the expression and its associated type to the symbol table.
if let Some(type_) = &output {
self.symbol_table.borrow_mut().insert_type(input.id(), type_.clone());
}
// Return the output type.
output
}
fn visit_access(&mut self, input: &'a AccessExpression, expected: &Self::AdditionalInput) -> Self::Output {
match input {
AccessExpression::Array(access) => {