55 lines
1.8 KiB
Python
55 lines
1.8 KiB
Python
# Copyright (c) 2014-2024, Chris Pressey, Cat's Eye Technologies.
|
|
# This file is distributed under a 2-clause BSD license. See LICENSES/ dir.
|
|
# SPDX-License-Identifier: LicenseRef-BSD-2-Clause-X-SixtyPical
|
|
|
|
# encoding: UTF-8
|
|
|
|
from sixtypical.model import (
|
|
TYPE_BIT, TYPE_BYTE, LocationRef,
|
|
)
|
|
|
|
|
|
class SymEntry(object):
|
|
def __init__(self, ast_node, type_):
|
|
self.ast_node = ast_node
|
|
self.type_ = type_
|
|
|
|
def __repr__(self):
|
|
return "%s(%r, %r)" % (self.__class__.__name__, self.ast_node, self.type_)
|
|
|
|
|
|
class SymbolTable(object):
|
|
def __init__(self):
|
|
self.symbols = {} # symbol name -> SymEntry
|
|
self.locals = {} # routine name -> (symbol name -> SymEntry)
|
|
self.typedefs = {} # type name -> Type AST
|
|
self.consts = {} # const name -> ConstantRef
|
|
|
|
for name in ('a', 'x', 'y'):
|
|
self.symbols[name] = SymEntry(None, TYPE_BYTE)
|
|
for name in ('c', 'z', 'n', 'v'):
|
|
self.symbols[name] = SymEntry(None, TYPE_BIT)
|
|
|
|
def __str__(self):
|
|
return "Symbols: {}\nLocals: {}\nTypedefs: {}\nConsts: {}".format(self.symbols, self.locals, self.typedefs, self.consts)
|
|
|
|
def has_local(self, routine_name, name):
|
|
return name in self.locals.get(routine_name, {})
|
|
|
|
def fetch_global_type(self, name):
|
|
return self.symbols[name].type_
|
|
|
|
def fetch_local_type(self, routine_name, name):
|
|
return self.locals[routine_name][name].type_
|
|
|
|
def fetch_global_ref(self, name):
|
|
if name in self.symbols:
|
|
return LocationRef(name)
|
|
return None
|
|
|
|
def fetch_local_ref(self, routine_name, name):
|
|
routine_locals = self.locals.get(routine_name, {})
|
|
if name in routine_locals:
|
|
return LocationRef(name)
|
|
return None
|