use std::cmp::Ordering;
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::{
Constraint, FreeKind, HasLevel, IntervalOp, ParamTy, Predicate, RefinementType, SubrKind,
SubrType, TyBound, TyParam, TyParamOrdering, Type,
};
use erg_common::value::{ValueObj, Visibility};
use erg_common::Str;
use erg_common::{
assume_unreachable, enum_unwrap, fmt_option, 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};
use Mutability::*;
use Visibility::*;
type Trait = Type;
#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum TyParamIdx {
Nth(usize),
Nested { idx: usize, inner: Box<TyParamIdx> },
}
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() => {}
TyParam::Type(inner) => {
if let Some(inner) = Self::search(&inner, target) {
return Some(Self::nested(i, inner));
}
}
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!(),
}
}
pub fn nested(idx: usize, inner: Self) -> Self {
Self::Nested {
idx,
inner: Box::new(inner),
}
}
}
#[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 {
ValueObj::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!()
}
}
ValueObj::Type(t) => TyParam::t(*t.clone()),
v => TyParam::Value(v.clone()),
},
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::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) {
t.clone()
} else if let Some(t) = self.get_typaram(&n) {
if let TyParam::Type(t) = t {
*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::Value(_) => 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(ValueObj),
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, ValueObj>,
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 muty = Mutability::from(&sig.inspect().unwrap()[..]);
match &sig.pat {
ast::VarPattern::Ident(ident) => {
if sig.t_spec.is_none() && opt_t.is_none() {
Err(TyCheckError::no_type_spec_error(
line!() as usize,
sig.loc(),
self.caused_by(),
ident.inspect(),
))
} else {
if self.registered(ident.inspect(), ident.is_const()) {
return Err(TyCheckError::duplicate_decl_error(
line!() as usize,
sig.loc(),
self.caused_by(),
ident.inspect(),
));
}
let vis = ident.vis();
let kind = id.map_or(VarKind::Declared, VarKind::Defined);
let sig_t = self.instantiate_var_sig_t(sig, opt_t, PreRegister)?;
self.decls
.insert(ident.name.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.ident.inspect();
let vis = sig.ident.vis();
let muty = Mutability::from(&name[..]);
let kind = id.map_or(VarKind::Declared, VarKind::Defined);
if self.registered(name, name.is_uppercase()) {
return Err(TyCheckError::duplicate_decl_error(
line!() as usize,
sig.loc(),
self.caused_by(),
name,
));
}
let t = self.instantiate_sub_sig_t(sig, opt_ret_t, PreRegister)?;
let vi = VarInfo::new(t, muty, vis, kind);
if let Some(_decl) = self.decls.remove(name) {
return Err(TyCheckError::duplicate_decl_error(
line!() as usize,
sig.loc(),
self.caused_by(),
name,
));
} else {
self.decls.insert(sig.ident.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 muty = Mutability::from(&sig.inspect().unwrap()[..]);
let (generalized, bounds) = self.generalize_t(body_t.clone());
let generalized = if !bounds.is_empty() {
if self.rec_supertype_of(&Type::mono("GenericCallable"), &generalized) {
Type::quantified(generalized, bounds)
} else {
panic!()
}
} else {
generalized
};
match &sig.pat {
ast::VarPattern::Discard(_token) => Ok(()),
ast::VarPattern::Ident(ident) => {
if self.registered(ident.inspect(), ident.is_const()) {
Err(TyCheckError::reassign_error(
line!() as usize,
ident.loc(),
self.caused_by(),
ident.inspect(),
))
} else {
if self.decls.remove(ident.inspect()).is_some() {
}
let vis = ident.vis();
let vi = VarInfo::new(generalized, muty, vis, VarKind::Defined(id));
self.params.push((Some(ident.name.clone()), vi));
Ok(())
}
}
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(
line!() as usize,
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.ident.is_const() {
Mutability::Const
} else {
Mutability::Immutable
};
let name = &sig.ident.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(
line!() as usize,
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(
line!() as usize,
name.loc(),
self.caused_by(),
name.inspect(),
))
} else {
let sub_t = if sig.ident.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_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_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_supertype_of(&vi.t, &found_t) {
return Err(TyCheckError::violate_decl_error(
line!() as usize,
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_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(
line!() as usize,
mod_name.loc(),
self.caused_by(),
"import::name",
&Str,
mod_name.ref_t(),
));
}
}
_ => {
return Err(TyCheckError::feature_error(
line!() as usize,
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_of(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::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::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::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::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::Value(_) | 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_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(
line!() as usize,
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_else(|| "".to_string());
let name = name
+ "::"
+ param_ty
.name
.as_ref()
.map(|s| readable_name(&s[..]))
.unwrap_or("");
TyCheckError::type_mismatch_error(
line!() as usize,
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(
line!() as usize,
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(
line!() as usize,
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() => {
let inner = fv.unwrap_linked();
self.deref_tp(inner)
}
TyParam::FreeVar(_fv) if self.level == 0 => {
Err(TyCheckError::dummy_infer_error(fn_name!(), line!()))
}
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!(),
TyParam::MonoProj { .. }
| TyParam::MonoQVar(_)
| TyParam::PolyQVar { .. }
| TyParam::Failure
if self.level == 0 =>
{
Err(TyCheckError::dummy_infer_error(fn_name!(), line!()))
}
t => Ok(t),
}
}
fn deref_constraint(&self, constraint: Constraint) -> TyCheckResult<Constraint> {
match constraint {
Constraint::Sandwiched { sub, sup } => Ok(Constraint::sandwiched(
self.deref_tyvar(sub)?,
self.deref_tyvar(sup)?,
)),
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_sandwiched() => {
let constraint = fv.crack_constraint();
let (sub_t, super_t) = constraint.sub_sup_type().unwrap();
if self.rec_same_type_of(sub_t, super_t) {
self.unify(sub_t, super_t, None, None)?;
let t = if sub_t.rec_eq(&Never) {
super_t.clone()
} else {
sub_t.clone()
};
drop(constraint);
fv.link(&t);
self.deref_tyvar(Type::FreeVar(fv))
} else if self.level == 0 || self.level <= fv.level().unwrap() {
let new_t = if sub_t.rec_eq(&Never) {
super_t.clone()
} else {
sub_t.clone()
};
drop(constraint);
fv.link(&new_t);
self.deref_tyvar(Type::FreeVar(fv))
} else {
drop(constraint);
Ok(Type::FreeVar(fv))
}
}
Type::FreeVar(fv) if fv.is_unbound() => {
if self.level == 0 {
match &*fv.crack_constraint() {
Constraint::TypeOf(_) => {
Err(TyCheckError::dummy_infer_error(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))
}
Type::Ref(t) => {
let t = self.deref_tyvar(*t)?;
Ok(Type::ref_(t))
}
Type::RefMut(t) => {
let t = self.deref_tyvar(*t)?;
Ok(Type::ref_mut(t))
}
Type::VarArgs(t) => {
let t = self.deref_tyvar(*t)?;
Ok(Type::var_args(t))
}
Type::Callable { .. } => todo!(),
Type::Record(_) => todo!(),
Type::Refinement(refine) => {
let t = self.deref_tyvar(*refine.t)?;
Ok(Type::refinement(refine.var, t, refine.preds))
}
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(_) | hir::Accessor::Public(_) => {}
_ => todo!(),
}
Ok(())
}
hir::Expr::Array(array) => match array {
hir::Array::Normal(arr) => {
arr.t = self.deref_tyvar(mem::take(&mut arr.t))?;
for elem in arr.elems.pos_args.iter_mut() {
self.deref_expr_t(&mut elem.expr)?;
}
Ok(())
}
_ => 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)?;
}
for arg in call.args.kw_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<()> {
if let Type::Subr(SubrType {
kind: SubrKind::ProcMethod {
after: Some(after), ..
},
..
}) = t
{
let receiver_t = callee.receiver_t().unwrap();
self.reunify(receiver_t, after, Some(callee.loc()), None)?;
}
Ok(())
}
fn instantiate_trait(&self, generic: Type) -> Result<Type, Type> {
match generic {
Type::Poly { name, params } => {
let t_name = name.clone();
let t_params = params.clone();
let t = Type::Poly { name, params };
let mut min = Type::Never;
for (concrete_t, concrete_trait) in self.rec_get_poly_trait_impls(&t_name) {
if self.rec_supertype_of(&concrete_trait, &t) {
min = self.rec_min(&min, &concrete_t).unwrap_or(&min).clone();
}
}
if min == Type::Never {
let mut instantiated = false;
let mut new_params = Vec::with_capacity(t_params.len());
for param in t_params.into_iter() {
match param {
TyParam::Type(t) => match self.instantiate_trait(*t) {
Ok(concrete) => {
instantiated = true;
new_params.push(TyParam::t(concrete));
}
Err(param) => {
new_params.push(TyParam::t(param));
}
},
other => {
new_params.push(other);
}
}
}
if instantiated {
Ok(Type::poly(t_name, new_params))
} else {
Err(t)
}
} else {
Ok(min)
}
}
Type::Subr(subr) => {
let mut instantiated = false;
let mut new_non_default_params = Vec::with_capacity(subr.non_default_params.len());
for param in subr.non_default_params.into_iter() {
match self.instantiate_trait(param.ty) {
Ok(t) => {
instantiated = true;
new_non_default_params.push(ParamTy::new(param.name, t));
}
Err(other) => {
new_non_default_params.push(ParamTy::new(param.name, other));
}
}
}
let mut new_default_params = Vec::with_capacity(subr.default_params.len());
for param in subr.default_params.into_iter() {
match self.instantiate_trait(param.ty) {
Ok(t) => {
instantiated = true;
new_default_params.push(ParamTy::new(param.name, t));
}
Err(other) => {
new_default_params.push(ParamTy::new(param.name, other));
}
}
}
let new_return_t = match self.instantiate_trait(*subr.return_t) {
Ok(t) => {
instantiated = true;
t
}
Err(other) => other,
};
let t = Type::subr(
subr.kind,
new_non_default_params,
new_default_params,
new_return_t,
);
if instantiated {
Ok(t)
} else {
Err(t)
}
}
other => Err(other),
}
}
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_supertype_of(&fv_t, &tp_t) {
if fv.level() < Some(self.level) {
let new_constraint = Constraint::subtype_of(tp_t);
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_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::Value(ValueObj::Mut(l)), TyParam::Value(ValueObj::Mut(r))) => {
*l.borrow_mut() = r.borrow().clone();
Ok(())
}
(TyParam::Value(ValueObj::Mut(l)), TyParam::Value(r)) => {
*l.borrow_mut() = r.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(
line!() as usize,
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(
line!() as usize,
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(
line!() as usize,
l_pred,
r_pred,
self.caused_by(),
)),
},
_ => Err(TyCheckError::pred_unification_error(
line!() as usize,
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_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_subtype_of(sup, t) {
*sup = t.clone();
} else {
self.sub_unify(t, sup, rhs_loc, lhs_loc)?;
}
}
}
} let new_constraint = Constraint::subtype_of(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.structural_supertype_of(&l.t, &r.t, None, None)
&& !self.structural_supertype_of(&r.t, &l.t, None, None)
{
return Err(TyCheckError::unification_error(
line!() as usize,
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(
line!() as usize,
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(
line!() as usize,
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(
line!() as usize,
&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.structural_same_type_of(l, r, None, None) => Ok(()),
(l, r) => Err(TyCheckError::re_unification_error(
line!() as usize,
l,
r,
bef_loc,
aft_loc,
self.caused_by(),
)),
}
}
pub(crate) 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_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(
line!() as usize,
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::Sandwiched { sub, sup } => {
if !self.rec_supertype_of(sup, l) {
return Err(TyCheckError::subtyping_error(
line!() as usize,
l,
sup, sub_loc,
sup_loc,
self.caused_by(),
));
}
if let Some(new_sub) = self.rec_max(l, sub) {
*constraint =
Constraint::sandwiched(new_sub.clone(), mem::take(sup));
} else {
let new_sub = self.rec_union(l, sub);
*constraint = Constraint::sandwiched(new_sub, mem::take(sup));
}
}
Constraint::TypeOf(ty) => {
if self.rec_supertype_of(&Type, ty) {
*constraint = Constraint::supertype_of(l.clone());
} else {
todo!()
}
}
_ => unreachable!(),
},
_ => {}
}
return Ok(());
}
(Type::FreeVar(fv), r) if fv.is_unbound() => {
match &mut *fv.borrow_mut() {
FreeKind::NamedUnbound { constraint, .. }
| FreeKind::Unbound { constraint, .. } => match constraint {
Constraint::Sandwiched { sub, sup } => {
if !self.rec_subtype_of(sub, r) || !self.rec_supertype_of(sup, r) {
return Err(TyCheckError::subtyping_error(
line!() as usize,
sub,
r,
sub_loc,
sup_loc,
self.caused_by(),
));
}
if let Some(new_sup) = self.rec_min(sup, r) {
*constraint =
Constraint::sandwiched(mem::take(sub), new_sup.clone());
} else {
let new_sup = self.rec_union(sup, r);
*constraint = Constraint::sandwiched(mem::take(sub), new_sup);
}
}
Constraint::TypeOf(ty) => {
if self.rec_supertype_of(&Type, ty) {
*constraint = Constraint::subtype_of(r.clone());
} else {
todo!()
}
}
_ => unreachable!(),
},
_ => {}
}
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_sup_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.structural_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.rec_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.structural_supertype_of(l, maybe_sub, Some(&bounds), Some(&variance))
&& self.structural_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.rec_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(
line!() as usize,
token.loc(),
self.caused_by(),
"_",
&spec_t,
body_t,
));
}
}
ast::VarPattern::Ident(ident) => {
if self.unify(&spec_t, body_t, None, Some(sig.loc())).is_err() {
return Err(TyCheckError::type_mismatch_error(
line!() as usize,
ident.loc(),
self.caused_by(),
ident.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.ident.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),
"Int" => Ok(Type::Int),
"Ratio" => Ok(Type::Ratio),
"Float" => Ok(Type::Float),
"Str" => Ok(Type::Str),
"Bool" => Ok(Type::Bool),
"None" => Ok(Type::NoneType),
"Ellipsis" => Ok(Type::Ellipsis),
"NotImplemented" => Ok(Type::NotImplemented),
"Inf" => Ok(Type::Inf),
"Obj" => Ok(Type::Obj),
"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::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::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.rec_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_of(
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(
line!() as usize,
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(
line!() as usize,
pos_arg.loc(),
self.caused_by(),
"match",
&Type::mono("LambdaFunc"),
t,
));
}
}
let match_target_expr_t = pos_args[0].expr.ref_t();
let mut union_pat_t = Type::Never;
for (i, pos_arg) in pos_args.iter().skip(1).enumerate() {
let lambda = erg_common::enum_unwrap!(&pos_arg.expr, hir::Expr::Lambda);
if !lambda.params.defaults.is_empty() {
todo!()
}
if lambda.params.len() != 1 {
return Err(TyCheckError::argument_error(
line!() as usize,
pos_args[i + 1].loc(),
self.caused_by(),
1,
pos_args[i + 1].expr.signature_t().unwrap().typarams_len(),
));
}
let rhs = self.instantiate_param_sig_t(&lambda.params.non_defaults[0], None, Normal)?;
union_pat_t = self.rec_union(&union_pat_t, &rhs);
}
if self
.sub_unify(match_target_expr_t, &union_pat_t, None, None)
.is_err()
{
return Err(TyCheckError::match_error(
line!() as usize,
pos_args[0].loc(),
self.caused_by(),
match_target_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.rec_union(&return_t, arg_t.ty.return_t().unwrap());
}
let param_ty = ParamTy::anonymous(match_target_expr_t.clone());
let param_ts = [vec![param_ty], branch_ts.to_vec()].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,
vis: Visibility,
namespace: &Str,
) -> TyCheckResult<Type> {
if let Some(vi) = self.get_current_scope_var(&name.inspect()[..]) {
if vi.vis == vis {
Ok(vi.t())
} else {
Err(TyCheckError::visibility_error(
line!() as usize,
name.loc(),
namespace.clone(),
name.inspect(),
vi.vis,
))
}
} else {
if let Some(parent) = self.outer.as_ref() {
return parent.get_var_t(name, vis, namespace);
}
Err(TyCheckError::no_var_error(
line!() as usize,
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, Public, namespace)?;
return Ok(t);
}
_ => {}
}
for ctx in self.rec_sorted_sup_type_ctxs(&self_t) {
if let Ok(t) = ctx.get_var_t(name, Public, 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(
line!() as usize,
name.loc(),
namespace.clone(),
&self_t,
name.inspect(),
self.get_similar_attr(&self_t, name.inspect()),
))
}
}
fn search_callee_t(
&self,
obj: &hir::Expr,
method_name: &Option<Token>,
namespace: &Str,
) -> TyCheckResult<Type> {
if let Some(method_name) = method_name.as_ref() {
for ctx in self.rec_sorted_sup_type_ctxs(obj.ref_t()) {
if let Some(vi) = ctx.locals.get(method_name.inspect()) {
return Ok(vi.t());
} else if let Some(vi) = ctx.decls.get(method_name.inspect()) {
return Ok(vi.t());
}
}
Err(TyCheckError::no_attr_error(
line!() as usize,
method_name.loc(),
namespace.clone(),
obj.ref_t(),
method_name.inspect(),
self.get_similar_attr(obj.ref_t(), method_name.inspect()),
))
} else {
if obj.ref_t().rec_eq(&ASTOmitted) {
let local = enum_unwrap!(obj, hir::Expr::Accessor:(hir::Accessor::Local:(_)));
self.get_var_t(&local.name, Private, namespace)
} else {
Ok(obj.t())
}
}
}
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 op = hir::Expr::Accessor(hir::Accessor::local(symbol, Type::ASTOmitted));
self.get_call_t(&op, &None, 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 op = hir::Expr::Accessor(hir::Accessor::local(symbol, Type::ASTOmitted));
self.get_call_t(&op, &None, 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,
obj: &hir::Expr,
method_name: &Option<Token>,
pos_args: &[hir::PosArg],
kw_args: &[hir::KwArg],
namespace: &Str,
) -> TyCheckResult<Type> {
match obj {
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_callee_t(obj, method_name, namespace)?;
log!(
"Found:\ncallee: {obj}{}\nfound: {found}",
fmt_option!(pre ".", method_name.as_ref().map(|t| &t.content))
);
let instance = self.instantiate(found, obj)?;
log!(
"Instantiated:\ninstance: {instance}\npos_args: ({})\nkw_args: ({})",
fmt_slice(pos_args),
fmt_slice(kw_args)
);
self.substitute_call(obj, &instance, pos_args, kw_args)?;
log!("Substituted:\ninstance: {instance}");
let res = self.deref_tyvar(instance)?;
log!("Derefed:\nres: {res}\n");
let res = match self.instantiate_trait(res) {
Ok(t) => t,
Err(e) => e,
};
log!("Trait instantiated:\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, obj)?;
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.structural_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.structural_supertype_of(l, r, bounds, None)
|| self.structural_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.structural_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.structural_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_supertype_of(&self, lhs: &Type, rhs: &Type) -> bool {
if self.supertype_of(lhs, rhs) {
return true;
}
if let Some(outer) = &self.outer {
if outer.rec_supertype_of(lhs, rhs) {
return true;
}
}
false
}
pub(crate) fn rec_subtype_of(&self, lhs: &Type, rhs: &Type) -> bool {
self.rec_supertype_of(rhs, lhs)
}
pub(crate) fn rec_same_type_of(&self, lhs: &Type, rhs: &Type) -> bool {
self.rec_supertype_of(lhs, rhs) && self.rec_subtype_of(lhs, rhs)
}
pub(crate) fn _rec_related(&self, lhs: &Type, rhs: &Type) -> bool {
self.rec_supertype_of(lhs, rhs) || self.rec_subtype_of(lhs, rhs)
}
fn related(&self, lhs: &Type, rhs: &Type) -> bool {
self.supertype_of(lhs, rhs) || self.subtype_of(lhs, rhs)
}
fn supertype_of(&self, lhs: &Type, rhs: &Type) -> bool {
self.structural_supertype_of(lhs, rhs, None, None) || self.nominal_supertype_of(lhs, rhs)
}
fn subtype_of(&self, lhs: &Type, rhs: &Type) -> bool {
self.structural_subtype_of(lhs, rhs, None, None) || self.nominal_subtype_of(lhs, rhs)
}
fn nominal_supertype_of(&self, lhs: &Type, rhs: &Type) -> bool {
for rhs_ctx in self.sorted_sup_type_ctxs(rhs) {
let r_bounds = rhs_ctx.type_params_bounds();
let bounds = if lhs.is_monomorphic() {
r_bounds
} else {
if let Some((_, lhs_ctx)) = self._just_type_ctxs(lhs) {
lhs_ctx.type_params_bounds().concat(r_bounds)
} else {
r_bounds
}
};
let variance = rhs_ctx.type_params_variance();
if rhs_ctx
.super_classes
.iter()
.chain(rhs_ctx.super_traits.iter())
.any(|sup| self.structural_supertype_of(lhs, sup, Some(&bounds), Some(&variance)))
{
return true;
}
}
for (patch_name, sub_type, sup_trait) 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.structural_supertype_of(sub_type, rhs, Some(&bounds), Some(&variance))
&& self.structural_subtype_of(sup_trait, lhs, Some(&bounds), Some(&variance))
{
return true;
}
}
false
}
fn nominal_subtype_of(&self, lhs: &Type, rhs: &Type) -> bool {
self.nominal_supertype_of(rhs, lhs)
}
fn structural_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.structural_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.structural_subtype_of(&l.ty, &r.ty, bounds, lhs_variance))
&& ls.default_params.iter()
.zip(rs.default_params.iter())
.all(|(l, r)| self.structural_subtype_of(&l.ty, &r.ty, bounds, lhs_variance))
}
(Ref(lhs), Ref(rhs)) | (VarArgs(lhs), VarArgs(rhs)) => {
self.structural_supertype_of(lhs, rhs, bounds, lhs_variance)
}
(FreeVar(v), rhs) => {
match &*v.borrow() {
FreeKind::Linked(t) => {
self.structural_supertype_of(t, rhs, bounds, lhs_variance)
}
FreeKind::Unbound { constraint, .. }
| FreeKind::NamedUnbound { constraint, .. } => match constraint {
Constraint::Sandwiched { sup, .. } => {
self.structural_supertype_of(sup, rhs, bounds, lhs_variance)
}
Constraint::TypeOf(t) => {
if self.structural_supertype_of(&Type, t, bounds, lhs_variance) {
true
} else {
panic!()
}
}
Constraint::Uninited => unreachable!(),
},
}
}
(lhs, FreeVar(fv)) => {
match &*fv.borrow() {
FreeKind::Linked(t) => {
self.structural_supertype_of(lhs, t, bounds, lhs_variance)
}
FreeKind::Unbound { constraint, .. }
| FreeKind::NamedUnbound { constraint, .. } => match constraint {
Constraint::Sandwiched { sub, sup: _ } => {
self.structural_supertype_of(lhs, sub, bounds, lhs_variance)
}
Constraint::TypeOf(t) => {
if self.structural_supertype_of(&Type, t, bounds, lhs_variance) {
true
} else {
panic!()
}
}
Constraint::Uninited => unreachable!(),
},
}
}
(Refinement(l), Refinement(r)) => {
if !self.structural_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.rec_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.structural_supertype_of(&nat, re, bounds, lhs_variance)
}
(re @ Refinement(_), Nat) => {
let nat = Type::Refinement(self.into_refinement(Nat));
self.structural_supertype_of(re, &nat, bounds, lhs_variance)
}
(l, Refinement(r)) => self.structural_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.structural_supertype_of(&l.t, r, bounds, lhs_variance)
}
(Quantified(l), Quantified(r)) => {
if bounds.is_some() {
panic!("Nested quantification")
} else {
self.structural_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.structural_supertype_of(
q.unbound_callable.as_ref(),
r,
Some(&q.bounds),
lhs_variance,
)
}
}
(lhs, Or(tys)) => tys
.iter()
.all(|t| self.structural_supertype_of(lhs, t, bounds, lhs_variance)),
(And(tys), rhs) => tys
.iter()
.all(|t| self.structural_supertype_of(t, rhs, bounds, lhs_variance)),
(VarArgs(lhs), rhs) => self.structural_supertype_of(lhs, rhs, bounds, lhs_variance),
(Ref(lhs), rhs) | (RefMut(lhs), rhs) => {
self.structural_supertype_of(lhs, rhs, bounds, lhs_variance)
}
(
Poly {
name: ln,
params: lps,
},
Poly {
name: rn,
params: rps,
},
) => {
if let Some(lhs_variance) = lhs_variance {
ln == rn
&& lps.len() == rps.len()
&& lps.iter().zip(rps.iter()).zip(lhs_variance.iter()).all(
|((lp, rp), variance)| match (lp, rp, variance) {
(TyParam::Type(l), TyParam::Type(r), Variance::Contravariant) => {
self.structural_subtype_of(l, r, bounds, Some(lhs_variance))
}
(TyParam::Type(l), TyParam::Type(r), Variance::Covariant) => {
log!("{l}, {r}");
self.structural_supertype_of(l, r, bounds, Some(lhs_variance))
}
_ => self.eq_tp(lp, rp, bounds, Some(lhs_variance)),
},
)
} else {
ln == rn
&& lps.len() == rps.len()
&& lps
.iter()
.zip(rps.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.structural_supertype_of(bound.t(), r, bounds, lhs_variance)
} else if let Some(bound) = bs.iter().find(|b| b.mentions_as_instance(name)) {
if self.structural_same_type_of(
bound.t(),
&Type::Type,
bounds,
lhs_variance,
) {
true
} else {
todo!()
}
} else {
log!("bs: {bs}\nname: {name}, r: {r}");
panic!("Unbound type variable: {name}")
}
} else {
panic!("No quantification")
}
}
(l, MonoQVar(name)) => {
if let Some(bs) = bounds {
if let Some(bound) = bs.iter().find(|b| b.mentions_as_subtype(name)) {
self.structural_supertype_of(l, bound.t(), bounds, lhs_variance)
} else if let Some(bound) = bs.iter().find(|b| b.mentions_as_instance(name)) {
if self.structural_same_type_of(
bound.t(),
&Type::Type,
bounds,
lhs_variance,
) {
true
} else {
todo!()
}
} else {
log!("bs: {bs}\nl: {l}, name: {name}");
panic!("Unbound type variable: {name}")
}
} else {
panic!("No quantification")
}
}
(PolyQVar { .. }, _r) => todo!(),
(_l, PolyQVar { .. }) => todo!(),
(_l, _r) => false,
}
}
pub(crate) fn structural_subtype_of(
&self,
lhs: &Type,
rhs: &Type,
bounds: Option<&Set<TyBound>>,
lhs_variance: Option<&Vec<Variance>>,
) -> bool {
self.structural_supertype_of(rhs, lhs, bounds, lhs_variance)
}
pub(crate) fn structural_same_type_of(
&self,
lhs: &Type,
rhs: &Type,
bounds: Option<&Set<TyBound>>,
lhs_variance: Option<&Vec<Variance>>,
) -> bool {
self.structural_supertype_of(lhs, rhs, bounds, lhs_variance)
&& self.structural_subtype_of(lhs, rhs, bounds, lhs_variance)
}
fn rec_try_cmp(
&self,
l: &TyParam,
r: &TyParam,
bounds: Option<&Set<TyBound>>,
) -> Option<TyParamOrdering> {
match (l, r) {
(TyParam::Value(l), TyParam::Value(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.rec_try_cmp(&l, r, bounds)
} else { Some(Any) }
},
(TyParam::FreeVar(fv), p) if fv.is_linked() => {
self.rec_try_cmp(&*fv.crack(), p, bounds)
}
(p, TyParam::FreeVar(fv)) if fv.is_linked() => {
self.rec_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_supertype_of(&l_t, &r_t) || self.rec_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.rec_inf(&t);
let sup = self.rec_sup(&t);
if let (Some(inf), Some(sup)) = (inf, sup) {
match (
self.rec_try_cmp(&inf, p, bounds).unwrap(),
self.rec_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.rec_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, t, set! {})
}
}
}
fn rec_union(&self, lhs: &Type, rhs: &Type) -> Type {
match (
self.rec_supertype_of(lhs, rhs),
self.rec_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.structural_supertype_of(&lhs.t, &rhs.t, None, None)
&& !self.structural_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 rec_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.rec_try_cmp(rhs, rhs2, bounds).unwrap().is_eq()
}
(
Pred::GreaterEqual { rhs, .. },
Pred::GreaterEqual { rhs: rhs2, .. } | Pred::Equal { rhs: rhs2, .. },
) => self.rec_try_cmp(rhs, rhs2, bounds).unwrap().is_le(),
(
Pred::LessEqual { rhs, .. },
Pred::LessEqual { rhs: rhs2, .. } | Pred::Equal { rhs: rhs2, .. },
) => self.rec_try_cmp(rhs, rhs2, bounds).unwrap().is_ge(),
(lhs @ (Pred::GreaterEqual { .. } | Pred::LessEqual { .. }), Pred::And(l, r)) => {
self.rec_is_super_pred_of(lhs, l, bounds)
|| self.rec_is_super_pred_of(lhs, r, bounds)
}
(lhs, Pred::Or(l, r)) => {
self.rec_is_super_pred_of(lhs, l, bounds)
&& self.rec_is_super_pred_of(lhs, r, bounds)
}
(Pred::Or(l, r), rhs @ (Pred::GreaterEqual { .. } | Pred::LessEqual { .. })) => {
self.rec_is_super_pred_of(l, rhs, bounds)
|| self.rec_is_super_pred_of(r, rhs, bounds)
}
(Pred::And(l, r), rhs) => {
self.rec_is_super_pred_of(l, rhs, bounds)
&& self.rec_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::TypeOf(lhs), Constraint::TypeOf(rhs)) => self.rec_subtype_of(lhs, rhs),
(Constraint::Sandwiched { sub: Never, .. }, Constraint::TypeOf(Type)) => true,
(
Constraint::Sandwiched {
sub: lsub,
sup: lsup,
},
Constraint::Sandwiched {
sub: rsub,
sup: rsup,
},
) => self.rec_supertype_of(lsub, rsub) && self.rec_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 rec_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.rec_try_cmp(rhs, max, None).unwrap() == Greater {
maybe_max = Some(rhs.clone());
}
} else {
maybe_max = Some(rhs.clone());
}
}
_ => {}
}
}
maybe_max
}
_other => None,
}
}
fn rec_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.rec_try_cmp(rhs, min, None).unwrap() == Less {
maybe_min = Some(rhs.clone());
}
} else {
maybe_min = Some(rhs.clone());
}
}
_ => {}
}
}
maybe_min
}
_other => None,
}
}
fn rec_min<'t>(&self, lhs: &'t Type, rhs: &'t Type) -> Option<&'t Type> {
match (
self.rec_supertype_of(lhs, rhs),
self.rec_subtype_of(lhs, rhs),
) {
(true, true) | (true, false) => Some(rhs),
(false, true) => Some(lhs),
(false, false) => None,
}
}
fn rec_max<'t>(&self, lhs: &'t Type, rhs: &'t Type) -> Option<&'t Type> {
match (
self.rec_supertype_of(lhs, rhs),
self.rec_subtype_of(lhs, rhs),
) {
(true, true) | (true, false) => Some(lhs),
(false, true) => Some(rhs),
(false, false) => None,
}
}
fn _rec_cmp_t<'t>(&self, lhs: &'t Type, rhs: &'t Type) -> TyParamOrdering {
match self.rec_min(lhs, rhs) {
Some(l) if l == lhs => TyParamOrdering::Less,
Some(_) => TyParamOrdering::Greater,
None => TyParamOrdering::NoRelation,
}
}
fn min<'t>(&self, lhs: &'t Type, rhs: &'t Type) -> Option<&'t Type> {
match (self.supertype_of(lhs, rhs), self.subtype_of(lhs, rhs)) {
(true, true) | (true, false) => Some(rhs),
(false, true) => Some(lhs),
(false, false) => None,
}
}
fn _max<'t>(&self, lhs: &'t Type, rhs: &'t Type) -> Option<&'t Type> {
match (self.supertype_of(lhs, rhs), self.subtype_of(lhs, rhs)) {
(true, true) | (true, false) => Some(lhs),
(false, true) => Some(rhs),
(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, Trait)>>(
&self,
type_and_traits: I,
) -> Option<(Type, Trait)> {
let mut type_and_traits = type_and_traits.collect::<Vec<_>>();
let mut cheap_sort_succeed = true;
type_and_traits.sort_by(|(_, lhs), (_, rhs)| match self.cmp_t(lhs, rhs).try_into() {
Ok(ord) => ord,
Err(_) => {
cheap_sort_succeed = false;
Ordering::Equal
}
});
let mut sorted = if cheap_sort_succeed {
type_and_traits
} else {
self.sort_type_pairs(type_and_traits.into_iter())
};
if sorted.first().is_some() {
Some(sorted.remove(0))
} else {
None
}
}
fn smallest_ref_t<'t, I: Iterator<Item = &'t Type>>(&self, ts: I) -> Option<&'t Type> {
let mut ts = ts.collect::<Vec<_>>();
let mut cheap_sort_succeed = true;
ts.sort_by(|lhs, rhs| match self.cmp_t(lhs, rhs).try_into() {
Ok(ord) => ord,
Err(_) => {
cheap_sort_succeed = false;
Ordering::Equal
}
});
let mut sorted = if cheap_sort_succeed {
ts
} else {
self.sort_types(ts.into_iter())
};
if sorted.first().is_some() {
Some(sorted.remove(0))
} else {
None
}
}
pub(crate) fn get_local(&self, name: &Token, namespace: &Str) -> TyCheckResult<ValueObj> {
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(
line!() as usize,
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<ValueObj> {
let self_t = obj.t();
for ctx in self.sorted_sup_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(
line!() as usize,
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_sup_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(
line!() as usize,
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 fn sort_types<'a>(&self, types: impl Iterator<Item = &'a Type>) -> Vec<&'a Type> {
let mut buffers: Vec<Vec<&Type>> = vec![];
for t in types {
let mut found = false;
for buf in buffers.iter_mut() {
if buf.iter().all(|buf_inner| self.related(buf_inner, t)) {
found = true;
buf.push(t);
break;
}
}
if !found {
buffers.push(vec![t]);
}
}
for buf in buffers.iter_mut() {
buf.sort_by(|lhs, rhs| self.cmp_t(lhs, rhs).try_into().unwrap());
}
let mut concatenated = buffers.into_iter().flatten().collect::<Vec<_>>();
let mut idx = 0;
let len = concatenated.len();
while let Some(maybe_sup) = concatenated.get(idx) {
if let Some(pos) = concatenated
.iter()
.take(len - idx - 1)
.rposition(|t| self.supertype_of(maybe_sup, t))
{
let sup = concatenated.remove(idx);
concatenated.insert(pos, sup); }
idx += 1;
}
concatenated
}
fn sort_type_ctxs<'a>(
&self,
type_and_ctxs: impl Iterator<Item = (&'a Type, &'a Context)>,
) -> Vec<(&'a Type, &'a Context)> {
let mut buffers: Vec<Vec<(&Type, &Context)>> = vec![];
for t_ctx in type_and_ctxs {
let mut found = false;
for buf in buffers.iter_mut() {
if buf
.iter()
.all(|(buf_inner, _)| self.related(buf_inner, t_ctx.0))
{
found = true;
buf.push(t_ctx);
break;
}
}
if !found {
buffers.push(vec![t_ctx]);
}
}
for buf in buffers.iter_mut() {
buf.sort_by(|(lhs, _), (rhs, _)| self.cmp_t(lhs, rhs).try_into().unwrap());
}
let mut concatenated = buffers.into_iter().flatten().collect::<Vec<_>>();
let mut idx = 0;
let len = concatenated.len();
while let Some((maybe_sup, _)) = concatenated.get(idx) {
if let Some(pos) = concatenated
.iter()
.take(len - idx - 1)
.rposition(|(t, _)| self.supertype_of(maybe_sup, t))
{
let sup = concatenated.remove(idx);
concatenated.insert(pos, sup); }
idx += 1;
}
concatenated
}
fn sort_type_pairs(
&self,
type_and_traits: impl Iterator<Item = (Type, Trait)>,
) -> Vec<(Type, Trait)> {
let mut buffers: Vec<Vec<(Type, Trait)>> = vec![];
for t_trait in type_and_traits {
let mut found = false;
for buf in buffers.iter_mut() {
if buf
.iter()
.all(|(_, buf_inner)| self.related(buf_inner, &t_trait.0))
{
found = true;
buf.push(t_trait.clone());
break;
}
}
if !found {
buffers.push(vec![t_trait]);
}
}
for buf in buffers.iter_mut() {
buf.sort_by(|(_, lhs), (_, rhs)| self.cmp_t(lhs, rhs).try_into().unwrap());
}
let mut concatenated = buffers.into_iter().flatten().collect::<Vec<_>>();
let mut idx = 0;
let len = concatenated.len();
while let Some((_, maybe_sup)) = concatenated.get(idx) {
if let Some(pos) = concatenated
.iter()
.take(len - idx - 1)
.rposition(|(_, t)| self.supertype_of(maybe_sup, t))
{
let sup = concatenated.remove(idx);
concatenated.insert(pos, sup); }
idx += 1;
}
concatenated
}
pub(crate) fn rec_sorted_sup_type_ctxs<'a>(
&'a self,
t: &'a Type,
) -> impl Iterator<Item = &'a Context> {
let i = self.sorted_sup_type_ctxs(t);
if i.size_hint().1 == Some(0) {
if let Some(outer) = &self.outer {
return outer.sorted_sup_type_ctxs(t);
}
}
i
}
fn sorted_sup_type_ctxs<'a>(&'a self, t: &'a Type) -> impl Iterator<Item = &'a Context> {
let mut ctxs = self._sup_type_ctxs(t).collect::<Vec<_>>();
let mut cheap_sort_succeed = true;
ctxs.sort_by(|(lhs, _), (rhs, _)| match self.cmp_t(lhs, rhs).try_into() {
Ok(ord) => ord,
Err(_) => {
cheap_sort_succeed = false;
Ordering::Equal
}
});
let sorted = if cheap_sort_succeed {
ctxs
} else {
self.sort_type_ctxs(ctxs.into_iter())
};
sorted.into_iter().map(|(_, ctx)| ctx)
}
fn _just_type_ctxs<'a>(&'a self, t: &'a Type) -> Option<(&'a Type, &'a Context)> {
self.types.iter().find(move |(maybe_sup, ctx)| {
let bounds = ctx.type_params_bounds();
let variance = ctx.type_params_variance();
self.structural_same_type_of(maybe_sup, t, Some(&bounds), Some(&variance))
})
}
fn _sup_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.structural_supertype_of(maybe_sup, t, Some(&bounds), Some(&variance)) {
Some((maybe_sup, ctx))
} else {
None
}
})
}
fn rec_get_poly_trait_impls(&self, name: &Str) -> Vec<(Type, Type)> {
let current = if let Some(impls) = self.poly_trait_impls.get(name) {
impls.clone()
} else {
vec![]
};
if let Some(outer) = &self.outer {
[current, outer.rec_get_poly_trait_impls(name)].concat()
} else {
current
}
}
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) {
Some(patch)
} else if let Some(outer) = &self.outer {
outer.rec_get_patch(name)
} else {
None
}
}
fn rec_get_mod(&self, name: &str) -> Option<&Context> {
if let Some(mod_) = self.mods.get(name) {
Some(mod_)
} else if let Some(outer) = &self.outer {
outer.rec_get_mod(name)
} else {
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: &[ast::Expr]) -> TyCheckResult<()> {
for expr in block.iter() {
if let ast::Expr::Def(def) = expr {
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(())
}
}
impl Context {
pub fn test_refinement_subtyping(&self) -> Result<(), ()> {
let lhs = Nat;
let var = Str::ever("I");
let rhs = Type::refinement(
var.clone(),
Type::Nat,
set! { Predicate::eq(var, TyParam::value(1)) },
);
if self.rec_supertype_of(&lhs, &rhs) {
Ok(())
} else {
Err(())
}
}
}