map.h 1.03 KB
Newer Older
1
2
3
4
typedef struct _mp_map_elem_t {
    mp_obj_t key;
    mp_obj_t value;
} mp_map_elem_t;
5

6
typedef struct _mp_map_t {
7
    struct {
8
9
        machine_uint_t all_keys_are_qstrs : 1;
        machine_uint_t used : (8 * sizeof(machine_uint_t) - 1);
10
11
    };
    machine_uint_t alloc;
12
13
14
15
16
17
18
19
    mp_map_elem_t *table;
} mp_map_t;

typedef struct _mp_set_t {
    machine_uint_t alloc;
    machine_uint_t used;
    mp_obj_t *table;
} mp_set_t;
20

21
typedef enum _mp_map_lookup_kind_t {
John R. Lenton's avatar
John R. Lenton committed
22
23
24
25
    MP_MAP_LOOKUP,                    // 0
    MP_MAP_LOOKUP_ADD_IF_NOT_FOUND,   // 1
    MP_MAP_LOOKUP_REMOVE_IF_FOUND,    // 2
    MP_MAP_LOOKUP_FIRST = 4,
26
27
} mp_map_lookup_kind_t;

28
int get_doubling_prime_greater_or_equal_to(int x);
29
30
31
void mp_map_init(mp_map_t *map, int n);
mp_map_t *mp_map_new(int n);
mp_map_elem_t* mp_map_lookup(mp_map_t *map, mp_obj_t index, mp_map_lookup_kind_t lookup_kind);
John R. Lenton's avatar
John R. Lenton committed
32
void mp_map_clear(mp_map_t *map);
33
34

void mp_set_init(mp_set_t *set, int n);
John R. Lenton's avatar
John R. Lenton committed
35
mp_obj_t mp_set_lookup(mp_set_t *set, mp_obj_t index, mp_map_lookup_kind_t lookup_kind);
John R. Lenton's avatar
John R. Lenton committed
36
void mp_set_clear(mp_set_t *set);