Changeset 525f7ad for src/ResolvExpr/CandidateFinder.cpp

Timestamp:

Jun 19, 2024, 3:20:39 PM (4 months ago)

Author:

JiadaL <j82liang@…>

Branches:

master

Children:

57e43cd

Parents:

1725989

Message:

1. Add count_e( enum_name ), a pseudo function that return the number of element in an enum; 2. Implementation of enum range loop.

File:

• src/ResolvExpr/CandidateFinder.cpp (modified) (2 diffs)

src/ResolvExpr/CandidateFinder.cpp

@@ -696 +696 @@
                 void postvisit( const ast::UntypedInitExpr * initExpr );
                 void postvisit( const ast::QualifiedNameExpr * qualifiedExpr );
+                void postvisit( const ast::CountExpr * countExpr );
 
                 const ast::Expr * makeEnumOffsetCast( const ast::EnumInstType * src, 
@@ -1527 +1528 @@
                         addCandidate( *choice, new ast::SizeofExpr{ sizeofExpr->location, choice->expr } );
                 }
+        }
+
+        void Finder::postvisit( const ast::CountExpr * countExpr ) {
+                assert( countExpr->type );
+                auto enumInst = countExpr->type.as<ast::EnumInstType>();
+                if ( !enumInst ) {
+                        SemanticError( countExpr, "Count Expression only supports Enum Type as operand: ");
+                }
+                addCandidate( ast::ConstantExpr::from_ulong(countExpr->location, enumInst->base->members.size()), tenv );
         }