~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/_static_tuple_c.c

- win32 get_console_size from Alexander

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/* Copyright (C) 2009, 2010 Canonical Ltd
2
 
 * 
3
 
 * This program is free software; you can redistribute it and/or modify
4
 
 * it under the terms of the GNU General Public License as published by
5
 
 * the Free Software Foundation; either version 2 of the License, or
6
 
 * (at your option) any later version.
7
 
 *
8
 
 * This program is distributed in the hope that it will be useful,
9
 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
 * GNU General Public License for more details.
12
 
 *
13
 
 * You should have received a copy of the GNU General Public License
14
 
 * along with this program; if not, write to the Free Software
15
 
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 */
17
 
 
18
 
/* Must be defined before importing _static_tuple_c.h so that we get the right
19
 
 * linkage.
20
 
 */
21
 
#define STATIC_TUPLE_MODULE
22
 
 
23
 
#include <Python.h>
24
 
#include "python-compat.h"
25
 
 
26
 
#include "_static_tuple_c.h"
27
 
#include "_export_c_api.h"
28
 
 
29
 
/* Pyrex 0.9.6.4 exports _simple_set_pyx_api as
30
 
 * import__simple_set_pyx(), while Pyrex 0.9.8.5 and Cython 0.11.3 export them
31
 
 * as import_bzrlib___simple_set_pyx(). As such, we just #define one to be
32
 
 * equivalent to the other in our internal code.
33
 
 */
34
 
#define import__simple_set_pyx import_bzrlib___simple_set_pyx
35
 
#include "_simple_set_pyx_api.h"
36
 
 
37
 
#if defined(__GNUC__)
38
 
#   define inline __inline__
39
 
#elif defined(_MSC_VER)
40
 
#   define inline __inline
41
 
#else
42
 
#   define inline
43
 
#endif
44
 
 
45
 
 
46
 
/* The one and only StaticTuple with no values */
47
 
static StaticTuple *_empty_tuple = NULL;
48
 
static PyObject *_interned_tuples = NULL;
49
 
 
50
 
 
51
 
static inline int
52
 
_StaticTuple_is_interned(StaticTuple *self)
53
 
{
54
 
    return self->flags & STATIC_TUPLE_INTERNED_FLAG;
55
 
}
56
 
 
57
 
 
58
 
 
59
 
static PyObject *
60
 
StaticTuple_as_tuple(StaticTuple *self)
61
 
{
62
 
    PyObject *tpl = NULL, *obj = NULL;
63
 
    int i, len;
64
 
 
65
 
    len = self->size;
66
 
    tpl = PyTuple_New(len);
67
 
    if (!tpl) {
68
 
        /* Malloc failure */
69
 
        return NULL;
70
 
    }
71
 
    for (i = 0; i < len; ++i) {
72
 
        obj = (PyObject *)self->items[i];
73
 
        Py_INCREF(obj);
74
 
        PyTuple_SET_ITEM(tpl, i, obj);
75
 
    }
76
 
    return tpl;
77
 
}
78
 
 
79
 
 
80
 
static char StaticTuple_as_tuple_doc[] = "as_tuple() => tuple";
81
 
 
82
 
static StaticTuple *
83
 
StaticTuple_Intern(StaticTuple *self)
84
 
{
85
 
    PyObject *canonical_tuple = NULL;
86
 
 
87
 
    if (_interned_tuples == NULL || _StaticTuple_is_interned(self)) {
88
 
        Py_INCREF(self);
89
 
        return self;
90
 
    }
91
 
    /* SimpleSet_Add returns whatever object is present at self
92
 
     * or the new object if it needs to add it.
93
 
     */
94
 
    canonical_tuple = SimpleSet_Add(_interned_tuples, (PyObject *)self);
95
 
    if (!canonical_tuple) {
96
 
        // Some sort of exception, propogate it.
97
 
        return NULL;
98
 
    }
99
 
    if (canonical_tuple != (PyObject *)self) {
100
 
        // There was already a tuple with that value
101
 
        return (StaticTuple *)canonical_tuple;
102
 
    }
103
 
    self->flags |= STATIC_TUPLE_INTERNED_FLAG;
104
 
    // The two references in the dict do not count, so that the StaticTuple
105
 
    // object does not become immortal just because it was interned.
106
 
    Py_REFCNT(self) -= 1;
107
 
    return self;
108
 
}
109
 
 
110
 
static char StaticTuple_Intern_doc[] = "intern() => unique StaticTuple\n"
111
 
    "Return a 'canonical' StaticTuple object.\n"
112
 
    "Similar to intern() for strings, this makes sure there\n"
113
 
    "is only one StaticTuple object for a given value\n."
114
 
    "Common usage is:\n"
115
 
    "  key = StaticTuple('foo', 'bar').intern()\n";
116
 
 
117
 
 
118
 
static void
119
 
StaticTuple_dealloc(StaticTuple *self)
120
 
{
121
 
    int i, len;
122
 
 
123
 
    if (_StaticTuple_is_interned(self)) {
124
 
        /* revive dead object temporarily for Discard */
125
 
        Py_REFCNT(self) = 2;
126
 
        if (SimpleSet_Discard(_interned_tuples, (PyObject*)self) != 1)
127
 
            Py_FatalError("deletion of interned StaticTuple failed");
128
 
        self->flags &= ~STATIC_TUPLE_INTERNED_FLAG;
129
 
    }
130
 
    len = self->size;
131
 
    for (i = 0; i < len; ++i) {
132
 
        Py_XDECREF(self->items[i]);
133
 
    }
134
 
    Py_TYPE(self)->tp_free((PyObject *)self);
135
 
}
136
 
 
137
 
 
138
 
