Changes in src/CodeGen/FixMain.h [60a8062:8e48fca4]

File:

• src/CodeGen/FixMain.h (modified) (2 diffs)

src/CodeGen/FixMain.h

@@ -9 +9 @@
 // Author           : Thierry Delisle
 // Created On       : Thr Jan 12 14:11:09 2017
-// Last Modified By : Peter A. Buhr
-// Last Modified On : Sun Feb 16 03:24:32 2020
-// Update Count     : 5
+// Last Modified By : Andrew Beach
+// Last Modified On : Fri Oct 29 16:20:00 2021
+// Update Count     : 8
 //
 
@@ -18 +18 @@
 #include <iosfwd>
 #include <memory>
+#include <list>
 
 #include "SynTree/LinkageSpec.h"
 
+class Declaration;
 class FunctionDecl;
+namespace ast {
+        class FunctionDecl;
+}
 
 namespace CodeGen {
-        class FixMain {
-          public :
-                static inline LinkageSpec::Spec mainLinkage() {
-                        return replace_main ? LinkageSpec::Cforall : LinkageSpec::C;
-                }
-                
-                static inline void setReplaceMain(bool val) {
-                        replace_main = val;
-                }
 
-                static void registerMain(FunctionDecl* val);
+class FixMain {
+public :
+        static inline LinkageSpec::Spec mainLinkage() {
+                return replace_main ? LinkageSpec::Cforall : LinkageSpec::C;
+        }
 
-                static void fix(std::ostream &os, const char* bootloader_filename);
+        static inline void setReplaceMain(bool val) {
+                replace_main = val;
+        }
 
-          private:
-                static bool replace_main;
-                static std::unique_ptr<FunctionDecl> main_signature;
-        };
+        static bool isMain(FunctionDecl* decl);
+        static bool isMain(const ast::FunctionDecl * decl);
+
+        static void fix( std::list< Declaration * > & decls,
+                        std::ostream &os, const char* bootloader_filename );
+
+private:
+        static bool replace_main;
+};
+
 } // namespace CodeGen