Changeset 0c577f7 for src/CodeGen/FixMain.cc

Timestamp:

Oct 29, 2021, 4:03:07 PM (3 years ago)

Author:

Andrew Beach <ajbeach@…>

Branches:

ADT, ast-experimental, enum, forall-pointer-decay, master, pthread-emulation, qualifiedEnum

Children:

8e48fca4

Parents:

f42fc13

Message:

Implemented new AST version of the Fix Names pass.

File:

• src/CodeGen/FixMain.cc (modified) (3 diffs)

src/CodeGen/FixMain.cc

@@ -22 +22 @@
 #include <string>                  // for operator<<
 
+#include "AST/Decl.hpp"
+#include "AST/Type.hpp"
 #include "Common/PassVisitor.h"
 #include "Common/SemanticError.h"  // for SemanticError
@@ -105 +107 @@
 }
 
+bool is_main( const std::string & mangled_name ) {
+        // This breaks if you move it out of the function.
+        static const std::string mangled_mains[] = {
+                mangled_0_argument_main(),
+                mangled_2_argument_main(),
+                //mangled_3_argument_main(),
+        };
+
+        for ( auto main_name : mangled_mains ) {
+                if ( main_name == mangled_name ) return true;
+        }
+        return false;
+}
+
 struct FindMainCore {
         void previsit( FunctionDecl * decl ) {
@@ -116 +132 @@
 
 bool FixMain::isMain( FunctionDecl * decl ) {
-        // This breaks if you move it out of the function.
-        static const std::string mangled_mains[] = {
-                mangled_0_argument_main(),
-                mangled_2_argument_main(),
-                //mangled_3_argument_main(),
-        };
-
         if ( std::string("main") != decl->name ) {
                 return false;
         }
-        auto mangled_name = SymTab::Mangler::mangle( decl, true, true );
-        for ( auto main_name : mangled_mains ) {
-                if ( main_name == mangled_name ) return true;
+        return is_main( SymTab::Mangler::mangle( decl, true, true ) );
+}
+
+bool FixMain::isMain( const ast::FunctionDecl * decl ) {
+        if ( std::string("main") != decl->name ) {
+                return false;
         }
-        return false;
+        return is_main( Mangle::mangle( decl, Mangle::Type ) );
 }