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

File:

• src/CodeGen/FixMain.h (modified) (2 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 "SynTree/LinkageSpec.h"
+
+class Declaration;
+class FunctionDecl;
 namespace ast {
         class FunctionDecl;
-        class TranslationUnit;
 }
 
 namespace CodeGen {
 
-/// Is this function a program main function?
-bool isMain( const ast::FunctionDecl * decl );
+class FixMain {
+public :
+        static inline LinkageSpec::Spec mainLinkage() {
+                return replace_main ? LinkageSpec::Cforall : LinkageSpec::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(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