use std::mem::swap;
use voile_util::loc::ToLoc;
use crate::syntax::abs::AbsDecl;
use crate::syntax::core::{Neutral, TraverseNeutral, Val, ValInfo, TYPE_OMEGA};
use super::monad::{ValTCM, TCE, TCM, TCS};
pub fn check_decls(tcs: TCS, decls: Vec<AbsDecl>) -> TCM {
decls.into_iter().try_fold(tcs, check_decl)
}
fn require_local_emptiness(tcs: &TCS) {
debug_assert!(tcs.local_env.is_empty());
debug_assert!(tcs.local_gamma.is_empty());
}
fn unimplemented_to_glob(v: &mut [ValInfo], i: usize) {
let mut placeholder = Default::default();
swap(&mut v[i], &mut placeholder);
placeholder = placeholder.map_ast(|ast| ast.unimplemented_to_glob());
swap(&mut placeholder, &mut v[i]);
}
pub fn inline_metas(mut tcs: TCS, val: ValInfo) -> ValTCM {
use Neutral::*;
let info = val.loc;
let val = val.ast.try_map_neutral(&mut |neut| match neut {
Meta(mi) => tcs
.meta_context
.take_meta(mi)
.ok_or_else(|| TCE::MetaUnsolved(mi)),
e => Ok(Val::Neut(e)),
})?;
Ok((val.into_info(info), tcs))
}
fn check_decl(tcs: TCS, decl: AbsDecl) -> TCM {
debug_assert_eq!(tcs.gamma.len(), tcs.env.len());
let tcs = match decl {
AbsDecl::Impl(impl_abs, sign_dbi) => {
let sign = tcs.glob_type(sign_dbi);
let sign_cloned = sign.ast.clone();
let (val_fake, tcs) = tcs.check(&impl_abs, &sign_cloned)?;
let val = val_fake.map_ast(|ast| ast.generated_to_var());
let (val, mut tcs) = inline_metas(tcs, val)?;
tcs.env[sign_dbi.0] = val;
for i in sign_dbi.0..tcs.glob_len() {
unimplemented_to_glob(&mut tcs.env, i);
}
for i in sign_dbi.0 + 1..tcs.glob_len() {
unimplemented_to_glob(&mut tcs.gamma, i);
}
tcs
}
AbsDecl::Sign(sign_abs, self_index) => {
let loc = sign_abs.loc();
let (sign_fake, tcs) = tcs.check(&sign_abs, &TYPE_OMEGA)?;
let (sign_fake, mut tcs) = inline_metas(tcs, sign_fake)?;
let sign = sign_fake.map_ast(|ast| ast.generated_to_var());
let val_info = Val::fresh_unimplemented(self_index).into_info(loc);
tcs.env.push(val_info);
tcs.gamma.push(sign);
tcs
}
AbsDecl::Decl(impl_abs) => {
let (inferred, tcs) = tcs.infer(&impl_abs)?;
let (inferred, tcs) = inline_metas(tcs, inferred)?;
let (compiled, tcs) = tcs.evaluate(impl_abs);
let (compiled, mut tcs) = inline_metas(tcs, compiled)?;
let compiled = compiled.map_ast(|ast| ast.generated_to_var());
let inferred = inferred.map_ast(|ast| ast.generated_to_var());
tcs.env.push(compiled);
tcs.gamma.push(inferred);
tcs
}
};
require_local_emptiness(&tcs);
Ok(tcs)
}
impl TCS {
#[inline]
pub fn check_decls(self, decls: Vec<AbsDecl>) -> TCM {
check_decls(self, decls)
}
}