Refactor symbols type 3

wolfv · Aug 3, 2023 · 4b7403b · 4b7403b
1 parent da80ea6
commit 4b7403b
Showing 1 changed file with 9 additions and 22 deletions.
diff --git a/src/libasr/pass/unique_symbols.cpp b/src/libasr/pass/unique_symbols.cpp
@@ -163,7 +163,8 @@ class SymbolRenameVisitor: public ASR::BaseWalkVisitor<SymbolRenameVisitor> {
         }
     }
 
-    void visit_AssociateBlock(const ASR::AssociateBlock_t &x) {
+    template <typename T>
+    void visit_symbols_3(T &x) {
         if (all_symbols_mangling || should_mangle) {
             ASR::symbol_t *sym = ASR::down_cast<ASR::symbol_t>((ASR::asr_t*)&x);
             sym_to_renamed[sym] = update_name(x.m_name);
@@ -173,34 +174,20 @@ class SymbolRenameVisitor: public ASR::BaseWalkVisitor<SymbolRenameVisitor> {
         }
     }
 
+    void visit_AssociateBlock(const ASR::AssociateBlock_t &x) {
+        visit_symbols_3(x);
+    }
+
     void visit_Block(const ASR::Block_t &x) {
-        if (all_symbols_mangling || should_mangle) {
-            ASR::symbol_t *sym = ASR::down_cast<ASR::symbol_t>((ASR::asr_t*)&x);
-            sym_to_renamed[sym] = update_name(x.m_name);
-        }
-        for (auto &a : x.m_symtab->get_scope()) {
-            this->visit_symbol(*a.second);
-        }
+        visit_symbols_3(x);
     }
 
     void visit_Requirement(const ASR::Requirement_t &x) {
-        if (all_symbols_mangling || should_mangle) {
-            ASR::symbol_t *sym = ASR::down_cast<ASR::symbol_t>((ASR::asr_t*)&x);
-            sym_to_renamed[sym] = update_name(x.m_name);
-        }
-        for (auto &a : x.m_symtab->get_scope()) {
-            this->visit_symbol(*a.second);
-        }
+        visit_symbols_3(x);
     }
 
     void visit_Template(const ASR::Template_t &x) {
-        if (all_symbols_mangling || should_mangle) {
-            ASR::symbol_t *sym = ASR::down_cast<ASR::symbol_t>((ASR::asr_t*)&x);
-            sym_to_renamed[sym] = update_name(x.m_name);
-        }
-        for (auto &a : x.m_symtab->get_scope()) {
-            this->visit_symbol(*a.second);
-        }
+        visit_symbols_3(x);
     }
 
 };