XX Wrap all the symbols from global scope into a function

Author: Thirumalai-Shaktivel

src/libasr/asr_scopes.cpp

@@ -134,4 +134,26 @@ std::string SymbolTable::get_unique_name(const std::string &name) {
     return unique_name;
 }
 
+void SymbolTable::move_symbols_from_global_scope(SymbolTable *fn_scope,
+        std::vector<std::string> &syms) {
+    for (auto &a : scope) {
+        switch (a.second->type) {
+            case (ASR::symbolType::Variable) : {
+                ASR::Variable_t *v = ASR::down_cast<ASR::Variable_t>(a.second);
+                v->m_parent_symtab = fn_scope;
+                fn_scope->add_symbol(a.first, (ASR::symbol_t *)v);
+                syms.push_back(a.first);
+                break;
+            }
+            case (ASR::symbolType::Function) : {
+                // Pass
+                break;
+            }
+            default : {
+                LCompilersException("Not implemented!");
+            };
+        }
+    }
+}
+
 } // namespace LFortran

src/libasr/asr_scopes.h

@@ -80,6 +80,9 @@ struct SymbolTable {
         size_t n_scope_names, char **m_scope_names);
 
     std::string get_unique_name(const std::string &name);
+
+    void move_symbols_from_global_scope(SymbolTable *fn_scope,
+        std::vector<std::string> &syms);
 };
 
 } // namespace LFortran

src/libasr/pass/global_stmts.cpp

@@ -100,6 +100,11 @@ void pass_wrap_global_stmts_into_function(Allocator &al,
                 throw LCompilersException("Unsupported type of global scope node");
             }
         }
+        std::vector<std::string> symbols;
+        unit.m_global_scope->move_symbols_from_global_scope(fn_scope, symbols);
+        for (auto &a: symbols) {
+            unit.m_global_scope->erase_symbol(a);
+        }
 
         if (return_var) {
             // The last item was an expression, create a function returning it