/* Similar to PyTuple_New() */
139
 
static StaticTuple *
140
 
StaticTuple_New(Py_ssize_t size)
141
 
{
142
 
    StaticTuple *stuple;
143
 
 
144
 
    if (size < 0 || size > 255) {
145
 
        /* Too big or too small */
146
 
        PyErr_SetString(PyExc_ValueError, "StaticTuple(...)"
147
 
            " takes from 0 to 255 items");
148
 
        return NULL;
149
 
    }
150
 
    if (size == 0 && _empty_tuple != NULL) {
151
 
        Py_INCREF(_empty_tuple);
152
 
        return _empty_tuple;
153
 
    }
154
 
    /* Note that we use PyObject_NewVar because we want to allocate a variable
155
 
     * width entry. However we *aren't* truly a PyVarObject because we don't
156
 
     * use a long for ob_size. Instead we use a plain 'size' that is an int,
157
 
     * and will be overloaded with flags in the future.
158
 
     * As such we do the alloc, and then have to clean up anything it does
159
 
     * incorrectly.
160
 
     */
161
 
    stuple = PyObject_NewVar(StaticTuple, &StaticTuple_Type, size);
162
 
    if (stuple == NULL) {
163
 
        return NULL;
164
 
    }
165
 
    stuple->size = size;
166
 
    stuple->flags = 0;
167
 
    stuple->_unused0 = 0;
168
 
    stuple->_unused1 = 0;
169
 
    if (size > 0) {
170
 
        memset(stuple->items, 0, sizeof(PyObject *) * size);
171
 
    }
172
 
#if STATIC_TUPLE_HAS_HASH
173
 
    stuple->hash = -1;
174
 
#endif
175
 
    return stuple;
176
 
}
177
 
 
178
 
 
179
 
static StaticTuple *
180
 
StaticTuple_FromSequence(PyObject *sequence)
181
 
{
182
 
    StaticTuple *new = NULL;
183
 
    PyObject *as_tuple = NULL;
184
 
    PyObject *item;
185
 
    Py_ssize_t i, size;
186
 
 
187
 
    if (StaticTuple_CheckExact(sequence)) {
188
 
        Py_INCREF(sequence);
189
 
        return (StaticTuple *)sequence;
190
 
    }
191
 
    if (!PySequence_Check(sequence)) {
192
 
        as_tuple = PySequence_Tuple(sequence);
193
 
        if (as_tuple == NULL)
194
 
            goto done;
195
 
        sequence = as_tuple;
196
 
    }
197
 
    size = PySequence_Size(sequence);
198
 
    if (size == -1) {
199
 
        goto done;
200
 
    }
201
 
    new = StaticTuple_New(size);
202
 
    if (new == NULL) {
203
 
        goto done;
204
 
    }
205
 
    for (i = 0; i < size; ++i) {
206
 
        // This returns a new reference, which we then 'steal' with 
207
 
        // StaticTuple_SET_ITEM
208
 
        item = PySequence_GetItem(sequence, i);
209
 
        if (item == NULL) {
210
 
            Py_DECREF(new);
211
 
            new = NULL;
212
 
            goto done;
213
 
        }
214
 
        StaticTuple_SET_ITEM(new, i, item);
215
 
    }
216
 
done:
217
 
    Py_XDECREF(as_tuple);
218
 
    return (StaticTuple *)new;
219
 
}
220
 
 
221
 
static StaticTuple *
222
 
StaticTuple_from_sequence(PyObject *self, PyObject *args, PyObject *kwargs)
223
 
{
224
 
    PyObject *sequence;
225
 
    if (!PyArg_ParseTuple(args, "O", &sequence))
226
 
        return NULL;
227
 
    return StaticTuple_FromSequence(sequence);
228
 
}
229
 
 
230
 
 
231
 
/* Check that all items we point to are 'valid' */
232
 
static int
233
 
StaticTuple_check_items(StaticTuple *self)
234
 
{
235
 
    int i;
236
 
    PyObject *obj;
237
 
 
238
 
    for (i = 0; i < self->size; ++i) {
239
 
        obj = self->items[i];
240
 
        if (obj == NULL) {
241
 
            PyErr_SetString(PyExc_RuntimeError, "StaticTuple(...)"
242
 
                " should not have a NULL entry.");
243
 
            return 0;
244
 
        }
245
 
        if (PyString_CheckExact(obj)
246
 
            || StaticTuple_CheckExact(obj)
247
 
            || obj == Py_None
248
 
            || PyBool_Check(obj)
249
 
            || PyInt_CheckExact(obj)
250
 
            || PyLong_CheckExact(obj)
251
 
            || PyFloat_CheckExact(obj)
252
 
            || PyUnicode_CheckExact(obj)
253
 
            ) continue;
254
 
        PyErr_Format(PyExc_TypeError, "StaticTuple(...)"
255
 
            " requires that all items are one of"
256
 
            " str, StaticTuple, None, bool, int, long, float, or unicode"
257
 
            " not %s.", Py_TYPE(obj)->tp_name);
258
 
        return 0;
259
 
    }
260
 
    return 1;
261
 
}
262
 
 
263
 
