source: src/CodeGen/FixMain.h @ f42fc13

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// FixMain.h -- 
//
// Author           : Thierry Delisle
// Created On       : Thr Jan 12 14:11:09 2017
// Last Modified By : Andrew Beach
// Last Modified On : Fri Oct 29 11:07:00 2021
// Update Count     : 6
//

#pragma once

#include <iosfwd>
#include <memory>
#include <list>

#include "SynTree/LinkageSpec.h"

class Declaration;
class FunctionDecl;

namespace CodeGen {
        class FixMain {
          public :
                static inline LinkageSpec::Spec mainLinkage() {
                        return replace_main ? LinkageSpec::Cforall : LinkageSpec::C;
                }
                
                static inline void setReplaceMain(bool val) {
                        replace_main = val;
                }

                static void registerMain(FunctionDecl* val);
                static bool isMain(FunctionDecl* decl);

                static void fix(std::ostream &os, const char* bootloader_filename);

                static void findMain( std::list< Declaration * > & decls );

          private:
                static bool replace_main;
                static std::unique_ptr<FunctionDecl> main_signature;
        };
} // namespace CodeGen