source: src/GenPoly/GenPoly.cc @ 69911c11

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

Restore Mangler to closer to previous behaviour

  • Property mode set to 100644
File size: 6.4 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        FunctionType * getFunctionType( Type *ty ) {
152                PointerType *ptrType;
153                if ( ( ptrType = dynamic_cast< PointerType* >( ty ) ) ) {
154                        return dynamic_cast< FunctionType* >( ptrType->get_base() ); // pointer if FunctionType, NULL otherwise
155                } else {
156                        return dynamic_cast< FunctionType* >( ty ); // pointer if FunctionType, NULL otherwise
157                }
158        }
159
160        void printTyVarMap( std::ostream &os, const TyVarMap &tyVarMap ) {
161                for ( TyVarMap::const_iterator i = tyVarMap.begin(); i != tyVarMap.end(); ++i ) {
162                        os << i->first << " (" << i->second << ") ";
163                } // for
164                os << std::endl;
165        }
166
167        std::string sizeofName( Type *ty ) {
168                return std::string( "_sizeof_" ) + SymTab::Mangler::mangleType( ty );
169        }
170
171        std::string alignofName( Type *ty ) {
172                return std::string( "_alignof_" ) + SymTab::Mangler::mangleType( ty );
173        }
174} // namespace GenPoly
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.