static PyObject *
264
 
StaticTuple_new_constructor(PyTypeObject *type, PyObject *args, PyObject *kwds)
265
 
{
266
 
    StaticTuple *self;
267
 
    PyObject *obj = NULL;
268
 
    Py_ssize_t i, len = 0;
269
 
 
270
 
    if (type != &StaticTuple_Type) {
271
 
        PyErr_SetString(PyExc_TypeError, "we only support creating StaticTuple");
272
 
        return NULL;
273
 
    }
274
 
    if (!PyTuple_CheckExact(args)) {
275
 
        PyErr_SetString(PyExc_TypeError, "args must be a tuple");
276
 
        return NULL;
277
 
    }
278
 
    len = PyTuple_GET_SIZE(args);
279
 
    if (len < 0 || len > 255) {
280
 
        /* Check the length here so we can raise a TypeError instead of
281
 
         * StaticTuple_New's ValueError.
282
 
         */
283
 
        PyErr_SetString(PyExc_TypeError, "StaticTuple(...)"
284
 
            " takes from 0 to 255 items");
285
 
        return NULL;
286
 
    }
287
 
    self = (StaticTuple *)StaticTuple_New(len);
288
 
    if (self == NULL) {
289
 
        return NULL;
290
 
    }
291
 
    for (i = 0; i < len; ++i) {
292
 
        obj = PyTuple_GET_ITEM(args, i);
293
 
        Py_INCREF(obj);
294
 
        self->items[i] = obj;
295
 
    }
296
 
    if (!StaticTuple_check_items(self)) {
297
 
        type->tp_dealloc((PyObject *)self);
298
 
        return NULL;
299
 
    }
300
 
    return (PyObject *)self;
301
 
}
302
 
 
303
 
static PyObject *
304
 
StaticTuple_repr(StaticTuple *self)
305
 
{
306
 
    PyObject *as_tuple, *tuple_repr, *result;
307
 
 
308
 
    as_tuple = StaticTuple_as_tuple(self);
309
 
    if (as_tuple == NULL) {
310
 
        return NULL;
311
 
    }
312
 
    tuple_repr = PyObject_Repr(as_tuple);
313
 
    Py_DECREF(as_tuple);
314
 
    if (tuple_repr == NULL) {
315
 
        return NULL;
316
 
    }
317
 
    result = PyString_FromFormat("StaticTuple%s",
318
 
                                 PyString_AsString(tuple_repr));
319
 
    return result;
320
 
}
321
 
 
322
 
static long
323
 
StaticTuple_hash(StaticTuple *self)
324
 
{
325
 
    /* adapted from tuplehash(), is the specific hash value considered
326
 
     * 'stable'?
327
 
     */
328
 
    register long x, y;
329
 
    Py_ssize_t len = self->size;
330
 
    PyObject **p;
331
 
    long mult = 1000003L;
332
 
 
333
 
#if STATIC_TUPLE_HAS_HASH
334
 
    if (self->hash != -1) {
335
 
        return self->hash;
336
 
    }
337
 
#endif
338
 
    x = 0x345678L;
339
 
    p = self->items;
340
 
    // TODO: We could set specific flags if we know that, for example, all the
341
 
    //       items are strings. I haven't seen a real-world benefit to that
342
 
    //       yet, though.
343
 
    while (--len >= 0) {
344
 
        y = PyObject_Hash(*p++);
345
 
        if (y == -1) /* failure */
346
 
            return -1;
347
 
        x = (x ^ y) * mult;
348
 
        /* the cast might truncate len; that doesn't change hash stability */
349
 
        mult += (long)(82520L + len + len);
350
 
    }
351
 
    x += 97531L;
352
 
    if (x == -1)
353
 
        x = -2;
354
 
#if STATIC_TUPLE_HAS_HASH
355
 
    self->hash = x;
356
 
#endif
357
 
    return x;
358
 
}
359
 
 
360
 
static PyObject *
361
 
StaticTuple_richcompare_to_tuple(StaticTuple *v, PyObject *wt, int op)
362
 
{
363
 
    PyObject *vt;
364
 
    PyObject *result = NULL;
365
 
    
366
 
    vt = StaticTuple_as_tuple((StaticTuple *)v);
367
 
    if (vt == NULL) {
368
 
        goto done;
369
 
    }
370
 
    if (!PyTuple_Check(wt)) {
371
 
        PyErr_BadInternalCall();
372
 
        goto done;
373
 
    }
374
 
    /* Now we have 2 tuples to compare, do it */
375
 
    result = PyTuple_Type.tp_richcompare(vt, wt, op);
376
 
done:
377
 
    Py_XDECREF(vt);
378
 
    return result;
379
 
}
380
 
 
381
 
