X-Git-Url: https://git.distorted.org.uk/~mdw/mLib-python/blobdiff_plain/20bce5e92b01cd928f26b61be78215117039c561..579d01693c86259110fe7a2c2a6f005f1bdbad5b:/assoc.pyx

diff --git a/assoc.pyx b/assoc.pyx
new file mode 100644
index 0000000..26369de
--- /dev/null
+++ b/assoc.pyx
@@ -0,0 +1,84 @@
+# -*-pyrex-*-
+#
+# $Id$
+#
+# Association tables
+#
+# (c) 2005 Straylight/Edgeware
+#
+
+#----- Licensing notice -----------------------------------------------------
+#
+# This file is part of the Python interface to mLib.
+#
+# mLib/Python is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+# 
+# mLib/Python is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+# 
+# You should have received a copy of the GNU General Public License
+# along with mLib/Python; if not, write to the Free Software Foundation,
+# Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+cdef struct _assoc_entry:
+  sym_base _b
+  PyObject *v
+
+cdef class AssocTable (Mapping):
+  cdef assoc_table _t
+  cdef int _init(me) except -1:
+    assoc_create(&me._t)
+    return 0
+  cdef void *_find(me, object key, unsigned *f) except NULL:
+    cdef void *p
+    cdef int n
+    cdef _assoc_entry *e
+    cdef atom *a
+    a = ATOM_A(atom_pyintern(key))
+    PyObject_AsReadBuffer(key, &p, &n)
+    if f:
+      f[0] = 0
+      e = <_assoc_entry *>assoc_find(&me._t, a, PSIZEOF(e), f)
+      if not f[0]:
+        e.v = NULL
+    else:
+      e = <_assoc_entry *>assoc_find(&me._t, a, 0, NULL)
+    return <void *>e
+  cdef object _key(me, void *e):
+    return atom_pywrap(ASSOC_ATOM(e))
+  cdef object _value(me, void *e):
+    cdef _assoc_entry *ee
+    ee = <_assoc_entry *>e
+    Py_INCREF(ee.v)
+    return <object>ee.v
+  cdef void _setval(me, void *e, object val): 
+    cdef _assoc_entry *ee
+    ee = <_assoc_entry *>e
+    if ee.v:
+      Py_DECREF(ee.v)
+    ee.v = <PyObject *>v
+    Py_INCREF(ee.v)
+  cdef void _del(me, void *e):
+    cdef _assoc_entry *ee
+    ee = <_assoc_entry *>e
+    if ee.v:
+      Py_DECREF(ee.v)
+    assoc_remove(&me._t, <void *>ee)
+  cdef _MapIterator _iter(me):
+    return _AssocIter(me)
+
+cdef class _AssocIter (_MapIterator):
+  cdef AssocTable t
+  cdef assoc_iter i
+  def __new__(me, AssocTable t):
+    me.t = t
+    assoc_mkiter(&me.i, &me.t._t)
+  cdef void *_next(me):
+    return assoc_next(&me.i)
+
+#----- That's all, folks ----------------------------------------------------