use std::option::Option; // conflicting to Type::Option use erg_common::levenshtein::get_similar_name; #[allow(unused)] use erg_common::log; use erg_common::traits::{Locational, Stream}; use erg_common::{assume_unreachable, dict, failable_map_mut, set, Str}; use ast::{ NonDefaultParamSignature, ParamTySpec, PreDeclTypeSpec, TypeBoundSpec, TypeBoundSpecs, TypeSpec, }; use erg_parser::ast::{ self, ConstApp, ConstArgs, ConstArray, ConstExpr, ConstSet, Identifier, VarName, VisModifierSpec, VisRestriction, }; use erg_parser::token::TokenKind; use erg_parser::Parser; use crate::ty::free::{CanbeFree, Constraint, HasLevel}; use crate::ty::typaram::{IntervalOp, OpKind, TyParam, TyParamLambda, TyParamOrdering}; use crate::ty::value::ValueObj; use crate::ty::{ constructors::*, CastTarget, GuardType, Predicate, RefinementType, VisibilityModifier, }; use crate::ty::{Field, HasType, ParamTy, SubrKind, SubrType, Type}; use crate::type_feature_error; use crate::varinfo::{AbsLocation, VarInfo}; use TyParamOrdering::*; use Type::*; use crate::context::instantiate::TyVarCache; use crate::context::{Context, DefaultInfo, RegistrationMode}; use crate::error::{Failable, TyCheckError, TyCheckErrors, TyCheckResult}; use crate::AccessKind; use RegistrationMode::*; pub fn token_kind_to_op_kind(kind: TokenKind) -> Option { match kind { TokenKind::Plus => Some(OpKind::Add), TokenKind::Minus => Some(OpKind::Sub), TokenKind::Star => Some(OpKind::Mul), TokenKind::Slash => Some(OpKind::Div), TokenKind::FloorDiv => Some(OpKind::FloorDiv), TokenKind::Mod => Some(OpKind::Mod), TokenKind::Pow => Some(OpKind::Pow), TokenKind::PrePlus => Some(OpKind::Pos), TokenKind::PreMinus => Some(OpKind::Neg), TokenKind::PreBitNot => Some(OpKind::Invert), TokenKind::DblEq => Some(OpKind::Eq), TokenKind::NotEq => Some(OpKind::Ne), TokenKind::Less => Some(OpKind::Lt), TokenKind::LessEq => Some(OpKind::Le), TokenKind::Gre => Some(OpKind::Gt), TokenKind::GreEq => Some(OpKind::Ge), TokenKind::AndOp => Some(OpKind::And), TokenKind::OrOp => Some(OpKind::Or), TokenKind::BitAnd => Some(OpKind::BitAnd), TokenKind::BitOr => Some(OpKind::BitOr), TokenKind::BitXor => Some(OpKind::BitXor), TokenKind::Shl => Some(OpKind::Shl), TokenKind::Shr => Some(OpKind::Shr), _ => None, } } #[derive(Debug, Clone, PartialEq, Eq)] pub enum ParamKind { NonDefault, Default(Type), VarParams, KwVarParams, } impl ParamKind { pub const fn is_var_params(&self) -> bool { matches!(self, ParamKind::VarParams) } pub const fn is_kw_var_params(&self) -> bool { matches!(self, ParamKind::KwVarParams) } pub const fn is_default(&self) -> bool { matches!(self, ParamKind::Default(_)) } pub const fn default_info(&self) -> DefaultInfo { match self { ParamKind::Default(_) => DefaultInfo::WithDefault, _ => DefaultInfo::NonDefault, } } } /// TODO: this struct will be removed when const functions are implemented. #[derive(Debug, Clone, PartialEq, Eq, Hash)] pub enum ConstTemplate { Obj(ValueObj), App { name: Str, non_default_args: Vec, default_args: Vec, }, } impl ConstTemplate { pub const fn app( name: &'static str, non_default_args: Vec, default_args: Vec, ) -> Self { ConstTemplate::App { name: Str::ever(name), non_default_args, default_args, } } } impl Context { pub(crate) fn instantiate_ty_bound( &self, bound: &TypeBoundSpec, tv_cache: &mut TyVarCache, mode: RegistrationMode, ) -> TyCheckResult<()> { // REVIEW: 型境界の左辺に来れるのは型変数だけか? // TODO: 高階型変数 match bound { TypeBoundSpec::Omitted(name) => { // TODO: other than type `Type` let constr = Constraint::new_type_of(Type); let tv = named_free_var(name.inspect().clone(), self.level, constr); tv_cache.push_or_init_tyvar(name, &tv, self) } TypeBoundSpec::NonDefault { lhs, spec } => { let constr = match spec.op.kind { TokenKind::SubtypeOf => Constraint::new_subtype_of( self.instantiate_typespec_full(&spec.t_spec, None, tv_cache, mode, true)?, ), TokenKind::SupertypeOf => Constraint::new_supertype_of( self.instantiate_typespec_full(&spec.t_spec, None, tv_cache, mode, true)?, ), TokenKind::Colon => Constraint::new_type_of(self.instantiate_typespec_full( &spec.t_spec, None, tv_cache, mode, true, )?), _ => unreachable!(), }; if constr.get_sub_sup().is_none() { let tp = TyParam::named_free_var(lhs.inspect().clone(), self.level, constr); tv_cache.push_or_init_typaram(lhs, &tp, self)?; } else { let tv = named_free_var(lhs.inspect().clone(), self.level, constr); tv_cache.push_or_init_tyvar(lhs, &tv, self)?; } Ok(()) } TypeBoundSpec::WithDefault { .. } => type_feature_error!( self, bound.loc(), "type boundary specification with default" ), } } pub(crate) fn instantiate_ty_bounds( &self, bounds: &TypeBoundSpecs, mode: RegistrationMode, ) -> Failable { let mut errs = TyCheckErrors::empty(); let mut tv_cache = TyVarCache::new(self.level, self); for bound in bounds.iter() { if let Err(es) = self.instantiate_ty_bound(bound, &mut tv_cache, mode) { errs.extend(es); } } for tv in tv_cache.tyvar_instances.values() { if tv.constraint().map(|c| c.is_uninited()).unwrap_or(false) { errs.push(TyCheckError::no_var_error( self.cfg.input.clone(), line!() as usize, bounds.loc(), self.caused_by(), &tv.local_name(), self.get_similar_name(&tv.local_name()), )); } } for tp in tv_cache.typaram_instances.values() { if tp.constraint().map(|c| c.is_uninited()).unwrap_or(false) { errs.push(TyCheckError::no_var_error( self.cfg.input.clone(), line!() as usize, bounds.loc(), self.caused_by(), &tp.to_string(), self.get_similar_name(&tp.to_string()), )); } } if errs.is_empty() { Ok(tv_cache) } else { Err((tv_cache, errs)) } } pub(crate) fn instantiate_var_sig_t( &self, t_spec: Option<&TypeSpec>, mode: RegistrationMode, ) -> TyCheckResult { let mut tmp_tv_cache = TyVarCache::new(self.level, self); let spec_t = if let Some(t_spec) = t_spec { self.instantiate_typespec_full(t_spec, None, &mut tmp_tv_cache, mode, false)? } else { free_var(self.level, Constraint::new_type_of(Type)) }; Ok(spec_t) } pub(crate) fn instantiate_sub_sig_t( &self, sig: &ast::SubrSignature, default_ts: Vec, mode: RegistrationMode, ) -> Failable { let mut errs = TyCheckErrors::empty(); // -> Result { let opt_decl_sig_t = match self .rec_get_decl_info(&sig.ident, AccessKind::Name, &self.cfg.input, self) .ok() .map(|vi| vi.t) { Some(Type::Subr(subr)) => Some(subr), Some(Type::FreeVar(fv)) if fv.is_unbound() => return Ok(Type::FreeVar(fv)), Some(other) => { let err = TyCheckError::unreachable( self.cfg.input.clone(), "instantiate_sub_sig_t", line!(), ); return Err((other, TyCheckErrors::from(err))); } None => None, }; let mut tmp_tv_cache = match self.instantiate_ty_bounds(&sig.bounds, PreRegister) { Ok(tv_cache) => tv_cache, Err((tv_cache, es)) => { errs.extend(es); tv_cache } }; let mut non_defaults = vec![]; for (n, param) in sig.params.non_defaults.iter().enumerate() { let opt_decl_t = opt_decl_sig_t .as_ref() .and_then(|subr| subr.non_default_params.get(n)); match self.instantiate_param_ty( param, opt_decl_t, &mut tmp_tv_cache, mode, ParamKind::NonDefault, false, ) { Ok(pt) => non_defaults.push(pt), Err((pt, es)) => { errs.extend(es); non_defaults.push(pt); } } } let var_params = if let Some(var_args) = sig.params.var_params.as_ref() { let opt_decl_t = opt_decl_sig_t .as_ref() .and_then(|subr| subr.var_params.as_ref().map(|v| v.as_ref())); let pt = match self.instantiate_param_ty( var_args, opt_decl_t, &mut tmp_tv_cache, mode, ParamKind::VarParams, false, ) { Ok(pt) => pt, Err((pt, es)) => { errs.extend(es); pt } }; Some(pt) } else { None }; let mut defaults = vec![]; for ((n, p), default_t) in sig.params.defaults.iter().enumerate().zip(default_ts) { let opt_decl_t = opt_decl_sig_t .as_ref() .and_then(|subr| subr.default_params.get(n)); match self.instantiate_param_ty( &p.sig, opt_decl_t, &mut tmp_tv_cache, mode, ParamKind::Default(default_t), false, ) { Ok(pt) => defaults.push(pt), Err((pt, es)) => { errs.extend(es); defaults.push(pt); } } } let kw_var_params = if let Some(kw_var_args) = sig.params.kw_var_params.as_ref() { let opt_decl_t = opt_decl_sig_t .as_ref() .and_then(|subr| subr.kw_var_params.as_ref().map(|v| v.as_ref())); let pt = match self.instantiate_param_ty( kw_var_args, opt_decl_t, &mut tmp_tv_cache, mode, ParamKind::KwVarParams, false, ) { Ok(pt) => pt, Err((pt, es)) => { errs.extend(es); pt } }; Some(pt) } else { None }; let spec_return_t = if let Some(t_spec) = sig.return_t_spec.as_ref() { let opt_decl_t = opt_decl_sig_t .as_ref() .map(|subr| ParamTy::Pos(subr.return_t.as_ref().clone())); match self.instantiate_typespec_full( &t_spec.t_spec, opt_decl_t.as_ref(), &mut tmp_tv_cache, mode, false, ) { Ok(ty) => { let params = non_defaults .iter() .chain(&var_params) .chain(&defaults) .chain(&kw_var_params) .filter_map(|pt| pt.name()); self.recover_guard(ty, params) } Err(es) => { errs.extend(es); Type::Failure } } } else { // preregisterならouter scopeで型宣言(see inference.md) let level = if mode.is_preregister() { self.level } else { self.level + 1 }; free_var(level, Constraint::new_type_of(Type)) }; // tmp_tv_cache.warn_isolated_vars(self); let typ = if sig.ident.is_procedural() { proc( non_defaults, var_params, defaults, kw_var_params, spec_return_t, ) } else { func( non_defaults, var_params, defaults, kw_var_params, spec_return_t, ) }; if errs.is_empty() { Ok(typ) } else { Err((typ, errs)) } } /// spec_t == Noneかつリテラル推論が不可能なら型変数を発行する pub(crate) fn instantiate_param_sig_t( &self, sig: &NonDefaultParamSignature, opt_decl_t: Option<&ParamTy>, tmp_tv_cache: &mut TyVarCache, mode: RegistrationMode, kind: ParamKind, not_found_is_qvar: bool, ) -> Failable { let gen_free_t = || { let level = if mode.is_preregister() { self.level } else { self.level + 1 }; free_var(level, Constraint::new_type_of(Type)) }; let spec_t = if let Some(spec_with_op) = &sig.t_spec { self.instantiate_typespec_full( &spec_with_op.t_spec, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, ) .map_err(|errs| { ( opt_decl_t.map_or(Type::Failure, |pt| pt.typ().clone()), // Ignore errors if `mode == Normal`, because the errors have already been collected. if mode.is_normal() { TyCheckErrors::empty() } else { errs }, ) })? } else { match &sig.pat { ast::ParamPattern::Lit(lit) => { let lit = self.eval_lit(lit).map_err(|errs| (Type::Failure, errs))?; v_enum(set![lit]) } ast::ParamPattern::Discard(_) => Type::Obj, ast::ParamPattern::Ref(_) => ref_(gen_free_t()), ast::ParamPattern::RefMut(_) => ref_mut(gen_free_t(), None), // ast::ParamPattern::VarName(name) if &name.inspect()[..] == "_" => Type::Obj, // TODO: Array _ => gen_free_t(), } }; if let Some(decl_pt) = opt_decl_t { if kind.is_var_params() { let spec_t = unknown_len_array_t(spec_t.clone()); self.sub_unify( decl_pt.typ(), &spec_t, &sig.t_spec.as_ref().ok_or(sig), None, ) .map_err(|errs| (spec_t, errs))?; } else { self.sub_unify( decl_pt.typ(), &spec_t, &sig.t_spec.as_ref().ok_or(sig), None, ) .map_err(|errs| (spec_t.clone(), errs))?; } } Ok(spec_t) } /// Given the type `T -> U`, if `T` is a known type, then this is a function type that takes `T` and returns `U`. /// If the type `T` is not defined, then `T` is considered a constant parameter. /// FIXME: The type bounds are processed regardless of the order in the specification, but in the current implementation, undefined type may be considered a constant parameter. pub(crate) fn instantiate_param_ty( &self, sig: &NonDefaultParamSignature, opt_decl_t: Option<&ParamTy>, tmp_tv_cache: &mut TyVarCache, mode: RegistrationMode, kind: ParamKind, not_found_is_qvar: bool, ) -> Failable { if let Some(value) = sig .name() .and_then(|name| self.get_const_local(name.token(), &self.name).ok()) { return Ok(ParamTy::Pos(v_enum(set! { value }))); } else if let Some((tp, _vi)) = sig .name() .and_then(|name| self.get_tp_from_tv_cache(name.inspect(), tmp_tv_cache)) { match tp { TyParam::Type(t) => return Ok(ParamTy::Pos(*t)), other => { let (t, errs) = match self.get_tp_t(&other) { Ok(t) => (t, TyCheckErrors::empty()), Err(errs) => (Type::Failure, errs), }; let pt = ParamTy::Pos(tp_enum(t, set! { other })); if errs.is_empty() { return Ok(pt); } else { return Err((pt, errs)); } } } } let (t, errs) = match self.instantiate_param_sig_t( sig, opt_decl_t, tmp_tv_cache, mode, kind.clone(), not_found_is_qvar, ) { Ok(t) => (t, TyCheckErrors::empty()), Err((t, errs)) => (t, errs), }; let pt = match (sig.inspect(), kind) { (Some(name), ParamKind::Default(default_t)) => { ParamTy::kw_default(name.clone(), t, default_t) } (Some(name), _) => ParamTy::kw(name.clone(), t), (None, _) => ParamTy::Pos(t), }; if errs.is_empty() { Ok(pt) } else { Err((pt, errs)) } } pub(crate) fn instantiate_predecl_t( &self, predecl: &PreDeclTypeSpec, opt_decl_t: Option<&ParamTy>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> TyCheckResult { self.inc_ref_predecl_typespec(predecl, self, tmp_tv_cache); match predecl { ast::PreDeclTypeSpec::Mono(simple) => { self.instantiate_mono_t(simple, opt_decl_t, tmp_tv_cache, not_found_is_qvar) } ast::PreDeclTypeSpec::Poly(poly) => match &poly.acc { ast::ConstAccessor::Local(local) => self .instantiate_local_poly_t( local, &poly.args, opt_decl_t, tmp_tv_cache, not_found_is_qvar, ) .map_err(|(_, err)| err), ast::ConstAccessor::Attr(attr) => { let ctxs = self.get_singular_ctxs(&attr.obj.clone().downgrade(), self)?; for ctx in ctxs { if let Ok(typ) = ctx.instantiate_local_poly_t( &attr.name, &poly.args, opt_decl_t, tmp_tv_cache, not_found_is_qvar, ) { return Ok(typ); } } Err(TyCheckErrors::from(TyCheckError::no_var_error( self.cfg.input.clone(), line!() as usize, attr.loc(), self.caused_by(), attr.name.inspect(), self.get_similar_name(attr.name.inspect()), ))) } _ => type_feature_error!(self, poly.loc(), &format!("instantiating type {poly}")), }, ast::PreDeclTypeSpec::Attr { namespace, t } => { if let Ok(receiver) = Parser::validate_const_expr(namespace.as_ref().clone()) { if let Ok(receiver_t) = self.instantiate_const_expr_as_type( &receiver, None, tmp_tv_cache, not_found_is_qvar, ) { return self.eval_proj( receiver_t, t.inspect().clone(), self.level, predecl, ); } } let ctxs = self.get_singular_ctxs(namespace.as_ref(), self)?; for ctx in ctxs { if let Some(ctx) = ctx.rec_local_get_type(t.inspect()) { // TODO: visibility check return Ok(ctx.typ.clone()); } } Err(TyCheckErrors::from(TyCheckError::no_var_error( self.cfg.input.clone(), line!() as usize, t.loc(), self.caused_by(), t.inspect(), self.get_similar_name(t.inspect()), ))) } other => type_feature_error!(self, other.loc(), &format!("instantiating type {other}")), } } pub(crate) fn instantiate_mono_t( &self, ident: &Identifier, opt_decl_t: Option<&ParamTy>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> TyCheckResult { match &ident.inspect()[..] { "_" | "Obj" => Ok(Type::Obj), "Nat" => Ok(Type::Nat), "Int" => Ok(Type::Int), "Ratio" => Ok(Type::Ratio), "Float" => Ok(Type::Float), "Str" => Ok(Type::Str), "Bool" => Ok(Type::Bool), "NoneType" => Ok(Type::NoneType), "Ellipsis" => Ok(Type::Ellipsis), "NotImplementedType" => Ok(Type::NotImplementedType), "Inf" => Ok(Type::Inf), "NegInf" => Ok(Type::NegInf), "Never" => Ok(Type::Never), "ClassType" => Ok(Type::ClassType), "TraitType" => Ok(Type::TraitType), "Type" => Ok(Type::Type), "Self" => self.rec_get_self_t().ok_or_else(|| { TyCheckErrors::from(TyCheckError::self_type_error( self.cfg.input.clone(), line!() as usize, ident.loc(), self.caused_by(), )) }), "True" | "False" | "None" => Err(TyCheckErrors::from(TyCheckError::not_a_type_error( self.cfg.input.clone(), line!() as usize, ident.loc(), self.caused_by(), ident.inspect(), ))), other => { if let Some((TyParam::Type(t), vi)) = self.get_tp_from_tv_cache(other, tmp_tv_cache) { self.inc_ref(ident.inspect(), vi, ident, self); return Ok(*t); } if let Some(typ) = self .consts .get(ident.inspect()) .and_then(|v| self.convert_value_into_type(v.clone()).ok()) { if let Some((_, vi)) = self.get_var_info(ident.inspect()) { self.inc_ref(ident.inspect(), vi, ident, self); } return Ok(typ); } if let Some(outer) = &self.outer { if let Ok(t) = outer.instantiate_mono_t(ident, opt_decl_t, tmp_tv_cache, not_found_is_qvar) { return Ok(t); } } if let Some(ctx) = self.get_type_ctx(ident.inspect()) { if let Some((_, vi)) = self.get_var_info(ident.inspect()) { self.inc_ref(ident.inspect(), vi, ident, self); } Ok(ctx.typ.clone()) } else if not_found_is_qvar { let tyvar = named_free_var(Str::rc(other), self.level, Constraint::Uninited); tmp_tv_cache.push_or_init_tyvar(&ident.name, &tyvar, self)?; Ok(tyvar) } else if let Some(decl_t) = opt_decl_t { Ok(decl_t.typ().clone()) } else { Err(TyCheckErrors::from(TyCheckError::no_type_error( self.cfg.input.clone(), line!() as usize, ident.loc(), self.caused_by(), other, self.get_similar_name(other), ))) } } } } fn instantiate_local_poly_t( &self, name: &Identifier, args: &ConstArgs, opt_decl_t: Option<&ParamTy>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> Failable { match name.inspect().trim_start_matches([':', '.']) { "Array" => { let ctx = &self .get_nominal_type_ctx(&array_t(Type::Obj, TyParam::Failure)) .unwrap() .ctx; // TODO: kw let mut pos_args = args.pos_args(); if let Some(first) = pos_args.next() { let t = self .instantiate_const_expr_as_type( &first.expr, Some((ctx, 0)), tmp_tv_cache, not_found_is_qvar, ) .map_err(|err| (Type::Failure, err))?; let len = if let Some(len) = pos_args.next() { self.instantiate_const_expr( &len.expr, Some((ctx, 1)), tmp_tv_cache, not_found_is_qvar, ) .map_err(|err| (Type::Failure, err))? } else { TyParam::erased(Nat) }; Ok(array_t(t, len)) } else { Ok(mono("GenericArray")) } } "Ref" => { let mut pos_args = args.pos_args(); let Some(first) = pos_args.next() else { return Err(( Failure, TyCheckErrors::from(TyCheckError::args_missing_error( self.cfg.input.clone(), line!() as usize, args.loc(), "Ref", self.caused_by(), vec![Str::from("T")], )), )); }; let t = self .instantiate_const_expr_as_type( &first.expr, None, tmp_tv_cache, not_found_is_qvar, ) .map_err(|err| (Type::Failure, err))?; Ok(ref_(t)) } "RefMut" => { // TODO after let mut pos_args = args.pos_args(); let Some(first) = pos_args.next() else { return Err(( Failure, TyCheckErrors::from(TyCheckError::args_missing_error( self.cfg.input.clone(), line!() as usize, args.loc(), "RefMut", self.caused_by(), vec![Str::from("T")], )), )); }; let t = self .instantiate_const_expr_as_type( &first.expr, None, tmp_tv_cache, not_found_is_qvar, ) .map_err(|err| (Type::Failure, err))?; Ok(ref_mut(t, None)) } "Structural" => { let mut pos_args = args.pos_args(); let Some(first) = pos_args.next() else { return Err(( Failure, TyCheckErrors::from(TyCheckError::args_missing_error( self.cfg.input.clone(), line!() as usize, args.loc(), "Structural", self.caused_by(), vec![Str::from("Type")], )), )); }; let t = self .instantiate_const_expr_as_type( &first.expr, None, tmp_tv_cache, not_found_is_qvar, ) .map_err(|err| (Type::Failure, err))?; Ok(t.structuralize()) } "NamedTuple" => { let mut pose_args = args.pos_args(); let Some(first) = pose_args.next() else { return Err(( Failure, TyCheckErrors::from(TyCheckError::args_missing_error( self.cfg.input.clone(), line!() as usize, args.loc(), "NamedTuple", self.caused_by(), vec![Str::from("Fields")], )), )); }; let ConstExpr::Record(fields) = &first.expr else { return Err(( Failure, TyCheckErrors::from(TyCheckError::type_mismatch_error( self.cfg.input.clone(), line!() as usize, first.expr.loc(), self.caused_by(), "NamedTuple", None, &mono("Record"), &self .instantiate_const_expr_as_type( &first.expr, None, tmp_tv_cache, not_found_is_qvar, ) .map_err(|err| (Type::Failure, err))?, None, None, )), )); }; let mut ts = vec![]; for def in fields.attrs.iter() { let t = self .instantiate_const_expr_as_type( &def.body.block[0], None, tmp_tv_cache, not_found_is_qvar, ) .map_err(|err| (Type::Failure, err))?; let vis = self .instantiate_vis_modifier(&def.ident.vis) .map_err(|err| (Type::Failure, err))?; ts.push((Field::new(vis, def.ident.inspect().clone()), t)); } Ok(Type::NamedTuple(ts)) } other => { let Some(ctx) = self.get_type_ctx(other).or_else(|| { self.consts .get(other) .and_then(|v| self.convert_value_into_type(v.clone()).ok()) .and_then(|typ| self.get_nominal_type_ctx(&typ)) }) else { if let Some(outer) = &self.outer { if let Ok(t) = outer.instantiate_local_poly_t( name, args, opt_decl_t, tmp_tv_cache, not_found_is_qvar, ) { return Ok(t); } } if let Some(decl_t) = opt_decl_t { return Ok(decl_t.typ().clone()); } return Err(( Failure, TyCheckErrors::from(TyCheckError::no_type_error( self.cfg.input.clone(), line!() as usize, name.loc(), self.caused_by(), other, self.get_similar_name(other), )), )); }; let mut errs = TyCheckErrors::empty(); let mut new_params = vec![]; for ((i, arg), (name, param_vi)) in args.pos_args().enumerate().zip(ctx.params.iter()) { match self.instantiate_arg( &arg.expr, param_vi, name.as_ref(), ctx, i, tmp_tv_cache, not_found_is_qvar, ) { Ok(tp) => new_params.push(tp), Err((tp, es)) => { errs.extend(es); new_params.push(tp); } } } let mut missing_args = vec![]; // Fill kw params for (_, param_vi) in ctx.params.iter().skip(args.pos_args.len()) { new_params.push(TyParam::erased(param_vi.t.clone())); } let mut passed_kw_args = set! {}; for (i, (name, param_vi)) in ctx.params.iter().skip(args.pos_args.len()).enumerate() { if let Some(idx) = name.as_ref().and_then(|name| { args.kw_args .iter() .position(|arg| arg.keyword.inspect() == name.inspect()) }) { let kw_arg = &args.kw_args[idx]; let already_passed = !passed_kw_args.insert(kw_arg.keyword.inspect().clone()); if already_passed { errs.push(TyCheckError::multiple_args_error( self.cfg.input.clone(), line!() as usize, kw_arg.loc(), other, self.caused_by(), name.as_ref().map_or("_", |n| &n.inspect()[..]), )); } let tp = match self.instantiate_arg( &kw_arg.expr, param_vi, name.as_ref(), ctx, i, tmp_tv_cache, not_found_is_qvar, ) { Ok(tp) => tp, Err((tp, es)) => { errs.extend(es); tp } }; if let Some(old) = new_params.get_mut(args.pos_args.len() + idx) { *old = tp; } else { log!(err "{tp} / {} / {idx}", args.pos_args.len()); // TODO: too many kw args } } else if !param_vi.kind.has_default() { missing_args .push(name.as_ref().map_or("_".into(), |n| n.inspect().clone())); } } if !missing_args.is_empty() { errs.push(TyCheckError::args_missing_error( self.cfg.input.clone(), line!() as usize, args.loc(), other, self.caused_by(), missing_args, )); } if ctx.params.len() < args.pos_args.len() + args.kw_args.len() { errs.push(TyCheckError::too_many_args_error( self.cfg.input.clone(), line!() as usize, args.loc(), other, self.caused_by(), ctx.params.len(), args.pos_args.len(), args.kw_args.len(), )); } let param_names = ctx .params .iter() .filter_map(|(n, _)| n.as_ref().map(|n| &n.inspect()[..])) .collect::>(); for unexpected in args .kw_args .iter() .filter(|kw| !passed_kw_args.contains(&kw.keyword.inspect()[..])) { let kw = unexpected.keyword.inspect(); errs.push(TyCheckError::unexpected_kw_arg_error( self.cfg.input.clone(), line!() as usize, unexpected.loc(), other, self.caused_by(), unexpected.keyword.inspect(), get_similar_name(param_names.iter(), kw).copied(), )); } // FIXME: non-builtin let t = poly(ctx.typ.qual_name(), new_params); if errs.is_empty() { Ok(t) } else { Err((t, errs)) } } } } #[allow(clippy::too_many_arguments)] fn instantiate_arg( &self, arg: &ConstExpr, param_vi: &VarInfo, name: Option<&VarName>, ctx: &Context, i: usize, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> Failable { let param = self.instantiate_const_expr(arg, Some((ctx, i)), tmp_tv_cache, not_found_is_qvar); let param = param .or_else(|e| { if not_found_is_qvar { let name = arg.to_string(); // FIXME: handle `::` as a right way let name = Str::rc(name.trim_start_matches("::")); let tp = TyParam::named_free_var(name.clone(), self.level, Constraint::Uninited); let varname = VarName::from_str(name); tmp_tv_cache.push_or_init_typaram(&varname, &tp, self)?; Ok(tp) } else { Err(e) } }) .map_err(|err| (TyParam::Failure, err))?; let arg_t = self .get_tp_t(¶m) .map_err(|err| { log!(err "{param}: {err}"); err }) .unwrap_or(Obj); if self.subtype_of(&arg_t, ¶m_vi.t) { Ok(param) } else { let tp = TyParam::erased(param_vi.t.clone()); let errs = TyCheckError::type_mismatch_error( self.cfg.input.clone(), line!() as usize, arg.loc(), self.caused_by(), name.as_ref().map_or("", |n| &n.inspect()[..]), Some(i), ¶m_vi.t, &arg_t, None, None, ) .into(); Err((tp, errs)) } } fn instantiate_acc( &self, acc: &ast::ConstAccessor, erased_idx: Option<(&Context, usize)>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> TyCheckResult { self.inc_ref_acc(&acc.clone().downgrade(), self, tmp_tv_cache); match acc { ast::ConstAccessor::Attr(attr) => match self.instantiate_const_expr( &attr.obj, erased_idx, tmp_tv_cache, not_found_is_qvar, ) { Ok(obj) => Ok(obj.proj(attr.name.inspect())), Err(errs) => { if let Ok(ctxs) = self.get_singular_ctxs(&attr.obj.clone().downgrade(), self) { for ctx in ctxs { if let Some(value) = ctx.rec_get_const_obj(attr.name.inspect()) { return Ok(TyParam::Value(value.clone())); } } } Err(errs) } }, ast::ConstAccessor::Local(local) => { self.instantiate_local(local, erased_idx, tmp_tv_cache, local, not_found_is_qvar) } other => type_feature_error!( self, other.loc(), &format!("instantiating const expression {other}") ), } } fn instantiate_local( &self, name: &Identifier, erased_idx: Option<(&Context, usize)>, tmp_tv_cache: &mut TyVarCache, loc: &impl Locational, not_found_is_qvar: bool, ) -> TyCheckResult { if &name.inspect()[..] == "_" { let t = if let Some((ctx, i)) = erased_idx { let param = ctx.params.get(i).ok_or_else(|| { TyCheckErrors::from(TyCheckError::too_many_args_error( self.cfg.input.clone(), line!() as usize, loc.loc(), &ctx.name, self.caused_by(), ctx.params.len(), i, 0, )) })?; param.1.t.clone() } else { Type::Uninited }; return Ok(TyParam::erased(t)); } if let Some((tp, _vi)) = self.get_tp_from_tv_cache(name.inspect(), tmp_tv_cache) { return Ok(tp); } if let Some(value) = self.rec_get_const_obj(name.inspect()) { return Ok(TyParam::Value(value.clone())); } if not_found_is_qvar { let tyvar = named_free_var(name.inspect().clone(), self.level, Constraint::Uninited); tmp_tv_cache.push_or_init_tyvar(&name.name, &tyvar, self)?; return Ok(TyParam::t(tyvar)); } if name.is_const() { if let Some((_, vi)) = self.get_var_info(name.inspect()) { self.inc_ref(name.inspect(), vi, name, self); return Ok(TyParam::mono(name.inspect())); } } Err(TyCheckErrors::from(TyCheckError::no_var_error( self.cfg.input.clone(), line!() as usize, loc.loc(), self.caused_by(), name.inspect(), self.get_similar_name(name.inspect()), ))) } fn instantiate_app( &self, app: &ConstApp, erased_idx: Option<(&Context, usize)>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> TyCheckResult { match self.instantiate_const_expr(&app.obj, erased_idx, tmp_tv_cache, not_found_is_qvar) { Ok(obj) => { let ctx = self .get_singular_ctxs(&app.obj.clone().downgrade(), self) .ok() .and_then(|ctxs| ctxs.first().copied()) .or_else(|| { let typ = self.get_tp_t(&obj).ok()?; self.get_nominal_type_ctx(&typ).map(|ctx| &ctx.ctx) }) .unwrap_or(self); let mut args = vec![]; for (i, arg) in app.args.pos_args().enumerate() { let arg_t = self.instantiate_const_expr( &arg.expr, Some((ctx, i)), tmp_tv_cache, not_found_is_qvar, )?; args.push(arg_t); } if let Some(attr_name) = app.attr_name.as_ref() { Ok(obj.proj_call(attr_name.inspect().clone(), args)) } else if ctx.kind.is_type() && !ctx.params.is_empty() { Ok(TyParam::t(poly(ctx.name.clone(), args))) } else { let ast::ConstExpr::Accessor(ast::ConstAccessor::Local(ident)) = app.obj.as_ref() else { return type_feature_error!(self, app.loc(), "instantiating const callee"); }; Ok(TyParam::app(ident.inspect().clone(), args)) } } Err(errs) => { let Some(attr_name) = app.attr_name.as_ref() else { return Err(errs); }; let acc = app.obj.clone().attr(attr_name.clone()); let attr = self.instantiate_acc(&acc, erased_idx, tmp_tv_cache, not_found_is_qvar)?; let ctxs = self.get_singular_ctxs(&acc.downgrade().into(), self)?; let ctx = ctxs.first().copied().unwrap_or(self); let mut args = vec![]; for (i, arg) in app.args.pos_args().enumerate() { let arg_t = self.instantiate_const_expr( &arg.expr, Some((ctx, i)), tmp_tv_cache, not_found_is_qvar, )?; args.push(arg_t); } Ok(TyParam::app(attr.qual_name().unwrap(), args)) } } } /// erased_index: /// e.g. `instantiate_const_expr(Array(Str, _), Some((self, 1))) => Array(Str, _: Nat)` pub(crate) fn instantiate_const_expr( &self, expr: &ast::ConstExpr, erased_idx: Option<(&Context, usize)>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> TyCheckResult { if let Ok(value) = self.eval_const_expr(&expr.clone().downgrade()) { return Ok(TyParam::Value(value)); } match expr { ast::ConstExpr::Lit(lit) => Ok(TyParam::Value(self.eval_lit(lit)?)), ast::ConstExpr::Accessor(acc) => { self.instantiate_acc(acc, erased_idx, tmp_tv_cache, not_found_is_qvar) } ast::ConstExpr::App(app) => { self.instantiate_app(app, erased_idx, tmp_tv_cache, not_found_is_qvar) } ast::ConstExpr::Array(ConstArray::Normal(array)) => { let mut tp_arr = vec![]; for (i, elem) in array.elems.pos_args().enumerate() { let el = self.instantiate_const_expr( &elem.expr, Some((self, i)), tmp_tv_cache, not_found_is_qvar, )?; tp_arr.push(el); } Ok(TyParam::Array(tp_arr)) } ast::ConstExpr::Array(ConstArray::WithLength(arr)) => { let elem = self.instantiate_const_expr( &arr.elem, erased_idx, tmp_tv_cache, not_found_is_qvar, )?; let length = self.instantiate_const_expr( &arr.length, erased_idx, tmp_tv_cache, not_found_is_qvar, )?; if length.is_erased() { if let Ok(elem_t) = self.instantiate_tp_as_type(elem, arr) { return Ok(TyParam::t(unknown_len_array_t(elem_t))); } } type_feature_error!( self, arr.loc(), &format!("instantiating const expression {expr}") ) } ast::ConstExpr::Set(ConstSet::Normal(set)) => { let mut tp_set = set! {}; for (i, elem) in set.elems.pos_args().enumerate() { let el = self.instantiate_const_expr( &elem.expr, Some((self, i)), tmp_tv_cache, not_found_is_qvar, )?; tp_set.insert(el); } Ok(TyParam::Set(tp_set)) } ast::ConstExpr::Set(ConstSet::Comprehension(set)) => { if set.layout.is_none() && set.generators.len() == 1 && set.guard.is_some() { let (ident, expr) = set.generators.first().unwrap(); let iter = self.instantiate_const_expr( expr, erased_idx, tmp_tv_cache, not_found_is_qvar, )?; let pred = self.instantiate_pred_from_expr(set.guard.as_ref().unwrap(), tmp_tv_cache)?; if let Ok(t) = self.instantiate_tp_as_type(iter, set) { return Ok(TyParam::t(refinement(ident.inspect().clone(), t, pred))); } } type_feature_error!( self, set.loc(), &format!("instantiating const expression {expr}") ) } ast::ConstExpr::Dict(dict) => { let mut tp_dict = dict! {}; for (i, elem) in dict.kvs.iter().enumerate() { let key = self.instantiate_const_expr( &elem.key, Some((self, i)), tmp_tv_cache, not_found_is_qvar, )?; let val = self.instantiate_const_expr( &elem.value, Some((self, i)), tmp_tv_cache, not_found_is_qvar, )?; tp_dict.insert(key, val); } Ok(TyParam::Dict(tp_dict)) } ast::ConstExpr::Tuple(tuple) => { let mut tp_tuple = vec![]; for (i, elem) in tuple.elems.pos_args().enumerate() { let el = self.instantiate_const_expr( &elem.expr, Some((self, i)), tmp_tv_cache, not_found_is_qvar, )?; tp_tuple.push(el); } Ok(TyParam::Tuple(tp_tuple)) } ast::ConstExpr::Record(rec) => { let mut tp_rec = dict! {}; for attr in rec.attrs.iter() { let field = self.instantiate_field(&attr.ident)?; let val = self.instantiate_const_expr( attr.body.block.get(0).unwrap(), None, tmp_tv_cache, not_found_is_qvar, )?; tp_rec.insert(field, val); } Ok(TyParam::Record(tp_rec)) } ast::ConstExpr::Lambda(lambda) => { let mut errs = TyCheckErrors::empty(); let _tmp_tv_cache = match self .instantiate_ty_bounds(&lambda.sig.bounds, RegistrationMode::Normal) { Ok(tv_cache) => tv_cache, Err((tv_cache, es)) => { errs.extend(es); tv_cache } }; // Since there are type variables and other variables that can be constrained within closures, // they are `merge`d once and then `purge`d of type variables that are only used internally after instantiation. tmp_tv_cache.merge(&_tmp_tv_cache); let mut nd_params = Vec::with_capacity(lambda.sig.params.non_defaults.len()); for sig in lambda.sig.params.non_defaults.iter() { let pt = match self.instantiate_param_ty( sig, None, tmp_tv_cache, RegistrationMode::Normal, ParamKind::NonDefault, not_found_is_qvar, ) { Ok(pt) => pt, Err((pt, es)) => { errs.extend(es); pt } }; nd_params.push(pt); } let var_params = if let Some(p) = lambda.sig.params.var_params.as_ref() { let pt = match self.instantiate_param_ty( p, None, tmp_tv_cache, RegistrationMode::Normal, ParamKind::VarParams, not_found_is_qvar, ) { Ok(pt) => pt, Err((pt, es)) => { errs.extend(es); pt } }; Some(pt) } else { None }; let mut d_params = Vec::with_capacity(lambda.sig.params.defaults.len()); for sig in lambda.sig.params.defaults.iter() { let expr = self.eval_const_expr(&sig.default_val)?; let pt = match self.instantiate_param_ty( &sig.sig, None, tmp_tv_cache, RegistrationMode::Normal, ParamKind::Default(expr.t()), not_found_is_qvar, ) { Ok(pt) => pt, Err((pt, es)) => { errs.extend(es); pt } }; d_params.push(pt); } let kw_var_params = if let Some(p) = lambda.sig.params.kw_var_params.as_ref() { let pt = match self.instantiate_param_ty( p, None, tmp_tv_cache, RegistrationMode::Normal, ParamKind::KwVarParams, not_found_is_qvar, ) { Ok(pt) => pt, Err((pt, es)) => { errs.extend(es); pt } }; Some(pt) } else { None }; let mut lambda_ctx = Context::instant( Str::ever(""), self.cfg.clone(), 0, self.shared.clone(), self.clone(), ); for non_default in nd_params.iter() { let name = non_default .name() .map(|name| VarName::from_str(name.clone())); let vi = VarInfo::nd_parameter( non_default.typ().clone(), AbsLocation::unknown(), lambda_ctx.name.clone(), ); lambda_ctx.params.push((name, vi)); } if let Some(var_param) = var_params.as_ref() { let name = var_param.name().map(|name| VarName::from_str(name.clone())); let vi = VarInfo::nd_parameter( var_param.typ().clone(), AbsLocation::unknown(), lambda_ctx.name.clone(), ); lambda_ctx.params.push((name, vi)); } for default in d_params.iter() { let name = default.name().map(|name| VarName::from_str(name.clone())); let vi = VarInfo::d_parameter( default.typ().clone(), AbsLocation::unknown(), lambda_ctx.name.clone(), ); lambda_ctx.params.push((name, vi)); } let mut body = vec![]; for expr in lambda.body.iter() { let param = lambda_ctx.instantiate_const_expr( expr, None, tmp_tv_cache, not_found_is_qvar, )?; body.push(param); } tmp_tv_cache.purge(&_tmp_tv_cache); if errs.is_empty() { Ok(TyParam::Lambda(TyParamLambda::new( lambda.clone(), nd_params, var_params, d_params, kw_var_params, body, ))) } else { Err(errs) } } ast::ConstExpr::BinOp(bin) => { let Some(op) = token_kind_to_op_kind(bin.op.kind) else { return type_feature_error!( self, bin.loc(), &format!("instantiating const expression {bin}") ); }; let lhs = self.instantiate_const_expr( &bin.lhs, erased_idx, tmp_tv_cache, not_found_is_qvar, )?; let rhs = self.instantiate_const_expr( &bin.rhs, erased_idx, tmp_tv_cache, not_found_is_qvar, )?; Ok(TyParam::bin(op, lhs, rhs)) } ast::ConstExpr::UnaryOp(unary) => { let Some(op) = token_kind_to_op_kind(unary.op.kind) else { return type_feature_error!( self, unary.loc(), &format!("instantiating const expression {unary}") ); }; let val = self.instantiate_const_expr( &unary.expr, erased_idx, tmp_tv_cache, not_found_is_qvar, )?; Ok(TyParam::unary(op, val)) } ast::ConstExpr::TypeAsc(tasc) => { let tp = self.instantiate_const_expr( &tasc.expr, erased_idx, tmp_tv_cache, not_found_is_qvar, )?; let spec_t = self.instantiate_typespec_full( &tasc.t_spec.t_spec, None, tmp_tv_cache, RegistrationMode::Normal, false, )?; if self.subtype_of(&self.get_tp_t(&tp)?, &spec_t) { Ok(tp) } else { Err(TyCheckErrors::from(TyCheckError::subtyping_error( self.cfg.input.clone(), line!() as usize, &self.get_tp_t(&tp)?, &spec_t, tasc.loc(), self.caused_by(), ))) } } other => type_feature_error!( self, other.loc(), &format!("instantiating const expression {other}") ), } } pub(crate) fn instantiate_const_expr_as_type( &self, expr: &ast::ConstExpr, erased_idx: Option<(&Context, usize)>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> TyCheckResult { let tp = self.instantiate_const_expr(expr, erased_idx, tmp_tv_cache, not_found_is_qvar)?; self.instantiate_tp_as_type(tp, expr) } fn instantiate_tp_as_type(&self, tp: TyParam, loc: &impl Locational) -> TyCheckResult { match tp { TyParam::FreeVar(fv) if fv.is_linked() => { let tp = fv.crack().clone(); self.instantiate_tp_as_type(tp, loc) } TyParam::Mono(name) => Ok(mono(name)), TyParam::Proj { obj, attr } => { let obj = self.instantiate_tp_as_type(*obj, loc)?; Ok(proj(obj, attr)) } TyParam::App { name, args } => Ok(poly(name, args)), TyParam::Type(t) => Ok(*t), #[allow(clippy::bind_instead_of_map)] TyParam::Value(value) => self.convert_value_into_type(value).or_else(|value| { type_feature_error!(self, loc.loc(), &format!("instantiate `{value}` as type")) }), TyParam::Array(arr) => { let len = TyParam::value(arr.len()); let mut union = Type::Never; for tp in arr { let t = self.instantiate_tp_as_type(tp, loc)?; union = self.union(&union, &t); } Ok(array_t(union, len)) } TyParam::Set(set) => { let t = set .iter() .next() .and_then(|tp| self.get_tp_t(tp).ok()) .unwrap_or(Type::Never); Ok(tp_enum(t, set)) } TyParam::Tuple(ts) => { let mut tps = vec![]; for tp in ts { let t = self.instantiate_tp_as_type(tp, loc)?; tps.push(t); } Ok(tuple_t(tps)) } TyParam::Record(rec) => { let mut rec_t = dict! {}; for (field, tp) in rec { let t = self.instantiate_tp_as_type(tp, loc)?; rec_t.insert(field, t); } Ok(Type::Record(rec_t)) } TyParam::Lambda(lambda) => { let return_t = self .instantiate_tp_as_type(lambda.body.last().unwrap().clone(), &lambda.const_)?; let subr = SubrType::new( SubrKind::from(lambda.const_.op.kind), lambda.nd_params, lambda.var_params, lambda.d_params, lambda.kw_var_params, return_t, ); Ok(Type::Subr(subr)) } TyParam::BinOp { op, lhs, rhs } => match op { OpKind::And => { let lhs = self.instantiate_tp_as_type(*lhs, loc)?; let rhs = self.instantiate_tp_as_type(*rhs, loc)?; Ok(lhs & rhs) } OpKind::Or => { let lhs = self.instantiate_tp_as_type(*lhs, loc)?; let rhs = self.instantiate_tp_as_type(*rhs, loc)?; Ok(lhs | rhs) } _ => type_feature_error!( self, loc.loc(), &format!("instantiate `{lhs} {op} {rhs}` as type") ), }, other => { #[allow(clippy::bind_instead_of_map)] self.convert_tp_into_type(other).or_else(|tp| { type_feature_error!(self, loc.loc(), &format!("instantiate `{tp}` as type")) }) } } } fn instantiate_func_param_spec( &self, p: &ParamTySpec, opt_decl_t: Option<&ParamTy>, default_t: Option<&TypeSpec>, tmp_tv_cache: &mut TyVarCache, mode: RegistrationMode, not_found_is_qvar: bool, ) -> Failable { let mut errs = TyCheckErrors::empty(); let t = match self.instantiate_typespec_full( &p.ty, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, ) { Ok(t) => t, Err(es) => { errs.extend(es); Type::Failure } }; let pt = if let Some(default_t) = default_t { let default = match self.instantiate_typespec_full( default_t, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, ) { Ok(t) => t, Err(es) => { errs.extend(es); Type::Failure } }; ParamTy::kw_default(p.name.as_ref().unwrap().inspect().to_owned(), t, default) } else { ParamTy::pos_or_kw(p.name.as_ref().map(|t| t.inspect().to_owned()), t) }; if errs.is_empty() { Ok(pt) } else { Err((pt, errs)) } } fn instantiate_pred_from_expr( &self, expr: &ast::ConstExpr, tmp_tv_cache: &mut TyVarCache, ) -> TyCheckResult { match expr { ast::ConstExpr::Lit(lit) => { let value = self.eval_lit(lit)?; Ok(Predicate::Value(value)) } ast::ConstExpr::Accessor(ast::ConstAccessor::Local(local)) => { self.inc_ref_local(local, self, tmp_tv_cache); Ok(Predicate::Const(local.inspect().clone())) } ast::ConstExpr::Accessor(ast::ConstAccessor::Attr(attr)) => { let obj = self.instantiate_const_expr(&attr.obj, None, tmp_tv_cache, false)?; Ok(Predicate::attr(obj, attr.name.inspect().clone())) } ast::ConstExpr::App(app) => { let receiver = self.instantiate_const_expr(&app.obj, None, tmp_tv_cache, false)?; let name = app.attr_name.as_ref().map(|n| n.inspect().to_owned()); let mut args = vec![]; for arg in app.args.pos_args() { let arg = self.instantiate_const_expr(&arg.expr, None, tmp_tv_cache, false)?; args.push(arg); } Ok(Predicate::Call { receiver, name, args, }) } ast::ConstExpr::BinOp(bin) => { let lhs = self.instantiate_pred_from_expr(&bin.lhs, tmp_tv_cache)?; let rhs = self.instantiate_pred_from_expr(&bin.rhs, tmp_tv_cache)?; match bin.op.kind { TokenKind::DblEq | TokenKind::NotEq | TokenKind::Less | TokenKind::LessEq | TokenKind::Gre | TokenKind::GreEq => { let var = match lhs { Predicate::Const(var) => var, other if bin.op.kind == TokenKind::DblEq => { return Ok(Predicate::general_eq(other, rhs)); } other if bin.op.kind == TokenKind::NotEq => { return Ok(Predicate::general_ne(other, rhs)); } other if bin.op.kind == TokenKind::GreEq => { return Ok(Predicate::general_ge(other, rhs)); } other if bin.op.kind == TokenKind::LessEq => { return Ok(Predicate::general_le(other, rhs)); } _ => { return type_feature_error!( self, bin.loc(), &format!("instantiating predicate `{expr}`") ); } }; let rhs = match rhs { Predicate::Value(value) => TyParam::Value(value), Predicate::Const(var) => TyParam::Mono(var), other if bin.op.kind == TokenKind::DblEq => { return Ok(Predicate::general_eq(Predicate::Const(var), other)); } other if bin.op.kind == TokenKind::NotEq => { return Ok(Predicate::general_ne(Predicate::Const(var), other)); } other if bin.op.kind == TokenKind::GreEq => { return Ok(Predicate::general_ge(Predicate::Const(var), other)); } other if bin.op.kind == TokenKind::LessEq => { return Ok(Predicate::general_le(Predicate::Const(var), other)); } _ => { return type_feature_error!( self, bin.loc(), &format!("instantiating predicate `{expr}`") ); } }; let pred = match bin.op.kind { TokenKind::DblEq => Predicate::eq(var, rhs), TokenKind::NotEq => Predicate::ne(var, rhs), TokenKind::Less => Predicate::lt(var, rhs), TokenKind::LessEq => Predicate::le(var, rhs), TokenKind::Gre => Predicate::gt(var, rhs), TokenKind::GreEq => Predicate::ge(var, rhs), _ => unreachable!(), }; Ok(pred) } TokenKind::OrOp => Ok(lhs | rhs), TokenKind::AndOp => Ok(lhs & rhs), _ => type_feature_error!( self, bin.loc(), &format!("instantiating predicate `{expr}`") ), } } ast::ConstExpr::UnaryOp(unop) => { let pred = self.instantiate_pred_from_expr(&unop.expr, tmp_tv_cache)?; match unop.op.kind { TokenKind::PreBitNot => Ok(!pred), _ => type_feature_error!( self, unop.loc(), &format!("instantiating predicate `{expr}`") ), } } _ => type_feature_error!( self, expr.loc(), &format!("instantiating predicate `{expr}`") ), } } #[inline] fn recover_guard<'a>(&self, return_t: Type, mut params: impl Iterator) -> Type { match return_t { Type::Guard(GuardType { namespace, target: CastTarget::Expr(expr), to, }) => { let target = if let Some(nth) = params.position(|p| Some(p) == expr.get_name()) { CastTarget::arg(nth, expr.get_name().unwrap().clone(), ().loc()) } else { CastTarget::Expr(expr) }; Type::Guard(GuardType { namespace, target, to, }) } _ => return_t, } } // FIXME: opt_decl_t must be disassembled for each polymorphic type pub(crate) fn instantiate_typespec_full( &self, t_spec: &TypeSpec, opt_decl_t: Option<&ParamTy>, tmp_tv_cache: &mut TyVarCache, mode: RegistrationMode, not_found_is_qvar: bool, ) -> TyCheckResult { match t_spec { TypeSpec::Infer(_) => Ok(free_var(self.level, Constraint::new_type_of(Type))), TypeSpec::PreDeclTy(predecl) => Ok(self.instantiate_predecl_t( predecl, opt_decl_t, tmp_tv_cache, not_found_is_qvar, )?), TypeSpec::And(lhs, rhs) => Ok(self.intersection( &self.instantiate_typespec_full( lhs, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, &self.instantiate_typespec_full( rhs, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, )), TypeSpec::Or(lhs, rhs) => Ok(self.union( &self.instantiate_typespec_full( lhs, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, &self.instantiate_typespec_full( rhs, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, )), TypeSpec::Not(ty) => Ok(self.complement(&self.instantiate_typespec_full( ty, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?)), TypeSpec::Array(arr) => { let elem_t = self.instantiate_typespec_full( &arr.ty, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?; let mut len = self.instantiate_const_expr(&arr.len, None, tmp_tv_cache, not_found_is_qvar)?; if let TyParam::Erased(t) = &mut len { *t.as_mut() = Type::Nat; } Ok(array_t(elem_t, len)) } TypeSpec::SetWithLen(set) => { let elem_t = self.instantiate_typespec_full( &set.ty, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?; let mut len = self.instantiate_const_expr(&set.len, None, tmp_tv_cache, not_found_is_qvar)?; if let TyParam::Erased(t) = &mut len { *t.as_mut() = Type::Nat; } Ok(set_t(elem_t, len)) } TypeSpec::Tuple(tup) => { let mut inst_tys = vec![]; for spec in tup.tys.iter() { inst_tys.push(self.instantiate_typespec_full( spec, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?); } Ok(tuple_t(inst_tys)) } TypeSpec::Dict(dict) => { let mut inst_tys = dict! {}; for (k, v) in dict.kvs.iter() { inst_tys.insert( self.instantiate_typespec_full( k, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, self.instantiate_typespec_full( v, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, ); } Ok(dict_t(inst_tys.into())) } TypeSpec::Record(rec) => { let mut inst_tys = dict! {}; for (k, v) in rec.attrs.iter() { inst_tys.insert( self.instantiate_field(k)?, self.instantiate_typespec_full( v, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, ); } Ok(Type::Record(inst_tys)) } // TODO: エラー処理(リテラルでない)はパーサーにやらせる TypeSpec::Enum(set) => { let mut new_set = set! {}; // guard type (e.g. {x in Int}) if set.pos_args.len() == 1 { let expr = &set.pos_args().next().unwrap().expr; match expr { ConstExpr::BinOp(bin) if bin.op.is(TokenKind::InOp) => { if let Ok(to) = self.instantiate_const_expr_as_type( &bin.rhs, None, tmp_tv_cache, not_found_is_qvar, ) { let target = CastTarget::expr(bin.lhs.clone().downgrade()); return Ok(Type::Guard(GuardType::new( self.name.clone(), target, to, ))); } } _ => {} } } for arg in set.pos_args() { new_set.insert(self.instantiate_const_expr( &arg.expr, None, tmp_tv_cache, not_found_is_qvar, )?); } let ty = new_set.iter().fold(Type::Never, |t, tp| { self.union(&t, &self.get_tp_t(tp).unwrap().derefine()) }); Ok(tp_enum(ty, new_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, None, tmp_tv_cache, not_found_is_qvar)?; let l = self.eval_tp(l)?; let r = self.instantiate_const_expr(rhs, None, tmp_tv_cache, not_found_is_qvar)?; let r = self.eval_tp(r)?; if let Some(Greater) = self.try_cmp(&l, &r) { panic!("{l}..{r} is not a valid interval type (should be lhs <= rhs)") } let l_t = self.get_tp_t(&l).unwrap_or(Obj).derefine(); let r_t = self.get_tp_t(&r).unwrap_or(Obj).derefine(); let t = self.union(&l_t, &r_t); Ok(interval(op, t, l, r)) } TypeSpec::Subr(subr) => { let mut errs = TyCheckErrors::empty(); let mut inner_tv_ctx = if !subr.bounds.is_empty() { let tv_cache = match self.instantiate_ty_bounds(&subr.bounds, mode) { Ok(tv_cache) => tv_cache, Err((tv_cache, es)) => { errs.extend(es); tv_cache } }; Some(tv_cache) } else { None }; if let Some(inner) = &mut inner_tv_ctx { inner.merge(tmp_tv_cache); } let tmp_tv_ctx = if let Some(inner) = &mut inner_tv_ctx { inner } else { tmp_tv_cache }; let non_defaults = match failable_map_mut(subr.non_defaults.iter(), |p| { self.instantiate_func_param_spec( p, opt_decl_t, None, tmp_tv_ctx, mode, not_found_is_qvar, ) }) { Ok(v) => v, Err((v, es)) => { for e in es { errs.extend(e); } v } }; let var_params = subr .var_params .as_ref() .map(|p| { self.instantiate_func_param_spec( p, opt_decl_t, None, tmp_tv_ctx, mode, not_found_is_qvar, ) .map_err(|(_, es)| es) }) .transpose()?; let defaults = failable_map_mut(subr.defaults.iter(), |p| { self.instantiate_func_param_spec( &p.param, opt_decl_t, Some(&p.default), tmp_tv_ctx, mode, not_found_is_qvar, ) }) .unwrap_or_else(|(pts, es)| { for e in es { errs.extend(e); } pts }) .into_iter() .collect(); let kw_var_params = subr .kw_var_params .as_ref() .map(|p| { self.instantiate_func_param_spec( p, opt_decl_t, None, tmp_tv_ctx, mode, not_found_is_qvar, ) .map_err(|(_, es)| es) }) .transpose()?; let return_t = match self.instantiate_typespec_full( &subr.return_t, opt_decl_t, tmp_tv_ctx, mode, not_found_is_qvar, ) { Ok(t) => t, Err(es) => { errs.extend(es); Type::Failure } }; let params = non_defaults .iter() .chain(&var_params) .chain(&defaults) .chain(&kw_var_params) .filter_map(|pt| pt.name()); let return_t = self.recover_guard(return_t, params); // no quantification at this point (in `generalize_t`) if errs.is_empty() { Ok(subr_t( SubrKind::from(subr.arrow.kind), non_defaults, var_params, defaults, kw_var_params, return_t, )) } else { Err(errs) } } TypeSpec::TypeApp { spec, args } => { type_feature_error!( self, t_spec.loc(), &format!("instantiating type spec {spec}{args}") ) } TypeSpec::Refinement(refine) => { let t = self.instantiate_typespec_full( &refine.typ, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?; let name = VarName::new(refine.var.clone()); tmp_tv_cache.push_refine_var(&name, t.clone(), self); let pred = self .instantiate_pred_from_expr(&refine.pred, tmp_tv_cache) .map_err(|errs| { tmp_tv_cache.remove(name.inspect()); errs })?; tmp_tv_cache.remove(name.inspect()); let refine = Type::Refinement(RefinementType::new(refine.var.inspect().clone(), t, pred)); Ok(refine) } } } pub(crate) fn instantiate_typespec(&self, t_spec: &ast::TypeSpec) -> TyCheckResult { let mut dummy_tv_cache = TyVarCache::new(self.level, self); self.instantiate_typespec_with_tv_cache(t_spec, &mut dummy_tv_cache) } pub(crate) fn instantiate_typespec_with_tv_cache( &self, t_spec: &ast::TypeSpec, tv_cache: &mut TyVarCache, ) -> TyCheckResult { let t = self.instantiate_typespec_full( t_spec, None, tv_cache, RegistrationMode::Normal, false, )?; t.lift(); let t = self.generalize_t(t); Ok(t) } pub(crate) fn instantiate_field(&self, ident: &Identifier) -> TyCheckResult { let vis = self.instantiate_vis_modifier(&ident.vis)?; Ok(Field::new(vis, ident.inspect().clone())) } pub(crate) fn instantiate_vis_modifier( &self, spec: &VisModifierSpec, ) -> TyCheckResult { match spec { VisModifierSpec::Auto => unreachable!(), VisModifierSpec::Private | VisModifierSpec::ExplicitPrivate(_) => { Ok(VisibilityModifier::Private) } VisModifierSpec::Public(_) => Ok(VisibilityModifier::Public), VisModifierSpec::Restricted(rest) => match &rest { VisRestriction::Namespaces(namespace) => { let mut namespaces = set! {}; for ns in namespace.iter() { let ast::Accessor::Ident(ident) = ns else { return type_feature_error!(self, ns.loc(), "namespace accessors"); }; let vi = self .rec_get_var_info(ident, AccessKind::Name, &self.cfg.input, self) .none_or_result(|| { TyCheckError::no_var_error( self.cfg.input.clone(), line!() as usize, ident.loc(), self.caused_by(), ident.inspect(), None, ) })?; let name = Str::from(format!( "{}{}{}", vi.vis.def_namespace, vi.vis.modifier.display_as_accessor(), ident.inspect() )); namespaces.insert(name); } Ok(VisibilityModifier::Restricted(namespaces)) } VisRestriction::SubtypeOf(typ) => { let t = self.instantiate_typespec(typ)?; Ok(VisibilityModifier::SubtypeRestricted(t)) } }, } } pub(crate) fn expr_to_type(&self, expr: ast::Expr) -> Option { let t_spec = Parser::expr_to_type_spec(expr).ok()?; self.instantiate_typespec(&t_spec).ok() } pub(crate) fn expr_to_value(&self, expr: ast::Expr) -> Option { let const_expr = Parser::validate_const_expr(expr).ok()?; let mut dummy = TyVarCache::new(self.level, self); self.instantiate_const_expr(&const_expr, None, &mut dummy, false) .ok() .and_then(|tp| ValueObj::try_from(tp).ok()) } pub(crate) fn expr_to_tp(&self, expr: ast::Expr) -> Option { let const_expr = Parser::validate_const_expr(expr).ok()?; let mut dummy = TyVarCache::new(self.level, self); self.instantiate_const_expr(&const_expr, None, &mut dummy, false) .ok() } }