/** Compare two objects to determine if they are equivalent.
382
 
 * The basic flow is as follows
383
 
 *  1) First make sure that both objects are StaticTuple instances. If they
384
 
 *     aren't then cast self to a tuple, and have the tuple do the comparison.
385
 
 *  2) Special case comparison to Py_None, because it happens to occur fairly
386
 
 *     often in the test suite.
387
 
 *  3) Special case when v and w are the same pointer. As we know the answer to
388
 
 *     all queries without walking individual items.
389
 
 *  4) For all operations, we then walk the items to find the first paired
390
 
 *     items that are not equal.
391
 
 *  5) If all items found are equal, we then check the length of self and
392
 
 *     other to determine equality.
393
 
 *  6) If an item differs, then we apply "op" to those last two items. (eg.
394
 
 *     StaticTuple(A, B) > StaticTuple(A, C) iff B > C)
395
 
 */
396
 
 
397
 
static PyObject *
398
 
StaticTuple_richcompare(PyObject *v, PyObject *w, int op)
399
 
{
400
 
    StaticTuple *v_st, *w_st;
401
 
    Py_ssize_t vlen, wlen, min_len, i;
402
 
    PyObject *v_obj, *w_obj;
403
 
    richcmpfunc string_richcompare;
404
 
 
405
 
    if (!StaticTuple_CheckExact(v)) {
406
 
        /* This has never triggered, according to python-dev it seems this
407
 
         * might trigger if '__op__' is defined but '__rop__' is not, sort of
408
 
         * case. Such as "None == StaticTuple()"
409
 
         */
410
 
        fprintf(stderr, "self is not StaticTuple\n");
411
 
        Py_INCREF(Py_NotImplemented);
412
 
        return Py_NotImplemented;
413
 
    }
414
 
    v_st = (StaticTuple *)v;
415
 
    if (StaticTuple_CheckExact(w)) {
416
 
        /* The most common case */
417
 
        w_st = (StaticTuple*)w;
418
 
    } else if (PyTuple_Check(w)) {
419
 
        /* One of v or w is a tuple, so we go the 'slow' route and cast up to
420
 
         * tuples to compare.
421
 
         */
422
 
        /* TODO: This seems to be triggering more than I thought it would...
423
 
         *       We probably want to optimize comparing self to other when
424
 
         *       other is a tuple.
425
 
         */
426
 
        return StaticTuple_richcompare_to_tuple(v_st, w, op);
427
 
    } else if (w == Py_None) {
428
 
        // None is always less than the object
429
 
        switch (op) {
430
 
        case Py_NE:case Py_GT:case Py_GE:
431
 
            Py_INCREF(Py_True);
432
 
            return Py_True;
433
 
        case Py_EQ:case Py_LT:case Py_LE:
434
 
            Py_INCREF(Py_False);
435
 
            return Py_False;
436
 
    default: // Should never happen
437
 
        return Py_NotImplemented;
438
 
        }
439
 
    } else {
440
 
        /* We don't special case this comparison, we just let python handle
441
 
         * it.
442
 
         */
443
 
         Py_INCREF(Py_NotImplemented);
444
 
         return Py_NotImplemented;
445
 
    }
446
 
    /* Now we know that we have 2 StaticTuple objects, so let's compare them.
447
 
     * This code is inspired from tuplerichcompare, except we know our
448
 
     * objects are limited in scope, so we can inline some comparisons.
449
 
     */
450
 
    if (v == w) {
451
 
        /* Identical pointers, we can shortcut this easily. */
452
 
        switch (op) {
453
 
        case Py_EQ:case Py_LE:case Py_GE:
454
 
            Py_INCREF(Py_True);
455
 
            return Py_True;
456
 
        case Py_NE:case Py_LT:case Py_GT:
457
 
            Py_INCREF(Py_False);
458
 
            return Py_False;
459
 
        }
460
 
    }
461
 
    if (op == Py_EQ
462
 
        && _StaticTuple_is_interned(v_st)
463
 
        && _StaticTuple_is_interned(w_st))
464
 
    {
465
 
        /* If both objects are interned, we know they are different if the
466
 
         * pointer is not the same, which would have been handled by the
467
 
         * previous if. No need to compare the entries.
468
 
         */
469
 
        Py_INCREF(Py_False);
470
 
        return Py_False;
471
 
    }
472
 
 
473
 
    /* The only time we are likely to compare items of different lengths is in
474
 
     * something like the interned_keys set. However, the hash is good enough
475
 
     * that it is rare. Note that 'tuple_richcompare' also does not compare
476
 
     * lengths here.
477
 
     */
478
 
    vlen = v_st->size;
479
 
    wlen = w_st->size;
480
 
    min_len = (vlen < wlen) ? vlen : wlen;
481
 
    string_richcompare = PyString_Type.tp_richcompare;
482
 
    for (i = 0; i < min_len; i++) {
483
 
        PyObject *result = NULL;
484
 
        v_obj = StaticTuple_GET_ITEM(v_st, i);
485
 
        w_obj = StaticTuple_GET_ITEM(w_st, i);
486
 
        if (v_obj == w_obj) {
487
 
            /* Shortcut case, these must be identical */
488
 
            continue;
489
 
        }
490
 
        if (PyString_CheckExact(v_obj) && PyString_CheckExact(w_obj)) {
491
 
            result = string_richcompare(v_obj, w_obj, Py_EQ);
492
 
        } else if (StaticTuple_CheckExact(v_obj) &&
493
 
                   StaticTuple_CheckExact(w_obj))
494
 
        {
495
 
            /* Both are StaticTuple types, so recurse */
496
 
            result = StaticTuple_richcompare(v_obj, w_obj, Py_EQ);
497
 
        } else {
498
 
            /* Fall back to generic richcompare */
499
 
            result = PyObject_RichCompare(v_obj, w_obj, Py_EQ);
500
 
        }
501
 
        if (result == NULL) {
502
 
            return NULL; /* There seems to be an error */
503
 
        }
504
 
        if (result == Py_False) {
505
 
            // This entry is not identical, Shortcut for Py_EQ
506
 
            if (op == Py_EQ) {
507
 
                return result;
508
 
            }
509
 
            Py_DECREF(result);
510
 
            break;
511
 
        }
512
 
        if (result != Py_True) {
513
 
            /* We don't know *what* richcompare is returning, but it
514
 
             * isn't something we recognize
515
 
             */
516
 
            PyErr_BadInternalCall();
517
 
            Py_DECREF(result);
518
 
            return NULL;
519
 
        }
520
 
        Py_DECREF(result);
521
 
    }
522
 
    if (i >= min_len) {
523
 
        /* We walked off one of the lists, but everything compared equal so
524
 
         * far. Just compare the size.
525
 
         */
526
 
        int cmp;
527
 
        PyObject *res;
528
 
        switch (op) {
529
 
        case Py_LT: cmp = vlen <  wlen; break;
530
 
        case Py_LE: cmp = vlen <= wlen; break;
531
 
        case Py_EQ: cmp = vlen == wlen; break;
532
 
        case Py_NE: cmp = vlen != wlen; break;
533
 
        case Py_GT: cmp = vlen >  wlen; break;
534
 
        case Py_GE: cmp = vlen >= wlen; break;
535
 
        default: return NULL; /* cannot happen */
536
 
        }
537
 
        if (cmp)
538
 
            res = Py_True;
539
 
        else
540
 
            res = Py_False;
541
 
        Py_INCREF(res);
542
 
        return res;
543
 
    }
544
 
    /* The last item differs, shortcut the Py_NE case */
545
 
    if (op == Py_NE) {
546
 
        Py_INCREF(Py_True);
547
 
        return Py_True;
548
 
    }
549
 
    /* It is some other comparison, go ahead and do the real check. */
550
 
    if (PyString_CheckExact(v_obj) && PyString_CheckExact(w_obj))
551
 
    {
552
 
        return string_richcompare(v_obj, w_obj, op);
553
 
    } else if (StaticTuple_CheckExact(v_obj) &&
554
 
               StaticTuple_CheckExact(w_obj))
555
 
    {
556
 
        /* Both are StaticTuple types, so recurse */
557
 
        return StaticTuple_richcompare(v_obj, w_obj, op);
558
 
    } else {
559
 
        return PyObject_RichCompare(v_obj, w_obj, op);
560
 
    }
561
 
}
562
 
 
563
 
 
564
 
