579d0169 |
1 | # -*-pyrex-*- |
2 | # |
3 | # $Id$ |
4 | # |
5 | # Association tables |
6 | # |
7 | # (c) 2005 Straylight/Edgeware |
8 | # |
9 | |
10 | #----- Licensing notice ----------------------------------------------------- |
11 | # |
12 | # This file is part of the Python interface to mLib. |
13 | # |
14 | # mLib/Python is free software; you can redistribute it and/or modify |
15 | # it under the terms of the GNU General Public License as published by |
16 | # the Free Software Foundation; either version 2 of the License, or |
17 | # (at your option) any later version. |
18 | # |
19 | # mLib/Python is distributed in the hope that it will be useful, |
20 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
21 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
22 | # GNU General Public License for more details. |
23 | # |
24 | # You should have received a copy of the GNU General Public License |
25 | # along with mLib/Python; if not, write to the Free Software Foundation, |
26 | # Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. |
27 | |
28 | cdef struct _assoc_entry: |
29 | sym_base _b |
30 | PyObject *v |
31 | |
32 | cdef class AssocTable (Mapping): |
33 | cdef assoc_table _t |
34 | cdef int _init(me) except -1: |
35 | assoc_create(&me._t) |
36 | return 0 |
37 | cdef void *_find(me, object key, unsigned *f) except NULL: |
38 | cdef void *p |
39 | cdef int n |
40 | cdef _assoc_entry *e |
41 | cdef atom *a |
42 | a = ATOM_A(atom_pyintern(key)) |
43 | PyObject_AsReadBuffer(key, &p, &n) |
44 | if f: |
45 | f[0] = 0 |
46 | e = <_assoc_entry *>assoc_find(&me._t, a, PSIZEOF(e), f) |
47 | if not f[0]: |
48 | e.v = NULL |
49 | else: |
50 | e = <_assoc_entry *>assoc_find(&me._t, a, 0, NULL) |
51 | return <void *>e |
52 | cdef object _key(me, void *e): |
53 | return atom_pywrap(ASSOC_ATOM(e)) |
54 | cdef object _value(me, void *e): |
55 | cdef _assoc_entry *ee |
56 | ee = <_assoc_entry *>e |
57 | Py_INCREF(ee.v) |
58 | return <object>ee.v |
59 | cdef void _setval(me, void *e, object val): |
60 | cdef _assoc_entry *ee |
61 | ee = <_assoc_entry *>e |
62 | if ee.v: |
63 | Py_DECREF(ee.v) |
64 | ee.v = <PyObject *>v |
65 | Py_INCREF(ee.v) |
66 | cdef void _del(me, void *e): |
67 | cdef _assoc_entry *ee |
68 | ee = <_assoc_entry *>e |
69 | if ee.v: |
70 | Py_DECREF(ee.v) |
71 | assoc_remove(&me._t, <void *>ee) |
72 | cdef _MapIterator _iter(me): |
73 | return _AssocIter(me) |
74 | |
75 | cdef class _AssocIter (_MapIterator): |
76 | cdef AssocTable t |
77 | cdef assoc_iter i |
78 | def __new__(me, AssocTable t): |
79 | me.t = t |
80 | assoc_mkiter(&me.i, &me.t._t) |
81 | cdef void *_next(me): |
82 | return assoc_next(&me.i) |
83 | |
84 | #----- That's all, folks ---------------------------------------------------- |