summaryrefslogtreecommitdiffhomepage
path: root/examples/advanced.c
blob: 33ceebbeac9f519b080d9d6701ef15c8a5f7570a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#include <stdio.h>
#include <stc/cmap.h>
#include <stc/cstr.h>

typedef struct Viking {
    cstr name;
    cstr country;
} Viking;

void viking_del(Viking* vk) {
    cstr_del(&vk->name);
    cstr_del(&vk->country);
}

// Define Viking raw struct with hash, equals, and convertion functions between Viking and VikingRaw structs:

typedef struct VikingRaw {
    const char* name;
    const char* country;
} VikingRaw;

uint64_t vikingraw_hash(const VikingRaw* raw, size_t ignore) {
    uint64_t hash = c_strhash(raw->name) ^ (c_strhash(raw->country) >> 15);
    return hash;
}
static inline int vikingraw_equals(const VikingRaw* rx, const VikingRaw* ry) {
    return strcmp(rx->name, ry->name) == 0 && strcmp(rx->country, ry->country) == 0;
}

static inline Viking viking_fromRaw(VikingRaw raw) { // note: parameter is by value
    Viking vk = {cstr_from(raw.name), cstr_from(raw.country)}; return vk;
}
static inline VikingRaw viking_toRaw(Viking* vk) {
    VikingRaw raw = {vk->name.str, vk->country.str}; return raw;
}

// With this in place, we use the using_cmap_keydef() macro to define {Viking -> int} hash map type:

using_cmap_keydef(vk, Viking, int, vikingraw_equals, vikingraw_hash,
                      viking_del, viking_fromRaw, viking_toRaw, VikingRaw);

int main()
{
    cmap_vk vikings = cmap_vk_init();
    c_emplace_items(&vikings, cmap_vk, {
        { {"Einar", "Norway"}, 20},
        { {"Olaf", "Denmark"}, 24},
        { {"Harald", "Iceland"}, 12},
    });
    VikingRaw bjorn = {"Bjorn", "Sweden"};
    cmap_vk_emplace_or_assign(&vikings, bjorn, 10);

    VikingRaw einar = {"Einar", "Norway"};
    cmap_vk_value_t *e = cmap_vk_find(&vikings, einar).ref;
    e->second += 3; // add 3 hp points to Einar
    cmap_vk_emplace(&vikings, einar, 0).ref->second += 5; // add 5 more to Einar

    c_foreach (k, cmap_vk, vikings) {
        printf("%s of %s has %d hp\n", k.ref->first.name.str, k.ref->first.country.str, k.ref->second);
    }
    cmap_vk_del(&vikings);
}