# -*-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 __cinit__(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 ----------------------------------------------------