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