Changeset 1e30df7 for src/Validate/LinkReferenceToTypes.cpp

Timestamp:

Oct 18, 2022, 5:48:13 PM (21 months ago)

Author:

JiadaL <j82liang@…>

Branches:

ADT, ast-experimental, master

Children:

5408b59

Parents:

a55472cc

Message:

Supports inline enums

File:

• src/Validate/LinkReferenceToTypes.cpp (modified) (1 diff)

src/Validate/LinkReferenceToTypes.cpp

@@ -203 +203 @@
         }
 
+        // The following section 
+        auto mut = ast::mutate( decl );
+        std::vector<ast::ptr<ast::Decl>> buffer;
+        for ( auto it = decl->members.begin(); it != decl->members.end(); ++it) {
+                auto member = (*it).as<ast::ObjectDecl>();
+                if ( member->enumInLine ) {
+                        auto targetEnum = symtab.lookupEnum( member->name );
+                        if (targetEnum) {                       
+                                for (auto singleMamber : targetEnum->members) {
+                                        auto tm = singleMamber.as<ast::ObjectDecl>();
+                                        auto t = new ast::ObjectDecl(
+                                                member->location, // use the "inline" location
+                                                tm->name,
+                                                new ast::EnumInstType( decl, ast::CV::Const ),
+                                                // Construct a new EnumInstType as the type
+                                                tm->init,
+                                                tm->storage,
+                                                tm->linkage,
+                                                tm->bitfieldWidth,
+                                                {}, // enum member doesn't have attribute
+                                                tm->funcSpec
+                                        );
+                                        buffer.push_back(t);
+                                }
+                        }
+                } else {
+                        buffer.push_back( *it );
+                }
+        }
+        mut->members = buffer;
+        decl = mut;
+
         ForwardEnumsType::iterator fwds = forwardEnums.find( decl->name );
         if ( fwds != forwardEnums.end() ) {