Changeset bb336a6 for src/Validate/ImplementEnumFunc.cpp

Timestamp:

Jul 10, 2024, 3:39:26 AM (4 months ago)

Author:

JiadaL <j82liang@…>

Branches:

master

Children:

dbff8ec

Parents:

550afde2

Message:

Fixed the problem when enum use another enumerator as initializer

File:

• src/Validate/ImplementEnumFunc.cpp (modified) (5 diffs)

src/Validate/ImplementEnumFunc.cpp

@@ -104 +104 @@
                 auto memAsObjectDecl = mem.as<ast::ObjectDecl>();
                 assert(memAsObjectDecl);
-                if (memAsObjectDecl->init) {
+                if (auto& init = memAsObjectDecl->init) {
+                        auto singleInit = init.strict_as<ast::SingleInit>();
+                        auto nameExpr = singleInit->value.as<ast::NameExpr>();
+                        if (nameExpr) {
+                                auto name = nameExpr->name;
+                                if (auto it = std::find_if(decl->members.begin(), decl->members.end(), 
+                                        [name](ast::ptr<ast::Decl> mem_decl) {
+                                                return (mem_decl->name == name);
+                                        }); it != std::end(decl->members)
+                                ) {
+                                        // ast::SingleInit* newInit = new ast::SingleInit(
+                                        //      init->location, new ast::CastExpr( nameExpr->location, 
+                                        //      nameExpr, new ast::BasicType( ast::BasicKind::UnsignedInt ), ast::ExplicitCast )
+                                        // );
+                                        // auto targetInit = (*it).strict_as<ast::ObjectDecl>()->init;
+                                        // inits.emplace_back( targetInit );
+                                        auto index = std::distance( decl->members.begin(), it );
+                                        auto targetInit = inits.at(index).strict_as<ast::SingleInit>();
+                                        auto targetExpr = targetInit->value;
+                                        inits.push_back( new ast::SingleInit( targetExpr->location, targetExpr ) );
+                                        continue;
+                                }
+                        }
+
                         inits.emplace_back(memAsObjectDecl->init);
                 } else {
@@ -406 +429 @@
 void EnumAttrFuncGenerator::genTypedEnumFunction(const ast::EnumAttribute attr) {
         if (attr == ast::EnumAttribute::Value) {
-                if (decl->base) {
+                if (decl->isTyped()) {
                         // TypedEnum's backing arrays
                         std::vector<ast::ptr<ast::Init>> inits = genValueInit();
@@ -441 +464 @@
 
 void EnumAttrFuncGenerator::genTypedEnumFuncs() {
-        genTypedEnumFunction(ast::EnumAttribute::Value);
-        genTypedEnumFunction(ast::EnumAttribute::Label);
-        genTypedEnumFunction(ast::EnumAttribute::Posn);
+        // genTypedEnumFunction(ast::EnumAttribute::Value);
+        // genTypedEnumFunction(ast::EnumAttribute::Label);
+        // genTypedEnumFunction(ast::EnumAttribute::Posn);
 }
 
@@ -456 +479 @@
         std::list<ast::ptr<ast::Decl>>& decls) {
         // Generate the functions (they go into forwards and definitions).
-        genTypeNameFunc();
+        // genTypeNameFunc();
         genTypedEnumFuncs();
-        genSerialTraitFuncs();
-        genBoundedFunctions();
+        // genSerialTraitFuncs();
+        // genBoundedFunctions();
         // Now export the lists contents.
         decls.splice(decls.end(), forwards);
@@ -483 +506 @@
         enumInst.base = enumDecl;
         EnumAttrFuncGenerator gen(enumDecl, &enumInst, functionNesting);
-        gen.generateAndAppendFunctions(declsToAddAfter);
+        // gen.generateAndAppendFunctions(declsToAddAfter);
 }