forked from M-Labs/nac3
1
0
Fork 0

starting cleaning up and further add Arc<Mutex>

This commit is contained in:
ychenfo 2021-08-11 17:35:23 +08:00
parent d46a4b2d38
commit 17ee8fe6d0
1 changed files with 38 additions and 27 deletions

View File

@ -63,19 +63,27 @@ pub fn name_mangling(mut class_name: String, method_name: &str) -> String {
class_name class_name
} }
// like adding some info on top of the TopLevelDef for later parsing the class bases, method,
// and function sigatures
pub struct TopLevelDefInfo { pub struct TopLevelDefInfo {
// like adding some info on top of the TopLevelDef for later parsing the class bases, method, // the definition entry
// and function sigatures def: TopLevelDef,
def: TopLevelDef, // the definition entry // the entry in the top_level unifier
ty: Type, // the entry in the top_level unifier ty: Type,
ast: Option<ast::Stmt<()>>, // the ast submitted by applications, primitives and class methods will have None value here // the ast submitted by applications, primitives and
// resolver: Option<&'a dyn SymbolResolver> // the resolver // class methods will have None value here
ast: Option<ast::Stmt<()>>,
} }
pub struct TopLevelComposer { pub struct TopLevelComposer {
pub definition_list: Vec<TopLevelDefInfo>, // list of top level definitions and their info
pub definition_list: RwLock<Vec<TopLevelDefInfo>>,
// primitive store
pub primitives: PrimitiveStore, pub primitives: PrimitiveStore,
// start as a primitive unifier, will add more top_level defs inside
pub unifier: Unifier, pub unifier: Unifier,
// class method to definition id
pub class_method_to_def_id: HashMap<String, DefinitionId>,
} }
impl TopLevelComposer { impl TopLevelComposer {
@ -111,8 +119,11 @@ impl TopLevelComposer {
(primitives, unifier) (primitives, unifier)
} }
pub fn new() -> Self { // return a composer and things to make a "primitive" symbol resolver, so that the symbol
// resolver can later figure out primitive type definitions when passed a primitive type name
pub fn new() -> (Vec<(String, DefinitionId, Type)>, Self) {
let primitives = Self::make_primitives(); let primitives = Self::make_primitives();
// the def list including the entries of primitive info
let definition_list: Vec<TopLevelDefInfo> = vec![ let definition_list: Vec<TopLevelDefInfo> = vec![
TopLevelDefInfo { TopLevelDefInfo {
def: Self::make_top_level_class_def(0, None), def: Self::make_top_level_class_def(0, None),
@ -139,8 +150,20 @@ impl TopLevelComposer {
ast: None, ast: None,
ty: primitives.0.none, ty: primitives.0.none,
}, },
]; // the entries for primitive types ];
TopLevelComposer { definition_list, primitives: primitives.0, unifier: primitives.1 } let composer = TopLevelComposer {
definition_list: definition_list.into(),
primitives: primitives.0,
unifier: primitives.1,
class_method_to_def_id: Default::default(),
};
(vec![
("int32".into(), DefinitionId(0), composer.primitives.int32),
("int64".into(), DefinitionId(1), composer.primitives.int64),
("float".into(), DefinitionId(2), composer.primitives.float),
("bool".into(), DefinitionId(3), composer.primitives.bool),
("none".into(), DefinitionId(4), composer.primitives.none),
], composer)
} }
/// already include the definition_id of itself inside the ancestors vector /// already include the definition_id of itself inside the ancestors vector
@ -172,18 +195,6 @@ impl TopLevelComposer {
} }
} }
// like to make and return a "primitive" symbol resolver? so that the symbol resolver
// can later figure out primitive type definitions when passed a primitive type name
pub fn get_primitives_definition(&self) -> Vec<(String, DefinitionId, Type)> {
vec![
("int32".into(), DefinitionId(0), self.primitives.int32),
("int64".into(), DefinitionId(1), self.primitives.int64),
("float".into(), DefinitionId(2), self.primitives.float),
("bool".into(), DefinitionId(3), self.primitives.bool),
("none".into(), DefinitionId(4), self.primitives.none),
]
}
pub fn register_top_level( pub fn register_top_level(
&mut self, &mut self,
ast: ast::Stmt<()>, ast: ast::Stmt<()>,
@ -192,7 +203,8 @@ impl TopLevelComposer {
match &ast.node { match &ast.node {
ast::StmtKind::ClassDef { name, body, .. } => { ast::StmtKind::ClassDef { name, body, .. } => {
let class_name = name.to_string(); let class_name = name.to_string();
let class_def_id = self.definition_list.len(); let def_list = self.definition_list.write();
let class_def_id = def_list.len();
// add the class to the unifier // add the class to the unifier
let ty = self.unifier.add_ty(TypeEnum::TObj { let ty = self.unifier.add_ty(TypeEnum::TObj {
@ -241,10 +253,10 @@ impl TopLevelComposer {
}) })
// FIXME: should we return this to the symbol resolver?, should be yes // FIXME: should we return this to the symbol resolver?, should be yes
} }
} else { }
} // else do nothing
} }
// add to the definition list
// add the class to the definition list
self.definition_list.push(TopLevelDefInfo { self.definition_list.push(TopLevelDefInfo {
def: Self::make_top_level_class_def(class_def_id, resolver), def: Self::make_top_level_class_def(class_def_id, resolver),
ast: Some(ast), ast: Some(ast),
@ -439,4 +451,3 @@ impl TopLevelComposer {
Ok(()) Ok(())
} }
} }