source: src/SynTree/TypeSubstitution.h @ 37a3b8f

aaron-thesisarm-ehcleanup-dtorsctordeferred_resndemanglergc_noraiijacob/cs343-translationjenkins-sandboxmemorynew-astnew-ast-unique-exprnew-envno_listpersistent-indexerresolv-newstringwith_gc
Last change on this file since 37a3b8f was 37a3b8f, checked in by Aaron Moss <a3moss@…>, 6 years ago

First code generation for generic types (incorrect)

  • Property mode set to 100644
File size: 6.3 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.h --
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 : Mon May 18 11:12:30 2015
13// Update Count     : 1
14//
15
16#ifndef TYPESUBSTITUTION_H
17#define TYPESUBSTITUTION_H
18
19#include <map>
20#include <set>
21#include <cassert>
22
23#include "SynTree/Mutator.h"
24#include "SynTree/Declaration.h"
25#include "SynTree/Expression.h"
26
27class TypeSubstitution : public Mutator {
28        typedef Mutator Parent;
29  public:
30        TypeSubstitution();
31        template< typename FormalIterator, typename ActualIterator >
32        TypeSubstitution( FormalIterator formalBegin, FormalIterator formalEnd, ActualIterator actualBegin );
33        TypeSubstitution( const TypeSubstitution &other );
34        virtual ~TypeSubstitution();
35       
36        TypeSubstitution &operator=( const TypeSubstitution &other );
37       
38        template< typename SynTreeClass > int apply( SynTreeClass *&input );
39        template< typename SynTreeClass > int applyFree( SynTreeClass *&input );
40       
41        void add( std::string formalType, Type *actualType );
42        void add( const TypeSubstitution &other );
43        void remove( std::string formalType );
44        Type *lookup( std::string formalType ) const;
45        bool empty() const;
46       
47        template< typename FormalIterator, typename ActualIterator >
48        void add( FormalIterator formalBegin, FormalIterator formalEnd, ActualIterator actualBegin );
49       
50        template< typename TypeInstListIterator >
51        void extract( TypeInstListIterator begin, TypeInstListIterator end, TypeSubstitution &result );
52       
53        void normalize();
54
55        void print( std::ostream &os, int indent = 0 ) const;
56        TypeSubstitution *clone() const { return new TypeSubstitution( *this ); }
57  private:
58        virtual Type* mutate(TypeInstType *aggregateUseType);
59        virtual Expression* mutate(NameExpr *nameExpr);
60
61        /// Records type variable bindings from forall-statements
62        template< typename TypeClass > Type *handleType( TypeClass *type );
63        /// Records type variable bindings from forall-statements and instantiations of generic types
64        template< typename TypeClass > Type *handleAggregateType( TypeClass *type );
65       
66        virtual Type* mutate(VoidType *basicType);
67        virtual Type* mutate(BasicType *basicType);
68        virtual Type* mutate(PointerType *pointerType);
69        virtual Type* mutate(ArrayType *arrayType);
70        virtual Type* mutate(FunctionType *functionType);
71        virtual Type* mutate(StructInstType *aggregateUseType);
72        virtual Type* mutate(UnionInstType *aggregateUseType);
73        virtual Type* mutate(EnumInstType *aggregateUseType);
74        virtual Type* mutate(ContextInstType *aggregateUseType);
75        virtual Type* mutate(TupleType *tupleType);
76       
77        // TODO: worry about traversing into a forall-qualified function type or type decl with assertions
78       
79        void initialize( const TypeSubstitution &src, TypeSubstitution &dest );
80
81        typedef std::map< std::string, Type* > TypeEnvType;
82        typedef std::map< std::string, Expression* > VarEnvType;
83        typedef std::set< std::string > BoundVarsType;
84        TypeEnvType typeEnv;
85        VarEnvType varEnv;
86        BoundVarsType boundVars;
87        int subCount;
88        bool freeOnly;
89};
90
91template< typename FormalIterator, typename ActualIterator >
92void TypeSubstitution::add( FormalIterator formalBegin, FormalIterator formalEnd, ActualIterator actualBegin ) {
93        // FormalIterator points to a TypeDecl
94        // ActualIterator points to a Type
95        FormalIterator formalIt = formalBegin;
96        ActualIterator actualIt = actualBegin;
97        for ( ; formalIt != formalEnd; ++formalIt, ++actualIt ) {
98                if ( TypeDecl *formal = dynamic_cast< TypeDecl* >( *formalIt ) ) {
99                        if ( TypeExpr *actual = dynamic_cast< TypeExpr* >( *actualIt ) ) {
100                                if ( formal->get_name() != "" ) {
101                                        TypeEnvType::iterator i = typeEnv.find( formal->get_name() );
102                                        if ( i != typeEnv.end() ) {
103                                                delete i->second;
104                                        } // if
105                                        typeEnv[ formal->get_name() ] = actual->get_type()->clone();
106                                } // if
107                        } else {
108                                throw SemanticError( "Attempt to provide non-type parameter for type parameter", formal );
109                        } // if
110                } else {
111                        // TODO: type check the formal and actual parameters
112                        if ( (*formalIt)->get_name() != "" ) {
113                                varEnv[ (*formalIt)->get_name() ] = (*actualIt)->clone();
114                        } // if
115                } // if
116        } // for
117}
118
119template< typename FormalIterator, typename ActualIterator >
120TypeSubstitution::TypeSubstitution( FormalIterator formalBegin, FormalIterator formalEnd, ActualIterator actualBegin )
121{
122        add( formalBegin, formalEnd, actualBegin );
123}
124
125template< typename SynTreeClass >
126int TypeSubstitution::apply( SynTreeClass *&input ) {
127        assert( input );
128        subCount = 0;
129        freeOnly = false;
130        input = dynamic_cast< SynTreeClass *>( input->acceptMutator( *this ) );
131        assert( input );
132///     std::cout << "substitution result is: ";
133///     newType->print( std::cout );
134///     std::cout << std::endl;
135        return subCount;
136}
137       
138template< typename SynTreeClass >
139int TypeSubstitution::applyFree( SynTreeClass *&input ) {
140        assert( input );
141        subCount = 0;
142        freeOnly = true;
143        input = dynamic_cast< SynTreeClass *>( input->acceptMutator( *this ) );
144        assert( input );
145///     std::cout << "substitution result is: ";
146///     newType->print( std::cout );
147///     std::cout << std::endl;
148        return subCount;
149}
150       
151template< typename TypeInstListIterator >
152void TypeSubstitution::extract( TypeInstListIterator begin, TypeInstListIterator end, TypeSubstitution &result ) {
153        while ( begin != end ) {
154                TypeEnvType::iterator cur = typeEnv.find( (*begin++)->get_name() );
155                if ( cur != typeEnv.end() ) {
156                        result.typeEnv[ cur->first ] = cur->second;
157                        typeEnv.erase( cur );
158                } // if
159        } // while
160}
161
162/// Instantiate each member of the context given the actual parameters specified, and store the
163/// instantiations for use by the indexer
164template< typename FormalIterator, typename ActualIterator, typename MemberIterator, typename OutputIterator >
165void applySubstitution( FormalIterator formalBegin, FormalIterator formalEnd, ActualIterator actual, MemberIterator memberBegin, MemberIterator memberEnd, OutputIterator out ) {
166        TypeSubstitution sub = TypeSubstitution( formalBegin, formalEnd, actual );
167        for ( std::list< Declaration* >::iterator i = memberBegin; i != memberEnd; ++i ) {
168                Declaration *newdecl = (*i)->clone();
169                sub.apply( newdecl );
170                *out++ = newdecl;
171        } // for
172}
173
174#endif // TYPESUBSTITUTION_H
175
176// Local Variables: //
177// tab-width: 4 //
178// mode: c++ //
179// compile-command: "make install" //
180// End: //
Note: See TracBrowser for help on using the repository browser.