chiark - git - mdw - mLib-python/blob - 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 ----------------------------------------------------