← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/_simple_set_pyx.pyx

  • Committer: John Arbash Meinel
  • Date: 2009-10-12 18:14:21 UTC
  • mto: This revision was merged to the branch mainline in revision 4736.
  • Revision ID: john@arbash-meinel.com-20091012181421-6s3yxopo9s40wqjr
NEWS entry for StaticTuple class.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/_simple_set_pyx.pyx
1
 
# Copyright (C) 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2009 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
21
21
 
22
22
cdef extern from "Python.h":
23
23
    ctypedef unsigned long size_t
24
 
    ctypedef long (*hashfunc)(PyObject*) except -1
25
 
    ctypedef object (*richcmpfunc)(PyObject *, PyObject *, int)
 
24
    ctypedef long (*hashfunc)(PyObject*)
 
25
    ctypedef PyObject *(*richcmpfunc)(PyObject *, PyObject *, int)
26
26
    ctypedef int (*visitproc)(PyObject *, void *)
27
27
    ctypedef int (*traverseproc)(PyObject *, visitproc, void *)
28
28
    int Py_EQ
 
29
    PyObject *Py_True
 
30
    PyObject *Py_NotImplemented
29
31
    void Py_INCREF(PyObject *)
30
32
    void Py_DECREF(PyObject *)
31
33
    ctypedef struct PyTypeObject:
34
36
        traverseproc tp_traverse
35
37
 
36
38
    PyTypeObject *Py_TYPE(PyObject *)
37
 
    # Note: *Don't* use hash(), Pyrex 0.9.8.5 thinks it returns an 'int', and
38
 
    #       thus silently truncates to 32-bits on 64-bit machines.
39
 
    long PyObject_Hash(PyObject *) except -1
 
39
    int PyObject_IsTrue(PyObject *)
40
40
        
41
41
    void *PyMem_Malloc(size_t nbytes)
42
42
    void PyMem_Free(void *)
55
55
_dummy = <PyObject *>_dummy_obj
56
56
 
57
57
 
58
 
cdef object _NotImplemented
59
 
_NotImplemented = NotImplemented
60
 
 
61
 
 
62
58
cdef int _is_equal(PyObject *this, long this_hash, PyObject *other) except -1:
63
59
    cdef long other_hash
 
60
    cdef PyObject *res
64
61
 
65
62
    if this == other:
66
63
        return 1
67
 
    other_hash = PyObject_Hash(other)
 
64
    other_hash = Py_TYPE(other).tp_hash(other)
 
65
    if other_hash == -1:
 
66
        # Even though other successfully hashed in the past, it seems to have
 
67
        # changed its mind, and failed this time, so propogate the failure.
 
68
        return -1
68
69
    if other_hash != this_hash:
69
70
        return 0
70
71
 
76
77
    #      equal. (It doesn't try to cast them both to some intermediate form
77
78
    #      that would compare equal.)
78
79
    res = Py_TYPE(this).tp_richcompare(this, other, Py_EQ)
79
 
    if res is _NotImplemented:
 
80
    if res == NULL: # Exception
 
81
        return -1
 
82
    if PyObject_IsTrue(res):
 
83
        Py_DECREF(res)
 
84
        return 1
 
85
    if res == Py_NotImplemented:
 
86
        Py_DECREF(res)
80
87
        res = Py_TYPE(other).tp_richcompare(other, this, Py_EQ)
81
 
        if res is _NotImplemented:
82
 
            return 0
83
 
    if res:
 
88
    if res == NULL:
 
89
        return -1
 
90
    if PyObject_IsTrue(res):
 
91
        Py_DECREF(res)
84
92
        return 1
 
93
    Py_DECREF(res)
85
94
    return 0
86
95
 
87
96
 
115
124
            raise MemoryError()
116
125
        memset(self._table, 0, n_bytes)
117
126
 
118
 
    def __sizeof__(self):
119
 
        # Note: Pyrex doesn't allow sizeof(class) so we re-implement it here.
120
 
        # Bits are:
121
 
        #   1: PyObject
122
 
        #   2: vtable *
123
 
        #   3: 3 Py_ssize_t
124
 
        #   4: PyObject**
125
 
        # Note that we might get alignment, etc, wrong, but at least this is
126
 
        # better than no estimate at all
127
 
        # return sizeof(SimpleSet) + (self._mask + 1) * (sizeof(PyObject*))
128
 
        return (sizeof(PyObject) + sizeof(void*)
129
 
                + 3*sizeof(Py_ssize_t) + sizeof(PyObject**)
130
 
                + (self._mask + 1) * sizeof(PyObject*))
131
 
 
132
127
    def __dealloc__(self):
133
128
        if self._table != NULL:
134
129
            PyMem_Free(self._table)
208
203
        mask = self._mask
209
204
        table = self._table
210
205
 
211
 
        the_hash = PyObject_Hash(key)
 
206
        the_hash = Py_TYPE(key).tp_hash(key)
 
207
        if the_hash == -1:
 
208
            return -1
212
209
        i = the_hash
213
210
        for n_lookup from 0 <= n_lookup <= <size_t>mask: # Don't loop forever
214
211
            slot = &table[i & mask]
461
458
    cdef long key_hash
462
459
    cdef PyObject **table, **slot, *cur, **free_slot, *py_key
463
460
 
464
 
    py_key = <PyObject *>key
465
 
    # Note: avoid using hash(obj) because of a bug w/ pyrex 0.9.8.5 and 64-bit
466
 
    #       (it treats hash() as returning an 'int' rather than a 'long')
467
 
    key_hash = PyObject_Hash(py_key)
 
461
    # hash is a signed long(), we are using an offset at unsigned size_t
 
462
    key_hash = hash(key)
468
463
    i = <size_t>key_hash
469
464
    mask = self._mask
470
465
    table = self._table
471
466
    free_slot = NULL
 
467
    py_key = <PyObject *>key
472
468
    for n_lookup from 0 <= n_lookup <= <size_t>mask: # Don't loop forever
473
469
        slot = &table[i & mask]
474
470
        cur = slot[0]
554
550
    return _check_self(self)._used
555
551
 
556
552
 
557
 
cdef api int SimpleSet_Next(object self, Py_ssize_t *pos,
558
 
                            PyObject **key) except -1:
 
553
cdef api int SimpleSet_Next(object self, Py_ssize_t *pos, PyObject **key):
559
554
    """Walk over items in a SimpleSet.
560
555
 
561
556
    :param pos: should be initialized to 0 by the caller, and will be updated
582
577
    return 1
583
578
 
584
579
 
585
 
cdef int SimpleSet_traverse(SimpleSet self, visitproc visit,
586
 
                            void *arg) except -1:
 
580
cdef int SimpleSet_traverse(SimpleSet self, visitproc visit, void *arg):
587
581
    """This is an implementation of 'tp_traverse' that hits the whole table.
588
582
 
589
583
    Cython/Pyrex don't seem to let you define a tp_traverse, and they only