Generate IR code for the sizeof expressions

Signed-off-by: Roberto Raggi <roberto.raggi@gmail.com>

Author: robertoraggi

src/mlir/cxx/mlir/codegen_expressions.cc

@@ -1135,25 +1135,33 @@ auto Codegen::ExpressionVisitor::operator()(AwaitExpressionAST* ast)
 
 auto Codegen::ExpressionVisitor::operator()(SizeofExpressionAST* ast)
     -> ExpressionResult {
+  if (auto size = ast->value) {
+    auto resultlType = gen.convertType(ast->type);
+    auto loc = gen.getLocation(ast->firstSourceLocation());
+    auto op = mlir::cxx::IntConstantOp::create(gen.builder_, loc, resultlType,
+                                               size.value());
+    return {op};
+  }
+
   auto op =
       gen.emitTodoExpr(ast->firstSourceLocation(), to_string(ast->kind()));
 
-#if false
-  auto expressionResult = gen.expression(ast->expression);
-#endif
-
   return {op};
 }
 
 auto Codegen::ExpressionVisitor::operator()(SizeofTypeExpressionAST* ast)
     -> ExpressionResult {
+  if (auto size = ast->value) {
+    auto resultlType = gen.convertType(ast->type);
+    auto loc = gen.getLocation(ast->firstSourceLocation());
+    auto op = mlir::cxx::IntConstantOp::create(gen.builder_, loc, resultlType,
+                                               size.value());
+    return {op};
+  }
+
   auto op =
       gen.emitTodoExpr(ast->firstSourceLocation(), to_string(ast->kind()));
 
-#if false
-  auto typeIdResult = gen.typeId(ast->typeId);
-#endif
-
   return {op};
 }
 

src/parser/cxx/ast.h

@@ -2300,6 +2300,7 @@ class SizeofExpressionAST final : public ExpressionAST {
 
   SourceLocation sizeofLoc;
   ExpressionAST* expression = nullptr;
+  std::optional<std::int64_t> value;
 
   void accept(ASTVisitor* visitor) override { visitor->visit(this); }
 
@@ -2317,6 +2318,7 @@ class SizeofTypeExpressionAST final : public ExpressionAST {
   SourceLocation lparenLoc;
   TypeIdAST* typeId = nullptr;
   SourceLocation rparenLoc;
+  std::optional<std::int64_t> value;
 
   void accept(ASTVisitor* visitor) override { visitor->visit(this); }
 

src/parser/cxx/type_checker.cc

@@ -24,6 +24,7 @@
 #include <cxx/ast.h>
 #include <cxx/control.h>
 #include <cxx/literals.h>
+#include <cxx/memory_layout.h>
 #include <cxx/name_lookup.h>
 #include <cxx/names.h>
 #include <cxx/symbols.h>
@@ -1040,10 +1041,18 @@ void TypeChecker::Visitor::operator()(AwaitExpressionAST* ast) {}
 
 void TypeChecker::Visitor::operator()(SizeofExpressionAST* ast) {
   ast->type = control()->getSizeType();
+
+  if (ast->expression) {
+    ast->value = control()->memoryLayout()->sizeOf(ast->expression->type);
+  }
 }
 
 void TypeChecker::Visitor::operator()(SizeofTypeExpressionAST* ast) {
   ast->type = control()->getSizeType();
+
+  if (ast->typeId) {
+    ast->value = control()->memoryLayout()->sizeOf(ast->typeId->type);
+  }
 }
 
 void TypeChecker::Visitor::operator()(SizeofPackExpressionAST* ast) {}