source: src/MakeLibCfa.cc @ a3b5cac

//
// 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 : Peter A. Buhr
// Last Modified On : Sat May 16 10:40:44 2015
// Update Count     : 13
// 

#include "MakeLibCfa.h"
#include "SynTree/Visitor.h"
#include "SynTree/Declaration.h"
#include "SynTree/Type.h"
#include "SynTree/Expression.h"
#include "SynTree/Statement.h"
#include "SynTree/Initializer.h"
#include "CodeGen/OperatorTable.h"
#include "UniqueName.h"

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;
  
                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() );

                if ( (*param)->get_name() == "" ) {
                        (*param)->set_name( paramNamer.newName() );
                        (*param)->set_linkage( LinkageSpec::C );
                } // if

                switch ( opInfo.type ) {
                  case CodeGen::OT_INDEX:
                  case CodeGen::OT_CALL:
                  case CodeGen::OT_PREFIX:
                  case CodeGen::OT_POSTFIX:
                  case CodeGen::OT_INFIX:
                        newExpr->get_args().push_back( new VariableExpr( *param ) );
                        break;
                  case CodeGen::OT_PREFIXASSIGN:
                  case CodeGen::OT_POSTFIXASSIGN:
                  case CodeGen::OT_INFIXASSIGN:
                        {
                                newExpr->get_args().push_back( new VariableExpr( *param ) );
                                // UntypedExpr *deref = new UntypedExpr( new NameExpr( "*?" ) );
                                // deref->get_args().push_back( new VariableExpr( *param ) );
                                // newExpr->get_args().push_back( deref );
                                break;
                        }
                  case CodeGen::OT_CONSTANT:
                        assert( false );
                } // switch

                for ( param++; 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 >() ) );
                funcDecl->get_statements()->get_kids().push_back( new ReturnStmt( std::list< Label >(), newExpr ) );
                newDecls.push_back( funcDecl );
        }

        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() ), noDesignators ) );
                newDecls.push_back( objDecl );
        }
} // namespace LibCfa

// Local Variables: //
// tab-width: 4 //
// mode: c++ //
// compile-command: "make install" //
// End: //