Implement add_generic_procedures()

wangvei · Mar 5, 2021 · 7b62892 · 7b62892
1 parent 3662ea5
commit 7b62892
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/src/lfortran/semantics/ast_to_asr.cpp b/src/lfortran/semantics/ast_to_asr.cpp
@@ -161,6 +161,16 @@ class SymbolTableVisitor : public AST::BaseVisitor<SymbolTableVisitor>
     SymbolTableVisitor(Allocator &al, SymbolTable *symbol_table)
         : al{al}, current_scope{symbol_table} { }
 
+    ASR::symbol_t* resolve_symbol(const Location &loc, const char* id) {
+        SymbolTable *scope = current_scope;
+        std::string sub_name = id;
+        ASR::symbol_t *sub = scope->resolve_symbol(sub_name);
+        if (!sub) {
+            throw SemanticError("Symbol '" + sub_name + "' not declared", loc);
+        }
+        return sub;
+    }
+
     void visit_TranslationUnit(const AST::TranslationUnit_t &x) {
         if (!current_scope) {
             current_scope = al.make_new<SymbolTable>(nullptr);
@@ -188,6 +198,7 @@ class SymbolTableVisitor : public AST::BaseVisitor<SymbolTableVisitor>
         for (size_t i=0; i<x.n_contains; i++) {
             visit_program_unit(*x.m_contains[i]);
         }
+        add_generic_procedures();
         asr = ASR::make_Module_t(
             al, x.base.base.loc,
             /* a_symtab */ current_scope,
@@ -404,6 +415,32 @@ class SymbolTableVisitor : public AST::BaseVisitor<SymbolTableVisitor>
         }
     }
 
+    void add_generic_procedures() {
+        for (auto &proc : generic_procedures) {
+            Location loc;
+            loc.first_line = 1;
+            loc.last_line = 1;
+            loc.first_column = 1;
+            loc.last_column = 1;
+            Str s;
+            s.from_str_view(proc.first);
+            char *generic_name = s.c_str(al);
+            Vec<ASR::symbol_t*> symbols;
+            symbols.reserve(al, proc.second.size());
+            for (auto &pname : proc.second) {
+                ASR::symbol_t *x;
+                Str s;
+                s.from_str_view(pname);
+                char *name = s.c_str(al);
+                x = resolve_symbol(loc, name);
+                symbols.push_back(al, x);
+            }
+            ASR::asr_t *v = ASR::make_GenericProcedure_t(al, loc,
+                generic_name, symbols.p, symbols.size());
+            current_scope->scope[proc.first] = ASR::down_cast<ASR::symbol_t>(v);
+        }
+    }
+
     void visit_Use(const AST::Use_t &x) {
         std::string msym = x.m_module;
         ASR::symbol_t *t = current_scope->parent->resolve_symbol(msym);