cpython/Objects/classobject.c

/* Class object implementation (dead now except for methods) */

#include "Python.h"
#include "structmember.h"

#define TP_DESCR_GET(t) ((t)->tp_descr_get)

/* Free list for method objects to safe malloc/free overhead
 * The im_self element is used to chain the elements.
 */
static PyMethodObject *free_list;
static int numfree = 0;
#ifndef PyMethod_MAXFREELIST
#define PyMethod_MAXFREELIST 256
#endif

PyObject *
PyMethod_Function(PyObject *im)
{
	if (!PyMethod_Check(im)) {
		PyErr_BadInternalCall();
		return NULL;
	}
	return ((PyMethodObject *)im)->im_func;
}

PyObject *
PyMethod_Self(PyObject *im)
{
	if (!PyMethod_Check(im)) {
		PyErr_BadInternalCall();
		return NULL;
	}
	return ((PyMethodObject *)im)->im_self;
}

/* Method objects are used for bound instance methods returned by
   instancename.methodname. ClassName.methodname returns an ordinary
   function.
*/

PyObject *
PyMethod_New(PyObject *func, PyObject *self)
{
	register PyMethodObject *im;
	if (!PyCallable_Check(func)) {
		PyErr_BadInternalCall();
		return NULL;
	}
	if (self == NULL) {
		PyErr_BadInternalCall();
		return NULL;
	}
	im = free_list;
	if (im != NULL) {
		free_list = (PyMethodObject *)(im->im_self);
		PyObject_INIT(im, &PyMethod_Type);
		numfree--;
	}
	else {
		im = PyObject_GC_New(PyMethodObject, &PyMethod_Type);
		if (im == NULL)
			return NULL;
	}
	im->im_weakreflist = NULL;
	Py_INCREF(func);
	im->im_func = func;
	Py_XINCREF(self);
	im->im_self = self;
	_PyObject_GC_TRACK(im);
	return (PyObject *)im;
}

/* Descriptors for PyMethod attributes */

/* im_func and im_self are stored in the PyMethod object */

#define MO_OFF(x) offsetof(PyMethodObject, x)

static PyMemberDef method_memberlist[] = {
	{"__func__", T_OBJECT, MO_OFF(im_func), READONLY|RESTRICTED,
	 "the function (or other callable) implementing a method"},
	{"__self__", T_OBJECT, MO_OFF(im_self), READONLY|RESTRICTED,
	 "the instance to which a method is bound"},
	{NULL}	/* Sentinel */
};

/* Christian Tismer argued convincingly that method attributes should
   (nearly) always override function attributes.
   The one exception is __doc__; there's a default __doc__ which
   should only be used for the class, not for instances */

static PyObject *
method_get_doc(PyMethodObject *im, void *context)
{
	static PyObject *docstr;
	if (docstr == NULL) {
		docstr= PyUnicode_InternFromString("__doc__");
		if (docstr == NULL)
			return NULL;
	}
	return PyObject_GetAttr(im->im_func, docstr);
}

static PyGetSetDef method_getset[] = {
	{"__doc__", (getter)method_get_doc, NULL, NULL},
	{0}
};

static PyObject *
method_getattro(PyObject *obj, PyObject *name)
{
	PyMethodObject *im = (PyMethodObject *)obj;
	PyTypeObject *tp = obj->ob_type;
	PyObject *descr = NULL;

	{
		if (tp->tp_dict == NULL) {
			if (PyType_Ready(tp) < 0)
				return NULL;
		}
		descr = _PyType_Lookup(tp, name);
	}

	if (descr != NULL) {
		descrgetfunc f = TP_DESCR_GET(descr->ob_type);
		if (f != NULL)
			return f(descr, obj, (PyObject *)obj->ob_type);
		else {
			Py_INCREF(descr);
			return descr;
		}
	}

	return PyObject_GetAttr(im->im_func, name);
}

