harec

commit 27e8b63812e8e31883d2ad9feccf854d76be7c89
parent 1c2d30ac9180186a1853609d31a036f9c770c875
Author: Drew DeVault <sir@cmpwn.com>
Date:   Thu, 24 Dec 2020 11:05:54 -0500

check: implement measurement expressions

Diffstat:
M
include/expr.h
 | 
1
+
M
src/check.c
 | 
32
+++++++++++++++++++++++++++++++-
M
src/gen.c
 | 
1
+

3 files changed, 33 insertions(+), 1 deletion(-)
diff --git a/include/expr.h b/include/expr.h
@@ -149,6 +149,7 @@ struct expression {
 		struct expression_binding binding;
 		union expression_constant constant;
 		struct expression_list list;
+		struct expression_measure measure;
 		struct expression_return _return;
 		struct expression_unarithm unarithm;
 	};
diff --git a/src/check.c b/src/check.c
@@ -263,6 +263,34 @@ check_expr_list(struct context *ctx,
 }
 
 static void
+check_expr_measure(struct context *ctx,
+	const struct ast_expression *aexpr,
+	struct expression *expr)
+{
+	trenter(TR_CHECK, "measure");
+	expr->type = EXPR_MEASURE;
+	expr->result = &builtin_type_size;
+	expr->measure.op = aexpr->measure.op;
+
+	switch (expr->measure.op) {
+	case M_LEN:
+		expr->measure.value = calloc(1, sizeof(struct expression));
+		check_expression(ctx, aexpr->measure.value, expr->measure.value);
+		enum type_storage vstor = expr->measure.value->result->storage;
+		expect(vstor == TYPE_STORAGE_ARRAY || vstor == TYPE_STORAGE_SLICE,
+			"len argument must be of an array or slice type");
+		// TODO: Check that array type is not unbounded
+		break;
+	case M_SIZE:
+		expr->measure.type = type_store_lookup_atype(
+			&ctx->store, aexpr->measure.type);
+		break;
+	case M_OFFSET:
+		assert(0); // TODO
+	}
+}
+
+static void
 check_expr_return(struct context *ctx,
 	const struct ast_expression *aexpr,
 	struct expression *expr)
@@ -368,8 +396,10 @@ check_expression(struct context *ctx,
 		check_expr_list(ctx, aexpr, expr);
 		break;
 	case EXPR_MATCH:
-	case EXPR_MEASURE:
 		assert(0); // TODO
+	case EXPR_MEASURE:
+		check_expr_measure(ctx, aexpr, expr);
+		break;
 	case EXPR_RETURN:
 		check_expr_return(ctx, aexpr, expr);
 		break;
diff --git a/src/gen.c b/src/gen.c
@@ -417,6 +417,7 @@ gen_expression(struct gen_context *ctx,
 		break;
 	case EXPR_MATCH:
 	case EXPR_MEASURE:
+		assert(0); // TODO
 	case EXPR_RETURN:
 		gen_expr_return(ctx, expr, out);
 		break;