source: src/ResolvExpr/AlternativeFinder.cc@ 6fd1955

ADT aaron-thesis arm-eh ast-experimental cleanup-dtors enum forall-pointer-decay jacob/cs343-translation jenkins-sandbox new-ast new-ast-unique-expr pthread-emulation qualifiedEnum
Last change on this file since 6fd1955 was da6032af, checked in by Aaron Moss <a3moss@…>, 7 years ago

Remove assertion cost from cost model

  • Property mode set to 100644
File size: 71.2 KB
Line 
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// AlternativeFinder.cc --
8//
9// Author : Richard C. Bilson
10// Created On : Sat May 16 23:52:08 2015
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Thu Nov 1 21:00:56 2018
13// Update Count : 35
14//
15
16#include <algorithm> // for copy
17#include <cassert> // for strict_dynamic_cast, assert, assertf
18#include <cstddef> // for size_t
19#include <iostream> // for operator<<, cerr, ostream, endl
20#include <iterator> // for back_insert_iterator, back_inserter
21#include <list> // for _List_iterator, list, _List_const_...
22#include <map> // for _Rb_tree_iterator, map, _Rb_tree_c...
23#include <memory> // for allocator_traits<>::value_type, unique_ptr
24#include <utility> // for pair
25#include <vector> // for vector
26
27#include "CompilationState.h" // for resolvep
28#include "Alternative.h" // for AltList, Alternative
29#include "AlternativeFinder.h"
30#include "Common/SemanticError.h" // for SemanticError
31#include "Common/utility.h" // for deleteAll, printAll, CodeLocation
32#include "Cost.h" // for Cost, Cost::zero, operator<<, Cost...
33#include "ExplodedActual.h" // for ExplodedActual
34#include "InitTweak/InitTweak.h" // for getFunctionName
35#include "RenameVars.h" // for RenameVars, global_renamer
36#include "ResolveAssertions.h" // for resolveAssertions
37#include "ResolveTypeof.h" // for resolveTypeof
38#include "Resolver.h" // for resolveStmtExpr
39#include "SymTab/Indexer.h" // for Indexer
40#include "SymTab/Mangler.h" // for Mangler
41#include "SymTab/Validate.h" // for validateType
42#include "SynTree/Constant.h" // for Constant
43#include "SynTree/Declaration.h" // for DeclarationWithType, TypeDecl, Dec...
44#include "SynTree/Expression.h" // for Expression, CastExpr, NameExpr
45#include "SynTree/Initializer.h" // for SingleInit, operator<<, Designation
46#include "SynTree/SynTree.h" // for UniqueId
47#include "SynTree/Type.h" // for Type, FunctionType, PointerType
48#include "Tuples/Explode.h" // for explode
49#include "Tuples/Tuples.h" // for isTtype, handleTupleAssignment
50#include "Unify.h" // for unify
51#include "typeops.h" // for adjustExprType, polyCost, castCost
52
53#define PRINT( text ) if ( resolvep ) { text }
54//#define DEBUG_COST
55
56using std::move;
57
58/// copies any copyable type
59template<typename T>
60T copy(const T& x) { return x; }
61
62namespace ResolvExpr {
63 struct AlternativeFinder::Finder : public WithShortCircuiting {
64 Finder( AlternativeFinder & altFinder ) : altFinder( altFinder ), indexer( altFinder.indexer ), alternatives( altFinder.alternatives ), env( altFinder.env ), targetType( altFinder.targetType ) {}
65
66 void previsit( BaseSyntaxNode * ) { visit_children = false; }
67
68 void postvisit( ApplicationExpr * applicationExpr );
69 void postvisit( UntypedExpr * untypedExpr );
70 void postvisit( AddressExpr * addressExpr );
71 void postvisit( LabelAddressExpr * labelExpr );
72 void postvisit( CastExpr * castExpr );
73 void postvisit( VirtualCastExpr * castExpr );
74 void postvisit( UntypedMemberExpr * memberExpr );
75 void postvisit( MemberExpr * memberExpr );
76 void postvisit( NameExpr * variableExpr );
77 void postvisit( VariableExpr * variableExpr );
78 void postvisit( ConstantExpr * constantExpr );
79 void postvisit( SizeofExpr * sizeofExpr );
80 void postvisit( AlignofExpr * alignofExpr );
81 void postvisit( UntypedOffsetofExpr * offsetofExpr );
82 void postvisit( OffsetofExpr * offsetofExpr );
83 void postvisit( OffsetPackExpr * offsetPackExpr );
84 void postvisit( AttrExpr * attrExpr );
85 void postvisit( LogicalExpr * logicalExpr );
86 void postvisit( ConditionalExpr * conditionalExpr );
87 void postvisit( CommaExpr * commaExpr );
88 void postvisit( ImplicitCopyCtorExpr * impCpCtorExpr );
89 void postvisit( ConstructorExpr * ctorExpr );
90 void postvisit( RangeExpr * rangeExpr );
91 void postvisit( UntypedTupleExpr * tupleExpr );
92 void postvisit( TupleExpr * tupleExpr );
93 void postvisit( TupleIndexExpr * tupleExpr );
94 void postvisit( TupleAssignExpr * tupleExpr );
95 void postvisit( UniqueExpr * unqExpr );
96 void postvisit( StmtExpr * stmtExpr );
97 void postvisit( UntypedInitExpr * initExpr );
98 void postvisit( InitExpr * initExpr );
99 void postvisit( DeletedExpr * delExpr );
100 void postvisit( GenericExpr * genExpr );
101
102 /// Adds alternatives for anonymous members
103 void addAnonConversions( const Alternative & alt );
104 /// Adds alternatives for member expressions, given the aggregate, conversion cost for that aggregate, and name of the member
105 template< typename StructOrUnionType > void addAggMembers( StructOrUnionType *aggInst, Expression *expr, const Alternative &alt, const Cost &newCost, const std::string & name );
106 /// Adds alternatives for member expressions where the left side has tuple type
107 void addTupleMembers( TupleType *tupleType, Expression *expr, const Alternative &alt, const Cost &newCost, Expression *member );
108 /// Adds alternatives for offsetof expressions, given the base type and name of the member
109 template< typename StructOrUnionType > void addOffsetof( StructOrUnionType *aggInst, const std::string &name );
110 /// Takes a final result and checks if its assertions can be satisfied
111 template<typename OutputIterator>
112 void validateFunctionAlternative( const Alternative &func, ArgPack& result, const std::vector<ArgPack>& results, OutputIterator out );
113 /// Finds matching alternatives for a function, given a set of arguments
114 template<typename OutputIterator>
115 void makeFunctionAlternatives( const Alternative &func, FunctionType *funcType, const ExplodedArgs& args, OutputIterator out );
116 /// Sets up parameter inference for an output alternative
117 template< typename OutputIterator >
118 void inferParameters( Alternative &newAlt, OutputIterator out );
119 private:
120 AlternativeFinder & altFinder;
121 const SymTab::Indexer &indexer;
122 AltList & alternatives;
123 const TypeEnvironment &env;
124 Type *& targetType;
125 };
126
127 Cost sumCost( const AltList &in ) {
128 Cost total = Cost::zero;
129 for ( AltList::const_iterator i = in.begin(); i != in.end(); ++i ) {
130 total += i->cost;
131 }
132 return total;
133 }
134
135 void printAlts( const AltList &list, std::ostream &os, unsigned int indentAmt ) {
136 Indenter indent = { Indenter::tabsize, indentAmt };
137 for ( AltList::const_iterator i = list.begin(); i != list.end(); ++i ) {
138 i->print( os, indent );
139 os << std::endl;
140 }
141 }
142
143 namespace {
144 void makeExprList( const AltList &in, std::list< Expression* > &out ) {
145 for ( AltList::const_iterator i = in.begin(); i != in.end(); ++i ) {
146 out.push_back( i->expr->clone() );
147 }
148 }
149
150 struct PruneStruct {
151 bool isAmbiguous;
152 AltList::iterator candidate;
153 PruneStruct() {}
154 PruneStruct( AltList::iterator candidate ): isAmbiguous( false ), candidate( candidate ) {}
155 };
156
157 /// Prunes a list of alternatives down to those that have the minimum conversion cost for a given return type; skips ambiguous interpretations
158 template< typename InputIterator, typename OutputIterator >
159 void pruneAlternatives( InputIterator begin, InputIterator end, OutputIterator out ) {
160 // select the alternatives that have the minimum conversion cost for a particular set of result types
161 std::map< std::string, PruneStruct > selected;
162 for ( AltList::iterator candidate = begin; candidate != end; ++candidate ) {
163 PruneStruct current( candidate );
164 std::string mangleName;
165 {
166 Type * newType = candidate->expr->get_result()->clone();
167 candidate->env.apply( newType );
168 mangleName = SymTab::Mangler::mangle( newType );
169 delete newType;
170 }
171 std::map< std::string, PruneStruct >::iterator mapPlace = selected.find( mangleName );
172 if ( mapPlace != selected.end() ) {
173 if ( candidate->cost < mapPlace->second.candidate->cost ) {
174 PRINT(
175 std::cerr << "cost " << candidate->cost << " beats " << mapPlace->second.candidate->cost << std::endl;
176 )
177 selected[ mangleName ] = current;
178 } else if ( candidate->cost == mapPlace->second.candidate->cost ) {
179 // if one of the candidates contains a deleted identifier, can pick the other, since
180 // deleted expressions should not be ambiguous if there is another option that is at least as good
181 if ( findDeletedExpr( candidate->expr ) ) {
182 // do nothing
183 PRINT( std::cerr << "candidate is deleted" << std::endl; )
184 } else if ( findDeletedExpr( mapPlace->second.candidate->expr ) ) {
185 PRINT( std::cerr << "current is deleted" << std::endl; )
186 selected[ mangleName ] = current;
187 } else {
188 PRINT(
189 std::cerr << "marking ambiguous" << std::endl;
190 )
191 mapPlace->second.isAmbiguous = true;
192 }
193 } else {
194 PRINT(
195 std::cerr << "cost " << candidate->cost << " loses to " << mapPlace->second.candidate->cost << std::endl;
196 )
197 }
198 } else {
199 selected[ mangleName ] = current;
200 }
201 }
202
203 // accept the alternatives that were unambiguous
204 for ( std::map< std::string, PruneStruct >::iterator target = selected.begin(); target != selected.end(); ++target ) {
205 if ( ! target->second.isAmbiguous ) {
206 Alternative &alt = *target->second.candidate;
207 alt.env.applyFree( alt.expr->get_result() );
208 *out++ = alt;
209 }
210 }
211 }
212
213 void renameTypes( Expression *expr ) {
214 renameTyVars( expr->result );
215 }
216 } // namespace
217
218 void referenceToRvalueConversion( Expression *& expr, Cost & cost ) {
219 if ( dynamic_cast< ReferenceType * >( expr->get_result() ) ) {
220 // cast away reference from expr
221 expr = new CastExpr( expr, expr->get_result()->stripReferences()->clone() );
222 cost.incReference();
223 }
224 }
225
226 template< typename InputIterator, typename OutputIterator >
227 void AlternativeFinder::findSubExprs( InputIterator begin, InputIterator end, OutputIterator out ) {
228 while ( begin != end ) {
229 AlternativeFinder finder( indexer, env );
230 finder.findWithAdjustment( *begin );
231 // XXX either this
232 //Designators::fixDesignations( finder, (*begin++)->get_argName() );
233 // or XXX this
234 begin++;
235 PRINT(
236 std::cerr << "findSubExprs" << std::endl;
237 printAlts( finder.alternatives, std::cerr );
238 )
239 *out++ = finder;
240 }
241 }
242
243 AlternativeFinder::AlternativeFinder( const SymTab::Indexer &indexer, const TypeEnvironment &env )
244 : indexer( indexer ), env( env ) {
245 }
246
247 void AlternativeFinder::find( Expression *expr, ResolvMode mode ) {
248 PassVisitor<Finder> finder( *this );
249 expr->accept( finder );
250 if ( mode.failFast && alternatives.empty() ) {
251 PRINT(
252 std::cerr << "No reasonable alternatives for expression " << expr << std::endl;
253 )
254 SemanticError( expr, "No reasonable alternatives for expression " );
255 }
256 if ( mode.resolveAssns || mode.prune ) {
257 // trim candidates just to those where the assertions resolve
258 // - necessary pre-requisite to pruning
259 AltList candidates;
260 for ( unsigned i = 0; i < alternatives.size(); ++i ) {
261 resolveAssertions( alternatives[i], indexer, candidates );
262 }
263 // fail early if none such
264 if ( mode.failFast && candidates.empty() ) {
265 std::ostringstream stream;
266 stream << "No resolvable alternatives for expression " << expr << "\n"
267 << "Alternatives with failing assertions are:\n";
268 printAlts( alternatives, stream, 1 );
269 SemanticError( expr->location, stream.str() );
270 }
271 // reset alternatives
272 alternatives = std::move( candidates );
273 }
274 if ( mode.prune ) {
275 auto oldsize = alternatives.size();
276 PRINT(
277 std::cerr << "alternatives before prune:" << std::endl;
278 printAlts( alternatives, std::cerr );
279 )
280 AltList pruned;
281 pruneAlternatives( alternatives.begin(), alternatives.end(), back_inserter( pruned ) );
282 if ( mode.failFast && pruned.empty() ) {
283 std::ostringstream stream;
284 AltList winners;
285 findMinCost( alternatives.begin(), alternatives.end(), back_inserter( winners ) );
286 stream << "Cannot choose between " << winners.size() << " alternatives for expression\n";
287 expr->print( stream );
288 stream << " Alternatives are:\n";
289 printAlts( winners, stream, 1 );
290 SemanticError( expr->location, stream.str() );
291 }
292 alternatives = move(pruned);
293 PRINT(
294 std::cerr << "there are " << oldsize << " alternatives before elimination" << std::endl;
295 )
296 PRINT(
297 std::cerr << "there are " << alternatives.size() << " alternatives after elimination" << std::endl;
298 )
299 }
300 // adjust types after pruning so that types substituted by pruneAlternatives are correctly adjusted
301 if ( mode.adjust ) {
302 for ( Alternative& i : alternatives ) {
303 adjustExprType( i.expr->get_result(), i.env, indexer );
304 }
305 }
306
307 // Central location to handle gcc extension keyword, etc. for all expression types.
308 for ( Alternative &iter: alternatives ) {
309 iter.expr->set_extension( expr->get_extension() );
310 iter.expr->location = expr->location;
311 } // for
312 }
313
314 void AlternativeFinder::findWithAdjustment( Expression *expr ) {
315 find( expr, ResolvMode::withAdjustment() );
316 }
317
318 void AlternativeFinder::findWithoutPrune( Expression * expr ) {
319 find( expr, ResolvMode::withoutPrune() );
320 }
321
322 void AlternativeFinder::maybeFind( Expression * expr ) {
323 find( expr, ResolvMode::withoutFailFast() );
324 }
325
326 void AlternativeFinder::Finder::addAnonConversions( const Alternative & alt ) {
327 // adds anonymous member interpretations whenever an aggregate value type is seen.
328 // it's okay for the aggregate expression to have reference type -- cast it to the base type to treat the aggregate as the referenced value
329 std::unique_ptr<Expression> aggrExpr( alt.expr->clone() );
330 alt.env.apply( aggrExpr->result );
331 Type * aggrType = aggrExpr->result;
332 if ( dynamic_cast< ReferenceType * >( aggrType ) ) {
333 aggrType = aggrType->stripReferences();
334 aggrExpr.reset( new CastExpr( aggrExpr.release(), aggrType->clone() ) );
335 }
336
337 if ( StructInstType *structInst = dynamic_cast< StructInstType* >( aggrExpr->result ) ) {
338 addAggMembers( structInst, aggrExpr.get(), alt, alt.cost+Cost::safe, "" );
339 } else if ( UnionInstType *unionInst = dynamic_cast< UnionInstType* >( aggrExpr->result ) ) {
340 addAggMembers( unionInst, aggrExpr.get(), alt, alt.cost+Cost::safe, "" );
341 } // if
342 }
343
344 template< typename StructOrUnionType >
345 void AlternativeFinder::Finder::addAggMembers( StructOrUnionType *aggInst, Expression *expr, const Alternative& alt, const Cost &newCost, const std::string & name ) {
346 std::list< Declaration* > members;
347 aggInst->lookup( name, members );
348
349 for ( Declaration * decl : members ) {
350 if ( DeclarationWithType *dwt = dynamic_cast< DeclarationWithType* >( decl ) ) {
351 // addAnonAlternatives uses vector::push_back, which invalidates references to existing elements, so
352 // can't construct in place and use vector::back
353 Alternative newAlt{ alt, new MemberExpr{ dwt, expr->clone() }, newCost };
354 renameTypes( newAlt.expr );
355 addAnonConversions( newAlt ); // add anonymous member interpretations whenever an aggregate value type is seen as a member expression.
356 alternatives.push_back( std::move(newAlt) );
357 } else {
358 assert( false );
359 }
360 }
361 }
362
363 void AlternativeFinder::Finder::addTupleMembers( TupleType *tupleType, Expression *expr, const Alternative &alt, const Cost &newCost, Expression *member ) {
364 if ( ConstantExpr * constantExpr = dynamic_cast< ConstantExpr * >( member ) ) {
365 // get the value of the constant expression as an int, must be between 0 and the length of the tuple type to have meaning
366 auto val = constantExpr->intValue();
367 std::string tmp;
368 if ( val >= 0 && (unsigned long long)val < tupleType->size() ) {
369 alternatives.push_back( Alternative{
370 alt, new TupleIndexExpr( expr->clone(), val ), newCost } );
371 } // if
372 } // if
373 }
374
375 void AlternativeFinder::Finder::postvisit( ApplicationExpr *applicationExpr ) {
376 alternatives.push_back( Alternative{ applicationExpr->clone(), env } );
377 }
378
379 Cost computeConversionCost( Type * actualType, Type * formalType, const SymTab::Indexer &indexer, const TypeEnvironment & env ) {
380 PRINT(
381 std::cerr << std::endl << "converting ";
382 actualType->print( std::cerr, 8 );
383 std::cerr << std::endl << " to ";
384 formalType->print( std::cerr, 8 );
385 std::cerr << std::endl << "environment is: ";
386 env.print( std::cerr, 8 );
387 std::cerr << std::endl;
388 )
389 Cost convCost = conversionCost( actualType, formalType, indexer, env );
390 PRINT(
391 std::cerr << std::endl << "cost is " << convCost << std::endl;
392 )
393 if ( convCost == Cost::infinity ) {
394 return convCost;
395 }
396 convCost.incPoly( polyCost( formalType, env, indexer ) + polyCost( actualType, env, indexer ) );
397 PRINT(
398 std::cerr << "cost with polycost is " << convCost << std::endl;
399 )
400 return convCost;
401 }
402
403 Cost computeExpressionConversionCost( Expression *& actualExpr, Type * formalType, const SymTab::Indexer &indexer, const TypeEnvironment & env ) {
404 Cost convCost = computeConversionCost( actualExpr->result, formalType, indexer, env );
405
406 // if there is a non-zero conversion cost, ignoring poly cost, then the expression requires conversion.
407 // ignore poly cost for now, since this requires resolution of the cast to infer parameters and this
408 // does not currently work for the reason stated below.
409 Cost tmpCost = convCost;
410 tmpCost.incPoly( -tmpCost.get_polyCost() );
411 if ( tmpCost != Cost::zero ) {
412 Type *newType = formalType->clone();
413 env.apply( newType );
414 actualExpr = new CastExpr( actualExpr, newType );
415 // xxx - SHOULD be able to resolve this cast, but at the moment pointers are not castable to zero_t, but are implicitly convertible. This is clearly
416 // inconsistent, once this is fixed it should be possible to resolve the cast.
417 // xxx - this isn't working, it appears because type1 (the formal type) is seen as widenable, but it shouldn't be, because this makes the conversion from DT* to DT* since commontype(zero_t, DT*) is DT*, rather than just nothing.
418
419 // AlternativeFinder finder( indexer, env );
420 // finder.findWithAdjustment( actualExpr );
421 // assertf( finder.get_alternatives().size() > 0, "Somehow castable expression failed to find alternatives." );
422 // assertf( finder.get_alternatives().size() == 1, "Somehow got multiple alternatives for known cast expression." );
423 // Alternative & alt = finder.get_alternatives().front();
424 // delete actualExpr;
425 // actualExpr = alt.expr->clone();
426 }
427 return convCost;
428 }
429
430 Cost computeApplicationConversionCost( Alternative &alt, const SymTab::Indexer &indexer ) {
431 ApplicationExpr *appExpr = strict_dynamic_cast< ApplicationExpr* >( alt.expr );
432 PointerType *pointer = strict_dynamic_cast< PointerType* >( appExpr->function->result );
433 FunctionType *function = strict_dynamic_cast< FunctionType* >( pointer->base );
434
435 Cost convCost = Cost::zero;
436 std::list< DeclarationWithType* >& formals = function->parameters;
437 std::list< DeclarationWithType* >::iterator formal = formals.begin();
438 std::list< Expression* >& actuals = appExpr->args;
439
440 for ( Expression*& actualExpr : actuals ) {
441 Type * actualType = actualExpr->result;
442 PRINT(
443 std::cerr << "actual expression:" << std::endl;
444 actualExpr->print( std::cerr, 8 );
445 std::cerr << "--- results are" << std::endl;
446 actualType->print( std::cerr, 8 );
447 )
448 if ( formal == formals.end() ) {
449 if ( function->isVarArgs ) {
450 convCost.incUnsafe();
451 PRINT( std::cerr << "end of formals with varargs function: inc unsafe: " << convCost << std::endl; ; )
452 // convert reference-typed expressions to value-typed expressions
453 referenceToRvalueConversion( actualExpr, convCost );
454 continue;
455 } else {
456 return Cost::infinity;
457 }
458 }
459 if ( DefaultArgExpr * def = dynamic_cast< DefaultArgExpr * >( actualExpr ) ) {
460 // default arguments should be free - don't include conversion cost.
461 // Unwrap them here because they are not relevant to the rest of the system.
462 actualExpr = def->expr;
463 ++formal;
464 continue;
465 }
466 // mark conversion cost to formal and also specialization cost of formal type
467 Type * formalType = (*formal)->get_type();
468 convCost += computeExpressionConversionCost( actualExpr, formalType, indexer, alt.env );
469 convCost.decSpec( specCost( formalType ) );
470 ++formal; // can't be in for-loop update because of the continue
471 }
472 if ( formal != formals.end() ) {
473 return Cost::infinity;
474 }
475
476 // specialization cost of return types can't be accounted for directly, it disables
477 // otherwise-identical calls, like this example based on auto-newline in the I/O lib:
478 //
479 // forall(otype OS) {
480 // void ?|?(OS&, int); // with newline
481 // OS& ?|?(OS&, int); // no newline, always chosen due to more specialization
482 // }
483
484 // mark type variable and specialization cost of forall clause
485 convCost.incVar( function->forall.size() );
486 for ( TypeDecl* td : function->forall ) {
487 convCost.decSpec( td->assertions.size() );
488 }
489
490 return convCost;
491 }
492
493 /// Adds type variables to the open variable set and marks their assertions
494 void makeUnifiableVars( Type *type, OpenVarSet &unifiableVars, AssertionSet &needAssertions ) {
495 for ( Type::ForallList::const_iterator tyvar = type->forall.begin(); tyvar != type->forall.end(); ++tyvar ) {
496 unifiableVars[ (*tyvar)->get_name() ] = TypeDecl::Data{ *tyvar };
497 for ( std::list< DeclarationWithType* >::iterator assert = (*tyvar)->assertions.begin(); assert != (*tyvar)->assertions.end(); ++assert ) {
498 needAssertions[ *assert ].isUsed = true;
499 }
500 }
501 }
502
503 /// Unique identifier for matching expression resolutions to their requesting expression
504 UniqueId globalResnSlot = 0;
505
506 template< typename OutputIterator >
507 void AlternativeFinder::Finder::inferParameters( Alternative &newAlt, OutputIterator out ) {
508 // Set need bindings for any unbound assertions
509 UniqueId crntResnSlot = 0; // matching ID for this expression's assertions
510 for ( auto& assn : newAlt.need ) {
511 // skip already-matched assertions
512 if ( assn.info.resnSlot != 0 ) continue;
513 // assign slot for expression if needed
514 if ( crntResnSlot == 0 ) { crntResnSlot = ++globalResnSlot; }
515 // fix slot to assertion
516 assn.info.resnSlot = crntResnSlot;
517 }
518 // pair slot to expression
519 if ( crntResnSlot != 0 ) { newAlt.expr->resnSlots.push_back( crntResnSlot ); }
520
521 // add to output list, assertion resolution is deferred
522 *out++ = newAlt;
523 }
524
525 /// Gets a default value from an initializer, nullptr if not present
526 ConstantExpr* getDefaultValue( Initializer* init ) {
527 if ( SingleInit* si = dynamic_cast<SingleInit*>( init ) ) {
528 if ( CastExpr* ce = dynamic_cast<CastExpr*>( si->value ) ) {
529 return dynamic_cast<ConstantExpr*>( ce->arg );
530 } else {
531 return dynamic_cast<ConstantExpr*>( si->value );
532 }
533 }
534 return nullptr;
535 }
536
537 /// State to iteratively build a match of parameter expressions to arguments
538 struct ArgPack {
539 std::size_t parent; ///< Index of parent pack
540 std::unique_ptr<Expression> expr; ///< The argument stored here
541 Cost cost; ///< The cost of this argument
542 TypeEnvironment env; ///< Environment for this pack
543 AssertionSet need; ///< Assertions outstanding for this pack
544 AssertionSet have; ///< Assertions found for this pack
545 OpenVarSet openVars; ///< Open variables for this pack
546 unsigned nextArg; ///< Index of next argument in arguments list
547 unsigned tupleStart; ///< Number of tuples that start at this index
548 unsigned nextExpl; ///< Index of next exploded element
549 unsigned explAlt; ///< Index of alternative for nextExpl > 0
550
551 ArgPack()
552 : parent(0), expr(), cost(Cost::zero), env(), need(), have(), openVars(), nextArg(0),
553 tupleStart(0), nextExpl(0), explAlt(0) {}
554
555 ArgPack(const TypeEnvironment& env, const AssertionSet& need, const AssertionSet& have,
556 const OpenVarSet& openVars)
557 : parent(0), expr(), cost(Cost::zero), env(env), need(need), have(have),
558 openVars(openVars), nextArg(0), tupleStart(0), nextExpl(0), explAlt(0) {}
559
560 ArgPack(std::size_t parent, Expression* expr, TypeEnvironment&& env, AssertionSet&& need,
561 AssertionSet&& have, OpenVarSet&& openVars, unsigned nextArg,
562 unsigned tupleStart = 0, Cost cost = Cost::zero, unsigned nextExpl = 0,
563 unsigned explAlt = 0 )
564 : parent(parent), expr(expr->clone()), cost(cost), env(move(env)), need(move(need)),
565 have(move(have)), openVars(move(openVars)), nextArg(nextArg), tupleStart(tupleStart),
566 nextExpl(nextExpl), explAlt(explAlt) {}
567
568 ArgPack(const ArgPack& o, TypeEnvironment&& env, AssertionSet&& need, AssertionSet&& have,
569 OpenVarSet&& openVars, unsigned nextArg, Cost added )
570 : parent(o.parent), expr(o.expr ? o.expr->clone() : nullptr), cost(o.cost + added),
571 env(move(env)), need(move(need)), have(move(have)), openVars(move(openVars)),
572 nextArg(nextArg), tupleStart(o.tupleStart), nextExpl(0), explAlt(0) {}
573
574 /// true iff this pack is in the middle of an exploded argument
575 bool hasExpl() const { return nextExpl > 0; }
576
577 /// Gets the list of exploded alternatives for this pack
578 const ExplodedActual& getExpl( const ExplodedArgs& args ) const {
579 return args[nextArg-1][explAlt];
580 }
581
582 /// Ends a tuple expression, consolidating the appropriate actuals
583 void endTuple( const std::vector<ArgPack>& packs ) {
584 // add all expressions in tuple to list, summing cost
585 std::list<Expression*> exprs;
586 const ArgPack* pack = this;
587 if ( expr ) { exprs.push_front( expr.release() ); }
588 while ( pack->tupleStart == 0 ) {
589 pack = &packs[pack->parent];
590 exprs.push_front( pack->expr->clone() );
591 cost += pack->cost;
592 }
593 // reset pack to appropriate tuple
594 expr.reset( new TupleExpr( exprs ) );
595 tupleStart = pack->tupleStart - 1;
596 parent = pack->parent;
597 }
598 };
599
600 /// Instantiates an argument to match a formal, returns false if no results left
601 bool instantiateArgument( Type* formalType, Initializer* initializer,
602 const ExplodedArgs& args, std::vector<ArgPack>& results, std::size_t& genStart,
603 const SymTab::Indexer& indexer, unsigned nTuples = 0 ) {
604 if ( TupleType * tupleType = dynamic_cast<TupleType*>( formalType ) ) {
605 // formalType is a TupleType - group actuals into a TupleExpr
606 ++nTuples;
607 for ( Type* type : *tupleType ) {
608 // xxx - dropping initializer changes behaviour from previous, but seems correct
609 // ^^^ need to handle the case where a tuple has a default argument
610 if ( ! instantiateArgument(
611 type, nullptr, args, results, genStart, indexer, nTuples ) )
612 return false;
613 nTuples = 0;
614 }
615 // re-consititute tuples for final generation
616 for ( auto i = genStart; i < results.size(); ++i ) {
617 results[i].endTuple( results );
618 }
619 return true;
620 } else if ( TypeInstType * ttype = Tuples::isTtype( formalType ) ) {
621 // formalType is a ttype, consumes all remaining arguments
622 // xxx - mixing default arguments with variadic??
623
624 // completed tuples; will be spliced to end of results to finish
625 std::vector<ArgPack> finalResults{};
626
627 // iterate until all results completed
628 std::size_t genEnd;
629 ++nTuples;
630 do {
631 genEnd = results.size();
632
633 // add another argument to results
634 for ( std::size_t i = genStart; i < genEnd; ++i ) {
635 auto nextArg = results[i].nextArg;
636
637 // use next element of exploded tuple if present
638 if ( results[i].hasExpl() ) {
639 const ExplodedActual& expl = results[i].getExpl( args );
640
641 unsigned nextExpl = results[i].nextExpl + 1;
642 if ( nextExpl == expl.exprs.size() ) {
643 nextExpl = 0;
644 }
645
646 results.emplace_back(
647 i, expl.exprs[results[i].nextExpl].get(), copy(results[i].env),
648 copy(results[i].need), copy(results[i].have),
649 copy(results[i].openVars), nextArg, nTuples, Cost::zero, nextExpl,
650 results[i].explAlt );
651
652 continue;
653 }
654
655 // finish result when out of arguments
656 if ( nextArg >= args.size() ) {
657 ArgPack newResult{
658 results[i].env, results[i].need, results[i].have,
659 results[i].openVars };
660 newResult.nextArg = nextArg;
661 Type* argType;
662
663 if ( nTuples > 0 || ! results[i].expr ) {
664 // first iteration or no expression to clone,
665 // push empty tuple expression
666 newResult.parent = i;
667 std::list<Expression*> emptyList;
668 newResult.expr.reset( new TupleExpr( emptyList ) );
669 argType = newResult.expr->get_result();
670 } else {
671 // clone result to collect tuple
672 newResult.parent = results[i].parent;
673 newResult.cost = results[i].cost;
674 newResult.tupleStart = results[i].tupleStart;
675 newResult.expr.reset( results[i].expr->clone() );
676 argType = newResult.expr->get_result();
677
678 if ( results[i].tupleStart > 0 && Tuples::isTtype( argType ) ) {
679 // the case where a ttype value is passed directly is special,
680 // e.g. for argument forwarding purposes
681 // xxx - what if passing multiple arguments, last of which is
682 // ttype?
683 // xxx - what would happen if unify was changed so that unifying
684 // tuple
685 // types flattened both before unifying lists? then pass in
686 // TupleType (ttype) below.
687 --newResult.tupleStart;
688 } else {
689 // collapse leftover arguments into tuple
690 newResult.endTuple( results );
691 argType = newResult.expr->get_result();
692 }
693 }
694
695 // check unification for ttype before adding to final
696 if ( unify( ttype, argType, newResult.env, newResult.need, newResult.have,
697 newResult.openVars, indexer ) ) {
698 finalResults.push_back( move(newResult) );
699 }
700
701 continue;
702 }
703
704 // add each possible next argument
705 for ( std::size_t j = 0; j < args[nextArg].size(); ++j ) {
706 const ExplodedActual& expl = args[nextArg][j];
707
708 // fresh copies of parent parameters for this iteration
709 TypeEnvironment env = results[i].env;
710 OpenVarSet openVars = results[i].openVars;
711
712 env.addActual( expl.env, openVars );
713
714 // skip empty tuple arguments by (near-)cloning parent into next gen
715 if ( expl.exprs.empty() ) {
716 results.emplace_back(
717 results[i], move(env), copy(results[i].need),
718 copy(results[i].have), move(openVars), nextArg + 1, expl.cost );
719
720 continue;
721 }
722
723 // add new result
724 results.emplace_back(
725 i, expl.exprs.front().get(), move(env), copy(results[i].need),
726 copy(results[i].have), move(openVars), nextArg + 1,
727 nTuples, expl.cost, expl.exprs.size() == 1 ? 0 : 1, j );
728 }
729 }
730
731 // reset for next round
732 genStart = genEnd;
733 nTuples = 0;
734 } while ( genEnd != results.size() );
735
736 // splice final results onto results
737 for ( std::size_t i = 0; i < finalResults.size(); ++i ) {
738 results.push_back( move(finalResults[i]) );
739 }
740 return ! finalResults.empty();
741 }
742
743 // iterate each current subresult
744 std::size_t genEnd = results.size();
745 for ( std::size_t i = genStart; i < genEnd; ++i ) {
746 auto nextArg = results[i].nextArg;
747
748 // use remainder of exploded tuple if present
749 if ( results[i].hasExpl() ) {
750 const ExplodedActual& expl = results[i].getExpl( args );
751 Expression* expr = expl.exprs[results[i].nextExpl].get();
752
753 TypeEnvironment env = results[i].env;
754 AssertionSet need = results[i].need, have = results[i].have;
755 OpenVarSet openVars = results[i].openVars;
756
757 Type* actualType = expr->get_result();
758
759 PRINT(
760 std::cerr << "formal type is ";
761 formalType->print( std::cerr );
762 std::cerr << std::endl << "actual type is ";
763 actualType->print( std::cerr );
764 std::cerr << std::endl;
765 )
766
767 if ( unify( formalType, actualType, env, need, have, openVars, indexer ) ) {
768 unsigned nextExpl = results[i].nextExpl + 1;
769 if ( nextExpl == expl.exprs.size() ) {
770 nextExpl = 0;
771 }
772
773 results.emplace_back(
774 i, expr, move(env), move(need), move(have), move(openVars), nextArg,
775 nTuples, Cost::zero, nextExpl, results[i].explAlt );
776 }
777
778 continue;
779 }
780
781 // use default initializers if out of arguments
782 if ( nextArg >= args.size() ) {
783 if ( ConstantExpr* cnstExpr = getDefaultValue( initializer ) ) {
784 if ( Constant* cnst = dynamic_cast<Constant*>( cnstExpr->get_constant() ) ) {
785 TypeEnvironment env = results[i].env;
786 AssertionSet need = results[i].need, have = results[i].have;
787 OpenVarSet openVars = results[i].openVars;
788
789 if ( unify( formalType, cnst->get_type(), env, need, have, openVars,
790 indexer ) ) {
791 results.emplace_back(
792 i, new DefaultArgExpr( cnstExpr ), move(env), move(need), move(have),
793 move(openVars), nextArg, nTuples );
794 }
795 }
796 }
797
798 continue;
799 }
800
801 // Check each possible next argument
802 for ( std::size_t j = 0; j < args[nextArg].size(); ++j ) {
803 const ExplodedActual& expl = args[nextArg][j];
804
805 // fresh copies of parent parameters for this iteration
806 TypeEnvironment env = results[i].env;
807 AssertionSet need = results[i].need, have = results[i].have;
808 OpenVarSet openVars = results[i].openVars;
809
810 env.addActual( expl.env, openVars );
811
812 // skip empty tuple arguments by (near-)cloning parent into next gen
813 if ( expl.exprs.empty() ) {
814 results.emplace_back(
815 results[i], move(env), move(need), move(have), move(openVars),
816 nextArg + 1, expl.cost );
817
818 continue;
819 }
820
821 // consider only first exploded actual
822 Expression* expr = expl.exprs.front().get();
823 Type* actualType = expr->result->clone();
824
825 PRINT(
826 std::cerr << "formal type is ";
827 formalType->print( std::cerr );
828 std::cerr << std::endl << "actual type is ";
829 actualType->print( std::cerr );
830 std::cerr << std::endl;
831 )
832
833 // attempt to unify types
834 if ( unify( formalType, actualType, env, need, have, openVars, indexer ) ) {
835 // add new result
836 results.emplace_back(
837 i, expr, move(env), move(need), move(have), move(openVars), nextArg + 1,
838 nTuples, expl.cost, expl.exprs.size() == 1 ? 0 : 1, j );
839 }
840 }
841 }
842
843 // reset for next parameter
844 genStart = genEnd;
845
846 return genEnd != results.size();
847 }
848
849 template<typename OutputIterator>
850 void AlternativeFinder::Finder::validateFunctionAlternative( const Alternative &func, ArgPack& result,
851 const std::vector<ArgPack>& results, OutputIterator out ) {
852 ApplicationExpr *appExpr = new ApplicationExpr( func.expr->clone() );
853 // sum cost and accumulate actuals
854 std::list<Expression*>& args = appExpr->args;
855 Cost cost = func.cost;
856 const ArgPack* pack = &result;
857 while ( pack->expr ) {
858 args.push_front( pack->expr->clone() );
859 cost += pack->cost;
860 pack = &results[pack->parent];
861 }
862 // build and validate new alternative
863 Alternative newAlt{ appExpr, result.env, result.openVars, result.need, cost };
864 PRINT(
865 std::cerr << "instantiate function success: " << appExpr << std::endl;
866 std::cerr << "need assertions:" << std::endl;
867 printAssertionSet( result.need, std::cerr, 8 );
868 )
869 inferParameters( newAlt, out );
870 }
871
872 template<typename OutputIterator>
873 void AlternativeFinder::Finder::makeFunctionAlternatives( const Alternative &func,
874 FunctionType *funcType, const ExplodedArgs &args, OutputIterator out ) {
875 OpenVarSet funcOpenVars;
876 AssertionSet funcNeed, funcHave;
877 TypeEnvironment funcEnv( func.env );
878 makeUnifiableVars( funcType, funcOpenVars, funcNeed );
879 // add all type variables as open variables now so that those not used in the parameter
880 // list are still considered open.
881 funcEnv.add( funcType->forall );
882
883 if ( targetType && ! targetType->isVoid() && ! funcType->returnVals.empty() ) {
884 // attempt to narrow based on expected target type
885 Type * returnType = funcType->returnVals.front()->get_type();
886 if ( ! unify( returnType, targetType, funcEnv, funcNeed, funcHave, funcOpenVars,
887 indexer ) ) {
888 // unification failed, don't pursue this function alternative
889 return;
890 }
891 }
892
893 // iteratively build matches, one parameter at a time
894 std::vector<ArgPack> results;
895 results.push_back( ArgPack{ funcEnv, funcNeed, funcHave, funcOpenVars } );
896 std::size_t genStart = 0;
897
898 for ( DeclarationWithType* formal : funcType->parameters ) {
899 ObjectDecl* obj = strict_dynamic_cast< ObjectDecl* >( formal );
900 if ( ! instantiateArgument(
901 obj->type, obj->init, args, results, genStart, indexer ) )
902 return;
903 }
904
905 if ( funcType->get_isVarArgs() ) {
906 // append any unused arguments to vararg pack
907 std::size_t genEnd;
908 do {
909 genEnd = results.size();
910
911 // iterate results
912 for ( std::size_t i = genStart; i < genEnd; ++i ) {
913 auto nextArg = results[i].nextArg;
914
915 // use remainder of exploded tuple if present
916 if ( results[i].hasExpl() ) {
917 const ExplodedActual& expl = results[i].getExpl( args );
918
919 unsigned nextExpl = results[i].nextExpl + 1;
920 if ( nextExpl == expl.exprs.size() ) {
921 nextExpl = 0;
922 }
923
924 results.emplace_back(
925 i, expl.exprs[results[i].nextExpl].get(), copy(results[i].env),
926 copy(results[i].need), copy(results[i].have),
927 copy(results[i].openVars), nextArg, 0, Cost::zero, nextExpl,
928 results[i].explAlt );
929
930 continue;
931 }
932
933 // finish result when out of arguments
934 if ( nextArg >= args.size() ) {
935 validateFunctionAlternative( func, results[i], results, out );
936
937 continue;
938 }
939
940 // add each possible next argument
941 for ( std::size_t j = 0; j < args[nextArg].size(); ++j ) {
942 const ExplodedActual& expl = args[nextArg][j];
943
944 // fresh copies of parent parameters for this iteration
945 TypeEnvironment env = results[i].env;
946 OpenVarSet openVars = results[i].openVars;
947
948 env.addActual( expl.env, openVars );
949
950 // skip empty tuple arguments by (near-)cloning parent into next gen
951 if ( expl.exprs.empty() ) {
952 results.emplace_back(
953 results[i], move(env), copy(results[i].need),
954 copy(results[i].have), move(openVars), nextArg + 1, expl.cost );
955
956 continue;
957 }
958
959 // add new result
960 results.emplace_back(
961 i, expl.exprs.front().get(), move(env), copy(results[i].need),
962 copy(results[i].have), move(openVars), nextArg + 1, 0,
963 expl.cost, expl.exprs.size() == 1 ? 0 : 1, j );
964 }
965 }
966
967 genStart = genEnd;
968 } while ( genEnd != results.size() );
969 } else {
970 // filter out results that don't use all the arguments
971 for ( std::size_t i = genStart; i < results.size(); ++i ) {
972 ArgPack& result = results[i];
973 if ( ! result.hasExpl() && result.nextArg >= args.size() ) {
974 validateFunctionAlternative( func, result, results, out );
975 }
976 }
977 }
978 }
979
980 void AlternativeFinder::Finder::postvisit( UntypedExpr *untypedExpr ) {
981 AlternativeFinder funcFinder( indexer, env );
982 funcFinder.findWithAdjustment( untypedExpr->function );
983 // if there are no function alternatives, then proceeding is a waste of time.
984 // xxx - findWithAdjustment throws, so this check and others like it shouldn't be necessary.
985 if ( funcFinder.alternatives.empty() ) return;
986
987 std::vector< AlternativeFinder > argAlternatives;
988 altFinder.findSubExprs( untypedExpr->begin_args(), untypedExpr->end_args(),
989 back_inserter( argAlternatives ) );
990
991 // take care of possible tuple assignments
992 // if not tuple assignment, assignment is taken care of as a normal function call
993 Tuples::handleTupleAssignment( altFinder, untypedExpr, argAlternatives );
994
995 // find function operators
996 static NameExpr *opExpr = new NameExpr( "?()" );
997 AlternativeFinder funcOpFinder( indexer, env );
998 // it's ok if there aren't any defined function ops
999 funcOpFinder.maybeFind( opExpr );
1000 PRINT(
1001 std::cerr << "known function ops:" << std::endl;
1002 printAlts( funcOpFinder.alternatives, std::cerr, 1 );
1003 )
1004
1005 // pre-explode arguments
1006 ExplodedArgs argExpansions;
1007 argExpansions.reserve( argAlternatives.size() );
1008
1009 for ( const AlternativeFinder& arg : argAlternatives ) {
1010 argExpansions.emplace_back();
1011 auto& argE = argExpansions.back();
1012 // argE.reserve( arg.alternatives.size() );
1013
1014 for ( const Alternative& actual : arg ) {
1015 argE.emplace_back( actual, indexer );
1016 }
1017 }
1018
1019 AltList candidates;
1020 SemanticErrorException errors;
1021 for ( AltList::iterator func = funcFinder.alternatives.begin(); func != funcFinder.alternatives.end(); ++func ) {
1022 try {
1023 PRINT(
1024 std::cerr << "working on alternative: " << std::endl;
1025 func->print( std::cerr, 8 );
1026 )
1027 // check if the type is pointer to function
1028 if ( PointerType *pointer = dynamic_cast< PointerType* >( func->expr->result->stripReferences() ) ) {
1029 if ( FunctionType *function = dynamic_cast< FunctionType* >( pointer->base ) ) {
1030 Alternative newFunc( *func );
1031 referenceToRvalueConversion( newFunc.expr, newFunc.cost );
1032 makeFunctionAlternatives( newFunc, function, argExpansions,
1033 std::back_inserter( candidates ) );
1034 }
1035 } else if ( TypeInstType *typeInst = dynamic_cast< TypeInstType* >( func->expr->result->stripReferences() ) ) { // handle ftype (e.g. *? on function pointer)
1036 if ( const EqvClass *eqvClass = func->env.lookup( typeInst->name ) ) {
1037 if ( FunctionType *function = dynamic_cast< FunctionType* >( eqvClass->type ) ) {
1038 Alternative newFunc( *func );
1039 referenceToRvalueConversion( newFunc.expr, newFunc.cost );
1040 makeFunctionAlternatives( newFunc, function, argExpansions,
1041 std::back_inserter( candidates ) );
1042 } // if
1043 } // if
1044 }
1045 } catch ( SemanticErrorException &e ) {
1046 errors.append( e );
1047 }
1048 } // for
1049
1050 // try each function operator ?() with each function alternative
1051 if ( ! funcOpFinder.alternatives.empty() ) {
1052 // add exploded function alternatives to front of argument list
1053 std::vector<ExplodedActual> funcE;
1054 funcE.reserve( funcFinder.alternatives.size() );
1055 for ( const Alternative& actual : funcFinder ) {
1056 funcE.emplace_back( actual, indexer );
1057 }
1058 argExpansions.insert( argExpansions.begin(), move(funcE) );
1059
1060 for ( AltList::iterator funcOp = funcOpFinder.alternatives.begin();
1061 funcOp != funcOpFinder.alternatives.end(); ++funcOp ) {
1062 try {
1063 // check if type is a pointer to function
1064 if ( PointerType* pointer = dynamic_cast<PointerType*>(
1065 funcOp->expr->result->stripReferences() ) ) {
1066 if ( FunctionType* function =
1067 dynamic_cast<FunctionType*>( pointer->base ) ) {
1068 Alternative newFunc( *funcOp );
1069 referenceToRvalueConversion( newFunc.expr, newFunc.cost );
1070 makeFunctionAlternatives( newFunc, function, argExpansions,
1071 std::back_inserter( candidates ) );
1072 }
1073 }
1074 } catch ( SemanticErrorException &e ) {
1075 errors.append( e );
1076 }
1077 }
1078 }
1079
1080 // Implement SFINAE; resolution errors are only errors if there aren't any non-erroneous resolutions
1081 if ( candidates.empty() && ! errors.isEmpty() ) { throw errors; }
1082
1083 // compute conversionsion costs
1084 for ( Alternative& withFunc : candidates ) {
1085 Cost cvtCost = computeApplicationConversionCost( withFunc, indexer );
1086
1087 PRINT(
1088 ApplicationExpr *appExpr = strict_dynamic_cast< ApplicationExpr* >( withFunc.expr );
1089 PointerType *pointer = strict_dynamic_cast< PointerType* >( appExpr->function->result );
1090 FunctionType *function = strict_dynamic_cast< FunctionType* >( pointer->base );
1091 std::cerr << "Case +++++++++++++ " << appExpr->function << std::endl;
1092 std::cerr << "formals are:" << std::endl;
1093 printAll( function->parameters, std::cerr, 8 );
1094 std::cerr << "actuals are:" << std::endl;
1095 printAll( appExpr->args, std::cerr, 8 );
1096 std::cerr << "bindings are:" << std::endl;
1097 withFunc.env.print( std::cerr, 8 );
1098 std::cerr << "cost is: " << withFunc.cost << std::endl;
1099 std::cerr << "cost of conversion is:" << cvtCost << std::endl;
1100 )
1101 if ( cvtCost != Cost::infinity ) {
1102 withFunc.cvtCost = cvtCost;
1103 alternatives.push_back( withFunc );
1104 } // if
1105 } // for
1106
1107 candidates = move(alternatives);
1108
1109 // use a new list so that alternatives are not examined by addAnonConversions twice.
1110 AltList winners;
1111 findMinCost( candidates.begin(), candidates.end(), std::back_inserter( winners ) );
1112
1113 // function may return struct or union value, in which case we need to add alternatives
1114 // for implicit conversions to each of the anonymous members, must happen after findMinCost
1115 // since anon conversions are never the cheapest expression
1116 for ( const Alternative & alt : winners ) {
1117 addAnonConversions( alt );
1118 }
1119 spliceBegin( alternatives, winners );
1120
1121 if ( alternatives.empty() && targetType && ! targetType->isVoid() ) {
1122 // xxx - this is a temporary hack. If resolution is unsuccessful with a target type, try again without a
1123 // target type, since it will sometimes succeed when it wouldn't easily with target type binding. For example,
1124 // forall( otype T ) lvalue T ?[?]( T *, ptrdiff_t );
1125 // const char * x = "hello world";
1126 // unsigned char ch = x[0];
1127 // Fails with simple return type binding. First, T is bound to unsigned char, then (x: const char *) is unified
1128 // with unsigned char *, which fails because pointer base types must be unified exactly. The new resolver should
1129 // fix this issue in a more robust way.
1130 targetType = nullptr;
1131 postvisit( untypedExpr );
1132 }
1133 }
1134
1135 bool isLvalue( Expression *expr ) {
1136 // xxx - recurse into tuples?
1137 return expr->result && ( expr->result->get_lvalue() || dynamic_cast< ReferenceType * >( expr->result ) );
1138 }
1139
1140 void AlternativeFinder::Finder::postvisit( AddressExpr *addressExpr ) {
1141 AlternativeFinder finder( indexer, env );
1142 finder.find( addressExpr->get_arg() );
1143 for ( Alternative& alt : finder.alternatives ) {
1144 if ( isLvalue( alt.expr ) ) {
1145 alternatives.push_back(
1146 Alternative{ alt, new AddressExpr( alt.expr->clone() ), alt.cost } );
1147 } // if
1148 } // for
1149 }
1150
1151 void AlternativeFinder::Finder::postvisit( LabelAddressExpr * expr ) {
1152 alternatives.push_back( Alternative{ expr->clone(), env } );
1153 }
1154
1155 Expression * restructureCast( Expression * argExpr, Type * toType, bool isGenerated ) {
1156 if ( argExpr->get_result()->size() > 1 && ! toType->isVoid() && ! dynamic_cast<ReferenceType *>( toType ) ) {
1157 // Argument expression is a tuple and the target type is not void and not a reference type.
1158 // Cast each member of the tuple to its corresponding target type, producing the tuple of those
1159 // cast expressions. If there are more components of the tuple than components in the target type,
1160 // then excess components do not come out in the result expression (but UniqueExprs ensure that
1161 // side effects will still be done).
1162 if ( Tuples::maybeImpureIgnoreUnique( argExpr ) ) {
1163 // expressions which may contain side effects require a single unique instance of the expression.
1164 argExpr = new UniqueExpr( argExpr );
1165 }
1166 std::list< Expression * > componentExprs;
1167 for ( unsigned int i = 0; i < toType->size(); i++ ) {
1168 // cast each component
1169 TupleIndexExpr * idx = new TupleIndexExpr( argExpr->clone(), i );
1170 componentExprs.push_back( restructureCast( idx, toType->getComponent( i ), isGenerated ) );
1171 }
1172 delete argExpr;
1173 assert( componentExprs.size() > 0 );
1174 // produce the tuple of casts
1175 return new TupleExpr( componentExprs );
1176 } else {
1177 // handle normally
1178 CastExpr * ret = new CastExpr( argExpr, toType->clone() );
1179 ret->isGenerated = isGenerated;
1180 return ret;
1181 }
1182 }
1183
1184 void AlternativeFinder::Finder::postvisit( CastExpr *castExpr ) {
1185 Type *& toType = castExpr->get_result();
1186 assert( toType );
1187 toType = resolveTypeof( toType, indexer );
1188 SymTab::validateType( toType, &indexer );
1189 adjustExprType( toType, env, indexer );
1190
1191 AlternativeFinder finder( indexer, env );
1192 finder.targetType = toType;
1193 finder.findWithAdjustment( castExpr->arg );
1194
1195 AltList candidates;
1196 for ( Alternative & alt : finder.alternatives ) {
1197 AssertionSet needAssertions( alt.need.begin(), alt.need.end() );
1198 AssertionSet haveAssertions;
1199 OpenVarSet openVars{ alt.openVars };
1200
1201 alt.env.extractOpenVars( openVars );
1202
1203 // It's possible that a cast can throw away some values in a multiply-valued expression. (An example is a
1204 // cast-to-void, which casts from one value to zero.) Figure out the prefix of the subexpression results
1205 // that are cast directly. The candidate is invalid if it has fewer results than there are types to cast
1206 // to.
1207 int discardedValues = alt.expr->result->size() - castExpr->result->size();
1208 if ( discardedValues < 0 ) continue;
1209 // xxx - may need to go into tuple types and extract relevant types and use unifyList. Note that currently, this does not
1210 // allow casting a tuple to an atomic type (e.g. (int)([1, 2, 3]))
1211 // unification run for side-effects
1212 unify( castExpr->result, alt.expr->result, alt.env, needAssertions,
1213 haveAssertions, openVars, indexer );
1214 Cost thisCost = castCost( alt.expr->result, castExpr->result, indexer,
1215 alt.env );
1216 PRINT(
1217 std::cerr << "working on cast with result: " << castExpr->result << std::endl;
1218 std::cerr << "and expr type: " << alt.expr->result << std::endl;
1219 std::cerr << "env: " << alt.env << std::endl;
1220 )
1221 if ( thisCost != Cost::infinity ) {
1222 PRINT(
1223 std::cerr << "has finite cost." << std::endl;
1224 )
1225 // count one safe conversion for each value that is thrown away
1226 thisCost.incSafe( discardedValues );
1227 Alternative newAlt{
1228 restructureCast( alt.expr->clone(), toType, castExpr->isGenerated ),
1229 alt.env, openVars, needAssertions, alt.cost, alt.cost + thisCost };
1230 inferParameters( newAlt, back_inserter( candidates ) );
1231 } // if
1232 } // for
1233
1234 // findMinCost selects the alternatives with the lowest "cost" members, but has the side effect of copying the
1235 // cvtCost member to the cost member (since the old cost is now irrelevant). Thus, calling findMinCost twice
1236 // selects first based on argument cost, then on conversion cost.
1237 AltList minArgCost;
1238 findMinCost( candidates.begin(), candidates.end(), std::back_inserter( minArgCost ) );
1239 findMinCost( minArgCost.begin(), minArgCost.end(), std::back_inserter( alternatives ) );
1240 }
1241
1242 void AlternativeFinder::Finder::postvisit( VirtualCastExpr * castExpr ) {
1243 assertf( castExpr->get_result(), "Implicit virtual cast targets not yet supported." );
1244 AlternativeFinder finder( indexer, env );
1245 // don't prune here, since it's guaranteed all alternatives will have the same type
1246 finder.findWithoutPrune( castExpr->get_arg() );
1247 for ( Alternative & alt : finder.alternatives ) {
1248 alternatives.push_back( Alternative{
1249 alt, new VirtualCastExpr{ alt.expr->clone(), castExpr->get_result()->clone() },
1250 alt.cost } );
1251 }
1252 }
1253
1254 namespace {
1255 /// Gets name from untyped member expression (member must be NameExpr)
1256 const std::string& get_member_name( UntypedMemberExpr *memberExpr ) {
1257 if ( dynamic_cast< ConstantExpr * >( memberExpr->get_member() ) ) {
1258 SemanticError( memberExpr, "Indexed access to struct fields unsupported: " );
1259 } // if
1260 NameExpr * nameExpr = dynamic_cast< NameExpr * >( memberExpr->get_member() );
1261 assert( nameExpr );
1262 return nameExpr->get_name();
1263 }
1264 }
1265
1266 void AlternativeFinder::Finder::postvisit( UntypedMemberExpr *memberExpr ) {
1267 AlternativeFinder funcFinder( indexer, env );
1268 funcFinder.findWithAdjustment( memberExpr->get_aggregate() );
1269 for ( AltList::const_iterator agg = funcFinder.alternatives.begin(); agg != funcFinder.alternatives.end(); ++agg ) {
1270 // it's okay for the aggregate expression to have reference type -- cast it to the base type to treat the aggregate as the referenced value
1271 Cost cost = agg->cost;
1272 Expression * aggrExpr = agg->expr->clone();
1273 referenceToRvalueConversion( aggrExpr, cost );
1274 std::unique_ptr<Expression> guard( aggrExpr );
1275
1276 // find member of the given type
1277 if ( StructInstType *structInst = dynamic_cast< StructInstType* >( aggrExpr->get_result() ) ) {
1278 addAggMembers( structInst, aggrExpr, *agg, cost, get_member_name(memberExpr) );
1279 } else if ( UnionInstType *unionInst = dynamic_cast< UnionInstType* >( aggrExpr->get_result() ) ) {
1280 addAggMembers( unionInst, aggrExpr, *agg, cost, get_member_name(memberExpr) );
1281 } else if ( TupleType * tupleType = dynamic_cast< TupleType * >( aggrExpr->get_result() ) ) {
1282 addTupleMembers( tupleType, aggrExpr, *agg, cost, memberExpr->get_member() );
1283 } // if
1284 } // for
1285 }
1286
1287 void AlternativeFinder::Finder::postvisit( MemberExpr *memberExpr ) {
1288 alternatives.push_back( Alternative{ memberExpr->clone(), env } );
1289 }
1290
1291 void AlternativeFinder::Finder::postvisit( NameExpr *nameExpr ) {
1292 std::list< SymTab::Indexer::IdData > declList;
1293 indexer.lookupId( nameExpr->name, declList );
1294 PRINT( std::cerr << "nameExpr is " << nameExpr->name << std::endl; )
1295 for ( auto & data : declList ) {
1296 Cost cost = Cost::zero;
1297 Expression * newExpr = data.combine( cost );
1298
1299 // addAnonAlternatives uses vector::push_back, which invalidates references to existing elements, so
1300 // can't construct in place and use vector::back
1301 Alternative newAlt{ newExpr, env, OpenVarSet{}, AssertionList{}, Cost::zero, cost };
1302 PRINT(
1303 std::cerr << "decl is ";
1304 data.id->print( std::cerr );
1305 std::cerr << std::endl;
1306 std::cerr << "newExpr is ";
1307 newExpr->print( std::cerr );
1308 std::cerr << std::endl;
1309 )
1310 renameTypes( newAlt.expr );
1311 addAnonConversions( newAlt ); // add anonymous member interpretations whenever an aggregate value type is seen as a name expression.
1312 alternatives.push_back( std::move(newAlt) );
1313 } // for
1314 }
1315
1316 void AlternativeFinder::Finder::postvisit( VariableExpr *variableExpr ) {
1317 // not sufficient to clone here, because variable's type may have changed
1318 // since the VariableExpr was originally created.
1319 alternatives.push_back( Alternative{ new VariableExpr{ variableExpr->var }, env } );
1320 }
1321
1322 void AlternativeFinder::Finder::postvisit( ConstantExpr *constantExpr ) {
1323 alternatives.push_back( Alternative{ constantExpr->clone(), env } );
1324 }
1325
1326 void AlternativeFinder::Finder::postvisit( SizeofExpr *sizeofExpr ) {
1327 if ( sizeofExpr->get_isType() ) {
1328 Type * newType = sizeofExpr->get_type()->clone();
1329 alternatives.push_back( Alternative{
1330 new SizeofExpr{ resolveTypeof( newType, indexer ) }, env } );
1331 } else {
1332 // find all alternatives for the argument to sizeof
1333 AlternativeFinder finder( indexer, env );
1334 finder.find( sizeofExpr->get_expr() );
1335 // find the lowest cost alternative among the alternatives, otherwise ambiguous
1336 AltList winners;
1337 findMinCost( finder.alternatives.begin(), finder.alternatives.end(), back_inserter( winners ) );
1338 if ( winners.size() != 1 ) {
1339 SemanticError( sizeofExpr->get_expr(), "Ambiguous expression in sizeof operand: " );
1340 } // if
1341 // return the lowest cost alternative for the argument
1342 Alternative &choice = winners.front();
1343 referenceToRvalueConversion( choice.expr, choice.cost );
1344 alternatives.push_back( Alternative{
1345 choice, new SizeofExpr( choice.expr->clone() ), Cost::zero } );
1346 } // if
1347 }
1348
1349 void AlternativeFinder::Finder::postvisit( AlignofExpr *alignofExpr ) {
1350 if ( alignofExpr->get_isType() ) {
1351 Type * newType = alignofExpr->get_type()->clone();
1352 alternatives.push_back( Alternative{
1353 new AlignofExpr{ resolveTypeof( newType, indexer ) }, env } );
1354 } else {
1355 // find all alternatives for the argument to sizeof
1356 AlternativeFinder finder( indexer, env );
1357 finder.find( alignofExpr->get_expr() );
1358 // find the lowest cost alternative among the alternatives, otherwise ambiguous
1359 AltList winners;
1360 findMinCost( finder.alternatives.begin(), finder.alternatives.end(), back_inserter( winners ) );
1361 if ( winners.size() != 1 ) {
1362 SemanticError( alignofExpr->get_expr(), "Ambiguous expression in alignof operand: " );
1363 } // if
1364 // return the lowest cost alternative for the argument
1365 Alternative &choice = winners.front();
1366 referenceToRvalueConversion( choice.expr, choice.cost );
1367 alternatives.push_back( Alternative{
1368 choice, new AlignofExpr{ choice.expr->clone() }, Cost::zero } );
1369 } // if
1370 }
1371
1372 template< typename StructOrUnionType >
1373 void AlternativeFinder::Finder::addOffsetof( StructOrUnionType *aggInst, const std::string &name ) {
1374 std::list< Declaration* > members;
1375 aggInst->lookup( name, members );
1376 for ( std::list< Declaration* >::const_iterator i = members.begin(); i != members.end(); ++i ) {
1377 if ( DeclarationWithType *dwt = dynamic_cast< DeclarationWithType* >( *i ) ) {
1378 alternatives.push_back( Alternative{
1379 new OffsetofExpr{ aggInst->clone(), dwt }, env } );
1380 renameTypes( alternatives.back().expr );
1381 } else {
1382 assert( false );
1383 }
1384 }
1385 }
1386
1387 void AlternativeFinder::Finder::postvisit( UntypedOffsetofExpr *offsetofExpr ) {
1388 AlternativeFinder funcFinder( indexer, env );
1389 // xxx - resolveTypeof?
1390 if ( StructInstType *structInst = dynamic_cast< StructInstType* >( offsetofExpr->get_type() ) ) {
1391 addOffsetof( structInst, offsetofExpr->member );
1392 } else if ( UnionInstType *unionInst = dynamic_cast< UnionInstType* >( offsetofExpr->get_type() ) ) {
1393 addOffsetof( unionInst, offsetofExpr->member );
1394 }
1395 }
1396
1397 void AlternativeFinder::Finder::postvisit( OffsetofExpr *offsetofExpr ) {
1398 alternatives.push_back( Alternative{ offsetofExpr->clone(), env } );
1399 }
1400
1401 void AlternativeFinder::Finder::postvisit( OffsetPackExpr *offsetPackExpr ) {
1402 alternatives.push_back( Alternative{ offsetPackExpr->clone(), env } );
1403 }
1404
1405 namespace {
1406 void resolveAttr( SymTab::Indexer::IdData data, FunctionType *function, Type *argType, const TypeEnvironment &env, AlternativeFinder & finder ) {
1407 // assume no polymorphism
1408 // assume no implicit conversions
1409 assert( function->get_parameters().size() == 1 );
1410 PRINT(
1411 std::cerr << "resolvAttr: funcDecl is ";
1412 data.id->print( std::cerr );
1413 std::cerr << " argType is ";
1414 argType->print( std::cerr );
1415 std::cerr << std::endl;
1416 )
1417 const SymTab::Indexer & indexer = finder.get_indexer();
1418 AltList & alternatives = finder.get_alternatives();
1419 if ( typesCompatibleIgnoreQualifiers( argType, function->get_parameters().front()->get_type(), indexer, env ) ) {
1420 Cost cost = Cost::zero;
1421 Expression * newExpr = data.combine( cost );
1422 alternatives.push_back( Alternative{
1423 new AttrExpr{ newExpr, argType->clone() }, env, OpenVarSet{},
1424 AssertionList{}, Cost::zero, cost } );
1425 for ( DeclarationWithType * retVal : function->returnVals ) {
1426 alternatives.back().expr->result = retVal->get_type()->clone();
1427 } // for
1428 } // if
1429 }
1430 }
1431
1432 void AlternativeFinder::Finder::postvisit( AttrExpr *attrExpr ) {
1433 // assume no 'pointer-to-attribute'
1434 NameExpr *nameExpr = dynamic_cast< NameExpr* >( attrExpr->get_attr() );
1435 assert( nameExpr );
1436 std::list< SymTab::Indexer::IdData > attrList;
1437 indexer.lookupId( nameExpr->get_name(), attrList );
1438 if ( attrExpr->get_isType() || attrExpr->get_expr() ) {
1439 for ( auto & data : attrList ) {
1440 DeclarationWithType * id = data.id;
1441 // check if the type is function
1442 if ( FunctionType *function = dynamic_cast< FunctionType* >( id->get_type() ) ) {
1443 // assume exactly one parameter
1444 if ( function->get_parameters().size() == 1 ) {
1445 if ( attrExpr->get_isType() ) {
1446 resolveAttr( data, function, attrExpr->get_type(), env, altFinder);
1447 } else {
1448 AlternativeFinder finder( indexer, env );
1449 finder.find( attrExpr->get_expr() );
1450 for ( AltList::iterator choice = finder.alternatives.begin(); choice != finder.alternatives.end(); ++choice ) {
1451 if ( choice->expr->get_result()->size() == 1 ) {
1452 resolveAttr(data, function, choice->expr->get_result(), choice->env, altFinder );
1453 } // fi
1454 } // for
1455 } // if
1456 } // if
1457 } // if
1458 } // for
1459 } else {
1460 for ( auto & data : attrList ) {
1461 Cost cost = Cost::zero;
1462 Expression * newExpr = data.combine( cost );
1463 alternatives.push_back( Alternative{
1464 newExpr, env, OpenVarSet{}, AssertionList{}, Cost::zero, cost } );
1465 renameTypes( alternatives.back().expr );
1466 } // for
1467 } // if
1468 }
1469
1470 void AlternativeFinder::Finder::postvisit( LogicalExpr *logicalExpr ) {
1471 AlternativeFinder firstFinder( indexer, env );
1472 firstFinder.findWithAdjustment( logicalExpr->get_arg1() );
1473 if ( firstFinder.alternatives.empty() ) return;
1474 AlternativeFinder secondFinder( indexer, env );
1475 secondFinder.findWithAdjustment( logicalExpr->get_arg2() );
1476 if ( secondFinder.alternatives.empty() ) return;
1477 for ( const Alternative & first : firstFinder.alternatives ) {
1478 for ( const Alternative & second : secondFinder.alternatives ) {
1479 TypeEnvironment compositeEnv{ first.env };
1480 compositeEnv.simpleCombine( second.env );
1481 OpenVarSet openVars{ first.openVars };
1482 mergeOpenVars( openVars, second.openVars );
1483 AssertionSet need;
1484 cloneAll( first.need, need );
1485 cloneAll( second.need, need );
1486
1487 LogicalExpr *newExpr = new LogicalExpr{
1488 first.expr->clone(), second.expr->clone(), logicalExpr->get_isAnd() };
1489 alternatives.push_back( Alternative{
1490 newExpr, std::move(compositeEnv), std::move(openVars),
1491 AssertionList( need.begin(), need.end() ), first.cost + second.cost } );
1492 }
1493 }
1494 }
1495
1496 void AlternativeFinder::Finder::postvisit( ConditionalExpr *conditionalExpr ) {
1497 // find alternatives for condition
1498 AlternativeFinder firstFinder( indexer, env );
1499 firstFinder.findWithAdjustment( conditionalExpr->arg1 );
1500 if ( firstFinder.alternatives.empty() ) return;
1501 // find alternatives for true expression
1502 AlternativeFinder secondFinder( indexer, env );
1503 secondFinder.findWithAdjustment( conditionalExpr->arg2 );
1504 if ( secondFinder.alternatives.empty() ) return;
1505 // find alterantives for false expression
1506 AlternativeFinder thirdFinder( indexer, env );
1507 thirdFinder.findWithAdjustment( conditionalExpr->arg3 );
1508 if ( thirdFinder.alternatives.empty() ) return;
1509 for ( const Alternative & first : firstFinder.alternatives ) {
1510 for ( const Alternative & second : secondFinder.alternatives ) {
1511 for ( const Alternative & third : thirdFinder.alternatives ) {
1512 TypeEnvironment compositeEnv{ first.env };
1513 compositeEnv.simpleCombine( second.env );
1514 compositeEnv.simpleCombine( third.env );
1515 OpenVarSet openVars{ first.openVars };
1516 mergeOpenVars( openVars, second.openVars );
1517 mergeOpenVars( openVars, third.openVars );
1518 AssertionSet need;
1519 cloneAll( first.need, need );
1520 cloneAll( second.need, need );
1521 cloneAll( third.need, need );
1522 AssertionSet have;
1523
1524 // unify true and false types, then infer parameters to produce new alternatives
1525 Type* commonType = nullptr;
1526 if ( unify( second.expr->result, third.expr->result, compositeEnv,
1527 need, have, openVars, indexer, commonType ) ) {
1528 ConditionalExpr *newExpr = new ConditionalExpr{
1529 first.expr->clone(), second.expr->clone(), third.expr->clone() };
1530 newExpr->result = commonType ? commonType : second.expr->result->clone();
1531 // convert both options to the conditional result type
1532 Cost cost = first.cost + second.cost + third.cost;
1533 cost += computeExpressionConversionCost(
1534 newExpr->arg2, newExpr->result, indexer, compositeEnv );
1535 cost += computeExpressionConversionCost(
1536 newExpr->arg3, newExpr->result, indexer, compositeEnv );
1537 // output alternative
1538 Alternative newAlt{
1539 newExpr, std::move(compositeEnv), std::move(openVars),
1540 AssertionList( need.begin(), need.end() ), cost };
1541 inferParameters( newAlt, back_inserter( alternatives ) );
1542 } // if
1543 } // for
1544 } // for
1545 } // for
1546 }
1547
1548 void AlternativeFinder::Finder::postvisit( CommaExpr *commaExpr ) {
1549 TypeEnvironment newEnv( env );
1550 Expression *newFirstArg = resolveInVoidContext( commaExpr->get_arg1(), indexer, newEnv );
1551 AlternativeFinder secondFinder( indexer, newEnv );
1552 secondFinder.findWithAdjustment( commaExpr->get_arg2() );
1553 for ( const Alternative & alt : secondFinder.alternatives ) {
1554 alternatives.push_back( Alternative{
1555 alt, new CommaExpr{ newFirstArg->clone(), alt.expr->clone() }, alt.cost } );
1556 } // for
1557 delete newFirstArg;
1558 }
1559
1560 void AlternativeFinder::Finder::postvisit( RangeExpr * rangeExpr ) {
1561 // resolve low and high, accept alternatives whose low and high types unify
1562 AlternativeFinder firstFinder( indexer, env );
1563 firstFinder.findWithAdjustment( rangeExpr->low );
1564 if ( firstFinder.alternatives.empty() ) return;
1565 AlternativeFinder secondFinder( indexer, env );
1566 secondFinder.findWithAdjustment( rangeExpr->high );
1567 if ( secondFinder.alternatives.empty() ) return;
1568 for ( const Alternative & first : firstFinder.alternatives ) {
1569 for ( const Alternative & second : secondFinder.alternatives ) {
1570 TypeEnvironment compositeEnv{ first.env };
1571 compositeEnv.simpleCombine( second.env );
1572 OpenVarSet openVars{ first.openVars };
1573 mergeOpenVars( openVars, second.openVars );
1574 AssertionSet need;
1575 cloneAll( first.need, need );
1576 cloneAll( second.need, need );
1577 AssertionSet have;
1578
1579 Type* commonType = nullptr;
1580 if ( unify( first.expr->result, second.expr->result, compositeEnv, need, have,
1581 openVars, indexer, commonType ) ) {
1582 RangeExpr * newExpr =
1583 new RangeExpr{ first.expr->clone(), second.expr->clone() };
1584 newExpr->result = commonType ? commonType : first.expr->result->clone();
1585 Alternative newAlt{
1586 newExpr, std::move(compositeEnv), std::move(openVars),
1587 AssertionList( need.begin(), need.end() ), first.cost + second.cost };
1588 inferParameters( newAlt, back_inserter( alternatives ) );
1589 } // if
1590 } // for
1591 } // for
1592 }
1593
1594 void AlternativeFinder::Finder::postvisit( UntypedTupleExpr *tupleExpr ) {
1595 std::vector< AlternativeFinder > subExprAlternatives;
1596 altFinder.findSubExprs( tupleExpr->get_exprs().begin(), tupleExpr->get_exprs().end(),
1597 back_inserter( subExprAlternatives ) );
1598 std::vector< AltList > possibilities;
1599 combos( subExprAlternatives.begin(), subExprAlternatives.end(),
1600 back_inserter( possibilities ) );
1601 for ( const AltList& alts : possibilities ) {
1602 std::list< Expression * > exprs;
1603 makeExprList( alts, exprs );
1604
1605 TypeEnvironment compositeEnv;
1606 OpenVarSet openVars;
1607 AssertionSet need;
1608 for ( const Alternative& alt : alts ) {
1609 compositeEnv.simpleCombine( alt.env );
1610 mergeOpenVars( openVars, alt.openVars );
1611 cloneAll( alt.need, need );
1612 }
1613
1614 alternatives.push_back( Alternative{
1615 new TupleExpr{ exprs }, std::move(compositeEnv), std::move(openVars),
1616 AssertionList( need.begin(), need.end() ), sumCost( alts ) } );
1617 } // for
1618 }
1619
1620 void AlternativeFinder::Finder::postvisit( TupleExpr *tupleExpr ) {
1621 alternatives.push_back( Alternative{ tupleExpr->clone(), env } );
1622 }
1623
1624 void AlternativeFinder::Finder::postvisit( ImplicitCopyCtorExpr * impCpCtorExpr ) {
1625 alternatives.push_back( Alternative{ impCpCtorExpr->clone(), env } );
1626 }
1627
1628 void AlternativeFinder::Finder::postvisit( ConstructorExpr * ctorExpr ) {
1629 AlternativeFinder finder( indexer, env );
1630 // don't prune here, since it's guaranteed all alternatives will have the same type
1631 // (giving the alternatives different types is half of the point of ConstructorExpr nodes)
1632 finder.findWithoutPrune( ctorExpr->get_callExpr() );
1633 for ( Alternative & alt : finder.alternatives ) {
1634 alternatives.push_back( Alternative{
1635 alt, new ConstructorExpr( alt.expr->clone() ), alt.cost } );
1636 }
1637 }
1638
1639 void AlternativeFinder::Finder::postvisit( TupleIndexExpr *tupleExpr ) {
1640 alternatives.push_back( Alternative{ tupleExpr->clone(), env } );
1641 }
1642
1643 void AlternativeFinder::Finder::postvisit( TupleAssignExpr *tupleAssignExpr ) {
1644 alternatives.push_back( Alternative{ tupleAssignExpr->clone(), env } );
1645 }
1646
1647 void AlternativeFinder::Finder::postvisit( UniqueExpr *unqExpr ) {
1648 AlternativeFinder finder( indexer, env );
1649 finder.findWithAdjustment( unqExpr->get_expr() );
1650 for ( Alternative & alt : finder.alternatives ) {
1651 // ensure that the id is passed on to the UniqueExpr alternative so that the expressions are "linked"
1652 UniqueExpr * newUnqExpr = new UniqueExpr( alt.expr->clone(), unqExpr->get_id() );
1653 alternatives.push_back( Alternative{ alt, newUnqExpr, alt.cost } );
1654 }
1655 }
1656
1657 void AlternativeFinder::Finder::postvisit( StmtExpr *stmtExpr ) {
1658 StmtExpr * newStmtExpr = stmtExpr->clone();
1659 ResolvExpr::resolveStmtExpr( newStmtExpr, indexer );
1660 // xxx - this env is almost certainly wrong, and needs to somehow contain the combined environments from all of the statements in the stmtExpr...
1661 alternatives.push_back( Alternative{ newStmtExpr, env } );
1662 }
1663
1664 void AlternativeFinder::Finder::postvisit( UntypedInitExpr *initExpr ) {
1665 // handle each option like a cast
1666 AltList candidates;
1667 PRINT(
1668 std::cerr << "untyped init expr: " << initExpr << std::endl;
1669 )
1670 // O(N^2) checks of d-types with e-types
1671 for ( InitAlternative & initAlt : initExpr->get_initAlts() ) {
1672 Type * toType = resolveTypeof( initAlt.type->clone(), indexer );
1673 SymTab::validateType( toType, &indexer );
1674 adjustExprType( toType, env, indexer );
1675 // Ideally the call to findWithAdjustment could be moved out of the loop, but unfortunately it currently has to occur inside or else
1676 // polymorphic return types are not properly bound to the initialization type, since return type variables are only open for the duration of resolving
1677 // the UntypedExpr. This is only actually an issue in initialization contexts that allow more than one possible initialization type, but it is still suboptimal.
1678 AlternativeFinder finder( indexer, env );
1679 finder.targetType = toType;
1680 finder.findWithAdjustment( initExpr->expr );
1681 for ( Alternative & alt : finder.get_alternatives() ) {
1682 TypeEnvironment newEnv( alt.env );
1683 AssertionSet need;
1684 cloneAll( alt.need, need );
1685 AssertionSet have;
1686 OpenVarSet openVars( alt.openVars );
1687 // xxx - find things in env that don't have a "representative type" and claim
1688 // those are open vars?
1689 PRINT(
1690 std::cerr << " @ " << toType << " " << initAlt.designation << std::endl;
1691 )
1692 // It's possible that a cast can throw away some values in a multiply-valued
1693 // expression. (An example is a cast-to-void, which casts from one value to
1694 // zero.) Figure out the prefix of the subexpression results that are cast
1695 // directly. The candidate is invalid if it has fewer results than there are
1696 // types to cast to.
1697 int discardedValues = alt.expr->result->size() - toType->size();
1698 if ( discardedValues < 0 ) continue;
1699 // xxx - may need to go into tuple types and extract relevant types and use
1700 // unifyList. Note that currently, this does not allow casting a tuple to an
1701 // atomic type (e.g. (int)([1, 2, 3]))
1702
1703 // unification run for side-effects
1704 unify( toType, alt.expr->result, newEnv, need, have, openVars, indexer );
1705 // xxx - do some inspecting on this line... why isn't result bound to initAlt.type?
1706
1707 Cost thisCost = castCost( alt.expr->result, toType, indexer, newEnv );
1708 if ( thisCost != Cost::infinity ) {
1709 // count one safe conversion for each value that is thrown away
1710 thisCost.incSafe( discardedValues );
1711 Alternative newAlt{
1712 new InitExpr{
1713 restructureCast( alt.expr->clone(), toType, true ), initAlt.designation->clone() },
1714 std::move(newEnv), std::move(openVars),
1715 AssertionList( need.begin(), need.end() ), alt.cost, thisCost };
1716 inferParameters( newAlt, back_inserter( candidates ) );
1717 }
1718 }
1719 }
1720
1721 // findMinCost selects the alternatives with the lowest "cost" members, but has the side effect of copying the
1722 // cvtCost member to the cost member (since the old cost is now irrelevant). Thus, calling findMinCost twice
1723 // selects first based on argument cost, then on conversion cost.
1724 AltList minArgCost;
1725 findMinCost( candidates.begin(), candidates.end(), std::back_inserter( minArgCost ) );
1726 findMinCost( minArgCost.begin(), minArgCost.end(), std::back_inserter( alternatives ) );
1727 }
1728
1729 void AlternativeFinder::Finder::postvisit( InitExpr * ) {
1730 assertf( false, "AlternativeFinder should never see a resolved InitExpr." );
1731 }
1732
1733 void AlternativeFinder::Finder::postvisit( DeletedExpr * ) {
1734 assertf( false, "AlternativeFinder should never see a DeletedExpr." );
1735 }
1736
1737 void AlternativeFinder::Finder::postvisit( GenericExpr * ) {
1738 assertf( false, "_Generic is not yet supported." );
1739 }
1740} // namespace ResolvExpr
1741
1742// Local Variables: //
1743// tab-width: 4 //
1744// mode: c++ //
1745// compile-command: "make install" //
1746// End: //
Note: See TracBrowser for help on using the repository browser.