c3c/lib/std/collections/enummap.c3

module std::collections::enummap<Enum, ValueType>;

struct EnumMap
{
    ValueType[Enum.len] values;
}

fn void EnumMap.init(EnumMap* this, ValueType init_value)
{
    foreach(&a : this.values)
    {
        *a = init_value;
    }
}


fn void! EnumMap.to_format(EnumMap* map, Formatter* formatter) @dynamic
{
	formatter.print("{ ")!;
	foreach (i, &value : map.values)
	{
		if (i != 0) formatter.print(", ")!;
		formatter.printf("%s: %s", (Enum)i, *value)!;
	}
	formatter.print(" }")!;
}

fn String EnumMap.to_string(EnumMap* map, Allocator* using = mem::heap()) @dynamic
{
	return string::printf("%s", *map);
}

fn uint EnumMap.len(EnumMap* this) @operator(len) => this.values.len;
fn ValueType EnumMap.get(EnumMap* this, Enum key) @operator([]) => this.values[key.ordinal];
fn void EnumMap.set(EnumMap* this, Enum key, ValueType value) @operator([]=) => this.values[key.ordinal] = value;