| 
									
										
										
										
											2011-02-23 10:52:22 +00:00
										 |  |  | /*
 | 
					
						
							| 
									
										
										
										
											2008-04-16 22:40:48 +00:00
										 |  |  |  * ***** BEGIN GPL LICENSE BLOCK ***** | 
					
						
							| 
									
										
										
										
											2006-11-29 23:31:46 +00: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-16 22:40:48 +00:00
										 |  |  |  * of the License, or (at your option) any later version. | 
					
						
							| 
									
										
										
										
											2006-11-29 23:31:46 +00: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 13:34:04 +00:00
										 |  |  |  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. | 
					
						
							| 
									
										
										
										
											2006-11-29 23:31:46 +00:00
										 |  |  |  * | 
					
						
							|  |  |  |  * | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  |  * Contributor(s): Joseph Eagar, Campbell Barton | 
					
						
							| 
									
										
										
										
											2006-11-29 23:31:46 +00:00
										 |  |  |  * | 
					
						
							| 
									
										
										
										
											2008-04-16 22:40:48 +00:00
										 |  |  |  * ***** END GPL LICENSE BLOCK ***** | 
					
						
							| 
									
										
										
										
											2006-11-29 23:31:46 +00:00
										 |  |  |  */ | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | /** \file blender/python/generic/idprop_py_api.c
 | 
					
						
							| 
									
										
										
										
											2011-02-27 20:10:08 +00:00
										 |  |  |  *  \ingroup pygen | 
					
						
							|  |  |  |  */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-02-14 04:15:25 +00:00
										 |  |  | #include <Python.h>
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 09:28:15 +00:00
										 |  |  | #include "idprop_py_api.h"
 | 
					
						
							| 
									
										
										
										
											2011-01-07 18:36:47 +00:00
										 |  |  | #include "MEM_guardedalloc.h"
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #include "BLI_string.h"
 | 
					
						
							|  |  |  | #include "BLI_utildefines.h"
 | 
					
						
							| 
									
										
										
										
											2010-12-05 23:50:55 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | #include "BKE_idprop.h"
 | 
					
						
							| 
									
										
										
										
											2011-01-07 19:18:31 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-09-01 14:13:48 +00:00
										 |  |  | #define USE_STRING_COERCE
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #ifdef USE_STRING_COERCE
 | 
					
						
							|  |  |  | #include "py_capi_utils.h"
 | 
					
						
							|  |  |  | #endif
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | /*********************** ID Property Main Wrapper Stuff ***************/ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | /* ----------------------------------------------------------------------------
 | 
					
						
							|  |  |  |  * static conversion functions to avoid duplicate code, no type checking. | 
					
						
							|  |  |  |  */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | static PyObject *idprop_py_from_idp_string(const IDProperty *prop) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	if (prop->subtype == IDP_STRING_SUB_BYTE) { | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | 		return PyBytes_FromStringAndSize(IDP_String(prop), prop->len); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	else { | 
					
						
							|  |  |  | #ifdef USE_STRING_COERCE
 | 
					
						
							|  |  |  | 		return PyC_UnicodeFromByteAndSize(IDP_Array(prop), prop->len - 1); | 
					
						
							|  |  |  | #else
 | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | 		return PyUnicode_FromStringAndSize(IDP_String(prop), prop->len - 1); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | #endif
 | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | static PyObject *idprop_py_from_idp_int(const IDProperty *prop) | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | 	return PyLong_FromLong((long)IDP_Int(prop)); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | static PyObject *idprop_py_from_idp_float(const IDProperty *prop) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | 	return PyFloat_FromDouble((double)IDP_Float(prop)); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2011-11-15 09:12:10 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | static PyObject *idprop_py_from_idp_double(const IDProperty *prop) | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | 	return PyFloat_FromDouble(IDP_Double(prop)); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2011-11-15 09:12:10 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | static PyObject *idprop_py_from_idp_group(ID *id, IDProperty *prop, IDProperty *parent) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	BPy_IDProperty *group = PyObject_New(BPy_IDProperty, &BPy_IDGroup_Type); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	group->id = id; | 
					
						
							|  |  |  | 	group->prop = prop; | 
					
						
							|  |  |  | 	group->parent = parent; /* can be NULL */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	return (PyObject *)group; | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | static PyObject *idprop_py_from_idp_array(ID *id, IDProperty *prop) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  | 	BPy_IDProperty *array = PyObject_New(BPy_IDProperty, &BPy_IDArray_Type); | 
					
						
							|  |  |  | 	array->id = id; | 
					
						
							|  |  |  | 	array->prop = prop; | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	return (PyObject *)array; | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | static PyObject *idprop_py_from_idp_idparray(ID *id, IDProperty *prop) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  | 	PyObject *seq = PyList_New(prop->len), *wrap; | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	IDProperty *array = IDP_IDPArray(prop); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	int i; | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	if (!seq) { | 
					
						
							|  |  |  | 		PyErr_Format(PyExc_RuntimeError, | 
					
						
							|  |  |  | 		             "%s: IDP_IDPARRAY: PyList_New(%d) failed", | 
					
						
							|  |  |  | 		             __func__, prop->len); | 
					
						
							|  |  |  | 		return NULL; | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	for (i = 0; i < prop->len; i++) { | 
					
						
							|  |  |  | 		wrap = BPy_IDGroup_WrapData(id, array++, prop); | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		if (!wrap) /* BPy_IDGroup_MapDataToPy sets the error */ | 
					
						
							|  |  |  | 			return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		PyList_SET_ITEM(seq, i, wrap); | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return seq; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* -------------------------------------------------------------------------- */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* use for both array and group */ | 
					
						
							| 
									
										
										
										
											2012-02-26 17:20:36 +00:00
										 |  |  | static Py_hash_t BPy_IDGroup_hash(BPy_IDProperty *self) | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	return _Py_HashPointer(self->prop); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject *BPy_IDGroup_repr(BPy_IDProperty *self) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2012-04-11 08:37:48 +00:00
										 |  |  | 	return PyUnicode_FromFormat("<bpy id prop: owner=\"%s\", name=\"%s\", address=%p>", | 
					
						
							|  |  |  | 	                            self->id ? self->id->name : "<NONE>", self->prop->name, self->prop); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyObject *BPy_IDGroup_WrapData(ID *id, IDProperty *prop, IDProperty *parent) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  | 	switch (prop->type) { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		case IDP_STRING:   return idprop_py_from_idp_string(prop); | 
					
						
							|  |  |  | 		case IDP_INT:      return idprop_py_from_idp_int(prop); | 
					
						
							|  |  |  | 		case IDP_FLOAT:    return idprop_py_from_idp_float(prop); | 
					
						
							|  |  |  | 		case IDP_DOUBLE:   return idprop_py_from_idp_double(prop); | 
					
						
							|  |  |  | 		case IDP_GROUP:    return idprop_py_from_idp_group(id, prop, parent); | 
					
						
							|  |  |  | 		case IDP_ARRAY:    return idprop_py_from_idp_array(id, prop); | 
					
						
							|  |  |  | 		case IDP_IDPARRAY: return idprop_py_from_idp_idparray(id, prop); /* this could be better a internal type */ | 
					
						
							|  |  |  | 		default: Py_RETURN_NONE; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-07-16 23:23:33 +00:00
										 |  |  | #if 0 /* UNUSED, currently assignment overwrites into new properties, rather than setting in-place */
 | 
					
						
							| 
									
										
										
										
											2011-02-13 10:52:18 +00:00
										 |  |  | static int BPy_IDGroup_SetData(BPy_IDProperty *self, IDProperty *prop, PyObject *value) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	switch (prop->type) { | 
					
						
							|  |  |  | 		case IDP_STRING: | 
					
						
							|  |  |  | 		{ | 
					
						
							|  |  |  | 			char *st; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 			if (!PyUnicode_Check(value)) { | 
					
						
							|  |  |  | 				PyErr_SetString(PyExc_TypeError, "expected a string!"); | 
					
						
							|  |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 			/* NOTE: if this code is enabled, bytes support needs to be added */ | 
					
						
							| 
									
										
										
										
											2010-09-01 14:13:48 +00:00
										 |  |  | #ifdef USE_STRING_COERCE
 | 
					
						
							|  |  |  | 			{ | 
					
						
							|  |  |  | 				int alloc_len; | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 				PyObject *value_coerce = NULL; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 				st = (char *)PyC_UnicodeAsByte(value, &value_coerce); | 
					
						
							|  |  |  | 				alloc_len = strlen(st) + 1; | 
					
						
							| 
									
										
										
										
											2010-09-01 14:13:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 				st = _PyUnicode_AsString(value); | 
					
						
							|  |  |  | 				IDP_ResizeArray(prop, alloc_len); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 				memcpy(IDP_Array(prop), st, alloc_len); | 
					
						
							| 
									
										
										
										
											2010-09-01 14:13:48 +00:00
										 |  |  | 				Py_XDECREF(value_coerce); | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | #else
 | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 			st = _PyUnicode_AsString(value); | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 			IDP_ResizeArray(prop, strlen(st) + 1); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 			strcpy(IDP_Array(prop), st); | 
					
						
							| 
									
										
										
										
											2010-09-01 14:13:48 +00:00
										 |  |  | #endif
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 			return 0; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		case IDP_INT: | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			int ivalue = PyLong_AsSsize_t(value); | 
					
						
							|  |  |  | 			if (ivalue == -1 && PyErr_Occurred()) { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 				PyErr_SetString(PyExc_TypeError, "expected an int type"); | 
					
						
							|  |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | 			IDP_Int(prop) = ivalue; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 			break; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		case IDP_FLOAT: | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			float fvalue = (float)PyFloat_AsDouble(value); | 
					
						
							|  |  |  | 			if (fvalue == -1 && PyErr_Occurred()) { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 				PyErr_SetString(PyExc_TypeError, "expected a float"); | 
					
						
							|  |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | 			IDP_Float(self->prop) = fvalue; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 			break; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2008-07-24 19:22:17 +00:00
										 |  |  | 		case IDP_DOUBLE: | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			double dvalue = PyFloat_AsDouble(value); | 
					
						
							|  |  |  | 			if (dvalue == -1 && PyErr_Occurred()) { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 				PyErr_SetString(PyExc_TypeError, "expected a float"); | 
					
						
							|  |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2012-11-03 18:23:30 +00:00
										 |  |  | 			IDP_Double(self->prop) = dvalue; | 
					
						
							| 
									
										
										
										
											2008-07-24 19:22:17 +00:00
										 |  |  | 			break; | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 		default: | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 			PyErr_SetString(PyExc_AttributeError, "attempt to set read-only attribute!"); | 
					
						
							|  |  |  | 			return -1; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	return 0; | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2011-02-25 02:27:54 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-02-13 10:52:18 +00:00
										 |  |  | static PyObject *BPy_IDGroup_GetName(BPy_IDProperty *self, void *UNUSED(closure)) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	return PyUnicode_FromString(self->prop->name); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-10-13 23:25:08 +00:00
										 |  |  | static int BPy_IDGroup_SetName(BPy_IDProperty *self, PyObject *value, void *UNUSED(closure)) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 	const char *name; | 
					
						
							|  |  |  | 	Py_ssize_t name_size; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	if (!PyUnicode_Check(value)) { | 
					
						
							|  |  |  | 		PyErr_SetString(PyExc_TypeError, "expected a string!"); | 
					
						
							|  |  |  | 		return -1; | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 	name = _PyUnicode_AsStringAndSize(value, &name_size); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	if (name_size > MAX_IDPROP_NAME) { | 
					
						
							| 
									
										
										
										
											2012-01-11 08:51:06 +00:00
										 |  |  | 		PyErr_SetString(PyExc_TypeError, "string length cannot exceed 63 characters!"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return -1; | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 	memcpy(self->prop->name, name, name_size); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	return 0; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | #if 0
 | 
					
						
							| 
									
										
										
										
											2008-09-29 17:08:11 +00:00
										 |  |  | static PyObject *BPy_IDGroup_GetType(BPy_IDProperty *self) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2012-11-21 02:28:36 +00:00
										 |  |  | 	return PyLong_FromLong(self->prop->type); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PyGetSetDef BPy_IDGroup_getseters[] = { | 
					
						
							| 
									
										
										
										
											2010-12-03 17:05:21 +00:00
										 |  |  | 	{(char *)"name", (getter)BPy_IDGroup_GetName, (setter)BPy_IDGroup_SetName, (char *)"The name of this Group.", NULL}, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	{NULL, NULL, NULL, NULL, NULL} | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | }; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static Py_ssize_t BPy_IDGroup_Map_Len(BPy_IDProperty *self) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	if (self->prop->type != IDP_GROUP) { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 		PyErr_SetString(PyExc_TypeError, "len() of unsized object"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return -1; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	return self->prop->len; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-09-29 17:08:11 +00:00
										 |  |  | static PyObject *BPy_IDGroup_Map_GetItem(BPy_IDProperty *self, PyObject *item) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	IDProperty *idprop; | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 	const char *name; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-05-20 19:49:27 +00:00
										 |  |  | 	if (self->prop->type != IDP_GROUP) { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 		PyErr_SetString(PyExc_TypeError, "unsubscriptable object"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return NULL; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	name = _PyUnicode_AsString(item); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	if (name == NULL) { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 		PyErr_SetString(PyExc_TypeError, "only strings are allowed as keys of ID properties"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return NULL; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	idprop = IDP_GetPropertyFromGroup(self->prop, name); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	if (idprop == NULL) { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 		PyErr_SetString(PyExc_KeyError, "key not in subgroup dict"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return NULL; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	return BPy_IDGroup_WrapData(self->id, idprop, self->prop); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | /* returns NULL on success, error string on failure */ | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | static int idp_sequence_type(PyObject *seq_fast) | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	PyObject *item; | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	int type = IDP_INT; | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 	Py_ssize_t i, len = PySequence_Fast_GET_SIZE(seq_fast); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	for (i = 0; i < len; i++) { | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 		item = PySequence_Fast_GET_ITEM(seq_fast, i); | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 		if (PyFloat_Check(item)) { | 
					
						
							| 
									
										
										
										
											2011-10-13 01:29:08 +00:00
										 |  |  | 			if (type == IDP_IDPARRAY) { /* mixed dict/int */ | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			type = IDP_DOUBLE; | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 		} | 
					
						
							|  |  |  | 		else if (PyLong_Check(item)) { | 
					
						
							| 
									
										
										
										
											2011-10-13 01:29:08 +00:00
										 |  |  | 			if (type == IDP_IDPARRAY) { /* mixed dict/int */ | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2010-03-02 14:57:28 +00:00
										 |  |  | 		else if (PyMapping_Check(item)) { | 
					
						
							| 
									
										
										
										
											2011-10-13 01:29:08 +00:00
										 |  |  | 			if (i != 0 && (type != IDP_IDPARRAY)) { /* mixed dict/int */ | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			type = IDP_IDPARRAY; | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 		} | 
					
						
							|  |  |  | 		else { | 
					
						
							|  |  |  | 			return -1; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return type; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | /* note: group can be a pointer array or a group.
 | 
					
						
							|  |  |  |  * assume we already checked key is a string. */ | 
					
						
							|  |  |  | const char *BPy_IDProperty_Map_ValidateAndCreate(PyObject *name_obj, IDProperty *group, PyObject *ob) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2007-05-22 04:41:21 +00:00
										 |  |  | 	IDProperty *prop = NULL; | 
					
						
							| 
									
										
										
										
											2007-03-01 21:30:48 +00:00
										 |  |  | 	IDPropertyTemplate val = {0}; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	const char *name = ""; | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	if (name_obj) { | 
					
						
							|  |  |  | 		Py_ssize_t name_size; | 
					
						
							|  |  |  | 		name = _PyUnicode_AsStringAndSize(name_obj, &name_size); | 
					
						
							|  |  |  | 		if (name_size > MAX_IDPROP_NAME) { | 
					
						
							| 
									
										
										
										
											2012-01-11 08:51:06 +00:00
										 |  |  | 			return "the length of IDProperty names is limited to 63 characters"; | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	if (PyFloat_Check(ob)) { | 
					
						
							| 
									
										
										
										
											2008-07-24 19:22:17 +00:00
										 |  |  | 		val.d = PyFloat_AsDouble(ob); | 
					
						
							| 
									
										
										
										
											2011-11-15 09:22:52 +00:00
										 |  |  | 		prop = IDP_New(IDP_DOUBLE, &val, name); | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	else if (PyLong_Check(ob)) { | 
					
						
							| 
									
										
										
										
											2012-11-21 02:28:36 +00:00
										 |  |  | 		val.i = (int)PyLong_AsLong(ob); | 
					
						
							| 
									
										
										
										
											2011-11-15 09:22:52 +00:00
										 |  |  | 		prop = IDP_New(IDP_INT, &val, name); | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	else if (PyUnicode_Check(ob)) { | 
					
						
							| 
									
										
										
										
											2010-09-01 14:13:48 +00:00
										 |  |  | #ifdef USE_STRING_COERCE
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		PyObject *value_coerce = NULL; | 
					
						
							| 
									
										
										
										
											2011-11-15 09:12:10 +00:00
										 |  |  | 		val.string.str = (char *)PyC_UnicodeAsByte(ob, &value_coerce); | 
					
						
							|  |  |  | 		val.string.subtype = IDP_STRING_SUB_UTF8; | 
					
						
							| 
									
										
										
										
											2011-11-15 09:22:52 +00:00
										 |  |  | 		prop = IDP_New(IDP_STRING, &val, name); | 
					
						
							| 
									
										
										
										
											2010-09-01 14:13:48 +00:00
										 |  |  | 		Py_XDECREF(value_coerce); | 
					
						
							|  |  |  | #else
 | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		val.str = _PyUnicode_AsString(ob); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		prop = IDP_New(IDP_STRING, val, name); | 
					
						
							| 
									
										
										
										
											2010-09-01 14:13:48 +00:00
										 |  |  | #endif
 | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2011-11-15 09:12:10 +00:00
										 |  |  | 	else if (PyBytes_Check(ob)) { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		val.string.str = PyBytes_AS_STRING(ob); | 
					
						
							|  |  |  | 		val.string.len = PyBytes_GET_SIZE(ob); | 
					
						
							|  |  |  | 		val.string.subtype = IDP_STRING_SUB_BYTE; | 
					
						
							| 
									
										
										
										
											2011-11-15 09:12:10 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 09:22:52 +00:00
										 |  |  | 		prop = IDP_New(IDP_STRING, &val, name); | 
					
						
							| 
									
										
										
										
											2011-11-15 09:12:10 +00:00
										 |  |  | 		//prop = IDP_NewString(PyBytes_AS_STRING(ob), name, PyBytes_GET_SIZE(ob));
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		//prop->subtype = IDP_STRING_SUB_BYTE;
 | 
					
						
							| 
									
										
										
										
											2011-11-15 09:12:10 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 	else if (PySequence_Check(ob)) { | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 		PyObject *ob_seq_fast = PySequence_Fast(ob, "py -> idprop"); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		PyObject *item; | 
					
						
							|  |  |  | 		int i; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 		if (ob_seq_fast == NULL) { | 
					
						
							|  |  |  | 			PyErr_Print(); | 
					
						
							|  |  |  | 			PyErr_Clear(); | 
					
						
							|  |  |  | 			return "error converting the sequence"; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if ((val.array.type = idp_sequence_type(ob_seq_fast)) == -1) { | 
					
						
							|  |  |  | 			Py_DECREF(ob_seq_fast); | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 			return "only floats, ints and dicts are allowed in ID property arrays"; | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-03 20:36:09 +00:00
										 |  |  | 		/* validate sequence and derive type.
 | 
					
						
							|  |  |  | 		 * we assume IDP_INT unless we hit a float | 
					
						
							|  |  |  | 		 * number; then we assume it's */ | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 		val.array.len = PySequence_Fast_GET_SIZE(ob_seq_fast); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-12-18 08:50:06 +00:00
										 |  |  | 		switch (val.array.type) { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			case IDP_DOUBLE: | 
					
						
							|  |  |  | 				prop = IDP_New(IDP_ARRAY, &val, name); | 
					
						
							|  |  |  | 				for (i = 0; i < val.array.len; i++) { | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 					item = PySequence_Fast_GET_ITEM(ob_seq_fast, i); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 					((double *)IDP_Array(prop))[i] = (float)PyFloat_AsDouble(item); | 
					
						
							|  |  |  | 				} | 
					
						
							|  |  |  | 				break; | 
					
						
							|  |  |  | 			case IDP_INT: | 
					
						
							|  |  |  | 				prop = IDP_New(IDP_ARRAY, &val, name); | 
					
						
							|  |  |  | 				for (i = 0; i < val.array.len; i++) { | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 					item = PySequence_Fast_GET_ITEM(ob_seq_fast, i); | 
					
						
							| 
									
										
										
										
											2012-11-21 02:28:36 +00:00
										 |  |  | 					((int *)IDP_Array(prop))[i] = (int)PyLong_AsLong(item); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 				} | 
					
						
							|  |  |  | 				break; | 
					
						
							|  |  |  | 			case IDP_IDPARRAY: | 
					
						
							|  |  |  | 				prop = IDP_NewIDPArray(name); | 
					
						
							|  |  |  | 				for (i = 0; i < val.array.len; i++) { | 
					
						
							|  |  |  | 					const char *error; | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 					item = PySequence_Fast_GET_ITEM(ob_seq_fast, i); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 					error = BPy_IDProperty_Map_ValidateAndCreate(NULL, prop, item); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 					if (error) { | 
					
						
							|  |  |  | 						Py_DECREF(ob_seq_fast); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 						return error; | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 					} | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 				} | 
					
						
							|  |  |  | 				break; | 
					
						
							| 
									
										
										
										
											2012-10-12 03:24:47 +00:00
										 |  |  | 			default: | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 				Py_DECREF(ob_seq_fast); | 
					
						
							| 
									
										
										
										
											2012-10-12 03:24:47 +00:00
										 |  |  | 				return "internal error with idp array.type"; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2012-10-12 03:59:50 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 		Py_DECREF(ob_seq_fast); | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	else if (PyMapping_Check(ob)) { | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		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 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-03 20:36:09 +00:00
										 |  |  | 		/* we allocate the group first; if we hit any invalid data,
 | 
					
						
							|  |  |  | 		 * we can delete it easily enough.*/ | 
					
						
							| 
									
										
										
										
											2011-11-15 09:22:52 +00:00
										 |  |  | 		prop = IDP_New(IDP_GROUP, &val, name); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		len = PyMapping_Length(ob); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		for (i = 0; i < len; i++) { | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 			key = PySequence_GetItem(keys, i); | 
					
						
							|  |  |  | 			pval = PySequence_GetItem(vals, i); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 			if (!PyUnicode_Check(key)) { | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 				IDP_FreeProperty(prop); | 
					
						
							|  |  |  | 				MEM_freeN(prop); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 				Py_XDECREF(keys); | 
					
						
							|  |  |  | 				Py_XDECREF(vals); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 				Py_XDECREF(key); | 
					
						
							|  |  |  | 				Py_XDECREF(pval); | 
					
						
							|  |  |  | 				return "invalid element in subgroup dict template!"; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 			if (BPy_IDProperty_Map_ValidateAndCreate(key, prop, pval)) { | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 				IDP_FreeProperty(prop); | 
					
						
							|  |  |  | 				MEM_freeN(prop); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 				Py_XDECREF(keys); | 
					
						
							|  |  |  | 				Py_XDECREF(vals); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00: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); | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	else return "invalid property value"; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	if (group->type == IDP_IDPARRAY) { | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 		IDP_AppendArray(group, prop); | 
					
						
							| 
									
										
										
										
											2012-10-12 03:24:47 +00:00
										 |  |  | 		// IDP_FreeProperty(item);  /* IDP_AppendArray does a shallow copy (memcpy), only free memory */
 | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 		MEM_freeN(prop); | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	else { | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 		IDP_ReplaceInGroup(group, prop); | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	return NULL; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | int BPy_Wrap_SetMapItem(IDProperty *prop, PyObject *key, PyObject *val) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2012-05-20 19:49:27 +00:00
										 |  |  | 	if (prop->type != IDP_GROUP) { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 		PyErr_SetString(PyExc_TypeError, "unsubscriptable object"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return -1; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | 	if (val == NULL) { /* del idprop[key] */ | 
					
						
							|  |  |  | 		IDProperty *pkey = IDP_GetPropertyFromGroup(prop, _PyUnicode_AsString(key)); | 
					
						
							| 
									
										
										
										
											2006-12-01 03:04:36 +00:00
										 |  |  | 		if (pkey) { | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | 			IDP_RemFromGroup(prop, pkey); | 
					
						
							| 
									
										
										
										
											2006-12-01 03:04:36 +00:00
										 |  |  | 			IDP_FreeProperty(pkey); | 
					
						
							|  |  |  | 			MEM_freeN(pkey); | 
					
						
							|  |  |  | 			return 0; | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 		} | 
					
						
							|  |  |  | 		else { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 			PyErr_SetString(PyExc_KeyError, "property not found in group"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 			return -1; | 
					
						
							|  |  |  | 		} | 
					
						
							| 
									
										
										
										
											2006-12-01 03:04:36 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | 	else { | 
					
						
							| 
									
										
										
										
											2010-11-19 02:14:18 +00:00
										 |  |  | 		const char *err; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | 		if (!PyUnicode_Check(key)) { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 			PyErr_SetString(PyExc_TypeError, "only strings are allowed as subgroup keys"); | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | 			return -1; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 		err = BPy_IDProperty_Map_ValidateAndCreate(key, prop, val); | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | 		if (err) { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			PyErr_SetString(PyExc_KeyError, err); | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | 			return -1; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		return 0; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 22:21:39 +00:00
										 |  |  | static int BPy_IDGroup_Map_SetItem(BPy_IDProperty *self, PyObject *key, PyObject *val) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2009-11-17 12:54:29 +00:00
										 |  |  | 	return BPy_Wrap_SetMapItem(self->prop, key, val); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | static PyObject *BPy_IDGroup_iter(BPy_IDProperty *self) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	BPy_IDGroup_Iter *iter = PyObject_New(BPy_IDGroup_Iter, &BPy_IDGroup_Iter_Type); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	iter->group = self; | 
					
						
							|  |  |  | 	iter->mode = IDPROP_ITER_KEYS; | 
					
						
							|  |  |  | 	iter->cur = self->prop->data.group.first; | 
					
						
							|  |  |  | 	Py_XINCREF(iter); | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	return (PyObject *)iter; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | /* for simple, non nested types this is the same as BPy_IDGroup_WrapData */ | 
					
						
							| 
									
										
										
										
											2008-09-29 17:08:11 +00:00
										 |  |  | static PyObject *BPy_IDGroup_MapDataToPy(IDProperty *prop) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	switch (prop->type) { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		case IDP_STRING: | 
					
						
							|  |  |  | 			return idprop_py_from_idp_string(prop); | 
					
						
							|  |  |  | 		case IDP_INT: | 
					
						
							|  |  |  | 			return idprop_py_from_idp_int(prop); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 		case IDP_FLOAT: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			return idprop_py_from_idp_float(prop); | 
					
						
							| 
									
										
										
										
											2008-07-24 19:22:17 +00:00
										 |  |  | 		case IDP_DOUBLE: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			return idprop_py_from_idp_double(prop); | 
					
						
							|  |  |  | 		case IDP_ARRAY: | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			PyObject *seq = PyList_New(prop->len); | 
					
						
							|  |  |  | 			int i; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			if (!seq) { | 
					
						
							|  |  |  | 				PyErr_Format(PyExc_RuntimeError, | 
					
						
							|  |  |  | 				             "%s: IDP_ARRAY: PyList_New(%d) failed", | 
					
						
							|  |  |  | 				             __func__, prop->len); | 
					
						
							|  |  |  | 				return NULL; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 			} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			switch (prop->subtype) { | 
					
						
							|  |  |  | 				case IDP_FLOAT: | 
					
						
							|  |  |  | 				{ | 
					
						
							|  |  |  | 					float *array = (float *)IDP_Array(prop); | 
					
						
							|  |  |  | 					for (i = 0; i < prop->len; i++) { | 
					
						
							|  |  |  | 						PyList_SET_ITEM(seq, i, PyFloat_FromDouble(array[i])); | 
					
						
							|  |  |  | 					} | 
					
						
							|  |  |  | 					break; | 
					
						
							|  |  |  | 				} | 
					
						
							|  |  |  | 				case IDP_DOUBLE: | 
					
						
							|  |  |  | 				{ | 
					
						
							|  |  |  | 					double *array = (double *)IDP_Array(prop); | 
					
						
							|  |  |  | 					for (i = 0; i < prop->len; i++) { | 
					
						
							|  |  |  | 						PyList_SET_ITEM(seq, i, PyFloat_FromDouble(array[i])); | 
					
						
							|  |  |  | 					} | 
					
						
							|  |  |  | 					break; | 
					
						
							|  |  |  | 				} | 
					
						
							|  |  |  | 				case IDP_INT: | 
					
						
							|  |  |  | 				{ | 
					
						
							|  |  |  | 					int *array = (int *)IDP_Array(prop); | 
					
						
							|  |  |  | 					for (i = 0; i < prop->len; i++) { | 
					
						
							|  |  |  | 						PyList_SET_ITEM(seq, i, PyLong_FromLong(array[i])); | 
					
						
							|  |  |  | 					} | 
					
						
							|  |  |  | 					break; | 
					
						
							|  |  |  | 				} | 
					
						
							|  |  |  | 				default: | 
					
						
							|  |  |  | 					PyErr_Format(PyExc_RuntimeError, | 
					
						
							|  |  |  | 					             "%s: invalid/corrupt array type '%d'!", | 
					
						
							|  |  |  | 					             __func__, prop->subtype); | 
					
						
							|  |  |  | 					Py_DECREF(seq); | 
					
						
							|  |  |  | 					return NULL; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			return seq; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		case IDP_IDPARRAY: | 
					
						
							|  |  |  | 		{ | 
					
						
							|  |  |  | 			PyObject *seq = PyList_New(prop->len), *wrap; | 
					
						
							|  |  |  | 			IDProperty *array = IDP_IDPArray(prop); | 
					
						
							|  |  |  | 			int i; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			if (!seq) { | 
					
						
							|  |  |  | 				PyErr_Format(PyExc_RuntimeError, | 
					
						
							|  |  |  | 				             "%s: IDP_IDPARRAY: PyList_New(%d) failed", | 
					
						
							|  |  |  | 				             __func__, prop->len); | 
					
						
							|  |  |  | 				return NULL; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			for (i = 0; i < prop->len; i++) { | 
					
						
							|  |  |  | 				wrap = BPy_IDGroup_MapDataToPy(array++); | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 				if (!wrap) /* BPy_IDGroup_MapDataToPy sets the error */ | 
					
						
							|  |  |  | 					return NULL; | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 				PyList_SET_ITEM(seq, i, wrap); | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 			return seq; | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		case IDP_GROUP: | 
					
						
							|  |  |  | 		{ | 
					
						
							|  |  |  | 			PyObject *dict = PyDict_New(), *wrap; | 
					
						
							|  |  |  | 			IDProperty *loop; | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			for (loop = prop->data.group.first; loop; loop = loop->next) { | 
					
						
							|  |  |  | 				wrap = BPy_IDGroup_MapDataToPy(loop); | 
					
						
							| 
									
										
										
										
											2010-03-02 11:35:28 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 				if (!wrap) /* BPy_IDGroup_MapDataToPy sets the error */ | 
					
						
							|  |  |  | 					return NULL; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 				PyDict_SetItemString(dict, loop->name, wrap); | 
					
						
							|  |  |  | 				Py_DECREF(wrap); | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 			return dict; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	PyErr_Format(PyExc_RuntimeError, | 
					
						
							|  |  |  | 	             "%s ERROR: '%s' property exists with a bad type code '%d'!", | 
					
						
							|  |  |  | 	             __func__, prop->name, prop->type); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	return NULL; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-09-29 17:08:11 +00:00
										 |  |  | static PyObject *BPy_IDGroup_Pop(BPy_IDProperty *self, PyObject *value) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	IDProperty *idprop; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	PyObject *pyform; | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 	const char *name = _PyUnicode_AsString(value); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2007-06-16 12:24:41 +00:00
										 |  |  | 	if (!name) { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  | 		             "pop expected at least a string argument, not %.200s", | 
					
						
							|  |  |  | 		             Py_TYPE(value)->tp_name); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return NULL; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	idprop = IDP_GetPropertyFromGroup(self->prop, name); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-10-13 01:29:08 +00:00
										 |  |  | 	if (idprop) { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		pyform = BPy_IDGroup_MapDataToPy(idprop); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if (!pyform) { | 
					
						
							| 
									
										
										
										
											2012-03-03 20:36:09 +00:00
										 |  |  | 			/* 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.*/ | 
					
						
							| 
									
										
										
										
											2011-04-21 15:53:30 +00:00
										 |  |  | 			return NULL; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 		IDP_RemFromGroup(self->prop, idprop); | 
					
						
							|  |  |  | 		return pyform; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 	PyErr_SetString(PyExc_KeyError, "item not in group"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	return NULL; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-09-29 17:08:11 +00:00
										 |  |  | static PyObject *BPy_IDGroup_IterItems(BPy_IDProperty *self) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	BPy_IDGroup_Iter *iter = PyObject_New(BPy_IDGroup_Iter, &BPy_IDGroup_Iter_Type); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	iter->group = self; | 
					
						
							|  |  |  | 	iter->mode = IDPROP_ITER_ITEMS; | 
					
						
							|  |  |  | 	iter->cur = self->prop->data.group.first; | 
					
						
							|  |  |  | 	Py_XINCREF(iter); | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	return (PyObject *)iter; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | /* utility function */ | 
					
						
							| 
									
										
										
										
											2011-10-31 01:50:04 +00:00
										 |  |  | static void BPy_IDGroup_CorrectListLen(IDProperty *prop, PyObject *seq, int len, const char *func) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2009-11-19 11:56:13 +00:00
										 |  |  | 	int j; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-10-31 01:50:04 +00:00
										 |  |  | 	printf("%s: ID Property Error found and corrected!\n", func); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 	/*fill rest of list with valid references to None*/ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	for (j = len; j < prop->len; j++) { | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 		Py_INCREF(Py_None); | 
					
						
							|  |  |  | 		PyList_SET_ITEM(seq, j, Py_None); | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 	/*set correct group length*/ | 
					
						
							| 
									
										
										
										
											2009-11-19 11:56:13 +00:00
										 |  |  | 	prop->len = len; | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | PyObject *BPy_Wrap_GetKeys(IDProperty *prop) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	PyObject *list = PyList_New(prop->len); | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 	IDProperty *loop; | 
					
						
							|  |  |  | 	int i; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	for (i = 0, loop = prop->data.group.first; loop && (i < prop->len); loop = loop->next, i++) | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		PyList_SET_ITEM(list, i, PyUnicode_FromString(loop->name)); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2010-02-01 18:38:33 +00:00
										 |  |  | 	/* if the id prop is corrupt, count the remaining */ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	for ( ; loop; loop = loop->next, i++) { | 
					
						
							|  |  |  | 		/* pass */ | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2010-02-01 18:38:33 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	if (i != prop->len) { /* if the loop didnt finish, we know the length is wrong */ | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		BPy_IDGroup_CorrectListLen(prop, list, i, __func__); | 
					
						
							|  |  |  | 		Py_DECREF(list); /*free the list*/ | 
					
						
							| 
									
										
										
										
											2008-02-23 02:12:50 +00:00
										 |  |  | 		/*call self again*/ | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 		return BPy_Wrap_GetKeys(prop); | 
					
						
							| 
									
										
										
										
											2008-02-23 02:12:50 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	return list; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | PyObject *BPy_Wrap_GetValues(ID *id, IDProperty *prop) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	PyObject *list = PyList_New(prop->len); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	IDProperty *loop; | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 	int i; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	for (i = 0, loop = prop->data.group.first; loop; loop = loop->next, i++) { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		PyList_SET_ITEM(list, i, BPy_IDGroup_WrapData(id, loop, prop)); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 	if (i != prop->len) { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		BPy_IDGroup_CorrectListLen(prop, list, i, __func__); | 
					
						
							|  |  |  | 		Py_DECREF(list); /*free the list*/ | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 		/*call self again*/ | 
					
						
							|  |  |  | 		return BPy_Wrap_GetValues(id, prop); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2008-02-23 02:12:50 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	return list; | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | PyObject *BPy_Wrap_GetItems(ID *id, IDProperty *prop) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  | 	PyObject *seq = PyList_New(prop->len); | 
					
						
							|  |  |  | 	IDProperty *loop; | 
					
						
							|  |  |  | 	int i; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	for (i = 0, loop = prop->data.group.first; loop; loop = loop->next, i++) { | 
					
						
							|  |  |  | 		PyObject *item = PyTuple_New(2); | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 		PyTuple_SET_ITEM(item, 0, PyUnicode_FromString(loop->name)); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		PyTuple_SET_ITEM(item, 1, BPy_IDGroup_WrapData(id, loop, prop)); | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 		PyList_SET_ITEM(seq, i, item); | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 	if (i != prop->len) { | 
					
						
							| 
									
										
										
										
											2011-10-31 01:50:04 +00:00
										 |  |  | 		BPy_IDGroup_CorrectListLen(prop, seq, i, __func__); | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 		Py_DECREF(seq); /*free the list*/ | 
					
						
							| 
									
										
										
										
											2008-02-23 02:12:50 +00:00
										 |  |  | 		/*call self again*/ | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 		return BPy_Wrap_GetItems(id, prop); | 
					
						
							| 
									
										
										
										
											2008-02-23 02:12:50 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	return seq; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00: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 18:56:58 +00:00
										 |  |  | static int BPy_IDGroup_Contains(BPy_IDProperty *self, PyObject *value) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-11-03 14:09:18 +00:00
										 |  |  | 	const char *name = _PyUnicode_AsString(value); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	if (!name) { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  | 		             "expected a string, not a %.200s", | 
					
						
							|  |  |  | 		             Py_TYPE(value)->tp_name); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return -1; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	return IDP_GetPropertyFromGroup(self->prop, name) ? 1 : 0; | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-12-20 08:41:46 +00:00
										 |  |  | static PyObject *BPy_IDGroup_Update(BPy_IDProperty *self, PyObject *value) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2008-12-20 08:41:46 +00:00
										 |  |  | 	PyObject *pkey, *pval; | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	Py_ssize_t i = 0; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-10-31 19:07:25 +00:00
										 |  |  | 	if (BPy_IDGroup_Check(value)) { | 
					
						
							|  |  |  | 		BPy_IDProperty *other = (BPy_IDProperty *)value; | 
					
						
							|  |  |  | 		if (UNLIKELY(self->prop == other->prop)) { | 
					
						
							|  |  |  | 			Py_RETURN_NONE; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		/* XXX, possible one is inside the other */ | 
					
						
							|  |  |  | 		IDP_MergeGroup(self->prop, other->prop, TRUE); | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	else if (PyDict_Check(value)) { | 
					
						
							|  |  |  | 		while (PyDict_Next(value, &i, &pkey, &pval)) { | 
					
						
							|  |  |  | 			BPy_IDGroup_Map_SetItem(self, pkey, pval); | 
					
						
							|  |  |  | 			if (PyErr_Occurred()) return NULL; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	else { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		PyErr_Format(PyExc_TypeError, | 
					
						
							| 
									
										
										
										
											2012-10-31 19:07:25 +00:00
										 |  |  | 		             "expected a dict or an IDPropertyGroup type, not a %.200s", | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		             Py_TYPE(value)->tp_name); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return NULL; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	Py_RETURN_NONE; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | static PyObject *BPy_IDGroup_to_dict(BPy_IDProperty *self) | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	return BPy_IDGroup_MapDataToPy(self->prop); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-19 17:04:28 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | /* Matches python dict.get(key, [default]) */ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | static PyObject *BPy_IDGroup_Get(BPy_IDProperty *self, PyObject *args) | 
					
						
							| 
									
										
										
										
											2009-11-19 17:04:28 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	IDProperty *idprop; | 
					
						
							|  |  |  | 	char *key; | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	PyObject *def = Py_None; | 
					
						
							| 
									
										
										
										
											2009-11-19 17:04:28 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	if (!PyArg_ParseTuple(args, "s|O:get", &key, &def)) | 
					
						
							|  |  |  | 		return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	idprop = IDP_GetPropertyFromGroup(self->prop, key); | 
					
						
							| 
									
										
										
										
											2009-11-19 17:04:28 +00:00
										 |  |  | 	if (idprop) { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		PyObject *pyobj = BPy_IDGroup_WrapData(self->id, idprop, self->prop); | 
					
						
							| 
									
										
										
										
											2009-11-19 17:04:28 +00:00
										 |  |  | 		if (pyobj) | 
					
						
							|  |  |  | 			return pyobj; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	Py_INCREF(def); | 
					
						
							|  |  |  | 	return def; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | static struct PyMethodDef BPy_IDGroup_methods[] = { | 
					
						
							| 
									
										
										
										
											2007-06-16 12:24:41 +00:00
										 |  |  | 	{"pop", (PyCFunction)BPy_IDGroup_Pop, METH_O, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "pop an item from the group; raises KeyError if the item doesn't exist"}, | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	{"iteritems", (PyCFunction)BPy_IDGroup_IterItems, METH_NOARGS, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "iterate through the items in the dict; behaves like dictionary method iteritems"}, | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	{"keys", (PyCFunction)BPy_IDGroup_GetKeys, METH_NOARGS, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "get the keys associated with this group as a list of strings"}, | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	{"values", (PyCFunction)BPy_IDGroup_GetValues, METH_NOARGS, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "get the values associated with this group"}, | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 	{"items", (PyCFunction)BPy_IDGroup_GetItems, METH_NOARGS, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "get the items associated with this group"}, | 
					
						
							| 
									
										
										
										
											2008-12-20 08:41:46 +00:00
										 |  |  | 	{"update", (PyCFunction)BPy_IDGroup_Update, METH_O, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "updates the values in the group with the values of another or a dict"}, | 
					
						
							| 
									
										
										
										
											2009-11-19 17:04:28 +00:00
										 |  |  | 	{"get", (PyCFunction)BPy_IDGroup_Get, METH_VARARGS, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "idprop.get(k[,d]) -> idprop[k] if k in idprop, else d.  d defaults to None"}, | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	{"to_dict", (PyCFunction)BPy_IDGroup_to_dict, METH_NOARGS, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "return a purely python version of the group"}, | 
					
						
							| 
									
										
										
										
											2011-02-13 10:52:18 +00:00
										 |  |  | 	{NULL, NULL, 0, NULL} | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | }; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | static PySequenceMethods BPy_IDGroup_Seq = { | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00: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*** */ | 
					
						
							|  |  |  | 	(objobjproc) BPy_IDGroup_Contains,  /* objobjproc sq_contains */ | 
					
						
							|  |  |  | 	NULL,                               /* binaryfunc sq_inplace_concat */ | 
					
						
							|  |  |  | 	NULL,                               /* ssizeargfunc sq_inplace_repeat */ | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-02-13 10:52:18 +00:00
										 |  |  | static PyMappingMethods BPy_IDGroup_Mapping = { | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	(lenfunc)BPy_IDGroup_Map_Len,           /*inquiry mp_length */ | 
					
						
							|  |  |  | 	(binaryfunc)BPy_IDGroup_Map_GetItem,    /*binaryfunc mp_subscript */ | 
					
						
							|  |  |  | 	(objobjargproc)BPy_IDGroup_Map_SetItem, /*objobjargproc mp_ass_subscript */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | PyTypeObject BPy_IDGroup_Type = { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	PyVarObject_HEAD_INIT(NULL, 0) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/*  For printing, in format "<module>.<name>" */ | 
					
						
							| 
									
										
										
										
											2012-04-15 14:54:15 +00:00
										 |  |  | 	"IDPropertyGroup",       /* char *tp_name; */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	sizeof(BPy_IDProperty),     /* int tp_basicsize; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	0,                          /* tp_itemsize;  For allocation */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	/* Methods to implement standard operations */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	NULL,                       /* destructor tp_dealloc; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* printfunc tp_print; */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	NULL,                       /* getattrfunc tp_getattr; */ | 
					
						
							|  |  |  | 	NULL,                       /* setattrfunc tp_setattr; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* cmpfunc tp_compare; */ | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	(reprfunc)BPy_IDGroup_repr,     /* reprfunc tp_repr; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	/* Method suites for standard classes */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	NULL,                       /* PyNumberMethods *tp_as_number; */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	&BPy_IDGroup_Seq,           /* PySequenceMethods *tp_as_sequence; */ | 
					
						
							|  |  |  | 	&BPy_IDGroup_Mapping,       /* PyMappingMethods *tp_as_mapping; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	/* More standard operations (here for binary compatibility) */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	(hashfunc)BPy_IDGroup_hash, /* hashfunc tp_hash; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	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; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Flags to define presence of optional/expanded features ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	Py_TPFLAGS_DEFAULT,         /* long tp_flags; */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	NULL,                       /*  char *tp_doc;  Documentation string */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Assigned meaning in release 2.0 ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/* call function for all accessible objects */ | 
					
						
							|  |  |  | 	NULL,                       /* traverseproc tp_traverse; */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	/* delete references to contained objects */ | 
					
						
							|  |  |  | 	NULL,                       /* inquiry tp_clear; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/***  Assigned meaning in release 2.1 ***/ | 
					
						
							|  |  |  | 	/*** rich comparisons ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* richcmpfunc tp_richcompare; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/***  weak reference enabler ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	0,                          /* long tp_weaklistoffset; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Added in release 2.2 ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/*   Iterators */ | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	(getiterfunc)BPy_IDGroup_iter, /* getiterfunc tp_iter; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* iternextfunc tp_iternext; */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Attribute descriptor and subclassing stuff ***/ | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	BPy_IDGroup_methods,        /* struct PyMethodDef *tp_methods; */ | 
					
						
							|  |  |  | 	NULL,                       /* struct PyMemberDef *tp_members; */ | 
					
						
							|  |  |  | 	BPy_IDGroup_getseters,       /* struct PyGetSetDef *tp_getset; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /********Array Wrapper********/ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | static PyTypeObject *idp_array_py_type(BPy_IDArray *self, short *is_double) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	switch (self->prop->subtype) { | 
					
						
							|  |  |  | 		case IDP_FLOAT: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			*is_double = 0; | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 			return &PyFloat_Type; | 
					
						
							|  |  |  | 		case IDP_DOUBLE: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			*is_double = 1; | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 			return &PyFloat_Type; | 
					
						
							|  |  |  | 		case IDP_INT: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			*is_double = 0; | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 			return &PyLong_Type; | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	*is_double = 0; | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	return NULL; | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | static PyObject *BPy_IDArray_repr(BPy_IDArray *self) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	return PyUnicode_FromFormat("<bpy id property array [%d]>", self->prop->len); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | static PyObject *BPy_IDArray_GetType(BPy_IDArray *self) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2011-12-18 08:50:06 +00:00
										 |  |  | 	switch (self->prop->subtype) { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		case IDP_FLOAT:  return PyUnicode_FromString("f"); | 
					
						
							|  |  |  | 		case IDP_DOUBLE: return PyUnicode_FromString("d"); | 
					
						
							|  |  |  | 		case IDP_INT:    return PyUnicode_FromString("i"); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	PyErr_Format(PyExc_RuntimeError, | 
					
						
							|  |  |  | 	             "%s: invalid/corrupt array type '%d'!", | 
					
						
							|  |  |  | 	             __func__, self->prop->subtype); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	return NULL; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | static PyGetSetDef BPy_IDArray_getseters[] = { | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/* matches pythons array.typecode */ | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	{(char *)"typecode", (getter)BPy_IDArray_GetType, (setter)NULL, (char *)"The type of the data in the array, is an int.", NULL}, | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	{NULL, NULL, NULL, NULL, NULL}, | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | static PyObject *BPy_IDArray_to_list(BPy_IDArray *self) | 
					
						
							| 
									
										
										
										
											2009-11-17 12:54:29 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	return BPy_IDGroup_MapDataToPy(self->prop); | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyMethodDef BPy_IDArray_methods[] = { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	{"to_list", (PyCFunction)BPy_IDArray_to_list, METH_NOARGS, | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	 "return the array as a list"}, | 
					
						
							| 
									
										
										
										
											2011-02-13 10:52:18 +00:00
										 |  |  | 	{NULL, NULL, 0, NULL} | 
					
						
							| 
									
										
										
										
											2009-11-17 12:54:29 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-09-29 17:08:11 +00:00
										 |  |  | static int BPy_IDArray_Len(BPy_IDArray *self) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	return self->prop->len; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-09-29 17:08:11 +00:00
										 |  |  | static PyObject *BPy_IDArray_GetItem(BPy_IDArray *self, int index) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	if (index < 0 || index >= self->prop->len) { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 		PyErr_SetString(PyExc_IndexError, "index out of range!"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return NULL; | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	switch (self->prop->subtype) { | 
					
						
							|  |  |  | 		case IDP_FLOAT: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			return PyFloat_FromDouble(((float *)IDP_Array(self->prop))[index]); | 
					
						
							| 
									
										
										
										
											2008-07-24 19:22:17 +00:00
										 |  |  | 		case IDP_DOUBLE: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			return PyFloat_FromDouble(((double *)IDP_Array(self->prop))[index]); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		case IDP_INT: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			return PyLong_FromLong((long)((int *)IDP_Array(self->prop))[index]); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	} | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 	PyErr_Format(PyExc_RuntimeError, | 
					
						
							|  |  |  | 	             "%s: invalid/corrupt array type '%d'!", | 
					
						
							|  |  |  | 	             __func__, self->prop->subtype); | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	return NULL; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | static int BPy_IDArray_SetItem(BPy_IDArray *self, int index, PyObject *value) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	int i; | 
					
						
							|  |  |  | 	float f; | 
					
						
							| 
									
										
										
										
											2008-07-24 19:22:17 +00:00
										 |  |  | 	double d; | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	if (index < 0 || index >= self->prop->len) { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 		PyErr_SetString(PyExc_RuntimeError, "index out of range!"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return -1; | 
					
						
							|  |  |  | 	} | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	switch (self->prop->subtype) { | 
					
						
							|  |  |  | 		case IDP_FLOAT: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			f = (float)PyFloat_AsDouble(value); | 
					
						
							|  |  |  | 			if (f == -1 && PyErr_Occurred()) { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 				PyErr_SetString(PyExc_TypeError, "expected a float"); | 
					
						
							|  |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			((float *)IDP_Array(self->prop))[index] = f; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 			break; | 
					
						
							| 
									
										
										
										
											2008-07-24 19:22:17 +00:00
										 |  |  | 		case IDP_DOUBLE: | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			d = PyFloat_AsDouble(value); | 
					
						
							|  |  |  | 			if (d == -1 && PyErr_Occurred()) { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 				PyErr_SetString(PyExc_TypeError, "expected a float"); | 
					
						
							|  |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			((double *)IDP_Array(self->prop))[index] = d; | 
					
						
							| 
									
										
										
										
											2008-07-24 19:22:17 +00:00
										 |  |  | 			break; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		case IDP_INT: | 
					
						
							| 
									
										
										
										
											2012-11-21 02:28:36 +00:00
										 |  |  | 			i = PyLong_AsLong(value); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			if (i == -1 && PyErr_Occurred()) { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 				PyErr_SetString(PyExc_TypeError, "expected an int type"); | 
					
						
							|  |  |  | 				return -1; | 
					
						
							|  |  |  | 			} | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			((int *)IDP_Array(self->prop))[index] = i; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 			break; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	return 0; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PySequenceMethods BPy_IDArray_Seq = { | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	(lenfunc) BPy_IDArray_Len,          /* inquiry sq_length */ | 
					
						
							|  |  |  | 	NULL,                               /* binaryfunc sq_concat */ | 
					
						
							|  |  |  | 	NULL,                               /* intargfunc sq_repeat */ | 
					
						
							|  |  |  | 	(ssizeargfunc)BPy_IDArray_GetItem,  /* intargfunc sq_item */ | 
					
						
							|  |  |  | 	NULL,                               /* intintargfunc sq_slice */ | 
					
						
							|  |  |  | 	(ssizeobjargproc)BPy_IDArray_SetItem, /* intobjargproc sq_ass_item */ | 
					
						
							|  |  |  | 	NULL,                               /* intintobjargproc sq_ass_slice */ | 
					
						
							|  |  |  | 	NULL,                               /* objobjproc sq_contains */ | 
					
						
							|  |  |  | 	/* Added in release 2.0 */ | 
					
						
							|  |  |  | 	NULL,                               /* binaryfunc sq_inplace_concat */ | 
					
						
							|  |  |  | 	NULL,                               /* intargfunc sq_inplace_repeat */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* sequence slice (get): idparr[a:b] */ | 
					
						
							|  |  |  | static PyObject *BPy_IDArray_slice(BPy_IDArray *self, int begin, int end) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	IDProperty *prop = self->prop; | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	PyObject *tuple; | 
					
						
							|  |  |  | 	int count; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	CLAMP(begin, 0, prop->len); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	if (end < 0) end = prop->len + end + 1; | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	CLAMP(end, 0, prop->len); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	begin = MIN2(begin, end); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	tuple = PyTuple_New(end - begin); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	switch (prop->subtype) { | 
					
						
							|  |  |  | 		case IDP_FLOAT: | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			float *array = (float *)IDP_Array(prop); | 
					
						
							| 
									
										
										
										
											2011-10-13 01:29:08 +00:00
										 |  |  | 			for (count = begin; count < end; count++) { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 				PyTuple_SET_ITEM(tuple, count - begin, PyFloat_FromDouble(array[count])); | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 			break; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		case IDP_DOUBLE: | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			double *array = (double *)IDP_Array(prop); | 
					
						
							| 
									
										
										
										
											2011-10-13 01:29:08 +00:00
										 |  |  | 			for (count = begin; count < end; count++) { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 				PyTuple_SET_ITEM(tuple, count - begin, PyFloat_FromDouble(array[count])); | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 			break; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		case IDP_INT: | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 			int *array = (int *)IDP_Array(prop); | 
					
						
							| 
									
										
										
										
											2011-10-13 01:29:08 +00:00
										 |  |  | 			for (count = begin; count < end; count++) { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 				PyTuple_SET_ITEM(tuple, count - begin, PyLong_FromLong(array[count])); | 
					
						
							|  |  |  | 			} | 
					
						
							|  |  |  | 			break; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return tuple; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | /* sequence slice (set): idparr[a:b] = value */ | 
					
						
							|  |  |  | static int BPy_IDArray_ass_slice(BPy_IDArray *self, int begin, int end, PyObject *seq) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	IDProperty *prop = self->prop; | 
					
						
							|  |  |  | 	short is_double = 0; | 
					
						
							|  |  |  | 	const PyTypeObject *py_type = idp_array_py_type(self, &is_double); | 
					
						
							|  |  |  | 	const size_t elem_size = is_double ? sizeof(double) : sizeof(float); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	size_t alloc_len; | 
					
						
							|  |  |  | 	size_t size; | 
					
						
							|  |  |  | 	void *vec; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	CLAMP(begin, 0, prop->len); | 
					
						
							|  |  |  | 	CLAMP(end, 0, prop->len); | 
					
						
							|  |  |  | 	begin = MIN2(begin, end); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	size = (end - begin); | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	alloc_len = size * elem_size; | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	vec = MEM_mallocN(alloc_len, "array assignment"); /* NOTE: we count on int/float being the same size here */ | 
					
						
							| 
									
										
										
										
											2011-10-13 01:29:08 +00:00
										 |  |  | 	if (PyC_AsArray(vec, seq, size, py_type, is_double, "slice assignment: ") == -1) { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 		MEM_freeN(vec); | 
					
						
							|  |  |  | 		return -1; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	memcpy((void *)(((char *)IDP_Array(prop)) + (begin * elem_size)), vec, alloc_len); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	MEM_freeN(vec); | 
					
						
							|  |  |  | 	return 0; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | static PyObject *BPy_IDArray_subscript(BPy_IDArray *self, PyObject *item) | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	if (PyIndex_Check(item)) { | 
					
						
							|  |  |  | 		Py_ssize_t i; | 
					
						
							|  |  |  | 		i = PyNumber_AsSsize_t(item, PyExc_IndexError); | 
					
						
							|  |  |  | 		if (i == -1 && PyErr_Occurred()) | 
					
						
							|  |  |  | 			return NULL; | 
					
						
							|  |  |  | 		if (i < 0) | 
					
						
							|  |  |  | 			i += self->prop->len; | 
					
						
							|  |  |  | 		return BPy_IDArray_GetItem(self, i); | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	else if (PySlice_Check(item)) { | 
					
						
							|  |  |  | 		Py_ssize_t start, stop, step, slicelength; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-11-05 13:48:42 +00:00
										 |  |  | 		if (PySlice_GetIndicesEx(item, self->prop->len, &start, &stop, &step, &slicelength) < 0) | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 			return NULL; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if (slicelength <= 0) { | 
					
						
							|  |  |  | 			return PyTuple_New(0); | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		else if (step == 1) { | 
					
						
							|  |  |  | 			return BPy_IDArray_slice(self, start, stop); | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		else { | 
					
						
							|  |  |  | 			PyErr_SetString(PyExc_TypeError, "slice steps not supported with vectors"); | 
					
						
							|  |  |  | 			return NULL; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	else { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  | 		             "vector indices must be integers, not %.200s", | 
					
						
							|  |  |  | 		             __func__, Py_TYPE(item)->tp_name); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 		return NULL; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | static int BPy_IDArray_ass_subscript(BPy_IDArray *self, PyObject *item, PyObject *value) | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 	if (PyIndex_Check(item)) { | 
					
						
							|  |  |  | 		Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError); | 
					
						
							|  |  |  | 		if (i == -1 && PyErr_Occurred()) | 
					
						
							|  |  |  | 			return -1; | 
					
						
							|  |  |  | 		if (i < 0) | 
					
						
							|  |  |  | 			i += self->prop->len; | 
					
						
							|  |  |  | 		return BPy_IDArray_SetItem(self, i, value); | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	else if (PySlice_Check(item)) { | 
					
						
							|  |  |  | 		Py_ssize_t start, stop, step, slicelength; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-11-05 13:48:42 +00:00
										 |  |  | 		if (PySlice_GetIndicesEx(item, self->prop->len, &start, &stop, &step, &slicelength) < 0) | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 			return -1; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if (step == 1) | 
					
						
							|  |  |  | 			return BPy_IDArray_ass_slice(self, start, stop, value); | 
					
						
							|  |  |  | 		else { | 
					
						
							|  |  |  | 			PyErr_SetString(PyExc_TypeError, "slice steps not supported with vectors"); | 
					
						
							|  |  |  | 			return -1; | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	else { | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 		PyErr_Format(PyExc_TypeError, | 
					
						
							|  |  |  | 		             "vector indices must be integers, not %.200s", | 
					
						
							|  |  |  | 		             Py_TYPE(item)->tp_name); | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 		return -1; | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyMappingMethods BPy_IDArray_AsMapping = { | 
					
						
							|  |  |  | 	(lenfunc)BPy_IDArray_Len, | 
					
						
							|  |  |  | 	(binaryfunc)BPy_IDArray_subscript, | 
					
						
							|  |  |  | 	(objobjargproc)BPy_IDArray_ass_subscript | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyTypeObject BPy_IDArray_Type = { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	PyVarObject_HEAD_INIT(NULL, 0) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/*  For printing, in format "<module>.<name>" */ | 
					
						
							| 
									
										
										
										
											2012-04-15 14:54:15 +00:00
										 |  |  | 	"IDPropertyArray",           /* char *tp_name; */ | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	sizeof(BPy_IDArray),       /* int tp_basicsize; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	0,                          /* tp_itemsize;  For allocation */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	/* Methods to implement standard operations */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	NULL,                       /* destructor tp_dealloc; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* printfunc tp_print; */ | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	NULL,     /* getattrfunc tp_getattr; */ | 
					
						
							|  |  |  | 	NULL,     /* setattrfunc tp_setattr; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* cmpfunc tp_compare; */ | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	(reprfunc)BPy_IDArray_repr,     /* reprfunc tp_repr; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	/* Method suites for standard classes */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	NULL,                       /* PyNumberMethods *tp_as_number; */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	&BPy_IDArray_Seq,           /* PySequenceMethods *tp_as_sequence; */ | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	&BPy_IDArray_AsMapping,     /* PyMappingMethods *tp_as_mapping; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	/* More standard operations (here for binary compatibility) */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	NULL,                       /* hashfunc tp_hash; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	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; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Flags to define presence of optional/expanded features ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	Py_TPFLAGS_DEFAULT,         /* long tp_flags; */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	NULL,                       /*  char *tp_doc;  Documentation string */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Assigned meaning in release 2.0 ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/* call function for all accessible objects */ | 
					
						
							|  |  |  | 	NULL,                       /* traverseproc tp_traverse; */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	/* delete references to contained objects */ | 
					
						
							|  |  |  | 	NULL,                       /* inquiry tp_clear; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/***  Assigned meaning in release 2.1 ***/ | 
					
						
							|  |  |  | 	/*** rich comparisons ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* richcmpfunc tp_richcompare; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/***  weak reference enabler ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	0,                          /* long tp_weaklistoffset; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Added in release 2.2 ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/*   Iterators */ | 
					
						
							|  |  |  | 	NULL,                       /* getiterfunc tp_iter; */ | 
					
						
							|  |  |  | 	NULL,                       /* iternextfunc tp_iternext; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Attribute descriptor and subclassing stuff ***/ | 
					
						
							|  |  |  | 	BPy_IDArray_methods,        /* struct PyMethodDef *tp_methods; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* struct PyMemberDef *tp_members; */ | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	BPy_IDArray_getseters,       /* struct PyGetSetDef *tp_getset; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00: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 17:08:11 +00:00
										 |  |  | static PyObject *IDGroup_Iter_repr(BPy_IDGroup_Iter *self) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							| 
									
										
										
										
											2010-10-13 23:25:08 +00:00
										 |  |  | 	return PyUnicode_FromFormat("(ID Property Group Iter \"%s\")", self->group->prop->name); | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2008-09-29 17:08:11 +00:00
										 |  |  | static PyObject *BPy_Group_Iter_Next(BPy_IDGroup_Iter *self) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | { | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	if (self->cur) { | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 		PyObject *ret; | 
					
						
							|  |  |  | 		IDProperty *cur; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 		cur = self->cur; | 
					
						
							|  |  |  | 		self->cur = self->cur->next; | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 		if (self->mode == IDPROP_ITER_ITEMS) { | 
					
						
							| 
									
										
										
										
											2009-11-16 20:16:45 +00:00
										 |  |  | 			ret = PyTuple_New(2); | 
					
						
							|  |  |  | 			PyTuple_SET_ITEM(ret, 0, PyUnicode_FromString(cur->name)); | 
					
						
							| 
									
										
										
										
											2011-11-15 10:19:44 +00:00
										 |  |  | 			PyTuple_SET_ITEM(ret, 1, BPy_IDGroup_WrapData(self->group->id, cur, self->group->prop)); | 
					
						
							| 
									
										
										
										
											2007-05-26 04:39:31 +00:00
										 |  |  | 			return ret; | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 		} | 
					
						
							|  |  |  | 		else { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 			return PyUnicode_FromString(cur->name); | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 		} | 
					
						
							| 
									
										
										
										
											2011-03-19 11:12:48 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | 	else { | 
					
						
							| 
									
										
										
										
											2010-11-23 16:45:17 +00:00
										 |  |  | 		PyErr_SetString(PyExc_StopIteration, "iterator at end"); | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 		return NULL; | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	} | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | PyTypeObject BPy_IDGroup_Iter_Type = { | 
					
						
							| 
									
										
										
										
											2009-11-16 18:56:58 +00:00
										 |  |  | 	PyVarObject_HEAD_INIT(NULL, 0) | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/*  For printing, in format "<module>.<name>" */ | 
					
						
							| 
									
										
										
										
											2012-04-15 14:54:15 +00:00
										 |  |  | 	"IDPropertyGroupIter",           /* char *tp_name; */ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	sizeof(BPy_IDGroup_Iter),       /* int tp_basicsize; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	0,                          /* tp_itemsize;  For allocation */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	/* Methods to implement standard operations */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	NULL,                       /* destructor tp_dealloc; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* printfunc tp_print; */ | 
					
						
							| 
									
										
										
										
											2006-12-16 23:54:45 +00:00
										 |  |  | 	NULL,     /* getattrfunc tp_getattr; */ | 
					
						
							|  |  |  | 	NULL,     /* setattrfunc tp_setattr; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* cmpfunc tp_compare; */ | 
					
						
							| 
									
										
										
										
											2012-03-16 21:39:56 +00:00
										 |  |  | 	(reprfunc) IDGroup_Iter_repr,     /* reprfunc tp_repr; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	/* Method suites for standard classes */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	NULL,                       /* PyNumberMethods *tp_as_number; */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	NULL,                       /* PySequenceMethods *tp_as_sequence; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00: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; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Flags to define presence of optional/expanded features ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	Py_TPFLAGS_DEFAULT,         /* long tp_flags; */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	NULL,                       /*  char *tp_doc;  Documentation string */ | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Assigned meaning in release 2.0 ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/* call function for all accessible objects */ | 
					
						
							|  |  |  | 	NULL,                       /* traverseproc tp_traverse; */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	/* delete references to contained objects */ | 
					
						
							|  |  |  | 	NULL,                       /* inquiry tp_clear; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/***  Assigned meaning in release 2.1 ***/ | 
					
						
							|  |  |  | 	/*** rich comparisons ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	NULL,                       /* richcmpfunc tp_richcompare; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/***  weak reference enabler ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	0,                          /* long tp_weaklistoffset; */ | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2012-03-26 06:55:09 +00:00
										 |  |  | 	/*** Added in release 2.2 ***/ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	/*   Iterators */ | 
					
						
							| 
									
										
										
										
											2012-03-09 00:01:38 +00:00
										 |  |  | 	PyObject_SelfIter,                  /* getiterfunc tp_iter; */ | 
					
						
							| 
									
										
										
										
											2006-11-17 04:46:48 +00:00
										 |  |  | 	(iternextfunc) BPy_Group_Iter_Next, /* iternextfunc tp_iternext; */ | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2006-11-29 23:31:46 +00:00
										 |  |  | void IDProp_Init_Types(void) | 
					
						
							|  |  |  | { | 
					
						
							| 
									
										
										
										
											2011-06-17 05:45:46 +00:00
										 |  |  | 	PyType_Ready(&BPy_IDGroup_Type); | 
					
						
							|  |  |  | 	PyType_Ready(&BPy_IDGroup_Iter_Type); | 
					
						
							|  |  |  | 	PyType_Ready(&BPy_IDArray_Type); | 
					
						
							| 
									
										
										
										
											2006-11-29 23:31:46 +00:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2012-04-15 14:54:15 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | /*----------------------------MODULE INIT-------------------------*/ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* --- */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static struct PyModuleDef IDProp_types_module_def = { | 
					
						
							| 
									
										
										
										
											2012-11-03 15:35:03 +00:00
										 |  |  | 	PyModuleDef_HEAD_INIT, | 
					
						
							|  |  |  | 	"idprop.types",  /* m_name */ | 
					
						
							|  |  |  | 	NULL,  /* m_doc */ | 
					
						
							|  |  |  | 	0,  /* m_size */ | 
					
						
							|  |  |  | 	NULL,  /* m_methods */ | 
					
						
							|  |  |  | 	NULL,  /* m_reload */ | 
					
						
							|  |  |  | 	NULL,  /* m_traverse */ | 
					
						
							|  |  |  | 	NULL,  /* m_clear */ | 
					
						
							|  |  |  | 	NULL,  /* m_free */ | 
					
						
							| 
									
										
										
										
											2012-04-15 14:54:15 +00:00
										 |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyObject *BPyInit_idprop_types(void) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  | 	PyObject *submodule; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	submodule = PyModule_Create(&IDProp_types_module_def); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #define MODULE_TYPE_ADD(s, t) \
 | 
					
						
							|  |  |  | 	PyModule_AddObject(s, t.tp_name, (PyObject *)&t); Py_INCREF((PyObject *)&t) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	/* bmesh_py_types.c */ | 
					
						
							|  |  |  | 	MODULE_TYPE_ADD(submodule, BPy_IDGroup_Type); | 
					
						
							|  |  |  | 	MODULE_TYPE_ADD(submodule, BPy_IDGroup_Iter_Type); | 
					
						
							|  |  |  | 	MODULE_TYPE_ADD(submodule, BPy_IDArray_Type); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #undef MODULE_TYPE_ADD
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return submodule; | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | /* --- */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | static PyMethodDef IDProp_methods[] = { | 
					
						
							|  |  |  | 	{NULL, NULL, 0, NULL} | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyDoc_STRVAR(IDProp_module_doc, | 
					
						
							|  |  |  | "This module provides access id property types (currently mainly for docs)." | 
					
						
							|  |  |  | ); | 
					
						
							|  |  |  | static struct PyModuleDef IDProp_module_def = { | 
					
						
							|  |  |  | 	PyModuleDef_HEAD_INIT, | 
					
						
							|  |  |  | 	"idprop",  /* m_name */ | 
					
						
							|  |  |  | 	IDProp_module_doc,  /* m_doc */ | 
					
						
							|  |  |  | 	0,  /* m_size */ | 
					
						
							|  |  |  | 	IDProp_methods,  /* m_methods */ | 
					
						
							|  |  |  | 	NULL,  /* m_reload */ | 
					
						
							|  |  |  | 	NULL,  /* m_traverse */ | 
					
						
							|  |  |  | 	NULL,  /* m_clear */ | 
					
						
							|  |  |  | 	NULL,  /* m_free */ | 
					
						
							|  |  |  | }; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | PyObject *BPyInit_idprop(void) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  | 	PyObject *mod; | 
					
						
							|  |  |  | 	PyObject *submodule; | 
					
						
							|  |  |  | 	PyObject *sys_modules = PyThreadState_GET()->interp->modules; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	mod = PyModule_Create(&IDProp_module_def); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	/* bmesh.types */ | 
					
						
							|  |  |  | 	PyModule_AddObject(mod, "types", (submodule = BPyInit_idprop_types())); | 
					
						
							|  |  |  | 	PyDict_SetItemString(sys_modules, PyModule_GetName(submodule), submodule); | 
					
						
							|  |  |  | 	Py_INCREF(submodule); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	return mod; | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2012-10-31 20:29:32 +00:00
										 |  |  | 
 | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #ifdef DEBUG
 | 
					
						
							|  |  |  | /* -------------------------------------------------------------------- */ | 
					
						
							|  |  |  | /* debug only function */ | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | void IDP_spit(IDProperty *prop) | 
					
						
							|  |  |  | { | 
					
						
							|  |  |  | 	if (prop) { | 
					
						
							|  |  |  | 		PyGILState_STATE gilstate; | 
					
						
							|  |  |  | 		int use_gil = TRUE; /* !PYC_INTERPRETER_ACTIVE; */ | 
					
						
							|  |  |  | 		PyObject *ret_dict; | 
					
						
							|  |  |  | 		PyObject *ret_str; | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if (use_gil) { | 
					
						
							|  |  |  | 			gilstate = PyGILState_Ensure(); | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		/* to_dict() */ | 
					
						
							|  |  |  | 		ret_dict = BPy_IDGroup_MapDataToPy(prop); | 
					
						
							|  |  |  | 		ret_str = PyObject_Repr(ret_dict); | 
					
						
							|  |  |  | 		Py_DECREF(ret_dict); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		printf("IDProperty: %s\n", _PyUnicode_AsString(ret_str)); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		Py_DECREF(ret_str); | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		if (use_gil) { | 
					
						
							|  |  |  | 			PyGILState_Release(gilstate); | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 	else { | 
					
						
							|  |  |  | 		printf("IDProperty: <NIL>\n"); | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | #endif
 |