Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 1 | /* Cell object implementation */ |
| 2 | |
| 3 | #include "Python.h" |
| 4 | |
| 5 | PyObject * |
| 6 | PyCell_New(PyObject *obj) |
| 7 | { |
| 8 | PyCellObject *op; |
| 9 | |
Neil Schemenauer | e83c00e | 2001-08-29 23:54:21 +0000 | [diff] [blame] | 10 | op = (PyCellObject *)PyObject_GC_New(PyCellObject, &PyCell_Type); |
Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 11 | if (op == NULL) |
| 12 | return NULL; |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 13 | op->ob_ref = obj; |
| 14 | Py_XINCREF(obj); |
| 15 | |
Neil Schemenauer | e83c00e | 2001-08-29 23:54:21 +0000 | [diff] [blame] | 16 | _PyObject_GC_TRACK(op); |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 17 | return (PyObject *)op; |
| 18 | } |
| 19 | |
| 20 | PyObject * |
| 21 | PyCell_Get(PyObject *op) |
| 22 | { |
| 23 | if (!PyCell_Check(op)) { |
| 24 | PyErr_BadInternalCall(); |
| 25 | return NULL; |
| 26 | } |
| 27 | Py_XINCREF(((PyCellObject*)op)->ob_ref); |
| 28 | return PyCell_GET(op); |
| 29 | } |
| 30 | |
| 31 | int |
| 32 | PyCell_Set(PyObject *op, PyObject *obj) |
| 33 | { |
Christian Heimes | 18c6689 | 2008-02-17 13:31:39 +0000 | [diff] [blame] | 34 | PyObject* oldobj; |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 35 | if (!PyCell_Check(op)) { |
| 36 | PyErr_BadInternalCall(); |
| 37 | return -1; |
| 38 | } |
Christian Heimes | 18c6689 | 2008-02-17 13:31:39 +0000 | [diff] [blame] | 39 | oldobj = PyCell_GET(op); |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 40 | Py_XINCREF(obj); |
| 41 | PyCell_SET(op, obj); |
Christian Heimes | 18c6689 | 2008-02-17 13:31:39 +0000 | [diff] [blame] | 42 | Py_XDECREF(oldobj); |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 43 | return 0; |
| 44 | } |
| 45 | |
| 46 | static void |
| 47 | cell_dealloc(PyCellObject *op) |
| 48 | { |
Neil Schemenauer | e83c00e | 2001-08-29 23:54:21 +0000 | [diff] [blame] | 49 | _PyObject_GC_UNTRACK(op); |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 50 | Py_XDECREF(op->ob_ref); |
Neil Schemenauer | e83c00e | 2001-08-29 23:54:21 +0000 | [diff] [blame] | 51 | PyObject_GC_Del(op); |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 52 | } |
| 53 | |
Christian Heimes | d5e2b6f | 2008-03-19 21:50:51 +0000 | [diff] [blame] | 54 | static int |
| 55 | cell_compare(PyCellObject *a, PyCellObject *b) |
| 56 | { |
| 57 | if (a->ob_ref == NULL) { |
| 58 | if (b->ob_ref == NULL) |
| 59 | return 0; |
| 60 | return -1; |
| 61 | } else if (b->ob_ref == NULL) |
| 62 | return 1; |
| 63 | return PyObject_Compare(a->ob_ref, b->ob_ref); |
| 64 | } |
| 65 | |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 66 | static PyObject * |
| 67 | cell_repr(PyCellObject *op) |
| 68 | { |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 69 | if (op->ob_ref == NULL) |
Walter Dörwald | 1ab8330 | 2007-05-18 17:15:44 +0000 | [diff] [blame] | 70 | return PyUnicode_FromFormat("<cell at %p: empty>", op); |
Barry Warsaw | 7ce3694 | 2001-08-24 18:34:26 +0000 | [diff] [blame] | 71 | |
Walter Dörwald | 1ab8330 | 2007-05-18 17:15:44 +0000 | [diff] [blame] | 72 | return PyUnicode_FromFormat("<cell at %p: %.80s object at %p>", |
Barry Warsaw | 7ce3694 | 2001-08-24 18:34:26 +0000 | [diff] [blame] | 73 | op, op->ob_ref->ob_type->tp_name, |
| 74 | op->ob_ref); |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 75 | } |
| 76 | |
| 77 | static int |
| 78 | cell_traverse(PyCellObject *op, visitproc visit, void *arg) |
| 79 | { |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 80 | Py_VISIT(op->ob_ref); |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 81 | return 0; |
| 82 | } |
| 83 | |
| 84 | static int |
| 85 | cell_clear(PyCellObject *op) |
| 86 | { |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 87 | Py_CLEAR(op->ob_ref); |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 88 | return 0; |
| 89 | } |
| 90 | |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 91 | static PyObject * |
| 92 | cell_get_contents(PyCellObject *op, void *closure) |
| 93 | { |
Guido van Rossum | 0d3fb8a | 2007-11-26 23:23:18 +0000 | [diff] [blame] | 94 | if (op->ob_ref == NULL) |
| 95 | { |
| 96 | PyErr_SetString(PyExc_ValueError, "Cell is empty"); |
| 97 | return NULL; |
| 98 | } |
| 99 | Py_INCREF(op->ob_ref); |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 100 | return op->ob_ref; |
| 101 | } |
| 102 | |
| 103 | static PyGetSetDef cell_getsetlist[] = { |
| 104 | {"cell_contents", (getter)cell_get_contents, NULL}, |
| 105 | {NULL} /* sentinel */ |
| 106 | }; |
| 107 | |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 108 | PyTypeObject PyCell_Type = { |
Martin v. Löwis | 9f2e346 | 2007-07-21 17:22:18 +0000 | [diff] [blame] | 109 | PyVarObject_HEAD_INIT(&PyType_Type, 0) |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 110 | "cell", |
Neil Schemenauer | e83c00e | 2001-08-29 23:54:21 +0000 | [diff] [blame] | 111 | sizeof(PyCellObject), |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 112 | 0, |
| 113 | (destructor)cell_dealloc, /* tp_dealloc */ |
| 114 | 0, /* tp_print */ |
| 115 | 0, /* tp_getattr */ |
| 116 | 0, /* tp_setattr */ |
Christian Heimes | d5e2b6f | 2008-03-19 21:50:51 +0000 | [diff] [blame] | 117 | (cmpfunc)cell_compare, /* tp_compare */ |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 118 | (reprfunc)cell_repr, /* tp_repr */ |
| 119 | 0, /* tp_as_number */ |
| 120 | 0, /* tp_as_sequence */ |
| 121 | 0, /* tp_as_mapping */ |
| 122 | 0, /* tp_hash */ |
| 123 | 0, /* tp_call */ |
| 124 | 0, /* tp_str */ |
Tim Peters | 6d6c1a3 | 2001-08-02 04:15:00 +0000 | [diff] [blame] | 125 | PyObject_GenericGetAttr, /* tp_getattro */ |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 126 | 0, /* tp_setattro */ |
| 127 | 0, /* tp_as_buffer */ |
Neil Schemenauer | e83c00e | 2001-08-29 23:54:21 +0000 | [diff] [blame] | 128 | Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */ |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 129 | 0, /* tp_doc */ |
| 130 | (traverseproc)cell_traverse, /* tp_traverse */ |
| 131 | (inquiry)cell_clear, /* tp_clear */ |
Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 132 | 0, /* tp_richcompare */ |
| 133 | 0, /* tp_weaklistoffset */ |
| 134 | 0, /* tp_iter */ |
| 135 | 0, /* tp_iternext */ |
| 136 | 0, /* tp_methods */ |
| 137 | 0, /* tp_members */ |
| 138 | cell_getsetlist, /* tp_getset */ |
Jeremy Hylton | fbd849f | 2001-01-25 20:04:14 +0000 | [diff] [blame] | 139 | }; |