source: src/SynTree/TypeSubstitution.cc @ ea6332d

aaron-thesisarm-ehcleanup-dtorsdeferred_resndemanglerjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprnew-envno_listpersistent-indexerresolv-newwith_gc
Last change on this file since ea6332d was ea6332d, checked in by Thierry Delisle <tdelisle@…>, 4 years ago

Big header cleaning pass - commit 3

  • Property mode set to 100644
File size: 7.6 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// TypeSubstitution.cc --
8//
9// Author           : Richard C. Bilson
10// Created On       : Mon May 18 07:44:20 2015
11// Last Modified By : Peter A. Buhr
12// Last Modified On : Thu Mar 16 15:54:35 2017
13// Update Count     : 4
14//
15
16#include <ostream>  // for ostream, basic_ostream, operator<<, endl
17
18#include "Type.h"   // for TypeInstType, Type, StructInstType, UnionInstType
19#include "TypeSubstitution.h"
20
21TypeSubstitution::TypeSubstitution() {
22}
23
24TypeSubstitution::TypeSubstitution( const TypeSubstitution &other ) {
25        initialize( other, *this );
26}
27
28TypeSubstitution::~TypeSubstitution() {
29        for ( TypeEnvType::iterator i = typeEnv.begin(); i != typeEnv.end(); ++i ) {
30                delete( i->second );
31        }
32        for ( VarEnvType::iterator i = varEnv.begin(); i != varEnv.end(); ++i ) {
33                delete( i->second );
34        }
35}
36
37TypeSubstitution &TypeSubstitution::operator=( const TypeSubstitution &other ) {
38        if ( this == &other ) return *this;
39        initialize( other, *this );
40        return *this;
41}
42
43void TypeSubstitution::initialize( const TypeSubstitution &src, TypeSubstitution &dest ) {
44        dest.typeEnv.clear();
45        dest.varEnv.clear();
46        dest.add( src );
47}
48
49void TypeSubstitution::add( const TypeSubstitution &other ) {
50        for ( TypeEnvType::const_iterator i = other.typeEnv.begin(); i != other.typeEnv.end(); ++i ) {
51                typeEnv[ i->first ] = i->second->clone();
52        } // for
53        for ( VarEnvType::const_iterator i = other.varEnv.begin(); i != other.varEnv.end(); ++i ) {
54                varEnv[ i->first ] = i->second->clone();
55        } // for
56}
57
58void TypeSubstitution::add( std::string formalType, Type *actualType ) {
59        TypeEnvType::iterator i = typeEnv.find( formalType );
60        if ( i != typeEnv.end() ) {
61                delete i->second;
62        } // if
63        typeEnv[ formalType ] = actualType->clone();
64}
65
66void TypeSubstitution::remove( std::string formalType ) {
67        TypeEnvType::iterator i = typeEnv.find( formalType );
68        if ( i != typeEnv.end() ) {
69                delete i->second;
70                typeEnv.erase( formalType );
71        } // if
72}
73
74Type *TypeSubstitution::lookup( std::string formalType ) const {
75        TypeEnvType::const_iterator i = typeEnv.find( formalType );
76
77        // break on not in substitution set
78        if ( i == typeEnv.end() ) return 0;
79
80        // attempt to transitively follow TypeInstType links.
81        while ( TypeInstType *actualType = dynamic_cast< TypeInstType* >( i->second ) ) {
82                const std::string& typeName = actualType->get_name();
83
84                // break cycles in the transitive follow
85                if ( formalType == typeName ) break;
86
87                // Look for the type this maps to, returning previous mapping if none-such
88                i = typeEnv.find( typeName );
89                if ( i == typeEnv.end() ) return actualType;
90        }
91
92        // return type from substitution set
93        return i->second;
94
95#if 0
96        if ( i == typeEnv.end() ) {
97                return 0;
98        } else {
99                return i->second;
100        } // if
101#endif
102}
103
104bool TypeSubstitution::empty() const {
105        return typeEnv.empty() && varEnv.empty();
106}
107
108void TypeSubstitution::normalize() {
109        do {
110                subCount = 0;
111                freeOnly = true;
112                for ( TypeEnvType::iterator i = typeEnv.begin(); i != typeEnv.end(); ++i ) {
113                        i->second = i->second->acceptMutator( *this );
114                }
115        } while ( subCount );
116}
117
118Type * TypeSubstitution::mutate( TypeInstType *inst ) {
119        BoundVarsType::const_iterator bound = boundVars.find( inst->get_name() );
120        if ( bound != boundVars.end() ) return inst;
121
122        TypeEnvType::const_iterator i = typeEnv.find( inst->get_name() );
123        if ( i == typeEnv.end() ) {
124                return inst;
125        } else {
126///         std::cout << "found " << inst->get_name() << ", replacing with ";
127///         i->second->print( std::cout );
128///         std::cout << std::endl;
129                subCount++;
130                Type *newtype = i->second->clone();
131                newtype->get_qualifiers() |= inst->get_qualifiers();
132                delete inst;
133                return newtype;
134        } // if
135}
136
137Expression * TypeSubstitution::mutate( NameExpr *nameExpr ) {
138        VarEnvType::const_iterator i = varEnv.find( nameExpr->get_name() );
139        if ( i == varEnv.end() ) {
140                return nameExpr;
141        } else {
142                subCount++;
143                delete nameExpr;
144                return i->second->clone();
145        } // if
146}
147
148template< typename TypeClass >
149Type *TypeSubstitution::handleType( TypeClass *type ) {
150        BoundVarsType oldBoundVars( boundVars );
151        // bind type variables from forall-qualifiers
152        if ( freeOnly ) {
153                for ( Type::ForallList::const_iterator tyvar = type->get_forall().begin(); tyvar != type->get_forall().end(); ++tyvar ) {
154                        boundVars.insert( (*tyvar )->get_name() );
155                } // for
156        } // if
157        Type *ret = Mutator::mutate( type );
158        boundVars = oldBoundVars;
159        return ret;
160}
161
162template< typename TypeClass >
163Type *TypeSubstitution::handleAggregateType( TypeClass *type ) {
164        BoundVarsType oldBoundVars( boundVars );
165        // bind type variables from forall-qualifiers
166        if ( freeOnly ) {
167                for ( Type::ForallList::const_iterator tyvar = type->get_forall().begin(); tyvar != type->get_forall().end(); ++tyvar ) {
168                        boundVars.insert( (*tyvar )->get_name() );
169                } // for
170                // bind type variables from generic type instantiations
171                std::list< TypeDecl* > *baseParameters = type->get_baseParameters();
172                if ( baseParameters && ! type->get_parameters().empty() ) {
173                        for ( std::list< TypeDecl* >::const_iterator tyvar = baseParameters->begin(); tyvar != baseParameters->end(); ++tyvar ) {
174                                boundVars.insert( (*tyvar)->get_name() );
175                        } // for
176                } // if
177        } // if
178        Type *ret = Mutator::mutate( type );
179        boundVars = oldBoundVars;
180        return ret;
181}
182
183Type * TypeSubstitution::mutate( VoidType *voidType ) {
184        return handleType( voidType );
185}
186
187Type * TypeSubstitution::mutate( BasicType *basicType ) {
188        return handleType( basicType );
189}
190
191Type * TypeSubstitution::mutate( PointerType *pointerType ) {
192        return handleType( pointerType );
193}
194
195Type * TypeSubstitution::mutate( ArrayType *arrayType ) {
196        return handleType( arrayType );
197}
198
199Type * TypeSubstitution::mutate( FunctionType *functionType ) {
200        return handleType( functionType );
201}
202
203Type * TypeSubstitution::mutate( StructInstType *aggregateUseType ) {
204        return handleAggregateType( aggregateUseType );
205}
206
207Type * TypeSubstitution::mutate( UnionInstType *aggregateUseType ) {
208        return handleAggregateType( aggregateUseType );
209}
210
211Type * TypeSubstitution::mutate( EnumInstType *aggregateUseType ) {
212        return handleType( aggregateUseType );
213}
214
215Type * TypeSubstitution::mutate( TraitInstType *aggregateUseType ) {
216        return handleType( aggregateUseType );
217}
218
219Type * TypeSubstitution::mutate( TupleType *tupleType ) {
220        return handleType( tupleType );
221}
222
223Type * TypeSubstitution::mutate( VarArgsType *varArgsType ) {
224        return handleType( varArgsType );
225}
226
227Type * TypeSubstitution::mutate( ZeroType *zeroType ) {
228        return handleType( zeroType );
229}
230
231Type * TypeSubstitution::mutate( OneType *oneType ) {
232        return handleType( oneType );
233}
234
235TypeSubstitution * TypeSubstitution::acceptMutator( Mutator & mutator ) {
236        for ( auto & p : typeEnv ) {
237                p.second = maybeMutate( p.second, mutator );
238        }
239        for ( auto & p : varEnv ) {
240                p.second = maybeMutate( p.second, mutator );
241        }
242        return this;
243}
244
245void TypeSubstitution::print( std::ostream &os, int indent ) const {
246        os << std::string( indent, ' ' ) << "Types:" << std::endl;
247        for ( TypeEnvType::const_iterator i = typeEnv.begin(); i != typeEnv.end(); ++i ) {
248                os << std::string( indent+2, ' ' ) << i->first << " -> ";
249                i->second->print( os, indent+4 );
250                os << std::endl;
251        } // for
252        os << std::string( indent, ' ' ) << "Non-types:" << std::endl;
253        for ( VarEnvType::const_iterator i = varEnv.begin(); i != varEnv.end(); ++i ) {
254                os << std::string( indent+2, ' ' ) << i->first << " -> ";
255                i->second->print( os, indent+4 );
256                os << std::endl;
257        } // for
258}
259
260std::ostream & operator<<( std::ostream & out, const TypeSubstitution & sub ) {
261        sub.print( out );
262        return out;
263}
264
265
266// Local Variables: //
267// tab-width: 4 //
268// mode: c++ //
269// compile-command: "make install" //
270// End: //
Note: See TracBrowser for help on using the repository browser.