chiark - git - mdw - mLib-python/blob - assoc.pyx

   1 ### -*-pyrex-*-
   2 ###
   3 ### Association tables
   4 ###
   5 ### (c) 2005 Straylight/Edgeware
   6 ###
   7
   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.
  25
  26 cdef struct _assoc_entry:
  27   sym_base _b
  28   PyObject *v
  29
  30 cdef 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 Py_ssize_t n
  38     cdef _assoc_entry *e
  39     cdef atom *a
  40     a = ATOM_A(atom_pyintern(key))
  41     PyObject_AsReadBuffer(key, <cvp *>&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
  57   cdef void _setval(me, void *e, object val):
  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
  73 cdef class _AssocIter (_MapIterator):
  74   cdef AssocTable t
  75   cdef assoc_iter i
  76   def __cinit__(me, AssocTable t not None):
  77     me.t = t
  78     assoc_mkiter(&me.i, &me.t._t)
  79   cdef void *_next(me):
  80     return assoc_next(&me.i)
  81
  82 ###----- That's all, folks --------------------------------------------------