use std::option::Option; // conflicting to Type::Option #[allow(unused)] use erg_common::log; use erg_common::traits::{Locational, Stream}; use erg_common::Str; use erg_common::{assume_unreachable, dict, set, try_map_mut}; use ast::{ NonDefaultParamSignature, ParamTySpec, PreDeclTypeSpec, SimpleTypeSpec, TypeBoundSpec, TypeBoundSpecs, TypeSpec, }; use erg_parser::ast::{self, Identifier, VisModifierSpec, VisRestriction}; use erg_parser::token::TokenKind; use erg_parser::Parser; use crate::feature_error; use crate::ty::free::{CanbeFree, Constraint}; use crate::ty::typaram::{IntervalOp, OpKind, TyParam, TyParamLambda, TyParamOrdering}; use crate::ty::value::ValueObj; use crate::ty::{constructors::*, VisibilityModifier}; use crate::ty::{Field, HasType, ParamTy, SubrKind, SubrType, Type}; use crate::type_feature_error; use TyParamOrdering::*; use Type::*; use crate::context::instantiate::TyVarCache; use crate::context::{Context, DefaultInfo, RegistrationMode}; use crate::error::{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, KwParams, } impl ParamKind { pub const fn is_var_params(&self) -> bool { matches!(self, ParamKind::VarParams) } pub const fn is_kw_params(&self) -> bool { matches!(self, ParamKind::KwParams) } 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.inspect(), &tv, self); Ok(()) } TypeBoundSpec::NonDefault { lhs, spec } => { let constr = match spec.op.kind { TokenKind::SubtypeOf => Constraint::new_subtype_of( self.instantiate_typespec(&spec.t_spec, None, tv_cache, mode, true)?, ), TokenKind::SupertypeOf => Constraint::new_supertype_of( self.instantiate_typespec(&spec.t_spec, None, tv_cache, mode, true)?, ), TokenKind::Colon => Constraint::new_type_of(self.instantiate_typespec( &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.inspect(), &tp, self); } else { let tv = named_free_var(lhs.inspect().clone(), self.level, constr); tv_cache.push_or_init_tyvar(lhs.inspect(), &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, ) -> TyCheckResult { let mut tv_cache = TyVarCache::new(self.level, self); for bound in bounds.iter() { self.instantiate_ty_bound(bound, &mut tv_cache, mode)?; } for tv in tv_cache.tyvar_instances.values() { if tv.constraint().map(|c| c.is_uninited()).unwrap_or(false) { return Err(TyCheckErrors::from(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) { return Err(TyCheckErrors::from(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()), ))); } } Ok(tv_cache) } 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(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, ) -> Result { 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((TyCheckErrors::from(err), other)); } None => None, }; let mut tmp_tv_cache = self .instantiate_ty_bounds(&sig.bounds, PreRegister) .map_err(|errs| (errs, Type::Failure))?; 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, ) { Ok(pt) => non_defaults.push(pt), Err(es) => { errs.extend(es); non_defaults.push(ParamTy::pos_or_kw(param.inspect().cloned(), Type::Failure)); } } } let var_args = 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, ) { Ok(pt) => pt, Err(es) => { errs.extend(es); ParamTy::pos_or_kw(var_args.inspect().cloned(), Type::Failure) } }; 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), ) { Ok(pt) => defaults.push(pt), Err(es) => { errs.extend(es); defaults.push(ParamTy::pos_or_kw(p.sig.inspect().cloned(), Type::Failure)); } } } 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( t_spec, opt_decl_t.as_ref(), &mut tmp_tv_cache, mode, false, ) { Ok(ty) => ty, Err(es) => { errs.extend(es); Type::Failure } } } else { // preregisterならouter scopeで型宣言(see inference.md) let level = if mode == PreRegister { self.level } else { self.level + 1 }; free_var(level, Constraint::new_type_of(Type)) }; let typ = if sig.ident.is_procedural() { proc(non_defaults, var_args, defaults, spec_return_t) } else { func(non_defaults, var_args, defaults, spec_return_t) }; if errs.is_empty() { Ok(typ) } else { Err((errs, typ)) } } /// 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, ) -> TyCheckResult { let gen_free_t = || { let level = if mode == 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(&spec_with_op.t_spec, opt_decl_t, tmp_tv_cache, mode, false)? } else { match &sig.pat { ast::ParamPattern::Lit(lit) => v_enum(set![self.eval_lit(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, )?; } else { self.sub_unify( decl_pt.typ(), &spec_t, &sig.t_spec.as_ref().ok_or(sig), None, )?; } } 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, ) -> TyCheckResult { 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) = 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 => { return Ok(ParamTy::Pos(tp_enum( self.get_tp_t(&other)?, set! { other }, ))) } } } let t = self.instantiate_param_sig_t(sig, opt_decl_t, tmp_tv_cache, mode, kind.clone())?; match (sig.inspect(), kind) { (Some(name), ParamKind::Default(default_t)) => { Ok(ParamTy::kw_default(name.clone(), t, default_t)) } (Some(name), _) => Ok(ParamTy::kw(name.clone(), t)), (None, _) => Ok(ParamTy::Pos(t)), } } 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); match predecl { ast::PreDeclTypeSpec::Simple(simple) => { self.instantiate_simple_t(simple, opt_decl_t, tmp_tv_cache, not_found_is_qvar) } 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, ) { let rhs = t.ident.inspect(); return Ok(proj(receiver_t, rhs)); } } let ctxs = self.get_singular_ctxs(namespace.as_ref(), self)?; for ctx in ctxs { if let Some((typ, _)) = ctx.rec_local_get_type(t.ident.inspect()) { // TODO: visibility check return Ok(typ.clone()); } } Err(TyCheckErrors::from(TyCheckError::no_var_error( self.cfg.input.clone(), line!() as usize, t.loc(), self.caused_by(), t.ident.inspect(), self.get_similar_name(t.ident.inspect()), ))) } other => type_feature_error!(self, other.loc(), &format!("instantiating type {other}")), } } pub(crate) fn instantiate_simple_t( &self, simple: &SimpleTypeSpec, opt_decl_t: Option<&ParamTy>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> TyCheckResult { match &simple.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), "Array" => { // TODO: kw let mut args = simple.args.pos_args(); if let Some(first) = args.next() { let t = self.instantiate_const_expr_as_type( &first.expr, None, tmp_tv_cache, not_found_is_qvar, )?; let len = if let Some(len) = args.next() { self.instantiate_const_expr( &len.expr, None, tmp_tv_cache, not_found_is_qvar, )? } else { TyParam::erased(Nat) }; Ok(array_t(t, len)) } else { Ok(mono("GenericArray")) } } "Ref" => { let mut args = simple.args.pos_args(); let Some(first) = args.next() else { return Err(TyCheckErrors::from(TyCheckError::args_missing_error( self.cfg.input.clone(), line!() as usize, simple.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, )?; Ok(ref_(t)) } "RefMut" => { // TODO after let mut args = simple.args.pos_args(); let Some(first) = args.next() else { return Err(TyCheckErrors::from(TyCheckError::args_missing_error( self.cfg.input.clone(), line!() as usize, simple.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, )?; Ok(ref_mut(t, None)) } "Structural" => { let mut args = simple.args.pos_args(); let Some(first) = args.next() else { return Err(TyCheckErrors::from(TyCheckError::args_missing_error( self.cfg.input.clone(), line!() as usize, simple.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, )?; Ok(t.structuralize()) } "Self" => self.rec_get_self_t().ok_or_else(|| { TyCheckErrors::from(TyCheckError::unreachable( self.cfg.input.clone(), erg_common::fn_name_full!(), line!(), )) }), other if simple.args.is_empty() => { if let Some(TyParam::Type(t)) = self.get_tp_from_tv_cache(other, tmp_tv_cache) { return Ok(*t); } if let Some(outer) = &self.outer { if let Ok(t) = outer.instantiate_simple_t( simple, 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()); } if let Some((typ, _)) = self.get_type(simple.ident.inspect()) { Ok(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(simple.ident.inspect(), &tyvar, self); Ok(tyvar) } else { Err(TyCheckErrors::from(TyCheckError::no_type_error( self.cfg.input.clone(), line!() as usize, simple.loc(), self.caused_by(), other, self.get_similar_name(other), ))) } } other => { let ctx = if let Some((_, ctx)) = self.get_type(&Str::rc(other)) { ctx } else { return Err(TyCheckErrors::from(TyCheckError::no_type_error( self.cfg.input.clone(), line!() as usize, simple.ident.loc(), self.caused_by(), other, self.get_similar_name(other), ))); }; // FIXME: kw args let mut new_params = vec![]; for (i, arg) in simple.args.pos_args().enumerate() { let params = self.instantiate_const_expr( &arg.expr, Some((ctx, i)), tmp_tv_cache, not_found_is_qvar, ); let params = params.or_else(|e| { if not_found_is_qvar { let name = arg.expr.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, ); tmp_tv_cache.push_or_init_typaram(&name, &tp, self); Ok(tp) } else { Err(e) } })?; new_params.push(params); } // FIXME: non-builtin Ok(poly(Str::rc(other), new_params)) } } } fn instantiate_acc( &self, acc: &ast::ConstAccessor, erased_idx: Option<(&Context, usize)>, tmp_tv_cache: &mut TyVarCache, not_found_is_qvar: bool, ) -> TyCheckResult { match acc { ast::ConstAccessor::Attr(attr) => { let obj = self.instantiate_const_expr( &attr.obj, erased_idx, tmp_tv_cache, not_found_is_qvar, )?; Ok(obj.proj(attr.name.inspect())) } ast::ConstAccessor::Local(local) => { self.inc_ref_local(local, self); self.instantiate_local( local.inspect(), 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: &Str, erased_idx: Option<(&Context, usize)>, tmp_tv_cache: &mut TyVarCache, loc: &impl Locational, not_found_is_qvar: bool, ) -> TyCheckResult { if &name[..] == "_" { 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) = self.get_tp_from_tv_cache(name, tmp_tv_cache) { return Ok(tp); } if let Some(value) = self.rec_get_const_obj(name) { return Ok(TyParam::Value(value.clone())); } if not_found_is_qvar { let tyvar = named_free_var(name.clone(), self.level, Constraint::Uninited); tmp_tv_cache.push_or_init_tyvar(name, &tyvar, self); return Ok(TyParam::t(tyvar)); } Err(TyCheckErrors::from(TyCheckError::no_var_error( self.cfg.input.clone(), line!() as usize, loc.loc(), self.caused_by(), name, self.get_similar_name(name), ))) } 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().downcast()) { return Ok(TyParam::Value(value)); } match expr { ast::ConstExpr::Lit(lit) => Ok(TyParam::Value(self.eval_lit(lit)?)), // TODO: inc_ref ast::ConstExpr::Accessor(acc) => { self.instantiate_acc(acc, erased_idx, tmp_tv_cache, not_found_is_qvar) } ast::ConstExpr::Array(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::Set(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::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 _tmp_tv_cache = self.instantiate_ty_bounds(&lambda.sig.bounds, RegistrationMode::Normal)?; // 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 = self.instantiate_param_ty( sig, None, tmp_tv_cache, RegistrationMode::Normal, ParamKind::NonDefault, )?; nd_params.push(pt); } let var_params = if let Some(p) = lambda.sig.params.var_params.as_ref() { let pt = self.instantiate_param_ty( p, None, tmp_tv_cache, RegistrationMode::Normal, ParamKind::VarParams, )?; 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 = self.instantiate_param_ty( &sig.sig, None, tmp_tv_cache, RegistrationMode::Normal, ParamKind::Default(expr.t()), )?; d_params.push(pt); } let mut body = vec![]; for expr in lambda.body.iter() { let param = self.instantiate_const_expr(expr, None, tmp_tv_cache, not_found_is_qvar)?; body.push(param); } tmp_tv_cache.purge(&_tmp_tv_cache); Ok(TyParam::Lambda(TyParamLambda::new( lambda.clone(), nd_params, var_params, d_params, body, ))) } 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( &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() => { self.instantiate_tp_as_type(fv.crack().clone(), loc) } TyParam::Type(t) => Ok(*t), #[allow(clippy::bind_instead_of_map)] TyParam::Value(value) => Type::try_from(value).or_else(|value| { type_feature_error!(self, loc.loc(), &format!("instantiate `{value}` as type")) }), 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, return_t, ); Ok(Type::Subr(subr)) } other => { type_feature_error!(self, loc.loc(), &format!("instantiate `{other}` 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, ) -> TyCheckResult { let t = self.instantiate_typespec(&p.ty, opt_decl_t, tmp_tv_cache, mode, false)?; if let Some(default_t) = default_t { Ok(ParamTy::kw_default( p.name.as_ref().unwrap().inspect().to_owned(), t, self.instantiate_typespec(default_t, opt_decl_t, tmp_tv_cache, mode, false)?, )) } else { Ok(ParamTy::pos_or_kw( p.name.as_ref().map(|t| t.inspect().to_owned()), t, )) } } pub(crate) fn instantiate_typespec( &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( lhs, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, &self.instantiate_typespec( rhs, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, )), TypeSpec::Or(lhs, rhs) => Ok(self.union( &self.instantiate_typespec( lhs, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, &self.instantiate_typespec( rhs, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, )), TypeSpec::Not(ty) => Ok(self.complement(&self.instantiate_typespec( ty, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?)), TypeSpec::Array(arr) => { let elem_t = self.instantiate_typespec( &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( &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( 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 { inst_tys.insert( self.instantiate_typespec( k, opt_decl_t, tmp_tv_cache, mode, not_found_is_qvar, )?, self.instantiate_typespec( 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 { inst_tys.insert( self.instantiate_field(k)?, self.instantiate_typespec( 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! {}; 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()) }); 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)") } Ok(int_interval(op, l, r)) } TypeSpec::Subr(subr) => { let mut inner_tv_ctx = if !subr.bounds.is_empty() { let tv_cache = self.instantiate_ty_bounds(&subr.bounds, mode)?; 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 = try_map_mut(subr.non_defaults.iter(), |p| { self.instantiate_func_param_spec(p, opt_decl_t, None, tmp_tv_ctx, mode) })?; let var_params = subr .var_params .as_ref() .map(|p| { self.instantiate_func_param_spec(p, opt_decl_t, None, tmp_tv_ctx, mode) }) .transpose()?; let defaults = try_map_mut(subr.defaults.iter(), |p| { self.instantiate_func_param_spec( &p.param, opt_decl_t, Some(&p.default), tmp_tv_ctx, mode, ) })? .into_iter() .collect(); let return_t = self.instantiate_typespec( &subr.return_t, opt_decl_t, tmp_tv_ctx, mode, not_found_is_qvar, )?; Ok(subr_t( SubrKind::from(subr.arrow.kind), non_defaults, var_params, defaults, return_t, )) } TypeSpec::TypeApp { spec, args } => { type_feature_error!( self, t_spec.loc(), &format!("instantiating type spec {spec}{args}") ) } } } pub fn instantiate_field(&self, ident: &Identifier) -> TyCheckResult { let vis = self.instantiate_vis_modifier(&ident.vis)?; Ok(Field::new(vis, ident.inspect().clone())) } pub 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 feature_error!(TyCheckErrors, TyCheckError, 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 mut dummy_tv_cache = TyVarCache::new(self.level, self); let t = self.instantiate_typespec( typ, None, &mut dummy_tv_cache, RegistrationMode::Normal, false, )?; 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()?; let mut dummy = TyVarCache::new(self.level, self); self.instantiate_typespec(&t_spec, None, &mut dummy, RegistrationMode::Normal, false) .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()) } }