From d97dd3f04d32d73c64562ed5a4735951dce083d7 Mon Sep 17 00:00:00 2001 From: yuyi98 Date: Sun, 27 Mar 2022 22:58:13 +0800 Subject: [PATCH] cgen: fix crash for casting bool to int --- vlib/v/gen/c/cgen.v | 12 ++++++++++++ vlib/v/tests/cast_bool_to_int_test.v | 12 ++++++++++++ 2 files changed, 24 insertions(+) create mode 100644 vlib/v/tests/cast_bool_to_int_test.v diff --git a/vlib/v/gen/c/cgen.v b/vlib/v/gen/c/cgen.v index b10df0b4a8..adeb3e0b23 100644 --- a/vlib/v/gen/c/cgen.v +++ b/vlib/v/gen/c/cgen.v @@ -3771,6 +3771,18 @@ fn (mut g Gen) cast_expr(node ast.CastExpr) { g.write('))') } else if sym.kind == .alias && g.table.final_sym(node.typ).kind == .array_fixed { g.expr(node.expr) + } else if node.expr_type != 0 && g.table.final_sym(node.expr_type).kind == .bool + && node.typ.is_int() { + styp := g.typ(node.typ) + tmp := g.new_tmp_var() + g.empty_line = true + s := g.go_before_stmt(0) + g.write('$styp $tmp = (') + g.expr(node.expr) + g.writeln(')?1:0;') + g.empty_line = false + g.write(s) + g.write(tmp) } else { styp := g.typ(node.typ) if (g.pref.translated || g.file.is_translated) && sym.kind == .function { diff --git a/vlib/v/tests/cast_bool_to_int_test.v b/vlib/v/tests/cast_bool_to_int_test.v new file mode 100644 index 0000000000..2158366fdd --- /dev/null +++ b/vlib/v/tests/cast_bool_to_int_test.v @@ -0,0 +1,12 @@ +module main + +fn test_cast_bool_to_int() { + i := true + a := [1, 2, 3] + + println(a[int(!i)]) + assert a[int(!i)] == 1 + + println(a[int(i)]) + assert a[int(i)] == 2 +}