objdict.c 2.32 KB
Newer Older
1
2
3
4
5
6
7
8
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <assert.h>

#include "nlr.h"
#include "misc.h"
#include "mpconfig.h"
9
#include "mpqstr.h"
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
#include "obj.h"
#include "runtime0.h"
#include "runtime.h"
#include "map.h"

typedef struct _mp_obj_dict_t {
    mp_obj_base_t base;
    mp_map_t map;
} mp_obj_dict_t;

void dict_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t self_in) {
    mp_obj_dict_t *self = self_in;
    bool first = true;
    print(env, "{");
    for (int i = 0; i < self->map.alloc; i++) {
        if (self->map.table[i].key != NULL) {
            if (!first) {
                print(env, ", ");
            }
            first = false;
            mp_obj_print_helper(print, env, self->map.table[i].key);
            print(env, ": ");
            mp_obj_print_helper(print, env, self->map.table[i].value);
        }
    }
    print(env, "}");
}

mp_obj_t dict_binary_op(int op, mp_obj_t lhs_in, mp_obj_t rhs_in) {
    mp_obj_dict_t *o = lhs_in;
    switch (op) {
        case RT_BINARY_OP_SUBSCR:
        {
            // dict load
            mp_map_elem_t *elem = mp_map_lookup_helper(&o->map, rhs_in, false);
            if (elem == NULL) {
46
                nlr_jump(mp_obj_new_exception_msg(MP_QSTR_KeyError, "<value>"));
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
            } else {
                return elem->value;
            }
        }
        default:
            // op not supported
            return NULL;
    }
}

const mp_obj_type_t dict_type = {
    { &mp_const_type },
    "dict",
    dict_print, // print
    NULL, // call_n
    NULL, // unary_op
    dict_binary_op, // binary_op
    NULL, // getiter
    NULL, // iternext
    {{NULL, NULL},}, // method list
};

mp_obj_t mp_obj_new_dict(int n_args) {
    mp_obj_dict_t *o = m_new_obj(mp_obj_dict_t);
    o->base.type = &dict_type;
    mp_map_init(&o->map, MP_MAP_OBJ, n_args);
    return o;
}
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

uint mp_obj_dict_len(mp_obj_t self_in) {
    mp_obj_dict_t *self = self_in;
    uint len = 0;
    for (int i = 0; i < self->map.alloc; i++) {
        if (self->map.table[i].key != NULL) {
            len += 1;
        }
    }
    return len;
}

mp_obj_t mp_obj_dict_store(mp_obj_t self_in, mp_obj_t key, mp_obj_t value) {
    assert(MP_OBJ_IS_TYPE(self_in, &dict_type));
    mp_obj_dict_t *self = self_in;
    mp_map_lookup_helper(&self->map, key, true)->value = value;
    return self_in;
}