PyDoc_STRVAR(method_doc,
"method(function, instance)\n\
\n\
Create a bound instance method object.");

static PyObject *
method_new(PyTypeObject* type, PyObject* args, PyObject *kw)
{
	PyObject *func;
	PyObject *self;

	if (!_PyArg_NoKeywords("method", kw))
		return NULL;
	if (!PyArg_UnpackTuple(args, "method", 2, 2,
			      &func, &self))
		return NULL;
	if (!PyCallable_Check(func)) {
		PyErr_SetString(PyExc_TypeError,
				"first argument must be callable");
		return NULL;
	}
	if (self == NULL || self == Py_None) {
		PyErr_SetString(PyExc_TypeError,
			"self must not be None");
		return NULL;
	}

	return PyMethod_New(func, self);
}

static void
method_dealloc(register PyMethodObject *im)
{
	_PyObject_GC_UNTRACK(im);
	if (im->im_weakreflist != NULL)
		PyObject_ClearWeakRefs((PyObject *)im);
	Py_DECREF(im->im_func);
	Py_XDECREF(im->im_self);
	if (numfree < PyMethod_MAXFREELIST) {
		im->im_self = (PyObject *)free_list;
		free_list = im;
		numfree++;
	}
	else {
		PyObject_GC_Del(im);
	}
}

static PyObject *
method_richcompare(PyObject *self, PyObject *other, int op)
{
	PyMethodObject *a, *b;
	PyObject *res;
	int eq;

	if ((op != Py_EQ && op != Py_NE) ||
	    !PyMethod_Check(self) ||
	    !PyMethod_Check(other))
	{
		Py_INCREF(Py_NotImplemented);
		return Py_NotImplemented;
	}
	a = (PyMethodObject *)self;
	b = (PyMethodObject *)other;
	eq = PyObject_RichCompareBool(a->im_func, b->im_func, Py_EQ);
	if (eq == 1) {
		if (a->im_self == NULL || b->im_self == NULL)
			eq = a->im_self == b->im_self;
		else
			eq = PyObject_RichCompareBool(a->im_self, b->im_self,
						      Py_EQ);
	}
	if (eq < 0)
		return NULL;
	if (op == Py_EQ)
		res = eq ? Py_True : Py_False;
	else
		res = eq ? Py_False : Py_True;
	Py_INCREF(res);
	return res;
}

static PyObject *
method_repr(PyMethodObject *a)
{
	PyObject *self = a->im_self;
	PyObject *func = a->im_func;
	PyObject *klass = (PyObject*)Py_TYPE(self);
	PyObject *funcname = NULL ,*klassname = NULL, *result = NULL;
	char *defname = "?";

	if (self == NULL) {
		PyErr_BadInternalCall();
		return NULL;
	}

	funcname = PyObject_GetAttrString(func, "__name__");
	if (funcname == NULL) {
		if (!PyErr_ExceptionMatches(PyExc_AttributeError))
			return NULL;
		PyErr_Clear();
	}
	else if (!PyUnicode_Check(funcname)) {
		Py_DECREF(funcname);
		funcname = NULL;
	}

	if (klass == NULL)
		klassname = NULL;
	else {
		klassname = PyObject_GetAttrString(klass, "__name__");
		if (klassname == NULL) {
			if (!PyErr_ExceptionMatches(PyExc_AttributeError))
				return NULL;
			PyErr_Clear();
		}
		else if (!PyUnicode_Check(klassname)) {
			Py_DECREF(klassname);
			klassname = NULL;
		}
	}

	/* XXX Shouldn't use repr()/%R here! */
	result = PyUnicode_FromFormat("<bound method %V.%V of %R>",
	                              klassname, defname,
	                              funcname, defname, self);

	Py_XDECREF(funcname);
	Py_XDECREF(klassname);
	return result;
}

