forked from M-Labs/nac3
1
0
Fork 0

core: cleanup with Unifier::generate_var_id

This commit is contained in:
lyken 2024-06-14 09:42:04 +08:00
parent 6979843431
commit 8b9df7252f
1 changed files with 15 additions and 14 deletions

View File

@ -255,7 +255,7 @@ pub struct Unifier {
pub(crate) top_level: Option<Arc<TopLevelContext>>, pub(crate) top_level: Option<Arc<TopLevelContext>>,
pub(crate) unification_table: UnificationTable<Rc<TypeEnum>>, pub(crate) unification_table: UnificationTable<Rc<TypeEnum>>,
pub(crate) calls: Vec<Rc<Call>>, pub(crate) calls: Vec<Rc<Call>>,
var_id: u32, var_id_counter: u32,
unify_cache: HashSet<(Type, Type)>, unify_cache: HashSet<(Type, Type)>,
snapshot: Option<(usize, u32)>, snapshot: Option<(usize, u32)>,
primitive_store: Option<PrimitiveStore>, primitive_store: Option<PrimitiveStore>,
@ -273,7 +273,7 @@ impl Unifier {
pub fn new() -> Unifier { pub fn new() -> Unifier {
Unifier { Unifier {
unification_table: UnificationTable::new(), unification_table: UnificationTable::new(),
var_id: 0, var_id_counter: 0,
calls: Vec::new(), calls: Vec::new(),
unify_cache: HashSet::new(), unify_cache: HashSet::new(),
top_level: None, top_level: None,
@ -312,7 +312,7 @@ impl Unifier {
let lock = unifier.lock().unwrap(); let lock = unifier.lock().unwrap();
Unifier { Unifier {
unification_table: UnificationTable::from_send(&lock.0), unification_table: UnificationTable::from_send(&lock.0),
var_id: lock.1, var_id_counter: lock.1,
calls: lock.2.iter().map(|v| Rc::new(v.clone())).collect_vec(), calls: lock.2.iter().map(|v| Rc::new(v.clone())).collect_vec(),
top_level: None, top_level: None,
unify_cache: HashSet::new(), unify_cache: HashSet::new(),
@ -325,7 +325,7 @@ impl Unifier {
pub fn get_shared_unifier(&self) -> SharedUnifier { pub fn get_shared_unifier(&self) -> SharedUnifier {
Arc::new(Mutex::new(( Arc::new(Mutex::new((
self.unification_table.get_send(), self.unification_table.get_send(),
self.var_id, self.var_id_counter,
self.calls.iter().map(|v| v.as_ref().clone()).collect_vec(), self.calls.iter().map(|v| v.as_ref().clone()).collect_vec(),
))) )))
} }
@ -337,8 +337,7 @@ impl Unifier {
} }
pub fn add_record(&mut self, fields: Mapping<RecordKey, RecordField>) -> Type { pub fn add_record(&mut self, fields: Mapping<RecordKey, RecordField>) -> Type {
let id = TypeVarId(self.var_id + 1); let id = self.generate_var_id();
self.var_id += 1;
self.add_ty(TypeEnum::TVar { self.add_ty(TypeEnum::TVar {
id, id,
range: vec![], range: vec![],
@ -389,8 +388,7 @@ impl Unifier {
} }
pub fn get_fresh_rigid_var(&mut self, name: Option<StrRef>, loc: Option<Location>) -> TypeVar { pub fn get_fresh_rigid_var(&mut self, name: Option<StrRef>, loc: Option<Location>) -> TypeVar {
let id = TypeVarId(self.var_id + 1); let id = self.generate_var_id();
self.var_id += 1;
let ty = self.add_ty(TypeEnum::TRigidVar { id, name, loc }); let ty = self.add_ty(TypeEnum::TRigidVar { id, name, loc });
TypeVar { id, ty } TypeVar { id, ty }
} }
@ -417,8 +415,7 @@ impl Unifier {
) -> TypeVar { ) -> TypeVar {
let range = range.to_vec(); let range = range.to_vec();
let id = TypeVarId(self.var_id + 1); let id = self.generate_var_id();
self.var_id += 1;
let ty = self.add_ty(TypeEnum::TVar { let ty = self.add_ty(TypeEnum::TVar {
id, id,
range, range,
@ -437,8 +434,7 @@ impl Unifier {
name: Option<StrRef>, name: Option<StrRef>,
loc: Option<Location>, loc: Option<Location>,
) -> TypeVar { ) -> TypeVar {
let id = TypeVarId(self.var_id + 1); let id = self.generate_var_id();
self.var_id += 1;
let ty = self.add_ty(TypeEnum::TVar { let ty = self.add_ty(TypeEnum::TVar {
id, id,
range: vec![ty], range: vec![ty],
@ -1397,8 +1393,7 @@ impl Unifier {
if range.is_empty() { if range.is_empty() {
Err(()) Err(())
} else { } else {
let id = TypeVarId(self.var_id + 1); let id = self.generate_var_id();
self.var_id += 1;
let ty = TVar { let ty = TVar {
id, id,
fields: fields.clone(), fields: fields.clone(),
@ -1466,4 +1461,10 @@ impl Unifier {
} }
Err(TypeError::new(TypeErrorKind::IncompatibleRange(b, range.to_vec()), None)) Err(TypeError::new(TypeErrorKind::IncompatibleRange(b, range.to_vec()), None))
} }
/// Generate a new [`TypeVarId`] from [`Unifier::var_id_counter`]
fn generate_var_id(&mut self) -> TypeVarId {
self.var_id_counter += 1;
TypeVarId(self.var_id_counter)
}
} }