2011-02-23 11:52:22 +01:00
|
|
|
/*
|
2008-04-17 00:40:48 +02:00
|
|
|
* ***** BEGIN GPL LICENSE BLOCK *****
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU General Public License
|
|
|
|
* as published by the Free Software Foundation; either version 2
|
2008-04-17 00:40:48 +02:00
|
|
|
* of the License, or (at your option) any later version.
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program; if not, write to the Free Software Foundation,
|
2010-02-12 14:34:04 +01:00
|
|
|
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
|
|
|
*
|
2009-11-16 19:56:58 +01:00
|
|
|
* Contributor(s): Joseph Eagar, Campbell Barton
|
2006-11-30 00:31:46 +01:00
|
|
|
*
|
2008-04-17 00:40:48 +02:00
|
|
|
* ***** END GPL LICENSE BLOCK *****
|
2006-11-30 00:31:46 +01:00
|
|
|
*/
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
/** \file blender/python/generic/idprop_py_api.c
|
2011-02-27 21:10:08 +01:00
|
|
|
* \ingroup pygen
|
|
|
|
*/
|
|
|
|
|
2011-02-14 05:15:25 +01:00
|
|
|
#include <Python.h>
|
|
|
|
|
2011-01-07 19:36:47 +01:00
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
|
|
|
|
#include "BLI_utildefines.h"
|
2018-05-04 07:26:42 +02:00
|
|
|
#include "BLI_string.h"
|
2010-12-06 00:50:55 +01:00
|
|
|
|
2013-10-17 04:57:59 +02:00
|
|
|
#include "idprop_py_api.h"
|
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
#include "BKE_idprop.h"
|
2011-01-07 20:18:31 +01:00
|
|
|
|
2010-09-01 16:13:48 +02:00
|
|
|
#define USE_STRING_COERCE
|
|
|
|
|
|
|
|
#ifdef USE_STRING_COERCE
|
|
|
|
#include "py_capi_utils.h"
|
|
|
|
#endif
|
|
|
|
|
2015-06-17 22:11:02 +02:00
|
|
|
#include "python_utildefines.h"
|
2015-01-06 06:42:22 +01:00
|
|
|
|
Datablock ID Properties
The absence of datablock properties "will certainly be resolved soon as the need for them is becoming obvious" said the [[http://wiki.blender.org/index.php/Dev:Ref/Release_Notes/2.67/Python_Nodes|Python Nodes release notes]]. So this patch allows Python scripts to create ID Properties which reference datablocks.
This functionality is implemented for `PointerProperty` and now such properties can be created with Python.
In addition to the standard update callback, `PointerProperty` can have a `poll` callback (standard RNA) which is useful for search menus. For details see the test included in this patch.
Original author: @artfunkel
Alexander (Blend4Web Team)
Reviewers: brecht, artfunkel, mont29, campbellbarton
Reviewed By: mont29, campbellbarton
Subscribers: jta, sergey, campbellbarton, wisaac, poseidon4o, mont29, homyachetser, Evgeny_Rodygin, AlexKowel, yurikovelenov, fjuhec, sharlybg, cardboard, duarteframos, blueprintrandom, a.romanov, BYOB, disnel, aditiapratama, bliblubli, dfelinto, lukastoenne
Maniphest Tasks: T37754
Differential Revision: https://developer.blender.org/D113
2017-04-13 11:30:03 +02:00
|
|
|
extern bool pyrna_id_FromPyObject(PyObject *obj, ID **id);
|
|
|
|
extern PyObject *pyrna_id_CreatePyObject(ID *id);
|
|
|
|
extern bool pyrna_id_CheckPyObject(PyObject *obj);
|
2015-01-06 06:42:22 +01:00
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
/*********************** ID Property Main Wrapper Stuff ***************/
|
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
/* ----------------------------------------------------------------------------
|
|
|
|
* static conversion functions to avoid duplicate code, no type checking.
|
|
|
|
*/
|
|
|
|
|
2012-11-03 19:23:30 +01:00
|
|
|
static PyObject *idprop_py_from_idp_string(const IDProperty *prop)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2011-11-15 11:19:44 +01:00
|
|
|
if (prop->subtype == IDP_STRING_SUB_BYTE) {
|
2012-11-03 19:23:30 +01:00
|
|
|
return PyBytes_FromStringAndSize(IDP_String(prop), prop->len);
|
2011-11-15 11:19:44 +01:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
#ifdef USE_STRING_COERCE
|
|
|
|
return PyC_UnicodeFromByteAndSize(IDP_Array(prop), prop->len - 1);
|
|
|
|
#else
|
2012-11-03 19:23:30 +01:00
|
|
|
return PyUnicode_FromStringAndSize(IDP_String(prop), prop->len - 1);
|
2011-11-15 11:19:44 +01:00
|
|
|
#endif
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2012-11-03 19:23:30 +01:00
|
|
|
static PyObject *idprop_py_from_idp_int(const IDProperty *prop)
|
2011-06-17 07:45:46 +02:00
|
|
|
{
|
2012-11-03 19:23:30 +01:00
|
|
|
return PyLong_FromLong((long)IDP_Int(prop));
|
2011-06-17 07:45:46 +02:00
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
2012-11-03 19:23:30 +01:00
|
|
|
static PyObject *idprop_py_from_idp_float(const IDProperty *prop)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2012-11-03 19:23:30 +01:00
|
|
|
return PyFloat_FromDouble((double)IDP_Float(prop));
|
2011-11-15 11:19:44 +01:00
|
|
|
}
|
2011-11-15 10:12:10 +01:00
|
|
|
|
2012-11-03 19:23:30 +01:00
|
|
|
static PyObject *idprop_py_from_idp_double(const IDProperty *prop)
|
2011-11-15 11:19:44 +01:00
|
|
|
{
|
2012-11-03 19:23:30 +01:00
|
|
|
return PyFloat_FromDouble(IDP_Double(prop));
|
2011-11-15 11:19:44 +01:00
|
|
|
}
|
2011-11-15 10:12:10 +01:00
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
static PyObject *idprop_py_from_idp_group(ID *id, IDProperty *prop, IDProperty *parent)
|
|
|
|
{
|
2012-03-16 22:39:56 +01:00
|
|
|
BPy_IDProperty *group = PyObject_New(BPy_IDProperty, &BPy_IDGroup_Type);
|
2011-11-15 11:19:44 +01:00
|
|
|
group->id = id;
|
|
|
|
group->prop = prop;
|
|
|
|
group->parent = parent; /* can be NULL */
|
2012-03-26 08:55:09 +02:00
|
|
|
return (PyObject *)group;
|
2011-11-15 11:19:44 +01:00
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
|
Datablock ID Properties
The absence of datablock properties "will certainly be resolved soon as the need for them is becoming obvious" said the [[http://wiki.blender.org/index.php/Dev:Ref/Release_Notes/2.67/Python_Nodes|Python Nodes release notes]]. So this patch allows Python scripts to create ID Properties which reference datablocks.
This functionality is implemented for `PointerProperty` and now such properties can be created with Python.
In addition to the standard update callback, `PointerProperty` can have a `poll` callback (standard RNA) which is useful for search menus. For details see the test included in this patch.
Original author: @artfunkel
Alexander (Blend4Web Team)
Reviewers: brecht, artfunkel, mont29, campbellbarton
Reviewed By: mont29, campbellbarton
Subscribers: jta, sergey, campbellbarton, wisaac, poseidon4o, mont29, homyachetser, Evgeny_Rodygin, AlexKowel, yurikovelenov, fjuhec, sharlybg, cardboard, duarteframos, blueprintrandom, a.romanov, BYOB, disnel, aditiapratama, bliblubli, dfelinto, lukastoenne
Maniphest Tasks: T37754
Differential Revision: https://developer.blender.org/D113
2017-04-13 11:30:03 +02:00
|
|
|
static PyObject *idprop_py_from_idp_id(IDProperty *prop)
|
|
|
|
{
|
|
|
|
return pyrna_id_CreatePyObject(prop->data.pointer);
|
|
|
|
}
|
|
|
|
|
2011-11-15 11:19:44 +01: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 08:55:09 +02:00
|
|
|
return (PyObject *)array;
|
2011-11-15 11:19:44 +01:00
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
static PyObject *idprop_py_from_idp_idparray(ID *id, IDProperty *prop)
|
|
|
|
{
|
2016-07-31 08:59:08 +02:00
|
|
|
PyObject *seq = PyList_New(prop->len);
|
2012-03-16 22:39:56 +01:00
|
|
|
IDProperty *array = IDP_IDPArray(prop);
|
2011-11-15 11:19:44 +01:00
|
|
|
int i;
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
if (!seq) {
|
|
|
|
PyErr_Format(PyExc_RuntimeError,
|
|
|
|
"%s: IDP_IDPARRAY: PyList_New(%d) failed",
|
|
|
|
__func__, prop->len);
|
|
|
|
return NULL;
|
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
for (i = 0; i < prop->len; i++) {
|
2016-07-31 08:59:08 +02:00
|
|
|
PyObject *wrap = BPy_IDGroup_WrapData(id, array++, prop);
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2016-07-31 08:59:08 +02:00
|
|
|
/* BPy_IDGroup_MapDataToPy sets the error */
|
|
|
|
if (UNLIKELY(wrap == NULL)) {
|
|
|
|
Py_DECREF(seq);
|
2011-11-15 11:19:44 +01:00
|
|
|
return NULL;
|
2016-07-31 08:59:08 +02:00
|
|
|
}
|
2011-11-15 11:19:44 +01:00
|
|
|
|
|
|
|
PyList_SET_ITEM(seq, i, wrap);
|
|
|
|
}
|
|
|
|
|
|
|
|
return seq;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* -------------------------------------------------------------------------- */
|
|
|
|
|
|
|
|
/* use for both array and group */
|
2012-02-26 18:20:36 +01:00
|
|
|
static Py_hash_t BPy_IDGroup_hash(BPy_IDProperty *self)
|
2011-11-15 11:19:44 +01:00
|
|
|
{
|
|
|
|
return _Py_HashPointer(self->prop);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *BPy_IDGroup_repr(BPy_IDProperty *self)
|
|
|
|
{
|
2012-04-11 10:37:48 +02: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 11:19:44 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
PyObject *BPy_IDGroup_WrapData(ID *id, IDProperty *prop, IDProperty *parent)
|
|
|
|
{
|
|
|
|
switch (prop->type) {
|
2012-03-16 22:39:56 +01: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 */
|
Datablock ID Properties
The absence of datablock properties "will certainly be resolved soon as the need for them is becoming obvious" said the [[http://wiki.blender.org/index.php/Dev:Ref/Release_Notes/2.67/Python_Nodes|Python Nodes release notes]]. So this patch allows Python scripts to create ID Properties which reference datablocks.
This functionality is implemented for `PointerProperty` and now such properties can be created with Python.
In addition to the standard update callback, `PointerProperty` can have a `poll` callback (standard RNA) which is useful for search menus. For details see the test included in this patch.
Original author: @artfunkel
Alexander (Blend4Web Team)
Reviewers: brecht, artfunkel, mont29, campbellbarton
Reviewed By: mont29, campbellbarton
Subscribers: jta, sergey, campbellbarton, wisaac, poseidon4o, mont29, homyachetser, Evgeny_Rodygin, AlexKowel, yurikovelenov, fjuhec, sharlybg, cardboard, duarteframos, blueprintrandom, a.romanov, BYOB, disnel, aditiapratama, bliblubli, dfelinto, lukastoenne
Maniphest Tasks: T37754
Differential Revision: https://developer.blender.org/D113
2017-04-13 11:30:03 +02:00
|
|
|
case IDP_ID: return idprop_py_from_idp_id(prop);
|
2012-03-16 22:39:56 +01:00
|
|
|
default: Py_RETURN_NONE;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
2012-07-17 01:23:33 +02:00
|
|
|
#if 0 /* UNUSED, currently assignment overwrites into new properties, rather than setting in-place */
|
2011-02-13 11:52:18 +01:00
|
|
|
static int BPy_IDGroup_SetData(BPy_IDProperty *self, IDProperty *prop, PyObject *value)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
switch (prop->type) {
|
|
|
|
case IDP_STRING:
|
|
|
|
{
|
|
|
|
char *st;
|
2009-11-16 19:56:58 +01:00
|
|
|
if (!PyUnicode_Check(value)) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a string!");
|
|
|
|
return -1;
|
|
|
|
}
|
2011-11-15 11:19:44 +01:00
|
|
|
/* NOTE: if this code is enabled, bytes support needs to be added */
|
2010-09-01 16:13:48 +02:00
|
|
|
#ifdef USE_STRING_COERCE
|
|
|
|
{
|
|
|
|
int alloc_len;
|
2012-03-16 22:39:56 +01:00
|
|
|
PyObject *value_coerce = NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
st = (char *)PyC_UnicodeAsByte(value, &value_coerce);
|
|
|
|
alloc_len = strlen(st) + 1;
|
2010-09-01 16:13:48 +02:00
|
|
|
|
|
|
|
st = _PyUnicode_AsString(value);
|
|
|
|
IDP_ResizeArray(prop, alloc_len);
|
2011-06-17 07:45:46 +02:00
|
|
|
memcpy(IDP_Array(prop), st, alloc_len);
|
2010-09-01 16:13:48 +02:00
|
|
|
Py_XDECREF(value_coerce);
|
|
|
|
}
|
|
|
|
#else
|
2009-11-16 19:56:58 +01:00
|
|
|
st = _PyUnicode_AsString(value);
|
2012-03-26 08:55:09 +02:00
|
|
|
IDP_ResizeArray(prop, strlen(st) + 1);
|
2011-06-17 07:45:46 +02:00
|
|
|
strcpy(IDP_Array(prop), st);
|
2010-09-01 16:13:48 +02:00
|
|
|
#endif
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
case IDP_INT:
|
|
|
|
{
|
2012-03-16 22:39:56 +01:00
|
|
|
int ivalue = PyLong_AsSsize_t(value);
|
|
|
|
if (ivalue == -1 && PyErr_Occurred()) {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "expected an int type");
|
|
|
|
return -1;
|
|
|
|
}
|
2012-11-03 19:23:30 +01:00
|
|
|
IDP_Int(prop) = ivalue;
|
2006-12-17 00:54:45 +01:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
case IDP_FLOAT:
|
|
|
|
{
|
2012-03-16 22:39:56 +01:00
|
|
|
float fvalue = (float)PyFloat_AsDouble(value);
|
|
|
|
if (fvalue == -1 && PyErr_Occurred()) {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a float");
|
|
|
|
return -1;
|
|
|
|
}
|
2012-11-03 19:23:30 +01:00
|
|
|
IDP_Float(self->prop) = fvalue;
|
2006-12-17 00:54:45 +01:00
|
|
|
break;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
|
|
|
{
|
2012-03-16 22:39:56 +01:00
|
|
|
double dvalue = PyFloat_AsDouble(value);
|
|
|
|
if (dvalue == -1 && PyErr_Occurred()) {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a float");
|
|
|
|
return -1;
|
|
|
|
}
|
2012-11-03 19:23:30 +01:00
|
|
|
IDP_Double(self->prop) = dvalue;
|
2008-07-24 21:22:17 +02:00
|
|
|
break;
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
default:
|
2009-11-16 19:56:58 +01:00
|
|
|
PyErr_SetString(PyExc_AttributeError, "attempt to set read-only attribute!");
|
|
|
|
return -1;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
return 0;
|
|
|
|
}
|
2011-02-25 03:27:54 +01:00
|
|
|
#endif
|
2006-11-17 05:46:48 +01:00
|
|
|
|
2011-02-13 11:52:18 +01:00
|
|
|
static PyObject *BPy_IDGroup_GetName(BPy_IDProperty *self, void *UNUSED(closure))
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyUnicode_FromString(self->prop->name);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2010-10-14 01:25:08 +02:00
|
|
|
static int BPy_IDGroup_SetName(BPy_IDProperty *self, PyObject *value, void *UNUSED(closure))
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2011-11-03 15:09:18 +01:00
|
|
|
const char *name;
|
|
|
|
Py_ssize_t name_size;
|
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
if (!PyUnicode_Check(value)) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "expected a string!");
|
|
|
|
return -1;
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2011-11-03 15:09:18 +01:00
|
|
|
name = _PyUnicode_AsStringAndSize(value, &name_size);
|
|
|
|
|
|
|
|
if (name_size > MAX_IDPROP_NAME) {
|
2012-01-11 09:51:06 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "string length cannot exceed 63 characters!");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2011-11-03 15:09:18 +01:00
|
|
|
memcpy(self->prop->name, name, name_size);
|
2006-12-17 00:54:45 +01:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
#if 0
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_GetType(BPy_IDProperty *self)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2012-11-21 03:28:36 +01:00
|
|
|
return PyLong_FromLong(self->prop->type);
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
#endif
|
2006-12-17 00:54:45 +01:00
|
|
|
|
|
|
|
static PyGetSetDef BPy_IDGroup_getseters[] = {
|
2010-12-03 18:05:21 +01:00
|
|
|
{(char *)"name", (getter)BPy_IDGroup_GetName, (setter)BPy_IDGroup_SetName, (char *)"The name of this Group.", NULL},
|
2012-03-26 08:55:09 +02:00
|
|
|
{NULL, NULL, NULL, NULL, NULL}
|
2006-12-17 00:54:45 +01:00
|
|
|
};
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
static Py_ssize_t BPy_IDGroup_Map_Len(BPy_IDProperty *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
if (self->prop->type != IDP_GROUP) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "len() of unsized object");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
return self->prop->len;
|
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_Map_GetItem(BPy_IDProperty *self, PyObject *item)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
IDProperty *idprop;
|
2011-11-03 15:09:18 +01:00
|
|
|
const char *name;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-05-20 21:49:27 +02:00
|
|
|
if (self->prop->type != IDP_GROUP) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "unsubscriptable object");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
name = _PyUnicode_AsString(item);
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
if (name == NULL) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "only strings are allowed as keys of ID properties");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
idprop = IDP_GetPropertyFromGroup(self->prop, name);
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
if (idprop == NULL) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_KeyError, "key not in subgroup dict");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
return BPy_IDGroup_WrapData(self->id, idprop, self->prop);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
/* returns NULL on success, error string on failure */
|
2014-12-18 02:06:57 +01:00
|
|
|
static char idp_sequence_type(PyObject *seq_fast)
|
2010-03-02 12:35:28 +01:00
|
|
|
{
|
2015-07-29 01:58:10 +02:00
|
|
|
PyObject **seq_fast_items = PySequence_Fast_ITEMS(seq_fast);
|
2010-03-02 12:35:28 +01:00
|
|
|
PyObject *item;
|
2014-12-18 02:06:57 +01:00
|
|
|
char type = IDP_INT;
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2012-10-12 05:59:50 +02:00
|
|
|
Py_ssize_t i, len = PySequence_Fast_GET_SIZE(seq_fast);
|
2015-07-29 01:58:10 +02:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
for (i = 0; i < len; i++) {
|
2015-07-29 01:58:10 +02:00
|
|
|
item = seq_fast_items[i];
|
2010-03-02 12:35:28 +01:00
|
|
|
if (PyFloat_Check(item)) {
|
2011-10-13 03:29:08 +02:00
|
|
|
if (type == IDP_IDPARRAY) { /* mixed dict/int */
|
2010-03-02 12:35:28 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2012-03-16 22:39:56 +01:00
|
|
|
type = IDP_DOUBLE;
|
2010-03-02 12:35:28 +01:00
|
|
|
}
|
|
|
|
else if (PyLong_Check(item)) {
|
2011-10-13 03:29:08 +02:00
|
|
|
if (type == IDP_IDPARRAY) { /* mixed dict/int */
|
2010-03-02 12:35:28 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
2010-03-02 15:57:28 +01:00
|
|
|
else if (PyMapping_Check(item)) {
|
2011-10-13 03:29:08 +02:00
|
|
|
if (i != 0 && (type != IDP_IDPARRAY)) { /* mixed dict/int */
|
2010-03-02 12:35:28 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2012-03-16 22:39:56 +01:00
|
|
|
type = IDP_IDPARRAY;
|
2010-03-02 12:35:28 +01:00
|
|
|
}
|
|
|
|
else {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return type;
|
|
|
|
}
|
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
static const char *idp_try_read_name(PyObject *name_obj)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2017-04-10 13:06:00 +02:00
|
|
|
const char *name = NULL;
|
2011-11-03 15:09:18 +01:00
|
|
|
if (name_obj) {
|
|
|
|
Py_ssize_t name_size;
|
|
|
|
name = _PyUnicode_AsStringAndSize(name_obj, &name_size);
|
2013-10-17 05:18:21 +02:00
|
|
|
|
|
|
|
if (name == NULL) {
|
|
|
|
PyErr_Format(PyExc_KeyError,
|
|
|
|
"invalid id-property key, expected a string, not a %.200s",
|
|
|
|
Py_TYPE(name_obj)->tp_name);
|
2017-04-10 13:06:00 +02:00
|
|
|
return NULL;
|
2013-10-17 05:18:21 +02:00
|
|
|
}
|
|
|
|
|
2011-11-03 15:09:18 +01:00
|
|
|
if (name_size > MAX_IDPROP_NAME) {
|
2013-10-17 04:57:59 +02:00
|
|
|
PyErr_SetString(PyExc_KeyError, "the length of IDProperty names is limited to 63 characters");
|
2017-04-10 13:06:00 +02:00
|
|
|
return NULL;
|
2011-11-03 15:09:18 +01:00
|
|
|
}
|
|
|
|
}
|
2013-10-17 04:57:59 +02:00
|
|
|
else {
|
|
|
|
name = "";
|
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
return name;
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
/* -------------------------------------------------------------------------- */
|
|
|
|
|
|
|
|
/**
|
|
|
|
* The 'idp_from_Py*' functions expect that the input type has been checked before
|
|
|
|
* and return NULL if the IDProperty can't be created.
|
|
|
|
*/
|
|
|
|
|
|
|
|
static IDProperty *idp_from_PyFloat(const char *name, PyObject *ob)
|
|
|
|
{
|
|
|
|
IDPropertyTemplate val = {0};
|
|
|
|
val.d = PyFloat_AsDouble(ob);
|
|
|
|
return IDP_New(IDP_DOUBLE, &val, name);
|
|
|
|
}
|
|
|
|
|
|
|
|
static IDProperty *idp_from_PyLong(const char *name, PyObject *ob)
|
|
|
|
{
|
|
|
|
IDPropertyTemplate val = {0};
|
2017-08-20 07:44:54 +02:00
|
|
|
val.i = PyC_Long_AsI32(ob);
|
2017-04-10 13:06:00 +02:00
|
|
|
if (val.i == -1 && PyErr_Occurred()) {
|
|
|
|
return NULL;
|
2011-03-19 12:12:48 +01:00
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
return IDP_New(IDP_INT, &val, name);
|
|
|
|
}
|
|
|
|
|
|
|
|
static IDProperty *idp_from_PyUnicode(const char *name, PyObject *ob)
|
|
|
|
{
|
|
|
|
IDProperty *prop;
|
|
|
|
IDPropertyTemplate val = {0};
|
2010-09-01 16:13:48 +02:00
|
|
|
#ifdef USE_STRING_COERCE
|
2017-04-10 13:06:00 +02:00
|
|
|
Py_ssize_t value_size;
|
|
|
|
PyObject *value_coerce = NULL;
|
|
|
|
val.string.str = PyC_UnicodeAsByteAndSize(ob, &value_size, &value_coerce);
|
|
|
|
val.string.len = (int)value_size + 1;
|
|
|
|
val.string.subtype = IDP_STRING_SUB_UTF8;
|
|
|
|
prop = IDP_New(IDP_STRING, &val, name);
|
|
|
|
Py_XDECREF(value_coerce);
|
2010-09-01 16:13:48 +02:00
|
|
|
#else
|
2017-04-10 13:06:00 +02:00
|
|
|
val.str = _PyUnicode_AsString(ob);
|
|
|
|
prop = IDP_New(IDP_STRING, val, name);
|
2010-09-01 16:13:48 +02:00
|
|
|
#endif
|
2017-04-10 13:06:00 +02:00
|
|
|
return prop;
|
|
|
|
}
|
2011-11-15 10:12:10 +01:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
static IDProperty *idp_from_PyBytes(const char *name, PyObject *ob)
|
|
|
|
{
|
|
|
|
IDPropertyTemplate val = {0};
|
|
|
|
val.string.str = PyBytes_AS_STRING(ob);
|
|
|
|
val.string.len = PyBytes_GET_SIZE(ob);
|
|
|
|
val.string.subtype = IDP_STRING_SUB_BYTE;
|
|
|
|
return IDP_New(IDP_STRING, &val, name);
|
|
|
|
}
|
|
|
|
|
|
|
|
static int idp_array_type_from_format_char(char format)
|
|
|
|
{
|
|
|
|
if (format == 'i') return IDP_INT;
|
|
|
|
if (format == 'f') return IDP_FLOAT;
|
|
|
|
if (format == 'd') return IDP_DOUBLE;
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
static const char *idp_format_from_array_type(int type)
|
|
|
|
{
|
|
|
|
if (type == IDP_INT) return "i";
|
|
|
|
if (type == IDP_FLOAT) return "f";
|
|
|
|
if (type == IDP_DOUBLE) return "d";
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
static IDProperty *idp_from_PySequence_Buffer(const char *name, Py_buffer *buffer)
|
|
|
|
{
|
|
|
|
IDProperty *prop;
|
|
|
|
IDPropertyTemplate val = {0};
|
|
|
|
|
|
|
|
int format = idp_array_type_from_format_char(*buffer->format);
|
|
|
|
if (format == -1) {
|
|
|
|
/* should never happen as the type has been checked before */
|
|
|
|
return NULL;
|
2011-11-15 10:12:10 +01:00
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
else {
|
|
|
|
val.array.type = format;
|
|
|
|
val.array.len = buffer->len / buffer->itemsize;
|
|
|
|
}
|
|
|
|
prop = IDP_New(IDP_ARRAY, &val, name);
|
|
|
|
memcpy(IDP_Array(prop), buffer->buf, buffer->len);
|
|
|
|
return prop;
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
static IDProperty *idp_from_PySequence_Fast(const char *name, PyObject *ob)
|
|
|
|
{
|
|
|
|
IDProperty *prop;
|
|
|
|
IDPropertyTemplate val = {0};
|
2012-10-12 05:59:50 +02:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
PyObject **ob_seq_fast_items;
|
|
|
|
PyObject *item;
|
|
|
|
int i;
|
2015-07-29 01:58:10 +02:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
ob_seq_fast_items = PySequence_Fast_ITEMS(ob);
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
if ((val.array.type = idp_sequence_type(ob)) == (char)-1) {
|
|
|
|
PyErr_SetString(PyExc_TypeError, "only floats, ints and dicts are allowed in ID property arrays");
|
|
|
|
return NULL;
|
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
/* validate sequence and derive type.
|
|
|
|
* we assume IDP_INT unless we hit a float
|
|
|
|
* number; then we assume it's */
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
val.array.len = PySequence_Fast_GET_SIZE(ob);
|
|
|
|
|
|
|
|
switch (val.array.type) {
|
|
|
|
case IDP_DOUBLE:
|
|
|
|
{
|
|
|
|
double *prop_data;
|
|
|
|
prop = IDP_New(IDP_ARRAY, &val, name);
|
|
|
|
prop_data = IDP_Array(prop);
|
|
|
|
for (i = 0; i < val.array.len; i++) {
|
|
|
|
item = ob_seq_fast_items[i];
|
|
|
|
if (((prop_data[i] = PyFloat_AsDouble(item)) == -1.0) && PyErr_Occurred()) {
|
|
|
|
return NULL;
|
2012-03-16 22:39:56 +01:00
|
|
|
}
|
2013-10-17 05:18:21 +02:00
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
case IDP_INT:
|
|
|
|
{
|
|
|
|
int *prop_data;
|
|
|
|
prop = IDP_New(IDP_ARRAY, &val, name);
|
|
|
|
prop_data = IDP_Array(prop);
|
|
|
|
for (i = 0; i < val.array.len; i++) {
|
|
|
|
item = ob_seq_fast_items[i];
|
2017-08-20 07:44:54 +02:00
|
|
|
if (((prop_data[i] = PyC_Long_AsI32(item)) == -1) && PyErr_Occurred()) {
|
2017-04-10 13:06:00 +02:00
|
|
|
return NULL;
|
2012-03-16 22:39:56 +01:00
|
|
|
}
|
2013-10-17 05:18:21 +02:00
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
case IDP_IDPARRAY:
|
|
|
|
{
|
|
|
|
prop = IDP_NewIDPArray(name);
|
|
|
|
for (i = 0; i < val.array.len; i++) {
|
|
|
|
item = ob_seq_fast_items[i];
|
|
|
|
if (BPy_IDProperty_Map_ValidateAndCreate(NULL, prop, item) == false) {
|
|
|
|
return NULL;
|
2012-03-16 22:39:56 +01:00
|
|
|
}
|
2013-10-17 05:18:21 +02:00
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
break;
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
/* should never happen */
|
|
|
|
PyErr_SetString(PyExc_RuntimeError, "internal error with idp array.type");
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
return prop;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static IDProperty *idp_from_PySequence(const char *name, PyObject *ob)
|
|
|
|
{
|
|
|
|
Py_buffer buffer;
|
|
|
|
bool use_buffer = false;
|
|
|
|
|
|
|
|
if (PyObject_CheckBuffer(ob)) {
|
|
|
|
PyObject_GetBuffer(ob, &buffer, PyBUF_SIMPLE | PyBUF_FORMAT);
|
|
|
|
char format = *buffer.format;
|
|
|
|
if (ELEM(format, 'i', 'f', 'd')) {
|
|
|
|
use_buffer = true;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
else {
|
|
|
|
PyBuffer_Release(&buffer);
|
|
|
|
}
|
|
|
|
}
|
2012-10-12 05:59:50 +02:00
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
if (use_buffer) {
|
|
|
|
IDProperty *prop = idp_from_PySequence_Buffer(name, &buffer);
|
|
|
|
PyBuffer_Release(&buffer);
|
|
|
|
return prop;
|
2011-03-19 12:12:48 +01:00
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
else {
|
|
|
|
PyObject *ob_seq_fast = PySequence_Fast(ob, "py -> idprop");
|
|
|
|
if (ob_seq_fast != NULL) {
|
|
|
|
IDProperty *prop = idp_from_PySequence_Fast(name, ob_seq_fast);
|
|
|
|
Py_DECREF(ob_seq_fast);
|
|
|
|
return prop;
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static IDProperty *idp_from_PyMapping(const char *name, PyObject *ob)
|
|
|
|
{
|
|
|
|
IDProperty *prop;
|
|
|
|
IDPropertyTemplate val = {0};
|
|
|
|
|
|
|
|
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 (BPy_IDProperty_Map_ValidateAndCreate(key, prop, pval) == false) {
|
|
|
|
IDP_FreeProperty(prop);
|
|
|
|
MEM_freeN(prop);
|
|
|
|
Py_XDECREF(keys);
|
|
|
|
Py_XDECREF(vals);
|
2006-11-17 05:46:48 +01:00
|
|
|
Py_XDECREF(key);
|
|
|
|
Py_XDECREF(pval);
|
2017-04-10 13:06:00 +02:00
|
|
|
/* error is already set */
|
|
|
|
return NULL;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
Py_XDECREF(key);
|
|
|
|
Py_XDECREF(pval);
|
|
|
|
}
|
|
|
|
Py_XDECREF(keys);
|
|
|
|
Py_XDECREF(vals);
|
|
|
|
return prop;
|
|
|
|
}
|
|
|
|
|
Datablock ID Properties
The absence of datablock properties "will certainly be resolved soon as the need for them is becoming obvious" said the [[http://wiki.blender.org/index.php/Dev:Ref/Release_Notes/2.67/Python_Nodes|Python Nodes release notes]]. So this patch allows Python scripts to create ID Properties which reference datablocks.
This functionality is implemented for `PointerProperty` and now such properties can be created with Python.
In addition to the standard update callback, `PointerProperty` can have a `poll` callback (standard RNA) which is useful for search menus. For details see the test included in this patch.
Original author: @artfunkel
Alexander (Blend4Web Team)
Reviewers: brecht, artfunkel, mont29, campbellbarton
Reviewed By: mont29, campbellbarton
Subscribers: jta, sergey, campbellbarton, wisaac, poseidon4o, mont29, homyachetser, Evgeny_Rodygin, AlexKowel, yurikovelenov, fjuhec, sharlybg, cardboard, duarteframos, blueprintrandom, a.romanov, BYOB, disnel, aditiapratama, bliblubli, dfelinto, lukastoenne
Maniphest Tasks: T37754
Differential Revision: https://developer.blender.org/D113
2017-04-13 11:30:03 +02:00
|
|
|
static IDProperty *idp_from_DatablockPointer(const char *name, PyObject *ob, IDPropertyTemplate *val)
|
|
|
|
{
|
|
|
|
pyrna_id_FromPyObject(ob, &val->id);
|
|
|
|
return IDP_New(IDP_ID, val, name);
|
|
|
|
}
|
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
static IDProperty *idp_from_PyObject(PyObject *name_obj, PyObject *ob)
|
|
|
|
{
|
Datablock ID Properties
The absence of datablock properties "will certainly be resolved soon as the need for them is becoming obvious" said the [[http://wiki.blender.org/index.php/Dev:Ref/Release_Notes/2.67/Python_Nodes|Python Nodes release notes]]. So this patch allows Python scripts to create ID Properties which reference datablocks.
This functionality is implemented for `PointerProperty` and now such properties can be created with Python.
In addition to the standard update callback, `PointerProperty` can have a `poll` callback (standard RNA) which is useful for search menus. For details see the test included in this patch.
Original author: @artfunkel
Alexander (Blend4Web Team)
Reviewers: brecht, artfunkel, mont29, campbellbarton
Reviewed By: mont29, campbellbarton
Subscribers: jta, sergey, campbellbarton, wisaac, poseidon4o, mont29, homyachetser, Evgeny_Rodygin, AlexKowel, yurikovelenov, fjuhec, sharlybg, cardboard, duarteframos, blueprintrandom, a.romanov, BYOB, disnel, aditiapratama, bliblubli, dfelinto, lukastoenne
Maniphest Tasks: T37754
Differential Revision: https://developer.blender.org/D113
2017-04-13 11:30:03 +02:00
|
|
|
IDPropertyTemplate val = {0};
|
2017-04-10 13:06:00 +02:00
|
|
|
const char *name = idp_try_read_name(name_obj);
|
|
|
|
if (name == NULL) {
|
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (PyFloat_Check(ob)) {
|
|
|
|
return idp_from_PyFloat(name, ob);
|
|
|
|
}
|
|
|
|
else if (PyLong_Check(ob)) {
|
|
|
|
return idp_from_PyLong(name, ob);
|
|
|
|
}
|
|
|
|
else if (PyUnicode_Check(ob)) {
|
|
|
|
return idp_from_PyUnicode(name, ob);
|
|
|
|
}
|
|
|
|
else if (PyBytes_Check(ob)) {
|
|
|
|
return idp_from_PyBytes(name, ob);
|
|
|
|
}
|
|
|
|
else if (PySequence_Check(ob)) {
|
|
|
|
return idp_from_PySequence(name, ob);
|
|
|
|
}
|
Datablock ID Properties
The absence of datablock properties "will certainly be resolved soon as the need for them is becoming obvious" said the [[http://wiki.blender.org/index.php/Dev:Ref/Release_Notes/2.67/Python_Nodes|Python Nodes release notes]]. So this patch allows Python scripts to create ID Properties which reference datablocks.
This functionality is implemented for `PointerProperty` and now such properties can be created with Python.
In addition to the standard update callback, `PointerProperty` can have a `poll` callback (standard RNA) which is useful for search menus. For details see the test included in this patch.
Original author: @artfunkel
Alexander (Blend4Web Team)
Reviewers: brecht, artfunkel, mont29, campbellbarton
Reviewed By: mont29, campbellbarton
Subscribers: jta, sergey, campbellbarton, wisaac, poseidon4o, mont29, homyachetser, Evgeny_Rodygin, AlexKowel, yurikovelenov, fjuhec, sharlybg, cardboard, duarteframos, blueprintrandom, a.romanov, BYOB, disnel, aditiapratama, bliblubli, dfelinto, lukastoenne
Maniphest Tasks: T37754
Differential Revision: https://developer.blender.org/D113
2017-04-13 11:30:03 +02:00
|
|
|
else if (ob == Py_None || pyrna_id_CheckPyObject(ob)) {
|
|
|
|
return idp_from_DatablockPointer(name, ob, &val);
|
|
|
|
}
|
2017-04-10 13:06:00 +02:00
|
|
|
else if (PyMapping_Check(ob)) {
|
|
|
|
return idp_from_PyMapping(name, ob);
|
2011-03-19 12:12:48 +01:00
|
|
|
}
|
2013-03-09 04:46:30 +01:00
|
|
|
else {
|
2013-10-17 04:57:59 +02:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"invalid id-property type %.200s not supported",
|
|
|
|
Py_TYPE(ob)->tp_name);
|
2017-04-10 13:06:00 +02:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* -------------------------------------------------------------------------- */
|
|
|
|
|
|
|
|
/**
|
|
|
|
* \note group can be a pointer array or a group.
|
|
|
|
* assume we already checked key is a string.
|
|
|
|
*
|
|
|
|
* \return success.
|
|
|
|
*/
|
|
|
|
bool BPy_IDProperty_Map_ValidateAndCreate(PyObject *name_obj, IDProperty *group, PyObject *ob)
|
|
|
|
{
|
|
|
|
IDProperty *prop = idp_from_PyObject(name_obj, ob);
|
|
|
|
if (prop == NULL) {
|
2013-10-17 04:57:59 +02:00
|
|
|
return false;
|
2013-03-09 04:46:30 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
if (group->type == IDP_IDPARRAY) {
|
2010-03-02 12:35:28 +01:00
|
|
|
IDP_AppendArray(group, prop);
|
2017-04-10 13:06:00 +02:00
|
|
|
/* IDP_AppendArray does a shallow copy (memcpy), only free memory */
|
2010-03-02 12:35:28 +01:00
|
|
|
MEM_freeN(prop);
|
2011-03-19 12:12:48 +01:00
|
|
|
}
|
|
|
|
else {
|
2014-11-10 17:10:58 +01:00
|
|
|
IDProperty *prop_exist;
|
|
|
|
|
2014-11-24 09:33:12 +01:00
|
|
|
/* avoid freeing when types match in case they are referenced by the UI, see: T37073
|
2014-11-10 17:10:58 +01:00
|
|
|
* obviously this isn't a complete solution, but helps for common cases. */
|
|
|
|
prop_exist = IDP_GetPropertyFromGroup(group, prop->name);
|
|
|
|
if ((prop_exist != NULL) &&
|
|
|
|
(prop_exist->type == prop->type) &&
|
|
|
|
(prop_exist->subtype == prop->subtype))
|
|
|
|
{
|
2014-11-14 16:38:11 +01:00
|
|
|
/* Preserve prev/next links!!! See T42593. */
|
|
|
|
prop->prev = prop_exist->prev;
|
|
|
|
prop->next = prop_exist->next;
|
|
|
|
|
2014-11-10 17:10:58 +01:00
|
|
|
IDP_FreeProperty(prop_exist);
|
|
|
|
*prop_exist = *prop;
|
|
|
|
MEM_freeN(prop);
|
|
|
|
}
|
|
|
|
else {
|
|
|
|
IDP_ReplaceInGroup_ex(group, prop, prop_exist);
|
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
}
|
|
|
|
|
2013-10-17 04:57:59 +02:00
|
|
|
return true;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2009-11-16 23:21:39 +01:00
|
|
|
int BPy_Wrap_SetMapItem(IDProperty *prop, PyObject *key, PyObject *val)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2012-05-20 21:49:27 +02:00
|
|
|
if (prop->type != IDP_GROUP) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_TypeError, "unsubscriptable object");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
2009-11-16 23:21:39 +01:00
|
|
|
if (val == NULL) { /* del idprop[key] */
|
2013-11-08 09:46:43 +01:00
|
|
|
IDProperty *pkey;
|
|
|
|
const char *name = _PyUnicode_AsString(key);
|
|
|
|
|
|
|
|
if (name == NULL) {
|
|
|
|
PyErr_Format(PyExc_KeyError,
|
|
|
|
"expected a string, not %.200s",
|
|
|
|
Py_TYPE(key)->tp_name);
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
pkey = IDP_GetPropertyFromGroup(prop, name);
|
2006-12-01 04:04:36 +01:00
|
|
|
if (pkey) {
|
2013-10-16 07:29:28 +02:00
|
|
|
IDP_FreeFromGroup(prop, pkey);
|
2006-12-01 04:04:36 +01:00
|
|
|
return 0;
|
2011-03-19 12:12:48 +01:00
|
|
|
}
|
|
|
|
else {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_KeyError, "property not found in group");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2006-12-01 04:04:36 +01:00
|
|
|
}
|
2009-11-16 23:21:39 +01:00
|
|
|
else {
|
2013-10-17 04:57:59 +02:00
|
|
|
bool ok;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2013-10-17 04:57:59 +02:00
|
|
|
ok = BPy_IDProperty_Map_ValidateAndCreate(key, prop, val);
|
|
|
|
if (ok == false) {
|
2009-11-16 23:21:39 +01:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
2009-11-16 19:56:58 +01:00
|
|
|
}
|
2009-11-16 23:21:39 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 23:21:39 +01:00
|
|
|
static int BPy_IDGroup_Map_SetItem(BPy_IDProperty *self, PyObject *key, PyObject *val)
|
|
|
|
{
|
2009-11-17 13:54:29 +01:00
|
|
|
return BPy_Wrap_SetMapItem(self->prop, key, val);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2011-06-17 07:45:46 +02:00
|
|
|
static PyObject *BPy_IDGroup_iter(BPy_IDProperty *self)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2011-06-17 07:45:46 +02:00
|
|
|
BPy_IDGroup_Iter *iter = PyObject_New(BPy_IDGroup_Iter, &BPy_IDGroup_Iter_Type);
|
2006-12-17 00:54:45 +01:00
|
|
|
iter->group = self;
|
|
|
|
iter->mode = IDPROP_ITER_KEYS;
|
|
|
|
iter->cur = self->prop->data.group.first;
|
|
|
|
Py_XINCREF(iter);
|
2012-03-26 08:55:09 +02:00
|
|
|
return (PyObject *)iter;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
/* for simple, non nested types this is the same as BPy_IDGroup_WrapData */
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDGroup_MapDataToPy(IDProperty *prop)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
switch (prop->type) {
|
2012-03-16 22:39:56 +01:00
|
|
|
case IDP_STRING:
|
|
|
|
return idprop_py_from_idp_string(prop);
|
|
|
|
case IDP_INT:
|
|
|
|
return idprop_py_from_idp_int(prop);
|
2006-12-17 00:54:45 +01:00
|
|
|
case IDP_FLOAT:
|
2012-03-16 22:39:56 +01:00
|
|
|
return idprop_py_from_idp_float(prop);
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
2012-03-16 22:39:56 +01:00
|
|
|
return idprop_py_from_idp_double(prop);
|
Datablock ID Properties
The absence of datablock properties "will certainly be resolved soon as the need for them is becoming obvious" said the [[http://wiki.blender.org/index.php/Dev:Ref/Release_Notes/2.67/Python_Nodes|Python Nodes release notes]]. So this patch allows Python scripts to create ID Properties which reference datablocks.
This functionality is implemented for `PointerProperty` and now such properties can be created with Python.
In addition to the standard update callback, `PointerProperty` can have a `poll` callback (standard RNA) which is useful for search menus. For details see the test included in this patch.
Original author: @artfunkel
Alexander (Blend4Web Team)
Reviewers: brecht, artfunkel, mont29, campbellbarton
Reviewed By: mont29, campbellbarton
Subscribers: jta, sergey, campbellbarton, wisaac, poseidon4o, mont29, homyachetser, Evgeny_Rodygin, AlexKowel, yurikovelenov, fjuhec, sharlybg, cardboard, duarteframos, blueprintrandom, a.romanov, BYOB, disnel, aditiapratama, bliblubli, dfelinto, lukastoenne
Maniphest Tasks: T37754
Differential Revision: https://developer.blender.org/D113
2017-04-13 11:30:03 +02:00
|
|
|
case IDP_ID:
|
|
|
|
return idprop_py_from_idp_id(prop);
|
2012-03-16 22:39:56 +01:00
|
|
|
case IDP_ARRAY:
|
2011-11-15 11:19:44 +01:00
|
|
|
{
|
2012-03-16 22:39:56 +01: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 19:56:58 +01:00
|
|
|
}
|
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
switch (prop->subtype) {
|
|
|
|
case IDP_FLOAT:
|
|
|
|
{
|
2014-04-26 16:21:23 +02:00
|
|
|
const float *array = (float *)IDP_Array(prop);
|
2012-03-16 22:39:56 +01:00
|
|
|
for (i = 0; i < prop->len; i++) {
|
|
|
|
PyList_SET_ITEM(seq, i, PyFloat_FromDouble(array[i]));
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case IDP_DOUBLE:
|
|
|
|
{
|
2014-04-26 16:21:23 +02:00
|
|
|
const double *array = (double *)IDP_Array(prop);
|
2012-03-16 22:39:56 +01:00
|
|
|
for (i = 0; i < prop->len; i++) {
|
|
|
|
PyList_SET_ITEM(seq, i, PyFloat_FromDouble(array[i]));
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case IDP_INT:
|
|
|
|
{
|
2014-04-26 16:21:23 +02:00
|
|
|
const int *array = (int *)IDP_Array(prop);
|
2012-03-16 22:39:56 +01:00
|
|
|
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 07:45:46 +02:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
return seq;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2012-03-16 22:39:56 +01:00
|
|
|
case IDP_IDPARRAY:
|
|
|
|
{
|
2016-07-31 08:59:08 +02:00
|
|
|
PyObject *seq = PyList_New(prop->len);
|
2012-03-16 22:39:56 +01:00
|
|
|
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 19:56:58 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
for (i = 0; i < prop->len; i++) {
|
2016-07-31 08:59:08 +02:00
|
|
|
PyObject *wrap = BPy_IDGroup_MapDataToPy(array++);
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2016-07-31 08:59:08 +02:00
|
|
|
/* BPy_IDGroup_MapDataToPy sets the error */
|
|
|
|
if (UNLIKELY(wrap == NULL)) {
|
|
|
|
Py_DECREF(seq);
|
2012-03-16 22:39:56 +01:00
|
|
|
return NULL;
|
2016-07-31 08:59:08 +02:00
|
|
|
}
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
PyList_SET_ITEM(seq, i, wrap);
|
|
|
|
}
|
|
|
|
return seq;
|
2010-03-02 12:35:28 +01:00
|
|
|
}
|
2012-03-16 22:39:56 +01:00
|
|
|
case IDP_GROUP:
|
|
|
|
{
|
2016-07-31 09:22:04 +02:00
|
|
|
PyObject *dict = _PyDict_NewPresized(prop->len);
|
2012-03-16 22:39:56 +01:00
|
|
|
IDProperty *loop;
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
for (loop = prop->data.group.first; loop; loop = loop->next) {
|
2016-07-31 08:59:08 +02:00
|
|
|
PyObject *wrap = BPy_IDGroup_MapDataToPy(loop);
|
2010-03-02 12:35:28 +01:00
|
|
|
|
2016-07-31 08:59:08 +02:00
|
|
|
/* BPy_IDGroup_MapDataToPy sets the error */
|
|
|
|
if (UNLIKELY(wrap == NULL)) {
|
|
|
|
Py_DECREF(dict);
|
2012-03-16 22:39:56 +01:00
|
|
|
return NULL;
|
2016-07-31 08:59:08 +02:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
PyDict_SetItemString(dict, loop->name, wrap);
|
|
|
|
Py_DECREF(wrap);
|
|
|
|
}
|
|
|
|
return dict;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
PyErr_Format(PyExc_RuntimeError,
|
|
|
|
"%s ERROR: '%s' property exists with a bad type code '%d'!",
|
|
|
|
__func__, prop->name, prop->type);
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_pop_doc,
|
|
|
|
".. method:: pop(key)\n"
|
|
|
|
"\n"
|
|
|
|
" Remove an item from the group, returning a Python representation.\n"
|
|
|
|
"\n"
|
|
|
|
" :raises KeyError: When the item doesn't exist.\n"
|
|
|
|
"\n"
|
|
|
|
" :arg key: Name of item to remove.\n"
|
|
|
|
" :type key: string\n"
|
|
|
|
);
|
|
|
|
static PyObject *BPy_IDGroup_pop(BPy_IDProperty *self, PyObject *value)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
IDProperty *idprop;
|
2006-12-17 00:54:45 +01:00
|
|
|
PyObject *pyform;
|
2011-11-03 15:09:18 +01:00
|
|
|
const char *name = _PyUnicode_AsString(value);
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2007-06-16 14:24:41 +02:00
|
|
|
if (!name) {
|
2011-11-15 11:19:44 +01:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"pop expected at least a string argument, not %.200s",
|
|
|
|
Py_TYPE(value)->tp_name);
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
idprop = IDP_GetPropertyFromGroup(self->prop, name);
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2011-10-13 03:29:08 +02:00
|
|
|
if (idprop) {
|
2009-11-16 19:56:58 +01:00
|
|
|
pyform = BPy_IDGroup_MapDataToPy(idprop);
|
|
|
|
|
|
|
|
if (!pyform) {
|
2012-03-03 21:36:09 +01: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 17:53:30 +02:00
|
|
|
return NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2013-10-16 07:29:28 +02:00
|
|
|
IDP_RemoveFromGroup(self->prop, idprop);
|
2009-11-16 19:56:58 +01:00
|
|
|
return pyform;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_KeyError, "item not in group");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_iter_items_doc,
|
|
|
|
".. method:: iteritems()\n"
|
|
|
|
"\n"
|
|
|
|
" Iterate through the items in the dict; behaves like dictionary method iteritems.\n"
|
|
|
|
);
|
|
|
|
static PyObject *BPy_IDGroup_iter_items(BPy_IDProperty *self)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2011-06-17 07:45:46 +02:00
|
|
|
BPy_IDGroup_Iter *iter = PyObject_New(BPy_IDGroup_Iter, &BPy_IDGroup_Iter_Type);
|
2006-12-17 00:54:45 +01:00
|
|
|
iter->group = self;
|
|
|
|
iter->mode = IDPROP_ITER_ITEMS;
|
|
|
|
iter->cur = self->prop->data.group.first;
|
|
|
|
Py_XINCREF(iter);
|
2012-03-26 08:55:09 +02:00
|
|
|
return (PyObject *)iter;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
/* utility function */
|
2011-10-31 02:50:04 +01:00
|
|
|
static void BPy_IDGroup_CorrectListLen(IDProperty *prop, PyObject *seq, int len, const char *func)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2009-11-19 12:56:13 +01:00
|
|
|
int j;
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2011-10-31 02:50:04 +01:00
|
|
|
printf("%s: ID Property Error found and corrected!\n", func);
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2015-01-06 06:42:22 +01:00
|
|
|
/* fill rest of list with valid references to None */
|
2012-03-16 22:39:56 +01:00
|
|
|
for (j = len; j < prop->len; j++) {
|
2015-01-06 06:42:22 +01:00
|
|
|
PyList_SET_ITEM(seq, j, Py_INCREF_RET(Py_None));
|
2009-11-16 21:16:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
/*set correct group length*/
|
2009-11-19 12:56:13 +01:00
|
|
|
prop->len = len;
|
2009-11-16 21:16:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
PyObject *BPy_Wrap_GetKeys(IDProperty *prop)
|
|
|
|
{
|
2011-11-15 11:19:44 +01:00
|
|
|
PyObject *list = PyList_New(prop->len);
|
2009-11-16 21:16:45 +01:00
|
|
|
IDProperty *loop;
|
|
|
|
int i;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
for (i = 0, loop = prop->data.group.first; loop && (i < prop->len); loop = loop->next, i++)
|
2011-11-15 11:19:44 +01:00
|
|
|
PyList_SET_ITEM(list, i, PyUnicode_FromString(loop->name));
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2010-02-01 19:38:33 +01:00
|
|
|
/* if the id prop is corrupt, count the remaining */
|
2012-03-16 22:39:56 +01:00
|
|
|
for ( ; loop; loop = loop->next, i++) {
|
|
|
|
/* pass */
|
|
|
|
}
|
2010-02-01 19:38:33 +01:00
|
|
|
|
|
|
|
if (i != prop->len) { /* if the loop didnt finish, we know the length is wrong */
|
2011-11-15 11:19:44 +01:00
|
|
|
BPy_IDGroup_CorrectListLen(prop, list, i, __func__);
|
|
|
|
Py_DECREF(list); /*free the list*/
|
2008-02-23 03:12:50 +01:00
|
|
|
/*call self again*/
|
2009-11-16 21:16:45 +01:00
|
|
|
return BPy_Wrap_GetKeys(prop);
|
2008-02-23 03:12:50 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
return list;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
PyObject *BPy_Wrap_GetValues(ID *id, IDProperty *prop)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2011-11-15 11:19:44 +01:00
|
|
|
PyObject *list = PyList_New(prop->len);
|
2006-12-17 00:54:45 +01:00
|
|
|
IDProperty *loop;
|
2009-11-16 21:16:45 +01:00
|
|
|
int i;
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
for (i = 0, loop = prop->data.group.first; loop; loop = loop->next, i++) {
|
2011-11-15 11:19:44 +01:00
|
|
|
PyList_SET_ITEM(list, i, BPy_IDGroup_WrapData(id, loop, prop));
|
2009-11-16 19:56:58 +01:00
|
|
|
}
|
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
if (i != prop->len) {
|
2011-11-15 11:19:44 +01:00
|
|
|
BPy_IDGroup_CorrectListLen(prop, list, i, __func__);
|
|
|
|
Py_DECREF(list); /*free the list*/
|
2009-11-16 21:16:45 +01:00
|
|
|
/*call self again*/
|
|
|
|
return BPy_Wrap_GetValues(id, prop);
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2008-02-23 03:12:50 +01:00
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
return list;
|
2009-11-16 21:16:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
PyObject *BPy_Wrap_GetItems(ID *id, IDProperty *prop)
|
|
|
|
{
|
|
|
|
PyObject *seq = PyList_New(prop->len);
|
|
|
|
IDProperty *loop;
|
|
|
|
int i;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
for (i = 0, loop = prop->data.group.first; loop; loop = loop->next, i++) {
|
|
|
|
PyObject *item = PyTuple_New(2);
|
2015-01-06 06:42:22 +01:00
|
|
|
PyTuple_SET_ITEMS(item,
|
|
|
|
PyUnicode_FromString(loop->name),
|
|
|
|
BPy_IDGroup_WrapData(id, loop, prop));
|
2009-11-16 21:16:45 +01:00
|
|
|
PyList_SET_ITEM(seq, i, item);
|
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2009-11-16 21:16:45 +01:00
|
|
|
if (i != prop->len) {
|
2011-10-31 02:50:04 +01:00
|
|
|
BPy_IDGroup_CorrectListLen(prop, seq, i, __func__);
|
2009-11-16 21:16:45 +01:00
|
|
|
Py_DECREF(seq); /*free the list*/
|
2008-02-23 03:12:50 +01:00
|
|
|
/*call self again*/
|
2009-11-16 21:16:45 +01:00
|
|
|
return BPy_Wrap_GetItems(id, prop);
|
2008-02-23 03:12:50 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
return seq;
|
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_keys_doc,
|
|
|
|
".. method:: keys()\n"
|
|
|
|
"\n"
|
|
|
|
" Return the keys associated with this group as a list of strings.\n"
|
|
|
|
);
|
|
|
|
static PyObject *BPy_IDGroup_keys(BPy_IDProperty *self)
|
2009-11-16 21:16:45 +01:00
|
|
|
{
|
|
|
|
return BPy_Wrap_GetKeys(self->prop);
|
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_values_doc,
|
|
|
|
".. method:: values()\n"
|
|
|
|
"\n"
|
|
|
|
" Return the values associated with this group.\n"
|
|
|
|
);
|
|
|
|
static PyObject *BPy_IDGroup_values(BPy_IDProperty *self)
|
2009-11-16 21:16:45 +01:00
|
|
|
{
|
|
|
|
return BPy_Wrap_GetValues(self->id, self->prop);
|
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_items_doc,
|
|
|
|
".. method:: items()\n"
|
|
|
|
"\n"
|
|
|
|
" Return the items associated with this group.\n"
|
|
|
|
);
|
|
|
|
static PyObject *BPy_IDGroup_items(BPy_IDProperty *self)
|
2009-11-16 21:16:45 +01:00
|
|
|
{
|
|
|
|
return BPy_Wrap_GetItems(self->id, self->prop);
|
|
|
|
}
|
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
static int BPy_IDGroup_Contains(BPy_IDProperty *self, PyObject *value)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2011-11-03 15:09:18 +01:00
|
|
|
const char *name = _PyUnicode_AsString(value);
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
if (!name) {
|
2011-11-15 11:19:44 +01:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"expected a string, not a %.200s",
|
|
|
|
Py_TYPE(value)->tp_name);
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
return IDP_GetPropertyFromGroup(self->prop, name) ? 1 : 0;
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_update_doc,
|
|
|
|
".. method:: update(other)\n"
|
|
|
|
"\n"
|
|
|
|
" Update key, values.\n"
|
|
|
|
"\n"
|
|
|
|
" :arg other: Updates the values in the group with this.\n"
|
|
|
|
" :type other: :class:`IDPropertyGroup` or dict\n"
|
|
|
|
);
|
|
|
|
static PyObject *BPy_IDGroup_update(BPy_IDProperty *self, PyObject *value)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
2008-12-20 09:41:46 +01:00
|
|
|
PyObject *pkey, *pval;
|
2012-03-16 22:39:56 +01:00
|
|
|
Py_ssize_t i = 0;
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2012-10-31 20:07:25 +01: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 */
|
2014-04-01 02:34:00 +02:00
|
|
|
IDP_MergeGroup(self->prop, other->prop, true);
|
2012-10-31 20:07:25 +01:00
|
|
|
}
|
|
|
|
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 11:19:44 +01:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
2012-10-31 20:07:25 +01:00
|
|
|
"expected a dict or an IDPropertyGroup type, not a %.200s",
|
2011-11-15 11:19:44 +01:00
|
|
|
Py_TYPE(value)->tp_name);
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_to_dict_doc,
|
|
|
|
".. method:: to_dict()\n"
|
|
|
|
"\n"
|
|
|
|
" Return a purely python version of the group.\n"
|
|
|
|
);
|
2011-06-17 07:45:46 +02:00
|
|
|
static PyObject *BPy_IDGroup_to_dict(BPy_IDProperty *self)
|
2006-12-17 00:54:45 +01:00
|
|
|
{
|
|
|
|
return BPy_IDGroup_MapDataToPy(self->prop);
|
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_clear_doc,
|
|
|
|
".. method:: clear()\n"
|
|
|
|
"\n"
|
|
|
|
" Clear all members from this group.\n"
|
|
|
|
);
|
2013-01-03 00:10:14 +01:00
|
|
|
static PyObject *BPy_IDGroup_clear(BPy_IDProperty *self)
|
|
|
|
{
|
|
|
|
IDP_ClearProperty(self->prop);
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
2009-11-19 18:04:28 +01:00
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDGroup_get_doc,
|
|
|
|
".. method:: get(key, default=None)\n"
|
|
|
|
"\n"
|
|
|
|
" Return the value for key, if it exists, else default.\n"
|
|
|
|
);
|
|
|
|
static PyObject *BPy_IDGroup_get(BPy_IDProperty *self, PyObject *args)
|
2009-11-19 18:04:28 +01:00
|
|
|
{
|
|
|
|
IDProperty *idprop;
|
2014-04-26 16:21:23 +02:00
|
|
|
const char *key;
|
2012-03-16 22:39:56 +01:00
|
|
|
PyObject *def = Py_None;
|
2009-11-19 18:04:28 +01:00
|
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "s|O:get", &key, &def))
|
|
|
|
return NULL;
|
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
idprop = IDP_GetPropertyFromGroup(self->prop, key);
|
2009-11-19 18:04:28 +01:00
|
|
|
if (idprop) {
|
2012-03-16 22:39:56 +01:00
|
|
|
PyObject *pyobj = BPy_IDGroup_WrapData(self->id, idprop, self->prop);
|
2009-11-19 18:04:28 +01:00
|
|
|
if (pyobj)
|
|
|
|
return pyobj;
|
|
|
|
}
|
|
|
|
|
|
|
|
Py_INCREF(def);
|
|
|
|
return def;
|
|
|
|
}
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
static struct PyMethodDef BPy_IDGroup_methods[] = {
|
2016-07-31 07:39:30 +02:00
|
|
|
{"pop", (PyCFunction)BPy_IDGroup_pop, METH_O, BPy_IDGroup_pop_doc},
|
|
|
|
{"iteritems", (PyCFunction)BPy_IDGroup_iter_items, METH_NOARGS, BPy_IDGroup_iter_items_doc},
|
|
|
|
{"keys", (PyCFunction)BPy_IDGroup_keys, METH_NOARGS, BPy_IDGroup_keys_doc},
|
|
|
|
{"values", (PyCFunction)BPy_IDGroup_values, METH_NOARGS, BPy_IDGroup_values_doc},
|
|
|
|
{"items", (PyCFunction)BPy_IDGroup_items, METH_NOARGS, BPy_IDGroup_items_doc},
|
|
|
|
{"update", (PyCFunction)BPy_IDGroup_update, METH_O, BPy_IDGroup_update_doc},
|
|
|
|
{"get", (PyCFunction)BPy_IDGroup_get, METH_VARARGS, BPy_IDGroup_get_doc},
|
|
|
|
{"to_dict", (PyCFunction)BPy_IDGroup_to_dict, METH_NOARGS, BPy_IDGroup_to_dict_doc},
|
|
|
|
{"clear", (PyCFunction)BPy_IDGroup_clear, METH_NOARGS, BPy_IDGroup_clear_doc},
|
2011-02-13 11:52:18 +01:00
|
|
|
{NULL, NULL, 0, NULL}
|
2006-12-17 00:54:45 +01:00
|
|
|
};
|
2009-11-16 19:56:58 +01:00
|
|
|
|
|
|
|
static PySequenceMethods BPy_IDGroup_Seq = {
|
2012-03-26 08:55:09 +02: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 19:56:58 +01:00
|
|
|
};
|
|
|
|
|
2011-02-13 11:52:18 +01:00
|
|
|
static PyMappingMethods BPy_IDGroup_Mapping = {
|
2012-03-26 08:55:09 +02: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 05:46:48 +01:00
|
|
|
};
|
|
|
|
|
2011-06-17 07:45:46 +02:00
|
|
|
PyTypeObject BPy_IDGroup_Type = {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
2006-11-17 05:46:48 +01:00
|
|
|
/* For printing, in format "<module>.<name>" */
|
2012-04-15 16:54:15 +02:00
|
|
|
"IDPropertyGroup", /* char *tp_name; */
|
2012-03-26 08:55:09 +02:00
|
|
|
sizeof(BPy_IDProperty), /* int tp_basicsize; */
|
2006-11-17 05:46:48 +01:00
|
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
NULL, /* destructor tp_dealloc; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* printfunc tp_print; */
|
2012-03-26 08:55:09 +02:00
|
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
|
|
NULL, /* setattrfunc tp_setattr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* cmpfunc tp_compare; */
|
2011-06-17 07:45:46 +02:00
|
|
|
(reprfunc)BPy_IDGroup_repr, /* reprfunc tp_repr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
2012-03-26 08:55:09 +02:00
|
|
|
&BPy_IDGroup_Seq, /* PySequenceMethods *tp_as_sequence; */
|
|
|
|
&BPy_IDGroup_Mapping, /* PyMappingMethods *tp_as_mapping; */
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
|
2011-06-17 07:45:46 +02:00
|
|
|
(hashfunc)BPy_IDGroup_hash, /* hashfunc tp_hash; */
|
2006-11-17 05:46:48 +01: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 08:55:09 +02:00
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Assigned meaning in release 2.0 ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
/* call function for all accessible objects */
|
|
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
|
|
|
|
/* delete references to contained objects */
|
|
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
|
|
/*** rich comparisons ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** weak reference enabler ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Added in release 2.2 ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
/* Iterators */
|
2011-06-17 07:45:46 +02:00
|
|
|
(getiterfunc)BPy_IDGroup_iter, /* getiterfunc tp_iter; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* iternextfunc tp_iternext; */
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Attribute descriptor and subclassing stuff ***/
|
2006-12-17 00:54:45 +01:00
|
|
|
BPy_IDGroup_methods, /* struct PyMethodDef *tp_methods; */
|
|
|
|
NULL, /* struct PyMemberDef *tp_members; */
|
|
|
|
BPy_IDGroup_getseters, /* struct PyGetSetDef *tp_getset; */
|
2006-11-17 05:46:48 +01:00
|
|
|
};
|
|
|
|
|
|
|
|
/********Array Wrapper********/
|
|
|
|
|
2014-01-27 17:52:21 +01:00
|
|
|
static PyTypeObject *idp_array_py_type(BPy_IDArray *self, bool *r_is_double)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2011-06-17 07:45:46 +02:00
|
|
|
switch (self->prop->subtype) {
|
|
|
|
case IDP_FLOAT:
|
2014-01-27 17:52:21 +01:00
|
|
|
*r_is_double = false;
|
2011-06-17 07:45:46 +02:00
|
|
|
return &PyFloat_Type;
|
|
|
|
case IDP_DOUBLE:
|
2014-01-27 17:52:21 +01:00
|
|
|
*r_is_double = true;
|
2011-06-17 07:45:46 +02:00
|
|
|
return &PyFloat_Type;
|
|
|
|
case IDP_INT:
|
2014-01-27 17:52:21 +01:00
|
|
|
*r_is_double = false;
|
2011-06-17 07:45:46 +02:00
|
|
|
return &PyLong_Type;
|
2014-01-27 17:52:21 +01:00
|
|
|
default:
|
|
|
|
*r_is_double = false;
|
|
|
|
return NULL;
|
2011-06-17 07:45:46 +02:00
|
|
|
}
|
|
|
|
}
|
2006-12-17 00:54:45 +01:00
|
|
|
|
2011-06-17 07:45:46 +02:00
|
|
|
static PyObject *BPy_IDArray_repr(BPy_IDArray *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2011-06-17 07:45:46 +02:00
|
|
|
return PyUnicode_FromFormat("<bpy id property array [%d]>", self->prop->len);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDArray_get_typecode_doc,
|
|
|
|
"The type of the data in the array {'f': float, 'd': double, 'i': int}."
|
|
|
|
);
|
|
|
|
static PyObject *BPy_IDArray_get_typecode(BPy_IDArray *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2011-12-18 09:50:06 +01:00
|
|
|
switch (self->prop->subtype) {
|
2012-03-16 22:39:56 +01: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 07:45:46 +02:00
|
|
|
}
|
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
PyErr_Format(PyExc_RuntimeError,
|
|
|
|
"%s: invalid/corrupt array type '%d'!",
|
|
|
|
__func__, self->prop->subtype);
|
|
|
|
|
2011-06-17 07:45:46 +02:00
|
|
|
return NULL;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
static PyGetSetDef BPy_IDArray_getseters[] = {
|
2012-03-26 08:55:09 +02:00
|
|
|
/* matches pythons array.typecode */
|
2016-07-31 07:39:30 +02:00
|
|
|
{(char *)"typecode", (getter)BPy_IDArray_get_typecode, (setter)NULL, BPy_IDArray_get_typecode_doc, NULL},
|
2006-12-17 00:54:45 +01:00
|
|
|
{NULL, NULL, NULL, NULL, NULL},
|
|
|
|
};
|
|
|
|
|
2016-07-31 07:39:30 +02:00
|
|
|
PyDoc_STRVAR(BPy_IDArray_to_list_doc,
|
|
|
|
".. method:: to_list()\n"
|
|
|
|
"\n"
|
|
|
|
" Return the array as a list.\n"
|
|
|
|
);
|
2011-06-17 07:45:46 +02:00
|
|
|
static PyObject *BPy_IDArray_to_list(BPy_IDArray *self)
|
2009-11-17 13:54:29 +01:00
|
|
|
{
|
|
|
|
return BPy_IDGroup_MapDataToPy(self->prop);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyMethodDef BPy_IDArray_methods[] = {
|
2016-07-31 07:39:30 +02:00
|
|
|
{"to_list", (PyCFunction)BPy_IDArray_to_list, METH_NOARGS, BPy_IDArray_to_list_doc},
|
2011-02-13 11:52:18 +01:00
|
|
|
{NULL, NULL, 0, NULL}
|
2009-11-17 13:54:29 +01:00
|
|
|
};
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static int BPy_IDArray_Len(BPy_IDArray *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
|
|
|
return self->prop->len;
|
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_IDArray_GetItem(BPy_IDArray *self, int index)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
if (index < 0 || index >= self->prop->len) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_IndexError, "index out of range!");
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
switch (self->prop->subtype) {
|
|
|
|
case IDP_FLOAT:
|
2012-03-16 22:39:56 +01:00
|
|
|
return PyFloat_FromDouble(((float *)IDP_Array(self->prop))[index]);
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
2012-03-16 22:39:56 +01:00
|
|
|
return PyFloat_FromDouble(((double *)IDP_Array(self->prop))[index]);
|
2006-11-17 05:46:48 +01:00
|
|
|
case IDP_INT:
|
2012-03-16 22:39:56 +01:00
|
|
|
return PyLong_FromLong((long)((int *)IDP_Array(self->prop))[index]);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
2009-11-16 19:56:58 +01:00
|
|
|
|
2011-11-15 11:19:44 +01:00
|
|
|
PyErr_Format(PyExc_RuntimeError,
|
|
|
|
"%s: invalid/corrupt array type '%d'!",
|
|
|
|
__func__, self->prop->subtype);
|
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2009-11-16 19:56:58 +01:00
|
|
|
static int BPy_IDArray_SetItem(BPy_IDArray *self, int index, PyObject *value)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2009-11-16 19:56:58 +01:00
|
|
|
if (index < 0 || index >= self->prop->len) {
|
2010-11-23 17:45:17 +01:00
|
|
|
PyErr_SetString(PyExc_RuntimeError, "index out of range!");
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
switch (self->prop->subtype) {
|
|
|
|
case IDP_FLOAT:
|
2013-10-17 04:36:33 +02:00
|
|
|
{
|
|
|
|
const float f = (float)PyFloat_AsDouble(value);
|
2012-03-16 22:39:56 +01:00
|
|
|
if (f == -1 && PyErr_Occurred()) {
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2012-03-16 22:39:56 +01:00
|
|
|
((float *)IDP_Array(self->prop))[index] = f;
|
2006-11-17 05:46:48 +01:00
|
|
|
break;
|
2013-10-17 04:36:33 +02:00
|
|
|
}
|
2008-07-24 21:22:17 +02:00
|
|
|
case IDP_DOUBLE:
|
2013-10-17 04:36:33 +02:00
|
|
|
{
|
|
|
|
const double d = PyFloat_AsDouble(value);
|
2012-03-16 22:39:56 +01:00
|
|
|
if (d == -1 && PyErr_Occurred()) {
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2012-03-16 22:39:56 +01:00
|
|
|
((double *)IDP_Array(self->prop))[index] = d;
|
2008-07-24 21:22:17 +02:00
|
|
|
break;
|
2013-10-17 04:36:33 +02:00
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
case IDP_INT:
|
2013-10-17 04:36:33 +02:00
|
|
|
{
|
2017-08-20 07:44:54 +02:00
|
|
|
const int i = PyC_Long_AsI32(value);
|
2012-03-16 22:39:56 +01:00
|
|
|
if (i == -1 && PyErr_Occurred()) {
|
2009-11-16 19:56:58 +01:00
|
|
|
return -1;
|
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
((int *)IDP_Array(self->prop))[index] = i;
|
2006-11-17 05:46:48 +01:00
|
|
|
break;
|
2013-10-17 04:36:33 +02:00
|
|
|
}
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PySequenceMethods BPy_IDArray_Seq = {
|
2012-03-26 08:55:09 +02: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 05:46:48 +01:00
|
|
|
};
|
|
|
|
|
2011-06-17 07:45:46 +02:00
|
|
|
|
|
|
|
|
|
|
|
/* sequence slice (get): idparr[a:b] */
|
|
|
|
static PyObject *BPy_IDArray_slice(BPy_IDArray *self, int begin, int end)
|
|
|
|
{
|
2012-03-16 22:39:56 +01:00
|
|
|
IDProperty *prop = self->prop;
|
2011-06-17 07:45:46 +02:00
|
|
|
PyObject *tuple;
|
|
|
|
int count;
|
|
|
|
|
|
|
|
CLAMP(begin, 0, prop->len);
|
2012-03-16 22:39:56 +01:00
|
|
|
if (end < 0) end = prop->len + end + 1;
|
2011-06-17 07:45:46 +02:00
|
|
|
CLAMP(end, 0, prop->len);
|
2012-03-16 22:39:56 +01:00
|
|
|
begin = MIN2(begin, end);
|
2011-06-17 07:45:46 +02:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
tuple = PyTuple_New(end - begin);
|
2011-06-17 07:45:46 +02:00
|
|
|
|
|
|
|
switch (prop->subtype) {
|
|
|
|
case IDP_FLOAT:
|
|
|
|
{
|
2014-04-26 16:21:23 +02:00
|
|
|
const float *array = (float *)IDP_Array(prop);
|
2011-10-13 03:29:08 +02:00
|
|
|
for (count = begin; count < end; count++) {
|
2011-06-17 07:45:46 +02:00
|
|
|
PyTuple_SET_ITEM(tuple, count - begin, PyFloat_FromDouble(array[count]));
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case IDP_DOUBLE:
|
|
|
|
{
|
2014-04-26 16:21:23 +02:00
|
|
|
const double *array = (double *)IDP_Array(prop);
|
2011-10-13 03:29:08 +02:00
|
|
|
for (count = begin; count < end; count++) {
|
2011-06-17 07:45:46 +02:00
|
|
|
PyTuple_SET_ITEM(tuple, count - begin, PyFloat_FromDouble(array[count]));
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
case IDP_INT:
|
|
|
|
{
|
2014-04-26 16:21:23 +02:00
|
|
|
const int *array = (int *)IDP_Array(prop);
|
2011-10-13 03:29:08 +02:00
|
|
|
for (count = begin; count < end; count++) {
|
2011-06-17 07:45:46 +02: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 22:39:56 +01:00
|
|
|
IDProperty *prop = self->prop;
|
2014-01-27 17:52:21 +01:00
|
|
|
bool is_double;
|
2012-03-16 22:39:56 +01:00
|
|
|
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 07:45:46 +02: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 22:39:56 +01:00
|
|
|
alloc_len = size * elem_size;
|
2011-06-17 07:45:46 +02:00
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
vec = MEM_mallocN(alloc_len, "array assignment"); /* NOTE: we count on int/float being the same size here */
|
2011-10-13 03:29:08 +02:00
|
|
|
if (PyC_AsArray(vec, seq, size, py_type, is_double, "slice assignment: ") == -1) {
|
2011-06-17 07:45:46 +02: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 22:39:56 +01:00
|
|
|
static PyObject *BPy_IDArray_subscript(BPy_IDArray *self, PyObject *item)
|
2011-06-17 07:45:46 +02: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 14:48:42 +01:00
|
|
|
if (PySlice_GetIndicesEx(item, self->prop->len, &start, &stop, &step, &slicelength) < 0)
|
2011-06-17 07:45:46 +02: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 11:19:44 +01:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"vector indices must be integers, not %.200s",
|
|
|
|
__func__, Py_TYPE(item)->tp_name);
|
2011-06-17 07:45:46 +02:00
|
|
|
return NULL;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2012-03-16 22:39:56 +01:00
|
|
|
static int BPy_IDArray_ass_subscript(BPy_IDArray *self, PyObject *item, PyObject *value)
|
2011-06-17 07:45:46 +02: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 14:48:42 +01:00
|
|
|
if (PySlice_GetIndicesEx(item, self->prop->len, &start, &stop, &step, &slicelength) < 0)
|
2011-06-17 07:45:46 +02: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 11:19:44 +01:00
|
|
|
PyErr_Format(PyExc_TypeError,
|
|
|
|
"vector indices must be integers, not %.200s",
|
|
|
|
Py_TYPE(item)->tp_name);
|
2011-06-17 07:45:46 +02:00
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyMappingMethods BPy_IDArray_AsMapping = {
|
|
|
|
(lenfunc)BPy_IDArray_Len,
|
|
|
|
(binaryfunc)BPy_IDArray_subscript,
|
|
|
|
(objobjargproc)BPy_IDArray_ass_subscript
|
|
|
|
};
|
|
|
|
|
2017-04-10 13:06:00 +02:00
|
|
|
static int itemsize_by_idarray_type(int array_type)
|
|
|
|
{
|
|
|
|
if (array_type == IDP_INT) return sizeof(int);
|
|
|
|
if (array_type == IDP_FLOAT) return sizeof(float);
|
|
|
|
if (array_type == IDP_DOUBLE) return sizeof(double);
|
|
|
|
return -1; /* should never happen */
|
|
|
|
}
|
|
|
|
|
|
|
|
static int BPy_IDArray_getbuffer(BPy_IDArray *self, Py_buffer *view, int flags)
|
|
|
|
{
|
|
|
|
IDProperty *prop = self->prop;
|
|
|
|
int itemsize = itemsize_by_idarray_type(prop->subtype);
|
|
|
|
int length = itemsize * prop->len;
|
|
|
|
|
|
|
|
if (PyBuffer_FillInfo(view, (PyObject *)self, IDP_Array(prop), length, false, flags) == -1) {
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
view->itemsize = itemsize;
|
|
|
|
view->format = (char *)idp_format_from_array_type(prop->subtype);
|
|
|
|
|
|
|
|
Py_ssize_t *shape = MEM_mallocN(sizeof(Py_ssize_t), __func__);
|
|
|
|
shape[0] = prop->len;
|
|
|
|
view->shape = shape;
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
static void BPy_IDArray_releasebuffer(BPy_IDArray *UNUSED(self), Py_buffer *view)
|
|
|
|
{
|
|
|
|
MEM_freeN(view->shape);
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyBufferProcs BPy_IDArray_Buffer = {
|
|
|
|
(getbufferproc)BPy_IDArray_getbuffer,
|
|
|
|
(releasebufferproc)BPy_IDArray_releasebuffer,
|
|
|
|
};
|
|
|
|
|
2011-06-17 07:45:46 +02:00
|
|
|
|
|
|
|
PyTypeObject BPy_IDArray_Type = {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
2006-11-17 05:46:48 +01:00
|
|
|
/* For printing, in format "<module>.<name>" */
|
2012-04-15 16:54:15 +02:00
|
|
|
"IDPropertyArray", /* char *tp_name; */
|
2011-06-17 07:45:46 +02:00
|
|
|
sizeof(BPy_IDArray), /* int tp_basicsize; */
|
2006-11-17 05:46:48 +01:00
|
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
NULL, /* destructor tp_dealloc; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* printfunc tp_print; */
|
2006-12-17 00:54:45 +01:00
|
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
|
|
NULL, /* setattrfunc tp_setattr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* cmpfunc tp_compare; */
|
2011-06-17 07:45:46 +02:00
|
|
|
(reprfunc)BPy_IDArray_repr, /* reprfunc tp_repr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
2012-03-26 08:55:09 +02:00
|
|
|
&BPy_IDArray_Seq, /* PySequenceMethods *tp_as_sequence; */
|
2011-06-17 07:45:46 +02:00
|
|
|
&BPy_IDArray_AsMapping, /* PyMappingMethods *tp_as_mapping; */
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
NULL, /* hashfunc tp_hash; */
|
2006-11-17 05:46:48 +01: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 */
|
2017-04-10 13:06:00 +02:00
|
|
|
&BPy_IDArray_Buffer, /* PyBufferProcs *tp_as_buffer; */
|
2006-11-17 05:46:48 +01:00
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Assigned meaning in release 2.0 ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
/* call function for all accessible objects */
|
|
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
|
|
|
|
/* delete references to contained objects */
|
|
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
|
|
/*** rich comparisons ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** weak reference enabler ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Added in release 2.2 ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
/* Iterators */
|
|
|
|
NULL, /* getiterfunc tp_iter; */
|
|
|
|
NULL, /* iternextfunc tp_iternext; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Attribute descriptor and subclassing stuff ***/
|
|
|
|
BPy_IDArray_methods, /* struct PyMethodDef *tp_methods; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* struct PyMemberDef *tp_members; */
|
2006-12-17 00:54:45 +01:00
|
|
|
BPy_IDArray_getseters, /* struct PyGetSetDef *tp_getset; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* struct _typeobject *tp_base; */
|
|
|
|
NULL, /* PyObject *tp_dict; */
|
|
|
|
NULL, /* descrgetfunc tp_descr_get; */
|
|
|
|
NULL, /* descrsetfunc tp_descr_set; */
|
|
|
|
0, /* long tp_dictoffset; */
|
|
|
|
NULL, /* initproc tp_init; */
|
|
|
|
NULL, /* allocfunc tp_alloc; */
|
|
|
|
NULL, /* newfunc tp_new; */
|
|
|
|
/* Low-level free-memory routine */
|
|
|
|
NULL, /* freefunc tp_free; */
|
|
|
|
/* For PyObject_IS_GC */
|
|
|
|
NULL, /* inquiry tp_is_gc; */
|
|
|
|
NULL, /* PyObject *tp_bases; */
|
|
|
|
/* method resolution order */
|
|
|
|
NULL, /* PyObject *tp_mro; */
|
|
|
|
NULL, /* PyObject *tp_cache; */
|
|
|
|
NULL, /* PyObject *tp_subclasses; */
|
|
|
|
NULL, /* PyObject *tp_weaklist; */
|
|
|
|
NULL
|
|
|
|
};
|
|
|
|
|
|
|
|
/*********** ID Property Group iterator ********/
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *IDGroup_Iter_repr(BPy_IDGroup_Iter *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
2010-10-14 01:25:08 +02:00
|
|
|
return PyUnicode_FromFormat("(ID Property Group Iter \"%s\")", self->group->prop->name);
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
|
2008-09-29 19:08:11 +02:00
|
|
|
static PyObject *BPy_Group_Iter_Next(BPy_IDGroup_Iter *self)
|
2006-11-17 05:46:48 +01:00
|
|
|
{
|
|
|
|
|
|
|
|
if (self->cur) {
|
2012-03-16 22:39:56 +01:00
|
|
|
PyObject *ret;
|
|
|
|
IDProperty *cur;
|
|
|
|
|
2006-11-17 05:46:48 +01:00
|
|
|
cur = self->cur;
|
|
|
|
self->cur = self->cur->next;
|
2012-03-16 22:39:56 +01:00
|
|
|
|
2006-12-17 00:54:45 +01:00
|
|
|
if (self->mode == IDPROP_ITER_ITEMS) {
|
2009-11-16 21:16:45 +01:00
|
|
|
ret = PyTuple_New(2);
|
2015-01-06 06:42:22 +01:00
|
|
|
PyTuple_SET_ITEMS(ret,
|
|
|
|
PyUnicode_FromString(cur->name),
|
|
|
|
BPy_IDGroup_WrapData(self->group->id, cur, self->group->prop));
|
2007-05-26 06:39:31 +02:00
|
|
|
return ret;
|
2011-03-19 12:12:48 +01:00
|
|
|
}
|
|
|
|
else {
|
2009-11-16 19:56:58 +01:00
|
|
|
return PyUnicode_FromString(cur->name);
|
2006-12-17 00:54:45 +01:00
|
|
|
}
|
2011-03-19 12:12:48 +01:00
|
|
|
}
|
|
|
|
else {
|
2013-11-17 05:09:57 +01:00
|
|
|
PyErr_SetNone(PyExc_StopIteration);
|
2009-11-16 19:56:58 +01:00
|
|
|
return NULL;
|
2006-11-17 05:46:48 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2011-06-17 07:45:46 +02:00
|
|
|
PyTypeObject BPy_IDGroup_Iter_Type = {
|
2009-11-16 19:56:58 +01:00
|
|
|
PyVarObject_HEAD_INIT(NULL, 0)
|
2006-11-17 05:46:48 +01:00
|
|
|
/* For printing, in format "<module>.<name>" */
|
2012-04-15 16:54:15 +02:00
|
|
|
"IDPropertyGroupIter", /* char *tp_name; */
|
2012-03-16 22:39:56 +01:00
|
|
|
sizeof(BPy_IDGroup_Iter), /* int tp_basicsize; */
|
2006-11-17 05:46:48 +01:00
|
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
NULL, /* destructor tp_dealloc; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* printfunc tp_print; */
|
2006-12-17 00:54:45 +01:00
|
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
|
|
NULL, /* setattrfunc tp_setattr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* cmpfunc tp_compare; */
|
2012-03-16 22:39:56 +01:00
|
|
|
(reprfunc) IDGroup_Iter_repr, /* reprfunc tp_repr; */
|
2006-11-17 05:46:48 +01:00
|
|
|
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
2012-03-26 08:55:09 +02:00
|
|
|
NULL, /* PySequenceMethods *tp_as_sequence; */
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* PyMappingMethods *tp_as_mapping; */
|
|
|
|
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
|
|
|
|
NULL, /* hashfunc tp_hash; */
|
|
|
|
NULL, /* ternaryfunc tp_call; */
|
|
|
|
NULL, /* reprfunc tp_str; */
|
|
|
|
NULL, /* getattrofunc tp_getattro; */
|
|
|
|
NULL, /* setattrofunc tp_setattro; */
|
|
|
|
|
|
|
|
/* Functions to access object as input/output buffer */
|
|
|
|
NULL, /* PyBufferProcs *tp_as_buffer; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Assigned meaning in release 2.0 ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
/* call function for all accessible objects */
|
|
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
|
|
|
|
/* delete references to contained objects */
|
|
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
|
|
/*** rich comparisons ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** weak reference enabler ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
|
2012-03-26 08:55:09 +02:00
|
|
|
/*** Added in release 2.2 ***/
|
2006-11-17 05:46:48 +01:00
|
|
|
/* Iterators */
|
2012-03-09 01:01:38 +01:00
|
|
|
PyObject_SelfIter, /* getiterfunc tp_iter; */
|
2006-11-17 05:46:48 +01:00
|
|
|
(iternextfunc) BPy_Group_Iter_Next, /* iternextfunc tp_iternext; */
|
|
|
|
};
|
|
|
|
|
2006-11-30 00:31:46 +01:00
|
|
|
void IDProp_Init_Types(void)
|
|
|
|
{
|
2011-06-17 07:45:46 +02:00
|
|
|
PyType_Ready(&BPy_IDGroup_Type);
|
|
|
|
PyType_Ready(&BPy_IDGroup_Iter_Type);
|
|
|
|
PyType_Ready(&BPy_IDArray_Type);
|
2006-11-30 00:31:46 +01:00
|
|
|
}
|
2012-04-15 16:54:15 +02:00
|
|
|
|
|
|
|
/*----------------------------MODULE INIT-------------------------*/
|
|
|
|
|
|
|
|
/* --- */
|
|
|
|
|
|
|
|
static struct PyModuleDef IDProp_types_module_def = {
|
2012-11-03 16:35:03 +01: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 16:54:15 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
static PyObject *BPyInit_idprop_types(void)
|
|
|
|
{
|
|
|
|
PyObject *submodule;
|
|
|
|
|
|
|
|
submodule = PyModule_Create(&IDProp_types_module_def);
|
|
|
|
|
2013-02-06 14:14:11 +01:00
|
|
|
IDProp_Init_Types();
|
|
|
|
|
2012-04-15 16:54:15 +02:00
|
|
|
#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);
|
|
|
|
|
2013-02-06 14:07:45 +01:00
|
|
|
/* idprop.types */
|
2012-04-15 16:54:15 +02:00
|
|
|
PyModule_AddObject(mod, "types", (submodule = BPyInit_idprop_types()));
|
2016-07-14 07:51:56 +02:00
|
|
|
PyDict_SetItem(sys_modules, PyModule_GetNameObject(submodule), submodule);
|
2012-04-15 16:54:15 +02:00
|
|
|
Py_INCREF(submodule);
|
|
|
|
|
|
|
|
return mod;
|
|
|
|
}
|
2012-10-31 21:29:32 +01:00
|
|
|
|
|
|
|
/* -------------------------------------------------------------------- */
|
|
|
|
/* debug only function */
|
|
|
|
|
2018-05-04 07:26:42 +02:00
|
|
|
char *IDP_reprN(const IDProperty *prop)
|
2012-10-31 21:29:32 +01:00
|
|
|
{
|
2018-05-04 07:26:42 +02:00
|
|
|
if (prop == NULL) {
|
|
|
|
return BLI_strdup("None");
|
|
|
|
}
|
2012-10-31 21:29:32 +01:00
|
|
|
|
2018-05-04 07:26:42 +02:00
|
|
|
PyGILState_STATE gilstate;
|
|
|
|
bool use_gil = true; /* !PyC_IsInterpreterActive(); */
|
|
|
|
PyObject *ret_dict;
|
|
|
|
PyObject *ret_str;
|
2012-10-31 21:29:32 +01:00
|
|
|
|
2018-05-04 07:26:42 +02:00
|
|
|
if (use_gil) {
|
|
|
|
gilstate = PyGILState_Ensure();
|
|
|
|
}
|
2012-10-31 21:29:32 +01:00
|
|
|
|
2018-05-04 07:26:42 +02:00
|
|
|
/* Note: non-const cast is safe here since we only repr the result. */
|
|
|
|
/* to_dict() */
|
|
|
|
ret_dict = BPy_IDGroup_MapDataToPy((IDProperty *)prop);
|
|
|
|
ret_str = PyObject_Repr(ret_dict);
|
|
|
|
Py_DECREF(ret_dict);
|
2012-10-31 21:29:32 +01:00
|
|
|
|
2018-05-04 07:26:42 +02:00
|
|
|
Py_ssize_t res_str_len = 0;
|
|
|
|
char *res_str_bytes = _PyUnicode_AsStringAndSize(ret_str, &res_str_len);
|
2012-10-31 21:29:32 +01:00
|
|
|
|
2018-05-04 07:26:42 +02:00
|
|
|
res_str_bytes = BLI_strdupn(res_str_bytes, res_str_len);
|
|
|
|
|
|
|
|
Py_DECREF(ret_str);
|
|
|
|
|
|
|
|
if (use_gil) {
|
|
|
|
PyGILState_Release(gilstate);
|
2012-10-31 21:29:32 +01:00
|
|
|
}
|
2018-05-04 07:26:42 +02:00
|
|
|
return res_str_bytes;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
void IDP_print(const IDProperty *prop)
|
|
|
|
{
|
|
|
|
char *repr = IDP_reprN(prop);
|
|
|
|
printf("IDProperty(%p): %s\n", prop, repr);
|
|
|
|
MEM_freeN(repr);
|
2012-10-31 21:29:32 +01:00
|
|
|
}
|
|
|
|
|