Changes in src/main.cc [626dbc10:3fe34ae]

File:

• src/main.cc (modified) (7 diffs)

src/main.cc

@@ -14 +14 @@
 //
 
+#include <memory>
 #include <iostream>
 #include <fstream>
@@ -65 +66 @@
         nopreludep = false,
         noprotop = false,
+        nomainp = false,
         parsep = false,
         resolvep = false,                                                                       // used in AlternativeFinder
@@ -77 +79 @@
 static void parse( FILE * input, LinkageSpec::Spec linkage, bool shouldExit = false );
 static void dump( list< Declaration * > & translationUnit, ostream & out = cout );
+
+bool translation_unit_nomain = true;
+std::unique_ptr<FunctionDecl> translation_unit_main_signature = nullptr;
 
 static void backtrace( int start ) {                                    // skip first N stack frames
@@ -154 +159 @@
 
         parse_cmdline( argc, argv, filename );                          // process command-line arguments
+        translation_unit_nomain = nomainp;
 
         try {
@@ -299 +305 @@
 
                 CodeGen::generate( translationUnit, *output, ! noprotop );
+
+                if( translation_unit_main_signature ) {
+                        *output << "int main(int argc, char** argv) { return ";
+
+                        *output << translation_unit_main_signature->get_scopedMangleName() << "(";
+                        if(translation_unit_main_signature->get_functionType()->get_parameters().size() != 0){
+                                *output << "argc, argv";
+                        }
+                        *output << ");";
+
+                        *output << " }\n";
+                }
 
                 if ( output != &cout ) {
@@ -360 +378 @@
 
         int c;
-        while ( (c = getopt_long( argc, argv, "abBcefglnpqrstTvyzD:F:", long_opts, &long_index )) != -1 ) {
+        while ( (c = getopt_long( argc, argv, "abBcefglmnpqrstTvyzD:F:", long_opts, &long_index )) != -1 ) {
                 switch ( c ) {
                   case Ast:
@@ -400 +418 @@
                   case 'p':                                                                             // generate prototypes for preamble functions
                         noprotop = true;
+                        break;
+                  case 'm':                                                                             // don't replace the main
+                        nomainp = true;
                         break;
                   case Parse: