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

File:

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

src/CodeGen/FixMain.h

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