mirror of https://github.com/python/cpython.git
240 lines
5.2 KiB
C
240 lines
5.2 KiB
C
#include "Python.h"
|
|
#include "structmember.h"
|
|
|
|
/* Examples showing how to subtype the builtin list and dict types from C. */
|
|
|
|
/* spamlist -- a list subtype */
|
|
|
|
typedef struct {
|
|
PyListObject list;
|
|
int state;
|
|
} spamlistobject;
|
|
|
|
static PyObject *
|
|
spamlist_getstate(spamlistobject *self, PyObject *args)
|
|
{
|
|
if (!PyArg_ParseTuple(args, ":getstate"))
|
|
return NULL;
|
|
return PyInt_FromLong(self->state);
|
|
}
|
|
|
|
static PyObject *
|
|
spamlist_setstate(spamlistobject *self, PyObject *args)
|
|
{
|
|
int state;
|
|
|
|
if (!PyArg_ParseTuple(args, "i:setstate", &state))
|
|
return NULL;
|
|
self->state = state;
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyMethodDef spamlist_methods[] = {
|
|
{"getstate", (PyCFunction)spamlist_getstate, METH_VARARGS,
|
|
"getstate() -> state"},
|
|
{"setstate", (PyCFunction)spamlist_setstate, METH_VARARGS,
|
|
"setstate(state)"},
|
|
{NULL, NULL},
|
|
};
|
|
|
|
staticforward PyTypeObject spamlist_type;
|
|
|
|
static int
|
|
spamlist_init(spamlistobject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
if (PyList_Type.tp_init((PyObject *)self, args, kwds) < 0)
|
|
return -1;
|
|
self->state = 0;
|
|
return 0;
|
|
}
|
|
|
|
static PyTypeObject spamlist_type = {
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
|
0,
|
|
"spamlist",
|
|
sizeof(spamlistobject),
|
|
0,
|
|
0, /* tp_dealloc */
|
|
0, /* tp_print */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_compare */
|
|
0, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
0, /* tp_getattro */
|
|
0, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
|
|
0, /* tp_doc */
|
|
0, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
0, /* tp_iter */
|
|
0, /* tp_iternext */
|
|
spamlist_methods, /* tp_methods */
|
|
0, /* tp_members */
|
|
0, /* tp_getset */
|
|
&PyList_Type, /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
(initproc)spamlist_init, /* tp_init */
|
|
0, /* tp_alloc */
|
|
0, /* tp_new */
|
|
};
|
|
|
|
/* spamdict -- a dict subtype */
|
|
|
|
typedef struct {
|
|
PyDictObject dict;
|
|
int state;
|
|
} spamdictobject;
|
|
|
|
static PyObject *
|
|
spamdict_getstate(spamdictobject *self, PyObject *args)
|
|
{
|
|
if (!PyArg_ParseTuple(args, ":getstate"))
|
|
return NULL;
|
|
return PyInt_FromLong(self->state);
|
|
}
|
|
|
|
static PyObject *
|
|
spamdict_setstate(spamdictobject *self, PyObject *args)
|
|
{
|
|
int state;
|
|
|
|
if (!PyArg_ParseTuple(args, "i:setstate", &state))
|
|
return NULL;
|
|
self->state = state;
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyMethodDef spamdict_methods[] = {
|
|
{"getstate", (PyCFunction)spamdict_getstate, METH_VARARGS,
|
|
"getstate() -> state"},
|
|
{"setstate", (PyCFunction)spamdict_setstate, METH_VARARGS,
|
|
"setstate(state)"},
|
|
{NULL, NULL},
|
|
};
|
|
|
|
staticforward PyTypeObject spamdict_type;
|
|
|
|
static int
|
|
spamdict_init(spamdictobject *self, PyObject *args, PyObject *kwds)
|
|
{
|
|
if (PyDict_Type.tp_init((PyObject *)self, args, kwds) < 0)
|
|
return -1;
|
|
self->state = 0;
|
|
return 0;
|
|
}
|
|
|
|
static struct memberlist spamdict_members[] = {
|
|
{"state", T_INT, offsetof(spamdictobject, state), READONLY},
|
|
{0}
|
|
};
|
|
|
|
static PyTypeObject spamdict_type = {
|
|
PyObject_HEAD_INIT(&PyType_Type)
|
|
0,
|
|
"spamdict",
|
|
sizeof(spamdictobject),
|
|
0,
|
|
0, /* tp_dealloc */
|
|
0, /* tp_print */
|
|
0, /* tp_getattr */
|
|
0, /* tp_setattr */
|
|
0, /* tp_compare */
|
|
0, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_hash */
|
|
0, /* tp_call */
|
|
0, /* tp_str */
|
|
0, /* tp_getattro */
|
|
0, /* tp_setattro */
|
|
0, /* tp_as_buffer */
|
|
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /* tp_flags */
|
|
0, /* tp_doc */
|
|
0, /* tp_traverse */
|
|
0, /* tp_clear */
|
|
0, /* tp_richcompare */
|
|
0, /* tp_weaklistoffset */
|
|
0, /* tp_iter */
|
|
0, /* tp_iternext */
|
|
spamdict_methods, /* tp_methods */
|
|
spamdict_members, /* tp_members */
|
|
0, /* tp_getset */
|
|
&PyDict_Type, /* tp_base */
|
|
0, /* tp_dict */
|
|
0, /* tp_descr_get */
|
|
0, /* tp_descr_set */
|
|
0, /* tp_dictoffset */
|
|
(initproc)spamdict_init, /* tp_init */
|
|
0, /* tp_alloc */
|
|
0, /* tp_new */
|
|
};
|
|
|
|
PyObject *
|
|
spam_bench(PyObject *self, PyObject *args)
|
|
{
|
|
PyObject *obj, *name, *res;
|
|
int n = 1000;
|
|
time_t t0, t1;
|
|
|
|
if (!PyArg_ParseTuple(args, "OS|i", &obj, &name, &n))
|
|
return NULL;
|
|
t0 = clock();
|
|
while (--n >= 0) {
|
|
res = PyObject_GetAttr(obj, name);
|
|
if (res == NULL)
|
|
return NULL;
|
|
Py_DECREF(res);
|
|
}
|
|
t1 = clock();
|
|
return PyFloat_FromDouble((double)(t1-t0) / CLOCKS_PER_SEC);
|
|
}
|
|
|
|
static PyMethodDef xxsubtype_functions[] = {
|
|
{"bench", spam_bench, METH_VARARGS},
|
|
{NULL, NULL} /* sentinel */
|
|
};
|
|
|
|
DL_EXPORT(void)
|
|
initxxsubtype(void)
|
|
{
|
|
PyObject *m, *d;
|
|
|
|
m = Py_InitModule("xxsubtype", xxsubtype_functions);
|
|
if (m == NULL)
|
|
return;
|
|
|
|
if (PyType_Ready(&spamlist_type) < 0)
|
|
return;
|
|
if (PyType_Ready(&spamdict_type) < 0)
|
|
return;
|
|
|
|
d = PyModule_GetDict(m);
|
|
if (d == NULL)
|
|
return;
|
|
|
|
Py_INCREF(&spamlist_type);
|
|
if (PyDict_SetItemString(d, "spamlist",
|
|
(PyObject *) &spamlist_type) < 0)
|
|
return;
|
|
|
|
Py_INCREF(&spamdict_type);
|
|
if (PyDict_SetItemString(d, "spamdict",
|
|
(PyObject *) &spamdict_type) < 0)
|
|
return;
|
|
}
|