Campbell Barton
87d5e34453
Regression in D1812: PyDriver variables as Objects Taking the Python representation is nice in general but for enums it would convert them into strings, breaking some existing drivers.
109 lines
2.8 KiB
C
109 lines
2.8 KiB
C
/*
|
|
* ***** BEGIN GPL 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.
|
|
*
|
|
* 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*
|
|
* ***** END GPL LICENSE BLOCK *****
|
|
*/
|
|
|
|
/** \file blender/python/intern/bpy_rna_driver.c
|
|
* \ingroup pythonintern
|
|
*
|
|
* This file defines utility functions that use the RNA API, from PyDrivers.
|
|
*/
|
|
|
|
#include <Python.h>
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
#include "BLI_utildefines.h"
|
|
|
|
#include "BKE_fcurve.h"
|
|
|
|
#include "RNA_access.h"
|
|
|
|
#include "bpy_rna.h"
|
|
|
|
#include "bpy_rna_driver.h" /* own include */
|
|
|
|
|
|
/**
|
|
* A version of #driver_get_variable_value which returns a PyObject.
|
|
*/
|
|
PyObject *pyrna_driver_get_variable_value(
|
|
struct ChannelDriver *driver, struct DriverTarget *dtar)
|
|
{
|
|
PyObject *driver_arg = NULL;
|
|
PointerRNA ptr;
|
|
PropertyRNA *prop = NULL;
|
|
int index;
|
|
|
|
if (driver_get_variable_property(driver, dtar, &ptr, &prop, &index)) {
|
|
if (prop) {
|
|
if (index != -1) {
|
|
if (index < RNA_property_array_length(&ptr, prop) && index >= 0) {
|
|
/* object, property & index */
|
|
driver_arg = pyrna_array_index(&ptr, prop, index);
|
|
}
|
|
else {
|
|
/* out of range, pass */
|
|
}
|
|
}
|
|
else {
|
|
/* object & property */
|
|
PropertyType type = RNA_property_type(prop);
|
|
if (type == PROP_ENUM) {
|
|
/* Note that enum's are converted to strings by default,
|
|
* we want to avoid that, see: T52213 */
|
|
driver_arg = PyLong_FromLong(RNA_property_enum_get(&ptr, prop));
|
|
}
|
|
else {
|
|
driver_arg = pyrna_prop_to_py(&ptr, prop);
|
|
}
|
|
}
|
|
}
|
|
else {
|
|
/* object only */
|
|
driver_arg = pyrna_struct_CreatePyObject(&ptr);
|
|
}
|
|
}
|
|
else {
|
|
/* can't resolve path, pass */
|
|
}
|
|
|
|
return driver_arg;
|
|
}
|
|
|
|
PyObject *pyrna_driver_self_from_anim_rna(PathResolvedRNA *anim_rna)
|
|
{
|
|
return pyrna_struct_CreatePyObject(&anim_rna->ptr);
|
|
}
|
|
|
|
bool pyrna_driver_is_equal_anim_rna(const PathResolvedRNA *anim_rna, const PyObject *py_anim_rna)
|
|
{
|
|
if (BPy_StructRNA_Check(py_anim_rna)) {
|
|
const PointerRNA *ptr_a = &anim_rna->ptr;
|
|
const PointerRNA *ptr_b = &(((const BPy_StructRNA *)py_anim_rna)->ptr);
|
|
|
|
if ((ptr_a->id.data == ptr_b->id.data) &&
|
|
(ptr_a->type == ptr_b->type) &&
|
|
(ptr_a->data == ptr_b->data))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|