Changeset 8e48fca4


Ignore:
Timestamp:
Oct 29, 2021, 4:47:16 PM (9 months ago)
Author:
Andrew Beach <ajbeach@…>
Branches:
enum, forall-pointer-decay, master
Children:
e58e423
Parents:
0c577f7
Message:

Combined the code in FixMain? so it is all done with one pass.

Location:
src
Files:
3 edited

Legend:

Unmodified
Added
Removed
  • src/CodeGen/FixMain.cc

    r0c577f7 r8e48fca4  
    3232
    3333namespace CodeGen {
     34
     35namespace {
     36
     37struct FindMainCore {
     38        FunctionDecl * main_signature = nullptr;
     39
     40        void previsit( FunctionDecl * decl ) {
     41                if ( FixMain::isMain( decl ) ) {
     42                        if ( main_signature ) {
     43                                SemanticError( decl, "Multiple definition of main routine\n" );
     44                        }
     45                        main_signature = decl;
     46                }
     47        }
     48};
     49
     50}
     51
    3452        bool FixMain::replace_main = false;
    35         std::unique_ptr<FunctionDecl> FixMain::main_signature = nullptr;
    3653
    3754        template<typename container>
     
    4057        }
    4158
    42         void FixMain::registerMain(FunctionDecl* functionDecl)
    43         {
    44                 if(main_signature) {
    45                         SemanticError(functionDecl, "Multiple definition of main routine\n");
    46                 }
    47                 main_signature.reset( functionDecl->clone() );
    48         }
     59        void FixMain::fix( std::list< Declaration * > & translationUnit,
     60                        std::ostream &os, const char* bootloader_filename ) {
     61                PassVisitor< FindMainCore > main_finder;
     62                acceptAll( translationUnit, main_finder );
     63                FunctionDecl * main_signature = main_finder.pass.main_signature;
    4964
    50         void FixMain::fix(std::ostream &os, const char* bootloader_filename) {
    5165                if( main_signature ) {
    5266                        os << "static inline int invoke_main(int argc, char* argv[], char* envp[]) { (void)argc; (void)argv; (void)envp; return ";
    53                         main_signature->mangleName = SymTab::Mangler::mangle(main_signature.get());
     67                        main_signature->mangleName = SymTab::Mangler::mangle(main_signature);
    5468
    5569                        os << main_signature->get_scopedMangleName() << "(";
     
    121135}
    122136
    123 struct FindMainCore {
    124         void previsit( FunctionDecl * decl ) {
    125                 if ( FixMain::isMain( decl ) ) {
    126                         FixMain::registerMain( decl );
    127                 }
    128         }
    129 };
    130 
    131137} // namespace
    132138
     
    145151}
    146152
    147 void FixMain::findMain( std::list< Declaration * > & translationUnit ) {
    148         PassVisitor< FindMainCore > mainFinder;
    149         acceptAll( translationUnit, mainFinder );
    150 }
    151 
    152153};
  • src/CodeGen/FixMain.h

    r0c577f7 r8e48fca4  
    1010// Created On       : Thr Jan 12 14:11:09 2017
    1111// Last Modified By : Andrew Beach
    12 // Last Modified On : Fri Oct 29 14:49:00 2021
    13 // Update Count     : 7
     12// Last Modified On : Fri Oct 29 16:20:00 2021
     13// Update Count     : 8
    1414//
    1515
     
    2929
    3030namespace CodeGen {
    31         class FixMain {
    32           public :
    33                 static inline LinkageSpec::Spec mainLinkage() {
    34                         return replace_main ? LinkageSpec::Cforall : LinkageSpec::C;
    35                 }
    3631
    37                 static inline void setReplaceMain(bool val) {
    38                         replace_main = val;
    39                 }
     32class FixMain {
     33public :
     34        static inline LinkageSpec::Spec mainLinkage() {
     35                return replace_main ? LinkageSpec::Cforall : LinkageSpec::C;
     36        }
    4037
    41                 static void registerMain(FunctionDecl* val);
    42                 static bool isMain(FunctionDecl* decl);
    43                 static bool isMain(const ast::FunctionDecl * decl);
     38        static inline void setReplaceMain(bool val) {
     39                replace_main = val;
     40        }
    4441
    45                 static void fix(std::ostream &os, const char* bootloader_filename);
     42        static bool isMain(FunctionDecl* decl);
     43        static bool isMain(const ast::FunctionDecl * decl);
    4644
    47                 static void findMain( std::list< Declaration * > & decls );
     45        static void fix( std::list< Declaration * > & decls,
     46                        std::ostream &os, const char* bootloader_filename );
    4847
    49           private:
    50                 static bool replace_main;
    51                 static std::unique_ptr<FunctionDecl> main_signature;
    52         };
     48private:
     49        static bool replace_main;
     50};
     51
    5352} // namespace CodeGen
  • src/main.cc

    r0c577f7 r8e48fca4  
    1010// Created On       : Fri May 15 23:12:02 2015
    1111// Last Modified By : Andrew Beach
    12 // Last Modified On : Fri Oct 29 10:10:00 2021
    13 // Update Count     : 654
     12// Last Modified On : Fri Oct 29 16:34:00 2021
     13// Update Count     : 655
    1414//
    1515
     
    472472                PASS( "Code Gen", CodeGen::generate( translationUnit, *output, ! genproto, prettycodegenp, true, linemarks ) );
    473473
    474                 CodeGen::FixMain::findMain( translationUnit );
    475                 CodeGen::FixMain::fix( *output, (PreludeDirector + "/bootloader.c").c_str() );
     474                CodeGen::FixMain::fix( translationUnit, *output,
     475                                (PreludeDirector + "/bootloader.c").c_str() );
    476476                if ( output != &cout ) {
    477477                        delete output;
Note: See TracChangeset for help on using the changeset viewer.