2006-05-07 14:57:58 +00:00
|
|
|
/*
|
|
|
|
* $Id$
|
|
|
|
*
|
|
|
|
* ***** BEGIN GPL/BL DUAL LICENSE BLOCK *****
|
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU General Public License
|
|
|
|
* as published by the Free Software Foundation; either version 2
|
|
|
|
* of the License, or (at your option) any later version. The Blender
|
|
|
|
* Foundation also sells licenses for use in proprietary software under
|
|
|
|
* the Blender License. See http://www.blender.org/BL/ for information
|
|
|
|
* about this.
|
|
|
|
*
|
|
|
|
* This program is distributed in the hope that it will be useful,
|
|
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
* GNU General Public License for more details.
|
|
|
|
*
|
|
|
|
* You should have received a copy of the GNU General Public License
|
|
|
|
* along with this program; if not, write to the Free Software Foundation,
|
|
|
|
* Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
|
|
*
|
|
|
|
* The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
|
|
|
|
* All rights reserved.
|
|
|
|
*
|
|
|
|
* This is a new part of Blender.
|
|
|
|
*
|
|
|
|
* Contributor(s): Joseph Gilbert, Ken Hughes
|
|
|
|
*
|
|
|
|
* ***** END GPL/BL DUAL LICENSE BLOCK *****
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include "Constraint.h" /*This must come first*/
|
|
|
|
|
|
|
|
#include "DNA_object_types.h"
|
|
|
|
#include "DNA_effect_types.h"
|
|
|
|
#include "DNA_vec_types.h"
|
|
|
|
#include "DNA_curve_types.h"
|
|
|
|
|
|
|
|
#include "BKE_main.h"
|
|
|
|
#include "BKE_global.h"
|
|
|
|
#include "BKE_library.h"
|
|
|
|
#include "BKE_action.h"
|
2006-05-12 04:34:02 +00:00
|
|
|
#include "BKE_armature.h"
|
2006-05-07 14:57:58 +00:00
|
|
|
#include "BLI_blenlib.h"
|
|
|
|
#include "BIF_editconstraint.h"
|
|
|
|
#include "BSE_editipo.h"
|
|
|
|
#include "MEM_guardedalloc.h"
|
|
|
|
#include "butspace.h"
|
|
|
|
#include "blendef.h"
|
|
|
|
#include "mydevice.h"
|
|
|
|
|
|
|
|
#include "Object.h"
|
|
|
|
#include "NLA.h"
|
|
|
|
#include "gen_utils.h"
|
|
|
|
|
|
|
|
enum constraint_constants {
|
|
|
|
EXPP_CONSTR_XROT = 0,
|
|
|
|
EXPP_CONSTR_YROT = 1,
|
|
|
|
EXPP_CONSTR_ZROT = 2,
|
|
|
|
|
|
|
|
EXPP_CONSTR_MAXX = TRACK_X,
|
|
|
|
EXPP_CONSTR_MAXY = TRACK_Y,
|
|
|
|
EXPP_CONSTR_MAXZ = TRACK_Z,
|
|
|
|
EXPP_CONSTR_MINX = TRACK_nX,
|
|
|
|
EXPP_CONSTR_MINY = TRACK_nY,
|
|
|
|
EXPP_CONSTR_MINZ = TRACK_nZ,
|
|
|
|
|
|
|
|
EXPP_CONSTR_TARGET = 100,
|
|
|
|
EXPP_CONSTR_TOLERANCE,
|
|
|
|
EXPP_CONSTR_ITERATIONS,
|
|
|
|
EXPP_CONSTR_BONE,
|
|
|
|
EXPP_CONSTR_CHAINLEN,
|
|
|
|
EXPP_CONSTR_POSWEIGHT,
|
|
|
|
EXPP_CONSTR_ROTWEIGHT,
|
|
|
|
EXPP_CONSTR_ROTATE,
|
|
|
|
EXPP_CONSTR_USETIP,
|
|
|
|
|
|
|
|
EXPP_CONSTR_ACTION,
|
|
|
|
EXPP_CONSTR_LOCAL,
|
|
|
|
EXPP_CONSTR_START,
|
|
|
|
EXPP_CONSTR_END,
|
|
|
|
EXPP_CONSTR_MIN,
|
|
|
|
EXPP_CONSTR_MAX,
|
|
|
|
EXPP_CONSTR_KEYON,
|
|
|
|
|
|
|
|
EXPP_CONSTR_TRACK,
|
|
|
|
EXPP_CONSTR_UP,
|
|
|
|
|
|
|
|
EXPP_CONSTR_RESTLENGTH,
|
|
|
|
EXPP_CONSTR_VOLVARIATION,
|
|
|
|
EXPP_CONSTR_VOLUMEMODE,
|
|
|
|
EXPP_CONSTR_PLANE,
|
|
|
|
|
|
|
|
EXPP_CONSTR_FOLLOW,
|
|
|
|
EXPP_CONSTR_OFFSET,
|
|
|
|
EXPP_CONSTR_FORWARD,
|
|
|
|
|
|
|
|
EXPP_CONSTR_LOCK,
|
|
|
|
|
|
|
|
EXPP_CONSTR_MINMAX,
|
|
|
|
EXPP_CONSTR_STICKY,
|
|
|
|
|
|
|
|
EXPP_CONSTR_COPY,
|
|
|
|
};
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python BPy_Constraint methods declarations: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static PyObject *Constraint_getName( BPy_Constraint * self );
|
|
|
|
static int Constraint_setName( BPy_Constraint * self, PyObject *arg );
|
|
|
|
static PyObject *Constraint_getType( BPy_Constraint * self );
|
|
|
|
static PyObject *Constraint_getInfluence( BPy_Constraint * self );
|
|
|
|
static int Constraint_setInfluence( BPy_Constraint * self, PyObject * arg );
|
|
|
|
|
|
|
|
static PyObject *Constraint_insertKey( BPy_Constraint * self, PyObject * arg );
|
|
|
|
|
|
|
|
static PyObject *Constraint_getData( BPy_Constraint * self, PyObject * key );
|
|
|
|
static int Constraint_setData( BPy_Constraint * self, PyObject * key,
|
|
|
|
PyObject * value );
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python BPy_Constraint methods table: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static PyMethodDef BPy_Constraint_methods[] = {
|
|
|
|
/* name, method, flags, doc */
|
|
|
|
{"insertKey", ( PyCFunction ) Constraint_insertKey, METH_VARARGS,
|
|
|
|
"Insert influence keyframe for constraint"},
|
|
|
|
{NULL, NULL, 0, NULL}
|
|
|
|
};
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python BPy_Constraint attributes get/set structure: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static PyGetSetDef BPy_Constraint_getseters[] = {
|
|
|
|
{"name",
|
|
|
|
(getter)Constraint_getName, (setter)Constraint_setName,
|
|
|
|
"Constraint name", NULL},
|
|
|
|
{"type",
|
|
|
|
(getter)Constraint_getType, (setter)NULL,
|
|
|
|
"Constraint type (read only)", NULL},
|
|
|
|
{"influence",
|
|
|
|
(getter)Constraint_getInfluence, (setter)Constraint_setInfluence,
|
|
|
|
"Constraint influence", NULL},
|
|
|
|
{NULL,NULL,NULL,NULL,NULL} /* Sentinel */
|
|
|
|
};
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python Constraint_Type Mapping Methods table: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static PyMappingMethods Constraint_as_mapping = {
|
|
|
|
NULL, /* mp_length */
|
|
|
|
( binaryfunc ) Constraint_getData, /* mp_subscript */
|
|
|
|
( objobjargproc ) Constraint_setData, /* mp_ass_subscript */
|
|
|
|
};
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python Constraint_Type callback function prototypes: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static void Constraint_dealloc( BPy_Constraint * self );
|
|
|
|
static PyObject *Constraint_repr( BPy_Constraint * self );
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python Constraint_Type structure definition: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
PyTypeObject Constraint_Type = {
|
|
|
|
PyObject_HEAD_INIT( NULL ) /* required py macro */
|
|
|
|
0, /* ob_size */
|
|
|
|
/* For printing, in format "<module>.<name>" */
|
|
|
|
"Blender Constraint", /* char *tp_name; */
|
|
|
|
sizeof( BPy_Constraint ), /* int tp_basicsize; */
|
|
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
|
|
|
|
( destructor ) Constraint_dealloc,/* destructor tp_dealloc; */
|
|
|
|
NULL, /* printfunc tp_print; */
|
|
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
|
|
NULL, /* setattrfunc tp_setattr; */
|
|
|
|
NULL, /* cmpfunc tp_compare; */
|
|
|
|
( reprfunc ) Constraint_repr, /* reprfunc tp_repr; */
|
|
|
|
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
|
|
|
NULL, /* PySequenceMethods *tp_as_sequence; */
|
|
|
|
&Constraint_as_mapping, /* PyMappingMethods *tp_as_mapping; */
|
|
|
|
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
|
|
|
|
NULL, /* hashfunc tp_hash; */
|
|
|
|
NULL, /* ternaryfunc tp_call; */
|
|
|
|
NULL, /* reprfunc tp_str; */
|
|
|
|
NULL, /* getattrofunc tp_getattro; */
|
|
|
|
NULL, /* setattrofunc tp_setattro; */
|
|
|
|
|
|
|
|
/* Functions to access object as input/output buffer */
|
|
|
|
NULL, /* PyBufferProcs *tp_as_buffer; */
|
|
|
|
|
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
|
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
|
|
|
/*** Assigned meaning in release 2.0 ***/
|
|
|
|
/* call function for all accessible objects */
|
|
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
|
|
|
|
/* delete references to contained objects */
|
|
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
|
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
|
|
/*** rich comparisons ***/
|
|
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
|
|
|
|
/*** weak reference enabler ***/
|
|
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
|
|
|
|
/*** Added in release 2.2 ***/
|
|
|
|
/* Iterators */
|
|
|
|
NULL, /* getiterfunc tp_iter; */
|
|
|
|
NULL, /* iternextfunc tp_iternext; */
|
|
|
|
|
|
|
|
/*** Attribute descriptor and subclassing stuff ***/
|
|
|
|
BPy_Constraint_methods, /* struct PyMethodDef *tp_methods; */
|
|
|
|
NULL, /* struct PyMemberDef *tp_members; */
|
|
|
|
BPy_Constraint_getseters, /* struct PyGetSetDef *tp_getset; */
|
|
|
|
NULL, /* struct _typeobject *tp_base; */
|
|
|
|
NULL, /* PyObject *tp_dict; */
|
|
|
|
NULL, /* descrgetfunc tp_descr_get; */
|
|
|
|
NULL, /* descrsetfunc tp_descr_set; */
|
|
|
|
0, /* long tp_dictoffset; */
|
|
|
|
NULL, /* initproc tp_init; */
|
|
|
|
NULL, /* allocfunc tp_alloc; */
|
|
|
|
NULL, /* newfunc tp_new; */
|
|
|
|
/* Low-level free-memory routine */
|
|
|
|
NULL, /* freefunc tp_free; */
|
|
|
|
/* For PyObject_IS_GC */
|
|
|
|
NULL, /* inquiry tp_is_gc; */
|
|
|
|
NULL, /* PyObject *tp_bases; */
|
|
|
|
/* method resolution order */
|
|
|
|
NULL, /* PyObject *tp_mro; */
|
|
|
|
NULL, /* PyObject *tp_cache; */
|
|
|
|
NULL, /* PyObject *tp_subclasses; */
|
|
|
|
NULL, /* PyObject *tp_weaklist; */
|
|
|
|
NULL
|
|
|
|
};
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python BPy_Constraint methods: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
|
|
|
|
/*
|
|
|
|
* return the name of this constraint
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *Constraint_getName( BPy_Constraint * self )
|
|
|
|
{
|
|
|
|
if( !self->con )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
return PyString_FromString( self->con->name );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* set the name of this constraint
|
|
|
|
*/
|
|
|
|
|
|
|
|
static int Constraint_setName( BPy_Constraint * self, PyObject * attr )
|
|
|
|
{
|
|
|
|
char *name = PyString_AsString( attr );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError, "expected string arg" );
|
|
|
|
|
|
|
|
if( !self->con )
|
|
|
|
return EXPP_ReturnIntError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
BLI_strncpy( self->con->name, name, sizeof( self->con->name ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* return the influence of this constraint
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *Constraint_getInfluence( BPy_Constraint * self )
|
|
|
|
{
|
|
|
|
if( !self->con )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
return PyFloat_FromDouble( (double)self->con->enforce );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* set the influence of this constraint
|
|
|
|
*/
|
|
|
|
|
|
|
|
static int Constraint_setInfluence( BPy_Constraint * self, PyObject * value )
|
|
|
|
{
|
|
|
|
if( !self->con )
|
|
|
|
return EXPP_ReturnIntError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
return EXPP_setFloatClamped( value, &self->con->enforce, 0.0, 1.0 );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* return the type of this constraint
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *Constraint_getType( BPy_Constraint * self )
|
|
|
|
{
|
|
|
|
if( !self->con )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
return PyInt_FromLong( self->con->type );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* add keyframe for influence
|
|
|
|
base on code in add_influence_key_to_constraint_func()
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *Constraint_insertKey( BPy_Constraint * self, PyObject * arg )
|
|
|
|
{
|
|
|
|
IpoCurve *icu;
|
|
|
|
float cfra;
|
|
|
|
char actname[32] = "";
|
|
|
|
Object *ob = self->obj;
|
|
|
|
bConstraint *con = self->con;
|
|
|
|
|
|
|
|
if( !self->con )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
/* get frame for inserting key */
|
|
|
|
if( !PyArg_ParseTuple( arg, "f", &cfra ) )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
|
|
"expected a float argument" );
|
|
|
|
|
|
|
|
// constraint_active_func(ob_v, con_v);
|
|
|
|
get_constraint_ipo_context( ob, actname );
|
|
|
|
icu= verify_ipocurve((ID *)ob, ID_CO, actname, con->name, CO_ENFORCE);
|
|
|
|
|
|
|
|
if( ob->action )
|
|
|
|
insert_vert_ipo( icu, get_action_frame(ob, cfra), con->enforce);
|
|
|
|
else
|
|
|
|
insert_vert_ipo( icu, cfra, con->enforce);
|
|
|
|
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Specific constraint get/set procedures */
|
|
|
|
/*****************************************************************************/
|
|
|
|
|
|
|
|
static PyObject *kinematic_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bKinematicConstraint *con = (bKinematicConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_TOLERANCE:
|
|
|
|
return PyFloat_FromDouble( (double)con->tolerance );
|
|
|
|
case EXPP_CONSTR_ITERATIONS:
|
|
|
|
return PyInt_FromLong( (long)con->iterations );
|
|
|
|
case EXPP_CONSTR_CHAINLEN:
|
|
|
|
return PyInt_FromLong( (long)con->rootbone );
|
|
|
|
case EXPP_CONSTR_POSWEIGHT:
|
|
|
|
return PyFloat_FromDouble( (double)con->weight );
|
|
|
|
case EXPP_CONSTR_ROTWEIGHT:
|
|
|
|
return PyFloat_FromDouble( (double)con->orientweight );
|
|
|
|
case EXPP_CONSTR_ROTATE:
|
|
|
|
return PyBool_FromLong( (long)( con->flag & CONSTRAINT_IK_ROT ) ) ;
|
|
|
|
case EXPP_CONSTR_USETIP:
|
|
|
|
return PyBool_FromLong( (long)( con->flag & CONSTRAINT_IK_TIP ) ) ;
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int kinematic_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bKinematicConstraint *con = (bKinematicConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_TOLERANCE:
|
|
|
|
return EXPP_setFloatClamped( value, &con->tolerance, 0.0001, 1.0 );
|
|
|
|
case EXPP_CONSTR_ITERATIONS:
|
|
|
|
return EXPP_setIValueClamped( value, &con->iterations, 1, 10000, 'h' );
|
|
|
|
case EXPP_CONSTR_CHAINLEN:
|
|
|
|
return EXPP_setIValueClamped( value, &con->rootbone, 0, 255, 'i' );
|
|
|
|
case EXPP_CONSTR_POSWEIGHT:
|
|
|
|
return EXPP_setFloatClamped( value, &con->weight, 0.01, 1.0 );
|
|
|
|
case EXPP_CONSTR_ROTWEIGHT:
|
|
|
|
return EXPP_setFloatClamped( value, &con->orientweight, 0.01, 1.0 );
|
|
|
|
case EXPP_CONSTR_ROTATE:
|
|
|
|
return EXPP_setBitfield( value, &con->flag, CONSTRAINT_IK_ROT, 'h' );
|
|
|
|
case EXPP_CONSTR_USETIP:
|
|
|
|
return EXPP_setBitfield( value, &con->flag, CONSTRAINT_IK_TIP, 'h' );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *action_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bActionConstraint *con = (bActionConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_ACTION:
|
|
|
|
return Action_CreatePyObject( con->act );
|
|
|
|
case EXPP_CONSTR_LOCAL:
|
|
|
|
return PyBool_FromLong( (long)( con->local & SELECT ) );
|
|
|
|
case EXPP_CONSTR_START:
|
|
|
|
return PyInt_FromLong( (long)con->start );
|
|
|
|
case EXPP_CONSTR_END:
|
|
|
|
return PyInt_FromLong( (long)con->end );
|
|
|
|
case EXPP_CONSTR_MIN:
|
|
|
|
return PyFloat_FromDouble( (double)con->min );
|
|
|
|
case EXPP_CONSTR_MAX:
|
|
|
|
return PyFloat_FromDouble( (double)con->max );
|
|
|
|
case EXPP_CONSTR_KEYON:
|
|
|
|
return PyInt_FromLong( (long)con->type );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int action_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bActionConstraint *con = (bActionConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_ACTION: {
|
|
|
|
bAction *act = (( BPy_Action * )value)->action;
|
|
|
|
if( !Action_CheckPyObject( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy action argument" );
|
|
|
|
con->act = act;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_LOCAL:
|
|
|
|
return EXPP_setBitfield( value, &con->local, SELECT, 'h' );
|
|
|
|
case EXPP_CONSTR_START:
|
|
|
|
return EXPP_setIValueClamped( value, &con->start, 1, MAXFRAME, 'h' );
|
|
|
|
case EXPP_CONSTR_END:
|
|
|
|
return EXPP_setIValueClamped( value, &con->end, 1, MAXFRAME, 'h' );
|
|
|
|
case EXPP_CONSTR_MIN:
|
|
|
|
return EXPP_setFloatClamped( value, &con->min, -180.0, 180.0 );
|
|
|
|
case EXPP_CONSTR_MAX:
|
|
|
|
return EXPP_setFloatClamped( value, &con->max, -180.0, 180.0 );
|
|
|
|
case EXPP_CONSTR_KEYON:
|
|
|
|
return EXPP_setIValueRange( value, &con->type,
|
|
|
|
EXPP_CONSTR_XROT, EXPP_CONSTR_ZROT, 'h' );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *trackto_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bTrackToConstraint *con = (bTrackToConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_TRACK:
|
|
|
|
return PyInt_FromLong( (long)con->reserved1 );
|
|
|
|
case EXPP_CONSTR_UP:
|
|
|
|
return PyInt_FromLong( (long)con->reserved2 );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int trackto_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bTrackToConstraint *con = (bTrackToConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_TRACK:
|
|
|
|
return EXPP_setIValueRange( value, &con->reserved1,
|
|
|
|
TRACK_X, TRACK_nZ, 'i' );
|
|
|
|
case EXPP_CONSTR_UP:
|
|
|
|
return EXPP_setIValueRange( value, &con->reserved2,
|
|
|
|
UP_X, UP_Z, 'i' );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *stretchto_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bStretchToConstraint *con = (bStretchToConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_RESTLENGTH:
|
|
|
|
return PyFloat_FromDouble( (double)con->orglength );
|
|
|
|
case EXPP_CONSTR_VOLVARIATION:
|
|
|
|
return PyFloat_FromDouble( (double)con->bulge );
|
|
|
|
case EXPP_CONSTR_VOLUMEMODE:
|
|
|
|
return PyInt_FromLong( (long)con->volmode );
|
|
|
|
case EXPP_CONSTR_PLANE:
|
|
|
|
return PyInt_FromLong( (long)con->plane );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int stretchto_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bStretchToConstraint *con = (bStretchToConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_RESTLENGTH:
|
|
|
|
return EXPP_setFloatClamped( value, &con->orglength, 0.0, 100.0 );
|
|
|
|
case EXPP_CONSTR_VOLVARIATION:
|
|
|
|
return EXPP_setFloatClamped( value, &con->bulge, 0.0, 100.0 );
|
|
|
|
case EXPP_CONSTR_VOLUMEMODE:
|
|
|
|
return EXPP_setIValueRange( value, &con->volmode,
|
|
|
|
VOLUME_XZ, NO_VOLUME, 'h' );
|
|
|
|
case EXPP_CONSTR_PLANE: {
|
|
|
|
int status, oldcode = con->plane;
|
|
|
|
status = EXPP_setIValueRange( value, &con->plane,
|
|
|
|
PLANE_X, PLANE_Z, 'h' );
|
|
|
|
if( !status && con->plane == PLANE_Y ) {
|
|
|
|
con->plane = oldcode;
|
|
|
|
return EXPP_ReturnIntError( PyExc_ValueError,
|
|
|
|
"value must be either PLANEX or PLANEZ" );
|
|
|
|
}
|
|
|
|
return status;
|
|
|
|
}
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *followpath_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bFollowPathConstraint *con = (bFollowPathConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_FOLLOW:
|
|
|
|
return PyBool_FromLong( (long)( con->followflag & SELECT ) );
|
|
|
|
case EXPP_CONSTR_OFFSET:
|
|
|
|
return PyFloat_FromDouble( (double)con->offset );
|
|
|
|
case EXPP_CONSTR_FORWARD:
|
|
|
|
return PyInt_FromLong( (long)con->trackflag );
|
|
|
|
case EXPP_CONSTR_UP:
|
|
|
|
return PyInt_FromLong( (long)con->upflag );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int followpath_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bFollowPathConstraint *con = (bFollowPathConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_FOLLOW:
|
|
|
|
return EXPP_setBitfield( value, &con->followflag, SELECT, 'i' );
|
|
|
|
case EXPP_CONSTR_OFFSET:
|
|
|
|
return EXPP_setFloatClamped( value, &con->offset,
|
|
|
|
-MAXFRAMEF, MAXFRAMEF );
|
|
|
|
case EXPP_CONSTR_FORWARD:
|
|
|
|
return EXPP_setIValueRange( value, &con->trackflag,
|
|
|
|
TRACK_X, TRACK_nZ, 'i' );
|
|
|
|
case EXPP_CONSTR_UP:
|
|
|
|
return EXPP_setIValueRange( value, &con->upflag,
|
|
|
|
UP_X, UP_Z, 'i' );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *locktrack_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bLockTrackConstraint *con = (bLockTrackConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_TRACK:
|
|
|
|
return PyInt_FromLong( (long)con->trackflag );
|
|
|
|
case EXPP_CONSTR_LOCK:
|
|
|
|
return PyInt_FromLong( (long)con->lockflag );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int locktrack_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bLockTrackConstraint *con = (bLockTrackConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_TRACK:
|
|
|
|
return EXPP_setIValueRange( value, &con->trackflag,
|
|
|
|
TRACK_X, TRACK_nZ, 'i' );
|
|
|
|
case EXPP_CONSTR_LOCK:
|
|
|
|
return EXPP_setIValueRange( value, &con->lockflag,
|
|
|
|
LOCK_X, LOCK_Z, 'i' );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *floor_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bMinMaxConstraint *con = (bMinMaxConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_MINMAX:
|
|
|
|
return PyInt_FromLong( (long)con->minmaxflag );
|
|
|
|
case EXPP_CONSTR_OFFSET:
|
|
|
|
return PyFloat_FromDouble( (double)con->offset );
|
|
|
|
case EXPP_CONSTR_STICKY:
|
|
|
|
return PyBool_FromLong( (long)( con->sticky & SELECT ) ) ;
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int floor_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bMinMaxConstraint *con = (bMinMaxConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_MINMAX:
|
|
|
|
return EXPP_setIValueRange( value, &con->minmaxflag,
|
|
|
|
EXPP_CONSTR_MAXX, EXPP_CONSTR_MINZ, 'i' );
|
|
|
|
case EXPP_CONSTR_OFFSET:
|
|
|
|
return EXPP_setFloatClamped( value, &con->offset, -100.0, 100.0 );
|
|
|
|
case EXPP_CONSTR_STICKY:
|
|
|
|
return EXPP_setBitfield( value, &con->sticky, SELECT, 'h' );
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *locatelike_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bLocateLikeConstraint *con = (bLocateLikeConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_COPY:
|
|
|
|
return PyInt_FromLong( (long)con->flag );
|
2006-05-12 04:34:02 +00:00
|
|
|
case EXPP_CONSTR_LOCAL:
|
|
|
|
if( get_armature( con->tar ) )
|
2006-05-12 13:10:55 +00:00
|
|
|
return PyBool_FromLong( (long)
|
|
|
|
( self->con->flag & CONSTRAINT_LOCAL ) ) ;
|
2006-05-12 04:34:02 +00:00
|
|
|
Py_RETURN_NONE;
|
2006-05-07 14:57:58 +00:00
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int locatelike_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bLocateLikeConstraint *con = (bLocateLikeConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_COPY:
|
|
|
|
return EXPP_setIValueRange( value, &con->flag,
|
|
|
|
0, LOCLIKE_X | LOCLIKE_Y | LOCLIKE_Z, 'i' );
|
2006-05-12 04:34:02 +00:00
|
|
|
case EXPP_CONSTR_LOCAL:
|
|
|
|
if( !get_armature( con->tar ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_RuntimeError,
|
|
|
|
"only armature targets have LOCAL key" );
|
|
|
|
return EXPP_setBitfield( value, &self->con->flag,
|
|
|
|
CONSTRAINT_LOCAL, 'h' );
|
2006-05-07 14:57:58 +00:00
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *rotatelike_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bRotateLikeConstraint *con = (bRotateLikeConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_COPY:
|
|
|
|
return PyInt_FromLong( (long)con->flag );
|
2006-05-12 04:34:02 +00:00
|
|
|
case EXPP_CONSTR_LOCAL:
|
|
|
|
if( get_armature( con->tar ) )
|
2006-05-12 13:10:55 +00:00
|
|
|
return PyBool_FromLong( (long)
|
|
|
|
( self->con->flag & CONSTRAINT_LOCAL ) ) ;
|
2006-05-12 04:34:02 +00:00
|
|
|
Py_RETURN_NONE;
|
2006-05-07 14:57:58 +00:00
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int rotatelike_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bRotateLikeConstraint *con = (bRotateLikeConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_COPY:
|
|
|
|
return EXPP_setIValueRange( value, &con->flag,
|
|
|
|
0, LOCLIKE_X | LOCLIKE_Y | LOCLIKE_Z, 'i' );
|
2006-05-12 04:34:02 +00:00
|
|
|
case EXPP_CONSTR_LOCAL:
|
|
|
|
if( !get_armature( con->tar ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_RuntimeError,
|
|
|
|
"only armature targets have LOCAL key" );
|
2006-05-12 13:10:55 +00:00
|
|
|
return EXPP_setBitfield( value, &self->con->flag,
|
|
|
|
CONSTRAINT_LOCAL, 'h' );
|
2006-05-07 14:57:58 +00:00
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *sizelike_getter( BPy_Constraint * self, int type )
|
|
|
|
{
|
|
|
|
bSizeLikeConstraint *con = (bSizeLikeConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET:
|
|
|
|
return Object_CreatePyObject( con->tar );
|
|
|
|
case EXPP_CONSTR_BONE:
|
|
|
|
return PyString_FromString( con->subtarget );
|
|
|
|
case EXPP_CONSTR_COPY:
|
|
|
|
return PyInt_FromLong( (long)con->flag );
|
2006-05-12 13:10:55 +00:00
|
|
|
#if 0
|
2006-05-12 04:34:02 +00:00
|
|
|
case EXPP_CONSTR_LOCAL:
|
|
|
|
if( get_armature( con->tar ) )
|
2006-05-12 13:10:55 +00:00
|
|
|
return PyBool_FromLong( (long)
|
|
|
|
( self->con->flag & CONSTRAINT_LOCAL ) ) ;
|
2006-05-12 04:34:02 +00:00
|
|
|
Py_RETURN_NONE;
|
2006-05-12 13:10:55 +00:00
|
|
|
#endif
|
2006-05-07 14:57:58 +00:00
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int sizelike_setter( BPy_Constraint *self, int type, PyObject *value )
|
|
|
|
{
|
|
|
|
bSizeLikeConstraint *con = (bSizeLikeConstraint *)(self->con->data);
|
|
|
|
|
|
|
|
switch( type ) {
|
|
|
|
case EXPP_CONSTR_TARGET: {
|
|
|
|
Object *obj = (( BPy_Object * )value)->object;
|
|
|
|
if( !BPy_Object_Check( value ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected BPy object argument" );
|
|
|
|
con->tar = obj;
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_BONE: {
|
|
|
|
char *name = PyString_AsString( value );
|
|
|
|
if( !name )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected string arg" );
|
|
|
|
|
|
|
|
BLI_strncpy( con->subtarget, name, sizeof( con->subtarget ) );
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
case EXPP_CONSTR_COPY:
|
|
|
|
return EXPP_setIValueRange( value, &con->flag,
|
|
|
|
0, LOCLIKE_X | LOCLIKE_Y | LOCLIKE_Z, 'i' );
|
2006-05-12 13:10:55 +00:00
|
|
|
#if 0
|
2006-05-12 04:34:02 +00:00
|
|
|
case EXPP_CONSTR_LOCAL:
|
|
|
|
if( !get_armature( con->tar ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_RuntimeError,
|
|
|
|
"only armature targets have LOCAL key" );
|
2006-05-12 13:10:55 +00:00
|
|
|
return EXPP_setBitfield( value, &self->con->flag,
|
|
|
|
CONSTRAINT_LOCAL, 'h' );
|
|
|
|
#endif
|
2006-05-07 14:57:58 +00:00
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* get data from a constraint
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *Constraint_getData( BPy_Constraint * self, PyObject * key )
|
|
|
|
{
|
|
|
|
int setting;
|
|
|
|
|
|
|
|
if( !PyInt_CheckExact( key ) )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
|
|
"expected an int arg" );
|
|
|
|
|
|
|
|
if( !self->con )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
setting = PyInt_AsLong( key );
|
|
|
|
switch( self->con->type ) {
|
|
|
|
case CONSTRAINT_TYPE_NULL:
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
case CONSTRAINT_TYPE_TRACKTO:
|
|
|
|
return trackto_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_KINEMATIC:
|
|
|
|
return kinematic_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_FOLLOWPATH:
|
|
|
|
return followpath_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_ACTION:
|
|
|
|
return action_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_LOCKTRACK:
|
|
|
|
return locktrack_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_STRETCHTO:
|
|
|
|
return stretchto_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_MINMAX:
|
|
|
|
return floor_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_LOCLIKE:
|
|
|
|
return locatelike_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_ROTLIKE:
|
|
|
|
return rotatelike_getter( self, setting );
|
|
|
|
case CONSTRAINT_TYPE_SIZELIKE:
|
|
|
|
return sizelike_getter( self, setting );
|
|
|
|
|
|
|
|
case CONSTRAINT_TYPE_CHILDOF: /* Unimplemented */
|
|
|
|
case CONSTRAINT_TYPE_ROTLIMIT:
|
|
|
|
case CONSTRAINT_TYPE_LOCLIMIT:
|
|
|
|
case CONSTRAINT_TYPE_SIZELIMIT:
|
|
|
|
case CONSTRAINT_TYPE_PYTHON:
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError,
|
|
|
|
"unknown constraint type" );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static int Constraint_setData( BPy_Constraint * self, PyObject * key,
|
|
|
|
PyObject * arg )
|
|
|
|
{
|
|
|
|
int key_int, result;
|
|
|
|
|
|
|
|
if( !PyNumber_Check( key ) )
|
|
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
|
|
"expected an int arg" );
|
|
|
|
if( !self->con )
|
|
|
|
return EXPP_ReturnIntError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
key_int = PyInt_AsLong( key );
|
|
|
|
switch( self->con->type ) {
|
|
|
|
case CONSTRAINT_TYPE_KINEMATIC:
|
|
|
|
result = kinematic_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_ACTION:
|
|
|
|
result = action_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_TRACKTO:
|
|
|
|
result = trackto_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_STRETCHTO:
|
|
|
|
result = stretchto_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_FOLLOWPATH:
|
|
|
|
result = followpath_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_LOCKTRACK:
|
|
|
|
result = locktrack_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_MINMAX:
|
|
|
|
result = floor_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_LOCLIKE:
|
|
|
|
result = locatelike_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_ROTLIKE:
|
|
|
|
result = rotatelike_setter( self, key_int, arg );
|
|
|
|
break;
|
|
|
|
case CONSTRAINT_TYPE_SIZELIKE:
|
|
|
|
result = sizelike_setter( self, key_int, arg );
|
|
|
|
break;
|
2006-05-12 04:34:02 +00:00
|
|
|
case CONSTRAINT_TYPE_NULL:
|
|
|
|
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
|
2006-05-07 14:57:58 +00:00
|
|
|
case CONSTRAINT_TYPE_CHILDOF: /* Unimplemented */
|
|
|
|
case CONSTRAINT_TYPE_ROTLIMIT:
|
|
|
|
case CONSTRAINT_TYPE_LOCLIMIT:
|
|
|
|
case CONSTRAINT_TYPE_SIZELIMIT:
|
|
|
|
case CONSTRAINT_TYPE_PYTHON:
|
|
|
|
default:
|
|
|
|
return EXPP_ReturnIntError( PyExc_RuntimeError,
|
|
|
|
"unsupported constraint setting" );
|
|
|
|
}
|
|
|
|
if( !result && self->pchan )
|
|
|
|
update_pose_constraint_flags( self->obj->pose );
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: Constraint_dealloc */
|
|
|
|
/* Description: This is a callback function for the BPy_Constraint type. It */
|
|
|
|
/* destroys data when the object is deleted. */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static void Constraint_dealloc( BPy_Constraint * self )
|
|
|
|
{
|
|
|
|
PyObject_DEL( self );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: Constraint_repr */
|
|
|
|
/* Description: This is a callback function for the BPy_Constraint type. It */
|
|
|
|
/* builds a meaningful string to represent constraint objects. */
|
|
|
|
/*****************************************************************************/
|
|
|
|
|
|
|
|
static PyObject *Constraint_repr( BPy_Constraint * self )
|
|
|
|
{
|
|
|
|
char type[32];
|
|
|
|
|
|
|
|
if( !self->con )
|
|
|
|
return PyString_FromString( "[Constraint - Removed]");
|
|
|
|
|
|
|
|
get_constraint_typestring (type, self->con);
|
|
|
|
return PyString_FromFormat( "[Constraint \"%s\", Type \"%s\"]",
|
|
|
|
self->con->name, type );
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Three Python Constraint_Type helper functions needed by the Object module: */
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: Constraint_CreatePyObject */
|
|
|
|
/* Description: This function will create a new BPy_Constraint from an */
|
|
|
|
/* existing Blender constraint structure. */
|
|
|
|
/*****************************************************************************/
|
|
|
|
PyObject *Constraint_CreatePyObject( bPoseChannel *pchan, Object *obj,
|
|
|
|
bConstraint *con )
|
|
|
|
{
|
|
|
|
BPy_Constraint *pycon;
|
|
|
|
pycon = ( BPy_Constraint * ) PyObject_NEW( BPy_Constraint,
|
|
|
|
&Constraint_Type );
|
|
|
|
if( !pycon )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
|
|
"couldn't create BPy_Constraint object" );
|
|
|
|
|
|
|
|
pycon->con = con;
|
|
|
|
|
|
|
|
/* one of these two will be NULL */
|
|
|
|
pycon->obj = obj;
|
|
|
|
pycon->pchan = pchan;
|
|
|
|
return ( PyObject * ) pycon;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: Constraint_CheckPyObject */
|
|
|
|
/* Description: This function returns true when the given PyObject is of the */
|
|
|
|
/* type Constraint. Otherwise it will return false. */
|
|
|
|
/*****************************************************************************/
|
|
|
|
int Constraint_CheckPyObject( PyObject * pyobj )
|
|
|
|
{
|
|
|
|
return ( pyobj->ob_type == &Constraint_Type );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: Constraint_FromPyObject */
|
|
|
|
/* Description: This function returns the Blender constraint from the given */
|
|
|
|
/* PyObject. */
|
|
|
|
/*****************************************************************************/
|
|
|
|
bConstraint *Constraint_FromPyObject( BPy_Constraint * self )
|
|
|
|
{
|
|
|
|
return self->con;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Constraint Sequence wrapper */
|
|
|
|
/*****************************************************************************/
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Initialize the interator
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *ConstraintSeq_getIter( BPy_ConstraintSeq * self )
|
|
|
|
{
|
|
|
|
if( self->pchan )
|
|
|
|
self->iter = (bConstraint *)self->pchan->constraints.first;
|
|
|
|
else
|
|
|
|
self->iter = (bConstraint *)self->obj->constraints.first;
|
|
|
|
return EXPP_incr_ret ( (PyObject *) self );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Get the next Constraint
|
|
|
|
*/
|
|
|
|
|
|
|
|
static PyObject *ConstraintSeq_nextIter( BPy_ConstraintSeq * self )
|
|
|
|
{
|
|
|
|
bConstraint *this = self->iter;
|
|
|
|
if( this ) {
|
|
|
|
self->iter = this->next;
|
|
|
|
return Constraint_CreatePyObject( self->pchan, self->obj, this );
|
|
|
|
}
|
|
|
|
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_StopIteration,
|
|
|
|
"iterator at end" );
|
|
|
|
}
|
|
|
|
|
|
|
|
/* return the number of constraints */
|
|
|
|
|
|
|
|
static int ConstraintSeq_length( BPy_ConstraintSeq * self )
|
|
|
|
{
|
|
|
|
return BLI_countlist( self->pchan ?
|
|
|
|
&self->pchan->constraints : &self->obj->constraints );
|
|
|
|
}
|
|
|
|
|
|
|
|
/* return a constraint */
|
|
|
|
|
|
|
|
static PyObject *ConstraintSeq_item( BPy_ConstraintSeq * self, int i )
|
|
|
|
{
|
|
|
|
bConstraint *con = NULL;
|
|
|
|
|
|
|
|
/* if index is negative, start counting from the end of the list */
|
|
|
|
if( i < 0 )
|
|
|
|
i += ConstraintSeq_length( self );
|
|
|
|
|
|
|
|
/* skip through the list until we get the constraint or end of list */
|
|
|
|
|
|
|
|
if( self->pchan )
|
|
|
|
con = self->pchan->constraints.first;
|
|
|
|
else
|
|
|
|
con = self->obj->constraints.first;
|
|
|
|
|
|
|
|
while( i && con ) {
|
|
|
|
--i;
|
|
|
|
con = con->next;
|
|
|
|
}
|
|
|
|
|
|
|
|
if( con )
|
|
|
|
return Constraint_CreatePyObject( self->pchan, self->obj, con );
|
|
|
|
else
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_IndexError,
|
|
|
|
"array index out of range" );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python BPy_ConstraintSeq sequence table: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static PySequenceMethods ConstraintSeq_as_sequence = {
|
|
|
|
( inquiry ) ConstraintSeq_length, /* sq_length */
|
|
|
|
( binaryfunc ) 0, /* sq_concat */
|
|
|
|
( intargfunc ) 0, /* sq_repeat */
|
|
|
|
( intargfunc ) ConstraintSeq_item, /* sq_item */
|
|
|
|
( intintargfunc ) 0, /* sq_slice */
|
|
|
|
( intobjargproc ) 0, /* sq_ass_item */
|
|
|
|
( intintobjargproc ) 0, /* sq_ass_slice */
|
|
|
|
( objobjproc ) 0, /* sq_contains */
|
|
|
|
( binaryfunc ) 0, /* sq_inplace_concat */
|
|
|
|
( intargfunc ) 0, /* sq_inplace_repeat */
|
|
|
|
};
|
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
/*
|
|
|
|
* helper function to check for a valid constraint argument
|
|
|
|
*/
|
|
|
|
|
|
|
|
static bConstraint *locate_constr( BPy_ConstraintSeq *self, PyObject * args )
|
|
|
|
{
|
|
|
|
BPy_Constraint *pyobj;
|
|
|
|
bConstraint *con;
|
|
|
|
|
|
|
|
/* check that argument is a modifier */
|
|
|
|
if( !PyArg_ParseTuple( args, "O!", &Constraint_Type, &pyobj ) )
|
|
|
|
return (bConstraint *)EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
|
|
"expected a constraint as an argument" );
|
|
|
|
|
|
|
|
/* check whether constraint has been removed */
|
|
|
|
if( !pyobj->con )
|
|
|
|
return (bConstraint *)EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
|
|
"This constraint has been removed!" );
|
|
|
|
|
|
|
|
/* verify the constraint is still exists in the stack */
|
|
|
|
if( self->pchan )
|
|
|
|
con = self->pchan->constraints.first;
|
|
|
|
else
|
|
|
|
con = self->obj->constraints.first;
|
|
|
|
while( con && con != pyobj->con )
|
|
|
|
con = con->next;
|
|
|
|
|
|
|
|
/* if we didn't find it, exception */
|
|
|
|
if( !con )
|
|
|
|
return (bConstraint *)EXPP_ReturnPyObjError( PyExc_AttributeError,
|
|
|
|
"This constraint is no longer in the object's stack" );
|
|
|
|
|
|
|
|
return con;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2006-05-07 14:57:58 +00:00
|
|
|
/* create a new constraint at the end of the list */
|
|
|
|
|
|
|
|
static PyObject *ConstraintSeq_append( BPy_ConstraintSeq *self, PyObject *args )
|
|
|
|
{
|
|
|
|
int type;
|
|
|
|
bConstraint *con;
|
|
|
|
|
|
|
|
if( !PyArg_ParseTuple( args, "i", &type ) )
|
|
|
|
EXPP_ReturnPyObjError( PyExc_TypeError, "expected int argument" );
|
|
|
|
|
|
|
|
/* type 0 is CONSTRAINT_TYPE_NULL, should we be able to add one of these? */
|
2006-05-12 04:34:02 +00:00
|
|
|
if( type < CONSTRAINT_TYPE_NULL || type > CONSTRAINT_TYPE_MINMAX )
|
2006-05-07 14:57:58 +00:00
|
|
|
return EXPP_ReturnPyObjError( PyExc_ValueError,
|
|
|
|
"int argument out of range" );
|
|
|
|
|
|
|
|
con = add_new_constraint( type );
|
|
|
|
if( self->pchan ) {
|
|
|
|
BLI_addtail( &self->pchan->constraints, con );
|
|
|
|
update_pose_constraint_flags( self->obj->pose );
|
|
|
|
}
|
|
|
|
else
|
|
|
|
BLI_addtail( &self->obj->constraints, con );
|
|
|
|
|
|
|
|
return Constraint_CreatePyObject( self->pchan, self->obj, con );
|
|
|
|
}
|
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
/* move the constraint up in the stack */
|
2006-05-07 14:57:58 +00:00
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
static PyObject *ConstraintSeq_moveUp( BPy_ConstraintSeq *self, PyObject *args )
|
2006-05-07 14:57:58 +00:00
|
|
|
{
|
2006-06-14 05:16:39 +00:00
|
|
|
bConstraint *con = locate_constr( self, args );
|
2006-05-07 14:57:58 +00:00
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
/* if we can't locate the constraint, return (exception already set) */
|
|
|
|
if( !con )
|
|
|
|
return (PyObject *)NULL;
|
2006-05-07 14:57:58 +00:00
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
const_moveUp( self->obj, con );
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
2006-05-07 14:57:58 +00:00
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
/* move the constraint down in the stack */
|
2006-05-07 14:57:58 +00:00
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
static PyObject *ConstraintSeq_moveDown( BPy_ConstraintSeq *self, PyObject *args )
|
|
|
|
{
|
|
|
|
bConstraint *con = locate_constr( self, args );
|
2006-05-07 14:57:58 +00:00
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
/* if we can't locate the constraint, return (exception already set) */
|
2006-05-07 14:57:58 +00:00
|
|
|
if( !con )
|
2006-06-14 05:16:39 +00:00
|
|
|
return (PyObject *)NULL;
|
|
|
|
|
|
|
|
const_moveDown( self->obj, con );
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* remove an existing constraint */
|
|
|
|
|
|
|
|
static PyObject *ConstraintSeq_remove( BPy_ConstraintSeq *self, PyObject *args )
|
|
|
|
{
|
|
|
|
BPy_Constraint *pyobj;
|
|
|
|
bConstraint *con = locate_constr( self, args );
|
|
|
|
|
|
|
|
/* if we can't locate the constraint, return (exception already set) */
|
|
|
|
if( !con )
|
|
|
|
return (PyObject *)NULL;
|
2006-05-07 14:57:58 +00:00
|
|
|
|
|
|
|
/* do the actual removal */
|
|
|
|
if( self->pchan )
|
2006-06-14 05:16:39 +00:00
|
|
|
BLI_remlink( &self->pchan->constraints, con );
|
2006-05-07 14:57:58 +00:00
|
|
|
else
|
2006-06-14 05:16:39 +00:00
|
|
|
BLI_remlink( &self->obj->constraints, con);
|
|
|
|
del_constr_func( self->obj, con );
|
2006-05-07 14:57:58 +00:00
|
|
|
|
2006-06-14 05:16:39 +00:00
|
|
|
/* erase the link to the constraint */
|
|
|
|
pyobj = ( BPy_Constraint * )PyTuple_GET_ITEM( args, 0 );
|
2006-05-07 14:57:58 +00:00
|
|
|
pyobj->con = NULL;
|
2006-06-14 05:16:39 +00:00
|
|
|
|
2006-05-07 14:57:58 +00:00
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: ConstraintSeq_dealloc */
|
|
|
|
/* Description: This is a callback function for the BPy_ConstraintSeq type. */
|
|
|
|
/* It destroys data when the object is deleted. */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static void ConstraintSeq_dealloc( BPy_Constraint * self )
|
|
|
|
{
|
|
|
|
PyObject_DEL( self );
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python BPy_ConstraintSeq methods table: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
static PyMethodDef BPy_ConstraintSeq_methods[] = {
|
|
|
|
/* name, method, flags, doc */
|
|
|
|
{"append", ( PyCFunction ) ConstraintSeq_append, METH_VARARGS,
|
|
|
|
"(type) - add a new constraint, where type is the constraint type"},
|
|
|
|
{"remove", ( PyCFunction ) ConstraintSeq_remove, METH_VARARGS,
|
|
|
|
"(con) - remove an existing constraint, where con is a constraint from this object."},
|
2006-06-14 05:16:39 +00:00
|
|
|
{"moveUp", ( PyCFunction ) ConstraintSeq_moveUp, METH_VARARGS,
|
|
|
|
"(con) - Move constraint up in stack"},
|
|
|
|
{"moveDown", ( PyCFunction ) ConstraintSeq_moveDown, METH_VARARGS,
|
|
|
|
"(con) - Move constraint down in stack"},
|
2006-05-07 14:57:58 +00:00
|
|
|
{NULL, NULL, 0, NULL}
|
|
|
|
};
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Python ConstraintSeq_Type structure definition: */
|
|
|
|
/*****************************************************************************/
|
|
|
|
PyTypeObject ConstraintSeq_Type = {
|
|
|
|
PyObject_HEAD_INIT( NULL ) /* required py macro */
|
|
|
|
0, /* ob_size */
|
|
|
|
/* For printing, in format "<module>.<name>" */
|
2006-06-14 05:16:39 +00:00
|
|
|
"Blender.Constraints", /* char *tp_name; */
|
2006-05-07 14:57:58 +00:00
|
|
|
sizeof( BPy_ConstraintSeq ), /* int tp_basicsize; */
|
|
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
|
|
|
|
( destructor ) ConstraintSeq_dealloc,/* destructor tp_dealloc; */
|
|
|
|
NULL, /* printfunc tp_print; */
|
|
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
|
|
NULL, /* setattrfunc tp_setattr; */
|
|
|
|
NULL, /* cmpfunc tp_compare; */
|
|
|
|
( reprfunc ) NULL, /* reprfunc tp_repr; */
|
|
|
|
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
|
|
|
&ConstraintSeq_as_sequence, /* PySequenceMethods *tp_as_sequence; */
|
|
|
|
NULL, /* PyMappingMethods *tp_as_mapping; */
|
|
|
|
|
|
|
|
/* More standard operations (here for binary compatibility) */
|
|
|
|
|
|
|
|
NULL, /* hashfunc tp_hash; */
|
|
|
|
NULL, /* ternaryfunc tp_call; */
|
|
|
|
NULL, /* reprfunc tp_str; */
|
|
|
|
NULL, /* getattrofunc tp_getattro; */
|
|
|
|
NULL, /* setattrofunc tp_setattro; */
|
|
|
|
|
|
|
|
/* Functions to access object as input/output buffer */
|
|
|
|
NULL, /* PyBufferProcs *tp_as_buffer; */
|
|
|
|
|
|
|
|
/*** Flags to define presence of optional/expanded features ***/
|
|
|
|
Py_TPFLAGS_DEFAULT, /* long tp_flags; */
|
|
|
|
|
|
|
|
NULL, /* char *tp_doc; Documentation string */
|
|
|
|
/*** Assigned meaning in release 2.0 ***/
|
|
|
|
/* call function for all accessible objects */
|
|
|
|
NULL, /* traverseproc tp_traverse; */
|
|
|
|
|
|
|
|
/* delete references to contained objects */
|
|
|
|
NULL, /* inquiry tp_clear; */
|
|
|
|
|
|
|
|
/*** Assigned meaning in release 2.1 ***/
|
|
|
|
/*** rich comparisons ***/
|
|
|
|
NULL, /* richcmpfunc tp_richcompare; */
|
|
|
|
|
|
|
|
/*** weak reference enabler ***/
|
|
|
|
0, /* long tp_weaklistoffset; */
|
|
|
|
|
|
|
|
/*** Added in release 2.2 ***/
|
|
|
|
/* Iterators */
|
|
|
|
( getiterfunc )ConstraintSeq_getIter, /* getiterfunc tp_iter; */
|
|
|
|
( iternextfunc )ConstraintSeq_nextIter, /* iternextfunc tp_iternext; */
|
|
|
|
|
|
|
|
/*** Attribute descriptor and subclassing stuff ***/
|
|
|
|
BPy_ConstraintSeq_methods, /* struct PyMethodDef *tp_methods; */
|
|
|
|
NULL, /* struct PyMemberDef *tp_members; */
|
|
|
|
NULL, /* struct PyGetSetDef *tp_getset; */
|
|
|
|
NULL, /* struct _typeobject *tp_base; */
|
|
|
|
NULL, /* PyObject *tp_dict; */
|
|
|
|
NULL, /* descrgetfunc tp_descr_get; */
|
|
|
|
NULL, /* descrsetfunc tp_descr_set; */
|
|
|
|
0, /* long tp_dictoffset; */
|
|
|
|
NULL, /* initproc tp_init; */
|
|
|
|
NULL, /* allocfunc tp_alloc; */
|
|
|
|
NULL, /* newfunc tp_new; */
|
|
|
|
/* Low-level free-memory routine */
|
|
|
|
NULL, /* freefunc tp_free; */
|
|
|
|
/* For PyObject_IS_GC */
|
|
|
|
NULL, /* inquiry tp_is_gc; */
|
|
|
|
NULL, /* PyObject *tp_bases; */
|
|
|
|
/* method resolution order */
|
|
|
|
NULL, /* PyObject *tp_mro; */
|
|
|
|
NULL, /* PyObject *tp_cache; */
|
|
|
|
NULL, /* PyObject *tp_subclasses; */
|
|
|
|
NULL, /* PyObject *tp_weaklist; */
|
|
|
|
NULL
|
|
|
|
};
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: PoseConstraintSeq_CreatePyObject */
|
|
|
|
/* Description: This function will create a new BPy_ConstraintSeq from an */
|
|
|
|
/* existing ListBase structure. */
|
|
|
|
/*****************************************************************************/
|
|
|
|
PyObject *PoseConstraintSeq_CreatePyObject( bPoseChannel *pchan )
|
|
|
|
{
|
|
|
|
BPy_ConstraintSeq *pyseq;
|
|
|
|
Object *ob;
|
|
|
|
|
|
|
|
for( ob = G.main->object.first; ob; ob = ob->id.next ) {
|
|
|
|
if( ob->type == OB_ARMATURE ) {
|
|
|
|
bPoseChannel *p = ob->pose->chanbase.first;
|
|
|
|
while( p ) {
|
|
|
|
if( p == pchan ) {
|
|
|
|
pyseq = ( BPy_ConstraintSeq * ) PyObject_NEW(
|
|
|
|
BPy_ConstraintSeq, &ConstraintSeq_Type );
|
|
|
|
if( !pyseq )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
|
|
"couldn't create BPy_ConstraintSeq object" );
|
|
|
|
pyseq->pchan = pchan;
|
|
|
|
pyseq->obj = ob;
|
|
|
|
return ( PyObject * ) pyseq;
|
|
|
|
} else
|
|
|
|
p = p->next;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
|
|
"couldn't find ANY armature with the pose!" );
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: ObConstraintSeq_CreatePyObject */
|
|
|
|
/* Description: This function will create a new BPy_ConstraintSeq from an */
|
|
|
|
/* existing ListBase structure. */
|
|
|
|
/*****************************************************************************/
|
|
|
|
PyObject *ObConstraintSeq_CreatePyObject( Object *obj )
|
|
|
|
{
|
|
|
|
BPy_ConstraintSeq *pyseq;
|
|
|
|
pyseq = ( BPy_ConstraintSeq * ) PyObject_NEW( BPy_ConstraintSeq,
|
|
|
|
&ConstraintSeq_Type );
|
|
|
|
if( !pyseq )
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
|
|
"couldn't create BPy_ConstraintSeq object" );
|
|
|
|
pyseq->obj = obj;
|
|
|
|
pyseq->pchan = NULL;
|
|
|
|
return ( PyObject * ) pyseq;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *M_Constraint_TypeDict( void )
|
|
|
|
{
|
|
|
|
PyObject *S = PyConstant_New( );
|
|
|
|
|
|
|
|
if( S ) {
|
|
|
|
BPy_constant *d = ( BPy_constant * ) S;
|
|
|
|
PyConstant_Insert( d, "NULL",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_NULL ) );
|
|
|
|
PyConstant_Insert( d, "TRACKTO",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_TRACKTO ) );
|
|
|
|
PyConstant_Insert( d, "IKSOLVER",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_KINEMATIC ) );
|
|
|
|
PyConstant_Insert( d, "FOLLOWPATH",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_FOLLOWPATH ) );
|
|
|
|
PyConstant_Insert( d, "COPYROT",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_ROTLIKE ) );
|
|
|
|
PyConstant_Insert( d, "COPYLOC",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_LOCLIKE ) );
|
|
|
|
PyConstant_Insert( d, "COPYSIZE",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_SIZELIKE ) );
|
|
|
|
PyConstant_Insert( d, "ACTION",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_ACTION ) );
|
|
|
|
PyConstant_Insert( d, "LOCKTRACK",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_LOCKTRACK ) );
|
|
|
|
PyConstant_Insert( d, "STRETCHTO",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_STRETCHTO ) );
|
|
|
|
PyConstant_Insert( d, "FLOOR",
|
|
|
|
PyInt_FromLong( CONSTRAINT_TYPE_MINMAX ) );
|
|
|
|
}
|
|
|
|
return S;
|
|
|
|
}
|
|
|
|
|
|
|
|
static PyObject *M_Constraint_SettingsDict( void )
|
|
|
|
{
|
|
|
|
PyObject *S = PyConstant_New( );
|
|
|
|
|
|
|
|
if( S ) {
|
|
|
|
BPy_constant *d = ( BPy_constant * ) S;
|
|
|
|
PyConstant_Insert( d, "XROT",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_XROT ) );
|
|
|
|
PyConstant_Insert( d, "YROT",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_YROT ) );
|
|
|
|
PyConstant_Insert( d, "ZROT",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_ZROT ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "UPX",
|
|
|
|
PyInt_FromLong( UP_X ) );
|
|
|
|
PyConstant_Insert( d, "UPY",
|
|
|
|
PyInt_FromLong( UP_Y ) );
|
|
|
|
PyConstant_Insert( d, "UPZ",
|
|
|
|
PyInt_FromLong( UP_Z ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "TRACKX",
|
|
|
|
PyInt_FromLong( TRACK_X ) );
|
|
|
|
PyConstant_Insert( d, "TRACKY",
|
|
|
|
PyInt_FromLong( TRACK_Y ) );
|
|
|
|
PyConstant_Insert( d, "TRACKZ",
|
|
|
|
PyInt_FromLong( TRACK_Z ) );
|
|
|
|
PyConstant_Insert( d, "TRACKNEGX",
|
|
|
|
PyInt_FromLong( TRACK_nX ) );
|
|
|
|
PyConstant_Insert( d, "TRACKNEGY",
|
|
|
|
PyInt_FromLong( TRACK_nY ) );
|
|
|
|
PyConstant_Insert( d, "TRACKNEGZ",
|
|
|
|
PyInt_FromLong( TRACK_nZ ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "VOLUMEXZ",
|
|
|
|
PyInt_FromLong( VOLUME_XZ ) );
|
|
|
|
PyConstant_Insert( d, "VOLUMEX",
|
|
|
|
PyInt_FromLong( VOLUME_X ) );
|
|
|
|
PyConstant_Insert( d, "VOLUMEZ",
|
|
|
|
PyInt_FromLong( VOLUME_Z ) );
|
|
|
|
PyConstant_Insert( d, "VOLUMENONE",
|
|
|
|
PyInt_FromLong( NO_VOLUME ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "PLANEX",
|
|
|
|
PyInt_FromLong( PLANE_X ) );
|
|
|
|
PyConstant_Insert( d, "PLANEY",
|
|
|
|
PyInt_FromLong( PLANE_Y ) );
|
|
|
|
PyConstant_Insert( d, "PLANEZ",
|
|
|
|
PyInt_FromLong( PLANE_Z ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "LOCKX",
|
|
|
|
PyInt_FromLong( LOCK_X ) );
|
|
|
|
PyConstant_Insert( d, "LOCKY",
|
|
|
|
PyInt_FromLong( LOCK_Y ) );
|
|
|
|
PyConstant_Insert( d, "LOCKZ",
|
|
|
|
PyInt_FromLong( LOCK_Z ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "MAXX",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_MAXX ) );
|
|
|
|
PyConstant_Insert( d, "MAXY",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_MAXY ) );
|
|
|
|
PyConstant_Insert( d, "MAXZ",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_MAXZ ) );
|
|
|
|
PyConstant_Insert( d, "MINX",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_MINX ) );
|
|
|
|
PyConstant_Insert( d, "MINY",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_MINY ) );
|
|
|
|
PyConstant_Insert( d, "MINZ",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_MINZ ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "COPYX",
|
|
|
|
PyInt_FromLong( LOCLIKE_X ) );
|
|
|
|
PyConstant_Insert( d, "COPYY",
|
|
|
|
PyInt_FromLong( LOCLIKE_Y ) );
|
|
|
|
PyConstant_Insert( d, "COPYZ",
|
|
|
|
PyInt_FromLong( LOCLIKE_Z ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "TARGET",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_TARGET ) );
|
|
|
|
PyConstant_Insert( d, "TOLERANCE",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_TOLERANCE ) );
|
|
|
|
PyConstant_Insert( d, "ITERATIONS",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_ITERATIONS ) );
|
|
|
|
PyConstant_Insert( d, "BONE",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_BONE ) );
|
|
|
|
PyConstant_Insert( d, "CHAINLEN",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_CHAINLEN ) );
|
|
|
|
PyConstant_Insert( d, "POSWEIGHT",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_POSWEIGHT ) );
|
|
|
|
PyConstant_Insert( d, "ROTWEIGHT",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_ROTWEIGHT ) );
|
|
|
|
PyConstant_Insert( d, "ROTATE",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_ROTATE ) );
|
|
|
|
PyConstant_Insert( d, "USETIP",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_USETIP ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "ACTION",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_ACTION ) );
|
|
|
|
PyConstant_Insert( d, "LOCAL",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_LOCAL ) );
|
|
|
|
PyConstant_Insert( d, "START",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_START ) );
|
|
|
|
PyConstant_Insert( d, "END",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_END ) );
|
|
|
|
PyConstant_Insert( d, "MIN",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_MIN ) );
|
|
|
|
PyConstant_Insert( d, "MAX",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_MAX ) );
|
|
|
|
PyConstant_Insert( d, "KEYON",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_KEYON ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "TRACK",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_TRACK ) );
|
|
|
|
PyConstant_Insert( d, "UP",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_UP ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "RESTLENGTH",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_RESTLENGTH ) );
|
|
|
|
PyConstant_Insert( d, "VOLVARIATION",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_VOLVARIATION ) );
|
|
|
|
PyConstant_Insert( d, "VOLUMEMODE",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_VOLUMEMODE ) );
|
|
|
|
PyConstant_Insert( d, "PLANE",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_PLANE ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "FOLLOW",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_FOLLOW ) );
|
|
|
|
PyConstant_Insert( d, "OFFSET",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_OFFSET ) );
|
|
|
|
PyConstant_Insert( d, "FORWARD",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_FORWARD ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "LOCK",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_LOCK ) );
|
|
|
|
|
|
|
|
PyConstant_Insert( d, "COPY",
|
|
|
|
PyInt_FromLong( EXPP_CONSTR_COPY ) );
|
|
|
|
}
|
|
|
|
return S;
|
|
|
|
}
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
|
|
/* Function: Constraint_Init */
|
|
|
|
/*****************************************************************************/
|
|
|
|
PyObject *Constraint_Init( void )
|
|
|
|
{
|
|
|
|
PyObject *submodule;
|
|
|
|
PyObject *TypeDict = M_Constraint_TypeDict( );
|
|
|
|
PyObject *SettingsDict = M_Constraint_SettingsDict( );
|
|
|
|
|
|
|
|
if( PyType_Ready( &ConstraintSeq_Type ) < 0
|
|
|
|
|| PyType_Ready( &Constraint_Type ) < 0 )
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
submodule = Py_InitModule3( "Blender.Constraint", NULL,
|
|
|
|
"Constraint module for accessing and creating constraint data" );
|
|
|
|
|
|
|
|
if( TypeDict )
|
|
|
|
PyModule_AddObject( submodule, "Type", TypeDict );
|
|
|
|
|
|
|
|
if( SettingsDict )
|
|
|
|
PyModule_AddObject( submodule, "Settings", SettingsDict );
|
|
|
|
|
|
|
|
return submodule;
|
|
|
|
}
|