chiark / gitweb /
setup.py: Update to use pkg-config.
[mLib-python] / assoc.pyx
CommitLineData
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.
d8d81d1b 18#
579d0169 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.
d8d81d1b 23#
579d0169 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
28cdef struct _assoc_entry:
29 sym_base _b
30 PyObject *v
31
32cdef 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
d8d81d1b 59 cdef void _setval(me, void *e, object val):
579d0169 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
75cdef 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 ----------------------------------------------------