static Py_ssize_t
565
 
StaticTuple_length(StaticTuple *self)
566
 
{
567
 
    return self->size;
568
 
}
569
 
 
570
 
 
571
 
static PyObject *
572
 
StaticTuple__is_interned(StaticTuple *self)
573
 
{
574
 
    if (_StaticTuple_is_interned(self)) {
575
 
        Py_INCREF(Py_True);
576
 
        return Py_True;
577
 
    }
578
 
    Py_INCREF(Py_False);
579
 
    return Py_False;
580
 
}
581
 
 
582
 
static char StaticTuple__is_interned_doc[] = "_is_interned() => True/False\n"
583
 
    "Check to see if this tuple has been interned.\n";
584
 
 
585
 
 
586
 
static PyObject *
587
 
StaticTuple_reduce(StaticTuple *self)
588
 
{
589
 
    PyObject *result = NULL, *as_tuple = NULL;
590
 
 
591
 
    result = PyTuple_New(2);
592
 
    if (!result) {
593
 
        return NULL;
594
 
    }
595
 
    as_tuple = StaticTuple_as_tuple(self);
596
 
    if (as_tuple == NULL) {
597
 
        Py_DECREF(result);
598
 
        return NULL;
599
 
    }
600
 
    Py_INCREF(&StaticTuple_Type);
601
 
    PyTuple_SET_ITEM(result, 0, (PyObject *)&StaticTuple_Type);
602
 
    PyTuple_SET_ITEM(result, 1, as_tuple);
603
 
    return result;
604
 
}
605
 
 
606
 
static char StaticTuple_reduce_doc[] = "__reduce__() => tuple\n";
607
 
 
608
 
 
609
 
static PyObject *
610
 
StaticTuple_add(PyObject *v, PyObject *w)
611
 