static long
method_hash(PyMethodObject *a)
{
	long x, y;
	if (a->im_self == NULL)
		x = PyObject_Hash(Py_None);
	else
		x = PyObject_Hash(a->im_self);
	if (x == -1)
		return -1;
	y = PyObject_Hash(a->im_func);
	if (y == -1)
		return -1;
	x = x ^ y;
	if (x == -1)
		x = -2;
	return x;
}

static int
method_traverse(PyMethodObject *im, visitproc visit, void *arg)
{
	Py_VISIT(im->im_func);
	Py_VISIT(im->im_self);
	return 0;
}

static PyObject *
method_call(PyObject *func, PyObject *arg, PyObject *kw)
{
	PyObject *self = PyMethod_GET_SELF(func);
	PyObject *result;

	func = PyMethod_GET_FUNCTION(func);
	if (self == NULL) {
		PyErr_BadInternalCall();
		return NULL;
	}
	else {
		Py_ssize_t argcount = PyTuple_Size(arg);
		PyObject *newarg = PyTuple_New(argcount + 1);
		int i;
		if (newarg == NULL)
			return NULL;
		Py_INCREF(self);
		PyTuple_SET_ITEM(newarg, 0, self);
		for (i = 0; i < argcount; i++) {
			PyObject *v = PyTuple_GET_ITEM(arg, i);
			Py_XINCREF(v);
			PyTuple_SET_ITEM(newarg, i+1, v);
		}
		arg = newarg;
	}
	result = PyObject_Call((PyObject *)func, arg, kw);
	Py_DECREF(arg);
	return result;
}

static PyObject *
method_descr_get(PyObject *meth, PyObject *obj, PyObject *cls)
{
	/* Don't rebind an already bound method of a class that's not a base
	   class of cls. */
	if (PyMethod_GET_SELF(meth) != NULL) {
		/* Already bound */
		Py_INCREF(meth);
		return meth;
	}
	/* Bind it to obj */
	return PyMethod_New(PyMethod_GET_FUNCTION(meth), obj);
}

PyTypeObject PyMethod_Type = {
	PyVarObject_HEAD_INIT(&PyType_Type, 0)
	"method",
	sizeof(PyMethodObject),
	0,
	(destructor)method_dealloc,		/* tp_dealloc */
	0,					/* tp_print */
	0,					/* tp_getattr */
	0,					/* tp_setattr */
	0,					/* tp_compare */
	(reprfunc)method_repr,			/* tp_repr */
	0,					/* tp_as_number */
	0,					/* tp_as_sequence */
	0,					/* tp_as_mapping */
	(hashfunc)method_hash,			/* tp_hash */
	method_call,				/* tp_call */
	0,					/* tp_str */
	method_getattro,			/* tp_getattro */
	PyObject_GenericSetAttr,		/* tp_setattro */
	0,					/* tp_as_buffer */
	Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /* tp_flags */
	method_doc,				/* tp_doc */
	(traverseproc)method_traverse,		/* tp_traverse */
	0,					/* tp_clear */
	method_richcompare,			/* tp_richcompare */
	offsetof(PyMethodObject, im_weakreflist), /* tp_weaklistoffset */
	0,					/* tp_iter */
	0,					/* tp_iternext */
	0,					/* tp_methods */
	method_memberlist,			/* tp_members */
	method_getset,				/* tp_getset */
	0,					/* tp_base */
	0,					/* tp_dict */
	method_descr_get,			/* tp_descr_get */
	0,					/* tp_descr_set */
	0,					/* tp_dictoffset */
	0,					/* tp_init */
	0,					/* tp_alloc */
	method_new,				/* tp_new */
};

/* Clear out the free list */

int
PyMethod_ClearFreeList(void)
{
	int freelist_size = numfree;
	
	while (free_list) {
		PyMethodObject *im = free_list;
		free_list = (PyMethodObject *)(im->im_self);
		PyObject_GC_Del(im);
		numfree--;
	}
	assert(numfree == 0);
	return freelist_size;
}

