use std::fmt;
use std::mem;
use std::option::Option;
use erg_common::dict::Dict;
use erg_common::error::{ErrorCore, Location};
use erg_common::levenshtein::levenshtein;
use erg_common::set::Set;
use erg_common::traits::{HasType, Locational, Stream};
use erg_common::ty::fresh_varname;
use erg_common::ty::{
ConstObj, Constraint, FreeKind, HasLevel, IntervalOp, ParamTy, Predicate, RefinementType,
SubrKind, SubrType, TyBound, TyParam, TyParamOrdering, Type,
};
use erg_common::value::ValueObj;
use erg_common::Str;
use erg_common::{
assume_unreachable, enum_unwrap, fmt_slice, fn_name, get_hash, log, set, try_map,
};
use Predicate as Pred;
use TyParamOrdering::*;
use Type::*;
use ValueObj::{Inf, NegInf};
use ast::{
DefId, ParamSignature, ParamTySpec, PreDeclTypeSpec, SimpleTypeSpec, TypeBoundSpec,
TypeBoundSpecs, TypeSpec, VarName,
};
use erg_parser::ast;
use erg_parser::token::{Token, TokenKind};
use crate::error::readable_name;
use crate::error::{binop_to_dname, unaryop_to_dname, TyCheckError, TyCheckErrors, TyCheckResult};
use crate::eval::Evaluator;
use crate::hir;
use crate::varinfo::{Mutability, ParamIdx, VarInfo, VarKind, Visibility};
use Mutability::*;
use Visibility::*;
type Trait = Type;
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum TyParamIdx {
Nth(usize),
Nested(Box<TyParamIdx>, usize),
}
impl TyParamIdx {
pub fn search(search_from: &Type, target: &Type) -> Option<Self> {
match search_from {
Type::Poly{ params, .. } => {
for (i, tp) in params.iter().enumerate() {
match tp {
TyParam::Type(t) if t.rec_eq(target) => { return Some(Self::Nth(i)) },
TyParam::Type(t) if t.is_monomorphic() => {},
other => todo!("{other:?}"),
}
}
None
},
_ => todo!(),
}
}
pub fn select(self, from: &Type) -> Type {
match self {
Self::Nth(n) => {
let tps = from.typarams();
let tp = tps.iter().nth(n).unwrap();
match tp {
TyParam::Type(t) => *t.clone(),
_ => todo!(),
}
},
Self::Nested(_, _) => todo!(),
}
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum DefaultInfo {
NonDefault,
WithDefault,
}
impl DefaultInfo {
pub const fn has_default(&self) -> bool {
matches!(self, DefaultInfo::WithDefault)
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, Default)]
pub enum Variance {
Covariant, Contravariant, #[default]
Invariant, }
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub struct ParamSpec {
pub(crate) name: Option<&'static str>, pub(crate) t: Type,
pub default_info: DefaultInfo,
}
impl ParamSpec {
pub const fn new(name: Option<&'static str>, t: Type, default: DefaultInfo) -> Self {
Self {
name,
t,
default_info: default,
}
}
pub const fn named(name: &'static str, t: Type, default: DefaultInfo) -> Self {
Self::new(Some(name), t, default)
}
pub const fn named_nd(name: &'static str, t: Type) -> Self {
Self::new(Some(name), t, DefaultInfo::NonDefault)
}
pub const fn t(name: &'static str, default: DefaultInfo) -> Self {
Self::new(Some(name), Type, default)
}
pub const fn t_nd(name: &'static str) -> Self {
Self::new(Some(name), Type, DefaultInfo::NonDefault)
}
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum ContextKind {
Func,
Proc,
Tuple,
Record,
Class,
Trait,
StructuralTrait,
Patch,
StructuralPatch,
Module,
Instant,
Dummy,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum RegistrationMode {
PreRegister,
Normal,
}
use RegistrationMode::*;
#[derive(Debug, Clone)]
pub struct TyVarContext {
level: usize,
pub(crate) tyvar_instances: Dict<Str, Type>,
pub(crate) typaram_instances: Dict<Str, TyParam>,
}
impl TyVarContext {
pub fn new(level: usize, bounds: Set<TyBound>, ctx: &Context) -> Self {
let mut self_ = Self {
level,
tyvar_instances: Dict::new(),
typaram_instances: Dict::new(),
};
for bound in bounds.into_iter() {
self_.instantiate_bound(bound, ctx);
}
self_
}
fn instantiate_const_template(&mut self, var_name: &str, _callee_name: &Str, ct: &ConstTemplate) -> TyParam {
match ct {
ConstTemplate::Obj(o) => {
match o {
ConstObj::Type(t) if t.is_mono_q() => {
if t.name() == "Self" {
let constraint = Constraint::TypeOf(Type);
let t = Type::named_free_var(Str::rc(var_name), self.level, constraint);
TyParam::t(t)
} else {
todo!()
}
},
ConstObj::Type(t) => TyParam::t(*t.clone()),
v @ ConstObj::Value(_) => TyParam::ConstObj(v.clone()),
other => todo!("{other}"),
}
},
ConstTemplate::App { .. } => {
todo!()
}
}
}
fn instantiate_poly(&mut self, tvar_name: &str, name: &Str, params: Vec<TyParam>, ctx: &Context) -> Type {
if let Some(temp_defaults) = ctx.rec_get_const_param_defaults(&name) {
let c = ctx.rec_type_ctx_by_name(name).unwrap_or_else(|| panic!("{} not found", name));
let defined_params_len = c.params.len();
let given_params_len = params.len();
if defined_params_len < given_params_len { panic!() }
let inst_non_defaults = params.into_iter().map(|p| self.instantiate_tp(p)).collect();
let mut inst_defaults = vec![];
for c in temp_defaults.into_iter().take(defined_params_len - given_params_len) {
let c = self.instantiate_const_template(tvar_name, name, c);
inst_defaults.push(c);
}
Type::poly(
name,
[inst_non_defaults, inst_defaults].concat(),
)
} else {
Type::poly(
name,
params.into_iter().map(|p| self.instantiate_tp(p)).collect(),
)
}
}
fn instantiate_bound(&mut self, bound: TyBound, ctx: &Context) {
match bound {
TyBound::Subtype { sub, sup } => {
let sup = match sup {
Type::Poly { name, params } => {
self.instantiate_poly(sub.name(), &name, params, ctx)
}
Type::MonoProj { lhs, rhs } => Type::mono_proj(self.instantiate_t(*lhs), rhs),
sup => sup,
};
let constraint = Constraint::SubtypeOf(sup);
if let Some(tv) = self.tyvar_instances.get(sub.name()) {
tv.update_constraint(constraint);
} else if let Some(tp) = self.typaram_instances.get(sub.name()) {
tp.update_constraint(constraint);
} else {
let name = Str::rc(sub.name());
self.push_tyvar(name.clone(), Type::named_free_var(name, self.level, constraint));
}
}
TyBound::Supertype { sup, sub } => {
let sub = match sub {
Type::Poly { name, params } => {
self.instantiate_poly(sup.name(), &name, params, ctx)
}
Type::MonoProj { lhs, rhs } => Type::mono_proj(self.instantiate_t(*lhs), rhs),
sub => sub,
};
let constraint = Constraint::SupertypeOf(sub);
if let Some(tv) = self.tyvar_instances.get(sup.name()) {
tv.update_constraint(constraint);
} else if let Some(tp) = self.typaram_instances.get(sup.name()) {
tp.update_constraint(constraint);
} else {
let name = Str::rc(sup.name());
self.push_tyvar(name.clone(), Type::named_free_var(name, self.level, constraint));
}
}
TyBound::Sandwiched { sub, mid, sup } => {
let sub = match sub {
Type::Poly { name, params } => {
self.instantiate_poly(mid.name(), &name, params, ctx)
}
Type::MonoProj { lhs, rhs } => Type::mono_proj(self.instantiate_t(*lhs), rhs),
sub => sub,
};
let sup = match sup {
Type::Poly { name, params } => {
self.instantiate_poly(mid.name(), &name, params, ctx)
}
Type::MonoProj { lhs, rhs } => Type::mono_proj(self.instantiate_t(*lhs), rhs),
sup => sup,
};
let constraint = Constraint::Sandwiched { sub, sup };
if let Some(tv) = self.tyvar_instances.get(mid.name()) {
tv.update_constraint(constraint);
} else if let Some(tp) = self.typaram_instances.get(mid.name()) {
tp.update_constraint(constraint);
} else {
let name = Str::rc(mid.name());
self.push_tyvar(name.clone(), Type::named_free_var(name, self.level, constraint));
}
}
TyBound::Instance { name, t } => {
let t = match t {
Type::Poly { name, params } => {
self.instantiate_poly(&name[..], &name, params, ctx)
}
t => t,
};
let constraint = Constraint::TypeOf(t.clone());
if &t == &Type {
if let Some(tv) = self.tyvar_instances.get(&name) {
tv.update_constraint(constraint);
} else if let Some(tp) = self.typaram_instances.get(&name) {
tp.update_constraint(constraint);
} else {
self.push_tyvar(
name.clone(),
Type::named_free_var(name, self.level, constraint),
);
}
} else {
if let Some(tp) = self.typaram_instances.get(&name) {
tp.update_constraint(constraint);
} else {
self.push_typaram(name.clone(), TyParam::named_free_var(name, self.level, t));
}
}
}
}
}
fn _instantiate_pred(&self, _pred: Predicate) -> Predicate {
todo!()
}
pub(crate) fn instantiate_t(&mut self, quantified: Type) -> Type {
match quantified {
Type::MonoQVar(n) => {
if let Some(t) = self.get_tyvar(&n) {
return t.clone();
} else if let Some(t) = self.get_typaram(&n) {
if let TyParam::Type(t) = t {
return *t.clone();
} else {
todo!()
}
} else {
let tv = Type::named_free_var(n.clone(), self.level, Constraint::Uninited);
self.push_tyvar(n, tv.clone());
tv
}
}
other => todo!("{other}"),
}
}
fn instantiate_tp(&mut self, quantified: TyParam) -> TyParam {
match quantified {
TyParam::MonoQVar(n) => {
if let Some(t) = self.get_typaram(&n) {
t.clone()
} else if let Some(t) = self.get_tyvar(&n) {
TyParam::t(t.clone())
} else {
let tp = TyParam::named_free_var(n.clone(), self.level, Type::Uninited);
self.push_typaram(n, tp.clone());
tp
}
}
TyParam::Type(t) => {
if let Type::MonoQVar(n) = *t {
if let Some(t) = self.get_typaram(&n) {
t.clone()
} else if let Some(t) = self.get_tyvar(&n) {
TyParam::t(t.clone())
} else {
let tv = Type::named_free_var(n.clone(), self.level, Constraint::Uninited);
self.push_tyvar(n, tv.clone());
TyParam::t(tv)
}
} else {
todo!("{t}")
}
}
TyParam::UnaryOp { op, val } => {
let res = self.instantiate_tp(*val);
TyParam::unary(op, res)
}
TyParam::BinOp { op, lhs, rhs } => {
let lhs = self.instantiate_tp(*lhs);
let rhs = self.instantiate_tp(*rhs);
TyParam::bin(op, lhs, rhs)
}
p @ TyParam::ConstObj(_) => p,
other => todo!("{other}"),
}
}
pub(crate) fn push_tyvar(&mut self, name: Str, t: Type) {
self.tyvar_instances.insert(name, t);
}
pub(crate) fn push_typaram(&mut self, name: Str, t: TyParam) {
self.typaram_instances.insert(name, t);
}
pub(crate) fn get_tyvar(&self, name: &str) -> Option<&Type> {
self.tyvar_instances.get(name)
}
pub(crate) fn get_typaram(&self, name: &str) -> Option<&TyParam> {
self.typaram_instances.get(name)
}
}
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum ConstTemplate {
Obj(ConstObj),
App{ name: Str, non_default_args: Vec<Type>, default_args: Vec<ConstTemplate> },
}
impl ConstTemplate {
pub const fn app(name: &'static str, non_default_args: Vec<Type>, default_args: Vec<ConstTemplate>) -> Self {
ConstTemplate::App {
name: Str::ever(name),
non_default_args,
default_args,
}
}
}
#[derive(Debug)]
pub struct Context {
pub(crate) name: Str,
pub(crate) kind: ContextKind,
pub(crate) bounds: Vec<TyBound>,
pub(crate) preds: Vec<Predicate>,
pub(crate) outer: Option<Box<Context>>,
pub(crate) const_param_defaults: Dict<Str, Vec<ConstTemplate>>,
pub(crate) super_classes: Vec<Type>, pub(crate) super_traits: Vec<Type>, pub(crate) method_impl_patches: Dict<VarName, Vec<VarName>>,
pub(crate) poly_trait_impls: Dict<Str, Vec<(Type, Trait)>>,
pub(crate) glue_patch_and_types: Vec<(VarName, Type, Trait)>,
pub(crate) decls: Dict<VarName, VarInfo>,
pub(crate) params: Vec<(Option<VarName>, VarInfo)>,
pub(crate) locals: Dict<VarName, VarInfo>,
pub(crate) consts: Dict<Str, ConstObj>,
pub(crate) eval: Evaluator,
pub(crate) types: Dict<Type, Context>,
pub(crate) patches: Dict<VarName, Context>,
pub(crate) mods: Dict<VarName, Context>,
pub(crate) _nlocals: usize, pub(crate) level: usize,
}
impl Default for Context {
#[inline]
fn default() -> Self {
Self::new(
"<dummy>".into(),
ContextKind::Dummy,
vec![],
None,
vec![],
vec![],
Self::TOP_LEVEL,
)
}
}
impl fmt::Display for Context {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
f.debug_struct("Context")
.field("name", &self.name)
.field("bounds", &self.bounds)
.field("preds", &self.preds)
.field("params", &self.params)
.field("decls", &self.decls)
.field("locals", &self.params)
.field("consts", &self.consts)
.field("eval", &self.eval)
.field("types", &self.types)
.field("patches", &self.patches)
.field("mods", &self.mods)
.finish()
}
}
impl Context {
#[inline]
pub fn new(
name: Str,
kind: ContextKind,
params: Vec<ParamSpec>,
outer: Option<Context>,
super_classes: Vec<Type>,
super_traits: Vec<Type>,
level: usize,
) -> Self {
Self::with_capacity(
name,
kind,
params,
outer,
super_classes,
super_traits,
0,
level,
)
}
pub fn with_capacity(
name: Str,
kind: ContextKind,
params: Vec<ParamSpec>,
outer: Option<Context>,
super_classes: Vec<Type>,
super_traits: Vec<Type>,
capacity: usize,
level: usize,
) -> Self {
let mut params_ = Vec::new();
for (idx, param) in params.into_iter().enumerate() {
let id = DefId(get_hash(&(&name, ¶m)));
if let Some(name) = param.name {
let idx = ParamIdx::Nth(idx);
let kind = VarKind::parameter(id, idx, param.default_info);
let vi = VarInfo::new(param.t, Immutable, Private, kind);
params_.push((Some(VarName::new(Token::static_symbol(name))), vi));
} else {
let idx = ParamIdx::Nth(idx);
let kind = VarKind::parameter(id, idx, param.default_info);
let vi = VarInfo::new(param.t, Immutable, Private, kind);
params_.push((None, vi));
}
}
Self {
name,
kind,
bounds: vec![],
preds: vec![],
outer: outer.map(Box::new),
super_classes,
super_traits,
const_param_defaults: Dict::default(),
method_impl_patches: Dict::default(),
poly_trait_impls: Dict::default(),
glue_patch_and_types: Vec::default(),
params: params_,
decls: Dict::default(),
locals: Dict::with_capacity(capacity),
consts: Dict::default(),
eval: Evaluator::default(),
types: Dict::default(),
mods: Dict::default(),
patches: Dict::default(),
_nlocals: 0,
level,
}
}
#[inline]
pub fn mono(
name: Str,
kind: ContextKind,
outer: Option<Context>,
super_classes: Vec<Type>,
super_traits: Vec<Type>,
level: usize,
) -> Self {
Self::with_capacity(
name,
kind,
vec![],
outer,
super_classes,
super_traits,
0,
level,
)
}
#[inline]
pub fn poly(
name: Str,
kind: ContextKind,
params: Vec<ParamSpec>,
outer: Option<Context>,
super_classes: Vec<Type>,
super_traits: Vec<Type>,
level: usize,
) -> Self {
Self::with_capacity(
name,
kind,
params,
outer,
super_classes,
super_traits,
0,
level,
)
}
pub fn poly_trait<S: Into<Str>>(
name: S,
params: Vec<ParamSpec>,
supers: Vec<Type>,
level: usize,
) -> Self {
let name = name.into();
Self::poly(
name,
ContextKind::Trait,
params,
None,
vec![],
supers,
level,
)
}
pub fn poly_class<S: Into<Str>>(
name: S,
params: Vec<ParamSpec>,
super_classes: Vec<Type>,
impl_traits: Vec<Type>,
level: usize,
) -> Self {
let name = name.into();
Self::poly(
name,
ContextKind::Class,
params,
None,
super_classes,
impl_traits,
level,
)
}
#[inline]
pub fn mono_trait<S: Into<Str>>(name: S, supers: Vec<Type>, level: usize) -> Self {
Self::poly_trait(name, vec![], supers, level)
}
#[inline]
pub fn mono_class<S: Into<Str>>(
name: S,
super_classes: Vec<Type>,
super_traits: Vec<Type>,
level: usize,
) -> Self {
Self::poly_class(name, vec![], super_classes, super_traits, level)
}
#[inline]
pub fn poly_patch<S: Into<Str>>(
name: S,
params: Vec<ParamSpec>,
patch_classes: Vec<Type>,
impl_traits: Vec<Type>,
level: usize,
) -> Self {
Self::poly(
name.into(),
ContextKind::Trait,
params,
None,
patch_classes,
impl_traits,
level,
)
}
#[inline]
pub fn module(name: Str, capacity: usize) -> Self {
Self::with_capacity(
name,
ContextKind::Module,
vec![],
None,
vec![],
vec![],
capacity,
Self::TOP_LEVEL,
)
}
#[inline]
pub fn caused_by(&self) -> Str {
self.name.clone()
}
fn registered(&self, name: &Str, recursive: bool) -> bool {
if self.params.iter().any(|(maybe_name, _)| {
maybe_name
.as_ref()
.map(|n| n.inspect() == name)
.unwrap_or(false)
}) || self.locals.contains_key(name)
{
return true;
}
if recursive {
if let Some(outer) = &self.outer {
outer.registered(name, recursive)
} else {
false
}
} else {
false
}
}
}
impl Context {
pub(crate) fn declare_var(
&mut self,
sig: &ast::VarSignature,
opt_t: Option<Type>,
id: Option<DefId>,
) -> TyCheckResult<()> {
self.declare_var_pat(sig, opt_t, id)
}
fn declare_var_pat(
&mut self,
sig: &ast::VarSignature,
opt_t: Option<Type>,
id: Option<DefId>,
) -> TyCheckResult<()> {
let vis = Private; let muty = Mutability::from(&sig.inspect().unwrap()[..]);
match &sig.pat {
ast::VarPattern::VarName(v) => {
if sig.t_spec.is_none() && opt_t.is_none() {
Err(TyCheckError::no_type_spec_error(
sig.loc(),
self.caused_by(),
v.inspect(),
))
} else {
if self.registered(v.inspect(), v.inspect().is_uppercase()) {
return Err(TyCheckError::duplicate_decl_error(
sig.loc(),
self.caused_by(),
v.inspect(),
));
}
let kind = id
.map(|id| VarKind::Defined(id))
.unwrap_or(VarKind::Declared);
let sig_t = self.instantiate_var_sig_t(sig, opt_t, PreRegister)?;
self.decls
.insert(v.clone(), VarInfo::new(sig_t, muty, vis, kind));
Ok(())
}
}
ast::VarPattern::Array(a) => {
if let Some(opt_ts) = opt_t.and_then(|t| t.non_default_params().cloned()) {
for (elem, p) in a.iter().zip(opt_ts.into_iter()) {
self.declare_var_pat(elem, Some(p.ty), None)?;
}
} else {
for elem in a.iter() {
self.declare_var_pat(elem, None, None)?;
}
}
Ok(())
}
_ => todo!(),
}
}
pub(crate) fn declare_sub(
&mut self,
sig: &ast::SubrSignature,
opt_ret_t: Option<Type>,
id: Option<DefId>,
) -> TyCheckResult<()> {
let name = sig.name.inspect();
let muty = Mutability::from(&name[..]);
let kind = id
.map(|id| VarKind::Defined(id))
.unwrap_or(VarKind::Declared);
if self.registered(name, name.is_uppercase()) {
return Err(TyCheckError::duplicate_decl_error(
sig.loc(),
self.caused_by(),
name,
));
}
let t = self.instantiate_sub_sig_t(sig, opt_ret_t, PreRegister)?;
let vi = VarInfo::new(t, muty, Private, kind);
if let Some(_decl) = self.decls.remove(name) {
return Err(TyCheckError::duplicate_decl_error(
sig.loc(),
self.caused_by(),
name,
));
} else {
self.decls.insert(sig.name.clone(), vi);
}
Ok(())
}
pub(crate) fn assign_var(
&mut self,
sig: &ast::VarSignature,
id: DefId,
body_t: &Type,
) -> TyCheckResult<()> {
self.assign_var_sig(sig, body_t, id)
}
fn assign_var_sig(
&mut self,
sig: &ast::VarSignature,
body_t: &Type,
id: DefId,
) -> TyCheckResult<()> {
self.validate_var_sig_t(sig, body_t, Normal)?;
let vis = Private; let muty = Mutability::from(&sig.inspect().unwrap()[..]);
let (generalized, bounds) = self.generalize_t(body_t.clone());
let generalized = if !bounds.is_empty() {
if self.rec_full_supertype_of(&Type::mono("GenericCallable"), &generalized) {
Type::quantified(generalized, bounds)
} else {
panic!()
}
} else {
generalized
};
match &sig.pat {
ast::VarPattern::Discard(_token) => Ok(()),
ast::VarPattern::VarName(v) => {
if self.registered(v.inspect(), v.inspect().is_uppercase()) {
Err(TyCheckError::reassign_error(
v.loc(),
self.caused_by(),
v.inspect(),
))
} else {
if let Some(_) = self.decls.remove(v.inspect()) {
}
let vi = VarInfo::new(generalized, muty, vis, VarKind::Defined(id));
self.params.push((Some(v.clone()), vi));
Ok(())
}
}
ast::VarPattern::SelfDot(_) => todo!(),
ast::VarPattern::Array(arr) => {
for (elem, inf) in arr.iter().zip(generalized.inner_ts().iter()) {
let id = DefId(get_hash(&(&self.name, elem)));
self.assign_var_sig(elem, inf, id)?;
}
Ok(())
}
ast::VarPattern::Tuple(_) => todo!(),
ast::VarPattern::Record { .. } => todo!(),
}
}
fn assign_param(
&mut self,
sig: &ast::ParamSignature,
outer: Option<ParamIdx>,
nth: usize,
opt_decl_t: Option<&ParamTy>,
) -> TyCheckResult<()> {
match &sig.pat {
ast::ParamPattern::Discard(_token) => Ok(()),
ast::ParamPattern::VarName(v) => {
if self.registered(v.inspect(), v.inspect().is_uppercase()) {
Err(TyCheckError::reassign_error(
v.loc(),
self.caused_by(),
v.inspect(),
))
} else {
let spec_t = self.instantiate_param_sig_t(sig, opt_decl_t, Normal)?;
let idx = if let Some(outer) = outer {
ParamIdx::nested(outer, nth)
} else {
ParamIdx::Nth(nth)
};
let default = if sig.opt_default_val.is_some() {
DefaultInfo::WithDefault
} else {
DefaultInfo::NonDefault
};
let kind = VarKind::parameter(DefId(get_hash(&(&self.name, v))), idx, default);
self.params.push((
Some(v.clone()),
VarInfo::new(spec_t, Immutable, Private, kind),
));
Ok(())
}
}
ast::ParamPattern::Array(arr) => {
let mut array_nth = 0;
let array_outer = if let Some(outer) = outer {
ParamIdx::nested(outer, nth)
} else {
ParamIdx::Nth(nth)
};
if let Some(decl_t) = opt_decl_t {
for (elem, p) in arr
.elems
.non_defaults
.iter()
.zip(decl_t.ty.non_default_params().unwrap())
{
self.assign_param(elem, Some(array_outer.clone()), array_nth, Some(p))?;
array_nth += 1;
}
for (elem, p) in arr
.elems
.defaults
.iter()
.zip(decl_t.ty.default_params().unwrap())
{
self.assign_param(elem, Some(array_outer.clone()), array_nth, Some(p))?;
array_nth += 1;
}
} else {
for elem in arr.elems.non_defaults.iter() {
self.assign_param(elem, Some(array_outer.clone()), array_nth, None)?;
array_nth += 1;
}
for elem in arr.elems.defaults.iter() {
self.assign_param(elem, Some(array_outer.clone()), array_nth, None)?;
array_nth += 1;
}
}
Ok(())
}
ast::ParamPattern::Lit(_) => Ok(()),
_ => todo!(),
}
}
pub(crate) fn assign_params(
&mut self,
params: &ast::Params,
opt_decl_subr_t: Option<SubrType>,
) -> TyCheckResult<()> {
if let Some(decl_subr_t) = opt_decl_subr_t {
for (nth, (sig, pt)) in params
.non_defaults
.iter()
.zip(decl_subr_t.non_default_params.iter())
.chain(
params
.defaults
.iter()
.zip(decl_subr_t.default_params.iter()),
)
.enumerate()
{
self.assign_param(sig, None, nth, Some(pt))?;
}
} else {
for (nth, sig) in params
.non_defaults
.iter()
.chain(params.defaults.iter())
.enumerate()
{
self.assign_param(sig, None, nth, None)?;
}
}
Ok(())
}
pub(crate) fn assign_subr(
&mut self,
sig: &ast::SubrSignature,
id: DefId,
body_t: &Type,
) -> TyCheckResult<()> {
let muty = if sig.name.is_const() {
Mutability::Const
} else {
Mutability::Immutable
};
let name = &sig.name;
let t = self
.get_current_scope_var(&name.inspect())
.map(|v| &v.t)
.unwrap();
let non_default_params = t.non_default_params().unwrap();
let default_params = t.default_params().unwrap();
if let Some(spec_ret_t) = t.return_t() {
self.unify(spec_ret_t, body_t, Some(sig.loc()), None)
.map_err(|e| {
TyCheckError::return_type_error(
e.core.loc,
e.caused_by,
readable_name(name.inspect()),
spec_ret_t,
body_t,
)
})?;
}
if self.registered(name.inspect(), name.inspect().is_uppercase()) {
Err(TyCheckError::reassign_error(
name.loc(),
self.caused_by(),
name.inspect(),
))
} else {
let sub_t = if sig.name.is_procedural() {
Type::proc(
non_default_params.clone(),
default_params.clone(),
body_t.clone(),
)
} else {
Type::func(
non_default_params.clone(),
default_params.clone(),
body_t.clone(),
)
};
sub_t.lift();
let (generalized, bounds) = self.generalize_t(sub_t);
let found_t = if !bounds.is_empty() {
if self.rec_full_supertype_of(&Type::mono("GenericCallable"), &generalized) {
Type::quantified(generalized, bounds)
} else {
panic!()
}
} else {
generalized
};
if let Some(mut vi) = self.decls.remove(name) {
if vi.t.has_unbound_var() {
vi.t.lift();
let (generalized, bounds) = self.generalize_t(vi.t.clone());
let generalized = if !bounds.is_empty() {
if self.rec_full_supertype_of(&Type::mono("GenericCallable"), &generalized)
{
Type::quantified(generalized, bounds)
} else {
panic!()
}
} else {
generalized
};
vi.t = generalized;
}
self.decls.insert(name.clone(), vi);
}
if let Some(vi) = self.decls.remove(name) {
if !self.rec_full_supertype_of(&vi.t, &found_t) {
return Err(TyCheckError::violate_decl_error(
sig.loc(),
self.caused_by(),
name.inspect(),
&vi.t,
&found_t,
));
}
}
let vi = VarInfo::new(found_t, muty, Private, VarKind::Defined(id));
log!("Registered {}::{name}: {}", self.name, &vi.t);
self.params.push((Some(name.clone()), vi));
Ok(())
}
}
pub(crate) fn import_mod(
&mut self,
var_name: &VarName,
mod_name: &hir::Expr,
) -> TyCheckResult<()> {
match mod_name {
hir::Expr::Lit(lit) => {
if self.rec_full_subtype_of(&lit.data.class(), &Str) {
let name = enum_unwrap!(lit.data.clone(), ValueObj::Str);
match &name[..] {
"math" => {
self.mods.insert(var_name.clone(), Self::init_py_math_mod());
}
"random" => {
self.mods
.insert(var_name.clone(), Self::init_py_random_mod());
}
other => todo!("importing {other}"),
}
} else {
return Err(TyCheckError::type_mismatch_error(
mod_name.loc(),
self.caused_by(),
"import::name",
&Str,
mod_name.ref_t(),
));
}
}
_ => {
return Err(TyCheckError::feature_error(
mod_name.loc(),
"non-literal importing",
self.caused_by(),
))
}
}
Ok(())
}
pub(crate) fn _push_subtype_bound(&mut self, sub: Type, sup: Type) {
self.bounds.push(TyBound::subtype(sub, sup));
}
pub(crate) fn _push_instance_bound(&mut self, name: Str, t: Type) {
self.bounds.push(TyBound::instance(name, t));
}
}
impl Context {
pub const TOP_LEVEL: usize = 1;
pub const GENERIC_LEVEL: usize = usize::MAX;
fn _independentise<'a>(_t: Type, _ts: &[Type]) -> Type {
todo!()
}
fn _generalize_tp(&self, free: TyParam) -> (TyParam, Set<TyBound>) {
match free {
TyParam::FreeVar(v) if v.is_linked() => {
let bounds: Set<TyBound>;
if let FreeKind::Linked(tp) = &mut *v.borrow_mut() {
(*tp, bounds) = self._generalize_tp(tp.clone());
} else {
assume_unreachable!()
}
(TyParam::FreeVar(v), bounds)
}
TyParam::FreeVar(fv) if fv.level() > Some(self.level) => match &*fv.borrow() {
FreeKind::Unbound { id, constraint, .. } => {
let name = id.to_string();
let bound = match constraint {
Constraint::SubtypeOf(sup) => {
TyBound::subtype(Type::mono(name.clone()), sup.clone())
}
Constraint::SupertypeOf(sub) => {
TyBound::supertype(Type::mono(name.clone()), sub.clone())
}
Constraint::Sandwiched { sub, sup } => {
TyBound::sandwiched(sub.clone(), Type::mono(name.clone()), sup.clone())
}
Constraint::TypeOf(t) => TyBound::instance(Str::rc(&name[..]), t.clone()),
Constraint::Uninited => unreachable!(),
};
(TyParam::mono_q(&name), set! {bound})
}
FreeKind::NamedUnbound {
name, constraint, ..
} => {
let bound = match constraint {
Constraint::SubtypeOf(sup) => {
TyBound::subtype(Type::mono(name.clone()), sup.clone())
}
Constraint::SupertypeOf(sub) => {
TyBound::supertype(Type::mono(name.clone()), sub.clone())
}
Constraint::Sandwiched { sub, sup } => {
TyBound::sandwiched(sub.clone(), Type::mono(name.clone()), sup.clone())
}
Constraint::TypeOf(t) => TyBound::instance(Str::rc(&name[..]), t.clone()),
Constraint::Uninited => unreachable!(),
};
(TyParam::mono_q(name), set! {bound})
}
_ => assume_unreachable!(),
},
other if other.has_no_unbound_var() => (other, set! {}),
other => todo!("{other}"),
}
}
fn generalize_t(&self, free: Type) -> (Type, Set<TyBound>) {
match free {
FreeVar(v) if v.is_linked() => {
let bounds: Set<TyBound>;
if let FreeKind::Linked(t) = &mut *v.borrow_mut() {
(*t, bounds) = self.generalize_t(t.clone());
} else {
assume_unreachable!()
}
(Type::FreeVar(v), bounds)
}
FreeVar(fv) if fv.level() > Some(self.level) => match &*fv.borrow() {
FreeKind::Unbound { id, constraint, .. } => {
let name = id.to_string();
let bound = match constraint {
Constraint::SubtypeOf(sup) => {
TyBound::subtype(Type::mono(name.clone()), sup.clone())
}
Constraint::SupertypeOf(sub) => {
TyBound::supertype(Type::mono(name.clone()), sub.clone())
}
Constraint::Sandwiched { sub, sup } => {
TyBound::sandwiched(sub.clone(), Type::mono(name.clone()), sup.clone())
}
Constraint::TypeOf(t) => TyBound::instance(Str::rc(&name[..]), t.clone()),
Constraint::Uninited => unreachable!(),
};
(Type::mono(&name), set! {bound})
}
FreeKind::NamedUnbound {
name, constraint, ..
} => {
let bound = match constraint {
Constraint::SubtypeOf(sup) => {
TyBound::subtype(Type::mono(name.clone()), sup.clone())
}
Constraint::SupertypeOf(sub) => {
TyBound::supertype(Type::mono(name.clone()), sub.clone())
}
Constraint::Sandwiched { sub, sup } => {
TyBound::sandwiched(sub.clone(), Type::mono(name.clone()), sup.clone())
}
Constraint::TypeOf(t) => TyBound::instance(Str::rc(&name[..]), t.clone()),
Constraint::Uninited => unreachable!(),
};
(Type::mono(name), set! {bound})
}
_ => assume_unreachable!(),
},
Subr(mut subr) => {
let mut bounds = set! {};
let kind = match subr.kind {
SubrKind::FuncMethod(self_t) => {
let (t, bs) = self.generalize_t(*self_t);
bounds.merge(bs);
SubrKind::fn_met(t)
}
SubrKind::ProcMethod { before, after } => {
let (before, bs) = self.generalize_t(*before);
bounds.merge(bs);
if let Some(after) = after {
let (after, bs) = self.generalize_t(*after);
bounds.merge(bs);
SubrKind::pr_met(before, Some(after))
} else {
SubrKind::pr_met(before, None)
}
}
other => other,
};
subr.non_default_params.iter_mut().for_each(|p| {
let (t, bs) = self.generalize_t(mem::take(&mut p.ty));
p.ty = t;
bounds.merge(bs);
});
subr.default_params.iter_mut().for_each(|p| {
let (t, bs) = self.generalize_t(mem::take(&mut p.ty));
p.ty = t;
bounds.merge(bs);
});
let (return_t, bs) = self.generalize_t(*subr.return_t);
bounds.merge(bs);
(
Type::subr(kind, subr.non_default_params, subr.default_params, return_t),
bounds,
)
}
other => (other, set! {}),
}
}
pub(crate) fn type_params_bounds(&self) -> Set<TyBound> {
self.params
.iter()
.filter(|(opt_name, vi)| vi.kind.is_parameter() && opt_name.is_some())
.map(|(name, vi)| {
TyBound::instance(name.as_ref().unwrap().inspect().clone(), vi.t.clone())
})
.collect()
}
pub(crate) fn type_params_variance(&self) -> Vec<Variance> {
self.params
.iter()
.map(|(opt_name, _)| {
if let Some(name) = opt_name {
if let Some(t) = self.super_traits.iter().find(|t| {
(t.name() == "Input" || t.name() == "Output")
&& t.inner_ts()
.first()
.map(|t| t.name() == &name.inspect()[..])
.unwrap_or(false)
}) {
match t.name() {
"Output" => Variance::Covariant,
"Input" => Variance::Contravariant,
_ => unreachable!(),
}
} else {
Variance::Invariant
}
} else {
Variance::Invariant
}
})
.collect()
}
fn instantiate_tp(quantified: TyParam, tv_ctx: TyVarContext) -> (TyParam, TyVarContext) {
match quantified {
TyParam::MonoQVar(n) => {
if let Some(tp) = tv_ctx.get_typaram(&n) {
(tp.clone(), tv_ctx)
} else if let Some(t) = tv_ctx.get_tyvar(&n) {
(TyParam::t(t.clone()), tv_ctx)
} else {
panic!("type parameter {n} is not defined")
}
}
TyParam::UnaryOp { op, val } => {
let (res, tv_ctx) = Self::instantiate_tp(*val, tv_ctx);
(TyParam::unary(op, res), tv_ctx)
}
TyParam::BinOp { op, lhs, rhs } => {
let (lhs, tv_ctx) = Self::instantiate_tp(*lhs, tv_ctx);
let (rhs, tv_ctx) = Self::instantiate_tp(*rhs, tv_ctx);
(TyParam::bin(op, lhs, rhs), tv_ctx)
}
TyParam::Type(t) => {
let (t, tv_ctx) = Self::instantiate_t(*t, tv_ctx);
(TyParam::t(t), tv_ctx)
}
p @ (TyParam::ConstObj(_) | TyParam::Mono(_)) => (p, tv_ctx),
other => todo!("{other}"),
}
}
pub(crate) fn instantiate_t(
quantified: Type,
mut tv_ctx: TyVarContext,
) -> (Type, TyVarContext) {
match quantified {
MonoQVar(n) => {
if let Some(t) = tv_ctx.get_tyvar(&n) {
(t.clone(), tv_ctx)
} else if let Some(tp) = tv_ctx.get_typaram(&n) {
if let TyParam::Type(t) = tp {
(*t.clone(), tv_ctx)
} else {
todo!(
"typaram_insts: {}\ntyvar_insts:{}\n{tp}",
tv_ctx.typaram_instances,
tv_ctx.tyvar_instances,
)
}
} else {
panic!("the type variable {n} is not defined")
}
}
PolyQVar { name, mut params } => {
for param in params.iter_mut() {
(*param, tv_ctx) = Self::instantiate_tp(mem::take(param), tv_ctx);
}
(Type::poly_q(name, params), tv_ctx)
}
Refinement(mut refine) => {
refine.preds = refine
.preds
.into_iter()
.map(|mut pred| {
for tp in pred.typarams_mut() {
(*tp, tv_ctx) = Self::instantiate_tp(mem::take(tp), tv_ctx.clone());
}
pred
})
.collect();
(Type::Refinement(refine), tv_ctx)
}
Subr(mut subr) => {
let kind = match subr.kind {
SubrKind::FuncMethod(self_t) => {
let (res, _tv_ctx) = Self::instantiate_t(*self_t, tv_ctx);
tv_ctx = _tv_ctx;
SubrKind::FuncMethod(Box::new(res))
}
SubrKind::ProcMethod { before, after } => {
let (before, _tv_ctx) = Self::instantiate_t(*before, tv_ctx);
let (after, _tv_ctx) = if let Some(after) = after {
let (after, _tv_ctx) = Self::instantiate_t(*after, _tv_ctx);
(Some(after), _tv_ctx)
} else {
(None, _tv_ctx)
};
tv_ctx = _tv_ctx;
SubrKind::pr_met(before, after)
}
other => other,
};
for p in subr.non_default_params.iter_mut() {
(p.ty, tv_ctx) = Self::instantiate_t(mem::take(&mut p.ty), tv_ctx);
}
for p in subr.default_params.iter_mut() {
(p.ty, tv_ctx) = Self::instantiate_t(mem::take(&mut p.ty), tv_ctx);
}
let (return_t, tv_ctx) = Self::instantiate_t(*subr.return_t, tv_ctx);
(
Type::subr(kind, subr.non_default_params, subr.default_params, return_t),
tv_ctx,
)
}
Record(mut dict) => {
for v in dict.values_mut() {
(*v, tv_ctx) = Self::instantiate_t(mem::take(v), tv_ctx);
}
(Type::Record(dict), tv_ctx)
}
Ref(t) => {
let (t, tv_ctx) = Self::instantiate_t(*t, tv_ctx);
(Type::ref_(t), tv_ctx)
}
RefMut(t) => {
let (t, tv_ctx) = Self::instantiate_t(*t, tv_ctx);
(Type::ref_mut(t), tv_ctx)
}
VarArgs(t) => {
let (t, tv_ctx) = Self::instantiate_t(*t, tv_ctx);
(Type::var_args(t), tv_ctx)
}
MonoProj { lhs, rhs } => {
let (lhs, tv_ctx) = Self::instantiate_t(*lhs, tv_ctx);
(Type::mono_proj(lhs, rhs), tv_ctx)
}
Poly { name, mut params } => {
for param in params.iter_mut() {
(*param, tv_ctx) = Self::instantiate_tp(mem::take(param), tv_ctx);
}
(Type::poly(name, params), tv_ctx)
}
other if other.is_monomorphic() => (other, tv_ctx),
other => todo!("{other}"),
}
}
fn instantiate(&self, quantified: Type, callee: &hir::Expr) -> TyCheckResult<Type> {
match quantified {
Quantified(quant) => {
let tv_ctx = TyVarContext::new(self.level, quant.bounds, &self);
let (t, _) = Self::instantiate_t(*quant.unbound_callable, tv_ctx);
match &t {
Type::Subr(subr) => {
match (subr.kind.self_t(), callee.receiver_t()) {
(Some(l), Some(r)) => {
self.unify(l, r, None, Some(callee.loc()))?;
}
(None, Some(r))
if self.rec_full_subtype_of(r, &Type::mono("Named")) => {}
(None, None) => {}
(l, r) => todo!("{l:?}, {r:?}"),
}
}
_ => unreachable!(),
}
Ok(t)
}
other => Ok(other),
}
}
fn substitute_call(
&self,
callee: &hir::Expr,
instance: &Type,
pos_args: &[hir::PosArg],
kw_args: &[hir::KwArg],
) -> TyCheckResult<()> {
match instance {
Type::Subr(subr) => {
let params_len = subr.non_default_params.len() + subr.default_params.len();
if params_len < pos_args.len() + kw_args.len() {
return Err(TyCheckError::too_many_args_error(
callee.loc(),
&callee.to_string(),
self.caused_by(),
params_len,
pos_args.len(),
kw_args.len(),
));
}
let mut passed_params = set! {};
let params = subr
.non_default_params
.iter()
.chain(subr.default_params.iter());
for (param_ty, pos_arg) in params.clone().zip(pos_args) {
let arg_t = pos_arg.expr.ref_t();
let param_t = ¶m_ty.ty;
self.sub_unify(
arg_t,
param_t,
None,
Some(pos_arg.loc()),
)
.map_err(|e| {
let name = callee.var_full_name().unwrap_or("".to_string());
let name =
name + "::" + param_ty.name.as_ref().map(|s| readable_name(&s[..])).unwrap_or("");
TyCheckError::type_mismatch_error(
e.core.loc,
e.caused_by,
&name[..],
param_t,
arg_t,
)
})?;
if let Some(name) = ¶m_ty.name {
if passed_params.contains(name) {
return Err(TyCheckError::multiple_args_error(
callee.loc(),
&callee.to_string(),
self.caused_by(),
name,
));
} else {
passed_params.insert(name);
}
}
}
let param_ts = {
let mut param_ts = Dict::new();
for param_ty in params {
if let Some(name) = ¶m_ty.name {
param_ts.insert(name, ¶m_ty.ty);
}
}
param_ts
};
for kw_arg in kw_args.iter() {
if let Some(param_ty) = param_ts.get(kw_arg.keyword.inspect()) {
self.sub_unify(kw_arg.expr.ref_t(), param_ty, None, Some(kw_arg.loc()))?;
} else {
return Err(TyCheckError::unexpected_kw_arg_error(
kw_arg.keyword.loc(),
&callee.to_string(),
self.caused_by(),
kw_arg.keyword.inspect(),
));
}
}
Ok(())
}
other => todo!("{other}"),
}
}
fn deref_tp(&self, tp: TyParam) -> TyCheckResult<TyParam> {
match tp {
TyParam::FreeVar(fv) if fv.is_linked() => Ok(fv.unwrap_linked()),
TyParam::Type(t) => Ok(TyParam::t(self.deref_tyvar(*t)?)),
TyParam::App { name, mut args } => {
for param in args.iter_mut() {
*param = self.deref_tp(mem::take(param))?;
}
Ok(TyParam::App { name, args })
}
TyParam::BinOp { .. } => todo!(),
TyParam::UnaryOp { .. } => todo!(),
TyParam::Array(_) | TyParam::Tuple(_) => todo!(),
t => Ok(t),
}
}
fn deref_constraint(&self, constraint: Constraint) -> TyCheckResult<Constraint> {
match constraint {
Constraint::SubtypeOf(sup) => {
Ok(Constraint::SubtypeOf(self.deref_tyvar(sup)?))
},
Constraint::Sandwiched { sub, sup } => {
Ok(Constraint::sandwiched(self.deref_tyvar(sub)?, self.deref_tyvar(sup)?))
},
Constraint::SupertypeOf(sub) => {
Ok(Constraint::SupertypeOf(self.deref_tyvar(sub)?))
},
Constraint::TypeOf(t) => {
Ok(Constraint::TypeOf(self.deref_tyvar(t)?))
},
_ => unreachable!(),
}
}
fn deref_tyvar(&self, t: Type) -> TyCheckResult<Type> {
match t {
Type::FreeVar(fv) if fv.constraint_is_typeof() => Ok(Type::FreeVar(fv)),
Type::FreeVar(fv) if fv.constraint_is_subtypeof() => {
if self.level <= fv.level().unwrap() {
Ok(fv.crack_constraint().super_type().unwrap().clone())
} else {
Ok(Type::FreeVar(fv))
}
},
Type::FreeVar(fv) if fv.is_unbound() => {
if self.level == 0 {
match &*fv.crack_constraint() {
Constraint::SupertypeOf(t)
| Constraint::SubtypeOf(t) => Ok(t.clone()),
Constraint::Sandwiched { sub, .. } => Ok(sub.clone()),
Constraint::TypeOf(_) => {
Err(TyCheckError::unreachable(fn_name!(), line!()))
},
_ => unreachable!(),
}
} else {
let new_constraint = fv.crack_constraint().clone();
let new_constraint = self.deref_constraint(new_constraint)?;
fv.update_constraint(new_constraint);
Ok(Type::FreeVar(fv))
}
},
Type::FreeVar(fv) if fv.is_linked() => {
let t = fv.unwrap_linked();
self.deref_tyvar(t)
},
Type::Poly { name, mut params } => {
for param in params.iter_mut() {
*param = self.deref_tp(mem::take(param))?;
}
Ok(Type::Poly { name, params })
}
Type::Subr(mut subr) => {
match &mut subr.kind {
SubrKind::FuncMethod(t) => {
*t = Box::new(self.deref_tyvar(mem::take(t))?);
}
SubrKind::ProcMethod { before, after } => {
*before = Box::new(self.deref_tyvar(mem::take(before))?);
if let Some(after) = after {
*after = Box::new(self.deref_tyvar(mem::take(after))?);
}
}
_ => {}
}
let params = subr
.non_default_params
.iter_mut()
.chain(subr.default_params.iter_mut());
for param in params {
param.ty = self.deref_tyvar(mem::take(&mut param.ty))?;
}
subr.return_t = Box::new(self.deref_tyvar(mem::take(&mut subr.return_t))?);
Ok(Type::Subr(subr))
}
t => Ok(t),
}
}
pub(crate) fn deref_toplevel(&mut self, mut hir: hir::HIR) -> TyCheckResult<hir::HIR> {
self.level = 0;
for chunk in hir.module.iter_mut() {
self.deref_expr_t(chunk).map_err(|e| e)?;
}
Ok(hir)
}
fn deref_expr_t(&self, expr: &mut hir::Expr) -> TyCheckResult<()> {
match expr {
hir::Expr::Lit(_) => Ok(()),
hir::Expr::Accessor(acc) => {
let t = acc.ref_mut_t();
*t = self.deref_tyvar(mem::take(t))?;
match acc {
hir::Accessor::Attr(attr) => {
self.deref_expr_t(&mut attr.obj)?;
}
hir::Accessor::Local(_) => {}
_ => todo!(),
}
Ok(())
},
hir::Expr::Array(_array) => {
todo!()
},
hir::Expr::Dict(_dict) => {
todo!()
}
hir::Expr::BinOp(binop) => {
let t = binop.signature_mut_t().unwrap();
*t = self.deref_tyvar(mem::take(t))?;
self.deref_expr_t(&mut binop.lhs)?;
self.deref_expr_t(&mut binop.rhs)?;
Ok(())
},
hir::Expr::UnaryOp(unaryop) => {
let t = unaryop.signature_mut_t().unwrap();
*t = self.deref_tyvar(mem::take(t))?;
self.deref_expr_t(&mut unaryop.expr)?;
Ok(())
},
hir::Expr::Call(call) => {
let t = call.signature_mut_t().unwrap();
*t = self.deref_tyvar(mem::take(t))?;
for arg in call.args.pos_args.iter_mut() {
self.deref_expr_t(&mut arg.expr)?;
}
Ok(())
}
hir::Expr::Decl(decl) => {
decl.t = self.deref_tyvar(mem::take(&mut decl.t))?;
Ok(())
},
hir::Expr::Def(def) => {
match &mut def.sig {
hir::Signature::Var(var) => {
var.t = self.deref_tyvar(mem::take(&mut var.t))?;
},
hir::Signature::Subr(subr) => {
subr.t = self.deref_tyvar(mem::take(&mut subr.t))?;
},
}
for chunk in def.body.block.iter_mut() {
self.deref_expr_t(chunk)?;
}
Ok(())
}
hir::Expr::Lambda(lambda) => {
lambda.t = self.deref_tyvar(mem::take(&mut lambda.t))?;
for chunk in lambda.body.iter_mut() {
self.deref_expr_t(chunk)?;
}
Ok(())
}
}
}
fn propagate(&self, t: &Type, callee: &hir::Expr) -> TyCheckResult<()> {
match t {
Type::Subr(subr) => match &subr.kind {
SubrKind::ProcMethod {
before: _,
after: Some(after),
} => {
let receiver_t = callee.receiver_t().unwrap();
self.reunify(receiver_t, after, Some(callee.loc()), None)?;
}
_ => {}
},
_ => {}
}
Ok(())
}
fn _occur(&self, _t: Type) -> TyCheckResult<Type> {
todo!()
}
pub(crate) fn unify_tp(
&self,
l: &TyParam,
r: &TyParam,
bounds: Option<&Set<TyBound>>,
lhs_variance: Option<&Vec<Variance>>,
allow_divergence: bool,
) -> TyCheckResult<()> {
if l.has_no_unbound_var() && r.has_no_unbound_var() && l.rec_eq(r) {
return Ok(());
}
match (l, r) {
(TyParam::Type(l), TyParam::Type(r)) => self.unify(&l, &r, None, None),
(ltp @ TyParam::FreeVar(lfv), rtp @ TyParam::FreeVar(rfv))
if lfv.is_unbound() && rfv.is_unbound() =>
{
if lfv.level().unwrap() > rfv.level().unwrap() {
lfv.link(rtp);
} else {
rfv.link(ltp);
}
Ok(())
}
(TyParam::FreeVar(fv), tp) | (tp, TyParam::FreeVar(fv)) => {
match &*fv.borrow() {
FreeKind::Linked(l) => {
return self.unify_tp(l, tp, bounds, lhs_variance, allow_divergence)
}
FreeKind::Unbound { .. } | FreeKind::NamedUnbound { .. } => {}
} let fv_t = fv.borrow().constraint().unwrap().typ().unwrap().clone(); let tp_t = self.eval.get_tp_t(tp, bounds, &self)?;
if self.rec_full_supertype_of(&fv_t, &tp_t) {
if fv.level() < Some(self.level) {
let new_constraint = Constraint::SubtypeOf(tp_t.clone());
if self.is_sub_constraint_of(
fv.borrow().constraint().unwrap(),
&new_constraint,
) || fv.borrow().constraint().unwrap().typ() == Some(&Type)
{
fv.update_constraint(new_constraint);
}
} else {
fv.link(tp);
}
Ok(())
} else {
if allow_divergence
&& (self.eq_tp(&tp, &TyParam::value(Inf), None, None)
|| self.eq_tp(&tp, &TyParam::value(NegInf), None, None))
&& self.rec_full_subtype_of(&fv_t, &Type::mono("Num"))
{
fv.link(tp);
Ok(())
} else {
Err(TyCheckError::unreachable(fn_name!(), line!()))
}
}
}
(TyParam::UnaryOp { op: lop, val: lval }, TyParam::UnaryOp { op: rop, val: rval })
if lop == rop =>
{
self.unify_tp(lval, rval, bounds, lhs_variance, allow_divergence)
}
(
TyParam::BinOp { op: lop, lhs, rhs },
TyParam::BinOp {
op: rop,
lhs: lhs2,
rhs: rhs2,
},
) if lop == rop => {
self.unify_tp(lhs, lhs2, bounds, lhs_variance, allow_divergence)?;
self.unify_tp(rhs, rhs2, bounds, lhs_variance, allow_divergence)
}
(l, r) => panic!("type-parameter unification failed:\nl:{l}\nr: {r}"),
}
}
fn reunify_tp(
&self,
before: &TyParam,
after: &TyParam,
bounds: Option<&Set<TyBound>>,
lhs_variance: Option<&Vec<Variance>>,
) -> TyCheckResult<()> {
match (before, after) {
(TyParam::ConstObj(ConstObj::MutValue(l)), TyParam::ConstObj(ConstObj::Value(r))) => {
*l.borrow_mut() = r.clone();
Ok(())
}
(
TyParam::ConstObj(ConstObj::MutValue(l)),
TyParam::ConstObj(ConstObj::MutValue(r)),
) => {
*l.borrow_mut() = r.borrow().clone();
Ok(())
}
(TyParam::Type(l), TyParam::Type(r)) => self.reunify(&l, &r, None, None),
(TyParam::UnaryOp { op: lop, val: lval }, TyParam::UnaryOp { op: rop, val: rval })
if lop == rop =>
{
self.reunify_tp(lval, rval, bounds, lhs_variance)
}
(
TyParam::BinOp { op: lop, lhs, rhs },
TyParam::BinOp {
op: rop,
lhs: lhs2,
rhs: rhs2,
},
) if lop == rop => {
self.reunify_tp(lhs, lhs2, bounds, lhs_variance)?;
self.reunify_tp(rhs, rhs2, bounds, lhs_variance)
}
(l, r) if self.eq_tp(l, r, bounds, lhs_variance) => Ok(()),
(l, r) => panic!("type-parameter re-unification failed:\nl: {l}\nr: {r}"),
}
}
fn unify_pred(&self, l_pred: &Predicate, r_pred: &Predicate) -> TyCheckResult<()> {
match (l_pred, r_pred) {
(Pred::Value(_), Pred::Value(_)) | (Pred::Const(_), Pred::Const(_)) => Ok(()),
(Pred::Equal { rhs, .. }, Pred::Equal { rhs: rhs2, .. })
| (Pred::GreaterEqual { rhs, .. }, Pred::GreaterEqual { rhs: rhs2, .. })
| (Pred::LessEqual { rhs, .. }, Pred::LessEqual { rhs: rhs2, .. })
| (Pred::NotEqual { rhs, .. }, Pred::NotEqual { rhs: rhs2, .. }) => {
self.unify_tp(rhs, rhs2, None, None, false)
}
(Pred::And(l1, r1), Pred::And(l2, r2))
| (Pred::Or(l1, r1), Pred::Or(l2, r2))
| (Pred::Not(l1, r1), Pred::Not(l2, r2)) => {
match (self.unify_pred(l1, l2), self.unify_pred(r1, r2)) {
(Ok(()), Ok(())) => Ok(()),
(Ok(()), Err(e)) | (Err(e), Ok(())) | (Err(e), Err(_)) => Err(e),
}
}
(Pred::GreaterEqual { rhs, .. }, Pred::And(l, r))
| (Predicate::And(l, r), Pred::GreaterEqual { rhs, .. }) => {
match (l.as_ref(), r.as_ref()) {
(
Pred::GreaterEqual { rhs: ge_rhs, .. },
Pred::LessEqual { rhs: le_rhs, .. },
)
| (
Pred::LessEqual { rhs: le_rhs, .. },
Pred::GreaterEqual { rhs: ge_rhs, .. },
) => {
self.unify_tp(rhs, ge_rhs, None, None, false)?;
self.unify_tp(le_rhs, &TyParam::value(Inf), None, None, true)
}
_ => Err(TyCheckError::pred_unification_error(
l_pred,
r_pred,
self.caused_by(),
)),
}
}
(Pred::LessEqual { rhs, .. }, Pred::And(l, r))
| (Pred::And(l, r), Pred::LessEqual { rhs, .. }) => match (l.as_ref(), r.as_ref()) {
(Pred::GreaterEqual { rhs: ge_rhs, .. }, Pred::LessEqual { rhs: le_rhs, .. })
| (Pred::LessEqual { rhs: le_rhs, .. }, Pred::GreaterEqual { rhs: ge_rhs, .. }) => {
self.unify_tp(rhs, le_rhs, None, None, false)?;
self.unify_tp(ge_rhs, &TyParam::value(NegInf), None, None, true)
}
_ => Err(TyCheckError::pred_unification_error(
l_pred,
r_pred,
self.caused_by(),
)),
},
(Pred::Equal { rhs, .. }, Pred::And(l, r))
| (Pred::And(l, r), Pred::Equal { rhs, .. }) => match (l.as_ref(), r.as_ref()) {
(Pred::GreaterEqual { rhs: ge_rhs, .. }, Pred::LessEqual { rhs: le_rhs, .. })
| (Pred::LessEqual { rhs: le_rhs, .. }, Pred::GreaterEqual { rhs: ge_rhs, .. }) => {
self.unify_tp(rhs, le_rhs, None, None, false)?;
self.unify_tp(rhs, ge_rhs, None, None, false)
}
_ => Err(TyCheckError::pred_unification_error(
l_pred,
r_pred,
self.caused_by(),
)),
},
_ => Err(TyCheckError::pred_unification_error(
l_pred,
r_pred,
self.caused_by(),
)),
}
}
pub(crate) fn unify(
&self,
lhs_t: &Type,
rhs_t: &Type,
lhs_loc: Option<Location>,
rhs_loc: Option<Location>,
) -> TyCheckResult<()> {
if lhs_t.has_no_unbound_var()
&& rhs_t.has_no_unbound_var()
&& self.rec_full_supertype_of(lhs_t, rhs_t)
{
return Ok(());
}
match (lhs_t, rhs_t) {
(lt @ Type::FreeVar(lfv), rt @ Type::FreeVar(rfv))
if lfv.is_unbound() && rfv.is_unbound() =>
{
if lfv.constraint_is_typeof() && !rfv.constraint_is_typeof() {
lfv.update_constraint(rfv.crack_constraint().clone());
} else if rfv.constraint_is_typeof() && !lfv.constraint_is_typeof() {
rfv.update_constraint(lfv.crack_constraint().clone());
}
if lfv.level().unwrap() > rfv.level().unwrap() {
lfv.link(rt);
} else {
rfv.link(lt);
}
Ok(())
}
(Type::FreeVar(fv), t) | (t, Type::FreeVar(fv)) => {
match &mut *fv.borrow_mut() {
FreeKind::Linked(l) => return self.unify(l, t, lhs_loc, rhs_loc),
FreeKind::Unbound {
lev, constraint, ..
}
| FreeKind::NamedUnbound {
lev, constraint, ..
} => {
t.update_level(*lev);
if let Some(sup) = constraint.super_type_mut() {
if self.rec_full_subtype_of(sup, t) {
*sup = t.clone();
} else {
self.sub_unify(t, sup, rhs_loc, lhs_loc)?;
}
}
}
} let new_constraint = Constraint::SubtypeOf(t.clone());
if fv.level() < Some(self.level) {
if self.is_sub_constraint_of(fv.borrow().constraint().unwrap(), &new_constraint)
|| fv.borrow().constraint().unwrap().typ() == Some(&Type)
{
fv.update_constraint(new_constraint);
}
} else {
fv.link(t);
}
Ok(())
}
(Type::Refinement(l), Type::Refinement(r)) => {
if !self.formal_supertype_of(&l.t, &r.t, None, None)
&& !self.formal_supertype_of(&r.t, &l.t, None, None)
{
return Err(TyCheckError::unification_error(
lhs_t,
rhs_t,
lhs_loc,
rhs_loc,
self.caused_by(),
));
}
for l_pred in l.preds.iter() {
for r_pred in r.preds.iter() {
self.unify_pred(l_pred, r_pred)?;
}
}
Ok(())
}
(Type::Refinement(_), r) => {
let rhs_t = self.into_refinement(r.clone());
self.unify(lhs_t, &Type::Refinement(rhs_t), lhs_loc, rhs_loc)
}
(l, Type::Refinement(_)) => {
let lhs_t = self.into_refinement(l.clone());
self.unify(&Type::Refinement(lhs_t), rhs_t, lhs_loc, rhs_loc)
}
(Type::Subr(ls), Type::Subr(rs)) if ls.kind.same_kind_as(&rs.kind) => {
if let (Some(l), Some(r)) = (ls.kind.self_t(), rs.kind.self_t()) {
self.unify(l, r, lhs_loc, rhs_loc)?;
}
for (l, r) in ls
.non_default_params
.iter()
.zip(rs.non_default_params.iter())
{
self.unify(&l.ty, &r.ty, lhs_loc, rhs_loc)?;
}
self.unify(&ls.return_t, &rs.return_t, lhs_loc, rhs_loc)
}
(Type::Ref(l), Type::Ref(r))
| (Type::RefMut(l), Type::RefMut(r))
| (VarArgs(l), VarArgs(r)) => self.unify(l, r, lhs_loc, rhs_loc),
(Type::Ref(l), r) | (Type::RefMut(l), r) => self.unify(l, r, lhs_loc, rhs_loc),
(l, Type::Ref(r)) | (l, Type::RefMut(r)) => self.unify(l, r, lhs_loc, rhs_loc),
(
Type::Poly {
name: ln,
params: lps,
},
Type::Poly {
name: rn,
params: rps,
},
) => {
if ln != rn {
return Err(TyCheckError::unification_error(
lhs_t,
rhs_t,
lhs_loc,
rhs_loc,
self.caused_by(),
));
}
for (l, r) in lps.iter().zip(rps.iter()) {
self.unify_tp(l, r, None, None, false)?;
}
Ok(())
}
(Type::Poly { name: _, params: _ }, _r) => {
todo!()
}
(l, r) => Err(TyCheckError::unification_error(
l,
r,
lhs_loc,
rhs_loc,
self.caused_by(),
)),
}
}
pub(crate) fn reunify(
&self,
before_t: &Type,
after_t: &Type,
bef_loc: Option<Location>,
aft_loc: Option<Location>,
) -> TyCheckResult<()> {
match (before_t, after_t) {
(Type::FreeVar(fv), r) if fv.is_linked() => {
self.reunify(&fv.crack(), r, bef_loc, aft_loc)
}
(l, Type::FreeVar(fv)) if fv.is_linked() => {
self.reunify(l, &fv.crack(), bef_loc, aft_loc)
}
(Type::Ref(l), Type::Ref(r))
| (Type::RefMut(l), Type::RefMut(r))
| (Type::VarArgs(l), Type::VarArgs(r)) => self.reunify(l, r, bef_loc, aft_loc),
(Type::Ref(l), r) | (Type::RefMut(l), r) => self.reunify(l, r, bef_loc, aft_loc),
(l, Type::Ref(r)) | (l, Type::RefMut(r)) => self.reunify(l, r, bef_loc, aft_loc),
(
Type::Poly {
name: ln,
params: lps,
},
Type::Poly {
name: rn,
params: rps,
},
) => {
if ln != rn {
let before_t = Type::poly(ln.clone(), lps.clone());
return Err(TyCheckError::re_unification_error(
&before_t,
after_t,
bef_loc,
aft_loc,
self.caused_by(),
));
}
for (l, r) in lps.iter().zip(rps.iter()) {
self.reunify_tp(l, r, None, None)?;
}
Ok(())
}
(l, r) if self.formal_same_type_of(l, r, None, None) => Ok(()),
(l, r) => Err(TyCheckError::re_unification_error(
l,
r,
bef_loc,
aft_loc,
self.caused_by(),
)),
}
}
fn sub_unify(
&self,
maybe_sub: &Type,
maybe_sup: &Type,
sub_loc: Option<Location>,
sup_loc: Option<Location>,
) -> TyCheckResult<()> {
let maybe_sub_is_sub = self.rec_full_subtype_of(maybe_sub, maybe_sup);
if maybe_sub.has_no_unbound_var() && maybe_sup.has_no_unbound_var() && maybe_sub_is_sub {
return Ok(());
}
if !maybe_sub_is_sub {
let loc = sub_loc.or(sup_loc).unwrap_or(Location::Unknown);
return Err(TyCheckError::type_mismatch_error(
loc,
self.caused_by(),
"<???>",
maybe_sup,
maybe_sub,
));
}
match (maybe_sub, maybe_sup) {
(l, Type::FreeVar(fv)) if fv.is_unbound() => {
match &mut *fv.borrow_mut() {
FreeKind::NamedUnbound { constraint, .. }
| FreeKind::Unbound { constraint, .. } => match constraint {
Constraint::SupertypeOf(sub) => {
if self.rec_full_supertype_of(l, sub) {
*constraint = Constraint::SupertypeOf(l.clone());
} else if self.rec_full_subtype_of(l, sub) {
} else {
todo!()
}
}
Constraint::SubtypeOf(sup) => {
if !self.rec_full_subtype_of(l, sup) {
return Err(TyCheckError::subtyping_error(
l,
sup,
sub_loc,
sup_loc,
self.caused_by(),
))
} else {
*constraint = Constraint::sandwiched(l.clone(), mem::take(sup));
}
}
Constraint::Sandwiched { sub, sup } => {
if self.rec_full_no_relation_of(l, sup) {
return Err(TyCheckError::subtyping_error(
l,
sup, sub_loc,
sup_loc,
self.caused_by(),
))
} else {
let union = self.union(l, sub);
*constraint = Constraint::sandwiched(union, mem::take(sub));
}
}
Constraint::TypeOf(_t) => {
*constraint = Constraint::SupertypeOf(l.clone());
},
_ => {}
},
_ => {}
}
return Ok(());
}
(Type::FreeVar(fv), r) if fv.is_unbound() => {
match &mut *fv.borrow_mut() {
FreeKind::NamedUnbound { constraint, .. }
| FreeKind::Unbound { constraint, .. } => match constraint {
Constraint::SupertypeOf(sub) => {
if !self.rec_full_subtype_of(sub, r) {
return Err(TyCheckError::subtyping_error(
sub,
r,
sub_loc,
sup_loc,
self.caused_by(),
))
} else {
*constraint = Constraint::sandwiched(sub.clone(), r.clone());
}
}
Constraint::SubtypeOf(sup)
if self.rec_full_supertype_of(sup, r) =>
{
*constraint = Constraint::SubtypeOf(r.clone());
}
Constraint::Sandwiched { sub, sup: _ }
if !self.rec_full_subtype_of(sub, r) =>
{
return Err(TyCheckError::subtyping_error(
sub,
r,
sub_loc,
sup_loc,
self.caused_by(),
))
}
Constraint::TypeOf(_t) => {
*constraint = Constraint::SubtypeOf(r.clone());
},
_ => {}
},
_ => {}
}
return Ok(());
},
(Type::FreeVar(_fv), _r) => todo!(),
(l @ Refinement(_), r @ Refinement(_)) => return self.unify(l, r, sub_loc, sup_loc),
_ => {}
}
let mut opt_smallest = None;
for ctx in self.rec_sorted_type_ctxs(maybe_sub) {
let bounds = ctx.type_params_bounds();
let variance = ctx.type_params_variance();
let instances = ctx
.super_classes
.iter()
.chain(ctx.super_traits.iter())
.filter(|t| self.formal_supertype_of(maybe_sup, t, Some(&bounds), Some(&variance)));
if let Some(t) = self.smallest_ref_t(instances) {
opt_smallest = if let Some(small) = opt_smallest {
self.min(small, t)
} else {
Some(t)
};
}
}
let glue_patch_and_types = self.rec_get_glue_patch_and_types();
let patch_instances = glue_patch_and_types
.iter()
.filter_map(|(patch_name, l, r)| {
let patch = self.rec_get_patch(patch_name).unwrap();
let bounds = patch.type_params_bounds();
let variance = patch.type_params_variance();
if self.formal_supertype_of(l, maybe_sub, Some(&bounds), Some(&variance))
&& self.formal_supertype_of(r, maybe_sup, Some(&bounds), Some(&variance))
{
let tv_ctx = TyVarContext::new(self.level, bounds, &self);
let (l, _) = Self::instantiate_t(l.clone(), tv_ctx.clone());
let (r, _) = Self::instantiate_t(r.clone(), tv_ctx);
Some((l, r))
} else {
None
}
});
let opt_smallest_pair = self.smallest_pair(patch_instances);
match (opt_smallest, opt_smallest_pair) {
(Some(smallest), Some((l, r))) => {
if self.min(smallest, &r) == Some(&r) {
self.unify(maybe_sub, &l, sub_loc, None)?;
self.unify(maybe_sup, &r, sup_loc, None)
} else {
self.unify(maybe_sup, smallest, sup_loc, None)
}
}
(Some(smallest), None) => self.unify(maybe_sup, smallest, sup_loc, None),
(None, Some((l, r))) => {
self.unify(maybe_sub, &l, sub_loc, None)?;
self.unify(maybe_sup, &r, sup_loc, None)?;
Ok(())
}
(None, None) => {
log!("{maybe_sub}, {maybe_sup}");
todo!()
}
}
}
}
impl Context {
fn validate_var_sig_t(
&self,
sig: &ast::VarSignature,
body_t: &Type,
mode: RegistrationMode,
) -> TyCheckResult<()> {
let spec_t = self.instantiate_var_sig_t(sig, None, mode)?;
match &sig.pat {
ast::VarPattern::Discard(token) => {
if self.unify(&spec_t, body_t, None, Some(sig.loc())).is_err() {
return Err(TyCheckError::type_mismatch_error(
token.loc(),
self.caused_by(),
"_",
&spec_t,
body_t,
));
}
}
ast::VarPattern::VarName(n) => {
if self.unify(&spec_t, body_t, None, Some(sig.loc())).is_err() {
return Err(TyCheckError::type_mismatch_error(
n.loc(),
self.caused_by(),
n.inspect(),
&spec_t,
body_t,
));
}
}
ast::VarPattern::Array(a) => {
for (elem, inf_elem_t) in a.iter().zip(body_t.inner_ts().iter()) {
self.validate_var_sig_t(elem, inf_elem_t, mode)?;
}
}
_ => todo!(),
}
Ok(())
}
pub(crate) fn instantiate_var_sig_t(
&self,
sig: &ast::VarSignature,
opt_t: Option<Type>,
mode: RegistrationMode,
) -> TyCheckResult<Type> {
let ty = if let Some(s) = sig.t_spec.as_ref() {
self.instantiate_typespec(s, mode)?
} else {
Type::free_var(self.level, Constraint::TypeOf(Type))
};
if let Some(t) = opt_t {
self.unify(&ty, &t, sig.t_spec.as_ref().map(|s| s.loc()), None)?;
}
Ok(ty)
}
pub(crate) fn instantiate_sub_sig_t(
&self,
sig: &ast::SubrSignature,
opt_ret_t: Option<Type>,
mode: RegistrationMode,
) -> TyCheckResult<Type> {
let non_defaults = sig
.params
.non_defaults
.iter()
.map(|p| {
ParamTy::new(
p.inspect().cloned(),
self.instantiate_param_sig_t(p, None, mode).unwrap(),
)
})
.collect::<Vec<_>>();
let defaults = sig
.params
.defaults
.iter()
.map(|p| {
ParamTy::new(
p.inspect().cloned(),
self.instantiate_param_sig_t(p, None, mode).unwrap(),
)
})
.collect::<Vec<_>>();
let return_t = if let Some(s) = sig.return_t_spec.as_ref() {
self.instantiate_typespec(s, mode)?
} else {
let level = if mode == PreRegister {
self.level
} else {
self.level + 1
};
Type::free_var(level, Constraint::TypeOf(Type))
};
if let Some(ret_t) = opt_ret_t {
self.unify(
&return_t,
&ret_t,
sig.return_t_spec.as_ref().map(|s| s.loc()),
None,
)?;
}
Ok(if sig.name.is_procedural() {
Type::proc(non_defaults, defaults, return_t)
} else {
Type::func(non_defaults, defaults, return_t)
})
}
pub(crate) fn instantiate_param_sig_t(
&self,
sig: &ParamSignature,
opt_decl_t: Option<&ParamTy>,
mode: RegistrationMode,
) -> TyCheckResult<Type> {
let t = if let Some(spec) = &sig.t_spec {
self.instantiate_typespec(spec, mode)?
} else {
match &sig.pat {
ast::ParamPattern::Lit(lit) => Type::enum_t(set![self.eval.eval_const_lit(&lit)]),
_ => {
let level = if mode == PreRegister {
self.level
} else {
self.level + 1
};
Type::free_var(level, Constraint::TypeOf(Type))
}
}
};
if let Some(decl_t) = opt_decl_t {
self.unify(&t, &decl_t.ty, sig.t_spec.as_ref().map(|s| s.loc()), None)?;
}
Ok(t)
}
pub(crate) fn instantiate_predecl_t(&self, _predecl: &PreDeclTypeSpec) -> TyCheckResult<Type> {
match _predecl {
ast::PreDeclTypeSpec::Simple(simple) => self.instantiate_simple_t(simple),
_ => todo!(),
}
}
pub(crate) fn instantiate_simple_t(&self, simple: &SimpleTypeSpec) -> TyCheckResult<Type> {
match &simple.name.inspect()[..] {
"Nat" => Ok(Type::Nat),
"Nat!" => Ok(Type::NatMut),
"Int" => Ok(Type::Int),
"Int!" => Ok(Type::IntMut),
"Ratio" => Ok(Type::Ratio),
"Ratio!" => Ok(Type::RatioMut),
"Float" => Ok(Type::Float),
"Float!" => Ok(Type::FloatMut),
"Str" => Ok(Type::Str),
"Str!" => Ok(Type::StrMut),
"Bool" => Ok(Type::Bool),
"Bool!" => Ok(Type::BoolMut),
"None" => Ok(Type::NoneType),
"Ellipsis" => Ok(Type::Ellipsis),
"NotImplemented" => Ok(Type::NotImplemented),
"Inf" => Ok(Type::Inf),
"Obj" => Ok(Type::Obj),
"Obj!" => Ok(Type::ObjMut),
"Array" => {
let mut args = simple.args.pos_args();
if let Some(first) = args.next() {
let t = self.instantiate_const_expr_as_type(&first.expr)?;
let len = args.next().unwrap();
let len = self.instantiate_const_expr(&len.expr);
Ok(Type::array(t, len))
} else {
Ok(Type::mono("GenericArray"))
}
}
other if simple.args.is_empty() => Ok(Type::mono(Str::rc(other))),
other => {
let params = simple.args.pos_args().map(|arg| match &arg.expr {
ast::ConstExpr::Lit(lit) => {
TyParam::ConstObj(ConstObj::Value(ValueObj::from(lit)))
}
_ => {
todo!()
}
});
Ok(Type::poly(Str::rc(other), params.collect()))
}
}
}
pub(crate) fn instantiate_const_expr(&self, expr: &ast::ConstExpr) -> TyParam {
match expr {
ast::ConstExpr::Lit(lit) => {
TyParam::ConstObj(ConstObj::Value(ValueObj::from(&lit.token)))
}
ast::ConstExpr::Accessor(ast::ConstAccessor::Local(name)) => {
TyParam::Mono(name.inspect().clone())
}
_ => todo!(),
}
}
pub(crate) fn instantiate_const_expr_as_type(
&self,
expr: &ast::ConstExpr,
) -> TyCheckResult<Type> {
match expr {
ast::ConstExpr::Accessor(ast::ConstAccessor::Local(name)) => {
Ok(Type::mono(name.inspect()))
}
_ => todo!(),
}
}
fn instantiate_func_param_spec(
&self,
p: &ParamTySpec,
mode: RegistrationMode,
) -> TyCheckResult<ParamTy> {
let t = self.instantiate_typespec(&p.ty, mode)?;
Ok(ParamTy::new(
p.name.as_ref().map(|t| t.inspect().to_owned()),
t,
))
}
pub(crate) fn instantiate_typespec(
&self,
spec: &TypeSpec,
mode: RegistrationMode,
) -> TyCheckResult<Type> {
match spec {
TypeSpec::PreDeclTy(predecl) => self.instantiate_predecl_t(predecl),
TypeSpec::And(lhs, rhs) => Ok(Type::And(vec![
self.instantiate_typespec(lhs, mode)?,
self.instantiate_typespec(rhs, mode)?,
])),
TypeSpec::Not(lhs, rhs) => Ok(Type::Not(vec![
self.instantiate_typespec(lhs, mode)?,
self.instantiate_typespec(rhs, mode)?,
])),
TypeSpec::Or(lhs, rhs) => Ok(Type::Or(vec![
self.instantiate_typespec(lhs, mode)?,
self.instantiate_typespec(rhs, mode)?,
])),
TypeSpec::Array { .. } => todo!(),
TypeSpec::Tuple(tys) => Ok(Type::tuple(
tys.iter()
.map(|spec| self.instantiate_typespec(spec, mode).unwrap())
.collect(),
)),
TypeSpec::Enum(set) => Ok(Type::enum_t(
set.pos_args()
.map(|arg| {
if let ast::ConstExpr::Lit(lit) = &arg.expr {
ValueObj::from(lit)
} else {
todo!()
}
})
.collect::<Set<_>>(),
)),
TypeSpec::Interval { op, lhs, rhs } => {
let op = match op.kind {
TokenKind::Closed => IntervalOp::Closed,
TokenKind::LeftOpen => IntervalOp::LeftOpen,
TokenKind::RightOpen => IntervalOp::RightOpen,
TokenKind::Open => IntervalOp::Open,
_ => assume_unreachable!(),
};
let l = self.instantiate_const_expr(lhs);
let l = self.eval.eval_tp(&l, self)?;
let r = self.instantiate_const_expr(rhs);
let r = self.eval.eval_tp(&r, self)?;
if let Some(Greater) = self.try_cmp(&l, &r, None) {
panic!("{l}..{r} is not a valid interval type (should be lhs <= rhs)")
}
Ok(Type::int_interval(op, l, r))
}
TypeSpec::Subr(subr) => {
let non_defaults = try_map(subr.non_defaults.iter(), |p| {
self.instantiate_func_param_spec(p, mode)
})?;
let defaults = try_map(subr.defaults.iter(), |p| {
self.instantiate_func_param_spec(p, mode)
})?;
let return_t = self.instantiate_typespec(&subr.return_t, mode)?;
Ok(Type::subr(
subr.kind.clone(),
non_defaults,
defaults,
return_t,
))
}
}
}
pub(crate) fn instantiate_ty_bound(
&self,
bound: &TypeBoundSpec,
mode: RegistrationMode,
) -> TyCheckResult<TyBound> {
match bound {
TypeBoundSpec::Subtype { sub, sup } => Ok(TyBound::subtype(
Type::mono_q(sub.inspect().clone()),
self.instantiate_typespec(sup, mode)?,
)),
TypeBoundSpec::Instance { name, ty } => Ok(TyBound::instance(
name.inspect().clone(),
self.instantiate_typespec(ty, mode)?,
)),
}
}
pub(crate) fn instantiate_ty_bounds(
&self,
bounds: &TypeBoundSpecs,
mode: RegistrationMode,
) -> TyCheckResult<Set<TyBound>> {
let mut new_bounds = set! {};
for bound in bounds.iter() {
new_bounds.insert(self.instantiate_ty_bound(bound, mode)?);
}
Ok(new_bounds)
}
pub(crate) fn get_current_scope_var(&self, name: &str) -> Option<&VarInfo> {
self.locals
.get(name)
.or_else(|| self.decls.get(name))
.or_else(|| {
self.params
.iter()
.find(|(opt_name, _)| {
opt_name
.as_ref()
.map(|n| &n.inspect()[..] == name)
.unwrap_or(false)
})
.map(|(_, vi)| vi)
})
}
fn get_context(
&self,
obj: &hir::Expr,
kind: Option<ContextKind>,
namespace: &Str,
) -> TyCheckResult<&Context> {
match obj {
hir::Expr::Accessor(hir::Accessor::Local(name)) => {
if kind == Some(ContextKind::Module) {
if let Some(ctx) = self.rec_get_mod(name.inspect()) {
Ok(ctx)
} else {
Err(TyCheckError::no_var_error(
obj.loc(),
namespace.clone(),
name.inspect(),
self.get_similar_name(name.inspect()),
))
}
} else {
todo!()
}
}
_ => todo!(),
}
}
fn get_match_call_t(
&self,
pos_args: &[hir::PosArg],
kw_args: &[hir::KwArg],
) -> TyCheckResult<Type> {
if !kw_args.is_empty() {
todo!()
}
for pos_arg in pos_args.iter().skip(1) {
let t = pos_arg.expr.ref_t();
if !matches!(&pos_arg.expr, hir::Expr::Lambda(_)) {
return Err(TyCheckError::type_mismatch_error(
pos_arg.loc(),
self.caused_by(),
"match",
&Type::mono("LambdaFunc"),
&t,
));
}
}
let expr_t = pos_args[0].expr.ref_t();
let mut union_pat_t = Type::Never;
for (i, a) in pos_args.iter().skip(1).enumerate() {
let lambda = erg_common::enum_unwrap!(&a.expr, hir::Expr::Lambda);
if !lambda.params.defaults.is_empty() {
todo!()
}
if lambda.params.len() != 1 {
return Err(TyCheckError::argument_error(
pos_args[i + 1].loc(),
self.caused_by(),
1,
pos_args[i + 1].expr.ref_t().typarams_len(),
));
}
let rhs = self.instantiate_param_sig_t(&lambda.params.non_defaults[0], None, Normal)?;
union_pat_t = self.union(&union_pat_t, &rhs);
}
if expr_t.has_no_unbound_var()
&& self.formal_supertype_of(&expr_t, &union_pat_t, None, None)
&& !self.formal_supertype_of(&union_pat_t, &expr_t, None, None)
{
return Err(TyCheckError::match_error(
pos_args[0].loc(),
self.caused_by(),
&expr_t,
));
}
let branch_ts = pos_args
.iter()
.skip(1)
.map(|a| ParamTy::anonymous(a.expr.ref_t().clone()))
.collect::<Vec<_>>();
let mut return_t = branch_ts[0].ty.return_t().unwrap().clone();
for arg_t in branch_ts.iter().skip(1) {
return_t = self.union(&return_t, arg_t.ty.return_t().unwrap());
}
let expr_t = if expr_t.has_unbound_var() {
union_pat_t
} else {
expr_t.clone()
};
let param_ts = [
vec![ParamTy::anonymous(expr_t)],
branch_ts.iter().map(|pt| pt.clone()).collect(),
]
.concat();
let t = Type::func(param_ts, vec![], return_t);
Ok(t)
}
pub(crate) fn get_local_uniq_obj_name(&self, name: &Token) -> Option<Str> {
if let Some(ctx) = self.rec_get_mod(name.inspect()) {
return Some(ctx.name.clone());
}
None
}
pub(crate) fn get_var_t(&self, name: &Token, namespace: &Str) -> TyCheckResult<Type> {
if let Some(vi) = self.get_current_scope_var(&name.inspect()[..]) {
Ok(vi.t())
} else {
if let Some(parent) = self.outer.as_ref() {
return parent.get_var_t(name, namespace);
}
Err(TyCheckError::no_var_error(
name.loc(),
namespace.clone(),
name.inspect(),
self.get_similar_name(name.inspect()),
))
}
}
pub(crate) fn get_attr_t(
&self,
obj: &hir::Expr,
name: &Token,
namespace: &Str,
) -> TyCheckResult<Type> {
let self_t = obj.t();
match self_t {
ASTOmitted => panic!(),
Type => todo!(),
Type::Record(_) => todo!(),
Module => {
let mod_ctx = self.get_context(obj, Some(ContextKind::Module), namespace)?;
let t = mod_ctx.get_var_t(name, namespace)?;
return Ok(t);
}
_ => {}
}
for ctx in self.rec_sorted_type_ctxs(&self_t) {
if let Ok(t) = ctx.get_var_t(name, namespace) {
return Ok(t);
}
}
if let Some(parent) = self.outer.as_ref() {
parent.get_attr_t(obj, name, namespace)
} else {
Err(TyCheckError::no_attr_error(
name.loc(),
namespace.clone(),
&self_t,
name.inspect(),
self.get_similar_attr(&self_t, name.inspect()),
))
}
}
fn search_call_t(&self, callee: &hir::Expr, namespace: &Str) -> TyCheckResult<Type> {
match callee {
hir::Expr::Accessor(hir::Accessor::Local(local)) => {
self.get_var_t(&local.name, namespace)
}
hir::Expr::Accessor(hir::Accessor::Attr(attr)) => {
self.get_attr_t(&attr.obj, &attr.name, namespace)
}
_ => todo!(),
}
}
pub(crate) fn get_binop_t(
&self,
op: &Token,
args: &[hir::PosArg],
namespace: &Str,
) -> TyCheckResult<Type> {
erg_common::debug_power_assert!(args.len() == 2);
let symbol = Token::symbol(binop_to_dname(op.inspect()));
let mut op = hir::Expr::Accessor(hir::Accessor::local(symbol, Type::ASTOmitted));
self.get_call_t(&mut op, args, &[], namespace).map_err(|e| {
let core = ErrorCore::new(
e.core.errno,
e.core.kind,
op.loc(),
e.core.desc,
e.core.hint,
);
TyCheckError::new(core, e.caused_by)
})
}
pub(crate) fn get_unaryop_t(
&self,
op: &Token,
args: &[hir::PosArg],
namespace: &Str,
) -> TyCheckResult<Type> {
erg_common::debug_power_assert!(args.len() == 1);
let symbol = Token::symbol(unaryop_to_dname(op.inspect()));
let mut op = hir::Expr::Accessor(hir::Accessor::local(symbol, Type::ASTOmitted));
self.get_call_t(&mut op, args, &[], namespace).map_err(|e| {
let core = ErrorCore::new(
e.core.errno,
e.core.kind,
op.loc(),
e.core.desc,
e.core.hint,
);
TyCheckError::new(core, e.caused_by)
})
}
pub(crate) fn get_call_t(
&self,
callee: &hir::Expr,
pos_args: &[hir::PosArg],
kw_args: &[hir::KwArg],
namespace: &Str,
) -> TyCheckResult<Type> {
match callee {
hir::Expr::Accessor(hir::Accessor::Local(local)) if &local.inspect()[..] == "match" => {
return self.get_match_call_t(pos_args, kw_args)
}
_ => {}
}
let found = self.search_call_t(callee, namespace)?;
log!("Found:\ncallee: {callee}\nfound: {found}");
let instance = self.instantiate(found, callee)?;
log!(
"Instantiated:\ninstance: {instance}\npos_args: ({})\nkw_args: ({})",
fmt_slice(pos_args),
fmt_slice(kw_args)
);
self.substitute_call(callee, &instance, pos_args, kw_args)?;
log!("Substituted:\ninstance: {instance}");
let res = self.deref_tyvar(instance)?;
log!("Derefed:\nres: {res}\n");
let res = self.eval.eval_t_params(res, &self, self.level)?;
log!("Params Evaluated:\nres: {res}\n");
let res = self.deref_tyvar(res)?;
log!("Derefed (2):\nres: {res}\n");
self.propagate(&res, callee)?;
log!("Propagated:\nres: {res}\n");
Ok(res)
}
fn eq_tp(
&self,
lhs: &TyParam,
rhs: &TyParam,
bounds: Option<&Set<TyBound>>,
lhs_variance: Option<&Vec<Variance>>,
) -> bool {
match (lhs, rhs) {
(TyParam::Type(lhs), TyParam::Type(rhs)) => {
return self.formal_same_type_of(lhs, rhs, bounds, lhs_variance)
}
(TyParam::Mono(l), TyParam::Mono(r)) => {
if let (Some((l, _)), Some((r, _))) = (
self.types.iter().find(|(t, _)| t.name() == &l[..]),
self.types.iter().find(|(t, _)| t.name() == &r[..]),
) {
return self.formal_supertype_of(l, r, bounds, None)
|| self.formal_subtype_of(l, r, bounds, lhs_variance);
}
}
(TyParam::MonoQVar(name), other) | (other, TyParam::MonoQVar(name)) => {
if let Some(bs) = bounds {
if let Some(bound) = bs.iter().find(|b| b.mentions_as_instance(name)) {
let other_t = self.type_of(other, bounds);
return self.formal_supertype_of(bound.t(), &other_t, bounds, lhs_variance);
} else {
todo!()
} }
}
(
TyParam::App {
name: ln,
args: largs,
},
TyParam::App {
name: rn,
args: rargs,
},
) => {
return ln == rn
&& largs.len() == rargs.len()
&& largs
.iter()
.zip(rargs.iter())
.all(|(l, r)| self.eq_tp(l, r, bounds, lhs_variance))
}
(TyParam::FreeVar(fv), other) | (other, TyParam::FreeVar(fv)) => match &*fv.borrow() {
FreeKind::Linked(tp) => return self.eq_tp(tp, other, bounds, lhs_variance),
FreeKind::Unbound { constraint, .. }
| FreeKind::NamedUnbound { constraint, .. } => {
let t = constraint.typ().unwrap();
let other_t = self.type_of(other, bounds);
return self.formal_supertype_of(&t, &other_t, bounds, lhs_variance);
}
},
(l, r) if l == r => return true,
_ => {}
}
self.eval.shallow_eq_tp(lhs, rhs, &self)
}
pub(crate) fn rec_full_supertype_of(&self, lhs: &Type, rhs: &Type) -> bool {
if self.full_supertype_of(lhs, rhs) {
return true;
}
if let Some(outer) = &self.outer {
if outer.rec_full_supertype_of(lhs, rhs) {
return true;
}
}
false
}
pub(crate) fn rec_full_subtype_of(&self, lhs: &Type, rhs: &Type) -> bool {
self.rec_full_supertype_of(rhs, lhs)
}
pub(crate) fn rec_full_no_relation_of(&self, lhs: &Type, rhs: &Type) -> bool {
!self.rec_full_supertype_of(lhs, rhs) && !self.rec_full_subtype_of(lhs, rhs)
}
pub(crate) fn _rec_full_same_type_of(&self, lhs: &Type, rhs: &Type) -> bool {
self.rec_full_supertype_of(lhs, rhs) && self.rec_full_subtype_of(lhs, rhs)
}
fn full_supertype_of(&self, lhs: &Type, rhs: &Type) -> bool {
if self.formal_supertype_of(lhs, rhs, None, None) {
return true;
}
for rhs_ctx in self.sorted_type_ctxs(rhs) {
let bounds = rhs_ctx.type_params_bounds();
let variance = rhs_ctx.type_params_variance();
if rhs_ctx
.super_classes
.iter()
.chain(rhs_ctx.super_traits.iter())
.any(|sup| self.formal_supertype_of(lhs, sup, Some(&bounds), Some(&variance)))
{
return true;
}
}
for (patch_name, sub, sup) in self.glue_patch_and_types.iter() {
let patch = self.rec_get_patch(patch_name).unwrap_or_else(|| panic!("{patch_name} not found"));
let bounds = patch.type_params_bounds();
let variance = patch.type_params_variance();
if self.formal_supertype_of(sub, rhs, Some(&bounds), Some(&variance))
&& self.formal_supertype_of(sup, lhs, Some(&bounds), Some(&variance))
{
return true;
}
}
false
}
fn formal_supertype_of(
&self,
lhs: &Type,
rhs: &Type,
bounds: Option<&Set<TyBound>>,
lhs_variance: Option<&Vec<Variance>>,
) -> bool {
if lhs.rec_eq(rhs) {
return true;
}
match (lhs, rhs) {
(Obj, _) | (_, Never) => true,
(_, Obj) | (Never, _) => false,
(Float | Ratio | Int | Nat | Bool, Bool)
| (Float | Ratio | Int | Nat, Nat)
| (Float | Ratio | Int, Int)
| (Float | Ratio, Ratio)
| (Float, Float) => true,
(
Type::Mono(n),
Subr(SubrType {
kind: SubrKind::Func,
..
}),
) if &n[..] == "GenericFunc" => true,
(
Type::Mono(n),
Subr(SubrType {
kind: SubrKind::Proc,
..
}),
) if &n[..] == "GenericProc" => true,
(
Type::Mono(n),
Subr(SubrType {
kind: SubrKind::FuncMethod(_),
..
}),
) if &n[..] == "GenericFuncMethod" => true,
(
Type::Mono(n),
Subr(SubrType {
kind: SubrKind::ProcMethod { .. },
..
}),
) if &n[..] == "GenericProcMethod" => true,
(Type::Mono(l), Type::Poly { name: r, .. })
if &l[..] == "GenericArray" && &r[..] == "Array" =>
{
true
}
(Type::Mono(l), Type::Poly { name: r, .. })
if &l[..] == "GenericDict" && &r[..] == "Dict" =>
{
true
}
(Type::Mono(l), Type::Mono(r))
if &l[..] == "GenericCallable"
&& (&r[..] == "GenericFunc"
|| &r[..] == "GenericProc"
|| &r[..] == "GenericFuncMethod"
|| &r[..] == "GenericProcMethod") =>
{
true
}
(Type::Mono(n), Subr(_)) if &n[..] == "GenericCallable" => true,
(Subr(ls), Subr(rs))
if ls.kind.same_kind_as(&rs.kind)
&& (ls.kind == SubrKind::Func || ls.kind == SubrKind::Proc) =>
{
ls.non_default_params.len() == rs.non_default_params.len()
&& ls.default_params.len() == rs.default_params.len()
&& self.formal_supertype_of(&ls.return_t, &rs.return_t, bounds, lhs_variance) && ls.non_default_params.iter()
.zip(rs.non_default_params.iter())
.all(|(l, r)| self.formal_subtype_of(&l.ty, &r.ty, bounds, lhs_variance))
&& ls.default_params.iter()
.zip(rs.default_params.iter())
.all(|(l, r)| self.formal_subtype_of(&l.ty, &r.ty, bounds, lhs_variance))
}
(Ref(lhs), Ref(rhs)) | (VarArgs(lhs), VarArgs(rhs)) => {
self.formal_supertype_of(lhs, rhs, bounds, lhs_variance)
}
(FreeVar(v), rhs) => {
match &*v.borrow() {
FreeKind::Linked(t) => self.formal_supertype_of(t, rhs, bounds, lhs_variance),
FreeKind::Unbound { constraint, .. }
| FreeKind::NamedUnbound { constraint, .. } => match constraint {
Constraint::SubtypeOf(sup) => {
self.formal_supertype_of(sup, rhs, bounds, lhs_variance)
}
Constraint::SupertypeOf(_sub) => {
true
},
Constraint::Sandwiched { sup, .. } => {
self.formal_supertype_of(sup, rhs, bounds, lhs_variance)
}
Constraint::TypeOf(t) => {
if self.formal_supertype_of(&Type, t, bounds, lhs_variance) {
true
} else {
panic!()
}
}
Constraint::Uninited => unreachable!(),
},
}
}
(lhs, FreeVar(fv)) => {
match &*fv.borrow() {
FreeKind::Linked(t) => self.formal_supertype_of(lhs, t, bounds, lhs_variance),
FreeKind::Unbound { constraint, .. }
| FreeKind::NamedUnbound { constraint, .. } => match constraint {
Constraint::SubtypeOf(sup) => {
self.formal_supertype_of(lhs, sup, bounds, lhs_variance)
|| self.formal_subtype_of(lhs, sup, bounds, lhs_variance)
},
Constraint::SupertypeOf(sub) => {
self.formal_supertype_of(lhs, sub, bounds, lhs_variance)
}
Constraint::Sandwiched { sub, sup: _ } => {
self.formal_supertype_of(lhs, sub, bounds, lhs_variance)
}
Constraint::TypeOf(t) => {
if self.formal_supertype_of(&Type, t, bounds, lhs_variance) {
true
} else {
panic!()
}
}
Constraint::Uninited => unreachable!(),
},
}
}
(Refinement(l), Refinement(r)) => {
if !self.formal_supertype_of(&l.t, &r.t, bounds, lhs_variance) {
return false;
}
let mut r_preds_clone = r.preds.clone();
for l_pred in l.preds.iter() {
for r_pred in r.preds.iter() {
if l_pred.subject().unwrap_or("") == &l.var[..]
&& r_pred.subject().unwrap_or("") == &r.var[..]
&& self.is_super_pred_of(l_pred, r_pred, bounds)
{
r_preds_clone.remove(r_pred);
}
}
}
r_preds_clone.is_empty()
}
(Nat, re @ Refinement(_)) => {
let nat = Type::Refinement(self.into_refinement(Nat));
self.formal_supertype_of(&nat, re, bounds, lhs_variance)
}
(re @ Refinement(_), Nat) => {
let nat = Type::Refinement(self.into_refinement(Nat));
self.formal_supertype_of(re, &nat, bounds, lhs_variance)
}
(l, Refinement(r)) => {
self.formal_supertype_of(l, &r.t, bounds, lhs_variance)
},
(Refinement(l), r) => {
if l.preds
.iter()
.any(|p| p.mentions(&l.var) && p.can_be_false())
{
return false;
}
self.formal_supertype_of(&l.t, r, bounds, lhs_variance)
}
(Quantified(l), Quantified(r)) => {
if bounds.is_some() {
panic!("Nested quantification")
} else {
self.formal_supertype_of(
l.unbound_callable.as_ref(),
r.unbound_callable.as_ref(),
Some(&l.bounds),
lhs_variance,
)
}
}
(Quantified(q), r) => {
if bounds.is_some() {
panic!("Nested quantification")
} else {
self.formal_supertype_of(
q.unbound_callable.as_ref(),
r,
Some(&q.bounds),
lhs_variance,
)
}
}
(lhs, Or(tys)) => tys
.iter()
.all(|t| self.formal_supertype_of(lhs, t, bounds, lhs_variance)),
(And(tys), rhs) => tys
.iter()
.all(|t| self.formal_supertype_of(t, rhs, bounds, lhs_variance)),
(VarArgs(lhs), rhs) => self.formal_supertype_of(lhs, rhs, bounds, lhs_variance),
(Ref(lhs), rhs) | (RefMut(lhs), rhs) => {
self.formal_supertype_of(lhs, rhs, bounds, lhs_variance)
}
(
Poly {
name: ln,
params: lp,
},
Poly {
name: rn,
params: rp,
},
) => {
if let Some(lhs_variance) = lhs_variance {
ln == rn
&& lp.len() == rp.len()
&& lp.iter().zip(rp.iter()).zip(lhs_variance.iter()).all(
|((l, r), variance)| match (l, r, variance) {
(TyParam::Type(l), TyParam::Type(r), Variance::Contravariant) => {
self.formal_subtype_of(l, r, bounds, Some(lhs_variance))
}
(TyParam::Type(l), TyParam::Type(r), Variance::Covariant) => {
self.formal_supertype_of(l, r, bounds, Some(lhs_variance))
}
_ => self.eq_tp(l, r, bounds, Some(lhs_variance)),
},
)
} else {
ln == rn
&& lp.len() == rp.len()
&& lp
.iter()
.zip(rp.iter())
.all(|(l, r)| self.eq_tp(l, r, bounds, None))
}
}
(MonoQVar(name), r) => {
if let Some(bs) = bounds {
if let Some(bound) = bs.iter().find(|b| b.mentions_as_subtype(name)) {
self.formal_supertype_of(bound.t(), r, bounds, lhs_variance)
} else if let Some(bound) = bs.iter().find(|b| b.mentions_as_instance(name)) {
if self.formal_same_type_of(bound.t(), &Type::Type, bounds, lhs_variance) {
true
} else {
todo!()
}
} else {
panic!("Unbound type variable: {name}")
}
} else {
panic!("No quantification")
}
}
(_l, MonoQVar(_name)) => {
if let Some(_bounds) = bounds {
todo!()
} else {
todo!()
}
}
(PolyQVar { .. }, _r) => todo!(),
(_l, PolyQVar { .. }) => todo!(),
(_l, _r) => false,
}
}
pub(crate) fn formal_subtype_of(
&self,
lhs: &Type,
rhs: &Type,
bounds: Option<&Set<TyBound>>,
lhs_variance: Option<&Vec<Variance>>,
) -> bool {
self.formal_supertype_of(rhs, lhs, bounds, lhs_variance)
}
pub(crate) fn formal_same_type_of(
&self,
lhs: &Type,
rhs: &Type,
bounds: Option<&Set<TyBound>>,
lhs_variance: Option<&Vec<Variance>>,
) -> bool {
self.formal_supertype_of(lhs, rhs, bounds, lhs_variance)
&& self.formal_subtype_of(lhs, rhs, bounds, lhs_variance)
}
fn try_cmp(
&self,
l: &TyParam,
r: &TyParam,
bounds: Option<&Set<TyBound>>,
) -> Option<TyParamOrdering> {
match (l, r) {
(TyParam::ConstObj(l), TyParam::ConstObj(r)) =>
l.try_cmp(r).map(Into::into),
(TyParam::BinOp{ op, lhs, rhs }, r) => {
if let Ok(l) = self.eval.eval_bin_tp(*op, lhs, rhs) {
self.try_cmp(&l, r, bounds)
} else { Some(Any) }
},
(TyParam::FreeVar(fv), p) if fv.is_linked() => {
self.try_cmp(&*fv.crack(), p, bounds)
}
(p, TyParam::FreeVar(fv)) if fv.is_linked() => {
self.try_cmp(p, &*fv.crack(), bounds)
}
(
l @ (TyParam::FreeVar(_) | TyParam::Erased(_) | TyParam::MonoQVar(_)),
r @ (TyParam::FreeVar(_) | TyParam::Erased(_) | TyParam::MonoQVar(_)),
) => {
let l_t = self.eval.get_tp_t(l, bounds, self).unwrap();
let r_t = self.eval.get_tp_t(r, bounds, self).unwrap();
if self.rec_full_supertype_of(&l_t, &r_t) || self.rec_full_subtype_of(&l_t, &r_t) {
Some(Any)
} else { Some(NotEqual) }
},
(l @ (TyParam::Erased(_) | TyParam::FreeVar(_) | TyParam::MonoQVar(_)), p) => {
let t = self.eval.get_tp_t(l, bounds, &self).unwrap();
let inf = self.inf(&t);
let sup = self.sup(&t);
if let (Some(inf), Some(sup)) = (inf, sup) {
match (
self.try_cmp(&inf, p, bounds).unwrap(),
self.try_cmp(&sup, p, bounds).unwrap()
) {
(Less, Less) => Some(Less),
(Less, Equal) => Some(LessEqual),
(Less, LessEqual) => Some(LessEqual),
(Less, NotEqual) => Some(NotEqual),
(Less, Greater | GreaterEqual | Any) => Some(Any),
(Equal, Less) => assume_unreachable!(),
(Equal, Equal) => Some(Equal),
(Equal, Greater) => Some(GreaterEqual),
(Equal, LessEqual) => Some(Equal),
(Equal, NotEqual) => Some(GreaterEqual),
(Equal, GreaterEqual | Any) => Some(GreaterEqual),
(Greater, Less) => assume_unreachable!(),
(Greater, Equal) => assume_unreachable!(),
(Greater, Greater | NotEqual | GreaterEqual | Any) => Some(Greater),
(Greater, LessEqual) => assume_unreachable!(),
(LessEqual, Less) => assume_unreachable!(),
(LessEqual, Equal | LessEqual) => Some(LessEqual),
(LessEqual, Greater | NotEqual | GreaterEqual | Any) => Some(Any),
(NotEqual, Less) => Some(Less),
(NotEqual, Equal | LessEqual) => Some(LessEqual),
(NotEqual, Greater | GreaterEqual | Any) => Some(Any),
(NotEqual, NotEqual) => Some(NotEqual),
(GreaterEqual, Less) => assume_unreachable!(),
(GreaterEqual, Equal | LessEqual) => Some(Equal),
(GreaterEqual, Greater | NotEqual | GreaterEqual | Any) => Some(GreaterEqual),
(Any, Less) => Some(Less),
(Any, Equal | LessEqual) => Some(LessEqual),
(Any, Greater | NotEqual | GreaterEqual | Any) => Some(Any),
(l, r) =>
todo!("cmp({inf}, {sup}) = {l:?}, cmp({inf}, {sup}) = {r:?}"),
}
} else { None }
}
(l, r @ (TyParam::Erased(_) | TyParam::MonoQVar(_) | TyParam::FreeVar(_))) =>
self.try_cmp(r, l, bounds).map(|ord| ord.reverse()),
(_l, _r) => {
erg_common::fmt_dbg!(_l, _r,);
None
},
}
}
fn into_refinement(&self, t: Type) -> RefinementType {
match t {
Nat => {
let var = Str::from(fresh_varname());
RefinementType::new(
var.clone(),
Int,
set! {Predicate::ge(var, TyParam::value(0))},
)
}
Refinement(r) => r,
t => {
let var = Str::from(fresh_varname());
RefinementType::new(var.clone(), t, set! {})
}
}
}
fn union(&self, lhs: &Type, rhs: &Type) -> Type {
match (
self.rec_full_supertype_of(lhs, rhs),
self.rec_full_subtype_of(lhs, rhs),
) {
(true, true) => return lhs.clone(), (true, false) => return lhs.clone(), (false, true) => return rhs.clone(),
(false, false) => {}
}
match (lhs, rhs) {
(Refinement(l), Refinement(r)) => Type::Refinement(self.union_refinement(l, r)),
(Or(ts), t) | (t, Or(ts)) => Or([vec![t.clone()], ts.clone()].concat()),
(t, Type::Never) | (Type::Never, t) => t.clone(),
(t, Refinement(r)) | (Refinement(r), t) => {
let t = self.into_refinement(t.clone());
Type::Refinement(self.union_refinement(&t, r))
}
(l, r) => Type::Or(vec![l.clone(), r.clone()]),
}
}
fn union_refinement(&self, lhs: &RefinementType, rhs: &RefinementType) -> RefinementType {
if !self.formal_supertype_of(&lhs.t, &rhs.t, None, None)
&& !self.formal_subtype_of(&lhs.t, &rhs.t, None, None)
{
log!("{lhs}\n{rhs}");
todo!()
} else {
let name = lhs.var.clone();
let rhs_preds = rhs
.preds
.iter()
.map(|p| p.clone().change_subject_name(name.clone()))
.collect();
RefinementType::new(
lhs.var.clone(),
*lhs.t.clone(),
lhs.preds.clone().concat(rhs_preds),
)
}
}
fn is_super_pred_of(
&self,
lhs: &Predicate,
rhs: &Predicate,
bounds: Option<&Set<TyBound>>,
) -> bool {
match (lhs, rhs) {
(Pred::LessEqual { rhs, .. }, _) if !rhs.has_upper_bound() => true,
(Pred::GreaterEqual { rhs, .. }, _) if !rhs.has_lower_bound() => true,
(
Pred::Equal { .. },
Pred::GreaterEqual { .. } | Pred::LessEqual { .. } | Pred::NotEqual { .. },
)
| (Pred::LessEqual { .. }, Pred::GreaterEqual { .. })
| (Pred::GreaterEqual { .. }, Pred::LessEqual { .. })
| (Pred::NotEqual { .. }, Pred::Equal { .. }) => false,
(Pred::Equal { rhs, .. }, Pred::Equal { rhs: rhs2, .. })
| (Pred::NotEqual { rhs, .. }, Pred::NotEqual { rhs: rhs2, .. }) => {
self.try_cmp(rhs, rhs2, bounds).unwrap().is_eq()
}
(
Pred::GreaterEqual { rhs, .. },
Pred::GreaterEqual { rhs: rhs2, .. } | Pred::Equal { rhs: rhs2, .. },
) => self.try_cmp(rhs, rhs2, bounds).unwrap().is_le(),
(
Pred::LessEqual { rhs, .. },
Pred::LessEqual { rhs: rhs2, .. } | Pred::Equal { rhs: rhs2, .. },
) => self.try_cmp(rhs, rhs2, bounds).unwrap().is_ge(),
(lhs @ (Pred::GreaterEqual { .. } | Pred::LessEqual { .. }), Pred::And(l, r)) => {
self.is_super_pred_of(lhs, l, bounds) || self.is_super_pred_of(lhs, r, bounds)
}
(lhs, Pred::Or(l, r)) => {
self.is_super_pred_of(lhs, l, bounds) && self.is_super_pred_of(lhs, r, bounds)
}
(Pred::Or(l, r), rhs @ (Pred::GreaterEqual { .. } | Pred::LessEqual { .. })) => {
self.is_super_pred_of(l, rhs, bounds) || self.is_super_pred_of(r, rhs, bounds)
}
(Pred::And(l, r), rhs) => {
self.is_super_pred_of(l, rhs, bounds) && self.is_super_pred_of(r, rhs, bounds)
}
(lhs, rhs) => todo!("{lhs}/{rhs}"),
}
}
fn is_sub_constraint_of(&self, l: &Constraint, r: &Constraint) -> bool {
match (l, r) {
(Constraint::SubtypeOf(lhs), Constraint::SubtypeOf(rhs))
| (Constraint::TypeOf(lhs), Constraint::TypeOf(rhs)) => {
self.rec_full_subtype_of(lhs, rhs)
}
(Constraint::SupertypeOf(lhs), Constraint::SupertypeOf(rhs)) => {
self.rec_full_supertype_of(lhs, rhs)
}
(Constraint::SubtypeOf(_), Constraint::TypeOf(Type)) => true,
(
Constraint::Sandwiched {
sub: lsub,
sup: lsup,
},
Constraint::Sandwiched {
sub: rsub,
sup: rsup,
},
) => self.rec_full_supertype_of(lsub, rsub) && self.rec_full_subtype_of(lsup, rsup),
_ => false,
}
}
#[inline]
fn type_of(&self, p: &TyParam, bounds: Option<&Set<TyBound>>) -> Type {
self.eval.get_tp_t(p, bounds, &self).unwrap()
}
fn sup(&self, t: &Type) -> Option<TyParam> {
match t {
Int | Nat | Float => Some(TyParam::value(Inf)),
Refinement(refine) => {
let mut maybe_max = None;
for pred in refine.preds.iter() {
match pred {
Pred::LessEqual { lhs, rhs } | Pred::Equal { lhs, rhs }
if lhs == &refine.var =>
{
if let Some(max) = &maybe_max {
if self.try_cmp(rhs, &max, None).unwrap() == Greater {
maybe_max = Some(rhs.clone());
}
} else {
maybe_max = Some(rhs.clone());
}
}
_ => {}
}
}
maybe_max
}
_other => None,
}
}
fn inf(&self, t: &Type) -> Option<TyParam> {
match t {
Int | Float => Some(TyParam::value(-Inf)),
Nat => Some(TyParam::value(0usize)),
Refinement(refine) => {
let mut maybe_min = None;
for pred in refine.preds.iter() {
match pred {
Predicate::GreaterEqual { lhs, rhs } | Predicate::Equal { lhs, rhs }
if lhs == &refine.var =>
{
if let Some(min) = &maybe_min {
if self.try_cmp(rhs, &min, None).unwrap() == Less {
maybe_min = Some(rhs.clone());
}
} else {
maybe_min = Some(rhs.clone());
}
}
_ => {}
}
}
maybe_min
}
_other => None,
}
}
fn min<'t>(&self, lhs: &'t Type, rhs: &'t Type) -> Option<&'t Type> {
match (
self.rec_full_supertype_of(lhs, rhs),
self.rec_full_subtype_of(lhs, rhs),
) {
(true, true) | (true, false) => Some(rhs),
(false, true) => Some(lhs),
(false, false) => None,
}
}
fn cmp_t<'t>(&self, lhs: &'t Type, rhs: &'t Type) -> TyParamOrdering {
match self.min(lhs, rhs) {
Some(l) if l == lhs => TyParamOrdering::Less,
Some(_) => TyParamOrdering::Greater,
None => TyParamOrdering::NoRelation,
}
}
fn smallest_pair<I: Iterator<Item = (Type, Type)>>(&self, ts: I) -> Option<(Type, Type)> {
ts.min_by(|(_, lhs), (_, rhs)| {
self.cmp_t(lhs, rhs).try_into().unwrap() })
}
fn smallest_ref_t<'t, I: Iterator<Item = &'t Type>>(&self, ts: I) -> Option<&'t Type> {
ts.min_by(|lhs, rhs| {
self.cmp_t(lhs, rhs).try_into().unwrap() })
}
pub(crate) fn get_local(&self, name: &Token, namespace: &Str) -> TyCheckResult<ConstObj> {
if let Some(obj) = self.consts.get(name.inspect()) {
Ok(obj.clone())
} else {
if let Some(parent) = self.outer.as_ref() {
return parent.get_local(name, namespace);
}
Err(TyCheckError::no_var_error(
name.loc(),
namespace.clone(),
name.inspect(),
self.get_similar_name(name.inspect()),
))
}
}
pub(crate) fn _get_attr(
&self,
obj: &hir::Expr,
name: &Token,
namespace: &Str,
) -> TyCheckResult<ConstObj> {
let self_t = obj.t();
for ctx in self.sorted_type_ctxs(&self_t) {
if let Ok(t) = ctx.get_local(name, namespace) {
return Ok(t);
}
}
if let Some(parent) = self.outer.as_ref() {
parent._get_attr(obj, name, namespace)
} else {
Err(TyCheckError::no_attr_error(
name.loc(),
namespace.clone(),
&self_t,
name.inspect(),
self.get_similar_attr(&self_t, name.inspect()),
))
}
}
pub(crate) fn get_similar_name(&self, name: &str) -> Option<&Str> {
if name.len() <= 1 {
return None;
}
let most_similar_name = self
.params
.iter()
.filter_map(|(opt_name, _)| opt_name.as_ref())
.chain(self.locals.keys())
.min_by_key(|v| levenshtein(v.inspect(), name))?
.inspect();
let len = most_similar_name.len();
if levenshtein(most_similar_name, name) >= len / 2 {
let outer = self.outer.as_ref()?;
outer.get_similar_name(name)
} else {
Some(most_similar_name)
}
}
pub(crate) fn get_similar_attr<'a>(&'a self, self_t: &'a Type, name: &str) -> Option<&'a Str> {
for ctx in self.rec_sorted_type_ctxs(self_t) {
if let Some(name) = ctx.get_similar_name(name) {
return Some(name);
}
}
None
}
}
impl Context {
pub(crate) fn grow(
&mut self,
name: &str,
kind: ContextKind,
vis: Visibility,
) -> TyCheckResult<()> {
let name = if vis.is_public() {
format!("{parent}.{name}", parent = self.name)
} else {
format!("{parent}::{name}", parent = self.name)
};
log!("{}: current namespace: {name}", fn_name!());
self.outer = Some(Box::new(mem::take(self)));
self.name = name.into();
self.kind = kind;
Ok(())
}
pub(crate) fn pop(&mut self) -> Result<(), TyCheckErrors> {
let mut uninited_errs = TyCheckErrors::empty();
for (name, vi) in self.decls.iter() {
uninited_errs.push(TyCheckError::uninitialized_error(
name.loc(),
self.caused_by(),
name.inspect(),
&vi.t,
));
}
if let Some(parent) = &mut self.outer {
*self = mem::take(parent);
log!("{}: current namespace: {}", fn_name!(), self.name);
if !uninited_errs.is_empty() {
Err(uninited_errs)
} else {
Ok(())
}
} else {
Err(TyCheckErrors::from(TyCheckError::checker_bug(
0,
Location::Unknown,
fn_name!(),
line!(),
)))
}
}
pub(crate) fn rec_sorted_type_ctxs<'a>(
&'a self,
t: &'a Type,
) -> impl Iterator<Item = &'a Context> {
let i = self.sorted_type_ctxs(t);
if i.size_hint().1 == Some(0) {
if let Some(outer) = &self.outer {
return outer.sorted_type_ctxs(t);
}
}
i
}
fn sorted_type_ctxs<'a>(&'a self, t: &'a Type) -> impl Iterator<Item = &'a Context> {
let mut ctxs = self._type_ctxs(t).collect::<Vec<_>>();
ctxs.sort_by(|(lhs, _), (rhs, _)| {
self.cmp_t(lhs, rhs).try_into().unwrap_or_else(|_| panic!("{lhs}, {rhs}")) });
ctxs.into_iter().map(|(_, ctx)| ctx)
}
fn _type_ctxs<'a>(&'a self, t: &'a Type) -> impl Iterator<Item = (&'a Type, &'a Context)> {
self.types.iter().filter_map(move |(maybe_sup, ctx)| {
let bounds = ctx.type_params_bounds();
let variance = ctx.type_params_variance();
if self.formal_supertype_of(maybe_sup, t, Some(&bounds), Some(&variance)) {
Some((maybe_sup, ctx))
} else {
None
}
})
}
fn rec_get_glue_patch_and_types(&self) -> Vec<(VarName, Type, Type)> {
if let Some(outer) = &self.outer {
[
&self.glue_patch_and_types[..],
&outer.rec_get_glue_patch_and_types(),
]
.concat()
} else {
self.glue_patch_and_types.clone()
}
}
fn rec_get_patch(&self, name: &VarName) -> Option<&Context> {
if let Some(patch) = self.patches.get(name) {
return Some(patch);
} else {
if let Some(outer) = &self.outer {
return outer.rec_get_patch(name);
}
}
None
}
fn rec_get_mod(&self, name: &str) -> Option<&Context> {
if let Some(mod_) = self.mods.get(name) {
return Some(mod_);
} else {
if let Some(outer) = &self.outer {
return outer.rec_get_mod(name);
}
}
None
}
pub(crate) fn rec_type_ctx_by_name<'a>(
&'a self,
t_name: &'a str,
) -> Option<&'a Context> {
if let Some((_, ctx)) = self.types
.iter()
.find(|(t, _ctx)| t.name() == t_name)
{
return Some(ctx);
}
if let Some(outer) = &self.outer {
outer.rec_type_ctx_by_name(t_name)
} else {
None
}
}
fn rec_get_const_param_defaults(&self, name: &str) -> Option<&Vec<ConstTemplate>> {
if let Some(impls) = self.const_param_defaults.get(name) {
return Some(impls)
}
if let Some(outer) = &self.outer {
outer.rec_get_const_param_defaults(name)
} else { None }
}
pub(crate) fn preregister(&mut self, block: &Vec<ast::Expr>) -> TyCheckResult<()> {
for expr in block.iter() {
match expr {
ast::Expr::Def(def) => {
let id = Some(def.body.id);
let eval_body_t = || {
self.eval
.eval_const_block(&def.body.block, &self)
.map(|c| Type::enum_t(set![c]))
};
match &def.sig {
ast::Signature::Subr(sig) => {
let opt_ret_t = if let Some(spec) = sig.return_t_spec.as_ref() {
Some(self.instantiate_typespec(spec, PreRegister)?)
} else {
eval_body_t()
};
self.declare_sub(&sig, opt_ret_t, id)?;
}
ast::Signature::Var(sig) if sig.is_const() => {
let t = if let Some(spec) = sig.t_spec.as_ref() {
Some(self.instantiate_typespec(spec, PreRegister)?)
} else {
eval_body_t()
};
self.declare_var(&sig, t, id)?;
}
_ => {}
}
}
_ => {}
}
}
Ok(())
}
}