source: src/MakeLibCfa.cc @ d6c1dd0

//
// 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.
//
// MakeLibCfa.cc --
//
// Author           : Richard C. Bilson
// Created On       : Sat May 16 10:33:33 2015
// Last Modified By : Rob Schluntz
// Last Modified On : Fri Apr 22 13:54:15 2016
// Update Count     : 40
//

#include "MakeLibCfa.h"

#include <cassert>                 // for assert
#include <string>                   // for operator==, string

#include "CodeGen/OperatorTable.h"  // for OperatorInfo, operatorLookup, Ope...
#include "Common/SemanticError.h"   // for SemanticError
#include "Common/UniqueName.h"      // for UniqueName
#include "Parser/LinkageSpec.h"     // for Spec, Intrinsic, C
#include "SynTree/Declaration.h"    // for FunctionDecl, ObjectDecl, Declara...
#include "SynTree/Expression.h"     // for NameExpr, UntypedExpr, VariableExpr
#include "SynTree/Initializer.h"    // for SingleInit
#include "SynTree/Label.h"          // for Label
#include "SynTree/Statement.h"      // for CompoundStmt, ReturnStmt
#include "SynTree/Type.h"           // for FunctionType
#include "SynTree/Visitor.h"        // for acceptAll, Visitor

namespace LibCfa {
        class MakeLibCfa : public Visitor {
          public:
                void visit( FunctionDecl* funcDecl );
                void visit( ObjectDecl* objDecl );

                std::list< Declaration* > &get_newDecls() { return newDecls; }
          private:
                std::list< Declaration* > newDecls;
        };

        void makeLibCfa( std::list< Declaration* > &prelude ) {
                MakeLibCfa maker;
                acceptAll( prelude, maker );
                prelude.splice( prelude.end(), maker.get_newDecls() );
        }

        void MakeLibCfa::visit( FunctionDecl* origFuncDecl ) {
                if ( origFuncDecl->get_linkage() != LinkageSpec::Intrinsic ) return;
                if ( origFuncDecl->get_statements() ) return;

                FunctionDecl *funcDecl = origFuncDecl->clone();
                CodeGen::OperatorInfo opInfo;
                bool lookResult = CodeGen::operatorLookup( funcDecl->get_name(), opInfo );
                assert( lookResult );
                assert( ! funcDecl->get_statements() );
                UntypedExpr *newExpr = new UntypedExpr( new NameExpr( funcDecl->get_name() ) );
                UniqueName paramNamer( "_p" );
                std::list< DeclarationWithType* >::iterator param = funcDecl->get_functionType()->get_parameters().begin();
                assert( param != funcDecl->get_functionType()->get_parameters().end() );

                for ( ; param != funcDecl->get_functionType()->get_parameters().end(); ++param ) {
                        if ( (*param)->get_name() == "" ) {
                                (*param)->set_name( paramNamer.newName() );
                                (*param)->set_linkage( LinkageSpec::C );
                        }
                        newExpr->get_args().push_back( new VariableExpr( *param ) );
                } // for

                funcDecl->set_statements( new CompoundStmt( std::list< Label >() ) );
                newDecls.push_back( funcDecl );

                switch ( opInfo.type ) {
                  case CodeGen::OT_INDEX:
                  case CodeGen::OT_CALL:
                  case CodeGen::OT_PREFIX:
                  case CodeGen::OT_POSTFIX:
                  case CodeGen::OT_INFIX:
                  case CodeGen::OT_PREFIXASSIGN:
                  case CodeGen::OT_POSTFIXASSIGN:
                  case CodeGen::OT_INFIXASSIGN:
                  case CodeGen::OT_CTOR:
                  case CodeGen::OT_DTOR:
                                funcDecl->get_statements()->get_kids().push_back( new ReturnStmt( std::list< Label >(), newExpr ) );
                                break;
                  case CodeGen::OT_CONSTANT:
                  case CodeGen::OT_LABELADDRESS:
                        // there are no intrinsic definitions of 0/1 or label addresses as functions
                        assert( false );
                } // switch
        }

        void MakeLibCfa::visit( ObjectDecl* origObjDecl ) {
                if ( origObjDecl->get_linkage() != LinkageSpec::Intrinsic ) return;

                ObjectDecl *objDecl = origObjDecl->clone();
                assert( ! objDecl->get_init() );
                std::list< Expression* > noDesignators;
                objDecl->set_init( new SingleInit( new NameExpr( objDecl->get_name() ), false ) ); // cannot be constructed
                newDecls.push_back( objDecl );
        }
} // namespace LibCfa