void
PyMethod_Fini(void)
{
	(void)PyMethod_ClearFreeList();
}

/* ------------------------------------------------------------------------
 * instance method
 */

PyObject *
PyInstanceMethod_New(PyObject *func) {
	PyInstanceMethodObject *method;
	method = PyObject_GC_New(PyInstanceMethodObject,
				 &PyInstanceMethod_Type);
	if (method == NULL) return NULL;
	Py_INCREF(func);
	method->func = func;
	_PyObject_GC_TRACK(method);
	return (PyObject *)method;
}

PyObject *
PyInstanceMethod_Function(PyObject *im)
{
	if (!PyInstanceMethod_Check(im)) {
		PyErr_BadInternalCall();
		return NULL;
	}
	return PyInstanceMethod_GET_FUNCTION(im);
}

#define IMO_OFF(x) offsetof(PyInstanceMethodObject, x)

static PyMemberDef instancemethod_memberlist[] = {
	{"__func__", T_OBJECT, IMO_OFF(func), READONLY|RESTRICTED,
	 "the function (or other callable) implementing a method"},
	{NULL}	/* Sentinel */
};

static PyObject *
instancemethod_get_doc(PyObject *self, void *context)
{
	static PyObject *docstr;
	if (docstr == NULL) {
		docstr = PyUnicode_InternFromString("__doc__");
		if (docstr == NULL)
			return NULL;
	}
	return PyObject_GetAttr(PyInstanceMethod_GET_FUNCTION(self), docstr);
}

static PyGetSetDef instancemethod_getset[] = {
	{"__doc__", (getter)instancemethod_get_doc, NULL, NULL},
	{0}
};

static PyObject *
instancemethod_getattro(PyObject *self, PyObject *name)
{
	PyTypeObject *tp = self->ob_type;
	PyObject *descr = NULL;

	if (tp->tp_dict == NULL) {
		if (PyType_Ready(tp) < 0)
			return NULL;
	}
	descr = _PyType_Lookup(tp, name);

	if (descr != NULL) {
		descrgetfunc f = TP_DESCR_GET(descr->ob_type);
		if (f != NULL)
			return f(descr, self, (PyObject *)self->ob_type);
		else {
			Py_INCREF(descr);
			return descr;
		}
	}

	return PyObject_GetAttr(PyInstanceMethod_GET_FUNCTION(self), name);
}

static void
instancemethod_dealloc(PyObject *self) {
	_PyObject_GC_UNTRACK(self);
	Py_DECREF(PyInstanceMethod_GET_FUNCTION(self));
	PyObject_GC_Del(self);
}

static int
instancemethod_traverse(PyObject *self, visitproc visit, void *arg) {
	Py_VISIT(PyInstanceMethod_GET_FUNCTION(self));
	return 0;
}

static PyObject *
instancemethod_call(PyObject *self, PyObject *arg, PyObject *kw)
{
	return PyObject_Call(PyMethod_GET_FUNCTION(self), arg, kw);
}

static PyObject *
instancemethod_descr_get(PyObject *descr, PyObject *obj, PyObject *type) {
	register PyObject *func = PyInstanceMethod_GET_FUNCTION(descr);
	if (obj == NULL)
		return func;
	else
		return PyMethod_New(func, obj);
}

static PyObject *
instancemethod_richcompare(PyObject *self, PyObject *other, int op)
{
	PyInstanceMethodObject *a, *b;
	PyObject *res;
	int eq;

	if ((op != Py_EQ && op != Py_NE) ||
	    !PyInstanceMethod_Check(self) ||
	    !PyInstanceMethod_Check(other))
	{
		Py_INCREF(Py_NotImplemented);
		return Py_NotImplemented;
	}
	a = (PyInstanceMethodObject *)self;
	b = (PyInstanceMethodObject *)other;
	eq = PyObject_RichCompareBool(a->func, b->func, Py_EQ);
	if (eq < 0)
		return NULL;
	if (op == Py_EQ)
		res = eq ? Py_True : Py_False;
	else
		res = eq ? Py_False : Py_True;
	Py_INCREF(res);
	return res;
}

