915 lines
		
	
	
		
			27 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			915 lines
		
	
	
		
			27 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/**
 | 
						|
 * $Id: IDProp.c
 | 
						|
 *
 | 
						|
 * ***** BEGIN GPL/BL DUAL LICENSE BLOCK *****
 | 
						|
 *
 | 
						|
 * 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
 | 
						|
 * of the License, or (at your option) any later version. The Blender
 | 
						|
 * Foundation also sells licenses for use in proprietary software under
 | 
						|
 * the Blender License.  See http://www.blender.org/BL/ for information
 | 
						|
 * about this.
 | 
						|
 *
 | 
						|
 * 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,
 | 
						|
 * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 | 
						|
 *
 | 
						|
 * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
 | 
						|
 * All rights reserved.
 | 
						|
 *
 | 
						|
 * Contributor(s): Joseph Eagar
 | 
						|
 *
 | 
						|
 * ***** END GPL/BL DUAL LICENSE BLOCK *****
 | 
						|
 */
 | 
						|
 
 | 
						|
#include "DNA_ID.h"
 | 
						|
 | 
						|
#include "BKE_idprop.h"
 | 
						|
 | 
						|
#include "IDProp.h"
 | 
						|
#include "gen_utils.h"
 | 
						|
 | 
						|
#include "MEM_guardedalloc.h"
 | 
						|
 | 
						|
#define BSTR_EQ(a, b)	(*(a) == *(b) && !strcmp(a, b))
 | 
						|
 | 
						|
/*** Function to wrap ID properties ***/
 | 
						|
PyObject *BPy_Wrap_IDProperty(ID *id, IDProperty *prop, IDProperty *parent);
 | 
						|
 | 
						|
extern PyTypeObject IDArray_Type;
 | 
						|
extern PyTypeObject IDGroup_Iter_Type;
 | 
						|
 | 
						|
/*********************** ID Property Main Wrapper Stuff ***************/
 | 
						|
void IDGroup_dealloc( BPy_IDProperty *self )
 | 
						|
{
 | 
						|
	PyObject_DEL( self );
 | 
						|
}
 | 
						|
 | 
						|
PyObject *IDGroup_repr( BPy_IDProperty *self )
 | 
						|
{
 | 
						|
	return Py_BuildValue( "s", "(ID Property)" );
 | 
						|
}
 | 
						|
 | 
						|
extern PyTypeObject IDGroup_Type;
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_WrapData( ID *id, IDProperty *prop )
 | 
						|
{
 | 
						|
	switch ( prop->type ) {
 | 
						|
		case IDP_STRING:
 | 
						|
			return Py_BuildValue( "s", prop->data.pointer );
 | 
						|
		case IDP_INT:
 | 
						|
			return Py_BuildValue( "i", prop->data.val );
 | 
						|
		case IDP_FLOAT:
 | 
						|
			return Py_BuildValue( "f", *(float*)(&prop->data.val) );
 | 
						|
		case IDP_GROUP:
 | 
						|
			/*blegh*/
 | 
						|
			{
 | 
						|
				BPy_IDProperty *group = PyObject_New(BPy_IDProperty, &IDGroup_Type);
 | 
						|
				if (!group)
 | 
						|
					return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
					   "PyObject_New() failed" );
 | 
						|
			
 | 
						|
				group->id = id;
 | 
						|
				group->prop = prop;
 | 
						|
				return (PyObject*) group;
 | 
						|
			}
 | 
						|
		case IDP_ARRAY:
 | 
						|
			{
 | 
						|
				BPy_IDProperty *array = PyObject_New(BPy_IDProperty, &IDArray_Type);
 | 
						|
				if (!array)
 | 
						|
					return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
					   "PyObject_New() failed" );
 | 
						|
					   
 | 
						|
				array->id = id;
 | 
						|
				array->prop = prop;
 | 
						|
				return (PyObject*) array;
 | 
						|
			}
 | 
						|
		case IDP_MATRIX:
 | 
						|
		case IDP_VECTOR:
 | 
						|
			break;
 | 
						|
	}
 | 
						|
	Py_RETURN_NONE;
 | 
						|
}
 | 
						|
 | 
						|
