map.c 5.23 KB
Newer Older
1
2
3
4
5
#include <stdint.h>
#include <stdlib.h>
#include <assert.h>

#include "misc.h"
6
#include "mpconfig.h"
7
#include "obj.h"
8
9
#include "runtime0.h"
#include "map.h"
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

// approximatelly doubling primes; made with Mathematica command: Table[Prime[Floor[(1.7)^n]], {n, 3, 24}]
static int doubling_primes[] = {7, 19, 43, 89, 179, 347, 647, 1229, 2297, 4243, 7829, 14347, 26017, 47149, 84947, 152443, 273253, 488399, 869927, 1547173, 2745121, 4861607};

int get_doubling_prime_greater_or_equal_to(int x) {
    for (int i = 0; i < sizeof(doubling_primes) / sizeof(int); i++) {
        if (doubling_primes[i] >= x) {
            return doubling_primes[i];
        }
    }
    // ran out of primes in the table!
    // return something sensible, at least make it odd
    return x | 1;
}

25
26
27
28
/******************************************************************************/
/* map                                                                        */

void mp_map_init(mp_map_t *map, mp_map_kind_t kind, int n) {
29
30
31
    map->kind = kind;
    map->used = 0;
    map->alloc = get_doubling_prime_greater_or_equal_to(n + 1);
32
    map->table = m_new0(mp_map_elem_t, map->alloc);
33
34
}

35
36
37
mp_map_t *mp_map_new(mp_map_kind_t kind, int n) {
    mp_map_t *map = m_new(mp_map_t, 1);
    mp_map_init(map, kind, n);
38
39
40
    return map;
}

41
42
mp_map_elem_t* mp_map_lookup_helper(mp_map_t *map, mp_obj_t index, bool add_if_not_found) {
    bool is_map_mp_obj = (map->kind == MP_MAP_OBJ);
43
    machine_uint_t hash;
44
45
    if (is_map_mp_obj) {
        hash = mp_obj_hash(index);
46
47
48
49
50
    } else {
        hash = (machine_uint_t)index;
    }
    uint pos = hash % map->alloc;
    for (;;) {
51
        mp_map_elem_t *elem = &map->table[pos];
52
53
54
55
56
57
        if (elem->key == NULL) {
            // not in table
            if (add_if_not_found) {
                if (map->used + 1 >= map->alloc) {
                    // not enough room in table, rehash it
                    int old_alloc = map->alloc;
58
                    mp_map_elem_t *old_table = map->table;
59
60
                    map->alloc = get_doubling_prime_greater_or_equal_to(map->alloc + 1);
                    map->used = 0;
61
                    map->table = m_new0(mp_map_elem_t, map->alloc);
62
63
                    for (int i = 0; i < old_alloc; i++) {
                        if (old_table[i].key != NULL) {
64
                            mp_map_lookup_helper(map, old_table[i].key, true)->value = old_table[i].value;
65
66
67
68
69
70
71
72
73
74
75
76
77
                        }
                    }
                    m_free(old_table);
                    // restart the search for the new element
                    pos = hash % map->alloc;
                } else {
                    map->used += 1;
                    elem->key = index;
                    return elem;
                }
            } else {
                return NULL;
            }
78
        } else if (elem->key == index || (is_map_mp_obj && mp_obj_equal(elem->key, index))) {
79
80
81
82
83
84
85
86
87
88
89
90
91
92
            // found it
            /* it seems CPython does not replace the index; try x={True:'true'};x[1]='one';x
            if (add_if_not_found) {
                elem->key = index;
            }
            */
            return elem;
        } else {
            // not yet found, keep searching in this table
            pos = (pos + 1) % map->alloc;
        }
    }
}

93
94
95
mp_map_elem_t* mp_qstr_map_lookup(mp_map_t *map, qstr index, bool add_if_not_found) {
    mp_obj_t o = (mp_obj_t)(machine_uint_t)index;
    return mp_map_lookup_helper(map, o, add_if_not_found);
96
97
}

98
99
100
101
102
103
104
/******************************************************************************/
/* set                                                                        */

void mp_set_init(mp_set_t *set, int n) {
    set->alloc = get_doubling_prime_greater_or_equal_to(n + 1);
    set->used = 0;
    set->table = m_new0(mp_obj_t, set->alloc);
105
106
}

107
108
109
mp_obj_t mp_set_lookup(mp_set_t *set, mp_obj_t index, bool add_if_not_found) {
    int hash = mp_obj_hash(index);
    int pos = hash % set->alloc;
110
    for (;;) {
111
112
        mp_obj_t elem = set->table[pos];
        if (elem == MP_OBJ_NULL) {
113
114
            // not in table
            if (add_if_not_found) {
115
                if (set->used + 1 >= set->alloc) {
116
                    // not enough room in table, rehash it
117
118
119
120
121
                    int old_alloc = set->alloc;
                    mp_obj_t *old_table = set->table;
                    set->alloc = get_doubling_prime_greater_or_equal_to(set->alloc + 1);
                    set->used = 0;
                    set->table = m_new(mp_obj_t, set->alloc);
122
123
                    for (int i = 0; i < old_alloc; i++) {
                        if (old_table[i] != NULL) {
124
                            mp_set_lookup(set, old_table[i], true);
125
126
127
128
                        }
                    }
                    m_free(old_table);
                    // restart the search for the new element
129
                    pos = hash % set->alloc;
130
                } else {
131
132
                    set->used += 1;
                    set->table[pos] = index;
133
134
135
                    return index;
                }
            } else {
136
                return MP_OBJ_NULL;
137
            }
138
        } else if (mp_obj_equal(elem, index)) {
139
140
141
142
            // found it
            return elem;
        } else {
            // not yet found, keep searching in this table
143
            pos = (pos + 1) % set->alloc;
144
145
146
        }
    }
}