chiark / gitweb /
setup.py: Update to use pkg-config.
[mLib-python] / assoc.pyx
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 ----------------------------------------------------