int BPy_IDGroup_SetData(BPy_IDProperty *self, IDProperty *prop, PyObject *value)
 | 
						|
{
 | 
						|
	switch (prop->type) {
 | 
						|
		case IDP_STRING:
 | 
						|
		{
 | 
						|
			char *st;
 | 
						|
			if (!PyString_Check(value))
 | 
						|
				return EXPP_ReturnIntError(PyExc_TypeError, "expected a string!");
 | 
						|
 | 
						|
			st = PyString_AsString(value);
 | 
						|
			IDP_ResizeArray(prop, strlen(st)+1);
 | 
						|
			strcpy(prop->data.pointer, st);
 | 
						|
			return 0;
 | 
						|
		}
 | 
						|
 | 
						|
		case IDP_INT:
 | 
						|
		{
 | 
						|
			int ivalue;
 | 
						|
			if (!PyNumber_Check(value))
 | 
						|
				return EXPP_ReturnIntError(PyExc_TypeError, "expected an int!");
 | 
						|
			value = PyNumber_Int(value);
 | 
						|
			if (!value)
 | 
						|
				return EXPP_ReturnIntError(PyExc_TypeError, "expected an int!");
 | 
						|
			ivalue = (int) PyInt_AsLong(value);
 | 
						|
			prop->data.val = ivalue;
 | 
						|
			Py_XDECREF(value);
 | 
						|
			break;
 | 
						|
		}
 | 
						|
		case IDP_FLOAT:
 | 
						|
		{
 | 
						|
			float fvalue;
 | 
						|
			if (!PyNumber_Check(value))
 | 
						|
				return EXPP_ReturnIntError(PyExc_TypeError, "expected a float!");
 | 
						|
			value = PyNumber_Float(value);
 | 
						|
			if (!value)
 | 
						|
				return EXPP_ReturnIntError(PyExc_TypeError, "expected a float!");
 | 
						|
			fvalue = (float) PyFloat_AsDouble(value);
 | 
						|
			*(float*)&self->prop->data.val = fvalue;
 | 
						|
			Py_XDECREF(value);
 | 
						|
			break;
 | 
						|
		}
 | 
						|
 | 
						|
		default:
 | 
						|
			return EXPP_ReturnIntError(PyExc_AttributeError, "attempt to set read-only attribute!");
 | 
						|
	}
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_GetName(BPy_IDProperty *self)
 | 
						|
{
 | 
						|
	return Py_BuildValue("s", self->prop->name);
 | 
						|
}
 | 
						|
 | 
						|
int BPy_IDGroup_SetName(BPy_IDProperty *self, PyObject *value)
 | 
						|
{
 | 
						|
	char *st;
 | 
						|
	if (!PyString_Check(value))
 | 
						|
		return EXPP_ReturnIntError(PyExc_TypeError, "expected a string!");
 | 
						|
 | 
						|
	st = PyString_AsString(value);
 | 
						|
	if (strlen(st) >= MAX_IDPROP_NAME)
 | 
						|
		return EXPP_ReturnIntError(PyExc_TypeError, "string length cannot exceed 31 characters!");
 | 
						|
 | 
						|
	strcpy(self->prop->name, st);
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_GetType(BPy_IDProperty *self)
 | 
						|
{
 | 
						|
	return Py_BuildValue("i", self->prop->type);
 | 
						|
}
 | 
						|
 | 
						|
static PyGetSetDef BPy_IDGroup_getseters[] = {
 | 
						|
	{"name",
 | 
						|
	 (getter)BPy_IDGroup_GetName, (setter)BPy_IDGroup_SetName,
 | 
						|
	 "The name of this Group.",
 | 
						|
	 NULL},
 | 
						|
	 {NULL, NULL, NULL, NULL, NULL}
 | 
						|
};
 | 
						|
	 
 | 
						|
int BPy_IDGroup_Map_Len(BPy_IDProperty *self)
 | 
						|
{
 | 
						|
	if (self->prop->type != IDP_GROUP)
 | 
						|
		return EXPP_ReturnIntError( PyExc_TypeError,
 | 
						|
			"len() of unsized object");
 | 
						|
			
 | 
						|
	return self->prop->len;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_Map_GetItem(BPy_IDProperty *self, PyObject *item)
 | 
						|
{
 | 
						|
	IDProperty *loop;
 | 
						|
	char *st;
 | 
						|
	
 | 
						|
	if (self->prop->type  != IDP_GROUP)
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_TypeError,
 | 
						|
			"unsubscriptable object");
 | 
						|
			
 | 
						|
	if (!PyString_Check(item)) 
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_TypeError,
 | 
						|
			"only strings are allowed as keys of ID properties");
 | 
						|
	
 | 
						|
	st = PyString_AsString(item);
 | 
						|
	for (loop=self->prop->data.group.first; loop; loop=loop->next) {
 | 
						|
		if (BSTR_EQ(loop->name, st)) return BPy_IDGroup_WrapData(self->id, loop);
 | 
						|
	}
 | 
						|
	return EXPP_ReturnPyObjError( PyExc_KeyError,
 | 
						|
		"key not in subgroup dict");
 | 
						|
}
 | 
						|
 | 
						|
/*returns NULL on success, error string on failure*/
 | 
						|
char *BPy_IDProperty_Map_ValidateAndCreate(char *name, IDProperty *group, PyObject *ob)
 | 
						|
{
 | 
						|
	IDProperty *prop = NULL, *prop2=NULL, *prev=NULL;
 | 
						|
	IDPropertyTemplate val = {0};
 | 
						|
	
 | 
						|
	if (PyFloat_Check(ob)) {
 | 
						|
		val.f = (float) PyFloat_AsDouble(ob);
 | 
						|
		prop = IDP_New(IDP_FLOAT, val, name);
 | 
						|
	} else if (PyInt_Check(ob)) {
 | 
						|
		val.i = (int) PyInt_AsLong(ob);
 | 
						|
		prop = IDP_New(IDP_INT, val, name);
 | 
						|
	} else if (PyString_Check(ob)) {
 | 
						|
		val.str = PyString_AsString(ob);
 | 
						|
		prop = IDP_New(IDP_STRING, val, name);
 | 
						|
	} else if (PySequence_Check(ob)) {
 | 
						|
		PyObject *item;
 | 
						|
		int i;
 | 
						|
		
 | 
						|
		/*validate sequence and derive type.
 | 
						|
		we assume IDP_INT unless we hit a float
 | 
						|
		number; then we assume it's */
 | 
						|
		val.array.type = IDP_INT;
 | 
						|
		val.array.len = PySequence_Length(ob);
 | 
						|
		for (i=0; i<val.array.len; i++) {
 | 
						|
			item = PySequence_GetItem(ob, i);
 | 
						|
			if (PyFloat_Check(item)) val.array.type = IDP_FLOAT;
 | 
						|
			else if (!PyInt_Check(item)) return "only floats and ints are allowed in ID property arrays";
 | 
						|
			Py_XDECREF(item);
 | 
						|
		}
 | 
						|
		
 | 
						|
		prop = IDP_New(IDP_ARRAY, val, name);
 | 
						|
		for (i=0; i<val.array.len; i++) {
 | 
						|
			item = PySequence_GetItem(ob, i);
 | 
						|
			if (val.array.type == IDP_INT) {
 | 
						|
				item = PyNumber_Int(item);
 | 
						|
				((int*)prop->data.pointer)[i] = (int)PyInt_AsLong(item);
 | 
						|
			} else {
 | 
						|
				item = PyNumber_Float(item);
 | 
						|
				((float*)prop->data.pointer)[i] = (float)PyFloat_AsDouble(item);
 | 
						|
			}
 | 
						|
			Py_XDECREF(item);
 | 
						|
		}
 | 
						|
	} else if (PyMapping_Check(ob)) {
 | 
						|
		PyObject *keys, *vals, *key, *pval;
 | 
						|
		int i, len;
 | 
						|
		/*yay! we get into recursive stuff now!*/
 | 
						|
		keys = PyMapping_Keys(ob);
 | 
						|
		vals = PyMapping_Values(ob);
 | 
						|
		
 | 
						|
		/*we allocate the group first; if we hit any invalid data,
 | 
						|
		  we can delete it easily enough.*/
 | 
						|
		prop = IDP_New(IDP_GROUP, val, name);
 | 
						|
		len = PyMapping_Length(ob);
 | 
						|
		for (i=0; i<len; i++) {
 | 
						|
			key = PySequence_GetItem(keys, i);
 | 
						|
			pval = PySequence_GetItem(vals, i);
 | 
						|
			if (!PyString_Check(key)) {
 | 
						|
				IDP_FreeProperty(prop);
 | 
						|
				MEM_freeN(prop);
 | 
						|
				Py_XDECREF(keys);
 | 
						|
				Py_XDECREF(vals);
 | 
						|
				Py_XDECREF(key);
 | 
						|
				Py_XDECREF(pval);
 | 
						|
				return "invalid element in subgroup dict template!";
 | 
						|
			}
 | 
						|
			if (BPy_IDProperty_Map_ValidateAndCreate(PyString_AsString(key), prop, pval)) {
 | 
						|
				IDP_FreeProperty(prop);
 | 
						|
				MEM_freeN(prop);
 | 
						|
				Py_XDECREF(keys);
 | 
						|
				Py_XDECREF(vals);
 | 
						|
				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);
 | 
						|
	} else return "invalid property value";
 | 
						|
	
 | 
						|
	prop2 = IDP_GetPropertyFromGroup(group, prop->name);
 | 
						|
	if (prop2) {
 | 
						|
		prev=prop2->prev; /*we want to insert new property in same place as old*/
 | 
						|
		IDP_RemFromGroup(group, prop2);
 | 
						|
		IDP_FreeProperty(prop2);
 | 
						|
		MEM_freeN(prop2);
 | 
						|
	}
 | 
						|
 | 
						|
	IDP_InsertToGroup(group, prev, prop);
 | 
						|
	return NULL;
 | 
						|
}
 | 
						|
 | 
						|
int BPy_IDGroup_Map_SetItem(BPy_IDProperty *self, PyObject *key, PyObject *val)
 | 
						|
{
 | 
						|
	char *err;
 | 
						|
	
 | 
						|
	if (self->prop->type  != IDP_GROUP)
 | 
						|
		return EXPP_ReturnIntError( PyExc_TypeError,
 | 
						|
			"unsubscriptable object");
 | 
						|
			
 | 
						|
	if (!PyString_Check(key))
 | 
						|
		return EXPP_ReturnIntError( PyExc_TypeError,
 | 
						|
		   "only strings are allowed as subgroup keys" );
 | 
						|
 | 
						|
	if (val == NULL) {
 | 
						|
		IDProperty *pkey = IDP_GetPropertyFromGroup(self->prop, PyString_AsString(key));
 | 
						|
		if (pkey) {
 | 
						|
			IDP_RemFromGroup(self->prop, pkey);
 | 
						|
			IDP_FreeProperty(pkey);
 | 
						|
			MEM_freeN(pkey);
 | 
						|
			return 0;
 | 
						|
		} else return EXPP_ReturnIntError( PyExc_RuntimeError, "property not found in group" );
 | 
						|
	}
 | 
						|
	
 | 
						|
	err = BPy_IDProperty_Map_ValidateAndCreate(PyString_AsString(key), self->prop, val);
 | 
						|
	if (err) return EXPP_ReturnIntError( PyExc_RuntimeError, err );
 | 
						|
	
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_SpawnIterator(BPy_IDProperty *self)
 | 
						|
{
 | 
						|
	BPy_IDGroup_Iter *iter = PyObject_New(BPy_IDGroup_Iter, &IDGroup_Iter_Type);
 | 
						|
	
 | 
						|
	if (!iter)
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
		   "PyObject_New() failed" );
 | 
						|
	iter->group = self;
 | 
						|
	iter->mode = IDPROP_ITER_KEYS;
 | 
						|
	iter->cur = self->prop->data.group.first;
 | 
						|
	Py_XINCREF(iter);
 | 
						|
	return (PyObject*) iter;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_MapDataToPy(IDProperty *prop)
 | 
						|
{
 | 
						|
	switch (prop->type) {
 | 
						|
		case IDP_STRING:
 | 
						|
			return Py_BuildValue("s", prop->data.pointer);
 | 
						|
			break;
 | 
						|
		case IDP_FLOAT:
 | 
						|
			return Py_BuildValue("f", *((float*)&prop->data.val));
 | 
						|
			break;
 | 
						|
		case IDP_INT:
 | 
						|
			return Py_BuildValue("i", prop->data.val);
 | 
						|
			break;
 | 
						|
		case IDP_ARRAY:
 | 
						|
		{
 | 
						|
			PyObject *seq = PyList_New(prop->len);
 | 
						|
			int i;
 | 
						|
			
 | 
						|
			if (!seq) 
 | 
						|
				return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
					   "PyList_New() failed" );
 | 
						|
			
 | 
						|
			for (i=0; i<prop->len; i++) {
 | 
						|
				if (prop->subtype == IDP_FLOAT)
 | 
						|
					PyList_SetItem(seq, i, Py_BuildValue("f", ((float*)prop->data.pointer)[i]));
 | 
						|
				else PyList_SetItem(seq, i, Py_BuildValue("i",   ((int*)prop->data.pointer)[i]));
 | 
						|
			}
 | 
						|
			return seq;
 | 
						|
		}
 | 
						|
		case IDP_GROUP:
 | 
						|
		{
 | 
						|
			PyObject *dict = PyDict_New(), *wrap;
 | 
						|
			IDProperty *loop;
 | 
						|
			
 | 
						|
			if (!dict)
 | 
						|
				return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
					   "PyDict_New() failed" );
 | 
						|
					   
 | 
						|
			for (loop=prop->data.group.first; loop; loop=loop->next) {
 | 
						|
				wrap = BPy_IDGroup_MapDataToPy(loop);
 | 
						|
				if (!wrap) 
 | 
						|
					return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
					   "BPy_IDGroup_MapDataToPy() failed" );
 | 
						|
					   
 | 
						|
				PyDict_SetItemString(dict, loop->name, wrap);
 | 
						|
			}
 | 
						|
			return dict;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	
 | 
						|
	return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
					   "eek!! a property exists with a bad type code!!!" );
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_Pop(BPy_IDProperty *self, PyObject *vars)
 | 
						|
{
 | 
						|
	IDProperty *loop;
 | 
						|
	PyObject *pyform;
 | 
						|
	char *name;
 | 
						|
	int ok = PyArg_ParseTuple(vars, "s", &name);
 | 
						|
	
 | 
						|
	if (!ok) {
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_TypeError,
 | 
						|
		   "pop expected at least 1 arguments, got 0" );
 | 
						|
	}
 | 
						|
	
 | 
						|
	for (loop=self->prop->data.group.first; loop; loop=loop->next) {
 | 
						|
		if (BSTR_EQ(loop->name, name)) {
 | 
						|
			pyform = BPy_IDGroup_MapDataToPy(loop);
 | 
						|
			
 | 
						|
			if (!pyform)
 | 
						|
				/*ok something bad happened with the pyobject,
 | 
						|
				  so don't remove the prop from the group.  if pyform is
 | 
						|
				  NULL, then it already should have raised an exception.*/
 | 
						|
				  return NULL;
 | 
						|
 | 
						|
			IDP_RemFromGroup(self->prop, loop);
 | 
						|
			return pyform;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	
 | 
						|
	return EXPP_ReturnPyObjError( PyExc_KeyError,
 | 
						|
		   "item not in group" );
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_IterItems(BPy_IDProperty *self)
 | 
						|
{
 | 
						|
	BPy_IDGroup_Iter *iter = PyObject_New(BPy_IDGroup_Iter, &IDGroup_Iter_Type);
 | 
						|
	
 | 
						|
	if (!iter)
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
		   "PyObject_New() failed" );
 | 
						|
	
 | 
						|
	iter->group = self;
 | 
						|
	iter->mode = IDPROP_ITER_ITEMS;
 | 
						|
	iter->cur = self->prop->data.group.first;
 | 
						|
	Py_XINCREF(iter);
 | 
						|
	return (PyObject*) iter;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_GetKeys(BPy_IDProperty *self)
 | 
						|
{
 | 
						|
	PyObject *seq = PyList_New(self->prop->len), *st;
 | 
						|
	IDProperty *loop;
 | 
						|
	int i;
 | 
						|
 | 
						|
	if (!seq) 
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
		   "PyList_New() failed" );
 | 
						|
		   
 | 
						|
	for (i=0, loop=self->prop->data.group.first; loop; loop=loop->next, i++) {
 | 
						|
		st = Py_BuildValue("s", loop->name);
 | 
						|
		PyList_SetItem(seq, i, st);
 | 
						|
	}
 | 
						|
	
 | 
						|
	return seq;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_GetValues(BPy_IDProperty *self)
 | 
						|
{
 | 
						|
	PyObject *seq = PyList_New(self->prop->len);
 | 
						|
	IDProperty *loop;
 | 
						|
	int i;
 | 
						|
 | 
						|
	if (!seq) 
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
		   "PyList_New() failed" );
 | 
						|
	
 | 
						|
	for (i=0, loop=self->prop->data.group.first; loop; loop=loop->next, i++) {
 | 
						|
		PyList_SetItem(seq, i, BPy_IDGroup_WrapData(self->id, loop));
 | 
						|
	}
 | 
						|
	
 | 
						|
	return seq;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_HasKey(BPy_IDProperty *self, PyObject *vars)
 | 
						|
{
 | 
						|
	IDProperty *loop;
 | 
						|
	char *name;
 | 
						|
	
 | 
						|
	if (!PyArg_ParseTuple(vars, "s", &name))
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_TypeError,
 | 
						|
		   "expected a string");
 | 
						|
		   
 | 
						|
	for (loop=self->prop->data.group.first; loop; loop=loop->next) {
 | 
						|
		if (BSTR_EQ(loop->name, name)) Py_RETURN_TRUE;
 | 
						|
	}
 | 
						|
	
 | 
						|
	Py_RETURN_FALSE;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_Update(BPy_IDProperty *self, PyObject *vars)
 | 
						|
{
 | 
						|
	PyObject *pyob, *pkey, *pval;
 | 
						|
	int i=0;
 | 
						|
	
 | 
						|
	if (PySequence_Size(vars) != 1)
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_TypeError,
 | 
						|
		   "expected an object derived from dict.");
 | 
						|
	  
 | 
						|
	pyob = PyTuple_GET_ITEM(vars, 0);
 | 
						|
	if (!PyDict_Check(pyob))
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_TypeError,
 | 
						|
		   "expected an object derived from dict.");
 | 
						|
		   
 | 
						|
	while (PyDict_Next(pyob, &i, &pkey, &pval)) {
 | 
						|
		BPy_IDGroup_Map_SetItem(self, pkey, pval);
 | 
						|
		if (PyErr_Occurred()) return NULL;
 | 
						|
	}
 | 
						|
	
 | 
						|
	Py_RETURN_NONE;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDGroup_ConvertToPy(BPy_IDProperty *self)
 | 
						|
{
 | 
						|
	return BPy_IDGroup_MapDataToPy(self->prop);
 | 
						|
}
 | 
						|
 | 
						|
static struct PyMethodDef BPy_IDGroup_methods[] = {
 | 
						|
	{"pop", (PyCFunction)BPy_IDGroup_Pop, METH_VARARGS,
 | 
						|
		"pop an item from the group; raises KeyError if the item doesn't exist."},
 | 
						|
	{"iteritems", (PyCFunction)BPy_IDGroup_IterItems, METH_NOARGS,
 | 
						|
		"iterate through the items in the dict; behaves like dictionary method iteritems."},
 | 
						|
	{"keys", (PyCFunction)BPy_IDGroup_GetKeys, METH_NOARGS,
 | 
						|
		"get the keys associated with this group as a list of strings."},
 | 
						|
	{"values", (PyCFunction)BPy_IDGroup_GetValues, METH_NOARGS,
 | 
						|
		"get the values associated with this group."},
 | 
						|
	{"has_key", (PyCFunction)BPy_IDGroup_HasKey, METH_VARARGS,
 | 
						|
		"returns true if the group contains a key, false if not."},
 | 
						|
	{"update", (PyCFunction)BPy_IDGroup_Update, METH_VARARGS,
 | 
						|
		"updates the values in the group with the values of another or a dict."},
 | 
						|
	{"convert_to_pyobject", (PyCFunction)BPy_IDGroup_ConvertToPy, METH_NOARGS,
 | 
						|
		"return a purely python version of the group."},
 | 
						|
	{0, NULL, 0, NULL}
 | 
						|
};
 | 
						|
		
 | 
						|
PyMappingMethods BPy_IDGroup_Mapping = {
 | 
						|
	(inquiry)BPy_IDGroup_Map_Len, 			/*inquiry mp_length */
 | 
						|
	(binaryfunc)BPy_IDGroup_Map_GetItem,		/*binaryfunc mp_subscript */
 | 
						|
	(objobjargproc)BPy_IDGroup_Map_SetItem,	/*objobjargproc mp_ass_subscript */
 | 
						|
};
 | 
						|
 | 
						|
PyTypeObject IDGroup_Type = {
 | 
						|
	PyObject_HEAD_INIT( NULL )  /* required py macro */
 | 
						|
	0,                          /* ob_size */
 | 
						|
	/*  For printing, in format "<module>.<name>" */
 | 
						|
	"Blender IDProperty",           /* char *tp_name; */
 | 
						|
	sizeof( BPy_IDProperty ),       /* int tp_basicsize; */
 | 
						|
	0,                          /* tp_itemsize;  For allocation */
 | 
						|
 | 
						|
	/* Methods to implement standard operations */
 | 
						|
 | 
						|
	( destructor ) IDGroup_dealloc,/* destructor tp_dealloc; */
 | 
						|
	NULL,                       /* printfunc tp_print; */
 | 
						|
	NULL,     /* getattrfunc tp_getattr; */
 | 
						|
	NULL,     /* setattrfunc tp_setattr; */
 | 
						|
	NULL,                       /* cmpfunc tp_compare; */
 | 
						|
	( reprfunc ) IDGroup_repr,     /* reprfunc tp_repr; */
 | 
						|
 | 
						|
	/* Method suites for standard classes */
 | 
						|
 | 
						|
	NULL,                       /* PyNumberMethods *tp_as_number; */
 | 
						|
	NULL,	        			/* PySequenceMethods *tp_as_sequence; */
 | 
						|
	&BPy_IDGroup_Mapping,     /* PyMappingMethods *tp_as_mapping; */
 | 
						|
 | 
						|
	/* More standard operations (here for binary compatibility) */
 | 
						|
 | 
						|
	NULL,                       /* hashfunc tp_hash; */
 | 
						|
	NULL,                       /* ternaryfunc tp_call; */
 | 
						|
	NULL,                       /* reprfunc tp_str; */
 | 
						|
	NULL,                       /* getattrofunc tp_getattro; */
 | 
						|
	NULL,                       /* setattrofunc tp_setattro; */
 | 
						|
 | 
						|
	/* Functions to access object as input/output buffer */
 | 
						|
	NULL,                       /* PyBufferProcs *tp_as_buffer; */
 | 
						|
 | 
						|
  /*** Flags to define presence of optional/expanded features ***/
 | 
						|
	Py_TPFLAGS_DEFAULT,         /* long tp_flags; */
 | 
						|
 | 
						|
	NULL,                       /*  char *tp_doc;  Documentation string */
 | 
						|
  /*** Assigned meaning in release 2.0 ***/
 | 
						|
	/* call function for all accessible objects */
 | 
						|
	NULL,                       /* traverseproc tp_traverse; */
 | 
						|
 | 
						|
	/* delete references to contained objects */
 | 
						|
	NULL,                       /* inquiry tp_clear; */
 | 
						|
 | 
						|
  /***  Assigned meaning in release 2.1 ***/
 | 
						|
  /*** rich comparisons ***/
 | 
						|
	NULL,                       /* richcmpfunc tp_richcompare; */
 | 
						|
 | 
						|
  /***  weak reference enabler ***/
 | 
						|
	0,                          /* long tp_weaklistoffset; */
 | 
						|
 | 
						|
  /*** Added in release 2.2 ***/
 | 
						|
	/*   Iterators */
 | 
						|
	(getiterfunc)BPy_IDGroup_SpawnIterator, /* getiterfunc tp_iter; */
 | 
						|
	NULL,                       /* iternextfunc tp_iternext; */
 | 
						|
  /*** Attribute descriptor and subclassing stuff ***/
 | 
						|
	BPy_IDGroup_methods,        /* struct PyMethodDef *tp_methods; */
 | 
						|
	NULL,                       /* struct PyMemberDef *tp_members; */
 | 
						|
	BPy_IDGroup_getseters,       /* struct PyGetSetDef *tp_getset; */
 | 
						|
};
 | 
						|
 | 
						|
/*********** Main external wrapping function *******/
 | 
						|
PyObject *BPy_Wrap_IDProperty(ID *id, IDProperty *prop, IDProperty *parent)
 | 
						|
{
 | 
						|
	BPy_IDProperty *wrap = PyObject_New(BPy_IDProperty, &IDGroup_Type);
 | 
						|
	
 | 
						|
	if (!wrap)
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
		   "PyObject_New() failed" );
 | 
						|
						   
 | 
						|
	wrap->prop = prop;
 | 
						|
	wrap->parent = parent;
 | 
						|
	wrap->id = id;
 | 
						|
	//wrap->destroy = 0;
 | 
						|
	return (PyObject*) wrap;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/********Array Wrapper********/
 | 
						|
 | 
						|
void IDArray_dealloc(void *self)
 | 
						|
{
 | 
						|
	PyObject_DEL(self);
 | 
						|
}
 | 
						|
 | 
						|
PyObject *IDArray_repr(BPy_IDArray *self)
 | 
						|
{
 | 
						|
	return Py_BuildValue("s", "(ID Array)");
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
PyObject *BPy_IDArray_GetType(BPy_IDArray *self)
 | 
						|
{
 | 
						|
	return Py_BuildValue("i", self->prop->subtype);
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDArray_GetLen(BPy_IDArray *self)
 | 
						|
{
 | 
						|
	return Py_BuildValue("i", self->prop->len);
 | 
						|
}
 | 
						|
 | 
						|
static PyGetSetDef BPy_IDArray_getseters[] = {
 | 
						|
	{"len",
 | 
						|
	 (getter)BPy_IDArray_GetLen, (setter)NULL,
 | 
						|
	 "The length of the array, can also be gotten with len(array).",
 | 
						|
	 NULL},
 | 
						|
	{"type",
 | 
						|
	 (getter)BPy_IDArray_GetType, (setter)NULL,
 | 
						|
	 "The type of the data in the array, is an ant.",
 | 
						|
	 NULL},	
 | 
						|
	{NULL, NULL, NULL, NULL, NULL},
 | 
						|
};
 | 
						|
 | 
						|
int BPy_IDArray_Len(BPy_IDArray *self)
 | 
						|
{
 | 
						|
	return self->prop->len;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_IDArray_GetItem(BPy_IDArray *self, int index)
 | 
						|
{
 | 
						|
	if (index < 0 || index >= self->prop->len)
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_IndexError,
 | 
						|
				"index out of range!");
 | 
						|
 | 
						|
	switch (self->prop->subtype) {
 | 
						|
		case IDP_FLOAT:
 | 
						|
			return Py_BuildValue("f", ((float*)self->prop->data.pointer)[index]);
 | 
						|
			break;
 | 
						|
		case IDP_INT:
 | 
						|
			return Py_BuildValue("i", ((int*)self->prop->data.pointer)[index]);
 | 
						|
			break;
 | 
						|
	}
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_RuntimeError,
 | 
						|
				"invalid/corrupt array type!");
 | 
						|
}
 | 
						|
 | 
						|
int BPy_IDArray_SetItem(BPy_IDArray *self, int index, PyObject *val)
 | 
						|
{
 | 
						|
	int i;
 | 
						|
	float f;
 | 
						|
 | 
						|
	if (index < 0 || index >= self->prop->len)
 | 
						|
		return EXPP_ReturnIntError( PyExc_RuntimeError,
 | 
						|
				"index out of range!");
 | 
						|
 | 
						|
	switch (self->prop->subtype) {
 | 
						|
		case IDP_FLOAT:
 | 
						|
			if (!PyNumber_Check(val)) return EXPP_ReturnIntError( PyExc_TypeError,
 | 
						|
				"expected a float");
 | 
						|
			val = PyNumber_Float(val);
 | 
						|
			if (!val) return EXPP_ReturnIntError( PyExc_TypeError,
 | 
						|
				"expected a float");
 | 
						|
 | 
						|
			f = (float) PyFloat_AsDouble(val);
 | 
						|
			((float*)self->prop->data.pointer)[index] = f;
 | 
						|
			Py_XDECREF(val);
 | 
						|
			break;
 | 
						|
		case IDP_INT:
 | 
						|
			if (!PyNumber_Check(val)) return EXPP_ReturnIntError( PyExc_TypeError,
 | 
						|
				"expected an int");
 | 
						|
			val = PyNumber_Int(val);
 | 
						|
			if (!val) return EXPP_ReturnIntError( PyExc_TypeError,
 | 
						|
				"expected an int");
 | 
						|
 | 
						|
			i = (int) PyInt_AsLong(val);
 | 
						|
			((int*)self->prop->data.pointer)[index] = i;
 | 
						|
			Py_XDECREF(val);
 | 
						|
			break;
 | 
						|
	}
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static PySequenceMethods BPy_IDArray_Seq = {
 | 
						|
	(inquiry) BPy_IDArray_Len,			/* inquiry sq_length */
 | 
						|
	0,									/* binaryfunc sq_concat */
 | 
						|
	0,									/* intargfunc sq_repeat */
 | 
						|
	(intargfunc)BPy_IDArray_GetItem,	/* intargfunc sq_item */
 | 
						|
	0,									/* intintargfunc sq_slice */
 | 
						|
	(intobjargproc)BPy_IDArray_SetItem,	/* intobjargproc sq_ass_item */
 | 
						|
	0,									/* intintobjargproc sq_ass_slice */
 | 
						|
	0,									/* objobjproc sq_contains */
 | 
						|
				/* Added in release 2.0 */
 | 
						|
	0,									/* binaryfunc sq_inplace_concat */
 | 
						|
	0,									/* intargfunc sq_inplace_repeat */
 | 
						|
};
 | 
						|
 | 
						|
PyTypeObject IDArray_Type = {
 | 
						|
	PyObject_HEAD_INIT( NULL )  /* required py macro */
 | 
						|
	0,                          /* ob_size */
 | 
						|
	/*  For printing, in format "<module>.<name>" */
 | 
						|
	"Blender IDArray",           /* char *tp_name; */
 | 
						|
	sizeof( BPy_IDArray ),       /* int tp_basicsize; */
 | 
						|
	0,                          /* tp_itemsize;  For allocation */
 | 
						|
 | 
						|
	/* Methods to implement standard operations */
 | 
						|
 | 
						|
	( destructor ) IDArray_dealloc,/* destructor tp_dealloc; */
 | 
						|
	NULL,                       /* printfunc tp_print; */
 | 
						|
	NULL,     /* getattrfunc tp_getattr; */
 | 
						|
	NULL,     /* setattrfunc tp_setattr; */
 | 
						|
	NULL,                       /* cmpfunc tp_compare; */
 | 
						|
	( reprfunc ) IDArray_repr,     /* reprfunc tp_repr; */
 | 
						|
 | 
						|
	/* Method suites for standard classes */
 | 
						|
 | 
						|
	NULL,                       /* PyNumberMethods *tp_as_number; */
 | 
						|
	&BPy_IDArray_Seq,   			/* PySequenceMethods *tp_as_sequence; */
 | 
						|
	NULL,                       /* PyMappingMethods *tp_as_mapping; */
 | 
						|
 | 
						|
	/* More standard operations (here for binary compatibility) */
 | 
						|
 | 
						|
	NULL,                       /* hashfunc tp_hash; */
 | 
						|
	NULL,                       /* ternaryfunc tp_call; */
 | 
						|
	NULL,                       /* reprfunc tp_str; */
 | 
						|
	NULL,                       /* getattrofunc tp_getattro; */
 | 
						|
	NULL,                       /* setattrofunc tp_setattro; */
 | 
						|
 | 
						|
	/* Functions to access object as input/output buffer */
 | 
						|
	NULL,                       /* PyBufferProcs *tp_as_buffer; */
 | 
						|
 | 
						|
  /*** Flags to define presence of optional/expanded features ***/
 | 
						|
	Py_TPFLAGS_DEFAULT,         /* long tp_flags; */
 | 
						|
 | 
						|
	NULL,                       /*  char *tp_doc;  Documentation string */
 | 
						|
  /*** Assigned meaning in release 2.0 ***/
 | 
						|
	/* call function for all accessible objects */
 | 
						|
	NULL,                       /* traverseproc tp_traverse; */
 | 
						|
 | 
						|
	/* delete references to contained objects */
 | 
						|
	NULL,                       /* inquiry tp_clear; */
 | 
						|
 | 
						|
  /***  Assigned meaning in release 2.1 ***/
 | 
						|
  /*** rich comparisons ***/
 | 
						|
	NULL,                       /* richcmpfunc tp_richcompare; */
 | 
						|
 | 
						|
  /***  weak reference enabler ***/
 | 
						|
	0,                          /* long tp_weaklistoffset; */
 | 
						|
 | 
						|
  /*** Added in release 2.2 ***/
 | 
						|
	/*   Iterators */
 | 
						|
	NULL,                       /* getiterfunc tp_iter; */
 | 
						|
	NULL,                       /* iternextfunc tp_iternext; */
 | 
						|
 | 
						|
  /*** Attribute descriptor and subclassing stuff ***/
 | 
						|
	NULL,                       /* struct PyMethodDef *tp_methods; */
 | 
						|
	NULL,                       /* struct PyMemberDef *tp_members; */
 | 
						|
	BPy_IDArray_getseters,       /* struct PyGetSetDef *tp_getset; */
 | 
						|
	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 ********/
 | 
						|
void IDGroup_Iter_dealloc(void *self)
 | 
						|
{
 | 
						|
	PyObject_DEL(self);
 | 
						|
}
 | 
						|
 | 
						|
PyObject *IDGroup_Iter_iterself(PyObject *self)
 | 
						|
{
 | 
						|
	Py_XINCREF(self);
 | 
						|
	return self;
 | 
						|
}
 | 
						|
 | 
						|
PyObject *IDGroup_Iter_repr(BPy_IDGroup_Iter *self)
 | 
						|
{
 | 
						|
	return Py_BuildValue("s", "(ID Property Group)");
 | 
						|
}
 | 
						|
 | 
						|
PyObject *BPy_Group_Iter_Next(BPy_IDGroup_Iter *self)
 | 
						|
{
 | 
						|
	IDProperty *cur=NULL;
 | 
						|
 | 
						|
	if (self->cur) {
 | 
						|
		cur = self->cur;
 | 
						|
		self->cur = self->cur->next;
 | 
						|
		if (self->mode == IDPROP_ITER_ITEMS) {
 | 
						|
			return Py_BuildValue("[s, O]", cur->name, BPy_IDGroup_WrapData(self->group->id, cur));
 | 
						|
		} else {
 | 
						|
			return Py_BuildValue("s", cur->name);
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		return EXPP_ReturnPyObjError( PyExc_StopIteration,
 | 
						|
				"iterator at end" );
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
PyTypeObject IDGroup_Iter_Type = {
 | 
						|
	PyObject_HEAD_INIT( NULL )  /* required py macro */
 | 
						|
	0,                          /* ob_size */
 | 
						|
	/*  For printing, in format "<module>.<name>" */
 | 
						|
	"Blender IDGroup_Iter",           /* char *tp_name; */
 | 
						|
	sizeof( BPy_IDGroup_Iter ),       /* int tp_basicsize; */
 | 
						|
	0,                          /* tp_itemsize;  For allocation */
 | 
						|
 | 
						|
	/* Methods to implement standard operations */
 | 
						|
 | 
						|
	( destructor ) IDGroup_Iter_dealloc,/* destructor tp_dealloc; */
 | 
						|
	NULL,                       /* printfunc tp_print; */
 | 
						|
	NULL,     /* getattrfunc tp_getattr; */
 | 
						|
	NULL,     /* setattrfunc tp_setattr; */
 | 
						|
	NULL,                       /* cmpfunc tp_compare; */
 | 
						|
	( reprfunc ) IDGroup_Iter_repr,     /* reprfunc tp_repr; */
 | 
						|
 | 
						|
	/* Method suites for standard classes */
 | 
						|
 | 
						|
	NULL,                       /* PyNumberMethods *tp_as_number; */
 | 
						|
	NULL,	        			/* PySequenceMethods *tp_as_sequence; */
 | 
						|
	NULL,                       /* PyMappingMethods *tp_as_mapping; */
 | 
						|
 | 
						|
	/* More standard operations (here for binary compatibility) */
 | 
						|
 | 
						|
	NULL,                       /* hashfunc tp_hash; */
 | 
						|
	NULL,                       /* ternaryfunc tp_call; */
 | 
						|
	NULL,                       /* reprfunc tp_str; */
 | 
						|
	NULL,                       /* getattrofunc tp_getattro; */
 | 
						|
	NULL,                       /* setattrofunc tp_setattro; */
 | 
						|
 | 
						|
	/* Functions to access object as input/output buffer */
 | 
						|
	NULL,                       /* PyBufferProcs *tp_as_buffer; */
 | 
						|
 | 
						|
  /*** Flags to define presence of optional/expanded features ***/
 | 
						|
	Py_TPFLAGS_DEFAULT,         /* long tp_flags; */
 | 
						|
 | 
						|
	NULL,                       /*  char *tp_doc;  Documentation string */
 | 
						|
  /*** Assigned meaning in release 2.0 ***/
 | 
						|
	/* call function for all accessible objects */
 | 
						|
	NULL,                       /* traverseproc tp_traverse; */
 | 
						|
 | 
						|
	/* delete references to contained objects */
 | 
						|
	NULL,                       /* inquiry tp_clear; */
 | 
						|
 | 
						|
  /***  Assigned meaning in release 2.1 ***/
 | 
						|
  /*** rich comparisons ***/
 | 
						|
	NULL,                       /* richcmpfunc tp_richcompare; */
 | 
						|
 | 
						|
  /***  weak reference enabler ***/
 | 
						|
	0,                          /* long tp_weaklistoffset; */
 | 
						|
 | 
						|
  /*** Added in release 2.2 ***/
 | 
						|
	/*   Iterators */
 | 
						|
	IDGroup_Iter_iterself,              /* getiterfunc tp_iter; */
 | 
						|
	(iternextfunc) BPy_Group_Iter_Next, /* iternextfunc tp_iternext; */
 | 
						|
};
 | 
						|
 | 
						|
void IDProp_Init_Types(void)
 | 
						|
{
 | 
						|
	PyType_Ready( &IDGroup_Type );
 | 
						|
	PyType_Ready( &IDGroup_Iter_Type );
 | 
						|
	PyType_Ready( &IDArray_Type );
 | 
						|
}
 |