nac3/nac3core/src/toplevel/mod.rs

144 lines
4.5 KiB
Rust
Raw Normal View History

2021-09-08 19:45:36 +08:00
use std::{
borrow::BorrowMut,
collections::{HashMap, HashSet},
fmt::Debug,
iter::FromIterator,
sync::Arc,
};
2021-08-03 13:38:27 +08:00
2021-09-30 17:07:48 +08:00
use super::codegen::CodeGenContext;
2021-08-05 14:55:23 +08:00
use super::typecheck::type_inferencer::PrimitiveStore;
use super::typecheck::typedef::{FunSignature, FuncArg, SharedUnifier, Type, TypeEnum, Unifier, VarMap};
use crate::{
codegen::CodeGenerator,
symbol_resolver::{SymbolResolver, ValueEnum},
typecheck::{type_inferencer::CodeLocation, typedef::CallId},
};
use inkwell::values::BasicValueEnum;
use itertools::{izip, Itertools};
use nac3parser::ast::{self, Location, Stmt, StrRef};
use parking_lot::RwLock;
2021-08-03 13:38:27 +08:00
#[derive(PartialEq, Eq, PartialOrd, Ord, Clone, Copy, Hash, Debug)]
pub struct DefinitionId(pub usize);
2021-08-23 02:52:54 +08:00
pub mod builtins;
pub mod composer;
pub mod helper;
pub mod numpy;
pub mod type_annotation;
use composer::*;
use type_annotation::*;
#[cfg(test)]
mod test;
2021-08-23 02:52:54 +08:00
2021-09-30 17:07:48 +08:00
type GenCallCallback = Box<
dyn for<'ctx, 'a> Fn(
&mut CodeGenContext<'ctx, 'a>,
Option<(Type, ValueEnum<'ctx>)>,
2021-09-30 17:07:48 +08:00
(&FunSignature, DefinitionId),
Vec<(Option<StrRef>, ValueEnum<'ctx>)>,
&mut dyn CodeGenerator,
) -> Result<Option<BasicValueEnum<'ctx>>, String>
2021-09-30 17:07:48 +08:00
+ Send
+ Sync,
>;
pub struct GenCall {
fp: GenCallCallback,
}
impl GenCall {
2023-12-08 17:43:32 +08:00
#[must_use]
2021-09-30 17:07:48 +08:00
pub fn new(fp: GenCallCallback) -> GenCall {
GenCall { fp }
}
2023-12-06 11:49:02 +08:00
pub fn run<'ctx>(
2021-09-30 17:07:48 +08:00
&self,
2023-12-06 11:49:02 +08:00
ctx: &mut CodeGenContext<'ctx, '_>,
obj: Option<(Type, ValueEnum<'ctx>)>,
2021-09-30 17:07:48 +08:00
fun: (&FunSignature, DefinitionId),
args: Vec<(Option<StrRef>, ValueEnum<'ctx>)>,
generator: &mut dyn CodeGenerator,
) -> Result<Option<BasicValueEnum<'ctx>>, String> {
(self.fp)(ctx, obj, fun, args, generator)
2021-09-30 17:07:48 +08:00
}
}
impl Debug for GenCall {
fn fmt(&self, _: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
Ok(())
}
}
#[derive(Clone, Debug)]
pub struct FunInstance {
2021-09-22 16:04:25 +08:00
pub body: Arc<Vec<Stmt<Option<Type>>>>,
pub calls: Arc<HashMap<CodeLocation, CallId>>,
pub subst: VarMap,
pub unifier_id: usize,
}
#[derive(Debug, Clone)]
2021-08-03 13:38:27 +08:00
pub enum TopLevelDef {
Class {
/// Name for error messages and symbols.
2021-09-22 17:19:27 +08:00
name: StrRef,
/// Object ID used for [TypeEnum].
object_id: DefinitionId,
2021-08-27 01:39:15 +08:00
/// type variables bounded to the class.
type_vars: Vec<Type>,
/// Class fields.
///
/// Name and type is mutable.
fields: Vec<(StrRef, Type, bool)>,
/// Class methods, pointing to the corresponding function definition.
2021-09-22 17:19:27 +08:00
methods: Vec<(StrRef, Type, DefinitionId)>,
/// Ancestor classes, including itself.
2021-08-23 02:52:54 +08:00
ancestors: Vec<TypeAnnotation>,
/// Symbol resolver of the module defined the class; [None] if it is built-in type.
2021-10-16 18:08:13 +08:00
resolver: Option<Arc<dyn SymbolResolver + Send + Sync>>,
/// Constructor type.
2021-09-19 22:54:06 +08:00
constructor: Option<Type>,
/// Definition location.
loc: Option<Location>,
2021-08-03 13:38:27 +08:00
},
Function {
/// Prefix for symbol, should be unique globally.
2021-08-07 15:06:39 +08:00
name: String,
/// Simple name, the same as in method/function definition.
2021-09-22 17:19:27 +08:00
simple_name: StrRef,
/// Function signature.
2021-08-03 13:38:27 +08:00
signature: Type,
/// Instantiated type variable IDs.
var_id: Vec<u32>,
2021-08-03 13:38:27 +08:00
/// Function instance to symbol mapping
///
/// * Key: String representation of type variable values, sorted by variable ID in ascending
2021-08-03 13:38:27 +08:00
/// order, including type variables associated with the class.
/// * Value: Function symbol name.
2021-08-03 13:38:27 +08:00
instance_to_symbol: HashMap<String, String>,
/// Function instances to annotated AST mapping
///
/// * Key: String representation of type variable values, sorted by variable ID in ascending
2021-08-03 13:38:27 +08:00
/// order, including type variables associated with the class. Excluding rigid type
/// variables.
///
/// Rigid type variables that would be substituted when the function is instantiated.
instance_to_stmt: HashMap<String, FunInstance>,
/// Symbol resolver of the module defined the class.
2021-10-16 18:08:13 +08:00
resolver: Option<Arc<dyn SymbolResolver + Send + Sync>>,
/// Custom code generation callback.
codegen_callback: Option<Arc<GenCall>>,
/// Definition location.
loc: Option<Location>,
2021-08-03 13:38:27 +08:00
},
}
pub struct TopLevelContext {
pub definitions: Arc<RwLock<Vec<Arc<RwLock<TopLevelDef>>>>>,
2021-08-11 14:37:26 +08:00
pub unifiers: Arc<RwLock<Vec<(SharedUnifier, PrimitiveStore)>>>,
pub personality_symbol: Option<String>,
}