← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/_static_tuple_c.h

Turn completion assertions into separate methods.

Many common assertions used to be expressed as arguments to the complete
method.  This makes the checks more explicit, and the code easier to read.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/_static_tuple_c.h
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
 
#ifndef _STATIC_TUPLE_H_
19
 
#define _STATIC_TUPLE_H_
20
 
#include <Python.h>
21
 
#include <string.h>
22
 
 
23
 
#define STATIC_TUPLE_HAS_HASH 0
24
 
/* Caching the hash adds memory, but allows us to save a little time during
25
 
 * lookups. TIMEIT hash(key) shows it as
26
 
 *  0.108usec w/ hash
27
 
 *  0.160usec w/o hash
28
 
 * Note that the entries themselves are strings, which already cache their
29
 
 * hashes. So while there is a 1.5:1 difference in the time for hash(), it is
30
 
 * already a function which is quite fast. Probably the only reason we might
31
 
 * want to do so, is if we customized SimpleSet to the point that the item
32
 
 * pointers were exactly certain types, and then accessed table[i]->hash
33
 
 * directly. So far StaticTuple_hash() is fast enough to not warrant the memory
34
 
 * difference.
35
 
 */
36
 
 
37
 
/* This defines a single variable-width key.
38
 
 * It is basically the same as a tuple, but
39
 
 * 1) Lighter weight in memory
40
 
 * 2) Only supports strings or other static types (that don't reference other
41
 
 *    objects.)
42
 
 */
43
 
 
44
 
#define STATIC_TUPLE_INTERNED_FLAG 0x01
45
 
typedef struct {
46
 
    PyObject_HEAD
47
 
    // We could go with unsigned short here, and support 64k width tuples
48
 
    // without any memory impact, might be worthwhile
49
 
    unsigned char size;
50
 
    unsigned char flags;
51
 
    unsigned char _unused0;
52
 
    unsigned char _unused1;
53
 
    // Note that on 64-bit, we actually have 4-more unused bytes
54
 
    // because items will always be aligned to a 64-bit boundary
55
 
#if STATIC_TUPLE_HAS_HASH
56
 
    long hash;
57
 
#endif
58
 
    PyObject *items[0];
59
 
} StaticTuple;
60
 
extern PyTypeObject StaticTuple_Type;
61
 
 
62
 
typedef struct {
63
 
    PyObject_VAR_HEAD
64
 
    PyObject *table[0];
65
 
} KeyIntern;
66
 
 
67
 
#define StaticTuple_SET_ITEM(key, offset, val) \
68
 
    ((((StaticTuple*)(key))->items[(offset)]) = ((PyObject *)(val)))
69
 
#define StaticTuple_GET_ITEM(key, offset) (((StaticTuple*)key)->items[offset])
70
 
#define StaticTuple_GET_SIZE(key) (((StaticTuple*)key)->size)
71
 
 
72
 
 
73
 
#ifdef STATIC_TUPLE_MODULE
74
 
/* Used when compiling _static_tuple_c.c */
75
 
 
76
 
static StaticTuple * StaticTuple_New(Py_ssize_t);
77
 
static StaticTuple * StaticTuple_Intern(StaticTuple *self);
78
 
static StaticTuple * StaticTuple_FromSequence(PyObject *);
79
 
#define StaticTuple_CheckExact(op) (Py_TYPE(op) == &StaticTuple_Type)
80
 
 
81
 
#else
82
 
/* Used as the foreign api */
83
 
 
84
 
#include "_import_c_api.h"
85
 
 
86
 
static StaticTuple *(*StaticTuple_New)(Py_ssize_t);
87
 
static StaticTuple *(*StaticTuple_Intern)(StaticTuple *);
88
 
static StaticTuple *(*StaticTuple_FromSequence)(PyObject *);
89
 
static PyTypeObject *_p_StaticTuple_Type;
90
 
 
91
 
#define StaticTuple_CheckExact(op) (Py_TYPE(op) == _p_StaticTuple_Type)
92
 
static int (*_StaticTuple_CheckExact)(PyObject *);
93
 
 
94
 
 
95
 
/* Return -1 and set exception on error, 0 on success */
96
 
static int
97
 
import_static_tuple_c(void)
98
 
{
99
 
    struct function_description functions[] = {
100
 
        {"StaticTuple_New", (void **)&StaticTuple_New,
101
 
            "StaticTuple *(Py_ssize_t)"},
102
 
        {"StaticTuple_Intern", (void **)&StaticTuple_Intern,
103
 
            "StaticTuple *(StaticTuple *)"},
104
 
        {"StaticTuple_FromSequence", (void **)&StaticTuple_FromSequence,
105
 
            "StaticTuple *(PyObject *)"},
106
 
        {"_StaticTuple_CheckExact", (void **)&_StaticTuple_CheckExact,
107
 
            "int(PyObject *)"},
108
 
        {NULL}};
109
 
    struct type_description types[] = {
110
 
        {"StaticTuple", &_p_StaticTuple_Type},
111
 
        {NULL}};
112
 
    return _import_extension_module("bzrlib._static_tuple_c",
113
 
        functions, types);
114
 
}
115
 
 
116
 
#endif // !STATIC_TUPLE_MODULE
117
 
#endif // !_STATIC_TUPLE_H_