language-servers/erg
1
0
Fork 0
mirror of https://github.com/erg-lang/erg.git synced 2025-08-03 18:29:00 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: support inherited refinement types

Browse source
This commit is contained in:
Shunsuke Shibayama 2023-10-31 12:24:41 +09:00
parent 678c02faf9
commit a6336fa896
6 changed files with 125 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

10
crates/erg_compiler/ty/constructors.rs
View file

@ -100,9 +100,9 @@ pub fn module_from_path<P: Into<PathBuf>>(path: P) -> Type {
module(TyParam::Value(s))
}
pub fn v_enum(s: Set<ValueObj>) -> Type {
pub fn try_v_enum(s: Set<ValueObj>) -> Result<Type, Set<ValueObj>> {
if !is_homogeneous(&s) {
panic!("{s} is not homogeneous");
return Err(s);
}
let name = FRESH_GEN.fresh_varname();
let t = inner_class(&s);
@ -111,7 +111,11 @@ pub fn v_enum(s: Set<ValueObj>) -> Type {
.map(|o| Predicate::eq(name.clone(), TyParam::value(o)))
.fold(Predicate::FALSE, |acc, p| acc | p);
let refine = RefinementType::new(name, t, preds);
Type::Refinement(refine)
Ok(Type::Refinement(refine))
}
pub fn v_enum(s: Set<ValueObj>) -> Type {
try_v_enum(s).unwrap_or_else(|set| panic!("not homogeneous: {}", set))
}
pub fn tp_enum(ty: Type, s: Set<TyParam>) -> Type {