{
612
 
    Py_ssize_t i, len_v, len_w;
613
 
    PyObject *item;
614
 
    StaticTuple *result;
615
 
     /* StaticTuples and plain tuples may be added (concatenated) to
616
 
      * StaticTuples.
617
 
      */
618
 
    if (StaticTuple_CheckExact(v)) {
619
 
        len_v = ((StaticTuple*)v)->size;
620
 
    } else if (PyTuple_Check(v)) {
621
 
        len_v = PyTuple_GET_SIZE(v);
622
 
    } else {
623
 
        Py_INCREF(Py_NotImplemented);
624
 
        return Py_NotImplemented;
625
 
    }
626
 
    if (StaticTuple_CheckExact(w)) {
627
 
        len_w = ((StaticTuple*)w)->size;
628
 
    } else if (PyTuple_Check(w)) {
629
 
        len_w = PyTuple_GET_SIZE(w);
630
 
    } else {
631
 
        Py_INCREF(Py_NotImplemented);
632
 
        return Py_NotImplemented;
633
 
    }
634
 
    result = StaticTuple_New(len_v + len_w);
635
 
    if (result == NULL)
636
 
        return NULL;
637
 
    for (i = 0; i < len_v; ++i) {
638
 
        // This returns a new reference, which we then 'steal' with 
639
 
        // StaticTuple_SET_ITEM
640
 
        item = PySequence_GetItem(v, i);
641
 
        if (item == NULL) {
642
 
            Py_DECREF(result);
643
 
            return NULL;
644
 
        }
645
 
        StaticTuple_SET_ITEM(result, i, item);
646
 
    }
647
 
    for (i = 0; i < len_w; ++i) {
648
 
        item = PySequence_GetItem(w, i);
649
 
        if (item == NULL) {
650
 
            Py_DECREF(result);
651
 
            return NULL;
652
 
        }
653
 
        StaticTuple_SET_ITEM(result, i+len_v, item);
654
 
    }
655
 
    if (!StaticTuple_check_items(result)) {
656
 
        Py_DECREF(result);
657
 
        return NULL;
658
 
    }
659
 
    return (PyObject *)result;
660
 
}
661
 
 
662
 
static PyObject *
663
 
StaticTuple_item(StaticTuple *self, Py_ssize_t offset)
664
 
{
665
 
    PyObject *obj;
666
 
    /* We cast to (int) to avoid worrying about whether Py_ssize_t is a
667
 
     * long long, etc. offsets should never be >2**31 anyway.
668
 
     */
669
 
    if (offset < 0) {
670
 
        PyErr_Format(PyExc_IndexError, "StaticTuple_item does not support"
671
 
            " negative indices: %d\n", (int)offset);
672
 
    } else if (offset >= self->size) {
673
 
        PyErr_Format(PyExc_IndexError, "StaticTuple index out of range"
674
 
            " %d >= %d", (int)offset, (int)self->size);
675
 
        return NULL;
676
 
    }
677
 
    obj = (PyObject *)self->items[offset];
678
 
    Py_INCREF(obj);
679
 
    return obj;
680
 
}
681
 
 
682
 
static PyObject *
683
 
StaticTuple_slice(StaticTuple *self, Py_ssize_t ilow, Py_ssize_t ihigh)
684
 
{
685
 
    PyObject *as_tuple, *result;
686
 
 
687
 
    as_tuple = StaticTuple_as_tuple(self);
688
 
    if (as_tuple == NULL) {
689
 
        return NULL;
690
 
    }
691
 
    result = PyTuple_Type.tp_as_sequence->sq_slice(as_tuple, ilow, ihigh);
692
 
    Py_DECREF(as_tuple);
693
 
    return result;
694
 
}
695
 
 
696
 
static int
697
 
StaticTuple_traverse(StaticTuple *self, visitproc visit, void *arg)
698
 
{
699
 
    Py_ssize_t i;
700
 
    for (i = self->size; --i >= 0;) {
701
 
        Py_VISIT(self->items[i]);
702
 
    }
703
 
    return 0;
704
 
}
705
 
 
706
 
 
707
 
static PyObject *
708
 
StaticTuple_sizeof(StaticTuple *self)
709
 
{
710
 
        Py_ssize_t res;
711
 
 
712
 
        res = _PyObject_SIZE(&StaticTuple_Type) + (int)self->size * sizeof(void*);
713
 
        return PyInt_FromSsize_t(res);
714
 
}
715
 
 
716
 
 
717
 
 
718
 
static char StaticTuple_doc[] =
719
 
    "C implementation of a StaticTuple structure."
720
 
    "\n This is used as StaticTuple(item1, item2, item3)"
721
 
    "\n This is similar to tuple, less flexible in what it"
722
 
    "\n supports, but also lighter memory consumption."
723
 
    "\n Note that the constructor mimics the () form of tuples"
724
 
    "\n Rather than the 'tuple()' constructor."
725
 
    "\n  eg. StaticTuple(a, b) == (a, b) == tuple((a, b))";
726
 
 
727
 
static PyMethodDef StaticTuple_methods[] = {
728
 
    {"as_tuple", (PyCFunction)StaticTuple_as_tuple, METH_NOARGS, StaticTuple_as_tuple_doc},
729
 
    {"intern", (PyCFunction)StaticTuple_Intern, METH_NOARGS, StaticTuple_Intern_doc},
730
 
    {"_is_interned", (PyCFunction)StaticTuple__is_interned, METH_NOARGS,
731
 
     StaticTuple__is_interned_doc},
732
 
    {"from_sequence", (PyCFunction)StaticTuple_from_sequence,
733
 
     METH_STATIC | METH_VARARGS,
734
 
     "Create a StaticTuple from a given sequence. This functions"
735
 
     " the same as the tuple() constructor."},
736
 
    {"__reduce__", (PyCFunction)StaticTuple_reduce, METH_NOARGS, StaticTuple_reduce_doc},
737
 
    {"__sizeof__",  (PyCFunction)StaticTuple_sizeof, METH_NOARGS}, 
738
 
    {NULL, NULL} /* sentinel */
739
 
};
740
 
 
741
 
 
742
 
