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

   1 # -*-pyrex-*-
   2 #
   3 # $Id$
   4 #
   5 # Common mapping stuff
   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 class Mapping
  29
  30 cdef class _MapIterator:
  31   cdef void *_next(me):
  32     return NULL
  33
  34 cdef class Mapping:
  35
  36   ## Subclasses must implement these
  37   cdef int _init(me) except -1:
  38     raise TypeError, 'abstract class'
  39   cdef void *_find(me, object key, unsigned *f) except NULL:
  40     raise SystemError, 'unimplemented _find'
  41   cdef object _key(me, void *e):
  42     return None
  43   cdef object _value(me, void *e):
  44     return None
  45   cdef void _setval(me, void *e, object val):
  46     pass
  47   cdef void _del(me, void *e):
  48     pass
  49   cdef _MapIterator _iter(me):
  50     raise SystemError, 'unimplemented _iter'
  51
  52   ## Initialization
  53   def __new__(me, *hunoz, **hukairz):
  54     me._init()
  55   def __init__(me, stuff = None, **kw):
  56     me.update(stuff, kw)
  57
  58   ## Bulk update
  59   def update(me, stuff = None, **kw):
  60     cdef unsigned f
  61     if stuff is None:
  62       pass
  63     elif hasattr(stuff, 'itemiter'):
  64       for k, v in stuff.itemiter:
  65         me._setval(me._find(k, &f), v)
  66     elif hasattr(stuff, 'keys'):
  67       for k in stuff.keys():
  68         me._setval(me._find( k, &f), stuff[k])
  69     else:
  70       for k, v in stuff:
  71         me._setval(me._find(k, &f), v)
  72     for k, v in kw.iteritems():
  73       me._setval(me._find(k, &f), v)
  74     return me
  75
  76   ## Item access
  77   def __getitem__(me, key):
  78     cdef void *e
  79     e = me._find(key, NULL)
  80     if not e:
  81       raise KeyError, key
  82     return me._value(e)
  83   def __setitem__(me, key, value):
  84     cdef unsigned f
  85     me._setval(me._find(key, &f), value)
  86   def __delitem__(me, key):
  87     cdef void *e
  88     cdef unsigned f
  89     e = me._find(key, &f)
  90     if not e:
  91       raise KeyError, key
  92     me._del(e)
  93   def get(me, key, default = None):
  94     cdef void *e
  95     e = me._find(key, NULL)
  96     if not e:
  97       return default
  98     return me._value(e)
  99   def setdefault(me, key, default = None):
 100     cdef void *e
 101     cdef unsigned f
 102     e = me._find(key, &f)
 103     if f:
 104       return me._value(e)
 105     else:
 106       me._setval(e, default)
 107       return default
 108   def pop(me, key, default = None):
 109     cdef void *e
 110     e = me._find(key, NULL)
 111     if not e:
 112       return default
 113     rc = me._value(e)
 114     me._del(e)
 115     return rc
 116   def popitem(me):
 117     cdef _MapIterator i
 118     cdef void *e
 119     i = me._iter()
 120     e = i._next()
 121     if not e:
 122       raise ValueError, 'popitem(): table is empty'
 123     return me._key(e), me._value(e)
 124
 125   ## Lists of items
 126   cdef object _list(me, object (*func)(Mapping m, void *e)):
 127     cdef _MapIterator i
 128     cdef void *e
 129     i = me._iter()
 130     l = []
 131     while 1:
 132       e = i._next()
 133       if not e:
 134         break
 135       l.append(func(me, e))
 136     return l
 137
 138   def keys(me):
 139     return me._list(_map_key)
 140   def values(me):
 141     return me._list(_map_value)
 142   def items(me):
 143     return me._list(_map_item)
 144
 145   def clear(me):
 146     cdef _MapIterator i
 147     cdef void *e
 148     i = me._iter()
 149     while 1:
 150       e = i._next()
 151       if not e:
 152         break
 153       me._del(e)
 154     return me
 155
 156   ## Iteration
 157   def __iter__(me):
 158     return MapKeyIter(me)
 159   def iterkeys(me):
 160     return MapKeyIter(me)
 161   def itervalues(me):
 162     return MapValueIter(me)
 163   def iteritems(me):
 164     return MapItemIter(me)
 165
 166 cdef class MapIterBase:
 167   cdef Mapping m
 168   cdef object (*func)(Mapping m, void *e)
 169   cdef _MapIterator i
 170   cdef int _init(me) except -1:
 171     raise TypeError, 'abstract class'
 172   def __new__(me):
 173     me.i = m._iter()
 174     me._init()
 175   def __iter__(me):
 176     return me
 177   def __next__(me):
 178     cdef void *e
 179     e = me.i._next()
 180     if not e:
 181       raise StopIteration
 182     return me.func(me.m, e)
 183 cdef class MapKeyIter (MapIterBase):
 184   cdef int _init(me) except -1:
 185     me.func = _map_key
 186     return 0
 187 cdef class MapValueIter (MapIterBase):
 188   cdef int _init(me) except -1:
 189     me.func = _map_value
 190     return 0
 191 cdef class MapItemIter (MapIterBase):
 192   cdef int _init(me) except -1:
 193     me.func = _map_item
 194     return 0
 195
 196 cdef object _map_key(Mapping m, void *e):
 197   return m._key(e)
 198 cdef object _map_value(Mapping m, void *e):
 199   return m._value(e)
 200 cdef object _map_item(Mapping m, void *e):
 201   return m._key(e), m._value(e)
 202
 203 #----- That's all, folks ----------------------------------------------------