2011-02-23 11:52:22 +01:00
|
|
|
/*
|
2010-03-21 02:14:04 +01:00
|
|
|
* $Id$
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
2008-04-17 00:40:48 +02:00
|
|
|
* ***** BEGIN GPL LICENSE BLOCK *****
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU General Public License
|
|
|
|
* as published by the Free Software Foundation; either version 2
|
2008-04-17 00:40:48 +02:00
|
|
|
* of the License, or (at your option) any later version.
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program; if not, write to the Free Software Foundation,
|
2010-02-12 14:34:04 +01:00
|
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
|
|
|
*
|
2009-11-16 19:56:58 +01:00
|
|
|
* Contributor(s): Joseph Eagar, Campbell Barton
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
2008-04-17 00:40:48 +02:00
|
|
|
* ***** END GPL LICENSE BLOCK *****
|
2006-11-30 00:31:46 +01:00
|
|
|
*/
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2011-02-14 05:15:25 +01:00
|
|
|
#include <Python.h>
|
|
|
|
|
2010-12-06 00:50:55 +01:00
|
|
|
#include "IDProp.h"
|
2011-01-07 19:36:47 +01:00
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
|
|
|
|
#include "BLI_string.h"
|
|
|
|
#include "BLI_utildefines.h"
|
2010-12-06 00:50:55 +01:00
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
#include "BKE_idprop.h"
|
2011-01-07 20:18:31 +01:00
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
|
2010-09-01 16:13:48 +02:00
|
|
|
#define USE_STRING_COERCE
|
|
|
|
|
|
|
|
#ifdef USE_STRING_COERCE
|
|
|
|
#include "py_capi_utils.h"
|
|
|
|
#endif
|
|
|
|
|
|
|
|
PyObject * PyC_UnicodeFromByte(const char *str);
|
2010-10-04 01:29:43 +02:00
|
|
|
const char * PyC_UnicodeAsByte(PyObject *py_str, PyObject **coerce); /* coerce must be NULL */
|
2010-09-01 16:13:48 +02:00
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
/*** Function to wrap ID properties ***/
|
2006-11-19 17:00:04 +01:00
|
|
|
PyObject *BPy_Wrap_IDProperty(ID *id, IDProperty *prop, IDProperty *parent);
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
extern PyTypeObject IDArray_Type;
|
2006-12-17 00:54:45 +01:00
|
|
|
extern PyTypeObject IDGroup_Iter_Type;
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
/*********************** ID Property Main Wrapper Stuff ***************/
|
|
|
|
|
2011-02-13 11:52:18 +01:00
|
|
|
static PyObject *IDGroup_repr( BPy_IDProperty *self )
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyUnicode_FromFormat( "<bpy ID property from \"%s\">", self->id->name);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
extern PyTypeObject IDGroup_Type;
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
PyObject *BPy_IDGroup_WrapData( ID *id, IDProperty *prop )
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2006-12-17 00:54:45 +01:00
|
|
|
switch ( prop->type ) {
|
|
|
|
case IDP_STRING:
|
2010-09-01 16:13:48 +02:00
|
|
|
#ifdef USE_STRING_COERCE
|
|
|
|
return PyC_UnicodeFromByte(prop->data.pointer);
|
|
|
|
#else
|
|
|
|
return PyUnicode_FromString(prop->data.pointer);
|
|
|
|
#endif
|
2006-12-17 00:54:45 +01:00
|
|
|
case IDP_INT:
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyLong_FromLong( (long)prop->data.val );
|
2006-12-17 00:54:45 +01:00
|
|
|
case IDP_FLOAT:
|
2007-05-26 06:39:31 +02:00
|
|
|
return PyFloat_FromDouble( (double)(*(float*)(&prop->data.val)) );
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
|
|
|
return PyFloat_FromDouble( (*(double*)(&prop->data.val)) );
|
2006-12-17 00:54:45 +01:00
|
|
|
case IDP_GROUP:
|
|
|
|
/*blegh*/
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2006-12-17 00:54:45 +01:00
|
|
|
BPy_IDProperty *group = PyObject_New(BPy_IDProperty, &IDGroup_Type);
|
|
|
|
group->id = id;
|
|
|
|
group->prop = prop;
|
|
|
|
return (PyObject*) group;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
case IDP_ARRAY:
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2006-12-17 00:54:45 +01:00
|
|
|
BPy_IDProperty *array = PyObject_New(BPy_IDProperty, &IDArray_Type);
|
|
|
|
array->id = id;
|
|
|
|
array->prop = prop;
|
|
|
|
return (PyObject*) array;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
case IDP_IDPARRAY: /* this could be better a internal type */
|
|
|
|
{
|
|
|
|
PyObject *seq = PyList_New(prop->len), *wrap;
|
|
|
|
IDProperty *array= IDP_IDPArray(prop);
|
|
|
|
int i;
|
|
|
|
|
|
|
|
if (!seq) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_Format(PyExc_RuntimeError, "BPy_IDGroup_MapDataToPy, IDP_IDPARRAY: PyList_New(%d) failed", prop->len);
|
2010-03-02 12:35:28 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
for (i=0; i<prop->len; i++) {
|
|
|
|
wrap= BPy_IDGroup_WrapData(id, array++);
|
|
|
|
|
|
|
|
if (!wrap) /* BPy_IDGroup_MapDataToPy sets the error */
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
PyList_SET_ITEM(seq, i, wrap);
|
|
|
|
}
|
|
|
|
|
|
|
|
return seq;
|
|
|
|
}
|
|
|
|
/* case IDP_IDPARRAY: TODO */
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
2011-02-25 03:27:54 +01:00
|
|
|
#if 0 /* UNUSED, currenly assignment overwrites into new properties, rather then setting in-place */
|
2011-02-13 11:52:18 +01:00
|
|
|
static int BPy_IDGroup_SetData(BPy_IDProperty *self, IDProperty *prop, PyObject *value)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
switch (prop->type) {
|
|
|
|
case IDP_STRING:
|
|
|
|
{
|
|
|
|
char *st;
|
2009-11-16 19:56:58 +01:00
|
|
|
if (!PyUnicode_Check(value)) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a string!");
|
|
|
|
return -1;
|
|
|
|
}
|
2010-09-01 16:13:48 +02:00
|
|
|
#ifdef USE_STRING_COERCE
|
|
|
|
{
|
|
|
|
int alloc_len;
|
|
|
|
PyObject *value_coerce= NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2010-10-04 01:29:43 +02:00
|
|
|
st= (char *)PyC_UnicodeAsByte(value, &value_coerce);
|
2010-09-01 16:13:48 +02:00
|
|
|
alloc_len= strlen(st) + 1;
|
|
|
|
|
|
|
|
st = _PyUnicode_AsString(value);
|
|
|
|
IDP_ResizeArray(prop, alloc_len);
|
|
|
|
memcpy(prop->data.pointer, st, alloc_len);
|
|
|
|
Py_XDECREF(value_coerce);
|
|
|
|
}
|
|
|
|
#else
|
2009-11-16 19:56:58 +01:00
|
|
|
st = _PyUnicode_AsString(value);
|
2006-12-17 00:54:45 +01:00
|
|
|
IDP_ResizeArray(prop, strlen(st)+1);
|
|
|
|
strcpy(prop->data.pointer, st);
|
2010-09-01 16:13:48 +02:00
|
|
|
#endif
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
case IDP_INT:
|
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
int ivalue= PyLong_AsSsize_t(value);
|
|
|
|
if (ivalue==-1 && PyErr_Occurred()) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected an int type");
|
|
|
|
return -1;
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
prop->data.val = ivalue;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case IDP_FLOAT:
|
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
float fvalue= (float)PyFloat_AsDouble(value);
|
|
|
|
if (fvalue==-1 && PyErr_Occurred()) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a float");
|
|
|
|
return -1;
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
*(float*)&self->prop->data.val = fvalue;
|
|
|
|
break;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
double dvalue= PyFloat_AsDouble(value);
|
|
|
|
if (dvalue==-1 && PyErr_Occurred()) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a float");
|
|
|
|
return -1;
|
|
|
|
}
|
2008-07-24 21:22:17 +02:00
|
|
|
*(double*)&self->prop->data.val = dvalue;
|
|
|
|
break;
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
default:
|
2009-11-16 19:56:58 +01:00
|
|
|
PyErr_SetString(PyExc_AttributeError, "attempt to set read-only attribute!");
|
|
|
|
return -1;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
return 0;
|
|
|
|
}
|
2011-02-25 03:27:54 +01:00
|
|
|
#endif
|
2006-11-17 05:46:48 +01:00
|
|
|
|
2011-02-13 11:52:18 +01:00
|
|
|
static PyObject *BPy_IDGroup_GetName(BPy_IDProperty *self, void *UNUSED(closure))
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyUnicode_FromString(self->prop->name);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2010-10-14 01:25:08 +02:00
|
|
|
static int BPy_IDGroup_SetName(BPy_IDProperty *self, PyObject *value, void *UNUSED(closure))
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
char *st;
|
2009-11-16 19:56:58 +01:00
|
|
|
if (!PyUnicode_Check(value)) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a string!");
|
|
|
|
return -1;
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
st = _PyUnicode_AsString(value);
|
2010-12-06 00:50:55 +01:00
|
|
|
if (BLI_strnlen(st, MAX_IDPROP_NAME) == MAX_IDPROP_NAME) {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "string length cannot exceed 31 characters!");
|
|
|
|
return -1;
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2011-02-13 04:21:27 +01:00
|
|
|
BLI_strncpy(self->prop->name, st, sizeof(self->prop->name));
|
2006-12-17 00:54:45 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
#if 0
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_GetType(BPy_IDProperty *self)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyLong_FromSsize_t(self->prop->type);
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
#endif
|
2006-12-17 00:54:45 +01:00
|
|
|
|
|
|
|
static PyGetSetDef BPy_IDGroup_getseters[] = {
|
2010-12-03 18:05:21 +01:00
|
|
|
{(char *)"name", (getter)BPy_IDGroup_GetName, (setter)BPy_IDGroup_SetName, (char *)"The name of this Group.", NULL},
|
2006-12-17 00:54:45 +01:00
|
|
|
{NULL, NULL, NULL, NULL, NULL}
|
|
|
|
};
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
static Py_ssize_t BPy_IDGroup_Map_Len(BPy_IDProperty *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
if (self->prop->type != IDP_GROUP) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "len() of unsized object");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
return self->prop->len;
|
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_Map_GetItem(BPy_IDProperty *self, PyObject *item)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
IDProperty *idprop;
|
|
|
|
char *name;
|
|
|
|
|
|
|
|
if (self->prop->type != IDP_GROUP) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "unsubscriptable object");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
name= _PyUnicode_AsString(item);
|
|
|
|
|
|
|
|
if (name == NULL) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "only strings are allowed as keys of ID properties");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
idprop= IDP_GetPropertyFromGroup(self->prop, name);
|
|
|
|
|
|
|
|
if(idprop==NULL) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_KeyError, "key not in subgroup dict");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
return BPy_IDGroup_WrapData(self->id, idprop);
|
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/*returns NULL on success, error string on failure*/
|
2010-03-02 12:35:28 +01:00
|
|
|
static int idp_sequence_type(PyObject *seq)
|
|
|
|
{
|
|
|
|
PyObject *item;
|
|
|
|
int type= IDP_INT;
|
|
|
|
|
2011-01-09 15:53:18 +01:00
|
|
|
int i, len = PySequence_Size(seq);
|
2010-03-02 12:35:28 +01:00
|
|
|
for (i=0; i < len; i++) {
|
|
|
|
item = PySequence_GetItem(seq, i);
|
|
|
|
if (PyFloat_Check(item)) {
|
|
|
|
if(type == IDP_IDPARRAY) { /* mixed dict/int */
|
|
|
|
Py_DECREF(item);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
type= IDP_DOUBLE;
|
|
|
|
}
|
|
|
|
else if (PyLong_Check(item)) {
|
|
|
|
if(type == IDP_IDPARRAY) { /* mixed dict/int */
|
|
|
|
Py_DECREF(item);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
2010-03-02 15:57:28 +01:00
|
|
|
else if (PyMapping_Check(item)) {
|
2010-03-02 12:35:28 +01:00
|
|
|
if(i != 0 && (type != IDP_IDPARRAY)) { /* mixed dict/int */
|
|
|
|
Py_DECREF(item);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
type= IDP_IDPARRAY;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
Py_XDECREF(item);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_DECREF(item);
|
|
|
|
}
|
|
|
|
|
|
|
|
return type;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* note: group can be a pointer array or a group */
|
2010-11-19 03:14:18 +01:00
|
|
|
const char *BPy_IDProperty_Map_ValidateAndCreate(const char *name, IDProperty *group, PyObject *ob)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2007-05-22 06:41:21 +02:00
|
|
|
IDProperty *prop = NULL;
|
2007-03-01 22:30:48 +01:00
|
|
|
IDPropertyTemplate val = {0};
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
if(strlen(name) >= sizeof(group->name))
|
|
|
|
return "the length of IDProperty names is limited to 31 characters";
|
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
if (PyFloat_Check(ob)) {
|
2008-07-24 21:22:17 +02:00
|
|
|
val.d = PyFloat_AsDouble(ob);
|
|
|
|
prop = IDP_New(IDP_DOUBLE, val, name);
|
2009-11-16 19:56:58 +01:00
|
|
|
} else if (PyLong_Check(ob)) {
|
|
|
|
val.i = (int) PyLong_AsSsize_t(ob);
|
2006-11-17 05:46:48 +01:00
|
|
|
prop = IDP_New(IDP_INT, val, name);
|
2009-11-16 19:56:58 +01:00
|
|
|
} else if (PyUnicode_Check(ob)) {
|
2010-09-01 16:13:48 +02:00
|
|
|
#ifdef USE_STRING_COERCE
|
|
|
|
PyObject *value_coerce= NULL;
|
2010-10-04 01:29:43 +02:00
|
|
|
val.str = (char *)PyC_UnicodeAsByte(ob, &value_coerce);
|
2010-09-01 16:13:48 +02:00
|
|
|
prop = IDP_New(IDP_STRING, val, name);
|
|
|
|
Py_XDECREF(value_coerce);
|
|
|
|
#else
|
2009-11-16 19:56:58 +01:00
|
|
|
val.str = _PyUnicode_AsString(ob);
|
2006-11-17 05:46:48 +01:00
|
|
|
prop = IDP_New(IDP_STRING, val, name);
|
2010-09-01 16:13:48 +02:00
|
|
|
#endif
|
2006-11-17 05:46:48 +01:00
|
|
|
} else if (PySequence_Check(ob)) {
|
|
|
|
PyObject *item;
|
|
|
|
int i;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-03-02 12:35:28 +01:00
|
|
|
if((val.array.type= idp_sequence_type(ob)) == -1)
|
|
|
|
return "only floats, ints and dicts are allowed in ID property arrays";
|
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
/*validate sequence and derive type.
|
|
|
|
we assume IDP_INT unless we hit a float
|
|
|
|
number; then we assume it's */
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2011-01-09 15:53:18 +01:00
|
|
|
val.array.len = PySequence_Size(ob);
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-03-02 12:35:28 +01:00
|
|
|
switch(val.array.type) {
|
|
|
|
case IDP_DOUBLE:
|
|
|
|
prop = IDP_New(IDP_ARRAY, val, name);
|
|
|
|
for (i=0; i<val.array.len; i++) {
|
|
|
|
item = PySequence_GetItem(ob, i);
|
2008-07-24 21:22:17 +02:00
|
|
|
((double*)prop->data.pointer)[i] = (float)PyFloat_AsDouble(item);
|
2010-03-02 12:35:28 +01:00
|
|
|
Py_DECREF(item);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case IDP_INT:
|
|
|
|
prop = IDP_New(IDP_ARRAY, val, name);
|
|
|
|
for (i=0; i<val.array.len; i++) {
|
|
|
|
item = PySequence_GetItem(ob, i);
|
|
|
|
((int*)prop->data.pointer)[i] = (int)PyLong_AsSsize_t(item);
|
|
|
|
Py_DECREF(item);
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
case IDP_IDPARRAY:
|
|
|
|
prop= IDP_NewIDPArray(name);
|
|
|
|
for (i=0; i<val.array.len; i++) {
|
2010-11-19 03:14:18 +01:00
|
|
|
const char *error;
|
2010-03-02 12:35:28 +01:00
|
|
|
item = PySequence_GetItem(ob, i);
|
|
|
|
error= BPy_IDProperty_Map_ValidateAndCreate("", prop, item);
|
|
|
|
Py_DECREF(item);
|
|
|
|
|
|
|
|
if(error)
|
|
|
|
return error;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
break;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
} else if (PyMapping_Check(ob)) {
|
|
|
|
PyObject *keys, *vals, *key, *pval;
|
|
|
|
int i, len;
|
|
|
|
/*yay! we get into recursive stuff now!*/
|
|
|
|
keys = PyMapping_Keys(ob);
|
|
|
|
vals = PyMapping_Values(ob);
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
/*we allocate the group first; if we hit any invalid data,
|
|
|
|
we can delete it easily enough.*/
|
|
|
|
prop = IDP_New(IDP_GROUP, val, name);
|
|
|
|
len = PyMapping_Length(ob);
|
|
|
|
for (i=0; i<len; i++) {
|
|
|
|
key = PySequence_GetItem(keys, i);
|
|
|
|
pval = PySequence_GetItem(vals, i);
|
2009-11-16 19:56:58 +01:00
|
|
|
if (!PyUnicode_Check(key)) {
|
2006-11-17 05:46:48 +01:00
|
|
|
IDP_FreeProperty(prop);
|
|
|
|
MEM_freeN(prop);
|
2006-12-17 00:54:45 +01:00
|
|
|
Py_XDECREF(keys);
|
|
|
|
Py_XDECREF(vals);
|
2006-11-17 05:46:48 +01:00
|
|
|
Py_XDECREF(key);
|
|
|
|
Py_XDECREF(pval);
|
|
|
|
return "invalid element in subgroup dict template!";
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
if (BPy_IDProperty_Map_ValidateAndCreate(_PyUnicode_AsString(key), prop, pval)) {
|
2006-11-17 05:46:48 +01:00
|
|
|
IDP_FreeProperty(prop);
|
|
|
|
MEM_freeN(prop);
|
2006-12-17 00:54:45 +01:00
|
|
|
Py_XDECREF(keys);
|
|
|
|
Py_XDECREF(vals);
|
2006-11-17 05:46:48 +01:00
|
|
|
Py_XDECREF(key);
|
|
|
|
Py_XDECREF(pval);
|
|
|
|
return "invalid element in subgroup dict template!";
|
|
|
|
}
|
|
|
|
Py_XDECREF(key);
|
|
|
|
Py_XDECREF(pval);
|
|
|
|
}
|
|
|
|
Py_XDECREF(keys);
|
|
|
|
Py_XDECREF(vals);
|
2006-12-01 04:04:36 +01:00
|
|
|
} else return "invalid property value";
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-03-02 12:35:28 +01:00
|
|
|
if(group->type==IDP_IDPARRAY) {
|
|
|
|
IDP_AppendArray(group, prop);
|
|
|
|
// IDP_FreeProperty(item); // IDP_AppendArray does a shallow copy (memcpy), only free memory
|
|
|
|
MEM_freeN(prop);
|
|
|
|
} else {
|
|
|
|
IDP_ReplaceInGroup(group, prop);
|
|
|
|
}
|
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2009-11-16 23:21:39 +01:00
|
|
|
int BPy_Wrap_SetMapItem(IDProperty *prop, PyObject *key, PyObject *val)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 23:21:39 +01:00
|
|
|
if (prop->type != IDP_GROUP) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "unsubscriptable object");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2009-11-16 23:21:39 +01:00
|
|
|
if (val == NULL) { /* del idprop[key] */
|
|
|
|
IDProperty *pkey = IDP_GetPropertyFromGroup(prop, _PyUnicode_AsString(key));
|
2006-12-01 04:04:36 +01:00
|
|
|
if (pkey) {
|
2009-11-16 23:21:39 +01:00
|
|
|
IDP_RemFromGroup(prop, pkey);
|
2006-12-01 04:04:36 +01:00
|
|
|
IDP_FreeProperty(pkey);
|
|
|
|
MEM_freeN(pkey);
|
|
|
|
return 0;
|
2009-11-16 19:56:58 +01:00
|
|
|
} else {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_KeyError, "property not found in group");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2006-12-01 04:04:36 +01:00
|
|
|
}
|
2009-11-16 23:21:39 +01:00
|
|
|
else {
|
2010-11-19 03:14:18 +01:00
|
|
|
const char *err;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 23:21:39 +01:00
|
|
|
if (!PyUnicode_Check(key)) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "only strings are allowed as subgroup keys");
|
2009-11-16 23:21:39 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
err = BPy_IDProperty_Map_ValidateAndCreate(_PyUnicode_AsString(key), prop, val);
|
|
|
|
if (err) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_KeyError, err );
|
2009-11-16 23:21:39 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
2009-11-16 19:56:58 +01:00
|
|
|
}
|
2009-11-16 23:21:39 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 23:21:39 +01:00
|
|
|
static int BPy_IDGroup_Map_SetItem(BPy_IDProperty *self, PyObject *key, PyObject *val)
|
|
|
|
{
|
2009-11-17 13:54:29 +01:00
|
|
|
return BPy_Wrap_SetMapItem(self->prop, key, val);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_SpawnIterator(BPy_IDProperty *self)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
BPy_IDGroup_Iter *iter = PyObject_New(BPy_IDGroup_Iter, &IDGroup_Iter_Type);
|
|
|
|
iter->group = self;
|
|
|
|
iter->mode = IDPROP_ITER_KEYS;
|
|
|
|
iter->cur = self->prop->data.group.first;
|
|
|
|
Py_XINCREF(iter);
|
|
|
|
return (PyObject*) iter;
|
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_MapDataToPy(IDProperty *prop)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
switch (prop->type) {
|
|
|
|
case IDP_STRING:
|
2010-09-01 16:13:48 +02:00
|
|
|
#ifdef USE_STRING_COERCE
|
|
|
|
return PyC_UnicodeFromByte(prop->data.pointer);
|
|
|
|
#else
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyUnicode_FromString(prop->data.pointer);
|
2010-09-01 16:13:48 +02:00
|
|
|
#endif
|
2006-12-17 00:54:45 +01:00
|
|
|
break;
|
|
|
|
case IDP_FLOAT:
|
2007-05-26 06:39:31 +02:00
|
|
|
return PyFloat_FromDouble(*((float*)&prop->data.val));
|
2006-12-17 00:54:45 +01:00
|
|
|
break;
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
|
|
|
return PyFloat_FromDouble(*((double*)&prop->data.val));
|
|
|
|
break;
|
2006-12-17 00:54:45 +01:00
|
|
|
case IDP_INT:
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyLong_FromSsize_t( prop->data.val );
|
2006-12-17 00:54:45 +01:00
|
|
|
break;
|
|
|
|
case IDP_ARRAY:
|
|
|
|
{
|
|
|
|
PyObject *seq = PyList_New(prop->len);
|
|
|
|
int i;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
if (!seq) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_Format(PyExc_RuntimeError, "BPy_IDGroup_MapDataToPy, IDP_ARRAY: PyList_New(%d) failed", prop->len);
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
for (i=0; i<prop->len; i++) {
|
2008-07-24 21:22:17 +02:00
|
|
|
if (prop->subtype == IDP_FLOAT) {
|
2009-11-16 21:16:45 +01:00
|
|
|
PyList_SET_ITEM(seq, i,
|
2007-05-26 06:39:31 +02:00
|
|
|
PyFloat_FromDouble(((float*)prop->data.pointer)[i]));
|
2008-07-24 21:22:17 +02:00
|
|
|
} else if (prop->subtype == IDP_DOUBLE) {
|
2009-11-16 21:16:45 +01:00
|
|
|
PyList_SET_ITEM(seq, i,
|
2009-11-16 19:56:58 +01:00
|
|
|
PyFloat_FromDouble(((double*)prop->data.pointer)[i]));
|
2009-11-16 21:16:45 +01:00
|
|
|
} else {
|
|
|
|
PyList_SET_ITEM(seq, i,
|
2009-11-16 19:56:58 +01:00
|
|
|
PyLong_FromLong(((int*)prop->data.pointer)[i]));
|
2008-07-24 21:22:17 +02:00
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
return seq;
|
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
case IDP_IDPARRAY:
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2010-03-02 12:35:28 +01:00
|
|
|
PyObject *seq = PyList_New(prop->len), *wrap;
|
|
|
|
IDProperty *array= IDP_IDPArray(prop);
|
|
|
|
int i;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-03-02 12:35:28 +01:00
|
|
|
if (!seq) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_Format(PyExc_RuntimeError, "BPy_IDGroup_MapDataToPy, IDP_IDPARRAY: PyList_New(%d) failed", prop->len);
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2010-03-02 12:35:28 +01:00
|
|
|
for (i=0; i<prop->len; i++) {
|
|
|
|
wrap= BPy_IDGroup_MapDataToPy(array++);
|
|
|
|
|
|
|
|
if (!wrap) /* BPy_IDGroup_MapDataToPy sets the error */
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
PyList_SET_ITEM(seq, i, wrap);
|
|
|
|
}
|
|
|
|
return seq;
|
|
|
|
}
|
|
|
|
case IDP_GROUP:
|
|
|
|
{
|
|
|
|
PyObject *dict = PyDict_New(), *wrap;
|
|
|
|
IDProperty *loop;
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
for (loop=prop->data.group.first; loop; loop=loop->next) {
|
|
|
|
wrap = BPy_IDGroup_MapDataToPy(loop);
|
2010-03-02 12:35:28 +01:00
|
|
|
|
|
|
|
if (!wrap) /* BPy_IDGroup_MapDataToPy sets the error */
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
PyDict_SetItemString(dict, loop->name, wrap);
|
|
|
|
}
|
|
|
|
return dict;
|
|
|
|
}
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-03-02 12:35:28 +01:00
|
|
|
PyErr_Format(PyExc_RuntimeError, "eek!! '%s' property exists with a bad type code '%d' !!!", prop->name, prop->type);
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_Pop(BPy_IDProperty *self, PyObject *value)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
IDProperty *idprop;
|
2006-12-17 00:54:45 +01:00
|
|
|
PyObject *pyform;
|
2009-11-16 19:56:58 +01:00
|
|
|
char *name = _PyUnicode_AsString(value);
|
|
|
|
|
2007-06-16 14:24:41 +02:00
|
|
|
if (!name) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "pop expected at least 1 argument, got 0");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
idprop= IDP_GetPropertyFromGroup(self->prop, name);
|
|
|
|
|
|
|
|
if(idprop) {
|
|
|
|
pyform = BPy_IDGroup_MapDataToPy(idprop);
|
|
|
|
|
|
|
|
if (!pyform) {
|
|
|
|
/*ok something bad happened with the pyobject,
|
|
|
|
so don't remove the prop from the group. if pyform is
|
|
|
|
NULL, then it already should have raised an exception.*/
|
|
|
|
return NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
IDP_RemFromGroup(self->prop, idprop);
|
|
|
|
return pyform;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_KeyError, "item not in group");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_IterItems(BPy_IDProperty *self)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
BPy_IDGroup_Iter *iter = PyObject_New(BPy_IDGroup_Iter, &IDGroup_Iter_Type);
|
|
|
|
iter->group = self;
|
|
|
|
iter->mode = IDPROP_ITER_ITEMS;
|
|
|
|
iter->cur = self->prop->data.group.first;
|
|
|
|
Py_XINCREF(iter);
|
|
|
|
return (PyObject*) iter;
|
|
|
|
}
|
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
/* utility function */
|
2009-11-17 13:54:29 +01:00
|
|
|
static void BPy_IDGroup_CorrectListLen(IDProperty *prop, PyObject *seq, int len)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-19 12:56:13 +01:00
|
|
|
int j;
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
printf("ID Property Error found and corrected in BPy_IDGroup_GetKeys/Values/Items!\n");
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
/*fill rest of list with valid references to None*/
|
2009-11-19 12:56:13 +01:00
|
|
|
for (j=len; j<prop->len; j++) {
|
2009-11-16 21:16:45 +01:00
|
|
|
Py_INCREF(Py_None);
|
|
|
|
PyList_SET_ITEM(seq, j, Py_None);
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
/*set correct group length*/
|
2009-11-19 12:56:13 +01:00
|
|
|
prop->len = len;
|
2009-11-16 21:16:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
PyObject *BPy_Wrap_GetKeys(IDProperty *prop)
|
|
|
|
{
|
|
|
|
PyObject *seq = PyList_New(prop->len);
|
|
|
|
IDProperty *loop;
|
|
|
|
int i;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-02-01 19:38:33 +01:00
|
|
|
for (i=0, loop=prop->data.group.first; loop && (i < prop->len); loop=loop->next, i++)
|
2009-11-16 21:16:45 +01:00
|
|
|
PyList_SET_ITEM(seq, i, PyUnicode_FromString(loop->name));
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-02-01 19:38:33 +01:00
|
|
|
/* if the id prop is corrupt, count the remaining */
|
|
|
|
for (; loop; loop=loop->next, i++) {}
|
|
|
|
|
|
|
|
if (i != prop->len) { /* if the loop didnt finish, we know the length is wrong */
|
2009-11-16 21:16:45 +01:00
|
|
|
BPy_IDGroup_CorrectListLen(prop, seq, i);
|
|
|
|
Py_DECREF(seq); /*free the list*/
|
2008-02-23 03:12:50 +01:00
|
|
|
/*call self again*/
|
2009-11-16 21:16:45 +01:00
|
|
|
return BPy_Wrap_GetKeys(prop);
|
2008-02-23 03:12:50 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
return seq;
|
|
|
|
}
|
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
PyObject *BPy_Wrap_GetValues(ID *id, IDProperty *prop)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-16 21:16:45 +01:00
|
|
|
PyObject *seq = PyList_New(prop->len);
|
2006-12-17 00:54:45 +01:00
|
|
|
IDProperty *loop;
|
2009-11-16 21:16:45 +01:00
|
|
|
int i;
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
for (i=0, loop=prop->data.group.first; loop; loop=loop->next, i++) {
|
|
|
|
PyList_SET_ITEM(seq, i, BPy_IDGroup_WrapData(id, loop));
|
2009-11-16 19:56:58 +01:00
|
|
|
}
|
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
if (i != prop->len) {
|
|
|
|
BPy_IDGroup_CorrectListLen(prop, seq, i);
|
|
|
|
Py_DECREF(seq); /*free the list*/
|
|
|
|
/*call self again*/
|
|
|
|
return BPy_Wrap_GetValues(id, prop);
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2008-02-23 03:12:50 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
return seq;
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
PyObject *BPy_Wrap_GetItems(ID *id, IDProperty *prop)
|
|
|
|
{
|
|
|
|
PyObject *seq = PyList_New(prop->len);
|
|
|
|
IDProperty *loop;
|
|
|
|
int i;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
for (i=0, loop=prop->data.group.first; loop; loop=loop->next, i++) {
|
|
|
|
PyObject *item= PyTuple_New(2);
|
|
|
|
PyTuple_SET_ITEM(item, 0, PyUnicode_FromString(loop->name));
|
|
|
|
PyTuple_SET_ITEM(item, 1, BPy_IDGroup_WrapData(id, loop));
|
|
|
|
PyList_SET_ITEM(seq, i, item);
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
if (i != prop->len) {
|
|
|
|
BPy_IDGroup_CorrectListLen(prop, seq, i);
|
|
|
|
Py_DECREF(seq); /*free the list*/
|
2008-02-23 03:12:50 +01:00
|
|
|
/*call self again*/
|
2009-11-16 21:16:45 +01:00
|
|
|
return BPy_Wrap_GetItems(id, prop);
|
2008-02-23 03:12:50 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
return seq;
|
|
|
|
}
|
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
|
|
|
|
static PyObject *BPy_IDGroup_GetKeys(BPy_IDProperty *self)
|
|
|
|
{
|
|
|
|
return BPy_Wrap_GetKeys(self->prop);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *BPy_IDGroup_GetValues(BPy_IDProperty *self)
|
|
|
|
{
|
|
|
|
return BPy_Wrap_GetValues(self->id, self->prop);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *BPy_IDGroup_GetItems(BPy_IDProperty *self)
|
|
|
|
{
|
|
|
|
return BPy_Wrap_GetItems(self->id, self->prop);
|
|
|
|
}
|
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
static int BPy_IDGroup_Contains(BPy_IDProperty *self, PyObject *value)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
char *name = _PyUnicode_AsString(value);
|
|
|
|
|
|
|
|
if (!name) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a string");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
return IDP_GetPropertyFromGroup(self->prop, name) ? 1:0;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2008-12-20 09:41:46 +01:00
|
|
|
static PyObject *BPy_IDGroup_Update(BPy_IDProperty *self, PyObject *value)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2008-12-20 09:41:46 +01:00
|
|
|
PyObject *pkey, *pval;
|
2008-04-17 23:14:55 +02:00
|
|
|
Py_ssize_t i=0;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
if (!PyDict_Check(value)) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "expected an object derived from dict");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2008-12-20 09:41:46 +01:00
|
|
|
while (PyDict_Next(value, &i, &pkey, &pval)) {
|
2006-12-17 00:54:45 +01:00
|
|
|
BPy_IDGroup_Map_SetItem(self, pkey, pval);
|
|
|
|
if (PyErr_Occurred()) return NULL;
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_ConvertToPy(BPy_IDProperty *self)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
return BPy_IDGroup_MapDataToPy(self->prop);
|
|
|
|
}
|
|
|
|
|
2009-11-19 18:04:28 +01:00
|
|
|
|
|
|
|
/* Matches python dict.get(key, [default]) */
|
2011-02-13 11:52:18 +01:00
|
|
|
static PyObject* BPy_IDGroup_Get(BPy_IDProperty *self, PyObject *args)
|
2009-11-19 18:04:28 +01:00
|
|
|
{
|
|
|
|
IDProperty *idprop;
|
|
|
|
char *key;
|
|
|
|
PyObject* def = Py_None;
|
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "s|O:get", &key, &def))
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
idprop= IDP_GetPropertyFromGroup(self->prop, key);
|
|
|
|
if (idprop) {
|
|
|
|
PyObject* pyobj = BPy_IDGroup_WrapData(self->id, idprop);
|
|
|
|
if (pyobj)
|
|
|
|
return pyobj;
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_INCREF(def);
|
|
|
|
return def;
|
|
|
|
}
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
static struct PyMethodDef BPy_IDGroup_methods[] = {
|
2007-06-16 14:24:41 +02:00
|
|
|
{"pop", (PyCFunction)BPy_IDGroup_Pop, METH_O,
|
2010-02-11 03:03:18 +01:00
|
|
|
"pop an item from the group; raises KeyError if the item doesn't exist"},
|
2006-12-17 00:54:45 +01:00
|
|
|
{"iteritems", (PyCFunction)BPy_IDGroup_IterItems, METH_NOARGS,
|
2010-02-11 03:03:18 +01:00
|
|
|
"iterate through the items in the dict; behaves like dictionary method iteritems"},
|
2006-12-17 00:54:45 +01:00
|
|
|
{"keys", (PyCFunction)BPy_IDGroup_GetKeys, METH_NOARGS,
|
2010-02-11 03:03:18 +01:00
|
|
|
"get the keys associated with this group as a list of strings"},
|
2006-12-17 00:54:45 +01:00
|
|
|
{"values", (PyCFunction)BPy_IDGroup_GetValues, METH_NOARGS,
|
2010-02-11 03:03:18 +01:00
|
|
|
"get the values associated with this group"},
|
2009-11-16 21:16:45 +01:00
|
|
|
{"items", (PyCFunction)BPy_IDGroup_GetItems, METH_NOARGS,
|
2010-02-11 03:03:18 +01:00
|
|
|
"get the items associated with this group"},
|
2008-12-20 09:41:46 +01:00
|
|
|
{"update", (PyCFunction)BPy_IDGroup_Update, METH_O,
|
2010-02-11 03:03:18 +01:00
|
|
|
"updates the values in the group with the values of another or a dict"},
|
2009-11-19 18:04:28 +01:00
|
|
|
{"get", (PyCFunction)BPy_IDGroup_Get, METH_VARARGS,
|
2010-02-11 03:03:18 +01:00
|
|
|
"idprop.get(k[,d]) -> idprop[k] if k in idprop, else d. d defaults to None"},
|
2006-12-17 00:54:45 +01:00
|
|
|
{"convert_to_pyobject", (PyCFunction)BPy_IDGroup_ConvertToPy, METH_NOARGS,
|
2010-02-11 03:03:18 +01:00
|
|
|
"return a purely python version of the group"},
|
2011-02-13 11:52:18 +01:00
|
|
|
{NULL, NULL, 0, NULL}
|
2006-12-17 00:54:45 +01:00
|
|
|
};
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
static PySequenceMethods BPy_IDGroup_Seq = {
|
2011-02-13 11:52:18 +01:00
|
|
|
(lenfunc) BPy_IDGroup_Map_Len, /* lenfunc sq_length */
|
|
|
|
NULL, /* binaryfunc sq_concat */
|
|
|
|
NULL, /* ssizeargfunc sq_repeat */
|
|
|
|
NULL, /* ssizeargfunc sq_item */ /* TODO - setting this will allow PySequence_Check to return True */
|
|
|
|
NULL, /* intintargfunc ***was_sq_slice*** */
|
|
|
|
NULL, /* intobjargproc sq_ass_item */
|
|
|
|
NULL, /* ssizeobjargproc ***was_sq_ass_slice*** */
|
2009-11-16 19:56:58 +01:00
|
|
|
(objobjproc) BPy_IDGroup_Contains, /* objobjproc sq_contains */
|
2011-02-13 11:52:18 +01:00
|
|
|
NULL, /* binaryfunc sq_inplace_concat */
|
|
|
|
NULL, /* ssizeargfunc sq_inplace_repeat */
|
2009-11-16 19:56:58 +01:00
|
|
|
};
|
|
|
|
|
2011-02-13 11:52:18 +01:00
|
|
|
static PyMappingMethods BPy_IDGroup_Mapping = {
|
2009-11-16 19:56:58 +01:00
|
|
|
(lenfunc)BPy_IDGroup_Map_Len, /*inquiry mp_length */
|
2006-12-17 00:54:45 +01:00
|
|
|
(binaryfunc)BPy_IDGroup_Map_GetItem, /*binaryfunc mp_subscript */
|
|
|
|
(objobjargproc)BPy_IDGroup_Map_SetItem, /*objobjargproc mp_ass_subscript */
|
2006-11-17 05:46:48 +01:00
|
|
|
};
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
PyTypeObject IDGroup_Type = {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
2006-11-17 05:46:48 +01:00
|
|
|
/* For printing, in format "<module>.<name>" */
|
|
|
|
"Blender IDProperty", /* char *tp_name; */
|
|
|
|
sizeof( BPy_IDProperty ), /* int tp_basicsize; */
|
|
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
|
2007-03-15 02:47:53 +01:00
|
|
|
NULL, /* destructor tp_dealloc; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* printfunc tp_print; */
|
2006-12-17 00:54:45 +01:00
|
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
|
|
NULL, /* setattrfunc tp_setattr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* cmpfunc tp_compare; */
|
2006-12-17 00:54:45 +01:00
|
|
|
( reprfunc ) IDGroup_repr, /* reprfunc tp_repr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
2009-11-16 19:56:58 +01:00
|
|
|
&BPy_IDGroup_Seq, /* PySequenceMethods *tp_as_sequence; */
|
|
|
|
&BPy_IDGroup_Mapping, /* PyMappingMethods *tp_as_mapping; */
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
|
|
|
|
NULL, /* hashfunc tp_hash; */
|
|
|
|
NULL, /* ternaryfunc tp_call; */
|
|
|
|
NULL, /* reprfunc tp_str; */
|
|
|
|
NULL, /* getattrofunc tp_getattro; */
|
|
|
|
NULL, /* setattrofunc tp_setattro; */
|
|
|
|
|
|
|
|
/* Functions to access object as input/output buffer */
|
|
|
|
NULL, /* PyBufferProcs *tp_as_buffer; */
|
|
|
|
|
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
|
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
|
|
|
/*** Assigned meaning in release 2.0 ***/
|
|
|
|
/* call function for all accessible objects */
|
|
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
|
|
|
|
/* delete references to contained objects */
|
|
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
|
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
|
|
/*** rich comparisons ***/
|
|
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
|
|
|
|
/*** weak reference enabler ***/
|
|
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
|
|
|
|
/*** Added in release 2.2 ***/
|
|
|
|
/* Iterators */
|
2006-12-17 00:54:45 +01:00
|
|
|
(getiterfunc)BPy_IDGroup_SpawnIterator, /* getiterfunc tp_iter; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* iternextfunc tp_iternext; */
|
2006-12-17 00:54:45 +01:00
|
|
|
/*** Attribute descriptor and subclassing stuff ***/
|
|
|
|
BPy_IDGroup_methods, /* struct PyMethodDef *tp_methods; */
|
|
|
|
NULL, /* struct PyMemberDef *tp_members; */
|
|
|
|
BPy_IDGroup_getseters, /* struct PyGetSetDef *tp_getset; */
|
2006-11-17 05:46:48 +01:00
|
|
|
};
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
/*********** Main external wrapping function *******/
|
2006-11-19 17:00:04 +01:00
|
|
|
PyObject *BPy_Wrap_IDProperty(ID *id, IDProperty *prop, IDProperty *parent)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2006-12-17 00:54:45 +01:00
|
|
|
BPy_IDProperty *wrap = PyObject_New(BPy_IDProperty, &IDGroup_Type);
|
2006-11-17 05:46:48 +01:00
|
|
|
wrap->prop = prop;
|
2006-11-19 17:00:04 +01:00
|
|
|
wrap->parent = parent;
|
2006-11-17 05:46:48 +01:00
|
|
|
wrap->id = id;
|
|
|
|
//wrap->destroy = 0;
|
|
|
|
return (PyObject*) wrap;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/********Array Wrapper********/
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *IDArray_repr(BPy_IDArray *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2010-10-14 01:25:08 +02:00
|
|
|
return PyUnicode_FromFormat("(ID Array [%d])", self->prop->len);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDArray_GetType(BPy_IDArray *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyLong_FromSsize_t( self->prop->subtype );
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDArray_GetLen(BPy_IDArray *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyLong_FromSsize_t( self->prop->len );
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
static PyGetSetDef BPy_IDArray_getseters[] = {
|
2010-12-03 18:05:21 +01:00
|
|
|
{(char *)"len", (getter)BPy_IDArray_GetLen, (setter)NULL, (char *)"The length of the array, can also be gotten with len(array).", NULL},
|
|
|
|
{(char *)"type", (getter)BPy_IDArray_GetType, (setter)NULL, (char *)"The type of the data in the array, is an ant.", NULL},
|
2006-12-17 00:54:45 +01:00
|
|
|
{NULL, NULL, NULL, NULL, NULL},
|
|
|
|
};
|
|
|
|
|
2009-11-17 13:54:29 +01:00
|
|
|
static PyObject *BPy_IDArray_ConvertToPy(BPy_IDArray *self)
|
|
|
|
{
|
|
|
|
return BPy_IDGroup_MapDataToPy(self->prop);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyMethodDef BPy_IDArray_methods[] = {
|
|
|
|
{"convert_to_pyobject", (PyCFunction)BPy_IDArray_ConvertToPy, METH_NOARGS,
|
2010-02-11 03:03:18 +01:00
|
|
|
"return a purely python version of the group"},
|
2011-02-13 11:52:18 +01:00
|
|
|
{NULL, NULL, 0, NULL}
|
2009-11-17 13:54:29 +01:00
|
|
|
};
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static int BPy_IDArray_Len(BPy_IDArray *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
|
|
|
return self->prop->len;
|
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDArray_GetItem(BPy_IDArray *self, int index)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
if (index < 0 || index >= self->prop->len) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_IndexError, "index out of range!");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
switch (self->prop->subtype) {
|
|
|
|
case IDP_FLOAT:
|
2007-05-26 06:39:31 +02:00
|
|
|
return PyFloat_FromDouble( (double)(((float*)self->prop->data.pointer)[index]));
|
2006-11-17 05:46:48 +01:00
|
|
|
break;
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
|
|
|
return PyFloat_FromDouble( (((double*)self->prop->data.pointer)[index]));
|
2009-11-16 19:56:58 +01:00
|
|
|
break;
|
2006-11-17 05:46:48 +01:00
|
|
|
case IDP_INT:
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyLong_FromLong( (long)((int*)self->prop->data.pointer)[index] );
|
2006-11-17 05:46:48 +01:00
|
|
|
break;
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_RuntimeError, "invalid/corrupt array type!");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
static int BPy_IDArray_SetItem(BPy_IDArray *self, int index, PyObject *value)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
|
|
|
int i;
|
|
|
|
float f;
|
2008-07-24 21:22:17 +02:00
|
|
|
double d;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
if (index < 0 || index >= self->prop->len) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_RuntimeError, "index out of range!");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
switch (self->prop->subtype) {
|
|
|
|
case IDP_FLOAT:
|
2009-11-16 19:56:58 +01:00
|
|
|
f= (float)PyFloat_AsDouble(value);
|
|
|
|
if (f==-1 && PyErr_Occurred()) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a float");
|
|
|
|
return -1;
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
((float*)self->prop->data.pointer)[index] = f;
|
|
|
|
break;
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
2009-11-16 19:56:58 +01:00
|
|
|
d= PyFloat_AsDouble(value);
|
|
|
|
if (d==-1 && PyErr_Occurred()) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a float");
|
|
|
|
return -1;
|
|
|
|
}
|
2008-07-24 21:22:17 +02:00
|
|
|
((double*)self->prop->data.pointer)[index] = d;
|
|
|
|
break;
|
2006-11-17 05:46:48 +01:00
|
|
|
case IDP_INT:
|
2009-11-16 19:56:58 +01:00
|
|
|
i= PyLong_AsSsize_t(value);
|
|
|
|
if (i==-1 && PyErr_Occurred()) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected an int type");
|
|
|
|
return -1;
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
((int*)self->prop->data.pointer)[index] = i;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PySequenceMethods BPy_IDArray_Seq = {
|
2009-11-16 19:56:58 +01:00
|
|
|
(lenfunc) BPy_IDArray_Len, /* inquiry sq_length */
|
2011-02-13 11:52:18 +01:00
|
|
|
NULL, /* binaryfunc sq_concat */
|
|
|
|
NULL, /* intargfunc sq_repeat */
|
2009-11-16 19:56:58 +01:00
|
|
|
(ssizeargfunc)BPy_IDArray_GetItem, /* intargfunc sq_item */
|
2011-02-13 11:52:18 +01:00
|
|
|
NULL, /* intintargfunc sq_slice */
|
|
|
|
(ssizeobjargproc)BPy_IDArray_SetItem,/* intobjargproc sq_ass_item */
|
|
|
|
NULL, /* intintobjargproc sq_ass_slice */
|
|
|
|
NULL, /* objobjproc sq_contains */
|
2006-11-17 05:46:48 +01:00
|
|
|
/* Added in release 2.0 */
|
2011-02-13 11:52:18 +01:00
|
|
|
NULL, /* binaryfunc sq_inplace_concat */
|
|
|
|
NULL, /* intargfunc sq_inplace_repeat */
|
2006-11-17 05:46:48 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
PyTypeObject IDArray_Type = {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
2006-11-17 05:46:48 +01:00
|
|
|
/* For printing, in format "<module>.<name>" */
|
|
|
|
"Blender IDArray", /* char *tp_name; */
|
|
|
|
sizeof( BPy_IDArray ), /* int tp_basicsize; */
|
|
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
|
2007-03-15 02:47:53 +01:00
|
|
|
NULL, /* destructor tp_dealloc; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* printfunc tp_print; */
|
2006-12-17 00:54:45 +01:00
|
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
|
|
NULL, /* setattrfunc tp_setattr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* cmpfunc tp_compare; */
|
|
|
|
( reprfunc ) IDArray_repr, /* reprfunc tp_repr; */
|
|
|
|
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
|
|
|
&BPy_IDArray_Seq, /* PySequenceMethods *tp_as_sequence; */
|
|
|
|
NULL, /* PyMappingMethods *tp_as_mapping; */
|
|
|
|
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
|
|
|
|
NULL, /* hashfunc tp_hash; */
|
|
|
|
NULL, /* ternaryfunc tp_call; */
|
|
|
|
NULL, /* reprfunc tp_str; */
|
|
|
|
NULL, /* getattrofunc tp_getattro; */
|
|
|
|
NULL, /* setattrofunc tp_setattro; */
|
|
|
|
|
|
|
|
/* Functions to access object as input/output buffer */
|
|
|
|
NULL, /* PyBufferProcs *tp_as_buffer; */
|
|
|
|
|
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
|
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
|
|
|
/*** Assigned meaning in release 2.0 ***/
|
|
|
|
/* call function for all accessible objects */
|
|
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
|
|
|
|
/* delete references to contained objects */
|
|
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
|
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
|
|
/*** rich comparisons ***/
|
|
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
|
|
|
|
/*** weak reference enabler ***/
|
|
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
|
|
|
|
/*** Added in release 2.2 ***/
|
|
|
|
/* Iterators */
|
|
|
|
NULL, /* getiterfunc tp_iter; */
|
|
|
|
NULL, /* iternextfunc tp_iternext; */
|
|
|
|
|
|
|
|
/*** Attribute descriptor and subclassing stuff ***/
|
2009-11-17 13:54:29 +01:00
|
|
|
BPy_IDArray_methods, /* struct PyMethodDef *tp_methods; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* struct PyMemberDef *tp_members; */
|
2006-12-17 00:54:45 +01:00
|
|
|
BPy_IDArray_getseters, /* struct PyGetSetDef *tp_getset; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* struct _typeobject *tp_base; */
|
|
|
|
NULL, /* PyObject *tp_dict; */
|
|
|
|
NULL, /* descrgetfunc tp_descr_get; */
|
|
|
|
NULL, /* descrsetfunc tp_descr_set; */
|
|
|
|
0, /* long tp_dictoffset; */
|
|
|
|
NULL, /* initproc tp_init; */
|
|
|
|
NULL, /* allocfunc tp_alloc; */
|
|
|
|
NULL, /* newfunc tp_new; */
|
|
|
|
/* Low-level free-memory routine */
|
|
|
|
NULL, /* freefunc tp_free; */
|
|
|
|
/* For PyObject_IS_GC */
|
|
|
|
NULL, /* inquiry tp_is_gc; */
|
|
|
|
NULL, /* PyObject *tp_bases; */
|
|
|
|
/* method resolution order */
|
|
|
|
NULL, /* PyObject *tp_mro; */
|
|
|
|
NULL, /* PyObject *tp_cache; */
|
|
|
|
NULL, /* PyObject *tp_subclasses; */
|
|
|
|
NULL, /* PyObject *tp_weaklist; */
|
|
|
|
NULL
|
|
|
|
};
|
|
|
|
|
|
|
|
/*********** ID Property Group iterator ********/
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *IDGroup_Iter_iterself(PyObject *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2006-12-17 00:54:45 +01:00
|
|
|
Py_XINCREF(self);
|
|
|
|
return self;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *IDGroup_Iter_repr(BPy_IDGroup_Iter *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2010-10-14 01:25:08 +02:00
|
|
|
return PyUnicode_FromFormat("(ID Property Group Iter \"%s\")", self->group->prop->name);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_Group_Iter_Next(BPy_IDGroup_Iter *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
|
|
|
IDProperty *cur=NULL;
|
2007-05-26 06:39:31 +02:00
|
|
|
PyObject *ret;
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
if (self->cur) {
|
|
|
|
cur = self->cur;
|
|
|
|
self->cur = self->cur->next;
|
2006-12-17 00:54:45 +01:00
|
|
|
if (self->mode == IDPROP_ITER_ITEMS) {
|
2009-11-16 21:16:45 +01:00
|
|
|
ret = PyTuple_New(2);
|
|
|
|
PyTuple_SET_ITEM(ret, 0, PyUnicode_FromString(cur->name));
|
|
|
|
PyTuple_SET_ITEM(ret, 1, BPy_IDGroup_WrapData(self->group->id, cur));
|
2007-05-26 06:39:31 +02:00
|
|
|
return ret;
|
2006-12-17 00:54:45 +01:00
|
|
|
} else {
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyUnicode_FromString(cur->name);
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
} else {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_StopIteration, "iterator at end");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
PyTypeObject IDGroup_Iter_Type = {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
2006-11-17 05:46:48 +01:00
|
|
|
/* For printing, in format "<module>.<name>" */
|
|
|
|
"Blender IDGroup_Iter", /* char *tp_name; */
|
|
|
|
sizeof( BPy_IDGroup_Iter ), /* int tp_basicsize; */
|
|
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
|
2007-03-15 02:47:53 +01:00
|
|
|
NULL, /* destructor tp_dealloc; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* printfunc tp_print; */
|
2006-12-17 00:54:45 +01:00
|
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
|
|
NULL, /* setattrfunc tp_setattr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* cmpfunc tp_compare; */
|
|
|
|
( reprfunc ) IDGroup_Iter_repr, /* reprfunc tp_repr; */
|
|
|
|
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
2009-11-16 19:56:58 +01:00
|
|
|
NULL, /* PySequenceMethods *tp_as_sequence; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* PyMappingMethods *tp_as_mapping; */
|
|
|
|
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
|
|
|
|
NULL, /* hashfunc tp_hash; */
|
|
|
|
NULL, /* ternaryfunc tp_call; */
|
|
|
|
NULL, /* reprfunc tp_str; */
|
|
|
|
NULL, /* getattrofunc tp_getattro; */
|
|
|
|
NULL, /* setattrofunc tp_setattro; */
|
|
|
|
|
|
|
|
/* Functions to access object as input/output buffer */
|
|
|
|
NULL, /* PyBufferProcs *tp_as_buffer; */
|
|
|
|
|
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
|
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
|
|
|
/*** Assigned meaning in release 2.0 ***/
|
|
|
|
/* call function for all accessible objects */
|
|
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
|
|
|
|
/* delete references to contained objects */
|
|
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
|
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
|
|
/*** rich comparisons ***/
|
|
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
|
|
|
|
/*** weak reference enabler ***/
|
|
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
|
|
|
|
/*** Added in release 2.2 ***/
|
|
|
|
/* Iterators */
|
2006-12-17 00:54:45 +01:00
|
|
|
IDGroup_Iter_iterself, /* getiterfunc tp_iter; */
|
2006-11-17 05:46:48 +01:00
|
|
|
(iternextfunc) BPy_Group_Iter_Next, /* iternextfunc tp_iternext; */
|
|
|
|
};
|
|
|
|
|
2006-11-30 00:31:46 +01:00
|
|
|
void IDProp_Init_Types(void)
|
|
|
|
{
|
|
|
|
PyType_Ready( &IDGroup_Type );
|
|
|
|
PyType_Ready( &IDGroup_Iter_Type );
|
|
|
|
PyType_Ready( &IDArray_Type );
|
|
|
|
}
|