Changeset e48aca8

Timestamp:

Mar 14, 2025, 10:42:12 AM (6 months ago)

Author:

Andrew Beach <ajbeach@…>

Branches:

master

Children:

a21aaff

Parents:

6f9f338

Message:

Added support for code generation of the CountofExpr. This doesn't do anything to final output but is useful for debugging.

Location:

src/CodeGen

Files:

• CodeGenerator.cpp (modified) (1 diff)
• CodeGenerator.hpp (modified) (2 diffs)

src/CodeGen/CodeGenerator.cpp

@@ -790 +790 @@
 }
 
+void CodeGenerator::postvisit( ast::CountofExpr const * expr ) {
+        assertf( !options.genC, "CountofExpr should not reach code generation." );
+        extension( expr );
+        output << "countof(";
+        if ( auto type = expr->type.as<ast::TypeofType>() ) {
+                type->expr->accept( *visitor );
+        } else {
+                output << genType( expr->type, "", options );
+        }
+        output << ")";
+}
+
 void CodeGenerator::postvisit( ast::UntypedOffsetofExpr const * expr ) {
         assertf( !options.genC, "UntypedOffsetofExpr should not reach code generation." );

src/CodeGen/CodeGenerator.hpp

@@ -42 +42 @@
         void previsit( ast::Expr const * );
 
-        void postvisit( ast::StructDecl const * );
         void postvisit( ast::FunctionDecl const * );
         // Yes, there is one visit that does modify the ast.
         ast::ObjectDecl const * postvisit( ast::ObjectDecl const * );
+        void postvisit( ast::StructDecl const * );
         void postvisit( ast::UnionDecl const * );
         void postvisit( ast::EnumDecl const * );
@@ -73 +73 @@
         void postvisit( ast::SizeofExpr const * );
         void postvisit( ast::AlignofExpr const * );
+        void postvisit( ast::CountofExpr const * );
         void postvisit( ast::UntypedOffsetofExpr const * );
         void postvisit( ast::OffsetofExpr const * );