| 1 | //
 | 
|---|
| 2 | // Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
 | 
|---|
| 3 | //
 | 
|---|
| 4 | // The contents of this file are covered under the licence agreement in the
 | 
|---|
| 5 | // file "LICENCE" distributed with Cforall.
 | 
|---|
| 6 | //
 | 
|---|
| 7 | // FunctionChecker.cc -- 
 | 
|---|
| 8 | //
 | 
|---|
| 9 | // Author           : Rodolfo G. Esteves
 | 
|---|
| 10 | // Created On       : Mon May 18 07:44:20 2015
 | 
|---|
| 11 | // Last Modified By : Peter A. Buhr
 | 
|---|
| 12 | // Last Modified On : Mon May 18 11:59:55 2015
 | 
|---|
| 13 | // Update Count     : 3
 | 
|---|
| 14 | //
 | 
|---|
| 15 | 
 | 
|---|
| 16 | #include "FunctionChecker.h"
 | 
|---|
| 17 | #include "FunctionFixer.h"
 | 
|---|
| 18 | #include "SemanticError.h"
 | 
|---|
| 19 | 
 | 
|---|
| 20 | #include <algorithm>
 | 
|---|
| 21 | #include <iostream>
 | 
|---|
| 22 | #include <cassert>
 | 
|---|
| 23 | 
 | 