static PyObject *
instancemethod_repr(PyObject *self)
{
	PyObject *func = PyInstanceMethod_Function(self);
	PyObject *funcname = NULL , *result = NULL;
	char *defname = "?";

	if (func == NULL) {
		PyErr_BadInternalCall();
		return NULL;
	}

	funcname = PyObject_GetAttrString(func, "__name__");
	if (funcname == NULL) {
		if (!PyErr_ExceptionMatches(PyExc_AttributeError))
			return NULL;
		PyErr_Clear();
	}
	else if (!PyUnicode_Check(funcname)) {
		Py_DECREF(funcname);
		funcname = NULL;
	}

	result = PyUnicode_FromFormat("<instancemethod %V at %p>",
	                              funcname, defname, self);

	Py_XDECREF(funcname);
	return result;
}

/*
static long
instancemethod_hash(PyObject *self)
{
	long x, y;
	x = (long)self;
	y = PyObject_Hash(PyInstanceMethod_GET_FUNCTION(self));
	if (y == -1)
		return -1;
	x = x ^ y;
	if (x == -1)
		x = -2;
	return x;
}
*/

PyDoc_STRVAR(instancemethod_doc,
"instancemethod(function)\n\
\n\
Bind a function to a class.");

static PyObject *
instancemethod_new(PyTypeObject* type, PyObject* args, PyObject *kw)
{
	PyObject *func;

	if (!_PyArg_NoKeywords("instancemethod", kw))
		return NULL;
	if (!PyArg_UnpackTuple(args, "instancemethod", 1, 1, &func))
		return NULL;
	if (!PyCallable_Check(func)) {
		PyErr_SetString(PyExc_TypeError,
				"first argument must be callable");
		return NULL;
	}

	return PyInstanceMethod_New(func);
}

PyTypeObject PyInstanceMethod_Type = {
	PyVarObject_HEAD_INIT(&PyType_Type, 0)
	"instancemethod",			/* tp_name */
	sizeof(PyInstanceMethodObject),		/* tp_basicsize */
	0,					/* tp_itemsize */
	instancemethod_dealloc,			/* tp_dealloc */
	0,					/* tp_print */
	0,					/* tp_getattr */
	0,					/* tp_setattr */
	0,					/* tp_compare */
	(reprfunc)instancemethod_repr,		/* tp_repr */
	0,					/* tp_as_number */
	0,					/* tp_as_sequence */
	0,					/* tp_as_mapping */
	0, /*(hashfunc)instancemethod_hash,	tp_hash  */
	instancemethod_call,			/* tp_call */
	0,					/* tp_str */
	instancemethod_getattro,		/* tp_getattro */
	PyObject_GenericSetAttr,		/* tp_setattro */
	0,					/* tp_as_buffer */
	Py_TPFLAGS_DEFAULT
		| Py_TPFLAGS_HAVE_GC,		/* tp_flags */
	instancemethod_doc,			/* tp_doc */
	instancemethod_traverse,		/* tp_traverse */
	0,					/* tp_clear */
	instancemethod_richcompare,		/* tp_richcompare */
	0,					/* tp_weaklistoffset */
	0,					/* tp_iter */
	0,					/* tp_iternext */
	0,					/* tp_methods */
	instancemethod_memberlist,		/* tp_members */
	instancemethod_getset,			/* tp_getset */
	0,					/* tp_base */
	0,					/* tp_dict */
	instancemethod_descr_get,		/* tp_descr_get */
	0,					/* tp_descr_set */
	0,					/* tp_dictoffset */
	0,					/* tp_init */
	0,					/* tp_alloc */
	instancemethod_new,			/* tp_new */
};