This repository has been archived on 2023-10-09. You can view files and clone it, but cannot push or open issues or pull requests.
Files
blender-archive/source/blender/python/api2_2x/Modifier.c
Matt Ebb 3bf0a6fed8 This adds support for the new-ish options:
QUATERNION
B_BONE_REST
INVERT_VERTGROUP
MULTIMODIFIER

to the bpy armature modifier API.

It also fixes a significant problem - In the docs, it refers to 
the 'VERTGROUP' field as being a string value, shared
by lattice, armature, etc, referring to the 'VGroup' field in the 
armature modifier that defines a vertex group 'mask'
which the armature's effect is restricted to.

However, previously, for some very mistaken reason, in processing 
the getters/setters for the armature modifier, the
VERTGROUP field was pointing to the quite different 'Vert. Groups' 
toggle in the armature modifier, that enables or
disables using vertex groups for deformation (as opposed to 
envelopes).

I've fixed this, so VERTGROUP points to the VGroup string, as is 
already defined in the docs and consistent with other
modifiers like Lattice. A new field: 'VGROUPS' has been added, 
which is the analog to the 'Vert Groups' toggle.
2008-09-18 00:51:51 +00:00

1749 lines
57 KiB
C

/*
* $Id$
*
* ***** 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*
* This is a new part of Blender.
*
* Contributor(s): Ken Hughes, Campbell Barton
*
* ***** END GPL LICENSE BLOCK *****
*/
/* TODO, accessing a modifier sequence of a deleted object will crash blender at the moment, not sure how to fix this. */
#include "Modifier.h" /*This must come first*/
#include "DNA_object_types.h"
#include "DNA_effect_types.h"
#include "DNA_armature_types.h"
#include "DNA_vec_types.h"
#include "BKE_main.h"
#include "BKE_global.h"
#include "BKE_modifier.h"
#include "BKE_library.h"
#include "BLI_blenlib.h"
#include "BLI_arithb.h"
#include "MEM_guardedalloc.h"
#include "BDR_editobject.h"
#include "butspace.h"
#include "blendef.h"
#include "mydevice.h"
#include "Object.h"
#include "Texture.h"
#include "Mathutils.h"
#include "gen_utils.h"
#include "gen_library.h"
/* checks for the scene being removed */
#define MODIFIER_DEL_CHECK_PY(bpy_modifier) if (!(bpy_modifier->md)) return ( EXPP_ReturnPyObjError( PyExc_RuntimeError, "Modifier has been removed" ) )
#define MODIFIER_DEL_CHECK_INT(bpy_modifier) if (!(bpy_modifier->md)) return ( EXPP_ReturnIntError( PyExc_RuntimeError, "Modifier has been removed" ) )
enum mod_constants {
/*Apply to all modifiers*/
EXPP_MOD_RENDER = 0,
EXPP_MOD_REALTIME,
EXPP_MOD_EDITMODE,
EXPP_MOD_ONCAGE,
/*GENERIC*/
EXPP_MOD_OBJECT, /*ARMATURE, LATTICE, CURVE, BOOLEAN, ARRAY*/
EXPP_MOD_VERTGROUP, /*ARMATURE, LATTICE, CURVE, SMOOTH, CAST*/
EXPP_MOD_LIMIT, /*ARRAY, MIRROR*/
EXPP_MOD_FLAG, /*MIRROR, WAVE*/
EXPP_MOD_COUNT, /*DECIMATOR, ARRAY*/
EXPP_MOD_LENGTH, /*BUILD, ARRAY*/
EXPP_MOD_FACTOR, /*SMOOTH, CAST*/
EXPP_MOD_ENABLE_X, /*SMOOTH, CAST*/
EXPP_MOD_ENABLE_Y, /*SMOOTH, CAST*/
EXPP_MOD_ENABLE_Z, /*SMOOTH, CAST*/
EXPP_MOD_TYPES, /*SUBSURF, CAST*/
/*SUBSURF SPECIFIC*/
EXPP_MOD_LEVELS,
EXPP_MOD_RENDLEVELS,
EXPP_MOD_OPTIMAL,
EXPP_MOD_UV,
/*ARMATURE SPECIFIC*/
EXPP_MOD_VGROUPS,
EXPP_MOD_ENVELOPES,
EXPP_MOD_QUATERNION,
EXPP_MOD_B_BONE_REST,
EXPP_MOD_INVERT_VERTGROUP,
EXPP_MOD_MULTIMODIFIER,
/*ARRAY SPECIFIC*/
EXPP_MOD_OBJECT_OFFSET,
EXPP_MOD_OBJECT_CURVE,
EXPP_MOD_OFFSET_VEC,
EXPP_MOD_SCALE_VEC,
EXPP_MOD_MERGE_DIST,
/*BUILD SPECIFIC*/
EXPP_MOD_START,
EXPP_MOD_SEED,
EXPP_MOD_RANDOMIZE,
/*MIRROR SPECIFIC*/
EXPP_MOD_AXIS_X,
EXPP_MOD_AXIS_Y,
EXPP_MOD_AXIS_Z,
/*DECIMATE SPECIFIC*/
EXPP_MOD_RATIO,
/*WAVE SPECIFIC*/
EXPP_MOD_STARTX,
EXPP_MOD_STARTY,
EXPP_MOD_HEIGHT,
EXPP_MOD_WIDTH,
EXPP_MOD_NARROW,
EXPP_MOD_SPEED,
EXPP_MOD_DAMP,
EXPP_MOD_LIFETIME,
EXPP_MOD_TIMEOFFS,
/*BOOLEAN SPECIFIC*/
EXPP_MOD_OPERATION,
/*EDGE SPLIT SPECIFIC */
EXPP_MOD_EDGESPLIT_ANGLE,
EXPP_MOD_EDGESPLIT_FROM_ANGLE,
EXPP_MOD_EDGESPLIT_FROM_SHARP,
/* DISPLACE */
EXPP_MOD_UVLAYER,
EXPP_MOD_MID_LEVEL,
EXPP_MOD_STRENGTH,
EXPP_MOD_TEXTURE,
EXPP_MOD_MAPPING,
EXPP_MOD_DIRECTION,
/* SMOOTH */
EXPP_MOD_REPEAT,
/* CAST */
EXPP_MOD_RADIUS,
EXPP_MOD_SIZE,
EXPP_MOD_USE_OB_TRANSFORM,
EXPP_MOD_SIZE_FROM_RADIUS
/* yet to be implemented */
/* EXPP_MOD_HOOK_,*/
/* , */
};
/*****************************************************************************/
/* Python BPy_Modifier methods declarations: */
/*****************************************************************************/
static PyObject *Modifier_getName( BPy_Modifier * self );
static int Modifier_setName( BPy_Modifier * self, PyObject *arg );
static PyObject *Modifier_getType( BPy_Modifier * self );
static PyObject *Modifier_reset( BPy_Modifier * self );
static PyObject *Modifier_getData( BPy_Modifier * self, PyObject * key );
static int Modifier_setData( BPy_Modifier * self, PyObject * key,
PyObject * value );
/*****************************************************************************/
/* Python BPy_Modifier methods table: */
/*****************************************************************************/
static PyMethodDef BPy_Modifier_methods[] = {
/* name, method, flags, doc */
{"reset", (PyCFunction)Modifier_reset, METH_NOARGS,
"resets a hook modifier location"},
{NULL, NULL, 0, NULL}
};
/*****************************************************************************/
/* Python BPy_Modifier attributes get/set structure: */
/*****************************************************************************/
static PyGetSetDef BPy_Modifier_getseters[] = {
{"name",
(getter)Modifier_getName, (setter)Modifier_setName,
"Modifier name", NULL},
{"type",
(getter)Modifier_getType, (setter)NULL,
"Modifier type (read only)", NULL},
{NULL,NULL,NULL,NULL,NULL} /* Sentinel */
};
/*****************************************************************************/
/* Python Modifier_Type Mapping Methods table: */
/*****************************************************************************/
static PyMappingMethods Modifier_as_mapping = {
NULL, /* mp_length */
( binaryfunc ) Modifier_getData, /* mp_subscript */
( objobjargproc ) Modifier_setData, /* mp_ass_subscript */
};
/*****************************************************************************/
/* Python Modifier_Type callback function prototypes: */
/*****************************************************************************/
static PyObject *Modifier_repr( BPy_Modifier * self );
/*****************************************************************************/
/* Python Modifier_Type structure definition: */
/*****************************************************************************/
PyTypeObject Modifier_Type = {
PyObject_HEAD_INIT( NULL ) /* required py macro */
0, /* ob_size */
/* For printing, in format "<module>.<name>" */
"Blender Modifier", /* char *tp_name; */
sizeof( BPy_Modifier ), /* int tp_basicsize; */
0, /* tp_itemsize; For allocation */
/* Methods to implement standard operations */
NULL, /* destructor tp_dealloc; */
NULL, /* printfunc tp_print; */
NULL, /* getattrfunc tp_getattr; */
NULL, /* setattrfunc tp_setattr; */
NULL, /* cmpfunc tp_compare; */
( reprfunc ) Modifier_repr, /* reprfunc tp_repr; */
/* Method suites for standard classes */
NULL, /* PyNumberMethods *tp_as_number; */
NULL, /* PySequenceMethods *tp_as_sequence; */
&Modifier_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_Modifier_methods, /* struct PyMethodDef *tp_methods; */
NULL, /* struct PyMemberDef *tp_members; */
BPy_Modifier_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_Modifier methods: */
/*****************************************************************************/
/*
* return the name of this modifier
*/
static PyObject *Modifier_getName( BPy_Modifier * self )
{
MODIFIER_DEL_CHECK_PY(self);
return PyString_FromString( self->md->name );
}
/*
* set the name of this modifier
*/
static int Modifier_setName( BPy_Modifier * self, PyObject * attr )
{
char *name = PyString_AsString( attr );
if( !name )
return EXPP_ReturnIntError( PyExc_TypeError, "expected string arg" );
MODIFIER_DEL_CHECK_INT(self);
BLI_strncpy( self->md->name, name, sizeof( self->md->name ) );
return 0;
}
/*
* return the type of this modifier
*/
static PyObject *Modifier_getType( BPy_Modifier * self )
{
MODIFIER_DEL_CHECK_PY(self);
return PyInt_FromLong( self->md->type );
}
static PyObject *subsurf_getter( BPy_Modifier * self, int type )
{
SubsurfModifierData *md = ( SubsurfModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_TYPES:
return PyInt_FromLong( ( long )md->subdivType );
case EXPP_MOD_LEVELS:
return PyInt_FromLong( ( long )md->levels );
case EXPP_MOD_RENDLEVELS:
return PyInt_FromLong( ( long )md->renderLevels );
case EXPP_MOD_OPTIMAL:
return PyBool_FromLong( ( long )
( md->flags & eSubsurfModifierFlag_ControlEdges ) ) ;
case EXPP_MOD_UV:
return PyBool_FromLong( ( long )
( md->flags & eSubsurfModifierFlag_SubsurfUv ) ) ;
default:
return EXPP_ReturnPyObjError( PyExc_KeyError,
"key not found" );
}
}
static int subsurf_setter( BPy_Modifier * self, int type,
PyObject *value )
{
SubsurfModifierData *md = (SubsurfModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_TYPES:
return EXPP_setIValueRange( value, &md->subdivType, 0, 1, 'h' );
case EXPP_MOD_LEVELS:
return EXPP_setIValueClamped( value, &md->levels, 1, 6, 'h' );
case EXPP_MOD_RENDLEVELS:
return EXPP_setIValueClamped( value, &md->renderLevels, 1, 6, 'h' );
case EXPP_MOD_OPTIMAL:
return EXPP_setBitfield( value, &md->flags,
eSubsurfModifierFlag_ControlEdges, 'h' );
case EXPP_MOD_UV:
return EXPP_setBitfield( value, &md->flags,
eSubsurfModifierFlag_SubsurfUv, 'h' );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
static PyObject *armature_getter( BPy_Modifier * self, int type )
{
ArmatureModifierData *md = (ArmatureModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_OBJECT:
return Object_CreatePyObject( md->object );
case EXPP_MOD_VERTGROUP:
return PyString_FromString( md->defgrp_name ) ;
case EXPP_MOD_VGROUPS:
return EXPP_getBitfield( &md->deformflag, ARM_DEF_VGROUP, 'h' );
case EXPP_MOD_ENVELOPES:
return EXPP_getBitfield( &md->deformflag, ARM_DEF_ENVELOPE, 'h' );
case EXPP_MOD_QUATERNION:
return EXPP_getBitfield( &md->deformflag, ARM_DEF_QUATERNION, 'h' );
case EXPP_MOD_B_BONE_REST:
return EXPP_getBitfield( &md->deformflag, ARM_DEF_B_BONE_REST, 'h' );
case EXPP_MOD_INVERT_VERTGROUP:
return EXPP_getBitfield( &md->deformflag, ARM_DEF_INVERT_VGROUP, 'h' );
case EXPP_MOD_MULTIMODIFIER:
return EXPP_getBitfield( &md->multi, 1, 'h' );
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
return 0;
}
static int armature_setter( BPy_Modifier *self, int type, PyObject *value )
{
ArmatureModifierData *md = (ArmatureModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_OBJECT:
return GenericLib_assignData(value, (void **) &md->object, 0, 0, ID_OB, OB_ARMATURE);
case EXPP_MOD_VERTGROUP: {
char *defgrp_name = PyString_AsString( value );
if( !defgrp_name )
return EXPP_ReturnIntError( PyExc_TypeError,
"expected string arg" );
BLI_strncpy( md->defgrp_name, defgrp_name, sizeof( md->defgrp_name ) );
break;
}
case EXPP_MOD_VGROUPS:
return EXPP_setBitfield( value, &md->deformflag, ARM_DEF_VGROUP, 'h' );
case EXPP_MOD_ENVELOPES:
return EXPP_setBitfield( value, &md->deformflag, ARM_DEF_ENVELOPE, 'h' );
case EXPP_MOD_QUATERNION:
return EXPP_setBitfield( value, &md->deformflag, ARM_DEF_QUATERNION, 'h' );
case EXPP_MOD_B_BONE_REST:
return EXPP_setBitfield( value, &md->deformflag, ARM_DEF_B_BONE_REST, 'h' );
case EXPP_MOD_INVERT_VERTGROUP:
return EXPP_setBitfield( value, &md->deformflag, ARM_DEF_INVERT_VGROUP, 'h' );
case EXPP_MOD_MULTIMODIFIER:
return EXPP_setBitfield( value, &md->multi, 1, 'h' );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
return 0;
}
static PyObject *lattice_getter( BPy_Modifier * self, int type )
{
LatticeModifierData *md = (LatticeModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_OBJECT:
return Object_CreatePyObject( md->object );
case EXPP_MOD_VERTGROUP:
return PyString_FromString( md->name ) ;
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int lattice_setter( BPy_Modifier *self, int type, PyObject *value )
{
LatticeModifierData *md = (LatticeModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_OBJECT:
return GenericLib_assignData(value, (void **) &md->object, (void **) &self->object, 0, ID_OB, OB_LATTICE);
case EXPP_MOD_VERTGROUP: {
char *name = PyString_AsString( value );
if( !name )
return EXPP_ReturnIntError( PyExc_TypeError,
"expected string arg" );
BLI_strncpy( md->name, name, sizeof( md->name ) );
break;
}
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
return 0;
}
static PyObject *curve_getter( BPy_Modifier * self, int type )
{
CurveModifierData *md = (CurveModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_OBJECT:
return Object_CreatePyObject( md->object );
case EXPP_MOD_VERTGROUP:
return PyString_FromString( md->name ) ;
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int curve_setter( BPy_Modifier *self, int type, PyObject *value )
{
CurveModifierData *md = (CurveModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_OBJECT:
return GenericLib_assignData(value, (void **) &md->object, (void **) &self->object, 0, ID_OB, OB_CURVE);
case EXPP_MOD_VERTGROUP: {
char *name = PyString_AsString( value );
if( !name )
return EXPP_ReturnIntError( PyExc_TypeError,
"expected string arg" );
BLI_strncpy( md->name, name, sizeof( md->name ) );
break;
}
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
return 0;
}
static PyObject *build_getter( BPy_Modifier * self, int type )
{
BuildModifierData *md = (BuildModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_START:
return PyFloat_FromDouble( ( float )md->start );
case EXPP_MOD_LENGTH:
return PyFloat_FromDouble( ( float )md->length );
case EXPP_MOD_SEED:
return PyInt_FromLong( ( long )md->seed );
case EXPP_MOD_RANDOMIZE:
return PyBool_FromLong( ( long )md->randomize ) ;
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int build_setter( BPy_Modifier *self, int type, PyObject *value )
{
BuildModifierData *md = (BuildModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_START:
return EXPP_setFloatClamped( value, &md->start, 1.0, MAXFRAMEF );
case EXPP_MOD_LENGTH:
return EXPP_setFloatClamped( value, &md->length, 1.0, MAXFRAMEF );
case EXPP_MOD_SEED:
return EXPP_setIValueClamped( value, &md->seed, 1, MAXFRAME, 'i' );
case EXPP_MOD_RANDOMIZE:
return EXPP_setBitfield( value, &md->randomize, 1, 'i' );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
static PyObject *mirror_getter( BPy_Modifier * self, int type )
{
MirrorModifierData *md = (MirrorModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_LIMIT:
return PyFloat_FromDouble( (double)md->tolerance );
case EXPP_MOD_FLAG:
return PyBool_FromLong( (long)( md->flag & MOD_MIR_CLIPPING ) ) ;
case EXPP_MOD_AXIS_X:
return PyBool_FromLong( ( long )
( md->flag & MOD_MIR_AXIS_X ) ) ;
case EXPP_MOD_AXIS_Y:
return PyBool_FromLong( ( long )
( md->flag & MOD_MIR_AXIS_Y ) ) ;
case EXPP_MOD_AXIS_Z:
return PyBool_FromLong( ( long )
( md->flag & MOD_MIR_AXIS_Z ) ) ;
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int mirror_setter( BPy_Modifier *self, int type, PyObject *value )
{
MirrorModifierData *md = (MirrorModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_LIMIT:
return EXPP_setFloatClamped( value, &md->tolerance, 0.0, 1.0 );
case EXPP_MOD_FLAG:
return EXPP_setBitfield( value, &md->flag, MOD_MIR_CLIPPING, 'i' );
case EXPP_MOD_AXIS_X:
return EXPP_setBitfield( value, &md->flag, MOD_MIR_AXIS_X, 'h' );
case EXPP_MOD_AXIS_Y:
return EXPP_setBitfield( value, &md->flag, MOD_MIR_AXIS_Y, 'h' );
case EXPP_MOD_AXIS_Z:
return EXPP_setBitfield( value, &md->flag, MOD_MIR_AXIS_Z, 'h' );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
static PyObject *decimate_getter( BPy_Modifier * self, int type )
{
DecimateModifierData *md = (DecimateModifierData *)(self->md);
if( type == EXPP_MOD_RATIO )
return PyFloat_FromDouble( (double)md->percent );
else if( type == EXPP_MOD_COUNT )
return PyInt_FromLong( (long)md->faceCount );
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
static int decimate_setter( BPy_Modifier *self, int type, PyObject *value )
{
DecimateModifierData *md = (DecimateModifierData *)(self->md);
if( type == EXPP_MOD_RATIO )
return EXPP_setFloatClamped( value, &md->percent, 0.0, 1.0 );
else if( type == EXPP_MOD_COUNT )
return EXPP_ReturnIntError( PyExc_AttributeError,
"value is read-only" );
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
static PyObject *smooth_getter( BPy_Modifier * self, int type )
{
SmoothModifierData *md = (SmoothModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_FACTOR:
return PyFloat_FromDouble( (double)md->fac );
case EXPP_MOD_REPEAT:
return PyInt_FromLong( (long)md->repeat );
case EXPP_MOD_VERTGROUP:
return PyString_FromString( md->defgrp_name ) ;
case EXPP_MOD_ENABLE_X:
return EXPP_getBitfield( &md->flag, MOD_SMOOTH_X, 'h' );
case EXPP_MOD_ENABLE_Y:
return EXPP_getBitfield( &md->flag, MOD_SMOOTH_Y, 'h' );
case EXPP_MOD_ENABLE_Z:
return EXPP_getBitfield( &md->flag, MOD_SMOOTH_Z, 'h' );
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int smooth_setter( BPy_Modifier *self, int type, PyObject *value )
{
SmoothModifierData *md = (SmoothModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_FACTOR:
return EXPP_setFloatClamped( value, &md->fac, -10.0, 10.0 );
case EXPP_MOD_REPEAT:
return EXPP_setIValueRange( value, &md->repeat, 0, 30, 'h' );
case EXPP_MOD_VERTGROUP: {
char *name = PyString_AsString( value );
if( !name ) return EXPP_ReturnIntError( PyExc_TypeError, "expected string arg" );
BLI_strncpy( md->defgrp_name, name, sizeof( md->defgrp_name ) );
return 0;
}
case EXPP_MOD_ENABLE_X:
return EXPP_setBitfield( value, &md->flag, MOD_SMOOTH_X, 'h' );
case EXPP_MOD_ENABLE_Y:
return EXPP_setBitfield( value, &md->flag, MOD_SMOOTH_Y, 'h' );
case EXPP_MOD_ENABLE_Z:
return EXPP_setBitfield( value, &md->flag, MOD_SMOOTH_Z, 'h' );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
static PyObject *cast_getter( BPy_Modifier * self, int type )
{
CastModifierData *md = (CastModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_TYPES:
return PyInt_FromLong( (long)md->type );
case EXPP_MOD_FACTOR:
return PyFloat_FromDouble( (double)md->fac );
case EXPP_MOD_RADIUS:
return PyFloat_FromDouble( (double)md->radius );
case EXPP_MOD_SIZE:
return PyFloat_FromDouble( (double)md->size );
case EXPP_MOD_OBJECT:
return Object_CreatePyObject( md->object );
case EXPP_MOD_VERTGROUP:
return PyString_FromString( md->defgrp_name ) ;
case EXPP_MOD_ENABLE_X:
return EXPP_getBitfield( &md->flag, MOD_CAST_X, 'h' );
case EXPP_MOD_ENABLE_Y:
return EXPP_getBitfield( &md->flag, MOD_CAST_Y, 'h' );
case EXPP_MOD_ENABLE_Z:
return EXPP_getBitfield( &md->flag, MOD_CAST_Z, 'h' );
case EXPP_MOD_USE_OB_TRANSFORM:
return EXPP_getBitfield( &md->flag, MOD_CAST_USE_OB_TRANSFORM, 'h' );
case EXPP_MOD_SIZE_FROM_RADIUS:
return EXPP_getBitfield( &md->flag, MOD_CAST_SIZE_FROM_RADIUS, 'h' );
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int cast_setter( BPy_Modifier *self, int type, PyObject *value )
{
CastModifierData *md = (CastModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_TYPES:
return EXPP_setIValueRange( value, &md->type, 0, MOD_CAST_TYPE_CUBOID, 'h' );
case EXPP_MOD_FACTOR:
return EXPP_setFloatClamped( value, &md->fac, -10.0, 10.0 );
case EXPP_MOD_RADIUS:
return EXPP_setFloatClamped( value, &md->radius, 0.0, 100.0 );
case EXPP_MOD_SIZE:
return EXPP_setFloatClamped( value, &md->size, 0.0, 100.0 );
case EXPP_MOD_OBJECT: {
Object *ob_new=NULL;
if (value == Py_None) {
md->object = NULL;
} else if (BPy_Object_Check( value )) {
ob_new = ((( BPy_Object * )value)->object);
md->object = ob_new;
} else {
return EXPP_ReturnIntError( PyExc_TypeError,
"Expected an Object or None value" );
}
return 0;
}
case EXPP_MOD_VERTGROUP: {
char *name = PyString_AsString( value );
if( !name ) return EXPP_ReturnIntError( PyExc_TypeError, "expected string arg" );
BLI_strncpy( md->defgrp_name, name, sizeof( md->defgrp_name ) );
return 0;
}
case EXPP_MOD_ENABLE_X:
return EXPP_setBitfield( value, &md->flag, MOD_CAST_X, 'h' );
case EXPP_MOD_ENABLE_Y:
return EXPP_setBitfield( value, &md->flag, MOD_CAST_Y, 'h' );
case EXPP_MOD_ENABLE_Z:
return EXPP_setBitfield( value, &md->flag, MOD_CAST_Z, 'h' );
case EXPP_MOD_USE_OB_TRANSFORM:
return EXPP_setBitfield( value, &md->flag, MOD_CAST_USE_OB_TRANSFORM, 'h' );
case EXPP_MOD_SIZE_FROM_RADIUS:
return EXPP_setBitfield( value, &md->flag, MOD_CAST_SIZE_FROM_RADIUS, 'h' );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
static PyObject *wave_getter( BPy_Modifier * self, int type )
{
WaveModifierData *md = (WaveModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_STARTX:
return PyFloat_FromDouble( (double)md->startx );
case EXPP_MOD_STARTY:
return PyFloat_FromDouble( (double)md->starty );
case EXPP_MOD_HEIGHT:
return PyFloat_FromDouble( (double)md->height );
case EXPP_MOD_WIDTH:
return PyFloat_FromDouble( (double)md->width );
case EXPP_MOD_NARROW:
return PyFloat_FromDouble( (double)md->narrow );
case EXPP_MOD_SPEED:
return PyFloat_FromDouble( (double)md->speed );
case EXPP_MOD_DAMP:
return PyFloat_FromDouble( (double)md->damp );
case EXPP_MOD_LIFETIME:
return PyFloat_FromDouble( (double)md->lifetime );
case EXPP_MOD_TIMEOFFS:
return PyFloat_FromDouble( (double)md->timeoffs );
case EXPP_MOD_FLAG:
return PyInt_FromLong( (long)md->flag );
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int wave_setter( BPy_Modifier *self, int type, PyObject *value )
{
WaveModifierData *md = (WaveModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_STARTX:
return EXPP_setFloatClamped( value, &md->startx, -100.0, 100.0 );
case EXPP_MOD_STARTY:
return EXPP_setFloatClamped( value, &md->starty, -100.0, 100.0 );
case EXPP_MOD_HEIGHT:
return EXPP_setFloatClamped( value, &md->height, -2.0, 2.0 );
case EXPP_MOD_WIDTH:
return EXPP_setFloatClamped( value, &md->width, 0.0, 5.0 );
case EXPP_MOD_NARROW:
return EXPP_setFloatClamped( value, &md->width, 0.0, 5.0 );
case EXPP_MOD_SPEED:
return EXPP_setFloatClamped( value, &md->speed, -2.0, 2.0 );
case EXPP_MOD_DAMP:
return EXPP_setFloatClamped( value, &md->damp, -MAXFRAMEF, MAXFRAMEF );
case EXPP_MOD_LIFETIME:
return EXPP_setFloatClamped( value, &md->lifetime, -MAXFRAMEF, MAXFRAMEF );
case EXPP_MOD_TIMEOFFS:
return EXPP_setFloatClamped( value, &md->timeoffs, -MAXFRAMEF, MAXFRAMEF );
case EXPP_MOD_FLAG:
return EXPP_setIValueRange( value, &md->flag, 0,
MOD_WAVE_X | MOD_WAVE_Y | MOD_WAVE_CYCL, 'h' );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
static PyObject *array_getter( BPy_Modifier * self, int type )
{
ArrayModifierData *md = (ArrayModifierData *)(self->md);
if( type == EXPP_MOD_OBJECT_OFFSET )
return Object_CreatePyObject( md->offset_ob );
else if( type == EXPP_MOD_OBJECT_CURVE )
return Object_CreatePyObject( md->curve_ob );
else if( type == EXPP_MOD_COUNT )
return PyInt_FromLong( (long)md->count );
else if( type == EXPP_MOD_LENGTH )
return PyFloat_FromDouble( md->length );
else if( type == EXPP_MOD_MERGE_DIST )
return PyFloat_FromDouble( md->merge_dist );
else if( type == EXPP_MOD_OFFSET_VEC)
return newVectorObject( md->offset, 3, Py_NEW );
else if( type == EXPP_MOD_SCALE_VEC)
return newVectorObject( md->scale, 3, Py_NEW );
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
static int array_setter( BPy_Modifier *self, int type, PyObject *value )
{
ArrayModifierData *md = (ArrayModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_OBJECT_OFFSET:
return GenericLib_assignData(value, (void **) &md->offset_ob, (void **) &self->object, 0, ID_OB, 0);
case EXPP_MOD_OBJECT_CURVE:
return GenericLib_assignData(value, (void **) &md->curve_ob, 0, 0, ID_OB, OB_CURVE);
case EXPP_MOD_COUNT:
return EXPP_setIValueClamped( value, &md->count, 1, 1000, 'i' );
case EXPP_MOD_LENGTH:
return EXPP_setFloatClamped( value, &md->length, 0.0, 1000.0 );
case EXPP_MOD_MERGE_DIST:
return EXPP_setFloatClamped( value, &md->merge_dist, 0.0, 1000.0 );
case EXPP_MOD_OFFSET_VEC:
return EXPP_setVec3Clamped( value, md->offset, -10000.0, 10000.0 );
case EXPP_MOD_SCALE_VEC:
return EXPP_setVec3Clamped( value, md->scale, -10000.0, 10000.0 );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
static PyObject *boolean_getter( BPy_Modifier * self, int type )
{
BooleanModifierData *md = (BooleanModifierData *)(self->md);
if( type == EXPP_MOD_OBJECT )
return Object_CreatePyObject( md->object );
else if( type == EXPP_MOD_OPERATION )
return PyInt_FromLong( ( long )md->operation ) ;
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
static int boolean_setter( BPy_Modifier *self, int type, PyObject *value )
{
BooleanModifierData *md = (BooleanModifierData *)(self->md);
if( type == EXPP_MOD_OBJECT )
return GenericLib_assignData(value, (void **) &md->object, (void **) &self->object, 0, ID_OB, OB_MESH);
else if( type == EXPP_MOD_OPERATION )
return EXPP_setIValueRange( value, &md->operation,
eBooleanModifierOp_Intersect, eBooleanModifierOp_Difference,
'h' );
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
static PyObject *edgesplit_getter( BPy_Modifier * self, int type )
{
EdgeSplitModifierData *md = (EdgeSplitModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_EDGESPLIT_ANGLE:
return PyFloat_FromDouble( (double)md->split_angle );
case EXPP_MOD_EDGESPLIT_FROM_ANGLE:
return PyBool_FromLong( ( long )
( md->flags & MOD_EDGESPLIT_FROMANGLE ) ) ;
case EXPP_MOD_EDGESPLIT_FROM_SHARP:
return PyBool_FromLong( ( long )
( md->flags & MOD_EDGESPLIT_FROMFLAG ) ) ;
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int edgesplit_setter( BPy_Modifier *self, int type, PyObject *value )
{
EdgeSplitModifierData *md = (EdgeSplitModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_EDGESPLIT_ANGLE:
return EXPP_setFloatClamped( value, &md->split_angle, 0.0, 180.0 );
case EXPP_MOD_EDGESPLIT_FROM_ANGLE:
return EXPP_setBitfield( value, &md->flags,
MOD_EDGESPLIT_FROMANGLE, 'h' );
case EXPP_MOD_EDGESPLIT_FROM_SHARP:
return EXPP_setBitfield( value, &md->flags,
MOD_EDGESPLIT_FROMFLAG, 'h' );
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
static PyObject *displace_getter( BPy_Modifier * self, int type )
{
DisplaceModifierData *md = (DisplaceModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_TEXTURE:
if (md->texture) Texture_CreatePyObject( md->texture );
else Py_RETURN_NONE;
case EXPP_MOD_STRENGTH:
return PyFloat_FromDouble( (double)md->strength );
case EXPP_MOD_DIRECTION:
PyInt_FromLong( md->direction );
case EXPP_MOD_VERTGROUP:
return PyString_FromString( md->defgrp_name ) ;
case EXPP_MOD_MID_LEVEL:
return PyFloat_FromDouble( (double)md->midlevel );
case EXPP_MOD_MAPPING:
PyInt_FromLong( md->texmapping );
case EXPP_MOD_OBJECT:
return Object_CreatePyObject( md->map_object );
case EXPP_MOD_UVLAYER:
return PyString_FromString( md->uvlayer_name );
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int displace_setter( BPy_Modifier *self, int type, PyObject *value )
{
DisplaceModifierData *md = (DisplaceModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_TEXTURE:
return GenericLib_assignData(value, (void **) &md->texture, 0, 1, ID_TE, 0);
case EXPP_MOD_STRENGTH:
return EXPP_setFloatClamped( value, &md->strength, -1000.0, 1000.0 );
case EXPP_MOD_DIRECTION:
return EXPP_setIValueClamped( value, &md->direction,
MOD_DISP_DIR_X, MOD_DISP_DIR_RGB_XYZ, 'i' );
case EXPP_MOD_VERTGROUP: {
char *name = PyString_AsString( value );
if( !name ) return EXPP_ReturnIntError( PyExc_TypeError, "expected string arg" );
BLI_strncpy( md->defgrp_name, name, sizeof( md->defgrp_name ) );
return 0;
}
case EXPP_MOD_MID_LEVEL:
return EXPP_setFloatClamped( value, &md->midlevel, 0.0, 1.0 );
case EXPP_MOD_MAPPING:
return EXPP_setIValueClamped( value, &md->texmapping,
MOD_DISP_MAP_LOCAL, MOD_DISP_MAP_UV, 'i' );
case EXPP_MOD_OBJECT: {
Object *ob_new=NULL;
if (value == Py_None) {
md->map_object = NULL;
} else if (BPy_Object_Check( value )) {
ob_new = ((( BPy_Object * )value)->object);
md->map_object = ob_new;
} else {
return EXPP_ReturnIntError( PyExc_TypeError,
"Expected an Object or None value" );
}
return 0;
}
case EXPP_MOD_UVLAYER: {
char *name = PyString_AsString( value );
if( !name ) return EXPP_ReturnIntError( PyExc_TypeError, "expected string arg" );
BLI_strncpy( md->uvlayer_name, name, sizeof( md->uvlayer_name ) );
return 0;
}
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
}
/* static PyObject *uvproject_getter( BPy_Modifier * self, int type )
{
DisplaceModifierData *md = (DisplaceModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_MID_LEVEL:
return PyFloat_FromDouble( (double)md->midlevel );
default:
return EXPP_ReturnPyObjError( PyExc_KeyError, "key not found" );
}
}
static int uvproject_setter( BPy_Modifier *self, int type, PyObject *value )
{
DisplaceModifierData *md = (DisplaceModifierData *)(self->md);
switch( type ) {
case EXPP_MOD_TEXTURE:
return 0;
default:
return EXPP_ReturnIntError( PyExc_KeyError, "key not found" );
}
} */
/*
* get data from a modifier
*/
static PyObject *Modifier_getData( BPy_Modifier * self, PyObject * key )
{
int setting;
if( !PyInt_Check( key ) )
return EXPP_ReturnPyObjError( PyExc_TypeError,
"expected an int arg as stored in Blender.Modifier.Settings" );
MODIFIER_DEL_CHECK_PY(self);
setting = PyInt_AsLong( key );
switch( setting ) {
case EXPP_MOD_RENDER:
return EXPP_getBitfield( &self->md->mode, eModifierMode_Render, 'h' );
case EXPP_MOD_REALTIME:
return EXPP_getBitfield( &self->md->mode, eModifierMode_Realtime, 'h' );
case EXPP_MOD_EDITMODE:
return EXPP_getBitfield( &self->md->mode, eModifierMode_Editmode, 'h' );
case EXPP_MOD_ONCAGE:
return EXPP_getBitfield( &self->md->mode, eModifierMode_OnCage, 'h' );
default:
switch( self->md->type ) {
case eModifierType_Subsurf:
return subsurf_getter( self, setting );
case eModifierType_Armature:
return armature_getter( self, setting );
case eModifierType_Lattice:
return lattice_getter( self, setting );
case eModifierType_Curve:
return curve_getter( self, setting );
case eModifierType_Build:
return build_getter( self, setting );
case eModifierType_Mirror:
return mirror_getter( self, setting );
case eModifierType_Decimate:
return decimate_getter( self, setting );
case eModifierType_Smooth:
return smooth_getter( self, setting );
case eModifierType_Cast:
return cast_getter( self, setting );
case eModifierType_Wave:
return wave_getter( self, setting );
case eModifierType_Boolean:
return boolean_getter( self, setting );
case eModifierType_Array:
return array_getter( self, setting );
case eModifierType_EdgeSplit:
return edgesplit_getter( self, setting );
case eModifierType_Displace:
return displace_getter( self, setting );
/*case eModifierType_UVProject:
return uvproject_getter( self, setting );*/
case eModifierType_Hook:
case eModifierType_Softbody:
case eModifierType_None:
Py_RETURN_NONE;
}
}
return EXPP_ReturnPyObjError( PyExc_KeyError,
"unknown key or modifier type" );
}
static int Modifier_setData( BPy_Modifier * self, PyObject * key,
PyObject * arg )
{
int key_int;
if( !PyNumber_Check( key ) )
return EXPP_ReturnIntError( PyExc_TypeError,
"expected an int arg as stored in Blender.Modifier.Settings" );
MODIFIER_DEL_CHECK_INT(self);
key_int = PyInt_AsLong( key );
/* Chach for standard modifier settings */
switch( key_int ) {
case EXPP_MOD_RENDER:
return EXPP_setBitfield( arg, &self->md->mode,
eModifierMode_Render, 'h' );
case EXPP_MOD_REALTIME:
return EXPP_setBitfield( arg, &self->md->mode,
eModifierMode_Realtime, 'h' );
case EXPP_MOD_EDITMODE:
return EXPP_setBitfield( arg, &self->md->mode,
eModifierMode_Editmode, 'h' );
case EXPP_MOD_ONCAGE:
return EXPP_setBitfield( arg, &self->md->mode,
eModifierMode_OnCage, 'h' );
}
switch( self->md->type ) {
case eModifierType_Subsurf:
return subsurf_setter( self, key_int, arg );
case eModifierType_Armature:
return armature_setter( self, key_int, arg );
case eModifierType_Lattice:
return lattice_setter( self, key_int, arg );
case eModifierType_Curve:
return curve_setter( self, key_int, arg );
case eModifierType_Build:
return build_setter( self, key_int, arg );
case eModifierType_Mirror:
return mirror_setter( self, key_int, arg );
case eModifierType_Array:
return array_setter( self, key_int, arg );
case eModifierType_Decimate:
return decimate_setter( self, key_int, arg );
case eModifierType_Smooth:
return smooth_setter( self, key_int, arg );
case eModifierType_Cast:
return cast_setter( self, key_int, arg );
case eModifierType_Wave:
return wave_setter( self, key_int, arg );
case eModifierType_Boolean:
return boolean_setter( self, key_int, arg );
case eModifierType_EdgeSplit:
return edgesplit_setter( self, key_int, arg );
case eModifierType_Displace:
return displace_setter( self, key_int, arg );
/*case eModifierType_UVProject:
return uvproject_setter( self, key_int, arg );*/
case eModifierType_Hook:
case eModifierType_Softbody:
case eModifierType_None:
return 0;
}
return EXPP_ReturnIntError( PyExc_RuntimeError,
"unsupported modifier setting" );
}
static PyObject *Modifier_reset( BPy_Modifier * self )
{
Object *ob = self->object;
ModifierData *md = self->md;
HookModifierData *hmd = (HookModifierData*) md;
MODIFIER_DEL_CHECK_PY(self);
if (md->type != eModifierType_Hook)
return EXPP_ReturnPyObjError( PyExc_TypeError,
"can only reset hooks" );
if (hmd->object) {
Mat4Invert(hmd->object->imat, hmd->object->obmat);
Mat4MulSerie(hmd->parentinv, hmd->object->imat, ob->obmat, NULL, NULL, NULL, NULL, NULL, NULL);
}
Py_RETURN_NONE;
}
/*****************************************************************************/
/* Function: Modifier_repr */
/* Description: This is a callback function for the BPy_Modifier type. It */
/* builds a meaningful string to represent modifier objects. */
/*****************************************************************************/
static PyObject *Modifier_repr( BPy_Modifier * self )
{
ModifierTypeInfo *mti;
if (self->md==NULL)
return PyString_FromString( "[Modifier - Removed");
mti= modifierType_getInfo(self->md->type);
return PyString_FromFormat( "[Modifier \"%s\", Type \"%s\"]", self->md->name, mti->name );
}
/* Three Python Modifier_Type helper functions needed by the Object module: */
/*****************************************************************************/
/* Function: Modifier_CreatePyObject */
/* Description: This function will create a new BPy_Modifier from an */
/* existing Blender modifier structure. */
/*****************************************************************************/
PyObject *Modifier_CreatePyObject( Object *ob, ModifierData * md )
{
BPy_Modifier *pymod;
pymod = ( BPy_Modifier * ) PyObject_NEW( BPy_Modifier, &Modifier_Type );
if( !pymod )
return EXPP_ReturnPyObjError( PyExc_MemoryError,
"couldn't create BPy_Modifier object" );
pymod->md = md;
pymod->object = ob;
return ( PyObject * ) pymod;
}
/*****************************************************************************/
/* Function: Modifier_FromPyObject */
/* Description: This function returns the Blender modifier from the given */
/* PyObject. */
/*****************************************************************************/
ModifierData *Modifier_FromPyObject( PyObject * pyobj )
{
return ( ( BPy_Modifier * ) pyobj )->md;
}
/*****************************************************************************/
/* Modifier Sequence wrapper */
/*****************************************************************************/
/*
* Initialize the interator
*/
static PyObject *ModSeq_getIter( BPy_ModSeq * self )
{
if (!self->iter) {
self->iter = (ModifierData *)self->object->modifiers.first;
return EXPP_incr_ret ( (PyObject *) self );
} else {
return ModSeq_CreatePyObject(self->object, (ModifierData *)self->object->modifiers.first);
}
}
/*
* Get the next Modifier
*/
static PyObject *ModSeq_nextIter( BPy_ModSeq * self )
{
ModifierData *iter = self->iter;
if( iter ) {
self->iter = iter->next;
return Modifier_CreatePyObject( self->object, iter );
}
self->iter= NULL; /* mark as not iterating */
return EXPP_ReturnPyObjError( PyExc_StopIteration,
"iterator at end" );
}
/* return the number of modifiers */
static int ModSeq_length( BPy_ModSeq * self )
{
return BLI_countlist( &self->object->modifiers );
}
/* return a modifier */
static PyObject *ModSeq_item( BPy_ModSeq * self, int i )
{
ModifierData *md = NULL;
/* if index is negative, start counting from the end of the list */
if( i < 0 )
i += ModSeq_length( self );
/* skip through the list until we get the modifier or end of list */
for( md = self->object->modifiers.first; i && md; --i ) md = md->next;
if( md )
return Modifier_CreatePyObject( self->object, md );
else
return EXPP_ReturnPyObjError( PyExc_IndexError,
"array index out of range" );
}
/*****************************************************************************/
/* Python BPy_ModSeq sequence table: */
/*****************************************************************************/
static PySequenceMethods ModSeq_as_sequence = {
( inquiry ) ModSeq_length, /* sq_length */
( binaryfunc ) 0, /* sq_concat */
( intargfunc ) 0, /* sq_repeat */
( intargfunc ) ModSeq_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 */
};
/*
* helper function to check for a valid modifier argument
*/
static ModifierData *locate_modifier( BPy_ModSeq *self, BPy_Modifier * value )
{
ModifierData *md;
/* check that argument is a modifier */
if( !BPy_Modifier_Check(value) )
return (ModifierData *)EXPP_ReturnPyObjError( PyExc_TypeError,
"expected an modifier as an argument" );
/* check whether modifier has been removed */
if( !value->md )
return (ModifierData *)EXPP_ReturnPyObjError( PyExc_RuntimeError,
"This modifier has been removed!" );
/* find the modifier in the object's list */
for( md = self->object->modifiers.first; md; md = md->next )
if( md == value->md )
return md;
/* return exception if we can't find the modifier */
return (ModifierData *)EXPP_ReturnPyObjError( PyExc_AttributeError,
"This modifier is not in the object's stack" );
}
/* create a new modifier at the end of the list */
static PyObject *ModSeq_append( BPy_ModSeq *self, PyObject *value )
{
int type = PyInt_AsLong(value);
/* type 0 is eModifierType_None, should we be able to add one of these? */
if( type <= 0 || type >= NUM_MODIFIER_TYPES )
return EXPP_ReturnPyObjError( PyExc_ValueError,
"Not an int or argument out of range, expected an int from Blender.Modifier.Type" );
BLI_addtail( &self->object->modifiers, modifier_new( type ) );
return Modifier_CreatePyObject( self->object, self->object->modifiers.last );
}
/* remove an existing modifier */
static PyObject *ModSeq_remove( BPy_ModSeq *self, BPy_Modifier *value )
{
ModifierData *md = locate_modifier( self, value );
/* if we can't locate the modifier, return (exception already set) */
if( !md )
return (PyObject *)NULL;
/* do the actual removal */
BLI_remlink( &self->object->modifiers, md );
modifier_free( md );
/* erase the link to the modifier */
value->md = NULL;
Py_RETURN_NONE;
}
/* move the modifier up in the stack */
static PyObject *ModSeq_moveUp( BPy_ModSeq * self, BPy_Modifier * value )
{
ModifierData *md = locate_modifier( self, value );
/* if we can't locate the modifier, return (exception already set) */
if( !md )
return (PyObject *)NULL;
if( mod_moveUp( self->object, md ) )
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
"cannot move above a modifier requiring original data" );
Py_RETURN_NONE;
}
/* move the modifier down in the stack */
static PyObject *ModSeq_moveDown( BPy_ModSeq * self, BPy_Modifier *value )
{
ModifierData *md = locate_modifier( self, value );
/* if we can't locate the modifier, return (exception already set) */
if( !md )
return (PyObject *)NULL;
if( mod_moveDown( self->object, md ) )
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
"cannot move beyond a non-deforming modifier" );
Py_RETURN_NONE;
}
/* quick hack for ZanQdo: add new hook modifier for selected verts */
static PyObject *ModSeq_ZanQdoHack(BPy_ModSeq *self)
{
/* this should add the hook (assumes that modifier stack is on same ob!) */
if ((self) && (G.obedit) && (self->object==G.obedit)) {
add_hook(1);
}
Py_RETURN_NONE;
}
/*****************************************************************************/
/* Python BPy_ModSeq methods table: */
/*****************************************************************************/
static PyMethodDef BPy_ModSeq_methods[] = {
/* name, method, flags, doc */
{"append", ( PyCFunction ) ModSeq_append, METH_O,
"(type) - add a new modifier, where type is the type of modifier"},
{"remove", ( PyCFunction ) ModSeq_remove, METH_O,
"(modifier) - remove an existing modifier, where modifier is a modifier from this object."},
{"moveUp", ( PyCFunction ) ModSeq_moveUp, METH_O,
"(modifier) - Move a modifier up in stack"},
{"moveDown", ( PyCFunction ) ModSeq_moveDown, METH_O,
"(modifier) - Move a modifier down in stack"},
{"ZanQdoHack", (PyCFunction)ModSeq_ZanQdoHack, METH_NOARGS,
"while in editmode, adds a hook for the selected verts (adds new modifier, and deselects object)"},
{NULL, NULL, 0, NULL}
};
/*****************************************************************************/
/* Python ModSeq_Type structure definition: */
/*****************************************************************************/
PyTypeObject ModSeq_Type = {
PyObject_HEAD_INIT( NULL ) /* required py macro */
0, /* ob_size */
/* For printing, in format "<module>.<name>" */
"Blender.Modifiers", /* char *tp_name; */
sizeof( BPy_ModSeq ), /* int tp_basicsize; */
0, /* tp_itemsize; For allocation */
/* Methods to implement standard operations */
NULL, /* 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; */
&ModSeq_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 )ModSeq_getIter, /* getiterfunc tp_iter; */
( iternextfunc )ModSeq_nextIter, /* iternextfunc tp_iternext; */
/*** Attribute descriptor and subclassing stuff ***/
BPy_ModSeq_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: ModSeq_CreatePyObject */
/* Description: This function will create a new BPy_ModSeq from an */
/* existing ListBase structure. */
/*****************************************************************************/
PyObject *ModSeq_CreatePyObject( Object *ob, ModifierData *iter )
{
BPy_ModSeq *pymod;
pymod = ( BPy_ModSeq * ) PyObject_NEW( BPy_ModSeq, &ModSeq_Type );
if( !pymod )
return EXPP_ReturnPyObjError( PyExc_MemoryError,
"couldn't create BPy_ModSeq object" );
pymod->object = ob;
pymod->iter = iter;
return ( PyObject * ) pymod;
}
static PyObject *M_Modifier_TypeDict( void )
{
PyObject *S = PyConstant_New( );
if( S ) {
BPy_constant *d = ( BPy_constant * ) S;
PyConstant_Insert( d, "SUBSURF",
PyInt_FromLong( eModifierType_Subsurf ) );
PyConstant_Insert( d, "ARMATURE",
PyInt_FromLong( eModifierType_Armature ) );
PyConstant_Insert( d, "LATTICE",
PyInt_FromLong( eModifierType_Lattice ) );
PyConstant_Insert( d, "CURVE",
PyInt_FromLong( eModifierType_Curve ) );
PyConstant_Insert( d, "BUILD",
PyInt_FromLong( eModifierType_Build ) );
PyConstant_Insert( d, "MIRROR",
PyInt_FromLong( eModifierType_Mirror ) );
PyConstant_Insert( d, "DECIMATE",
PyInt_FromLong( eModifierType_Decimate ) );
PyConstant_Insert( d, "WAVE",
PyInt_FromLong( eModifierType_Wave ) );
PyConstant_Insert( d, "BOOLEAN",
PyInt_FromLong( eModifierType_Boolean ) );
PyConstant_Insert( d, "ARRAY",
PyInt_FromLong( eModifierType_Array ) );
PyConstant_Insert( d, "EDGESPLIT",
PyInt_FromLong( eModifierType_EdgeSplit ) );
PyConstant_Insert( d, "SMOOTH",
PyInt_FromLong( eModifierType_Smooth ) );
PyConstant_Insert( d, "CAST",
PyInt_FromLong( eModifierType_Cast ) );
PyConstant_Insert( d, "DISPLACE",
PyInt_FromLong( eModifierType_Displace ) );
}
return S;
}
static PyObject *M_Modifier_SettingsDict( void )
{
PyObject *S = PyConstant_New( );
if( S ) {
BPy_constant *d = ( BPy_constant * ) S;
/*
# The lines below are a python script that uses the enum variables to create
# the lines below
# START PYSCRIPT
st='''
EXPP_MOD_RENDER = 0,
EXPP_MOD_REALTIME,
EXPP_MOD_EDITMODE,
etc.. copy from above
'''
base= '''
PyConstant_Insert( d, "%s",
PyInt_FromLong( EXPP_MOD_%s ) );
'''
for var in st.replace(',','').split('\n'):
var= var.split()
if not var: continue
var= var[0]
if (not var) or var.startswith('/'): continue
var='_'.join(var.split('_')[2:])
print base % (var, var),
# END PYSCRIPT
*/
/*Auto generated from the above script*/
PyConstant_Insert( d, "RENDER",
PyInt_FromLong( EXPP_MOD_RENDER ) );
PyConstant_Insert( d, "REALTIME",
PyInt_FromLong( EXPP_MOD_REALTIME ) );
PyConstant_Insert( d, "EDITMODE",
PyInt_FromLong( EXPP_MOD_EDITMODE ) );
PyConstant_Insert( d, "ONCAGE",
PyInt_FromLong( EXPP_MOD_ONCAGE ) );
PyConstant_Insert( d, "OBJECT",
PyInt_FromLong( EXPP_MOD_OBJECT ) );
PyConstant_Insert( d, "VERTGROUP",
PyInt_FromLong( EXPP_MOD_VERTGROUP ) );
PyConstant_Insert( d, "LIMIT",
PyInt_FromLong( EXPP_MOD_LIMIT ) );
PyConstant_Insert( d, "FLAG",
PyInt_FromLong( EXPP_MOD_FLAG ) );
PyConstant_Insert( d, "COUNT",
PyInt_FromLong( EXPP_MOD_COUNT ) );
PyConstant_Insert( d, "LENGTH",
PyInt_FromLong( EXPP_MOD_LENGTH ) );
PyConstant_Insert( d, "FACTOR",
PyInt_FromLong( EXPP_MOD_FACTOR ) );
PyConstant_Insert( d, "ENABLE_X",
PyInt_FromLong( EXPP_MOD_ENABLE_X ) );
PyConstant_Insert( d, "ENABLE_Y",
PyInt_FromLong( EXPP_MOD_ENABLE_Y ) );
PyConstant_Insert( d, "ENABLE_Z",
PyInt_FromLong( EXPP_MOD_ENABLE_Z ) );
PyConstant_Insert( d, "TYPES",
PyInt_FromLong( EXPP_MOD_TYPES ) );
PyConstant_Insert( d, "LEVELS",
PyInt_FromLong( EXPP_MOD_LEVELS ) );
PyConstant_Insert( d, "RENDLEVELS",
PyInt_FromLong( EXPP_MOD_RENDLEVELS ) );
PyConstant_Insert( d, "OPTIMAL",
PyInt_FromLong( EXPP_MOD_OPTIMAL ) );
PyConstant_Insert( d, "UV",
PyInt_FromLong( EXPP_MOD_UV ) );
PyConstant_Insert( d, "VGROUPS",
PyInt_FromLong( EXPP_MOD_VGROUPS ) );
PyConstant_Insert( d, "ENVELOPES",
PyInt_FromLong( EXPP_MOD_ENVELOPES ) );
PyConstant_Insert( d, "QUATERNION",
PyInt_FromLong( EXPP_MOD_QUATERNION ) );
PyConstant_Insert( d, "B_BONE_REST",
PyInt_FromLong( EXPP_MOD_B_BONE_REST ) );
PyConstant_Insert( d, "INVERT_VERTGROUP",
PyInt_FromLong( EXPP_MOD_INVERT_VERTGROUP ) );
PyConstant_Insert( d, "MULTIMODIFIER",
PyInt_FromLong( EXPP_MOD_MULTIMODIFIER ) );
PyConstant_Insert( d, "OBJECT_OFFSET",
PyInt_FromLong( EXPP_MOD_OBJECT_OFFSET ) );
PyConstant_Insert( d, "OBJECT_CURVE",
PyInt_FromLong( EXPP_MOD_OBJECT_CURVE ) );
PyConstant_Insert( d, "OFFSET_VEC",
PyInt_FromLong( EXPP_MOD_OFFSET_VEC ) );
PyConstant_Insert( d, "SCALE_VEC",
PyInt_FromLong( EXPP_MOD_SCALE_VEC ) );
PyConstant_Insert( d, "MERGE_DIST",
PyInt_FromLong( EXPP_MOD_MERGE_DIST ) );
PyConstant_Insert( d, "START",
PyInt_FromLong( EXPP_MOD_START ) );
PyConstant_Insert( d, "SEED",
PyInt_FromLong( EXPP_MOD_SEED ) );
PyConstant_Insert( d, "RANDOMIZE",
PyInt_FromLong( EXPP_MOD_RANDOMIZE ) );
PyConstant_Insert( d, "AXIS_X",
PyInt_FromLong( EXPP_MOD_AXIS_X ) );
PyConstant_Insert( d, "AXIS_Y",
PyInt_FromLong( EXPP_MOD_AXIS_Y ) );
PyConstant_Insert( d, "AXIS_Z",
PyInt_FromLong( EXPP_MOD_AXIS_Z ) );
PyConstant_Insert( d, "RATIO",
PyInt_FromLong( EXPP_MOD_RATIO ) );
PyConstant_Insert( d, "STARTX",
PyInt_FromLong( EXPP_MOD_STARTX ) );
PyConstant_Insert( d, "STARTY",
PyInt_FromLong( EXPP_MOD_STARTY ) );
PyConstant_Insert( d, "HEIGHT",
PyInt_FromLong( EXPP_MOD_HEIGHT ) );
PyConstant_Insert( d, "WIDTH",
PyInt_FromLong( EXPP_MOD_WIDTH ) );
PyConstant_Insert( d, "NARROW",
PyInt_FromLong( EXPP_MOD_NARROW ) );
PyConstant_Insert( d, "SPEED",
PyInt_FromLong( EXPP_MOD_SPEED ) );
PyConstant_Insert( d, "DAMP",
PyInt_FromLong( EXPP_MOD_DAMP ) );
PyConstant_Insert( d, "LIFETIME",
PyInt_FromLong( EXPP_MOD_LIFETIME ) );
PyConstant_Insert( d, "TIMEOFFS",
PyInt_FromLong( EXPP_MOD_TIMEOFFS ) );
PyConstant_Insert( d, "OPERATION",
PyInt_FromLong( EXPP_MOD_OPERATION ) );
PyConstant_Insert( d, "EDGESPLIT_ANGLE",
PyInt_FromLong( EXPP_MOD_EDGESPLIT_ANGLE ) );
PyConstant_Insert( d, "EDGESPLIT_FROM_ANGLE",
PyInt_FromLong( EXPP_MOD_EDGESPLIT_FROM_ANGLE ) );
PyConstant_Insert( d, "EDGESPLIT_FROM_SHARP",
PyInt_FromLong( EXPP_MOD_EDGESPLIT_FROM_SHARP ) );
PyConstant_Insert( d, "UVLAYER",
PyInt_FromLong( EXPP_MOD_UVLAYER ) );
PyConstant_Insert( d, "MID_LEVEL",
PyInt_FromLong( EXPP_MOD_MID_LEVEL ) );
PyConstant_Insert( d, "STRENGTH",
PyInt_FromLong( EXPP_MOD_STRENGTH ) );
PyConstant_Insert( d, "TEXTURE",
PyInt_FromLong( EXPP_MOD_TEXTURE ) );
PyConstant_Insert( d, "MAPPING",
PyInt_FromLong( EXPP_MOD_MAPPING ) );
PyConstant_Insert( d, "DIRECTION",
PyInt_FromLong( EXPP_MOD_DIRECTION ) );
PyConstant_Insert( d, "REPEAT",
PyInt_FromLong( EXPP_MOD_REPEAT ) );
PyConstant_Insert( d, "RADIUS",
PyInt_FromLong( EXPP_MOD_RADIUS ) );
PyConstant_Insert( d, "SIZE",
PyInt_FromLong( EXPP_MOD_SIZE ) );
PyConstant_Insert( d, "USE_OB_TRANSFORM",
PyInt_FromLong( EXPP_MOD_USE_OB_TRANSFORM ) );
PyConstant_Insert( d, "SIZE_FROM_RADIUS",
PyInt_FromLong( EXPP_MOD_SIZE_FROM_RADIUS ) );
/*End Auto generated code*/
}
return S;
}
/*****************************************************************************/
/* Function: Modifier_Init */
/*****************************************************************************/
PyObject *Modifier_Init( void )
{
PyObject *submodule;
PyObject *TypeDict = M_Modifier_TypeDict( );
PyObject *SettingsDict = M_Modifier_SettingsDict( );
if( PyType_Ready( &ModSeq_Type ) < 0 ||
PyType_Ready( &Modifier_Type ) < 0 )
return NULL;
submodule = Py_InitModule3( "Blender.Modifier", NULL,
"Modifer module for accessing and creating object modifier data" );
if( TypeDict ) {
PyModule_AddObject( submodule, "Type", TypeDict ); /* deprecated */
/* since PyModule_AddObject() steals a reference, we need to
incref TypeDict to use it again */
Py_INCREF( TypeDict);
PyModule_AddObject( submodule, "Types", TypeDict );
}
if( SettingsDict )
PyModule_AddObject( submodule, "Settings", SettingsDict );
return submodule;
}