|---|
| 24 | namespace Tuples {
 | 
|---|
| 25 |         using namespace std;
 | 
|---|
| 26 | 
 | 
|---|
| 27 |         void checkFunctions( std::list< Declaration * > translationUnit ) {
 | 
|---|
| 28 |                 FunctionChecker fchk( true );
 | 
|---|
| 29 |                 TupleDistrib td;
 | 
|---|
| 30 |                 FunctionFixer ff;
 | 
|---|
| 31 | 
 | 
|---|
| 32 |                 mutateAll( translationUnit , fchk );
 | 
|---|
| 33 |                 mutateAll( translationUnit , ff );
 | 
|---|
| 34 |                 mutateAll( translationUnit , td );
 | 
|---|
| 35 |                 return;
 | 
|---|
| 36 |         }
 | 
|---|
| 37 | 
 | 
|---|
| 38 |         FunctionChecker::FunctionChecker( bool _topLevel, UniqueName *_nameGen ) : topLevel( _topLevel ), nameGen( _nameGen ) {
 | 
|---|
| 39 |                 if ( topLevel ) {
 | 
|---|
| 40 |                         assert( ! nameGen );
 | 
|---|
| 41 |                         nameGen = new UniqueName("_MVR_");
 | 
|---|
| 42 |                 } else
 | 
|---|
| 43 |                         assert( nameGen );
 | 
|---|
| 44 |         }
 | 
|---|
| 45 | 
 | 
|---|
| 46 |         FunctionChecker::~FunctionChecker() {
 | 
|---|
| 47 |                 if ( topLevel ) {
 | 
|---|
| 48 |                         delete nameGen;
 | 
|---|
| 49 |                         nameGen = 0;
 | 
|---|
| 50 |                 }
 | 
|---|
| 51 |         }
 | 
|---|
| 52 | 
 | 
|---|
| 53 |         Statement* FunctionChecker::mutate( ExprStmt *exprStmt ) {
 | 
|---|
| 54 |                 exprStmt->set_expr( maybeMutate( exprStmt->get_expr(), *this ) );
 | 
|---|
| 55 |                 if ( ! tempExpr.empty() ) {
 | 
|---|
| 56 |                         assert ( ! temporaries.empty() );
 | 
|---|
| 57 |                         CompoundStmt *newBlock = new CompoundStmt( std::list< Label >() );
 | 
|---|
| 58 |                         // declarations
 | 
|---|
| 59 |                         for ( std::list< ObjectDecl *>::iterator d = temporaries.begin(); d != temporaries.end(); ++d )
 | 
|---|
| 60 |                                 newBlock->get_kids().push_back( new DeclStmt( std::list<Label>(), *d ) );
 | 
|---|
| 61 |                         // new expression statements
 | 
|---|
| 62 |                         for ( std::list< Expression *>::iterator e = tempExpr.begin(); e != tempExpr.end(); ++e )
 | 
|---|
| 63 |                                 newBlock->get_kids().push_back( new ExprStmt( std::list<Label>(), *e ) );
 | 
|---|
| 64 | 
 | 
|---|
| 65 |                         newBlock->get_kids().push_back( exprStmt );
 | 
|---|
| 66 |                         return newBlock;
 | 
|---|
| 67 |                 } else
 | 
|---|
| 68 |                         return exprStmt;
 | 
|---|
| 69 |         }
 | 
|---|
| 70 | 
 | 
|---|
| 71 |         Expression* FunctionChecker::mutate( ApplicationExpr *applicationExpr ) {
 | 
|---|
| 72 |                 if ( topLevel )
 | 
|---|
| 73 |                         ; // In top level of Functionchecker
 | 
|---|
| 74 | 
 | 
|---|
| 75 |                 if ( applicationExpr->get_results().size() > 1 ) {
 | 
|---|
| 76 |                         for ( std::list< Type *>::iterator res = applicationExpr->get_results().begin(); res != applicationExpr->get_results().end(); res++ )
 | 
|---|
| 77 |                                 temporaries.push_back( new ObjectDecl( nameGen->newName(),Declaration::Auto,LinkageSpec::AutoGen, 0, (*res )->clone(), 0 ) );
 | 
|---|
| 78 | 
 | 
|---|
| 79 |                         assert( ! temporaries.empty() );
 | 
|---|
| 80 |                 }
 | 
|---|
| 81 | 
 | 
|---|
| 82 |                 applicationExpr->set_function(  maybeMutate( applicationExpr->get_function(), *this ) );
 | 
|---|
| 83 | 
 | 
|---|
| 84 |                 std::list< Expression * > newArgs;
 | 
|---|
| 85 |                 for ( std::list< Expression *>::iterator e = applicationExpr->get_args().begin(); e != applicationExpr->get_args().end(); ++e ) {
 | 
|---|
| 86 |                         FunctionChecker rec( false, nameGen );
 | 
|---|
| 87 |                         (*e )->acceptMutator( rec );
 | 
|---|
| 88 | 
 | 
|---|
| 89 |                         if ( ! rec.temporaries.empty() ) {
 | 
|---|
| 90 |                                 TupleExpr *lhs = new TupleExpr;
 | 
|---|
| 91 |                                 std::list< Expression * > &tmem = lhs->get_exprs();
 | 
|---|
| 92 |                                 for ( std::list<ObjectDecl *>::iterator d = rec.temporaries.begin();  d != rec.temporaries.end(); ++d ) {
 | 
|---|
| 93 |                                         tmem.push_back( new VariableExpr( *d ) );
 | 
|---|
| 94 |                                         newArgs.push_back( new VariableExpr( *d ) );
 | 
|---|
| 95 |                                 }
 | 
|---|
| 96 | 
 | 
|---|
| 97 |                                 // construct tuple assignment
 | 
|---|
| 98 |                                 std::list<Expression *> args;
 | 
|---|
| 99 |                                 args.push_back( new AddressExpr( lhs ) );
 | 
|---|
| 100 |                                 args.push_back( *e );
 | 
|---|
| 101 |                                 tempExpr.push_back( new UntypedExpr( new NameExpr("?=?"), args ) );
 | 
|---|
| 102 | 
 | 
|---|
| 103 |                                 temporaries.splice( temporaries.end(), rec.temporaries );
 | 
|---|
| 104 |                         } else
 | 
|---|
| 105 |                                 newArgs.push_back( *e );
 | 
|---|
| 106 |                         // percolate to recursive calls
 | 
|---|
| 107 |                 }
 | 
|---|
| 108 | 
 | 
|---|
| 109 |                 applicationExpr->get_args().clear();
 | 
|---|
| 110 |                 std::copy( newArgs.begin(), newArgs.end(), back_inserter( applicationExpr->get_args()) );
 | 
|---|
| 111 | 
 | 
|---|
| 112 |                 return applicationExpr;
 | 
|---|
| 113 |         }
 | 
|---|
| 114 | 
 | 
|---|
| 115 |         Expression* TupleDistrib::mutate( UntypedExpr *expr ) {
 | 
|---|
| 116 |                 if (  NameExpr *assgnop = dynamic_cast< NameExpr * >( expr->get_function()) ) {
 | 
|---|
| 117 |                         if ( assgnop->get_name() == std::string("?=?") ) {
 | 
|---|
| 118 |                                 std::list<Expression *> &args = expr->get_args();
 | 
|---|
| 119 |                                 assert( args.size() == 2 );
 | 
|---|
| 120 |                                 //if args.front() points to a tuple and if args.back() is already resolved
 | 
|---|
| 121 |                                 if ( AddressExpr *addr = dynamic_cast<AddressExpr *>( args.front()) )
 | 
|---|
| 122 |                                         if ( TupleExpr *lhs = dynamic_cast<TupleExpr *>( addr->get_arg()) )
 | 
|---|
| 123 |                                                 if ( ApplicationExpr *rhs = dynamic_cast<ApplicationExpr *>( args.back() ) ) {
 | 
|---|
| 124 |                                                         for ( std::list<Expression *>::iterator tc = lhs->get_exprs().begin(); tc != lhs->get_exprs().end(); ++tc )
 | 
|---|
| 125 |                                                                 rhs->get_args().push_back( new AddressExpr( *tc ) );
 | 
|---|
| 126 |                                                         return rhs; // XXX
 | 
|---|
| 127 |                                                 } // if
 | 
|---|
| 128 |                         } else
 | 
|---|
| 129 |                                 assert( false ); // It's not an assignment, shouldn't be here
 | 
|---|
| 130 |                 } // if
 | 
|---|
| 131 |                 return expr;
 | 
|---|
| 132 |         }
 | 
|---|
| 133 | 
 | 
|---|
| 134 | }
 | 
|---|
| 135 | 
 | 
|---|
| 136 | // namespace Tuples
 | 
|---|
| 137 | // Local Variables: //
 | 
|---|
| 138 | // tab-width: 4 //
 | 
|---|
| 139 | // mode: c++ //
 | 
|---|
| 140 | // compile-command: "make install" //
 | 
|---|
| 141 | // End: //
 | 
|---|