diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/resolver.rs | 8 | ||||
-rw-r--r-- | src/source_unit.rs | 23 |
2 files changed, 27 insertions, 4 deletions
diff --git a/src/resolver.rs b/src/resolver.rs index 23dc73a..37d0911 100644 --- a/src/resolver.rs +++ b/src/resolver.rs @@ -71,15 +71,15 @@ impl Resolver { match symbol.role { SymbolRole::Definition(_) => { // Check if the symbol has already been defined. - let equal = |d: &TrackedSymbol| { &d.symbol.name == &symbol.name }; + let equal = |d: &TrackedSymbol| { &d.symbol == &symbol }; if let Some(original) = self.definitions.iter().position(equal) { let definition = TrackedSymbol { symbol, source_id, source_role }; let redefinition = (definition, original); self.redefinitions.push(redefinition); } else { // Resolve all unresolved references that match this symbol. - let equal = |s: &mut TrackedSymbol| s.symbol.name == symbol.name; - for symbol in self.unresolved.extract_if(equal) { + let defines = |r: &mut TrackedSymbol| symbol.defines(&r.symbol); + for symbol in self.unresolved.extract_if(defines) { self.resolved.push(symbol); } let definition = TrackedSymbol { symbol, source_id, source_role }; @@ -88,7 +88,7 @@ impl Resolver { } SymbolRole::Reference => { let reference = TrackedSymbol { symbol, source_id, source_role }; - match self.definitions.contains(&reference) { + match self.definitions.iter().any(|d| d.symbol.defines(&reference.symbol)) { true => self.resolved.push(reference), false => self.unresolved.push(reference), } diff --git a/src/source_unit.rs b/src/source_unit.rs index 7ee6993..bc6d4ab 100644 --- a/src/source_unit.rs +++ b/src/source_unit.rs @@ -115,6 +115,29 @@ pub struct Symbol { pub role: SymbolRole, } +impl Symbol { + /// True if this symbol is a valid definition for a reference symbol + pub fn defines(&self, reference: &Symbol) -> bool { + self.name == reference.name && + std::iter::zip(&self.namespace, &reference.namespace).all(|(a, b)| a == b) + } +} + +impl PartialEq for Symbol { + fn eq(&self, other: &Symbol) -> bool { + self.name == other.name && self.namespace == other.namespace + } +} + +impl std::fmt::Debug for Symbol { + fn fmt(&self, f: &mut std::fmt::Formatter) -> Result<(), std::fmt::Error> { + for name in &self.namespace { + write!(f, "{name}:")? + } + write!(f, "{}", self.name) + } +} + #[derive(Copy, Clone, Debug, PartialEq)] pub enum SymbolRole { Definition(DefinitionType), |