source: src/ResolvExpr/Unify.cc @ 3096ec1

aaron-thesisarm-ehcleanup-dtorsdeferred_resndemanglerjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprnew-envno_listpersistent-indexerresolv-newwith_gc
Last change on this file since 3096ec1 was 3096ec1, checked in by Rob Schluntz <rschlunt@…>, 4 years ago

Convert TtypeExpander? to PassVisitor?

  • Property mode set to 100644
File size: 33.1 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// Unify.cc --
8//
9// Author           : Richard C. Bilson
10// Created On       : Sun May 17 12:27:10 2015
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Thu Mar 16 16:22:54 2017
13// Update Count     : 42
14//
15
16#include <cassert>                // for assertf, assert
17#include <iterator>               // for back_insert_iterator, back_inserter
18#include <map>                    // for _Rb_tree_const_iterator, _Rb_tree_i...
19#include <memory>                 // for unique_ptr, auto_ptr
20#include <set>                    // for set
21#include <string>                 // for string, operator==, operator!=, bas...
22#include <utility>                // for pair
23
24#include "Common/PassVisitor.h"   // for PassVisitor
25#include "FindOpenVars.h"         // for findOpenVars
26#include "Parser/LinkageSpec.h"   // for C
27#include "SynTree/Constant.h"     // for Constant
28#include "SynTree/Declaration.h"  // for TypeDecl, TypeDecl::Data, Declarati...
29#include "SynTree/Expression.h"   // for TypeExpr, Expression, ConstantExpr
30#include "SynTree/Mutator.h"      // for Mutator
31#include "SynTree/Type.h"         // for Type, TypeInstType, FunctionType
32#include "SynTree/Visitor.h"      // for Visitor
33#include "Tuples/Tuples.h"        // for isTtype
34#include "TypeEnvironment.h"      // for EqvClass, AssertionSet, OpenVarSet
35#include "Unify.h"
36#include "typeops.h"              // for flatten, occurs, commonType
37
38namespace SymTab {
39class Indexer;
40}  // namespace SymTab
41
42// #define DEBUG
43
44namespace ResolvExpr {
45
46        class Unify : public Visitor {
47          public:
48                Unify( Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, WidenMode widenMode, const SymTab::Indexer &indexer );
49
50                bool get_result() const { return result; }
51          private:
52                virtual void visit(VoidType *voidType);
53                virtual void visit(BasicType *basicType);
54                virtual void visit(PointerType *pointerType);
55                virtual void visit(ArrayType *arrayType);
56                virtual void visit(ReferenceType *refType);
57                virtual void visit(FunctionType *functionType);
58                virtual void visit(StructInstType *aggregateUseType);
59                virtual void visit(UnionInstType *aggregateUseType);
60                virtual void visit(EnumInstType *aggregateUseType);
61                virtual void visit(TraitInstType *aggregateUseType);
62                virtual void visit(TypeInstType *aggregateUseType);
63                virtual void visit(TupleType *tupleType);
64                virtual void visit(VarArgsType *varArgsType);
65                virtual void visit(ZeroType *zeroType);
66                virtual void visit(OneType *oneType);
67
68                template< typename RefType > void handleRefType( RefType *inst, Type *other );
69                template< typename RefType > void handleGenericRefType( RefType *inst, Type *other );
70
71                bool result;
72                Type *type2;                            // inherited
73                TypeEnvironment &env;
74                AssertionSet &needAssertions;
75                AssertionSet &haveAssertions;
76                const OpenVarSet &openVars;
77                WidenMode widenMode;
78                const SymTab::Indexer &indexer;
79        };
80
81        /// Attempts an inexact unification of type1 and type2.
82        /// Returns false if no such unification; if the types can be unified, sets common (unless they unify exactly and have identical type qualifiers)
83        bool unifyInexact( Type *type1, Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, WidenMode widenMode, const SymTab::Indexer &indexer, Type *&common );
84        bool unifyExact( Type *type1, Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, WidenMode widenMode, const SymTab::Indexer &indexer );
85
86        bool typesCompatible( Type *first, Type *second, const SymTab::Indexer &indexer, const TypeEnvironment &env ) {
87                TypeEnvironment newEnv;
88                OpenVarSet openVars, closedVars; // added closedVars
89                AssertionSet needAssertions, haveAssertions;
90                Type *newFirst = first->clone(), *newSecond = second->clone();
91                env.apply( newFirst );
92                env.apply( newSecond );
93
94                // do we need to do this? Seems like we do, types should be able to be compatible if they
95                // have free variables that can unify
96                findOpenVars( newFirst, openVars, closedVars, needAssertions, haveAssertions, false );
97                findOpenVars( newSecond, openVars, closedVars, needAssertions, haveAssertions, true );
98
99                bool result = unifyExact( newFirst, newSecond, newEnv, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
100                delete newFirst;
101                delete newSecond;
102                return result;
103        }
104
105        bool typesCompatibleIgnoreQualifiers( Type *first, Type *second, const SymTab::Indexer &indexer, const TypeEnvironment &env ) {
106                TypeEnvironment newEnv;
107                OpenVarSet openVars;
108                AssertionSet needAssertions, haveAssertions;
109                Type *newFirst = first->clone(), *newSecond = second->clone();
110                env.apply( newFirst );
111                env.apply( newSecond );
112                newFirst->get_qualifiers() = Type::Qualifiers();
113                newSecond->get_qualifiers() = Type::Qualifiers();
114///   std::cerr << "first is ";
115///   first->print( std::cerr );
116///   std::cerr << std::endl << "second is ";
117///   second->print( std::cerr );
118///   std::cerr << std::endl << "newFirst is ";
119///   newFirst->print( std::cerr );
120///   std::cerr << std::endl << "newSecond is ";
121///   newSecond->print( std::cerr );
122///   std::cerr << std::endl;
123                bool result = unifyExact( newFirst, newSecond, newEnv, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
124                delete newFirst;
125                delete newSecond;
126                return result;
127        }
128
129        bool isFtype( Type *type ) {
130                if ( dynamic_cast< FunctionType* >( type ) ) {
131                        return true;
132                } else if ( TypeInstType *typeInst = dynamic_cast< TypeInstType* >( type ) ) {
133                        return typeInst->get_isFtype();
134                } // if
135                return false;
136        }
137
138        bool tyVarCompatible( const TypeDecl::Data & data, Type *type ) {
139                switch ( data.kind ) {
140                  case TypeDecl::Any:
141                  case TypeDecl::Dtype:
142                        // to bind to an object type variable, the type must not be a function type.
143                        // if the type variable is specified to be a complete type then the incoming
144                        // type must also be complete
145                        // xxx - should this also check that type is not a tuple type and that it's not a ttype?
146                        return ! isFtype( type ) && (! data.isComplete || type->isComplete() );
147                  case TypeDecl::Ftype:
148                        return isFtype( type );
149                  case TypeDecl::Ttype:
150                        // ttype unifies with any tuple type
151                        return dynamic_cast< TupleType * >( type ) || Tuples::isTtype( type );
152                } // switch
153                return false;
154        }
155
156        bool bindVar( TypeInstType *typeInst, Type *other, const TypeDecl::Data & data, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, WidenMode widenMode, const SymTab::Indexer &indexer ) {
157                // remove references from other, so that type variables can only bind to value types
158                other = other->stripReferences();
159                OpenVarSet::const_iterator tyvar = openVars.find( typeInst->get_name() );
160                assert( tyvar != openVars.end() );
161                if ( ! tyVarCompatible( tyvar->second, other ) ) {
162                        return false;
163                } // if
164                if ( occurs( other, typeInst->get_name(), env ) ) {
165                        return false;
166                } // if
167                EqvClass curClass;
168                if ( env.lookup( typeInst->get_name(), curClass ) ) {
169                        if ( curClass.type ) {
170                                Type *common = 0;
171                                // attempt to unify equivalence class type (which has qualifiers stripped, so they must be restored) with the type to bind to
172                                std::auto_ptr< Type > newType( curClass.type->clone() );
173                                newType->get_qualifiers() = typeInst->get_qualifiers();
174                                if ( unifyInexact( newType.get(), other, env, needAssertions, haveAssertions, openVars, widenMode & WidenMode( curClass.allowWidening, true ), indexer, common ) ) {
175                                        if ( common ) {
176                                                common->get_qualifiers() = Type::Qualifiers();
177                                                delete curClass.type;
178                                                curClass.type = common;
179                                                env.add( curClass );
180                                        } // if
181                                        return true;
182                                } else {
183                                        return false;
184                                } // if
185                        } else {
186                                curClass.type = other->clone();
187                                curClass.type->get_qualifiers() = Type::Qualifiers();
188                                curClass.allowWidening = widenMode.widenFirst && widenMode.widenSecond;
189                                env.add( curClass );
190                        } // if
191                } else {
192                        EqvClass newClass;
193                        newClass.vars.insert( typeInst->get_name() );
194                        newClass.type = other->clone();
195                        newClass.type->get_qualifiers() = Type::Qualifiers();
196                        newClass.allowWidening = widenMode.widenFirst && widenMode.widenSecond;
197                        newClass.data = data;
198                        env.add( newClass );
199                } // if
200                return true;
201        }
202
203        bool bindVarToVar( TypeInstType *var1, TypeInstType *var2, const TypeDecl::Data & data, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, WidenMode widenMode, const SymTab::Indexer &indexer ) {
204                bool result = true;
205                EqvClass class1, class2;
206                bool hasClass1 = false, hasClass2 = false;
207                bool widen1 = false, widen2 = false;
208                Type *type1 = 0, *type2 = 0;
209
210                if ( env.lookup( var1->get_name(), class1 ) ) {
211                        hasClass1 = true;
212                        if ( class1.type ) {
213                                if ( occurs( class1.type, var2->get_name(), env ) ) {
214                                        return false;
215                                } // if
216                                type1 = class1.type->clone();
217                        } // if
218                        widen1 = widenMode.widenFirst && class1.allowWidening;
219                } // if
220                if ( env.lookup( var2->get_name(), class2 ) ) {
221                        hasClass2 = true;
222                        if ( class2.type ) {
223                                if ( occurs( class2.type, var1->get_name(), env ) ) {
224                                        return false;
225                                } // if
226                                type2 = class2.type->clone();
227                        } // if
228                        widen2 = widenMode.widenSecond && class2.allowWidening;
229                } // if
230
231                if ( type1 && type2 ) {
232//    std::cerr << "has type1 && type2" << std::endl;
233                        WidenMode newWidenMode ( widen1, widen2 );
234                        Type *common = 0;
235                        if ( unifyInexact( type1, type2, env, needAssertions, haveAssertions, openVars, newWidenMode, indexer, common ) ) {
236                                class1.vars.insert( class2.vars.begin(), class2.vars.end() );
237                                class1.allowWidening = widen1 && widen2;
238                                if ( common ) {
239                                        common->get_qualifiers() = Type::Qualifiers();
240                                        delete class1.type;
241                                        class1.type = common;
242                                } // if
243                                env.add( class1 );
244                        } else {
245                                result = false;
246                        } // if
247                } else if ( hasClass1 && hasClass2 ) {
248                        if ( type1 ) {
249                                class1.vars.insert( class2.vars.begin(), class2.vars.end() );
250                                class1.allowWidening = widen1;
251                                env.add( class1 );
252                        } else {
253                                class2.vars.insert( class1.vars.begin(), class1.vars.end() );
254                                class2.allowWidening = widen2;
255                                env.add( class2 );
256                        } // if
257                } else if ( hasClass1 ) {
258                        class1.vars.insert( var2->get_name() );
259                        class1.allowWidening = widen1;
260                        env.add( class1 );
261                } else if ( hasClass2 ) {
262                        class2.vars.insert( var1->get_name() );
263                        class2.allowWidening = widen2;
264                        env.add( class2 );
265                } else {
266                        EqvClass newClass;
267                        newClass.vars.insert( var1->get_name() );
268                        newClass.vars.insert( var2->get_name() );
269                        newClass.allowWidening = widen1 && widen2;
270                        newClass.data = data;
271                        env.add( newClass );
272                } // if
273                delete type1;
274                delete type2;
275                return result;
276        }
277
278        bool unify( Type *type1, Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, OpenVarSet &openVars, const SymTab::Indexer &indexer ) {
279                OpenVarSet closedVars;
280                findOpenVars( type1, openVars, closedVars, needAssertions, haveAssertions, false );
281                findOpenVars( type2, openVars, closedVars, needAssertions, haveAssertions, true );
282                Type *commonType = 0;
283                if ( unifyInexact( type1, type2, env, needAssertions, haveAssertions, openVars, WidenMode( true, true ), indexer, commonType ) ) {
284                        if ( commonType ) {
285                                delete commonType;
286                        } // if
287                        return true;
288                } else {
289                        return false;
290                } // if
291        }
292
293        bool unify( Type *type1, Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, OpenVarSet &openVars, const SymTab::Indexer &indexer, Type *&commonType ) {
294                OpenVarSet closedVars;
295                findOpenVars( type1, openVars, closedVars, needAssertions, haveAssertions, false );
296                findOpenVars( type2, openVars, closedVars, needAssertions, haveAssertions, true );
297                return unifyInexact( type1, type2, env, needAssertions, haveAssertions, openVars, WidenMode( true, true ), indexer, commonType );
298        }
299
300        bool unifyExact( Type *type1, Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, WidenMode widenMode, const SymTab::Indexer &indexer ) {
301#ifdef DEBUG
302                TypeEnvironment debugEnv( env );
303#endif
304                if ( type1->get_qualifiers() != type2->get_qualifiers() ) {
305                        return false;
306                }
307
308                bool result;
309                TypeInstType *var1 = dynamic_cast< TypeInstType* >( type1 );
310                TypeInstType *var2 = dynamic_cast< TypeInstType* >( type2 );
311                OpenVarSet::const_iterator entry1, entry2;
312                if ( var1 ) {
313                        entry1 = openVars.find( var1->get_name() );
314                } // if
315                if ( var2 ) {
316                        entry2 = openVars.find( var2->get_name() );
317                } // if
318                bool isopen1 = var1 && ( entry1 != openVars.end() );
319                bool isopen2 = var2 && ( entry2 != openVars.end() );
320
321                if ( isopen1 && isopen2 && entry1->second == entry2->second ) {
322                        result = bindVarToVar( var1, var2, entry1->second, env, needAssertions, haveAssertions, openVars, widenMode, indexer );
323                } else if ( isopen1 ) {
324                        result = bindVar( var1, type2, entry1->second, env, needAssertions, haveAssertions, openVars, widenMode, indexer );
325                } else if ( isopen2 ) {
326                        result = bindVar( var2, type1, entry2->second, env, needAssertions, haveAssertions, openVars, widenMode, indexer );
327                } else {
328                        Unify comparator( type2, env, needAssertions, haveAssertions, openVars, widenMode, indexer );
329                        type1->accept( comparator );
330                        result = comparator.get_result();
331                } // if
332#ifdef DEBUG
333                std::cerr << "============ unifyExact" << std::endl;
334                std::cerr << "type1 is ";
335                type1->print( std::cerr );
336                std::cerr << std::endl << "type2 is ";
337                type2->print( std::cerr );
338                std::cerr << std::endl << "openVars are ";
339                printOpenVarSet( openVars, std::cerr, 8 );
340                std::cerr << std::endl << "input env is " << std::endl;
341                debugEnv.print( std::cerr, 8 );
342                std::cerr << std::endl << "result env is " << std::endl;
343                env.print( std::cerr, 8 );
344                std::cerr << "result is " << result << std::endl;
345#endif
346                return result;
347        }
348
349        bool unifyExact( Type *type1, Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, OpenVarSet &openVars, const SymTab::Indexer &indexer ) {
350                return unifyExact( type1, type2, env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
351        }
352
353        bool unifyInexact( Type *type1, Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, WidenMode widenMode, const SymTab::Indexer &indexer, Type *&common ) {
354                Type::Qualifiers tq1 = type1->get_qualifiers(), tq2 = type2->get_qualifiers();
355                type1->get_qualifiers() = Type::Qualifiers();
356                type2->get_qualifiers() = Type::Qualifiers();
357                bool result;
358#ifdef DEBUG
359                std::cerr << "unifyInexact type 1 is ";
360                type1->print( std::cerr );
361                std::cerr << " type 2 is ";
362                type2->print( std::cerr );
363                std::cerr << std::endl;
364#endif
365                if ( ! unifyExact( type1, type2, env, needAssertions, haveAssertions, openVars, widenMode, indexer ) ) {
366#ifdef DEBUG
367                        std::cerr << "unifyInexact: no exact unification found" << std::endl;
368#endif
369                        if ( ( common = commonType( type1, type2, widenMode.widenFirst, widenMode.widenSecond, indexer, env, openVars ) ) ) {
370                                common->get_qualifiers() = tq1 | tq2;
371#ifdef DEBUG
372                                std::cerr << "unifyInexact: common type is ";
373                                common->print( std::cerr );
374                                std::cerr << std::endl;
375#endif
376                                result = true;
377                        } else {
378#ifdef DEBUG
379                                std::cerr << "unifyInexact: no common type found" << std::endl;
380#endif
381                                result = false;
382                        } // if
383                } else {
384                        if ( tq1 != tq2 ) {
385                                if ( ( tq1 > tq2 || widenMode.widenFirst ) && ( tq2 > tq1 || widenMode.widenSecond ) ) {
386                                        common = type1->clone();
387                                        common->get_qualifiers() = tq1 | tq2;
388                                        result = true;
389                                } else {
390                                        result = false;
391                                } // if
392                        } else {
393                                common = type1->clone();
394                                common->get_qualifiers() = tq1 | tq2;
395                                result = true;
396                        } // if
397                } // if
398                type1->get_qualifiers() = tq1;
399                type2->get_qualifiers() = tq2;
400                return result;
401        }
402
403        Unify::Unify( Type *type2, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, WidenMode widenMode, const SymTab::Indexer &indexer )
404                : result( false ), type2( type2 ), env( env ), needAssertions( needAssertions ), haveAssertions( haveAssertions ), openVars( openVars ), widenMode( widenMode ), indexer( indexer ) {
405        }
406
407        void Unify::visit( __attribute__((unused)) VoidType *voidType) {
408                result = dynamic_cast< VoidType* >( type2 );
409        }
410
411        void Unify::visit(BasicType *basicType) {
412                if ( BasicType *otherBasic = dynamic_cast< BasicType* >( type2 ) ) {
413                        result = basicType->get_kind() == otherBasic->get_kind();
414                } // if
415        }
416
417        void markAssertionSet( AssertionSet &assertions, DeclarationWithType *assert ) {
418///   std::cerr << "assertion set is" << std::endl;
419///   printAssertionSet( assertions, std::cerr, 8 );
420///   std::cerr << "looking for ";
421///   assert->print( std::cerr );
422///   std::cerr << std::endl;
423                AssertionSet::iterator i = assertions.find( assert );
424                if ( i != assertions.end() ) {
425///     std::cerr << "found it!" << std::endl;
426                        i->second.isUsed = true;
427                } // if
428        }
429
430        void markAssertions( AssertionSet &assertion1, AssertionSet &assertion2, Type *type ) {
431                for ( std::list< TypeDecl* >::const_iterator tyvar = type->get_forall().begin(); tyvar != type->get_forall().end(); ++tyvar ) {
432                        for ( std::list< DeclarationWithType* >::const_iterator assert = (*tyvar)->get_assertions().begin(); assert != (*tyvar)->get_assertions().end(); ++assert ) {
433                                markAssertionSet( assertion1, *assert );
434                                markAssertionSet( assertion2, *assert );
435                        } // for
436                } // for
437        }
438
439        void Unify::visit(PointerType *pointerType) {
440                if ( PointerType *otherPointer = dynamic_cast< PointerType* >( type2 ) ) {
441                        result = unifyExact( pointerType->get_base(), otherPointer->get_base(), env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
442                        markAssertions( haveAssertions, needAssertions, pointerType );
443                        markAssertions( haveAssertions, needAssertions, otherPointer );
444                } // if
445        }
446
447        void Unify::visit(ReferenceType *refType) {
448                if ( ReferenceType *otherRef = dynamic_cast< ReferenceType* >( type2 ) ) {
449                        result = unifyExact( refType->get_base(), otherRef->get_base(), env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
450                        markAssertions( haveAssertions, needAssertions, refType );
451                        markAssertions( haveAssertions, needAssertions, otherRef );
452                } // if
453        }
454
455        void Unify::visit(ArrayType *arrayType) {
456                ArrayType *otherArray = dynamic_cast< ArrayType* >( type2 );
457                // to unify, array types must both be VLA or both not VLA
458                // and must both have a dimension expression or not have a dimension
459                if ( otherArray && arrayType->get_isVarLen() == otherArray->get_isVarLen() ) {
460
461                        // not positive this is correct in all cases, but it's needed for typedefs
462                        if ( arrayType->get_isVarLen() || otherArray->get_isVarLen() ) {
463                                return;
464                        }
465
466                        if ( ! arrayType->get_isVarLen() && ! otherArray->get_isVarLen() &&
467                                arrayType->get_dimension() != 0 && otherArray->get_dimension() != 0 ) {
468                                ConstantExpr * ce1 = dynamic_cast< ConstantExpr * >( arrayType->get_dimension() );
469                                ConstantExpr * ce2 = dynamic_cast< ConstantExpr * >( otherArray->get_dimension() );
470                                // see C11 Reference Manual 6.7.6.2.6
471                                // two array types with size specifiers that are integer constant expressions are
472                                // compatible if both size specifiers have the same constant value
473                                if ( ce1 && ce2 ) {
474                                        Constant * c1 = ce1->get_constant();
475                                        Constant * c2 = ce2->get_constant();
476
477                                        if ( c1->get_value() != c2->get_value() ) {
478                                                // does not unify if the dimension is different
479                                                return;
480                                        }
481                                }
482                        }
483
484                        result = unifyExact( arrayType->get_base(), otherArray->get_base(), env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
485                } // if
486        }
487
488        template< typename Iterator, typename Func >
489        std::unique_ptr<Type> combineTypes( Iterator begin, Iterator end, Func & toType ) {
490                std::list< Type * > types;
491                for ( ; begin != end; ++begin ) {
492                        // it's guaranteed that a ttype variable will be bound to a flat tuple, so ensure that this results in a flat tuple
493                        flatten( toType( *begin ), back_inserter( types ) );
494                }
495                return std::unique_ptr<Type>( new TupleType( Type::Qualifiers(), types ) );
496        }
497
498        template< typename Iterator1, typename Iterator2 >
499        bool unifyDeclList( Iterator1 list1Begin, Iterator1 list1End, Iterator2 list2Begin, Iterator2 list2End, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, const SymTab::Indexer &indexer ) {
500                auto get_type = [](DeclarationWithType * dwt){ return dwt->get_type(); };
501                for ( ; list1Begin != list1End && list2Begin != list2End; ++list1Begin, ++list2Begin ) {
502                        Type * t1 = (*list1Begin)->get_type();
503                        Type * t2 = (*list2Begin)->get_type();
504                        bool isTtype1 = Tuples::isTtype( t1 );
505                        bool isTtype2 = Tuples::isTtype( t2 );
506                        // xxx - assumes ttype must be last parameter
507                        // xxx - there may be a nice way to refactor this, but be careful because the argument positioning might matter in some cases.
508                        if ( isTtype1 && ! isTtype2 ) {
509                                // combine all of the things in list2, then unify
510                                return unifyExact( t1, combineTypes( list2Begin, list2End, get_type ).get(), env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
511                        } else if ( isTtype2 && ! isTtype1 ) {
512                                // combine all of the things in list1, then unify
513                                return unifyExact( combineTypes( list1Begin, list1End, get_type ).get(), t2, env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
514                        } else if ( ! unifyExact( t1, t2, env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer ) ) {
515                                return false;
516                        } // if
517                } // for
518                // may get to the end of one argument list before the end of the other. This is only okay when the other is a ttype
519                if ( list1Begin != list1End ) {
520                        // try unifying empty tuple type with ttype
521                        Type * t1 = (*list1Begin)->get_type();
522                        if ( Tuples::isTtype( t1 ) ) {
523                                return unifyExact( t1, combineTypes( list2Begin, list2End, get_type ).get(), env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
524                        } else return false;
525                } else if ( list2Begin != list2End ) {
526                        // try unifying empty tuple type with ttype
527                        Type * t2 = (*list2Begin)->get_type();
528                        if ( Tuples::isTtype( t2 ) ) {
529                                return unifyExact( combineTypes( list1Begin, list1End, get_type ).get(), t2, env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
530                        } else return false;
531                } else {
532                        return true;
533                } // if
534        }
535
536        /// Finds ttypes and replaces them with their expansion, if known.
537        /// This needs to be done so that satisfying ttype assertions is easier.
538        /// If this isn't done then argument lists can have wildly different
539        /// size and structure, when they should be compatible.
540        struct TtypeExpander : public WithShortCircuiting {
541                TypeEnvironment & tenv;
542                TtypeExpander( TypeEnvironment & tenv ) : tenv( tenv ) {}
543                void premutate( TypeInstType * ) { visit_children = false; }
544                Type * postmutate( TypeInstType * typeInst ) {
545                        EqvClass eqvClass;
546                        if ( tenv.lookup( typeInst->get_name(), eqvClass ) ) {
547                                if ( eqvClass.data.kind == TypeDecl::Ttype ) {
548                                        // expand ttype parameter into its actual type
549                                        if ( eqvClass.type ) {
550                                                delete typeInst;
551                                                return eqvClass.type->clone();
552                                        }
553                                }
554                        }
555                        return typeInst;
556                }
557        };
558
559        /// flattens a list of declarations, so that each tuple type has a single declaration.
560        /// makes use of TtypeExpander to ensure ttypes are flat as well.
561        void flattenList( std::list< DeclarationWithType * > src, std::list< DeclarationWithType * > & dst, TypeEnvironment & env ) {
562                dst.clear();
563                for ( DeclarationWithType * dcl : src ) {
564                        PassVisitor<TtypeExpander> expander( env );
565                        dcl->acceptMutator( expander );
566                        std::list< Type * > types;
567                        flatten( dcl->get_type(), back_inserter( types ) );
568                        for ( Type * t : types ) {
569                                dst.push_back( new ObjectDecl( "", Type::StorageClasses(), LinkageSpec::C, nullptr, t, nullptr ) );
570                        }
571                        delete dcl;
572                }
573        }
574
575        void Unify::visit(FunctionType *functionType) {
576                FunctionType *otherFunction = dynamic_cast< FunctionType* >( type2 );
577                if ( otherFunction && functionType->get_isVarArgs() == otherFunction->get_isVarArgs() ) {
578                        // flatten the parameter lists for both functions so that tuple structure
579                        // doesn't affect unification. Must be a clone so that the types don't change.
580                        std::unique_ptr<FunctionType> flatFunc( functionType->clone() );
581                        std::unique_ptr<FunctionType> flatOther( otherFunction->clone() );
582                        flattenList( flatFunc->get_parameters(), flatFunc->get_parameters(), env );
583                        flattenList( flatOther->get_parameters(), flatOther->get_parameters(), env );
584
585                        // sizes don't have to match if ttypes are involved; need to be more precise wrt where the ttype is to prevent errors
586                        if ( (flatFunc->get_parameters().size() == flatOther->get_parameters().size() && flatFunc->get_returnVals().size() == flatOther->get_returnVals().size()) || flatFunc->isTtype() || flatOther->isTtype() ) {
587                                if ( unifyDeclList( flatFunc->get_parameters().begin(), flatFunc->get_parameters().end(), flatOther->get_parameters().begin(), flatOther->get_parameters().end(), env, needAssertions, haveAssertions, openVars, indexer ) ) {
588                                        if ( unifyDeclList( flatFunc->get_returnVals().begin(), flatFunc->get_returnVals().end(), flatOther->get_returnVals().begin(), flatOther->get_returnVals().end(), env, needAssertions, haveAssertions, openVars, indexer ) ) {
589
590                                                // the original types must be used in mark assertions, since pointer comparisons are used
591                                                markAssertions( haveAssertions, needAssertions, functionType );
592                                                markAssertions( haveAssertions, needAssertions, otherFunction );
593
594                                                result = true;
595                                        } // if
596                                } // if
597                        } // if
598                } // if
599        }
600
601        template< typename RefType >
602        void Unify::handleRefType( RefType *inst, Type *other ) {
603                // check that other type is compatible and named the same
604                RefType *otherStruct = dynamic_cast< RefType* >( other );
605                result = otherStruct && inst->get_name() == otherStruct->get_name();
606        }
607
608        template< typename RefType >
609        void Unify::handleGenericRefType( RefType *inst, Type *other ) {
610                // Check that other type is compatible and named the same
611                handleRefType( inst, other );
612                if ( ! result ) return;
613                // Check that parameters of types unify, if any
614                std::list< Expression* > params = inst->get_parameters();
615                std::list< Expression* > otherParams = ((RefType*)other)->get_parameters();
616
617                std::list< Expression* >::const_iterator it = params.begin(), jt = otherParams.begin();
618                for ( ; it != params.end() && jt != otherParams.end(); ++it, ++jt ) {
619                        TypeExpr *param = dynamic_cast< TypeExpr* >(*it);
620                        assertf(param, "Aggregate parameters should be type expressions");
621                        TypeExpr *otherParam = dynamic_cast< TypeExpr* >(*jt);
622                        assertf(otherParam, "Aggregate parameters should be type expressions");
623
624                        Type* paramTy = param->get_type();
625                        Type* otherParamTy = otherParam->get_type();
626
627                        bool tupleParam = Tuples::isTtype( paramTy );
628                        bool otherTupleParam = Tuples::isTtype( otherParamTy );
629
630                        if ( tupleParam && otherTupleParam ) {
631                                ++it; ++jt;  // skip ttype parameters for break
632                        } else if ( tupleParam ) {
633                                // bundle other parameters into tuple to match
634                                std::list< Type * > binderTypes;
635
636                                do {
637                                        binderTypes.push_back( otherParam->get_type()->clone() );
638                                        ++jt;
639
640                                        if ( jt == otherParams.end() ) break;
641
642                                        otherParam = dynamic_cast< TypeExpr* >(*jt);
643                                        assertf(otherParam, "Aggregate parameters should be type expressions");
644                                } while (true);
645
646                                otherParamTy = new TupleType{ paramTy->get_qualifiers(), binderTypes };
647                                ++it;  // skip ttype parameter for break
648                        } else if ( otherTupleParam ) {
649                                // bundle parameters into tuple to match other
650                                std::list< Type * > binderTypes;
651
652                                do {
653                                        binderTypes.push_back( param->get_type()->clone() );
654                                        ++it;
655
656                                        if ( it == params.end() ) break;
657
658                                        param = dynamic_cast< TypeExpr* >(*it);
659                                        assertf(param, "Aggregate parameters should be type expressions");
660                                } while (true);
661
662                                paramTy = new TupleType{ otherParamTy->get_qualifiers(), binderTypes };
663                                ++jt;  // skip ttype parameter for break
664                        }
665
666                        if ( ! unifyExact( paramTy, otherParamTy, env, needAssertions, haveAssertions, openVars, WidenMode(false, false), indexer ) ) {
667                                result = false;
668                                return;
669                        }
670
671                        // ttype parameter should be last
672                        if ( tupleParam || otherTupleParam ) break;
673                }
674                result = ( it == params.end() && jt == otherParams.end() );
675        }
676
677        void Unify::visit(StructInstType *structInst) {
678                handleGenericRefType( structInst, type2 );
679        }
680
681        void Unify::visit(UnionInstType *unionInst) {
682                handleGenericRefType( unionInst, type2 );
683        }
684
685        void Unify::visit(EnumInstType *enumInst) {
686                handleRefType( enumInst, type2 );
687        }
688
689        void Unify::visit(TraitInstType *contextInst) {
690                handleRefType( contextInst, type2 );
691        }
692
693        void Unify::visit(TypeInstType *typeInst) {
694                assert( openVars.find( typeInst->get_name() ) == openVars.end() );
695                TypeInstType *otherInst = dynamic_cast< TypeInstType* >( type2 );
696                if ( otherInst && typeInst->get_name() == otherInst->get_name() ) {
697                        result = true;
698///   } else {
699///     NamedTypeDecl *nt = indexer.lookupType( typeInst->get_name() );
700///     if ( nt ) {
701///       TypeDecl *type = dynamic_cast< TypeDecl* >( nt );
702///       assert( type );
703///       if ( type->get_base() ) {
704///         result = unifyExact( type->get_base(), typeInst, env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
705///       }
706///     }
707                } // if
708        }
709
710        template< typename Iterator1, typename Iterator2 >
711        bool unifyList( Iterator1 list1Begin, Iterator1 list1End, Iterator2 list2Begin, Iterator2 list2End, TypeEnvironment &env, AssertionSet &needAssertions, AssertionSet &haveAssertions, const OpenVarSet &openVars, const SymTab::Indexer &indexer ) {
712                auto get_type = [](Type * t) { return t; };
713                for ( ; list1Begin != list1End && list2Begin != list2End; ++list1Begin, ++list2Begin ) {
714                        Type * t1 = *list1Begin;
715                        Type * t2 = *list2Begin;
716                        bool isTtype1 = Tuples::isTtype( t1 );
717                        bool isTtype2 = Tuples::isTtype( t2 );
718                        // xxx - assumes ttype must be last parameter
719                        // xxx - there may be a nice way to refactor this, but be careful because the argument positioning might matter in some cases.
720                        if ( isTtype1 && ! isTtype2 ) {
721                                // combine all of the things in list2, then unify
722                                return unifyExact( t1, combineTypes( list2Begin, list2End, get_type ).get(), env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
723                        } else if ( isTtype2 && ! isTtype1 ) {
724                                // combine all of the things in list1, then unify
725                                return unifyExact( combineTypes( list1Begin, list1End, get_type ).get(), t2, env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
726                        } else if ( ! unifyExact( t1, t2, env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer ) ) {
727                                return false;
728                        } // if
729
730                } // for
731                if ( list1Begin != list1End ) {
732                        // try unifying empty tuple type with ttype
733                        Type * t1 = *list1Begin;
734                        if ( Tuples::isTtype( t1 ) ) {
735                                return unifyExact( t1, combineTypes( list2Begin, list2End, get_type ).get(), env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
736                        } else return false;
737                } else if ( list2Begin != list2End ) {
738                        // try unifying empty tuple type with ttype
739                        Type * t2 = *list2Begin;
740                        if ( Tuples::isTtype( t2 ) ) {
741                                return unifyExact( combineTypes( list1Begin, list1End, get_type ).get(), t2, env, needAssertions, haveAssertions, openVars, WidenMode( false, false ), indexer );
742                        } else return false;
743                } else {
744                        return true;
745                } // if
746        }
747
748        void Unify::visit(TupleType *tupleType) {
749                if ( TupleType *otherTuple = dynamic_cast< TupleType* >( type2 ) ) {
750                        std::unique_ptr<TupleType> flat1( tupleType->clone() );
751                        std::unique_ptr<TupleType> flat2( otherTuple->clone() );
752                        std::list<Type *> types1, types2;
753
754                        PassVisitor<TtypeExpander> expander( env );
755                        flat1->acceptMutator( expander );
756                        flat2->acceptMutator( expander );
757
758                        flatten( flat1.get(), back_inserter( types1 ) );
759                        flatten( flat2.get(), back_inserter( types2 ) );
760
761                        result = unifyList( types1.begin(), types1.end(), types2.begin(), types2.end(), env, needAssertions, haveAssertions, openVars, indexer );
762                } // if
763        }
764
765        void Unify::visit( __attribute__((unused)) VarArgsType *varArgsType ) {
766                result = dynamic_cast< VarArgsType* >( type2 );
767        }
768
769        void Unify::visit( __attribute__((unused)) ZeroType *zeroType ) {
770                result = dynamic_cast< ZeroType* >( type2 );
771        }
772
773        void Unify::visit( __attribute__((unused)) OneType *oneType ) {
774                result = dynamic_cast< OneType* >( type2 );
775        }
776
777        // xxx - compute once and store in the FunctionType?
778        Type * extractResultType( FunctionType * function ) {
779                if ( function->get_returnVals().size() == 0 ) {
780                        return new VoidType( Type::Qualifiers() );
781                } else if ( function->get_returnVals().size() == 1 ) {
782                        return function->get_returnVals().front()->get_type()->clone();
783                } else {
784                        std::list< Type * > types;
785                        for ( DeclarationWithType * decl : function->get_returnVals() ) {
786                                types.push_back( decl->get_type()->clone() );
787                        } // for
788                        return new TupleType( Type::Qualifiers(), types );
789                }
790        }
791} // namespace ResolvExpr
792
793// Local Variables: //
794// tab-width: 4 //
795// mode: c++ //
796// compile-command: "make install" //
797// End: //
Note: See TracBrowser for help on using the repository browser.