static PyNumberMethods StaticTuple_as_number = {
743
 
    (binaryfunc) StaticTuple_add,   /* nb_add */
744
 
    0,                              /* nb_subtract */
745
 
    0,                              /* nb_multiply */
746
 
    0,                              /* nb_divide */
747
 
    0,                              /* nb_remainder */
748
 
    0,                              /* nb_divmod */
749
 
    0,                              /* nb_power */
750
 
    0,                              /* nb_negative */
751
 
    0,                              /* nb_positive */
752
 
    0,                              /* nb_absolute */
753
 
    0,                              /* nb_nonzero */
754
 
    0,                              /* nb_invert */
755
 
    0,                              /* nb_lshift */
756
 
    0,                              /* nb_rshift */
757
 
    0,                              /* nb_and */
758
 
    0,                              /* nb_xor */
759
 
    0,                              /* nb_or */
760
 
    0,                              /* nb_coerce */
761
 
};
762
 
    
763
 
 
764
 
static PySequenceMethods StaticTuple_as_sequence = {
765
 
    (lenfunc)StaticTuple_length,            /* sq_length */
766
 
    0,                              /* sq_concat */
767
 
    0,                              /* sq_repeat */
768
 
    (ssizeargfunc)StaticTuple_item,         /* sq_item */
769
 
    (ssizessizeargfunc)StaticTuple_slice,   /* sq_slice */
770
 
    0,                              /* sq_ass_item */
771
 
    0,                              /* sq_ass_slice */
772
 
    0,                              /* sq_contains */
773
 
};
774
 
 
775
 
/* TODO: Implement StaticTuple_as_mapping.
776
 
 *       The only thing we really want to support from there is mp_subscript,
777
 
 *       so that we could support extended slicing (foo[::2]). Not worth it
778
 
 *       yet, though.
779
 
 */
780
 
 
781
 
 
782
 
PyTypeObject StaticTuple_Type = {
783
 
    PyObject_HEAD_INIT(NULL)
784
 
    0,                                           /* ob_size */
785
 
    "bzrlib._static_tuple_c.StaticTuple",        /* tp_name */
786
 
    sizeof(StaticTuple),                         /* tp_basicsize */
787
 
    sizeof(PyObject *),                          /* tp_itemsize */
788
 
    (destructor)StaticTuple_dealloc,             /* tp_dealloc */
789
 
    0,                                           /* tp_print */
790
 
    0,                                           /* tp_getattr */
791
 
    0,                                           /* tp_setattr */
792
 
    0,                                           /* tp_compare */
793
 
    (reprfunc)StaticTuple_repr,                  /* tp_repr */
794
 
    &StaticTuple_as_number,                      /* tp_as_number */
795
 
    &StaticTuple_as_sequence,                    /* tp_as_sequence */
796
 
    0,                                           /* tp_as_mapping */
797
 
    (hashfunc)StaticTuple_hash,                  /* tp_hash */
798
 
    0,                                           /* tp_call */
799
 
    0,                                           /* tp_str */
800
 
    0,                                           /* tp_getattro */
801
 
    0,                                           /* tp_setattro */
802
 
    0,                                           /* tp_as_buffer */
803
 
    /* Py_TPFLAGS_CHECKTYPES tells the number operations that they shouldn't
804
 
     * try to 'coerce' but instead stuff like 'add' will check it arguments.
805
 
     */
806
 
    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES,  /* tp_flags*/
807
 
    StaticTuple_doc,                             /* tp_doc */
808
 
    /* gc.get_referents checks the IS_GC flag before it calls tp_traverse
809
 
     * And we don't include this object in the garbage collector because we
810
 
     * know it doesn't create cycles. However, 'meliae' will follow
811
 
     * tp_traverse, even if the object isn't GC, and we want that.
812
 
     */
813
 
    (traverseproc)StaticTuple_traverse,          /* tp_traverse */
814
 
    0,                                           /* tp_clear */
815
 
    StaticTuple_richcompare,                     /* tp_richcompare */
816
 
    0,                                           /* tp_weaklistoffset */
817
 
    // without implementing tp_iter, Python will fall back to PySequence*
818
 
    // which seems to work ok, we may need something faster/lighter in the
819
 
    // future.
820
 
    0,                                           /* tp_iter */
821
 
    0,                                           /* tp_iternext */
822
 
    StaticTuple_methods,                         /* tp_methods */
823
 
    0,                                           /* tp_members */
824
 
    0,                                           /* tp_getset */
825
 
    0,                                           /* tp_base */
826
 
    0,                                           /* tp_dict */
827
 
    0,                                           /* tp_descr_get */
828
 
    0,                                           /* tp_descr_set */
829
 
    0,                                           /* tp_dictoffset */
830
 
    0,                                           /* tp_init */
831
 
    0,                                           /* tp_alloc */
832
 
    StaticTuple_new_constructor,                 /* tp_new */
833
 
};
834
 
 
835
 
 
836
 
static PyMethodDef static_tuple_c_methods[] = {
837
 
    {NULL, NULL}
838
 
};
839
 
 
840
 
 
841
 
static void
842
 
setup_interned_tuples(PyObject *m)
843
 
