mirrors/Odin
1
0
Fork 0
mirror of https://github.com/odin-lang/Odin.git synced 2026-04-18 20:40:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

Constant evaluation for in expression for bit_sets

Browse Source
This commit is contained in:
gingerBill
2018-08-16 00:07:26 +01:00
parent 042550cf87
commit 85ac95f81b
2 changed files with 19 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/demo/demo.odin
View File

@@ -727,7 +727,7 @@ bit_set_type :: proc() {
WEEKEND :: Days{Sunday, Saturday};
d: Days;
d = Days{Sunday} | Days{Monday};
d = {Sunday, Monday};
x := Tuesday;
e := d | WEEKEND;
e |= {Monday};
@@ -738,6 +738,8 @@ bit_set_type :: proc() {
if Saturday in e {
fmt.println("Saturday in", e);
}
X :: Saturday in WEEKEND; // Constant evaluation
fmt.println(X);
}
main :: proc() {

16
src/check_expr.cpp
View File

@@ -2041,6 +2041,22 @@ void check_binary_expr(CheckerContext *c, Operand *x, Ast *node, bool use_lhs_as
} else if (is_type_bit_set(y->type)) {
Type *yt = base_type(y->type);
check_assignment(c, x, yt->BitSet.base, str_lit("bit_set 'in'"));
if (x->mode == Addressing_Constant && y->mode == Addressing_Constant) {
ExactValue k = exact_value_to_integer(x->value);
ExactValue v = exact_value_to_integer(y->value);
GB_ASSERT(k.kind == ExactValue_Integer);
GB_ASSERT(v.kind == ExactValue_Integer);
i64 bit = 1ll<<big_int_to_i64(&k.value_integer);
i64 bits = big_int_to_i64(&v.value_integer);
x->mode = Addressing_Constant;
x->type = t_untyped_bool;
x->value = exact_value_bool((bit & bits) != 0);
x->expr = node;
return;
}
} else {
gbString t = type_to_string(y->type);
error(x->expr, "expected either a map or bitset for 'in', got %s", t);