Changes in src/CodeGen/FixMain.h [61efa42:c6b4432]

File:

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

src/CodeGen/FixMain.h

@@ -5 +5 @@
 // file "LICENCE" distributed with Cforall.
 //
-// FixMain.h -- Tools to change a Cforall main into a C main.
+// FixMain.h -- 
 //
 // Author           : Thierry Delisle
@@ -17 +17 @@
 
 #include <iosfwd>
+#include <memory>
+#include <list>
+
+#include "AST/LinkageSpec.hpp"
 
 namespace ast {
@@ -25 +29 @@
 namespace CodeGen {
 
-/// Is this function a program main function?
-bool isMain( const ast::FunctionDecl * decl );
+class FixMain {
+public :
+        static inline ast::Linkage::Spec getMainLinkage() {
+                return replace_main ? ast::Linkage::Cforall : ast::Linkage::C;
+        }
 
-/// Adjust the linkage of main functions.
-void fixMainLinkage( ast::TranslationUnit & transUnit, bool replaceMain );
+        static inline void setReplaceMain(bool val) {
+                replace_main = val;
+        }
 
-/// Add a wrapper around to run the Cforall main.
-void fixMainInvoke( ast::TranslationUnit & transUnit,
-                std::ostream & os, const char * bootloaderFilename );
+        static bool isMain(const ast::FunctionDecl * decl);
+
+        static void fix( ast::TranslationUnit & translationUnit,
+                        std::ostream &os, const char * bootloader_filename );
+
+private:
+        static bool replace_main;
+};
 
 } // namespace CodeGen