{
844
 
    _interned_tuples = (PyObject *)SimpleSet_New();
845
 
    if (_interned_tuples != NULL) {
846
 
        Py_INCREF(_interned_tuples);
847
 
        PyModule_AddObject(m, "_interned_tuples", _interned_tuples);
848
 
    }
849
 
}
850
 
 
851
 
 
852
 
static void
853
 
setup_empty_tuple(PyObject *m)
854
 
{
855
 
    StaticTuple *stuple;
856
 
    if (_interned_tuples == NULL) {
857
 
        fprintf(stderr, "You need to call setup_interned_tuples() before"
858
 
                " setup_empty_tuple, because we intern it.\n");
859
 
    }
860
 
    // We need to create the empty tuple
861
 
    stuple = (StaticTuple *)StaticTuple_New(0);
862
 
    _empty_tuple = StaticTuple_Intern(stuple);
863
 
    assert(_empty_tuple == stuple);
864
 
    // At this point, refcnt is 2: 1 from New(), and 1 from the return from
865
 
    // intern(). We will keep 1 for the _empty_tuple global, and use the other
866
 
    // for the module reference.
867
 
    PyModule_AddObject(m, "_empty_tuple", (PyObject *)_empty_tuple);
868
 
}
869
 
 
870
 
static int
871
 
_StaticTuple_CheckExact(PyObject *obj)
872
 
{
873
 
    return StaticTuple_CheckExact(obj);
874
 
}
875
 
 
876
 
static void
877
 
setup_c_api(PyObject *m)
878
 
{
879
 
    _export_function(m, "StaticTuple_New", StaticTuple_New,
880
 
        "StaticTuple *(Py_ssize_t)");
881
 
    _export_function(m, "StaticTuple_Intern", StaticTuple_Intern,
882
 
        "StaticTuple *(StaticTuple *)");
883
 
    _export_function(m, "StaticTuple_FromSequence", StaticTuple_FromSequence,
884
 
        "StaticTuple *(PyObject *)");
885
 
    _export_function(m, "_StaticTuple_CheckExact", _StaticTuple_CheckExact,
886
 
        "int(PyObject *)");
887
 
}
888
 
 
889
 
 
890
 
static int
891
 
_workaround_pyrex_096(void)
892
 
{
893
 
    /* Work around an incompatibility in how pyrex 0.9.6 exports a module,
894
 
     * versus how pyrex 0.9.8 and cython 0.11 export it.
895
 
     * Namely 0.9.6 exports import__simple_set_pyx and tries to
896
 
     * "import _simple_set_pyx" but it is available only as
897
 
     * "import bzrlib._simple_set_pyx"
898
 
     * It is a shame to hack up sys.modules, but that is what we've got to do.
899
 
     */
900
 
    PyObject *sys_module = NULL, *modules = NULL, *set_module = NULL;
901
 
    int retval = -1;
902
 
 
903
 
    /* Clear out the current ImportError exception, and try again. */
904
 
    PyErr_Clear();
905
 
    /* Note that this only seems to work if somewhere else imports
906
 
     * bzrlib._simple_set_pyx before importing bzrlib._static_tuple_c
907
 
     */
908
 
    set_module = PyImport_ImportModule("bzrlib._simple_set_pyx");
909
 
    if (set_module == NULL) {
910
 
        goto end;
911
 
    }
912
 
    /* Add the _simple_set_pyx into sys.modules at the appropriate location. */
913
 
    sys_module = PyImport_ImportModule("sys");
914
 
    if (sys_module == NULL) {
915
 
        goto end;
916
 
    }
917
 
    modules = PyObject_GetAttrString(sys_module, "modules");
918
 
    if (modules == NULL || !PyDict_Check(modules)) {
919
 
        goto end;
920
 
    }
921
 
    PyDict_SetItemString(modules, "_simple_set_pyx", set_module);
922
 
    /* Now that we have hacked it in, try the import again. */
923
 
    retval = import_bzrlib___simple_set_pyx();
924
 
end:
925
 
    Py_XDECREF(set_module);
926
 
    Py_XDECREF(sys_module);
927
 
    Py_XDECREF(modules);
928
 
    return retval;
929
 
}
930
 
 
931
 
 
932
 
PyMODINIT_FUNC
933
 
init_static_tuple_c(void)
934
 
{
935
 
    PyObject* m;
936
 
 
937
 
    StaticTuple_Type.tp_getattro = PyObject_GenericGetAttr;
938
 
    if (PyType_Ready(&StaticTuple_Type) < 0)
939
 
        return;
940
 
 
941
 
    m = Py_InitModule3("_static_tuple_c", static_tuple_c_methods,
942
 
                       "C implementation of a StaticTuple structure");
943
 
    if (m == NULL)
944
 
      return;
945
 
 
946
 
    Py_INCREF(&StaticTuple_Type);
947
 
    PyModule_AddObject(m, "StaticTuple", (PyObject *)&StaticTuple_Type);
948
 
    if (import_bzrlib___simple_set_pyx() == -1
949
 
        && _workaround_pyrex_096() == -1)
950
 
    {
951
 
        return;
952
 
    }
953
 
    setup_interned_tuples(m);
954
 
    setup_empty_tuple(m);
955
 
    setup_c_api(m);
956
 
}
957
 
 
958
 
// vim: tabstop=4 sw=4 expandtab