| 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 | // Autogen.cc --
 | 
|---|
| 8 | //
 | 
|---|
| 9 | // Author           : Rob Schluntz
 | 
|---|
| 10 | // Created On       : Thu Mar 03 15:45:56 2016
 | 
|---|
| 11 | // Last Modified By : Peter A. Buhr
 | 
|---|
| 12 | // Last Modified On : Tue Jul 12 17:47:17 2016
 | 
|---|
| 13 | // Update Count     : 2
 | 
|---|
| 14 | //
 | 
|---|
| 15 | 
 | 
|---|
| 16 | #include <list>
 | 
|---|
| 17 | #include <iterator>
 | 
|---|
| 18 | #include "SynTree/Visitor.h"
 | 
|---|
| 19 | #include "SynTree/Type.h"
 | 
|---|
| 20 | #include "SynTree/Statement.h"
 | 
|---|
| 21 | #include "SynTree/TypeSubstitution.h"
 | 
|---|
| 22 | #include "Common/utility.h"
 | 
|---|
| 23 | #include "AddVisit.h"
 | 
|---|
| 24 | #include "MakeLibCfa.h"
 | 
|---|
| 25 | #include "Autogen.h"
 | 
|---|
| 26 | 
 | 
|---|
| 27 | namespace SymTab {
 | 
|---|
| 28 |         Type * SizeType = 0;
 | 
|---|
| 29 | 
 | 
|---|
| 30 |         class AutogenerateRoutines : public Visitor {
 | 
|---|
| 31 |                 public:
 | 
|---|
| 32 |                 std::list< Declaration * > &get_declsToAdd() { return declsToAdd; }
 | 
|---|
| 33 | 
 | 
|---|
| 34 |                 virtual void visit( EnumDecl *enumDecl );
 | 
|---|
| 35 |                 virtual void visit( StructDecl *structDecl );
 | 
|---|
| 36 |                 virtual void visit( UnionDecl *structDecl );
 | 
|---|
| 37 |                 virtual void visit( TypeDecl *typeDecl );
 | 
|---|
| 38 |                 virtual void visit( TraitDecl *ctxDecl );
 | 
|---|
| 39 |                 virtual void visit( FunctionDecl *functionDecl );
 | 
|---|
| 40 | 
 | 
|---|
| 41 |                 virtual void visit( FunctionType *ftype );
 | 
|---|
| 42 |                 virtual void visit( PointerType *ftype );
 | 
|---|
| 43 | 
 | 
|---|
| 44 |                 virtual void visit( CompoundStmt *compoundStmt );
 | 
|---|
| 45 |                 virtual void visit( SwitchStmt *switchStmt );
 | 
|---|
| 46 | 
 | 
|---|
| 47 |                 AutogenerateRoutines() : functionNesting( 0 ) {}
 | 
|---|
| 48 |                 private:
 | 
|---|
| 49 |                 template< typename StmtClass > void visitStatement( StmtClass *stmt );
 | 
|---|
| 50 | 
 | 
|---|
| 51 |                 std::list< Declaration * > declsToAdd;
 | 
|---|
| 52 |                 std::set< std::string > structsDone;
 | 
|---|
| 53 |                 unsigned int functionNesting;     // current level of nested functions
 | 
|---|
| 54 |         };
 | 
|---|
| 55 | 
 | 
|---|
| 56 |         void autogenerateRoutines( std::list< Declaration * > &translationUnit ) {
 | 
|---|
| 57 |                 AutogenerateRoutines visitor;
 | 
|---|
| 58 |                 acceptAndAdd( translationUnit, visitor, false );
 | 
|---|
| 59 |         }
 | 
|---|
| 60 | 
 | 
|---|
| 61 |         bool isUnnamedBitfield( ObjectDecl * obj ) {
 | 
|---|
| 62 |                 return obj != NULL && obj->get_name() == "" && obj->get_bitfieldWidth() != NULL;
 | 
|---|
| 63 |         }
 | 
|---|
| 64 | 
 | 
|---|
| 65 |         template< typename OutputIterator >
 | 
|---|
| 66 |         void makeUnionFieldsAssignment( ObjectDecl *srcParam, ObjectDecl *dstParam, UnionInstType *unionType, OutputIterator out ) {
 | 
|---|
| 67 |                 UntypedExpr *copy = new UntypedExpr( new NameExpr( "__builtin_memcpy" ) );
 | 
|---|
| 68 |                 copy->get_args().push_back( new VariableExpr( dstParam ) );
 | 
|---|
| 69 |                 copy->get_args().push_back( new AddressExpr( new VariableExpr( srcParam ) ) );
 | 
|---|
| 70 |                 copy->get_args().push_back( new SizeofExpr( unionType ) );
 | 
|---|
| 71 | 
 | 
|---|
| 72 |                 *out++ = new ExprStmt( noLabels, copy );
 | 
|---|
| 73 |         }
 | 
|---|
| 74 | 
 | 
|---|
| 75 |         //E ?=?(E volatile*, int),
 | 
|---|
| 76 |         //  ?=?(E _Atomic volatile*, int);
 | 
|---|
| 77 |         void makeEnumFunctions( EnumDecl *enumDecl, EnumInstType *refType, unsigned int functionNesting, std::list< Declaration * > &declsToAdd ) {
 | 
|---|
| 78 |                 FunctionType *assignType = new FunctionType( Type::Qualifiers(), false );
 | 
|---|
| 79 | 
 | 
|---|
| 80 |                 ObjectDecl *dstParam = new ObjectDecl( "_dst", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, new PointerType( Type::Qualifiers(), refType ), 0 );
 | 
|---|
| 81 |                 assignType->get_parameters().push_back( dstParam );
 | 
|---|
| 82 | 
 | 
|---|
| 83 |                 // void ?{}(E *); void ^?{}(E *);
 | 
|---|
| 84 |                 FunctionType * ctorType = assignType->clone();
 | 
|---|
| 85 |                 FunctionType * dtorType = assignType->clone();
 | 
|---|
| 86 | 
 | 
|---|
| 87 |                 ObjectDecl *srcParam = new ObjectDecl( "_src", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, refType->clone(), 0 );
 | 
|---|
| 88 |                 assignType->get_parameters().push_back( srcParam );
 | 
|---|
| 89 |                 // void ?{}(E *, E);
 | 
|---|
| 90 |                 FunctionType *copyCtorType = assignType->clone();
 | 
|---|
| 91 | 
 | 
|---|
| 92 |                 // T ?=?(E *, E);
 | 
|---|
| 93 |                 ObjectDecl *returnVal = new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, refType->clone(), 0 );
 | 
|---|
| 94 |                 assignType->get_returnVals().push_back( returnVal );
 | 
|---|
| 95 | 
 | 
|---|
| 96 |                 // xxx - should we also generate void ?{}(E *, int) and E ?{}(E *, E)?
 | 
|---|
| 97 |                 // right now these cases work, but that might change.
 | 
|---|
| 98 | 
 | 
|---|
| 99 |                 // Routines at global scope marked "static" to prevent multiple definitions is separate translation units
 | 
|---|
| 100 |                 // because each unit generates copies of the default routines for each aggregate.
 | 
|---|
| 101 |                 // xxx - Temporary: make these functions intrinsic so they codegen as C assignment.
 | 
|---|
| 102 |                 // Really they're something of a cross between instrinsic and autogen, so should
 | 
|---|
| 103 |                 // probably make a new linkage type
 | 
|---|
| 104 |                 FunctionDecl *assignDecl = new FunctionDecl( "?=?", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::Intrinsic, assignType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 105 |                 FunctionDecl *ctorDecl = new FunctionDecl( "?{}", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::Intrinsic, ctorType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 106 |                 FunctionDecl *copyCtorDecl = new FunctionDecl( "?{}", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::Intrinsic, copyCtorType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 107 |                 FunctionDecl *dtorDecl = new FunctionDecl( "^?{}", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::Intrinsic, dtorType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 108 |                 assignDecl->fixUniqueId();
 | 
|---|
| 109 |                 ctorDecl->fixUniqueId();
 | 
|---|
| 110 |                 copyCtorDecl->fixUniqueId();
 | 
|---|
| 111 |                 dtorDecl->fixUniqueId();
 | 
|---|
| 112 | 
 | 
|---|
| 113 |                 // enum copy construct and assignment is just C-style assignment.
 | 
|---|
| 114 |                 // this looks like a bad recursive call, but code gen will turn it into
 | 
|---|
| 115 |                 // a C-style assignment.
 | 
|---|
| 116 |                 // This happens before function pointer type conversion, so need to do it manually here
 | 
|---|
| 117 |                 VariableExpr * assignVarExpr = new VariableExpr( assignDecl );
 | 
|---|
| 118 |                 Type *& assignVarExprType = assignVarExpr->get_results().front();
 | 
|---|
| 119 |                 assignVarExprType = new PointerType( Type::Qualifiers(), assignVarExprType );
 | 
|---|
| 120 |                 ApplicationExpr * assignExpr = new ApplicationExpr( assignVarExpr );
 | 
|---|
| 121 |                 assignExpr->get_args().push_back( new VariableExpr( dstParam ) );
 | 
|---|
| 122 |                 assignExpr->get_args().push_back( new VariableExpr( srcParam ) );
 | 
|---|
| 123 | 
 | 
|---|
| 124 |                 // body is either return stmt or expr stmt
 | 
|---|
| 125 |                 assignDecl->get_statements()->get_kids().push_back( new ReturnStmt( noLabels, assignExpr ) );
 | 
|---|
| 126 |                 copyCtorDecl->get_statements()->get_kids().push_back( new ExprStmt( noLabels, assignExpr->clone() ) );
 | 
|---|
| 127 | 
 | 
|---|
| 128 |                 declsToAdd.push_back( assignDecl );
 | 
|---|
| 129 |                 declsToAdd.push_back( ctorDecl );
 | 
|---|
| 130 |                 declsToAdd.push_back( copyCtorDecl );
 | 
|---|
| 131 |                 declsToAdd.push_back( dtorDecl );
 | 
|---|
| 132 |         }
 | 
|---|
| 133 | 
 | 
|---|
| 134 |         /// Clones a reference type, replacing any parameters it may have with a clone of the provided list
 | 
|---|
| 135 |         template< typename GenericInstType >
 | 
|---|
| 136 |         GenericInstType *cloneWithParams( GenericInstType *refType, const std::list< Expression* >& params ) {
 | 
|---|
| 137 |                 GenericInstType *clone = refType->clone();
 | 
|---|
| 138 |                 clone->get_parameters().clear();
 | 
|---|
| 139 |                 cloneAll( params, clone->get_parameters() );
 | 
|---|
| 140 |                 return clone;
 | 
|---|
| 141 |         }
 | 
|---|
| 142 | 
 | 
|---|
| 143 |         /// Creates a new type decl that's the same as src, but renamed and with only the ?=?, ?{} (default and copy), and ^?{} assertions (for complete types only)
 | 
|---|
| 144 |         TypeDecl *cloneAndRename( TypeDecl *src, const std::string &name ) {
 | 
|---|
| 145 |                 // TypeDecl *dst = new TypeDecl( name, src->get_storageClass(), 0, src->get_kind() );
 | 
|---|
| 146 | 
 | 
|---|
| 147 |                 // if ( src->get_kind() == TypeDecl::Any ) {
 | 
|---|
| 148 |                 //      TypeInstType *opParamType = new TypeInstType( Type::Qualifiers(), name, dst );
 | 
|---|
| 149 |                 //      FunctionType *opFunctionType = new FunctionType( Type::Qualifiers(), false );
 | 
|---|
| 150 |                 //      opFunctionType->get_parameters().push_back(
 | 
|---|
| 151 |                 //              new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, new PointerType( Type::Qualifiers(), opParamType->clone() ), 0 ) );
 | 
|---|
| 152 |                 //      FunctionDecl *ctorAssert = new FunctionDecl( "?{}", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, opFunctionType->clone(), 0, false, false );
 | 
|---|
| 153 |                 //      FunctionDecl *dtorAssert = new FunctionDecl( "^?{}", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, opFunctionType->clone(), 0, false, false );
 | 
|---|
| 154 | 
 | 
|---|
| 155 |                 //      opFunctionType->get_parameters().push_back(
 | 
|---|
| 156 |                 //              new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, opParamType, 0 ) );
 | 
|---|
| 157 |                 //      FunctionDecl *copyCtorAssert = new FunctionDecl( "?{}", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, opFunctionType->clone(), 0, false, false );
 | 
|---|
| 158 | 
 | 
|---|
| 159 |                 //      opFunctionType->get_returnVals().push_back(
 | 
|---|
| 160 |                 //              new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, opParamType->clone(), 0 ) );
 | 
|---|
| 161 |                 //      FunctionDecl *assignAssert = new FunctionDecl( "?=?", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, opFunctionType, 0, false, false );
 | 
|---|
| 162 | 
 | 
|---|
| 163 | 
 | 
|---|
| 164 |                 //      dst->get_assertions().push_back( assignAssert );
 | 
|---|
| 165 |                 //      dst->get_assertions().push_back( ctorAssert );
 | 
|---|
| 166 |                 //      dst->get_assertions().push_back( dtorAssert );
 | 
|---|
| 167 |                 //      dst->get_assertions().push_back( copyCtorAssert );
 | 
|---|
| 168 |                 // }
 | 
|---|
| 169 | 
 | 
|---|
| 170 |                 TypeDecl *dst = new TypeDecl( src->get_name(), src->get_storageClass(), 0, src->get_kind() );
 | 
|---|
| 171 |                 cloneAll(src->get_assertions(), dst->get_assertions());
 | 
|---|
| 172 |                 return dst;
 | 
|---|
| 173 |         }
 | 
|---|
| 174 | 
 | 
|---|
| 175 |         void makeStructMemberOp( ObjectDecl * dstParam, Expression * src, DeclarationWithType * field, FunctionDecl * func, TypeSubstitution & genericSubs, bool isDynamicLayout, bool forward = true ) {
 | 
|---|
| 176 |                 if ( isDynamicLayout && src ) {
 | 
|---|
| 177 |                         genericSubs.apply( src );
 | 
|---|
| 178 |                 }
 | 
|---|
| 179 | 
 | 
|---|
| 180 |                 ObjectDecl * returnVal = NULL;
 | 
|---|
| 181 |                 if ( ! func->get_functionType()->get_returnVals().empty() ) {
 | 
|---|
| 182 |                         returnVal = dynamic_cast<ObjectDecl*>( func->get_functionType()->get_returnVals().front() );
 | 
|---|
| 183 |                 }
 | 
|---|
| 184 | 
 | 
|---|
| 185 |                 InitTweak::InitExpander srcParam( src );
 | 
|---|
| 186 | 
 | 
|---|
| 187 |                 // assign to destination (and return value if generic)
 | 
|---|
| 188 |                 UntypedExpr *derefExpr = new UntypedExpr( new NameExpr( "*?" ) );
 | 
|---|
| 189 |                 derefExpr->get_args().push_back( new VariableExpr( dstParam ) );
 | 
|---|
| 190 |                 Expression *dstselect = new MemberExpr( field, derefExpr );
 | 
|---|
| 191 |                 genImplicitCall( srcParam, dstselect, func->get_name(), back_inserter( func->get_statements()->get_kids() ), field, forward );
 | 
|---|
| 192 | 
 | 
|---|
| 193 |                 if ( isDynamicLayout && returnVal ) {
 | 
|---|
| 194 |                         UntypedExpr *derefRet = new UntypedExpr( new NameExpr( "*?" ) );
 | 
|---|
| 195 |                         derefRet->get_args().push_back( new VariableExpr( returnVal ) );
 | 
|---|
| 196 |                         Expression *retselect = new MemberExpr( field, derefRet );
 | 
|---|
| 197 |                         genImplicitCall( srcParam, retselect, func->get_name(), back_inserter( func->get_statements()->get_kids() ), field, forward );
 | 
|---|
| 198 |                 } // if
 | 
|---|
| 199 |         }
 | 
|---|
| 200 | 
 | 
|---|
| 201 |         template<typename Iterator>
 | 
|---|
| 202 |         void makeStructFunctionBody( Iterator member, Iterator end, FunctionDecl * func, TypeSubstitution & genericSubs, bool isDynamicLayout, bool forward = true ) {
 | 
|---|
| 203 |                 for ( ; member != end; ++member ) {
 | 
|---|
| 204 |                         if ( DeclarationWithType *field = dynamic_cast< DeclarationWithType * >( *member ) ) { // otherwise some form of type declaration, e.g. Aggregate
 | 
|---|
| 205 |                                 // query the type qualifiers of this field and skip assigning it if it is marked const.
 | 
|---|
| 206 |                                 // If it is an array type, we need to strip off the array layers to find its qualifiers.
 | 
|---|
| 207 |                                 Type * type = field->get_type();
 | 
|---|
| 208 |                                 while ( ArrayType * at = dynamic_cast< ArrayType * >( type ) ) {
 | 
|---|
| 209 |                                         type = at->get_base();
 | 
|---|
| 210 |                                 }
 | 
|---|
| 211 | 
 | 
|---|
| 212 |                                 if ( type->get_qualifiers().isConst && func->get_name() == "?=?" ) {
 | 
|---|
| 213 |                                         // don't assign const members, but do construct/destruct
 | 
|---|
| 214 |                                         continue;
 | 
|---|
| 215 |                                 }
 | 
|---|
| 216 | 
 | 
|---|
| 217 |                                 if ( field->get_name() == "" ) {
 | 
|---|
| 218 |                                         // don't assign to anonymous members
 | 
|---|
| 219 |                                         // xxx - this is a temporary fix. Anonymous members tie into
 | 
|---|
| 220 |                                         // our inheritance model. I think the correct way to handle this is to
 | 
|---|
| 221 |                                         // cast the structure to the type of the member and let the resolver
 | 
|---|
| 222 |                                         // figure out whether it's valid and have a pass afterwards that fixes
 | 
|---|
| 223 |                                         // the assignment to use pointer arithmetic with the offset of the
 | 
|---|
| 224 |                                         // member, much like how generic type members are handled.
 | 
|---|
| 225 |                                         continue;
 | 
|---|
| 226 |                                 }
 | 
|---|
| 227 | 
 | 
|---|
| 228 |                                 assert( ! func->get_functionType()->get_parameters().empty() );
 | 
|---|
| 229 |                                 ObjectDecl * dstParam = dynamic_cast<ObjectDecl*>( func->get_functionType()->get_parameters().front() );
 | 
|---|
| 230 |                                 ObjectDecl * srcParam = NULL;
 | 
|---|
| 231 |                                 if ( func->get_functionType()->get_parameters().size() == 2 ) {
 | 
|---|
| 232 |                                         srcParam = dynamic_cast<ObjectDecl*>( func->get_functionType()->get_parameters().back() );
 | 
|---|
| 233 |                                 }
 | 
|---|
| 234 |                                 // srcParam may be NULL, in which case we have default ctor/dtor
 | 
|---|
| 235 |                                 assert( dstParam );
 | 
|---|
| 236 | 
 | 
|---|
| 237 |                                 Expression *srcselect = srcParam ? new MemberExpr( field, new VariableExpr( srcParam ) ) : NULL;
 | 
|---|
| 238 |                                 makeStructMemberOp( dstParam, srcselect, field, func, genericSubs, isDynamicLayout, forward );
 | 
|---|
| 239 |                         } // if
 | 
|---|
| 240 |                 } // for
 | 
|---|
| 241 |         } // makeStructFunctionBody
 | 
|---|
| 242 | 
 | 
|---|
| 243 |         /// generate the body of a constructor which takes parameters that match fields, e.g.
 | 
|---|
| 244 |         /// void ?{}(A *, int) and void?{}(A *, int, int) for a struct A which has two int fields.
 | 
|---|
| 245 |         template<typename Iterator>
 | 
|---|
| 246 |         void makeStructFieldCtorBody( Iterator member, Iterator end, FunctionDecl * func, TypeSubstitution & genericSubs, bool isDynamicLayout ) {
 | 
|---|
| 247 |                 FunctionType * ftype = func->get_functionType();
 | 
|---|
| 248 |                 std::list<DeclarationWithType*> & params = ftype->get_parameters();
 | 
|---|
| 249 |                 assert( params.size() >= 2 );  // should not call this function for default ctor, etc.
 | 
|---|
| 250 | 
 | 
|---|
| 251 |                 // skip 'this' parameter
 | 
|---|
| 252 |                 ObjectDecl * dstParam = dynamic_cast<ObjectDecl*>( params.front() );
 | 
|---|
| 253 |                 assert( dstParam );
 | 
|---|
| 254 |                 std::list<DeclarationWithType*>::iterator parameter = params.begin()+1;
 | 
|---|
| 255 |                 for ( ; member != end; ++member ) {
 | 
|---|
| 256 |                         if ( DeclarationWithType * field = dynamic_cast<DeclarationWithType*>( *member ) ) {
 | 
|---|
| 257 |                                 if ( isUnnamedBitfield( dynamic_cast< ObjectDecl * > ( field ) ) ) {
 | 
|---|
| 258 |                                         // don't make a function whose parameter is an unnamed bitfield
 | 
|---|
| 259 |                                         continue;
 | 
|---|
| 260 |                                 } else if ( field->get_name() == "" ) {
 | 
|---|
| 261 |                                         // don't assign to anonymous members
 | 
|---|
| 262 |                                         // xxx - this is a temporary fix. Anonymous members tie into
 | 
|---|
| 263 |                                         // our inheritance model. I think the correct way to handle this is to
 | 
|---|
| 264 |                                         // cast the structure to the type of the member and let the resolver
 | 
|---|
| 265 |                                         // figure out whether it's valid and have a pass afterwards that fixes
 | 
|---|
| 266 |                                         // the assignment to use pointer arithmetic with the offset of the
 | 
|---|
| 267 |                                         // member, much like how generic type members are handled.
 | 
|---|
| 268 |                                         continue;
 | 
|---|
| 269 |                                 } else if ( parameter != params.end() ) {
 | 
|---|
| 270 |                                         // matching parameter, initialize field with copy ctor
 | 
|---|
| 271 |                                         Expression *srcselect = new VariableExpr(*parameter);
 | 
|---|
| 272 |                                         makeStructMemberOp( dstParam, srcselect, field, func, genericSubs, isDynamicLayout );
 | 
|---|
| 273 |                                         ++parameter;
 | 
|---|
| 274 |                                 } else {
 | 
|---|
| 275 |                                         // no matching parameter, initialize field with default ctor
 | 
|---|
| 276 |                                         makeStructMemberOp( dstParam, NULL, field, func, genericSubs, isDynamicLayout );
 | 
|---|
| 277 |                                 }
 | 
|---|
| 278 |                         }
 | 
|---|
| 279 |                 }
 | 
|---|
| 280 |         }
 | 
|---|
| 281 | 
 | 
|---|
| 282 |         void makeStructFunctions( StructDecl *aggregateDecl, StructInstType *refType, unsigned int functionNesting, std::list< Declaration * > & declsToAdd ) {
 | 
|---|
| 283 |                 FunctionType *assignType = new FunctionType( Type::Qualifiers(), false );
 | 
|---|
| 284 | 
 | 
|---|
| 285 |                 // Make function polymorphic in same parameters as generic struct, if applicable
 | 
|---|
| 286 |                 bool isDynamicLayout = false;  // NOTE this flag is an incredibly ugly kludge; we should fix the assignment signature instead (ditto for union)
 | 
|---|
| 287 |                 std::list< TypeDecl* >& genericParams = aggregateDecl->get_parameters();
 | 
|---|
| 288 |                 std::list< Expression* > structParams;  // List of matching parameters to put on types
 | 
|---|
| 289 |                 TypeSubstitution genericSubs; // Substitutions to make to member types of struct
 | 
|---|
| 290 |                 for ( std::list< TypeDecl* >::const_iterator param = genericParams.begin(); param != genericParams.end(); ++param ) {
 | 
|---|
| 291 |                         if ( (*param)->get_kind() == TypeDecl::Any ) isDynamicLayout = true;
 | 
|---|
| 292 |                         TypeDecl *typeParam = cloneAndRename( *param, "_autoassign_" + aggregateDecl->get_name() + "_" + (*param)->get_name() );
 | 
|---|
| 293 |                         assignType->get_forall().push_back( typeParam );
 | 
|---|
| 294 |                         TypeInstType *newParamType = new TypeInstType( Type::Qualifiers(), typeParam->get_name(), typeParam );
 | 
|---|
| 295 |                         genericSubs.add( (*param)->get_name(), newParamType );
 | 
|---|
| 296 |                         structParams.push_back( new TypeExpr( newParamType ) );
 | 
|---|
| 297 |                 }
 | 
|---|
| 298 | 
 | 
|---|
| 299 |                 ObjectDecl *dstParam = new ObjectDecl( "_dst", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, new PointerType( Type::Qualifiers(), cloneWithParams( refType, structParams ) ), 0 );
 | 
|---|
| 300 |                 assignType->get_parameters().push_back( dstParam );
 | 
|---|
| 301 | 
 | 
|---|
| 302 |                 // void ?{}(T *); void ^?{}(T *);
 | 
|---|
| 303 |                 FunctionType *ctorType = assignType->clone();
 | 
|---|
| 304 |                 FunctionType *dtorType = assignType->clone();
 | 
|---|
| 305 | 
 | 
|---|
| 306 |                 ObjectDecl *srcParam = new ObjectDecl( "_src", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, cloneWithParams( refType, structParams ), 0 );
 | 
|---|
| 307 |                 assignType->get_parameters().push_back( srcParam );
 | 
|---|
| 308 | 
 | 
|---|
| 309 |                 // void ?{}(T *, T);
 | 
|---|
| 310 |                 FunctionType *copyCtorType = assignType->clone();
 | 
|---|
| 311 | 
 | 
|---|
| 312 |                 // T ?=?(T *, T);
 | 
|---|
| 313 |                 ObjectDecl *returnVal = new ObjectDecl( "_ret", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, cloneWithParams( refType, structParams ), 0 );
 | 
|---|
| 314 |                 assignType->get_returnVals().push_back( returnVal );
 | 
|---|
| 315 | 
 | 
|---|
| 316 |                 // Routines at global scope marked "static" to prevent multiple definitions is separate translation units
 | 
|---|
| 317 |                 // because each unit generates copies of the default routines for each aggregate.
 | 
|---|
| 318 |                 FunctionDecl *assignDecl = new FunctionDecl( "?=?", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, assignType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 319 |                 FunctionDecl *ctorDecl = new FunctionDecl( "?{}", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, ctorType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 320 |                 FunctionDecl *copyCtorDecl = new FunctionDecl( "?{}", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, copyCtorType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 321 |                 FunctionDecl *dtorDecl = new FunctionDecl( "^?{}", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, dtorType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 322 |                 assignDecl->fixUniqueId();
 | 
|---|
| 323 |                 ctorDecl->fixUniqueId();
 | 
|---|
| 324 |                 copyCtorDecl->fixUniqueId();
 | 
|---|
| 325 |                 dtorDecl->fixUniqueId();
 | 
|---|
| 326 | 
 | 
|---|
| 327 |                 // create constructors which take each member type as a parameter.
 | 
|---|
| 328 |                 // for example, for struct A { int x, y; }; generate
 | 
|---|
| 329 |                 // void ?{}(A *, int) and void ?{}(A *, int, int)
 | 
|---|
| 330 |                 std::list<Declaration *> memCtors;
 | 
|---|
| 331 |                 FunctionType * memCtorType = ctorType->clone();
 | 
|---|
| 332 |                 for ( std::list<Declaration *>::iterator i = aggregateDecl->get_members().begin(); i != aggregateDecl->get_members().end(); ++i ) {
 | 
|---|
| 333 |                         DeclarationWithType * member = dynamic_cast<DeclarationWithType *>( *i );
 | 
|---|
| 334 |                         assert( member );
 | 
|---|
| 335 |                         if ( isUnnamedBitfield( dynamic_cast< ObjectDecl * > ( member ) ) ) {
 | 
|---|
| 336 |                                 // don't make a function whose parameter is an unnamed bitfield
 | 
|---|
| 337 |                                 continue;
 | 
|---|
| 338 |                         } else if ( member->get_name() == "" ) {
 | 
|---|
| 339 |                                 // don't assign to anonymous members
 | 
|---|
| 340 |                                 // xxx - this is a temporary fix. Anonymous members tie into
 | 
|---|
| 341 |                                 // our inheritance model. I think the correct way to handle this is to
 | 
|---|
| 342 |                                 // cast the structure to the type of the member and let the resolver
 | 
|---|
| 343 |                                 // figure out whether it's valid and have a pass afterwards that fixes
 | 
|---|
| 344 |                                 // the assignment to use pointer arithmetic with the offset of the
 | 
|---|
| 345 |                                 // member, much like how generic type members are handled.
 | 
|---|
| 346 |                                 continue;
 | 
|---|
| 347 |                         }
 | 
|---|
| 348 |                         memCtorType->get_parameters().push_back( new ObjectDecl( member->get_name(), DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, member->get_type()->clone(), 0 ) );
 | 
|---|
| 349 |                         FunctionDecl * ctor = new FunctionDecl( "?{}", functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, memCtorType->clone(), new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 350 |                         ctor->fixUniqueId();
 | 
|---|
| 351 |                         makeStructFieldCtorBody( aggregateDecl->get_members().begin(), aggregateDecl->get_members().end(), ctor, genericSubs, isDynamicLayout );
 | 
|---|
| 352 |                         memCtors.push_back( ctor );
 | 
|---|
| 353 |                 }
 | 
|---|
| 354 |                 delete memCtorType;
 | 
|---|
| 355 | 
 | 
|---|
| 356 |                 // generate appropriate calls to member ctor, assignment
 | 
|---|
| 357 |                 makeStructFunctionBody( aggregateDecl->get_members().begin(), aggregateDecl->get_members().end(), assignDecl, genericSubs, isDynamicLayout );
 | 
|---|
| 358 |                 makeStructFunctionBody( aggregateDecl->get_members().begin(), aggregateDecl->get_members().end(), ctorDecl, genericSubs, isDynamicLayout );
 | 
|---|
| 359 |                 makeStructFunctionBody( aggregateDecl->get_members().begin(), aggregateDecl->get_members().end(), copyCtorDecl, genericSubs, isDynamicLayout );
 | 
|---|
| 360 |                 // needs to do everything in reverse, so pass "forward" as false
 | 
|---|
| 361 |                 makeStructFunctionBody( aggregateDecl->get_members().rbegin(), aggregateDecl->get_members().rend(), dtorDecl, genericSubs, isDynamicLayout, false );
 | 
|---|
| 362 | 
 | 
|---|
| 363 |                 if ( ! isDynamicLayout ) assignDecl->get_statements()->get_kids().push_back( new ReturnStmt( noLabels, new VariableExpr( srcParam ) ) );
 | 
|---|
| 364 | 
 | 
|---|
| 365 |                 declsToAdd.push_back( assignDecl );
 | 
|---|
| 366 |                 declsToAdd.push_back( ctorDecl );
 | 
|---|
| 367 |                 declsToAdd.push_back( copyCtorDecl );
 | 
|---|
| 368 |                 declsToAdd.push_back( dtorDecl );
 | 
|---|
| 369 |                 declsToAdd.splice( declsToAdd.end(), memCtors );
 | 
|---|
| 370 |         }
 | 
|---|
| 371 | 
 | 
|---|
| 372 |         void makeUnionFunctions( UnionDecl *aggregateDecl, UnionInstType *refType, unsigned int functionNesting, std::list< Declaration * > & declsToAdd ) {
 | 
|---|
| 373 |                 FunctionType *assignType = new FunctionType( Type::Qualifiers(), false );
 | 
|---|
| 374 | 
 | 
|---|
| 375 |                 // Make function polymorphic in same parameters as generic union, if applicable
 | 
|---|
| 376 |                 bool isDynamicLayout = false; // NOTE this flag is an incredibly ugly kludge; we should fix the assignment signature instead (ditto for struct)
 | 
|---|
| 377 |                 std::list< TypeDecl* >& genericParams = aggregateDecl->get_parameters();
 | 
|---|
| 378 |                 std::list< Expression* > unionParams;  // List of matching parameters to put on types
 | 
|---|
| 379 |                 for ( std::list< TypeDecl* >::const_iterator param = genericParams.begin(); param != genericParams.end(); ++param ) {
 | 
|---|
| 380 |                         if ( (*param)->get_kind() == TypeDecl::Any ) isDynamicLayout = true;
 | 
|---|
| 381 |                         TypeDecl *typeParam = cloneAndRename( *param, "_autoassign_" + aggregateDecl->get_name() + "_" + (*param)->get_name() );
 | 
|---|
| 382 |                         assignType->get_forall().push_back( typeParam );
 | 
|---|
| 383 |                         unionParams.push_back( new TypeExpr( new TypeInstType( Type::Qualifiers(), typeParam->get_name(), typeParam ) ) );
 | 
|---|
| 384 |                 }
 | 
|---|
| 385 | 
 | 
|---|
| 386 |                 ObjectDecl *dstParam = new ObjectDecl( "_dst", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, new PointerType( Type::Qualifiers(), cloneWithParams( refType, unionParams ) ), 0 );
 | 
|---|
| 387 |                 assignType->get_parameters().push_back( dstParam );
 | 
|---|
| 388 | 
 | 
|---|
| 389 |                 // default ctor/dtor need only first parameter
 | 
|---|
| 390 |                 FunctionType * ctorType = assignType->clone();
 | 
|---|
| 391 |                 FunctionType * dtorType = assignType->clone();
 | 
|---|
| 392 | 
 | 
|---|
| 393 |                 ObjectDecl *srcParam = new ObjectDecl( "_src", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, cloneWithParams( refType, unionParams ), 0 );
 | 
|---|
| 394 |                 assignType->get_parameters().push_back( srcParam );
 | 
|---|
| 395 | 
 | 
|---|
| 396 |                 // copy ctor needs both parameters
 | 
|---|
| 397 |                 FunctionType * copyCtorType = assignType->clone();
 | 
|---|
| 398 | 
 | 
|---|
| 399 |                 // assignment needs both and return value
 | 
|---|
| 400 |                 ObjectDecl *returnVal = new ObjectDecl( "_ret", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, cloneWithParams( refType, unionParams ), 0 );
 | 
|---|
| 401 |                 assignType->get_returnVals().push_back( returnVal );
 | 
|---|
| 402 | 
 | 
|---|
| 403 |                 // Routines at global scope marked "static" to prevent multiple definitions is separate translation units
 | 
|---|
| 404 |                 // because each unit generates copies of the default routines for each aggregate.
 | 
|---|
| 405 |                 FunctionDecl *assignDecl = new FunctionDecl( "?=?",  functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, assignType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 406 |                 FunctionDecl *ctorDecl = new FunctionDecl( "?{}",  functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, ctorType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 407 |                 FunctionDecl *copyCtorDecl = new FunctionDecl( "?{}",  functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, copyCtorType, NULL, true, false );
 | 
|---|
| 408 |                 FunctionDecl *dtorDecl = new FunctionDecl( "^?{}",  functionNesting > 0 ? DeclarationNode::NoStorageClass : DeclarationNode::Static, LinkageSpec::AutoGen, dtorType, new CompoundStmt( noLabels ), true, false );
 | 
|---|
| 409 | 
 | 
|---|
| 410 |                 assignDecl->fixUniqueId();
 | 
|---|
| 411 |                 ctorDecl->fixUniqueId();
 | 
|---|
| 412 |                 copyCtorDecl->fixUniqueId();
 | 
|---|
| 413 |                 dtorDecl->fixUniqueId();
 | 
|---|
| 414 | 
 | 
|---|
| 415 |                 makeUnionFieldsAssignment( srcParam, dstParam, cloneWithParams( refType, unionParams ), back_inserter( assignDecl->get_statements()->get_kids() ) );
 | 
|---|
| 416 |                 if ( isDynamicLayout ) makeUnionFieldsAssignment( srcParam, returnVal, cloneWithParams( refType, unionParams ), back_inserter( assignDecl->get_statements()->get_kids() ) );
 | 
|---|
| 417 |                 else assignDecl->get_statements()->get_kids().push_back( new ReturnStmt( noLabels, new VariableExpr( srcParam ) ) );
 | 
|---|
| 418 | 
 | 
|---|
| 419 |                 // body of assignment and copy ctor is the same
 | 
|---|
| 420 |                 copyCtorDecl->set_statements( assignDecl->get_statements()->clone() );
 | 
|---|
| 421 | 
 | 
|---|
| 422 |                 declsToAdd.push_back( assignDecl );
 | 
|---|
| 423 |                 declsToAdd.push_back( ctorDecl );
 | 
|---|
| 424 |                 declsToAdd.push_back( copyCtorDecl );
 | 
|---|
| 425 |                 declsToAdd.push_back( dtorDecl );
 | 
|---|
| 426 |         }
 | 
|---|
| 427 | 
 | 
|---|
| 428 |         void AutogenerateRoutines::visit( EnumDecl *enumDecl ) {
 | 
|---|
| 429 |                 if ( ! enumDecl->get_members().empty() ) {
 | 
|---|
| 430 |                         EnumInstType *enumInst = new EnumInstType( Type::Qualifiers(), enumDecl->get_name() );
 | 
|---|
| 431 |                         // enumInst->set_baseEnum( enumDecl );
 | 
|---|
| 432 |                         // declsToAdd.push_back(
 | 
|---|
| 433 |                         makeEnumFunctions( enumDecl, enumInst, functionNesting, declsToAdd );
 | 
|---|
| 434 |                 }
 | 
|---|
| 435 |         }
 | 
|---|
| 436 | 
 | 
|---|
| 437 |         void AutogenerateRoutines::visit( StructDecl *structDecl ) {
 | 
|---|
| 438 |                 if ( ! structDecl->get_members().empty() && structsDone.find( structDecl->get_name() ) == structsDone.end() ) {
 | 
|---|
| 439 |                         StructInstType structInst( Type::Qualifiers(), structDecl->get_name() );
 | 
|---|
| 440 |                         structInst.set_baseStruct( structDecl );
 | 
|---|
| 441 |                         makeStructFunctions( structDecl, &structInst, functionNesting, declsToAdd );
 | 
|---|
| 442 |                         structsDone.insert( structDecl->get_name() );
 | 
|---|
| 443 |                 } // if
 | 
|---|
| 444 |         }
 | 
|---|
| 445 | 
 | 
|---|
| 446 |         void AutogenerateRoutines::visit( UnionDecl *unionDecl ) {
 | 
|---|
| 447 |                 if ( ! unionDecl->get_members().empty() ) {
 | 
|---|
| 448 |                         UnionInstType unionInst( Type::Qualifiers(), unionDecl->get_name() );
 | 
|---|
| 449 |                         unionInst.set_baseUnion( unionDecl );
 | 
|---|
| 450 |                         makeUnionFunctions( unionDecl, &unionInst, functionNesting, declsToAdd );
 | 
|---|
| 451 |                 } // if
 | 
|---|
| 452 |         }
 | 
|---|
| 453 | 
 | 
|---|
| 454 |         void AutogenerateRoutines::visit( TypeDecl *typeDecl ) {
 | 
|---|
| 455 |                 CompoundStmt *stmts = 0;
 | 
|---|
| 456 |                 TypeInstType *typeInst = new TypeInstType( Type::Qualifiers(), typeDecl->get_name(), false );
 | 
|---|
| 457 |                 typeInst->set_baseType( typeDecl );
 | 
|---|
| 458 |                 ObjectDecl *src = new ObjectDecl( "_src", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, typeInst->clone(), 0 );
 | 
|---|
| 459 |                 ObjectDecl *dst = new ObjectDecl( "_dst", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, new PointerType( Type::Qualifiers(), typeInst->clone() ), 0 );
 | 
|---|
| 460 |                 if ( typeDecl->get_base() ) {
 | 
|---|
| 461 |                         // xxx - generate ctor/dtors for typedecls, e.g.
 | 
|---|
| 462 |                         // otype T = int *;
 | 
|---|
| 463 |                         stmts = new CompoundStmt( std::list< Label >() );
 | 
|---|
| 464 |                         UntypedExpr *assign = new UntypedExpr( new NameExpr( "?=?" ) );
 | 
|---|
| 465 |                         assign->get_args().push_back( new CastExpr( new VariableExpr( dst ), new PointerType( Type::Qualifiers(), typeDecl->get_base()->clone() ) ) );
 | 
|---|
| 466 |                         assign->get_args().push_back( new CastExpr( new VariableExpr( src ), typeDecl->get_base()->clone() ) );
 | 
|---|
| 467 |                         stmts->get_kids().push_back( new ReturnStmt( std::list< Label >(), assign ) );
 | 
|---|
| 468 |                 } // if
 | 
|---|
| 469 |                 FunctionType *type = new FunctionType( Type::Qualifiers(), false );
 | 
|---|
| 470 |                 type->get_returnVals().push_back( new ObjectDecl( "", DeclarationNode::NoStorageClass, LinkageSpec::Cforall, 0, typeInst, 0 ) );
 | 
|---|
| 471 |                 type->get_parameters().push_back( dst );
 | 
|---|
| 472 |                 type->get_parameters().push_back( src );
 | 
|---|
| 473 |                 FunctionDecl *func = new FunctionDecl( "?=?", DeclarationNode::NoStorageClass, LinkageSpec::AutoGen, type, stmts, true, false );
 | 
|---|
| 474 |                 declsToAdd.push_back( func );
 | 
|---|
| 475 |         }
 | 
|---|
| 476 | 
 | 
|---|
| 477 |         void addDecls( std::list< Declaration * > &declsToAdd, std::list< Statement * > &statements, std::list< Statement * >::iterator i ) {
 | 
|---|
| 478 |                 for ( std::list< Declaration * >::iterator decl = declsToAdd.begin(); decl != declsToAdd.end(); ++decl ) {
 | 
|---|
| 479 |                         statements.insert( i, new DeclStmt( noLabels, *decl ) );
 | 
|---|
| 480 |                 } // for
 | 
|---|
| 481 |                 declsToAdd.clear();
 | 
|---|
| 482 |         }
 | 
|---|
| 483 | 
 | 
|---|
| 484 |         void AutogenerateRoutines::visit( FunctionType *) {
 | 
|---|
| 485 |                 // ensure that we don't add assignment ops for types defined as part of the function
 | 
|---|
| 486 |         }
 | 
|---|
| 487 | 
 | 
|---|
| 488 |         void AutogenerateRoutines::visit( PointerType *) {
 | 
|---|
| 489 |                 // ensure that we don't add assignment ops for types defined as part of the pointer
 | 
|---|
| 490 |         }
 | 
|---|
| 491 | 
 | 
|---|
| 492 |         void AutogenerateRoutines::visit( TraitDecl *) {
 | 
|---|
| 493 |                 // ensure that we don't add assignment ops for types defined as part of the trait
 | 
|---|
| 494 |         }
 | 
|---|
| 495 | 
 | 
|---|
| 496 |         template< typename StmtClass >
 | 
|---|
| 497 |         inline void AutogenerateRoutines::visitStatement( StmtClass *stmt ) {
 | 
|---|
| 498 |                 std::set< std::string > oldStructs = structsDone;
 | 
|---|
| 499 |                 addVisit( stmt, *this );
 | 
|---|
| 500 |                 structsDone = oldStructs;
 | 
|---|
| 501 |         }
 | 
|---|
| 502 | 
 | 
|---|
| 503 |         void AutogenerateRoutines::visit( FunctionDecl *functionDecl ) {
 | 
|---|
| 504 |                 maybeAccept( functionDecl->get_functionType(), *this );
 | 
|---|
| 505 |                 acceptAll( functionDecl->get_oldDecls(), *this );
 | 
|---|
| 506 |                 functionNesting += 1;
 | 
|---|
| 507 |                 maybeAccept( functionDecl->get_statements(), *this );
 | 
|---|
| 508 |                 functionNesting -= 1;
 | 
|---|
| 509 |         }
 | 
|---|
| 510 | 
 | 
|---|
| 511 |         void AutogenerateRoutines::visit( CompoundStmt *compoundStmt ) {
 | 
|---|
| 512 |                 visitStatement( compoundStmt );
 | 
|---|
| 513 |         }
 | 
|---|
| 514 | 
 | 
|---|
| 515 |         void AutogenerateRoutines::visit( SwitchStmt *switchStmt ) {
 | 
|---|
| 516 |                 visitStatement( switchStmt );
 | 
|---|
| 517 |         }
 | 
|---|
| 518 | } // SymTab
 | 
|---|