modelverse/interface/HUTN/hutn_compiler/declare_functions_visitor.py

import hutn_compiler.symbol_table as st
import hutn_compiler.types_mv as types_mv
from hutn_compiler.visitor import Visitor
from hutn_compiler.hutnparser import Tree

# Declare the function but do not visit its body
class DeclareFunctionsVisitor(Visitor):
    def __init__(self, symbol_table, inputfiles):
        Visitor.__init__(self, [])
        self.symbol_table = symbol_table
        self.inputfiles = inputfiles

    def compute_parameter_types(self, tree):
        parameter_types = []
        for parameter in Tree.get_children(tree, 'parameter'):
            type_specifier = Tree.get_text(Tree.get_children(parameter, "type_specifier")[0])
            parameter_types.append(types_mv.string_to_type(type_specifier))
        return parameter_types

    def compute_func_type(self, tree):
        self.visit(Tree.get_tail(tree)[0])
        func_type = self.get_type(Tree.get_tail(tree)[0])
        return func_type

    def visit_funcdecl(self, tree):
        func_name = Tree.get_text(Tree.get_children(tree, "func_name")[0])
        func_type = self.compute_func_type(tree)
        parameter_types = self.compute_parameter_types(tree)
        self.set_type(tree, func_type)

        #TODO should check type compatibility if declared multiple times
        try:
            s = st.Symbol(func_name, func_type, is_global=True, params=parameter_types)
            self.symbol_table.add(s)
        except Exception:
            if "COLON" in Tree.get_tail(tree) or "ASSIGN" in Tree.get_tail(tree):
                raise RuntimeError(
                    "{}:{}:{}: error: redeclaration of '{}'".format(
                        self.inputfiles[0], tree['startpos']['line'],
                        tree['startpos']['column'], func_name))
        self.set_symbol(tree, s)

    def visit_func_type(self, tree):
        self.set_type(tree, types_mv.string_to_type(Tree.get_text(tree)))