source: src/GenPoly/GenPoly.cc @ 0f889a77

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

Added unfiltered version of isPoly{Type,Ptr}

  • Property mode set to 100644
File size: 6.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// GenPoly.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 : Tue Dec 15 16:11:18 2015
13// Update Count     : 13
14//
15
16#include "GenPoly.h"
17
18#include "SymTab/Mangler.h"
19#include "SynTree/Expression.h"
20#include "SynTree/Type.h"
21
22#include <iostream>
23using namespace std;
24
25namespace GenPoly {
26        bool needsAdapter( FunctionType *adaptee, const TyVarMap &tyVars ) {
27                if ( ! adaptee->get_returnVals().empty() && isPolyType( adaptee->get_returnVals().front()->get_type(), tyVars ) ) {
28                        return true;
29                } // if
30                for ( std::list< DeclarationWithType* >::const_iterator innerArg = adaptee->get_parameters().begin(); innerArg != adaptee->get_parameters().end(); ++innerArg ) {
31                        if ( isPolyType( (*innerArg)->get_type(), tyVars ) ) {
32                                return true;
33                        } // if
34                } // for
35                return false;
36        }
37
38        bool isPolyRet( FunctionType *function, std::string &name, const TyVarMap &otherTyVars ) {
39                bool doTransform = false;
40                if ( ! function->get_returnVals().empty() ) {
41                        if ( TypeInstType *typeInst = dynamic_cast< TypeInstType *>( function->get_returnVals().front()->get_type() ) ) {
42       
43                                // figure out if the return type is specified by a type parameter
44                                for ( std::list< TypeDecl *>::const_iterator tyVar = function->get_forall().begin(); tyVar != function->get_forall().end(); ++tyVar ) {
45                                        if ( (*tyVar)->get_name() == typeInst->get_name() ) {
46                                                doTransform = true;
47                                                name = typeInst->get_name();
48                                                break;
49                                        } // if
50                                } // for
51                                if ( ! doTransform && otherTyVars.find( typeInst->get_name() ) != otherTyVars.end() ) {
52                                        doTransform = true;
53                                } // if
54                        } // if
55                } // if
56                return doTransform;
57        }
58
59        bool isPolyRet( FunctionType *function, std::string &name ) {
60                TyVarMap dummyTyVars;
61                return isPolyRet( function, name, dummyTyVars );
62        }
63
64        bool isPolyRet( FunctionType *function, const TyVarMap &otherTyVars ) {
65                std::string dummyString;
66                return isPolyRet( function, dummyString, otherTyVars );
67        }
68
69        namespace {
70                /// Checks a parameter list for polymorphic parameters; will substitute according to env if present
71                bool hasPolyParams( std::list< Expression* >& params, const TypeSubstitution *env ) {
72                        for ( std::list< Expression* >::iterator param = params.begin(); param != params.end(); ++param ) {
73                                TypeExpr *paramType = dynamic_cast< TypeExpr* >( *param );
74                                assert(paramType && "Aggregate parameters should be type expressions");
75                                if ( isPolyType( paramType->get_type(), env ) ) return true;
76                        }
77                        return false;
78                }
79
80                /// Checks a parameter list for polymorphic parameters from tyVars; will substitute according to env if present
81                bool hasPolyParams( std::list< Expression* >& params, const TyVarMap &tyVars, const TypeSubstitution *env ) {
82                        for ( std::list< Expression* >::iterator param = params.begin(); param != params.end(); ++param ) {
83                                TypeExpr *paramType = dynamic_cast< TypeExpr* >( *param );
84                                assert(paramType && "Aggregate parameters should be type expressions");
85                                if ( isPolyType( paramType->get_type(), tyVars, env ) ) return true;
86                        }
87                        return false;
88                }
89        }
90
91        Type *isPolyType( Type *type, const TypeSubstitution *env ) {
92                if ( TypeInstType *typeInst = dynamic_cast< TypeInstType * >( type ) ) {
93                        if ( env ) {
94                                if ( Type *newType = env->lookup( typeInst->get_name() ) ) {
95                                        return isPolyType( newType, env );
96                                } // if
97                        } // if
98                        return type;
99                } else if ( StructInstType *structType = dynamic_cast< StructInstType* >( type ) ) {
100                        if ( hasPolyParams( structType->get_parameters(), env ) ) return type;
101                } else if ( UnionInstType *unionType = dynamic_cast< UnionInstType* >( type ) ) {
102                        if ( hasPolyParams( unionType->get_parameters(), env ) ) return type;
103                }
104                return 0;
105        }
106       
107        Type *isPolyType( Type *type, const TyVarMap &tyVars, const TypeSubstitution *env ) {
108                if ( TypeInstType *typeInst = dynamic_cast< TypeInstType * >( type ) ) {
109                        if ( env ) {
110                                if ( Type *newType = env->lookup( typeInst->get_name() ) ) {
111                                        return isPolyType( newType, tyVars, env );
112                                } // if
113                        } // if
114                        if ( tyVars.find( typeInst->get_name() ) != tyVars.end() ) {
115                                return type;
116                        }
117                } else if ( StructInstType *structType = dynamic_cast< StructInstType* >( type ) ) {
118                        if ( hasPolyParams( structType->get_parameters(), tyVars, env ) ) return type;
119                } else if ( UnionInstType *unionType = dynamic_cast< UnionInstType* >( type ) ) {
120                        if ( hasPolyParams( unionType->get_parameters(), tyVars, env ) ) return type;
121                }
122                return 0;
123        }
124
125        Type *isPolyPtr( Type *type, const TypeSubstitution *env ) {
126                if ( PointerType *ptr = dynamic_cast< PointerType *>( type ) ) {
127                        return isPolyType( ptr->get_base(), env );
128                } else if ( env ) {
129                        if ( TypeInstType *typeInst = dynamic_cast< TypeInstType *>( type ) ) {
130                                if ( Type *newType = env->lookup( typeInst->get_name() ) ) {
131                                        return isPolyPtr( newType, env );
132                                } // if
133                        } // if
134                } // if
135                return 0;
136        }
137       
138        Type *isPolyPtr( Type *type, const TyVarMap &tyVars, const TypeSubstitution *env ) {
139                if ( PointerType *ptr = dynamic_cast< PointerType *>( type ) ) {
140                        return isPolyType( ptr->get_base(), tyVars, env );
141                } else if ( env ) {
142                        if ( TypeInstType *typeInst = dynamic_cast< TypeInstType *>( type ) ) {
143                                if ( Type *newType = env->lookup( typeInst->get_name() ) ) {
144                                        return isPolyPtr( newType, tyVars, env );
145                                } // if
146                        } // if
147                } // if
148                return 0;
149        }
150
151        void printTyVarMap( std::ostream &os, const TyVarMap &tyVarMap ) {
152                for ( TyVarMap::const_iterator i = tyVarMap.begin(); i != tyVarMap.end(); ++i ) {
153                        os << i->first << " (" << i->second << ") ";
154                } // for
155                os << std::endl;
156        }
157
158        std::string sizeofName( Type *ty ) {
159                return std::string( "_sizeof_" ) + SymTab::Mangler::mangle( ty, false, false );
160        }
161
162        std::string alignofName( Type *ty ) {
163                return std::string( "_alignof_" ) + SymTab::Mangler::mangle( ty, false, false );
164        }
165} // namespace GenPoly
166
167// Local Variables: //
168// tab-width: 4 //
169// mode: c++ //
170// compile-command: "make install" //
171// End: //
Note: See TracBrowser for help on using the repository browser.