1884 lines
52 KiB
C
1884 lines
52 KiB
C
/*
|
|
* $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): Jacques Guignot RIP 2005, Nathan Letwory,
|
|
* Stephen Swaney, Ken Hughes
|
|
*
|
|
* ***** END GPL/BL DUAL LICENSE BLOCK *****
|
|
*/
|
|
|
|
#include "Ipo.h" /*This must come first*/
|
|
|
|
#include "BKE_main.h"
|
|
#include "BKE_global.h"
|
|
#include "BKE_library.h"
|
|
#include "BKE_ipo.h"
|
|
#include "BLI_blenlib.h"
|
|
#include "BIF_space.h"
|
|
#include "BSE_editipo.h"
|
|
#include "MEM_guardedalloc.h"
|
|
#include "DNA_key_types.h"
|
|
#include "mydevice.h"
|
|
#include "Ipocurve.h"
|
|
#include "gen_utils.h"
|
|
|
|
extern int ob_ar[];
|
|
extern int la_ar[];
|
|
extern int ma_ar[];
|
|
extern int ac_ar[];
|
|
extern int cam_ar[];
|
|
extern int co_ar[];
|
|
extern int cu_ar[];
|
|
extern int seq_ar[];
|
|
extern int te_ar[];
|
|
extern int wo_ar[];
|
|
|
|
PyObject *submodule;
|
|
|
|
/*****************************************************************************/
|
|
/* Python API function prototypes for the Ipo module. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Ipo_New( PyObject * self, PyObject * args );
|
|
static PyObject *M_Ipo_Get( PyObject * self, PyObject * args );
|
|
static PyObject *M_Ipo_Recalc( PyObject * self, PyObject * args );
|
|
|
|
/*****************************************************************************/
|
|
/* The following string definitions are used for documentation strings. */
|
|
/* In Python these will be written to the console when doing a */
|
|
/* Blender.Ipo.__doc__ */
|
|
/*****************************************************************************/
|
|
char M_Ipo_doc[] = "";
|
|
char M_Ipo_New_doc[] = "";
|
|
char M_Ipo_Get_doc[] = "";
|
|
|
|
/*****************************************************************************/
|
|
/* Python method structure definition for Blender.Ipo module: */
|
|
/*****************************************************************************/
|
|
|
|
struct PyMethodDef M_Ipo_methods[] = {
|
|
{"New", ( PyCFunction ) M_Ipo_New, METH_VARARGS | METH_KEYWORDS,
|
|
M_Ipo_New_doc},
|
|
{"Get", M_Ipo_Get, METH_VARARGS, M_Ipo_Get_doc},
|
|
{"get", M_Ipo_Get, METH_VARARGS, M_Ipo_Get_doc},
|
|
{"Recalc", M_Ipo_Recalc, METH_VARARGS, M_Ipo_Get_doc},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
/*****************************************************************************/
|
|
/* Python BPy_Ipo methods declarations: */
|
|
/*****************************************************************************/
|
|
static PyObject *Ipo_getName( BPy_Ipo * self );
|
|
static PyObject *Ipo_oldsetName( BPy_Ipo * self, PyObject * args );
|
|
static int Ipo_setName( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_getBlocktype( BPy_Ipo * self );
|
|
static PyObject *Ipo_oldsetBlocktype( BPy_Ipo * self, PyObject * args );
|
|
static int Ipo_setBlocktype( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_getRctf( BPy_Ipo * self );
|
|
static PyObject *Ipo_oldsetRctf( BPy_Ipo * self, PyObject * args );
|
|
static int Ipo_setRctf( BPy_Ipo * self, PyObject * args );
|
|
|
|
static PyObject *Ipo_getCurve( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_getCurves( BPy_Ipo * self );
|
|
static PyObject *Ipo_getCurveNames( BPy_Ipo * self );
|
|
static PyObject *Ipo_addCurve( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_delCurve( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_getNcurves( BPy_Ipo * self );
|
|
static PyObject *Ipo_getNBezPoints( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_DeleteBezPoints( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_getCurveBP( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_getCurvecurval( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_EvaluateCurveOn( BPy_Ipo * self, PyObject * args );
|
|
|
|
static PyObject *Ipo_setCurveBeztriple( BPy_Ipo * self, PyObject * args );
|
|
static PyObject *Ipo_getCurveBeztriple( BPy_Ipo * self, PyObject * args );
|
|
|
|
static PyObject *Ipo_getChannel( BPy_Ipo * self );
|
|
static int Ipo_setChannel( BPy_Ipo * self, PyObject * args );
|
|
|
|
static int Ipo_length( BPy_Ipo * inst );
|
|
static PyObject *Ipo_getIpoCurveByName( BPy_Ipo * self, PyObject * key );
|
|
static int Ipo_setIpoCurveByName( BPy_Ipo * self, PyObject * key,
|
|
PyObject * value );
|
|
static int Ipo_contains( BPy_Ipo * self, PyObject * key );
|
|
|
|
/*****************************************************************************/
|
|
/* Python BPy_Ipo methods table: */
|
|
/*****************************************************************************/
|
|
static PyMethodDef BPy_Ipo_methods[] = {
|
|
/* name, method, flags, doc */
|
|
{"getName", ( PyCFunction ) Ipo_getName, METH_NOARGS,
|
|
"() - Return Ipo Data name"},
|
|
{"setName", ( PyCFunction ) Ipo_oldsetName, METH_VARARGS,
|
|
"(str) - Change Ipo Data name"},
|
|
{"getBlocktype", ( PyCFunction ) Ipo_getBlocktype, METH_NOARGS,
|
|
"() - Return Ipo blocktype"},
|
|
{"setBlocktype", ( PyCFunction ) Ipo_oldsetBlocktype, METH_VARARGS,
|
|
"(str) - Change Ipo blocktype"},
|
|
{"getRctf", ( PyCFunction ) Ipo_getRctf, METH_NOARGS,
|
|
"() - Return Ipo rctf"},
|
|
{"setRctf", ( PyCFunction ) Ipo_oldsetRctf, METH_VARARGS,
|
|
"(flt,flt,flt,flt) - Change Ipo rctf"},
|
|
{"addCurve", ( PyCFunction ) Ipo_addCurve, METH_VARARGS,
|
|
"() - Add a curve to Ipo"},
|
|
{"delCurve", ( PyCFunction ) Ipo_delCurve, METH_VARARGS,
|
|
"(str) - Delete curve from Ipo"},
|
|
{"getNcurves", ( PyCFunction ) Ipo_getNcurves, METH_NOARGS,
|
|
"() - Return number of Ipo curves"},
|
|
{"getCurves", ( PyCFunction ) Ipo_getCurves, METH_NOARGS,
|
|
"() - Return list of all defined Ipo curves"},
|
|
{"getCurve", ( PyCFunction ) Ipo_getCurve, METH_VARARGS,
|
|
"(str|int) - Returns specified Ipo curve"},
|
|
{"getNBezPoints", ( PyCFunction ) Ipo_getNBezPoints, METH_VARARGS,
|
|
"(int) - Return number of Bez points on an Ipo curve"},
|
|
{"delBezPoint", ( PyCFunction ) Ipo_DeleteBezPoints, METH_VARARGS,
|
|
"(int) - deprecated: use ipocurve.delBezier()"},
|
|
{"getCurveBP", ( PyCFunction ) Ipo_getCurveBP, METH_VARARGS,
|
|
"() - unsupported"},
|
|
{"EvaluateCurveOn", ( PyCFunction ) Ipo_EvaluateCurveOn, METH_VARARGS,
|
|
"(int,flt) - deprecated: see ipocurve.evaluate()"},
|
|
{"getCurveCurval", ( PyCFunction ) Ipo_getCurvecurval, METH_VARARGS,
|
|
"(int) - deprecated: see ipocurve.evaluate()"},
|
|
{"getCurveBeztriple", ( PyCFunction ) Ipo_getCurveBeztriple, METH_VARARGS,
|
|
"(int,int) - deprecated: see ipocurve.bezierPoints[]"},
|
|
{"setCurveBeztriple", ( PyCFunction ) Ipo_setCurveBeztriple, METH_VARARGS,
|
|
"(int,int,list) - set a BezTriple"},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
/*****************************************************************************/
|
|
/* Python BPy_Ipo attributes get/set structure: */
|
|
/*****************************************************************************/
|
|
static PyGetSetDef BPy_Ipo_getseters[] = {
|
|
{"name",
|
|
(getter)Ipo_getName, (setter)Ipo_setName,
|
|
"Ipo data name",
|
|
NULL},
|
|
{"curves",
|
|
(getter)Ipo_getCurves, (setter)NULL,
|
|
"Ipo curves",
|
|
NULL},
|
|
{"curveConsts",
|
|
(getter)Ipo_getCurveNames, (setter)NULL,
|
|
"Ipo curve constants (values depend on Ipo type)",
|
|
NULL},
|
|
{"channel",
|
|
(getter)Ipo_getChannel, (setter)Ipo_setChannel,
|
|
"Ipo texture channel (world, lamp, material Ipos only)",
|
|
NULL},
|
|
|
|
{"blocktype",
|
|
(getter)Ipo_getBlocktype, (setter)NULL,
|
|
"Ipo block type",
|
|
NULL},
|
|
{"rcft",
|
|
(getter)Ipo_getRctf, (setter)Ipo_setRctf,
|
|
"Ipo type",
|
|
NULL},
|
|
{NULL,NULL,NULL,NULL,NULL} /* Sentinel */
|
|
};
|
|
|
|
/*****************************************************************************/
|
|
/* Python Ipo_Type Mapping Methods table: */
|
|
/*****************************************************************************/
|
|
static PyMappingMethods Ipo_as_mapping = {
|
|
( inquiry ) Ipo_length, /* mp_length */
|
|
( binaryfunc ) Ipo_getIpoCurveByName, /* mp_subscript */
|
|
( objobjargproc ) Ipo_setIpoCurveByName, /* mp_ass_subscript */
|
|
};
|
|
|
|
static PySequenceMethods Ipo_as_sequence = {
|
|
( inquiry ) 0, /* sq_length */
|
|
( binaryfunc ) 0, /* sq_concat */
|
|
( intargfunc ) 0, /* sq_repeat */
|
|
( intargfunc ) 0, /* sq_item */
|
|
( intintargfunc ) 0, /* sq_slice */
|
|
( intobjargproc ) 0, /* sq_ass_item */
|
|
( intintobjargproc ) 0, /* sq_ass_slice */
|
|
( objobjproc ) Ipo_contains, /* sq_contains */
|
|
( binaryfunc ) 0, /* sq_inplace_concat */
|
|
( intargfunc ) 0, /* sq_inplace_repeat */
|
|
};
|
|
|
|
/*****************************************************************************/
|
|
/* Python Ipo_Type callback function prototypes: */
|
|
/*****************************************************************************/
|
|
static void Ipo_dealloc( BPy_Ipo * self );
|
|
//static int IpoPrint (BPy_Ipo *self, FILE *fp, int flags);
|
|
static int Ipo_compare( BPy_Ipo * a, BPy_Ipo * b );
|
|
static PyObject *Ipo_repr( BPy_Ipo * self );
|
|
static PyObject *Ipo_getIter( BPy_Ipo * self );
|
|
static PyObject *Ipo_nextIter( BPy_Ipo * self );
|
|
|
|
/* #define CURVEATTRS */ /* uncomment to enable curves as Ipo attributes */
|
|
|
|
#ifdef CURVEATTRS
|
|
static PyGetSetDef BPy_Ipocurve_getseter = {
|
|
"noname",
|
|
(getter)NULL, (setter)NULL,
|
|
"Ipo curve name",
|
|
NULL
|
|
};
|
|
|
|
void generate_curveattrs( PyObject* dict, int blocktype )
|
|
{
|
|
typedef char * (*namefunc)(int, ... );
|
|
namefunc lookup_name;
|
|
int size;
|
|
int *vals = NULL;
|
|
char name[32];
|
|
PyObject*desc;
|
|
|
|
switch ( blocktype ) {
|
|
case ID_OB:
|
|
lookup_name = (namefunc)getname_ob_ei;
|
|
vals = ob_ar;
|
|
size = OB_TOTIPO;
|
|
break;
|
|
case ID_MA:
|
|
lookup_name = (namefunc)getname_mat_ei;
|
|
vals = ma_ar;
|
|
size = MA_TOTIPO;
|
|
break;
|
|
case ID_CA:
|
|
lookup_name = (namefunc)getname_cam_ei;
|
|
vals = cam_ar;
|
|
size = CAM_TOTIPO;
|
|
break;
|
|
case ID_LA:
|
|
lookup_name = (namefunc)getname_la_ei;
|
|
vals = la_ar;
|
|
size = LA_TOTIPO;
|
|
break;
|
|
case ID_TE:
|
|
lookup_name = (namefunc)getname_tex_ei;
|
|
vals = te_ar;
|
|
size = TE_TOTIPO;
|
|
break;
|
|
case ID_WO:
|
|
lookup_name = (namefunc)getname_world_ei;
|
|
vals = wo_ar;
|
|
size = WO_TOTIPO;
|
|
break;
|
|
case ID_PO:
|
|
lookup_name = (namefunc)getname_ac_ei;
|
|
vals = ac_ar;
|
|
size = AC_TOTIPO;
|
|
break;
|
|
case ID_CO:
|
|
lookup_name = (namefunc)getname_co_ei;
|
|
vals = co_ar;
|
|
size = CO_TOTIPO;
|
|
break;
|
|
case ID_CU:
|
|
lookup_name = (namefunc)getname_cu_ei;
|
|
vals = cu_ar;
|
|
size = CU_TOTIPO;
|
|
break;
|
|
case ID_SEQ:
|
|
lookup_name = (namefunc)getname_seq_ei;
|
|
vals = seq_ar;
|
|
size = SEQ_TOTIPO;
|
|
break;
|
|
}
|
|
|
|
desc = PyDescr_NewGetSet( &Ipo_Type, &BPy_Ipocurve_getseter );
|
|
while( size-- ) {
|
|
strcpy( name, lookup_name( *vals ) );
|
|
*name = tolower( *name );
|
|
PyDict_SetItemString( dict, name, desc );
|
|
++vals;
|
|
}
|
|
Py_DECREF( desc );
|
|
}
|
|
|
|
static short lookup_curve_name( char *, int , int );
|
|
|
|
static PyObject *getattro( PyObject *self, PyObject *value )
|
|
{
|
|
char *name = PyString_AS_STRING( value );
|
|
Ipo *ipo = ((BPy_Ipo *)self)->ipo;
|
|
int adrcode;
|
|
IpoCurve *icu;
|
|
|
|
if( !strcmp(name, "__class__") )
|
|
return PyObject_GenericGetAttr( self, value );
|
|
|
|
if( !strcmp(name, "__dict__") ) /* no effect */
|
|
{
|
|
PyObject *dict;
|
|
dict = PyDict_Copy( self->ob_type->tp_dict );
|
|
generate_curveattrs( dict, ipo->blocktype );
|
|
return dict;
|
|
}
|
|
|
|
adrcode = lookup_curve_name( name, ipo->blocktype,
|
|
((BPy_Ipo *)self)->mtex );
|
|
|
|
if( adrcode != -1 ) {
|
|
for( icu = ipo->curve.first; icu; icu = icu->next )
|
|
if( icu->adrcode == adrcode )
|
|
return IpoCurve_CreatePyObject( icu );
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
return PyObject_GenericGetAttr( self, value );
|
|
}
|
|
#endif
|
|
|
|
/*****************************************************************************/
|
|
/* Python Ipo_Type structure definition: */
|
|
/*****************************************************************************/
|
|
PyTypeObject Ipo_Type = {
|
|
PyObject_HEAD_INIT( NULL ) /* required py macro */
|
|
0, /* ob_size */
|
|
/* For printing, in format "<module>.<name>" */
|
|
"Blender Ipo", /* char *tp_name; */
|
|
sizeof( BPy_Ipo ), /* int tp_basicsize; */
|
|
0, /* tp_itemsize; For allocation */
|
|
|
|
/* Methods to implement standard operations */
|
|
|
|
( destructor ) Ipo_dealloc,/* destructor tp_dealloc; */
|
|
NULL, /* printfunc tp_print; */
|
|
NULL, /* getattrfunc tp_getattr; */
|
|
NULL, /* setattrfunc tp_setattr; */
|
|
( cmpfunc ) Ipo_compare, /* cmpfunc tp_compare; */
|
|
( reprfunc ) Ipo_repr, /* reprfunc tp_repr; */
|
|
|
|
/* Method suites for standard classes */
|
|
|
|
NULL, /* PyNumberMethods *tp_as_number; */
|
|
&Ipo_as_sequence, /* PySequenceMethods *tp_as_sequence; */
|
|
&Ipo_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; */
|
|
#ifdef CURVEATTRS
|
|
(getattrofunc)getattro,
|
|
#else
|
|
NULL, /* getattrofunc tp_getattro; */
|
|
#endif
|
|
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) Ipo_getIter, /* getiterfunc tp_iter; */
|
|
( iternextfunc ) Ipo_nextIter, /* iternextfunc tp_iternext; */
|
|
|
|
/*** Attribute descriptor and subclassing stuff ***/
|
|
BPy_Ipo_methods, /* struct PyMethodDef *tp_methods; */
|
|
NULL, /* struct PyMemberDef *tp_members; */
|
|
BPy_Ipo_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
|
|
};
|
|
|
|
/*****************************************************************************/
|
|
/* internal utility routines */
|
|
/*****************************************************************************/
|
|
|
|
/*
|
|
* Search through list of known Ipocurves for a particular name.
|
|
*
|
|
* str: name of curve we are searching for
|
|
* blocktype: type of Ipo
|
|
* channel: texture channel number, for World/Lamp/Material curves
|
|
*
|
|
* returns the adrcode for the named curve if it exists, -1 otherwise
|
|
*/
|
|
|
|
/* this is needed since getname_ob_ei() is different from the rest */
|
|
|
|
typedef char * (*namefunc)(int, ... );
|
|
|
|
static short lookup_curve_name( char *str, int blocktype, int channel )
|
|
{
|
|
namefunc lookup_name;
|
|
int *adrcodes = NULL;
|
|
int size = 0;
|
|
|
|
/* make sure channel type is ignored when it should be */
|
|
if( blocktype != ID_WO && blocktype != ID_LA && blocktype != ID_MA )
|
|
channel = -1;
|
|
|
|
switch ( blocktype ) {
|
|
case ID_OB:
|
|
lookup_name = (namefunc)getname_ob_ei;
|
|
adrcodes = ob_ar;
|
|
size = OB_TOTIPO;
|
|
break;
|
|
case ID_MA:
|
|
lookup_name = (namefunc)getname_mat_ei;
|
|
adrcodes = ma_ar;
|
|
size = MA_TOTIPO;
|
|
break;
|
|
case ID_CA:
|
|
lookup_name = (namefunc)getname_cam_ei;
|
|
adrcodes = cam_ar;
|
|
size = CAM_TOTIPO;
|
|
break;
|
|
case ID_LA:
|
|
lookup_name = (namefunc)getname_la_ei;
|
|
adrcodes = la_ar;
|
|
size = LA_TOTIPO;
|
|
break;
|
|
case ID_TE:
|
|
lookup_name = (namefunc)getname_tex_ei;
|
|
adrcodes = te_ar;
|
|
size = TE_TOTIPO;
|
|
break;
|
|
case ID_WO:
|
|
lookup_name = (namefunc)getname_world_ei;
|
|
adrcodes = wo_ar;
|
|
size = WO_TOTIPO;
|
|
break;
|
|
case ID_PO:
|
|
lookup_name = (namefunc)getname_ac_ei;
|
|
adrcodes = ac_ar;
|
|
size = AC_TOTIPO;
|
|
break;
|
|
case ID_CO:
|
|
lookup_name = (namefunc)getname_co_ei;
|
|
adrcodes = co_ar;
|
|
size = CO_TOTIPO;
|
|
break;
|
|
case ID_CU:
|
|
lookup_name = (namefunc)getname_cu_ei;
|
|
adrcodes = cu_ar;
|
|
size = CU_TOTIPO;
|
|
break;
|
|
case ID_SEQ:
|
|
lookup_name = (namefunc)getname_seq_ei;
|
|
adrcodes = seq_ar;
|
|
size = SEQ_TOTIPO;
|
|
break;
|
|
case ID_KE: /* shouldn't happen */
|
|
default:
|
|
return -1;
|
|
}
|
|
|
|
while ( size-- ) {
|
|
char *name = lookup_name ( *adrcodes );
|
|
|
|
/* if not a texture channel, just return the adrcode */
|
|
if( !strcmp( str, name ) ) {
|
|
if( channel == -1 || *adrcodes < MA_MAP1 )
|
|
return (short)*adrcodes;
|
|
|
|
/* otherwise adjust adrcode to include current channel */
|
|
else {
|
|
int param = (short)*adrcodes & ~MA_MAP1;
|
|
param |= texchannel_to_adrcode( channel );
|
|
return (short)param;
|
|
}
|
|
}
|
|
++adrcodes;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
static short lookup_curve_key( char *str, Ipo *ipo )
|
|
{
|
|
Key *keyiter;
|
|
|
|
/* find the ipo in the keylist */
|
|
for( keyiter = G.main->key.first; keyiter; keyiter = keyiter->id.next ) {
|
|
if( keyiter->ipo == ipo ) {
|
|
KeyBlock *block = keyiter->block.first;
|
|
|
|
/* look for a matching string, get the adrcode */
|
|
for( block = keyiter->block.first; block; block = block->next )
|
|
if( !strcmp( str, block->name ) )
|
|
return block->adrcode;
|
|
|
|
/* no match; no addr code */
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
/* error if the ipo isn't in the list */
|
|
return -2;
|
|
}
|
|
|
|
/*
|
|
* Search through list of known Ipocurves for a particular adrcode.
|
|
*
|
|
* code: adrcode of curve we are searching for
|
|
* blocktype: type of Ipo
|
|
* channel: texture channel number, for World/Lamp/Material curves
|
|
*
|
|
* returns the adrcode for the named curve if it exists, -1 otherwise
|
|
*/
|
|
|
|
static short lookup_curve_adrcode( int code, int blocktype, int channel )
|
|
{
|
|
int *adrcodes = NULL;
|
|
int size = 0;
|
|
|
|
switch ( blocktype ) {
|
|
case ID_OB:
|
|
adrcodes = ob_ar;
|
|
size = OB_TOTIPO;
|
|
break;
|
|
case ID_MA:
|
|
adrcodes = ma_ar;
|
|
size = MA_TOTIPO;
|
|
break;
|
|
case ID_CA:
|
|
adrcodes = cam_ar;
|
|
size = CAM_TOTIPO;
|
|
break;
|
|
case ID_LA:
|
|
adrcodes = la_ar;
|
|
size = LA_TOTIPO;
|
|
break;
|
|
case ID_TE:
|
|
adrcodes = te_ar;
|
|
size = TE_TOTIPO;
|
|
break;
|
|
case ID_WO:
|
|
adrcodes = wo_ar;
|
|
size = WO_TOTIPO;
|
|
break;
|
|
case ID_PO:
|
|
adrcodes = ac_ar;
|
|
size = AC_TOTIPO;
|
|
break;
|
|
case ID_CO:
|
|
adrcodes = co_ar;
|
|
size = CO_TOTIPO;
|
|
break;
|
|
case ID_CU:
|
|
adrcodes = cu_ar;
|
|
size = CU_TOTIPO;
|
|
break;
|
|
case ID_SEQ:
|
|
adrcodes = seq_ar;
|
|
size = SEQ_TOTIPO;
|
|
break;
|
|
case ID_KE:
|
|
default:
|
|
return -1;
|
|
}
|
|
|
|
while ( size-- ) {
|
|
if( *adrcodes == code ) {
|
|
|
|
/* if not a texture channel, just return the adrcode */
|
|
if( channel == -1 || *adrcodes < MA_MAP1 )
|
|
return (short)*adrcodes;
|
|
|
|
/* otherwise adjust adrcode to include current channel */
|
|
else {
|
|
int param = *adrcodes & ~MA_MAP1;
|
|
param |= texchannel_to_adrcode( channel );
|
|
return (short)param;
|
|
}
|
|
}
|
|
++adrcodes;
|
|
}
|
|
return -1;
|
|
}
|
|
|
|
/*
|
|
* Delete an IpoCurve from an Ipo
|
|
*/
|
|
|
|
static void del_ipocurve( Ipo * ipo, IpoCurve * icu ) {
|
|
BLI_remlink( &( ipo->curve ), icu );
|
|
if( icu->bezt )
|
|
MEM_freeN( icu->bezt );
|
|
if( icu->driver )
|
|
MEM_freeN( icu->driver );
|
|
MEM_freeN( icu );
|
|
|
|
/* have to do this to avoid crashes in the IPO window */
|
|
allspace( REMAKEIPO, 0 );
|
|
EXPP_allqueue( REDRAWIPO, 0 );
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Python BPy_Ipo functions: */
|
|
/*****************************************************************************/
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Ipo_New */
|
|
/* Python equivalent: Blender.Ipo.New */
|
|
/*****************************************************************************/
|
|
|
|
static PyObject *M_Ipo_New( PyObject * self_unused, PyObject * args )
|
|
{
|
|
char *name = NULL, *code = NULL;
|
|
int idcode = -1;
|
|
Ipo *blipo;
|
|
|
|
if( !PyArg_ParseTuple( args, "ss", &code, &name ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected two string arguments" );
|
|
|
|
if( !strcmp( code, "Object" ) )
|
|
idcode = ID_OB;
|
|
else if( !strcmp( code, "Camera" ) )
|
|
idcode = ID_CA;
|
|
else if( !strcmp( code, "World" ) )
|
|
idcode = ID_WO;
|
|
else if( !strcmp( code, "Material" ) )
|
|
idcode = ID_MA;
|
|
else if( !strcmp( code, "Texture" ) )
|
|
idcode = ID_TE;
|
|
else if( !strcmp( code, "Lamp" ) )
|
|
idcode = ID_LA;
|
|
else if( !strcmp( code, "Action" ) )
|
|
idcode = ID_PO;
|
|
else if( !strcmp( code, "Constraint" ) )
|
|
idcode = ID_CO;
|
|
else if( !strcmp( code, "Sequence" ) )
|
|
idcode = ID_SEQ;
|
|
else if( !strcmp( code, "Curve" ) )
|
|
idcode = ID_CU;
|
|
else if( !strcmp( code, "Key" ) )
|
|
idcode = ID_KE;
|
|
else return EXPP_ReturnPyObjError( PyExc_ValueError,
|
|
"unknown Ipo code" );
|
|
|
|
blipo = add_ipo( name, idcode );
|
|
|
|
if( blipo ) {
|
|
/* return user count to zero because add_ipo() inc'd it */
|
|
blipo->id.us = 0;
|
|
/* create python wrapper obj */
|
|
return Ipo_CreatePyObject( blipo );
|
|
} else
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't create Ipo Data in Blender" );
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Ipo_Get */
|
|
/* Python equivalent: Blender.Ipo.Get */
|
|
/* Description: Receives a string and returns the ipo data obj */
|
|
/* whose name matches the string. If no argument is */
|
|
/* passed in, a list of all ipo data names in the */
|
|
/* current scene is returned. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Ipo_Get( PyObject * self_unused, PyObject * args )
|
|
{
|
|
char *name = NULL;
|
|
Ipo *ipo_iter;
|
|
PyObject *ipolist, *pyobj;
|
|
char error_msg[64];
|
|
|
|
if( !PyArg_ParseTuple( args, "|s", &name ) )
|
|
return ( EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected string argument (or nothing)" ) );
|
|
|
|
ipo_iter = G.main->ipo.first;
|
|
|
|
if( name ) { /* (name) - Search ipo by name */
|
|
while( ipo_iter ) {
|
|
if( !strcmp( name, ipo_iter->id.name + 2 ) ) {
|
|
return Ipo_CreatePyObject( ipo_iter );
|
|
}
|
|
ipo_iter = ipo_iter->id.next;
|
|
}
|
|
|
|
PyOS_snprintf( error_msg, sizeof( error_msg ),
|
|
"Ipo \"%s\" not found", name );
|
|
return EXPP_ReturnPyObjError( PyExc_NameError, error_msg );
|
|
}
|
|
|
|
else { /* () - return a list with all ipos in the scene */
|
|
int index = 0;
|
|
|
|
ipolist = PyList_New( BLI_countlist( &( G.main->ipo ) ) );
|
|
|
|
if( !ipolist )
|
|
return EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
"couldn't create PyList" );
|
|
|
|
while( ipo_iter ) {
|
|
pyobj = Ipo_CreatePyObject( ipo_iter );
|
|
|
|
if( !pyobj )
|
|
return NULL;
|
|
|
|
PyList_SET_ITEM( ipolist, index, pyobj );
|
|
|
|
ipo_iter = ipo_iter->id.next;
|
|
index++;
|
|
}
|
|
|
|
return ipolist;
|
|
}
|
|
}
|
|
|
|
/*
|
|
* This should probably be deprecated too? Or else documented in epydocs.
|
|
* Seems very similar to Ipocurve.recalc().
|
|
*/
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Ipo_Recalc */
|
|
/* Python equivalent: Blender.Ipo.Recalc */
|
|
/* Description: Receives (presumably) an IpoCurve object and */
|
|
/* updates the curve after changes to control points. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Ipo_Recalc( PyObject * self_unused, PyObject * args )
|
|
{
|
|
PyObject *pyobj;
|
|
|
|
if( !PyArg_ParseTuple( args, "O!", &IpoCurve_Type, &pyobj ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected Ipo curve argument" );
|
|
|
|
testhandles_ipocurve( IpoCurve_FromPyObject( pyobj ) );
|
|
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Python BPy_Ipo methods: */
|
|
/*****************************************************************************/
|
|
|
|
static PyObject *Ipo_getName( BPy_Ipo * self )
|
|
{
|
|
PyObject *attr = PyString_FromString( self->ipo->id.name + 2 );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Ipo.name attribute" );
|
|
}
|
|
|
|
static int Ipo_setName( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
char *name;
|
|
char buf[21];
|
|
|
|
name = PyString_AsString( args );
|
|
if( !name )
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
"expected string argument" );
|
|
|
|
PyOS_snprintf( buf, sizeof( buf ), "%s", name );
|
|
|
|
rename_id( &self->ipo->id, buf );
|
|
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *Ipo_getBlocktype( BPy_Ipo * self )
|
|
{
|
|
PyObject *attr = PyInt_FromLong( self->ipo->blocktype );
|
|
|
|
if( attr )
|
|
return attr;
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"couldn't get Ipo.blocktype attribute" );
|
|
}
|
|
|
|
static int Ipo_setBlocktype( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
if( !PyInt_CheckExact( args ) )
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
"expected int argument" );
|
|
|
|
self->ipo->blocktype = (short)PyInt_AS_LONG( args );
|
|
|
|
return 0;
|
|
}
|
|
|
|
static PyObject *Ipo_getRctf( BPy_Ipo * self )
|
|
{
|
|
PyObject *l = PyList_New( 0 );
|
|
PyList_Append( l, PyFloat_FromDouble( self->ipo->cur.xmin ) );
|
|
PyList_Append( l, PyFloat_FromDouble( self->ipo->cur.xmax ) );
|
|
PyList_Append( l, PyFloat_FromDouble( self->ipo->cur.ymin ) );
|
|
PyList_Append( l, PyFloat_FromDouble( self->ipo->cur.ymax ) );
|
|
return l;
|
|
}
|
|
|
|
static int Ipo_setRctf( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
float v[4];
|
|
|
|
if( !PyArg_ParseTuple( args, "ffff", v, v + 1, v + 2, v + 3 ) )
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
"expected a tuple of 4 floats" );
|
|
|
|
self->ipo->cur.xmin = v[0];
|
|
self->ipo->cur.xmax = v[1];
|
|
self->ipo->cur.ymin = v[2];
|
|
self->ipo->cur.ymax = v[3];
|
|
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Get total number of Ipo curves for this Ipo. NOTE: this function
|
|
* returns all curves for Ipos which have texture channels, unlike
|
|
* Ipo_length().
|
|
*/
|
|
|
|
static PyObject *Ipo_getNcurves( BPy_Ipo * self )
|
|
{
|
|
IpoCurve *icu;
|
|
int i = 0;
|
|
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next ) {
|
|
i++;
|
|
}
|
|
|
|
return PyInt_FromLong( (long)i );
|
|
}
|
|
|
|
/*
|
|
Function: Ipo_addCurve
|
|
Bpy: Blender.Ipo.addCurve( 'curname')
|
|
|
|
add a new curve to an existing IPO.
|
|
example:
|
|
ipo = Blender.Ipo.New('Object','ObIpo')
|
|
cu = ipo.addCurve('LocX')
|
|
*/
|
|
|
|
static PyObject *Ipo_addCurve( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
short param; /* numeric curve name constant */
|
|
char *cur_name = 0; /* input arg: curve name */
|
|
Ipo *ipo = 0;
|
|
IpoCurve *icu = 0;
|
|
Link *link;
|
|
|
|
if( !PyArg_ParseTuple( args, "s", &cur_name ) )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "expected string argument" ) );
|
|
|
|
|
|
/* chase down the ipo list looking for ours */
|
|
link = G.main->ipo.first;
|
|
|
|
while( link ) {
|
|
ipo = ( Ipo * ) link;
|
|
if( ipo == self->ipo )
|
|
break;
|
|
link = link->next;
|
|
}
|
|
|
|
if( !link )
|
|
return EXPP_ReturnPyObjError
|
|
( PyExc_RuntimeError, "Ipo not found" );
|
|
|
|
/*
|
|
* Check if the input arg curve name is valid depending on the block
|
|
* type, and set param to numeric value. Invalid names will return
|
|
* param = -1.
|
|
*/
|
|
|
|
if( ipo->blocktype != ID_KE ) {
|
|
param = lookup_curve_name( cur_name, ipo->blocktype, self->mtex );
|
|
} else {
|
|
param = lookup_curve_key( cur_name, ipo );
|
|
if( param == -2 )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"unable to find matching key data for Ipo" );
|
|
}
|
|
|
|
if( param == -1 )
|
|
return EXPP_ReturnPyObjError( PyExc_NameError,
|
|
"curve name is not valid" );
|
|
|
|
/* see if the curve already exists */
|
|
for( icu = ipo->curve.first; icu; icu = icu->next )
|
|
if( icu->adrcode == param )
|
|
return EXPP_ReturnPyObjError( PyExc_ValueError,
|
|
"Ipo curve already exists" );
|
|
|
|
/* create the new ipo curve */
|
|
icu = MEM_callocN( sizeof(IpoCurve), "Python added ipocurve");
|
|
icu->blocktype = ipo->blocktype;
|
|
icu->adrcode = param;
|
|
icu->flag |= IPO_VISIBLE|IPO_AUTO_HORIZ;
|
|
set_icu_vars( icu );
|
|
BLI_addtail( &(ipo->curve), icu);
|
|
|
|
allspace( REMAKEIPO, 0 );
|
|
EXPP_allqueue( REDRAWIPO, 0 );
|
|
|
|
/* create a bpy wrapper for the new ipo curve */
|
|
return IpoCurve_CreatePyObject( icu );
|
|
}
|
|
|
|
/*
|
|
Function: Ipo_delCurve
|
|
Bpy: Blender.Ipo.delCurve(curtype)
|
|
|
|
delete an existing curve from IPO.
|
|
example:
|
|
ipo = Blender.Ipo.New('Object','ObIpo')
|
|
cu = ipo.delCurve('LocX')
|
|
*/
|
|
|
|
static PyObject *Ipo_delCurve( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
IpoCurve *icu;
|
|
char *strname;
|
|
|
|
if( !PyArg_ParseTuple( args, "s", &strname ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected string argument" );
|
|
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next ) {
|
|
if( !strcmp( strname, getIpoCurveName( icu ) ) ) {
|
|
del_ipocurve( self->ipo, icu );
|
|
Py_RETURN_NONE;
|
|
}
|
|
}
|
|
|
|
return EXPP_ReturnPyObjError( PyExc_ValueError, "IpoCurve not found" );
|
|
}
|
|
/*
|
|
*/
|
|
|
|
static PyObject *Ipo_getCurve( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
IpoCurve *icu = NULL;
|
|
short adrcode;
|
|
PyObject *value = NULL;
|
|
|
|
if( !PyArg_ParseTuple( args, "|O", &value ) )
|
|
goto typeError;
|
|
|
|
/* if no name give, get all the Ipocurves */
|
|
if( !value )
|
|
return Ipo_getCurves( self );
|
|
|
|
/* if arg is a string or int, look up the adrcode */
|
|
if( PyString_Check( value ) ) {
|
|
char *str = PyString_AsString( value );
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next ) {
|
|
if( !strcmp( str, getIpoCurveName( icu ) ) )
|
|
return IpoCurve_CreatePyObject( icu );
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
else if( PyInt_Check( value ) ) {
|
|
adrcode = ( short )PyInt_AsLong( value );
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next ) {
|
|
if( icu->adrcode == adrcode )
|
|
return IpoCurve_CreatePyObject( icu );
|
|
}
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
typeError:
|
|
return EXPP_ReturnPyObjError(PyExc_TypeError,
|
|
"expected string or int argument" );
|
|
}
|
|
|
|
static PyObject *Ipo_getCurves( BPy_Ipo * self )
|
|
{
|
|
PyObject *attr = PyList_New( 0 );
|
|
IpoCurve *icu;
|
|
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next )
|
|
PyList_Append( attr, IpoCurve_CreatePyObject( icu ) );
|
|
|
|
return attr;
|
|
}
|
|
|
|
/*
|
|
* return a list of valid curve name constants for the Ipo
|
|
*/
|
|
|
|
static PyObject *Ipo_getCurveNames( BPy_Ipo * self )
|
|
{
|
|
namefunc lookup_name;
|
|
int size;
|
|
PyObject *dict;
|
|
int *vals = NULL;
|
|
char name[32];
|
|
PyObject *attr = Py_None;
|
|
|
|
/* determine what type of Ipo we are */
|
|
|
|
switch ( self->ipo->blocktype ) {
|
|
case ID_OB:
|
|
lookup_name = (namefunc)getname_ob_ei;
|
|
vals = ob_ar;
|
|
size = OB_TOTIPO;
|
|
strcpy( name, "OB_" );
|
|
break;
|
|
case ID_MA:
|
|
lookup_name = (namefunc)getname_mat_ei;
|
|
vals = ma_ar;
|
|
size = MA_TOTIPO;
|
|
strcpy( name, "MA_" );
|
|
break;
|
|
case ID_CA:
|
|
lookup_name = (namefunc)getname_cam_ei;
|
|
vals = cam_ar;
|
|
size = CAM_TOTIPO;
|
|
strcpy( name, "CA_" );
|
|
break;
|
|
case ID_LA:
|
|
lookup_name = (namefunc)getname_la_ei;
|
|
vals = la_ar;
|
|
size = LA_TOTIPO;
|
|
strcpy( name, "LA_" );
|
|
break;
|
|
case ID_TE:
|
|
lookup_name = (namefunc)getname_tex_ei;
|
|
vals = te_ar;
|
|
size = TE_TOTIPO;
|
|
strcpy( name, "TE_" );
|
|
break;
|
|
case ID_WO:
|
|
lookup_name = (namefunc)getname_world_ei;
|
|
vals = wo_ar;
|
|
size = WO_TOTIPO;
|
|
strcpy( name, "WO_" );
|
|
break;
|
|
case ID_PO:
|
|
lookup_name = (namefunc)getname_ac_ei;
|
|
vals = ac_ar;
|
|
size = AC_TOTIPO;
|
|
strcpy( name, "PO_" );
|
|
break;
|
|
case ID_CO:
|
|
lookup_name = (namefunc)getname_co_ei;
|
|
vals = co_ar;
|
|
size = CO_TOTIPO;
|
|
strcpy( name, "CO_" );
|
|
break;
|
|
case ID_CU:
|
|
lookup_name = (namefunc)getname_cu_ei;
|
|
vals = cu_ar;
|
|
size = CU_TOTIPO;
|
|
strcpy( name, "CU_" );
|
|
break;
|
|
case ID_SEQ:
|
|
lookup_name = (namefunc)getname_seq_ei;
|
|
vals = seq_ar;
|
|
size = SEQ_TOTIPO;
|
|
strcpy( name, "SQ_" );
|
|
break;
|
|
case ID_KE:
|
|
{
|
|
Key *key;
|
|
|
|
/* find the ipo in the keylist */
|
|
for( key = G.main->key.first; key; key = key->id.next ) {
|
|
if( key->ipo == self->ipo ) {
|
|
KeyBlock *block = key->block.first;
|
|
attr = PyList_New( 0 );
|
|
|
|
/* add each name to the list */
|
|
for( block = key->block.first; block; block = block->next )
|
|
PyList_Append( attr,
|
|
PyString_FromString( block->name ) );
|
|
|
|
return attr;
|
|
}
|
|
}
|
|
|
|
/* error if the ipo isn't in the list */
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"unable to find matching key data for Ipo" );
|
|
}
|
|
default:
|
|
Py_DECREF( attr );
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"unknown Ipo type" );
|
|
}
|
|
|
|
/*
|
|
* go through the list of adrcodes to find names, then add to dictionary
|
|
* with string as key and adrcode as value
|
|
*/
|
|
|
|
dict = PyModule_GetDict( submodule );
|
|
attr = PyConstant_New();
|
|
|
|
while( size-- ) {
|
|
char *ptr = name+3;
|
|
strcpy( name+3, lookup_name( *vals ) );
|
|
while( *ptr ) {
|
|
*ptr = (char)toupper( *ptr );
|
|
++ptr;
|
|
}
|
|
PyConstant_Insert( (BPy_constant *)attr, name,
|
|
PyInt_FromLong( *vals ) );
|
|
++vals;
|
|
}
|
|
return attr;
|
|
}
|
|
|
|
void generate_curveconsts( PyObject* module )
|
|
{
|
|
namefunc lookup_name = NULL;
|
|
int size = 0;
|
|
int *vals = NULL;
|
|
char name[32];
|
|
|
|
unsigned int i = 0;
|
|
static short curvelist[] = {
|
|
ID_OB, ID_MA, ID_CA, ID_LA, ID_TE, ID_WO, ID_PO, ID_CO, ID_CU, ID_SEQ
|
|
};
|
|
|
|
for( i = 0; i < sizeof(curvelist)/sizeof(short); ++i ) {
|
|
switch ( curvelist[i] ) {
|
|
case ID_OB:
|
|
lookup_name = (namefunc)getname_ob_ei;
|
|
vals = ob_ar;
|
|
size = OB_TOTIPO;
|
|
strcpy( name, "OB_" );
|
|
break;
|
|
case ID_MA:
|
|
lookup_name = (namefunc)getname_mat_ei;
|
|
vals = ma_ar;
|
|
size = MA_TOTIPO;
|
|
strcpy( name, "MA_" );
|
|
break;
|
|
case ID_CA:
|
|
lookup_name = (namefunc)getname_cam_ei;
|
|
vals = cam_ar;
|
|
size = CAM_TOTIPO;
|
|
strcpy( name, "CA_" );
|
|
break;
|
|
case ID_LA:
|
|
lookup_name = (namefunc)getname_la_ei;
|
|
vals = la_ar;
|
|
size = LA_TOTIPO;
|
|
strcpy( name, "LA_" );
|
|
break;
|
|
case ID_TE:
|
|
lookup_name = (namefunc)getname_tex_ei;
|
|
vals = te_ar;
|
|
size = TE_TOTIPO;
|
|
strcpy( name, "TE_" );
|
|
break;
|
|
case ID_WO:
|
|
lookup_name = (namefunc)getname_world_ei;
|
|
vals = wo_ar;
|
|
size = WO_TOTIPO;
|
|
strcpy( name, "WO_" );
|
|
break;
|
|
case ID_PO:
|
|
lookup_name = (namefunc)getname_ac_ei;
|
|
vals = ac_ar;
|
|
size = AC_TOTIPO;
|
|
strcpy( name, "PO_" );
|
|
break;
|
|
case ID_CO:
|
|
lookup_name = (namefunc)getname_co_ei;
|
|
vals = co_ar;
|
|
size = CO_TOTIPO;
|
|
strcpy( name, "CO_" );
|
|
break;
|
|
case ID_CU:
|
|
lookup_name = (namefunc)getname_cu_ei;
|
|
vals = cu_ar;
|
|
size = CU_TOTIPO;
|
|
strcpy( name, "CU_" );
|
|
break;
|
|
case ID_SEQ:
|
|
lookup_name = (namefunc)getname_seq_ei;
|
|
vals = seq_ar;
|
|
size = SEQ_TOTIPO;
|
|
strcpy( name, "SQ_" );
|
|
break;
|
|
}
|
|
|
|
while( size-- ) {
|
|
char *ptr = name+3;
|
|
strcpy( name+3, lookup_name( *vals ) );
|
|
while( *ptr ) {
|
|
*ptr = (char)toupper( *ptr );
|
|
++ptr;
|
|
}
|
|
PyModule_AddIntConstant( module, name, *vals );
|
|
++vals;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
* get the current texture channel number, if defined
|
|
*/
|
|
|
|
static PyObject *Ipo_getChannel( BPy_Ipo * self )
|
|
{
|
|
if( self->mtex != -1 )
|
|
return Py_BuildValue( "h", self->mtex );
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/*
|
|
* set the current texture channel number, if defined
|
|
*/
|
|
|
|
static int Ipo_setChannel( BPy_Ipo * self, PyObject * value )
|
|
{
|
|
if( self->mtex != -1 )
|
|
return EXPP_setIValueRange( value, &self->mtex, 0, 9, 'h' );
|
|
return 0;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Ipo_dealloc */
|
|
/* Description: This is a callback function for the BPy_Ipo type. It is */
|
|
/* the destructor function. */
|
|
/*****************************************************************************/
|
|
static void Ipo_dealloc( BPy_Ipo * self )
|
|
{
|
|
PyObject_DEL( self );
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Ipo_compare */
|
|
/* Description: This compares 2 ipo python types, == or != only. */
|
|
/*****************************************************************************/
|
|
static int Ipo_compare( BPy_Ipo * a, BPy_Ipo * b )
|
|
{
|
|
return ( a->ipo == b->ipo ) ? 0 : -1;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Ipo_repr */
|
|
/* Description: This is a callback function for the BPy_Ipo type. It */
|
|
/* builds a meaningful string to represent ipo objects. */
|
|
/*****************************************************************************/
|
|
static PyObject *Ipo_repr( BPy_Ipo * self )
|
|
{
|
|
char *param;
|
|
|
|
switch ( self->ipo->blocktype ) {
|
|
case ID_OB:
|
|
param = "Object"; break;
|
|
case ID_CA:
|
|
param = "Camera"; break;
|
|
case ID_LA:
|
|
param = "Lamp"; break;
|
|
case ID_TE:
|
|
param = "Texture"; break;
|
|
case ID_WO:
|
|
param = "World"; break;
|
|
case ID_MA:
|
|
param = "Material"; break;
|
|
case ID_PO:
|
|
param = "Action"; break;
|
|
case ID_CO:
|
|
param = "Constriant"; break;
|
|
case ID_CU:
|
|
param = "Curve"; break;
|
|
case ID_SEQ:
|
|
param = "Sequence"; break;
|
|
case ID_KE:
|
|
param = "Key"; break;
|
|
default:
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"unknown Ipo type" );
|
|
}
|
|
return PyString_FromFormat( "[Ipo \"%s\" (%s)]", self->ipo->id.name + 2,
|
|
param );
|
|
}
|
|
|
|
/* Three Python Ipo_Type helper functions needed by the Object module: */
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Ipo_CreatePyObject */
|
|
/* Description: This function will create a new BPy_Ipo from an existing */
|
|
/* Blender ipo structure. */
|
|
/*****************************************************************************/
|
|
PyObject *Ipo_CreatePyObject( Ipo * ipo )
|
|
{
|
|
BPy_Ipo *pyipo;
|
|
pyipo = ( BPy_Ipo * ) PyObject_NEW( BPy_Ipo, &Ipo_Type );
|
|
if( !pyipo )
|
|
return EXPP_ReturnPyObjError( PyExc_MemoryError,
|
|
"couldn't create BPy_Ipo object" );
|
|
pyipo->ipo = ipo;
|
|
pyipo->iter = 0;
|
|
if( pyipo->ipo->blocktype == ID_WO || pyipo->ipo->blocktype == ID_LA ||
|
|
pyipo->ipo->blocktype == ID_MA )
|
|
pyipo->mtex = 0;
|
|
else
|
|
pyipo->mtex = -1;
|
|
return ( PyObject * ) pyipo;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Ipo_CheckPyObject */
|
|
/* Description: This function returns true when the given PyObject is of the */
|
|
/* type Ipo. Otherwise it will return false. */
|
|
/*****************************************************************************/
|
|
int Ipo_CheckPyObject( PyObject * pyobj )
|
|
{
|
|
return ( pyobj->ob_type == &Ipo_Type );
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Ipo_FromPyObject */
|
|
/* Description: This function returns the Blender ipo from the given */
|
|
/* PyObject. */
|
|
/*****************************************************************************/
|
|
Ipo *Ipo_FromPyObject( PyObject * pyobj )
|
|
{
|
|
return ( ( BPy_Ipo * ) pyobj )->ipo;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Ipo_length */
|
|
/* Description: This function counts the number of curves accessible for the */
|
|
/* PyObject. */
|
|
/*****************************************************************************/
|
|
static int Ipo_length( BPy_Ipo * self )
|
|
{
|
|
IpoCurve *icu;
|
|
int len = 0;
|
|
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next ) {
|
|
if( self->mtex == -1 || icu->adrcode < MA_MAP1 ||
|
|
icu->adrcode & texchannel_to_adrcode( self->mtex ) )
|
|
++len;
|
|
}
|
|
return len;
|
|
}
|
|
|
|
/*
|
|
* "mapping" operator getter: return an IpoCurve it we can find it
|
|
*/
|
|
|
|
static PyObject *Ipo_getIpoCurveByName( BPy_Ipo * self, PyObject * key )
|
|
{
|
|
IpoCurve *icu = NULL;
|
|
int adrcode;
|
|
|
|
/* if Ipo is not ShapeKey and arg is an int, look up the adrcode */
|
|
if( self->ipo->blocktype != ID_KE && PyNumber_Check( key ) )
|
|
adrcode = lookup_curve_adrcode( PyInt_AsLong( key ),
|
|
self->ipo->blocktype, self->mtex );
|
|
/* if Ipo is ShapeKey and arg is string, look up the adrcode */
|
|
else if( self->ipo->blocktype == ID_KE && PyString_Check( key ) ) {
|
|
adrcode = lookup_curve_key( PyString_AS_STRING( key ), self->ipo );
|
|
if( adrcode == -2 )
|
|
return EXPP_ReturnPyObjError( PyExc_RuntimeError,
|
|
"unable to find matching key data for Ipo" );
|
|
}
|
|
else
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected int or string key" );
|
|
|
|
/* if no adrcode found, value error */
|
|
if( adrcode == -1 )
|
|
return EXPP_ReturnPyObjError( PyExc_KeyError, "invalid curve key" );
|
|
|
|
/* search for a matching adrcode */
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next )
|
|
if( icu->adrcode == adrcode )
|
|
return IpoCurve_CreatePyObject( icu );
|
|
|
|
/* no curve found */
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
/*
|
|
* "mapping" operator setter: create or delete an IpoCurve it we can find it
|
|
*/
|
|
|
|
static int Ipo_setIpoCurveByName( BPy_Ipo * self, PyObject * key,
|
|
PyObject * arg )
|
|
{
|
|
IpoCurve *icu;
|
|
Ipo *ipo = self->ipo;
|
|
short adrcode;
|
|
|
|
/* "del ipo[const]" will send NULL here; give an error */
|
|
if( !arg )
|
|
return EXPP_ReturnIntError( PyExc_NotImplementedError,
|
|
"del operator not supported" );
|
|
|
|
/* "del ipo[const]" will send NULL here; give an error */
|
|
if( self->ipo->blocktype == ID_KE )
|
|
return EXPP_ReturnIntError( PyExc_TypeError,
|
|
"creation or deletion of Shape Keys not supported" );
|
|
|
|
/* check for int argument */
|
|
if( !PyNumber_Check( key ) )
|
|
return EXPP_ReturnIntError( PyExc_TypeError, "expected int key" );
|
|
|
|
/* look up the key, return error if not found */
|
|
adrcode = lookup_curve_adrcode( PyInt_AsLong( key ),
|
|
self->ipo->blocktype, self->mtex );
|
|
|
|
if( adrcode == -1 )
|
|
return EXPP_ReturnIntError( PyExc_KeyError,
|
|
"invalid curve specified" );
|
|
|
|
/* if arg is None, delete the curve */
|
|
if( arg == Py_None ) {
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next ) {
|
|
if( icu->adrcode == adrcode ) {
|
|
del_ipocurve( ipo, icu );
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
return EXPP_ReturnIntError( PyExc_ValueError, "IpoCurve not found" );
|
|
} else {
|
|
|
|
/* create the new ipo curve */
|
|
float time, curval;
|
|
PyObject *tmp, *flt=NULL, *val=NULL;
|
|
|
|
/* error if not a sequence or sequence with other than 2 values */
|
|
if( PySequence_Size( arg ) != 2 )
|
|
goto AttrError;
|
|
|
|
/* get the time and curval */
|
|
tmp = PySequence_ITEM( arg, 0 );
|
|
flt = PyNumber_Float( tmp );
|
|
Py_DECREF( tmp );
|
|
tmp = PySequence_ITEM( arg, 1 );
|
|
val = PyNumber_Float( tmp );
|
|
Py_DECREF( tmp );
|
|
|
|
if( !flt || !val )
|
|
goto AttrError;
|
|
|
|
time = (float)PyFloat_AS_DOUBLE( flt );
|
|
curval = (float)PyFloat_AS_DOUBLE( val );
|
|
Py_DECREF( flt );
|
|
Py_DECREF( val );
|
|
|
|
/* if curve already exist, delete the original */
|
|
for( icu = ipo->curve.first; icu; icu = icu->next )
|
|
if( icu->adrcode == adrcode ) {
|
|
del_ipocurve( ipo, icu );
|
|
break;
|
|
}
|
|
|
|
/* create the new curve, then add the key */
|
|
icu = MEM_callocN( sizeof(IpoCurve), "Python added ipocurve");
|
|
icu->blocktype = ipo->blocktype;
|
|
icu->adrcode = adrcode;
|
|
icu->flag |= IPO_VISIBLE|IPO_AUTO_HORIZ;
|
|
set_icu_vars( icu );
|
|
BLI_addtail( &(ipo->curve), icu);
|
|
insert_vert_ipo( icu, time, curval );
|
|
|
|
allspace( REMAKEIPO, 0 );
|
|
EXPP_allqueue( REDRAWIPO, 0 );
|
|
|
|
return 0;
|
|
|
|
AttrError:
|
|
Py_XDECREF( val );
|
|
Py_XDECREF( flt );
|
|
return EXPP_ReturnIntError( PyExc_AttributeError,
|
|
"expected sequence of two floats" );
|
|
}
|
|
}
|
|
|
|
/*
|
|
* sequence __contains__ method (implements "x in ipo")
|
|
*/
|
|
|
|
static int Ipo_contains( BPy_Ipo *self, PyObject *key )
|
|
{
|
|
IpoCurve *icu = NULL;
|
|
int adrcode;
|
|
|
|
/* take a Ipo curve name: key must be a int */
|
|
|
|
if( self->ipo->blocktype != ID_KE && PyNumber_Check( key ) ) {
|
|
adrcode = lookup_curve_adrcode( PyInt_AsLong( key ),
|
|
self->ipo->blocktype, self->mtex );
|
|
|
|
/* if we found an adrcode for the key, search the ipo's curve */
|
|
if( adrcode != -1 ) {
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next )
|
|
if( icu->adrcode == adrcode )
|
|
return 1;
|
|
}
|
|
} else if( self->ipo->blocktype == ID_KE && PyString_Check( key ) ) {
|
|
adrcode = lookup_curve_key( PyString_AS_STRING( key ), self->ipo );
|
|
|
|
/* if we found an adrcode for the key, search the ipo's curve */
|
|
if( adrcode >= 0 ) {
|
|
for( icu = self->ipo->curve.first; icu; icu = icu->next )
|
|
if( icu->adrcode == adrcode )
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
/* no curve found */
|
|
return 0;
|
|
}
|
|
|
|
/*
|
|
* Initialize the interator index
|
|
*/
|
|
|
|
static PyObject *Ipo_getIter( BPy_Ipo * self )
|
|
{
|
|
self->iter = 0;
|
|
return EXPP_incr_ret ( (PyObject *) self );
|
|
}
|
|
|
|
/*
|
|
* Get the next Ipo curve
|
|
*/
|
|
|
|
static PyObject *Ipo_nextIter( BPy_Ipo * self )
|
|
{
|
|
int i;
|
|
IpoCurve *icu = self->ipo->curve.first;
|
|
|
|
++self->iter;
|
|
|
|
/*
|
|
* count curves only if
|
|
* (a) Ipo has no texture channels
|
|
* (b) Ipo has texture channels, but curve is not that type
|
|
* (c) Ipo has texture channels, and curve is that type, and it is
|
|
* in the active texture channel
|
|
*/
|
|
for( i = 0; icu; icu = icu->next ) {
|
|
if( self->mtex == -1 || icu->adrcode < MA_MAP1 ||
|
|
icu->adrcode & texchannel_to_adrcode( self->mtex ) ) {
|
|
++i;
|
|
|
|
/* if indices match, return the curve */
|
|
if( i == self->iter )
|
|
return IpoCurve_CreatePyObject( icu );
|
|
}
|
|
}
|
|
|
|
/* ran out of curves */
|
|
return EXPP_ReturnPyObjError( PyExc_StopIteration,
|
|
"iterator at end" );
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: Ipo_Init */
|
|
/*****************************************************************************/
|
|
PyObject *Ipo_Init( void )
|
|
{
|
|
// PyObject *submodule;
|
|
|
|
if( PyType_Ready( &Ipo_Type ) < 0 )
|
|
return NULL;
|
|
|
|
submodule = Py_InitModule3( "Blender.Ipo", M_Ipo_methods, M_Ipo_doc );
|
|
generate_curveconsts( submodule );
|
|
|
|
return submodule;
|
|
}
|
|
|
|
/*
|
|
* The following methods should be deprecated when there are equivalent
|
|
* methods in Ipocurve (if there aren't already).
|
|
*/
|
|
|
|
static PyObject *Ipo_getNBezPoints( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
int num = 0;
|
|
IpoCurve *icu = NULL;
|
|
|
|
if( !PyArg_ParseTuple( args, "i", &num ) )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"expected int argument" );
|
|
|
|
icu = self->ipo->curve.first;
|
|
while( icu && num > 0 ) {
|
|
icu = icu->next;
|
|
--num;
|
|
}
|
|
|
|
if( num < 0 && !icu )
|
|
return EXPP_ReturnPyObjError( PyExc_IndexError,
|
|
"index out of range" );
|
|
|
|
return PyInt_FromLong( icu->totvert );
|
|
}
|
|
|
|
static PyObject *Ipo_DeleteBezPoints( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
int num = 0, i = 0;
|
|
IpoCurve *icu = 0;
|
|
if( !PyArg_ParseTuple( args, "i", &num ) )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "expected int argument" ) );
|
|
icu = self->ipo->curve.first;
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "No IPO curve" ) );
|
|
for( i = 0; i < num; i++ ) {
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "Bad curve number" ) );
|
|
icu = icu->next;
|
|
|
|
}
|
|
icu->totvert--;
|
|
return ( PyInt_FromLong( icu->totvert ) );
|
|
}
|
|
|
|
/*
|
|
* Ipo_getCurveBP()
|
|
* this method is UNSUPPORTED.
|
|
* Calling this method throws a TypeError Exception.
|
|
*
|
|
* it looks like the original intent was to return the first point
|
|
* of a BPoint Ipo curve. However, BPoint ipos are not currently
|
|
* implemented.
|
|
*/
|
|
|
|
static PyObject *Ipo_getCurveBP( BPy_Ipo * self_unused, PyObject * args_unused )
|
|
{
|
|
return EXPP_ReturnPyObjError( PyExc_NotImplementedError,
|
|
"bpoint ipos are not supported" );
|
|
}
|
|
|
|
static PyObject *Ipo_getCurveBeztriple( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
struct BezTriple *ptrbt;
|
|
int num = 0, pos, i, j;
|
|
IpoCurve *icu;
|
|
PyObject *l = PyList_New( 0 );
|
|
|
|
if( !PyArg_ParseTuple( args, "ii", &num, &pos ) )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "expected int argument" ) );
|
|
icu = self->ipo->curve.first;
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "No IPO curve" ) );
|
|
for( i = 0; i < num; i++ ) {
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "Bad ipo number" ) );
|
|
icu = icu->next;
|
|
}
|
|
if( pos >= icu->totvert )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"Bad bezt number" );
|
|
|
|
ptrbt = icu->bezt + pos;
|
|
if( !ptrbt )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"No bez triple" );
|
|
|
|
for( i = 0; i < 3; i++ )
|
|
for( j = 0; j < 3; j++ )
|
|
PyList_Append( l,
|
|
PyFloat_FromDouble( ptrbt->
|
|
vec[i][j] ) );
|
|
return l;
|
|
}
|
|
|
|
static PyObject *Ipo_setCurveBeztriple( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
struct BezTriple *ptrbt;
|
|
int num = 0, pos, i;
|
|
IpoCurve *icu;
|
|
PyObject *listargs = 0;
|
|
|
|
if( !PyArg_ParseTuple( args, "iiO", &num, &pos, &listargs ) )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError,
|
|
"expected int int object argument" ) );
|
|
if( !PyTuple_Check( listargs ) )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "3rd arg should be a tuple" ) );
|
|
icu = self->ipo->curve.first;
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "No IPO curve" ) );
|
|
for( i = 0; i < num; i++ ) {
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "Bad ipo number" ) );
|
|
icu = icu->next;
|
|
}
|
|
if( pos >= icu->totvert )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"Bad bezt number" );
|
|
|
|
ptrbt = icu->bezt + pos;
|
|
if( !ptrbt )
|
|
return EXPP_ReturnPyObjError( PyExc_TypeError,
|
|
"No bez triple" );
|
|
|
|
for( i = 0; i < 9; i++ ) {
|
|
PyObject *xx = PyTuple_GetItem( listargs, i );
|
|
ptrbt->vec[i / 3][i % 3] = (float)PyFloat_AsDouble( xx );
|
|
}
|
|
|
|
Py_INCREF( Py_None );
|
|
return Py_None;
|
|
}
|
|
|
|
static PyObject *Ipo_EvaluateCurveOn( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
int num = 0, i;
|
|
IpoCurve *icu;
|
|
float time = 0;
|
|
|
|
if( !PyArg_ParseTuple( args, "if", &num, &time ) )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "expected int argument" ) );
|
|
|
|
icu = self->ipo->curve.first;
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "No IPO curve" ) );
|
|
|
|
for( i = 0; i < num; i++ ) {
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "Bad ipo number" ) );
|
|
icu = icu->next;
|
|
|
|
}
|
|
return PyFloat_FromDouble( eval_icu( icu, time ) );
|
|
}
|
|
|
|
static PyObject *Ipo_getCurvecurval( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
int numcurve = 0, i;
|
|
IpoCurve *icu;
|
|
char *stringname = 0, *str1 = 0;
|
|
|
|
icu = self->ipo->curve.first;
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError, "No IPO curve" ) );
|
|
|
|
if( PyNumber_Check( PyTuple_GetItem( args, 0 ) ) ) // args is an integer
|
|
{
|
|
if( !PyArg_ParseTuple( args, "i", &numcurve ) )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError,
|
|
"expected int or string argument" ) );
|
|
for( i = 0; i < numcurve; i++ ) {
|
|
if( !icu )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError,
|
|
"Bad ipo number" ) );
|
|
icu = icu->next;
|
|
}
|
|
}
|
|
|
|
else // args is a string
|
|
{
|
|
if( !PyArg_ParseTuple( args, "s", &stringname ) )
|
|
return ( EXPP_ReturnPyObjError
|
|
( PyExc_TypeError,
|
|
"expected int or string argument" ) );
|
|
while( icu ) {
|
|
str1 = getIpoCurveName( icu );
|
|
if( !strcmp( str1, stringname ) )
|
|
break;
|
|
icu = icu->next;
|
|
}
|
|
}
|
|
|
|
if( icu )
|
|
return PyFloat_FromDouble( icu->curval );
|
|
Py_INCREF( Py_None );
|
|
return Py_None;
|
|
}
|
|
|
|
/*
|
|
* The following methods should be deprecated when methods are pruned out.
|
|
*/
|
|
|
|
static PyObject *Ipo_oldsetRctf( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
return EXPP_setterWrapperTuple( (void *)self, args,
|
|
(setter)Ipo_setRctf );
|
|
}
|
|
|
|
static PyObject *Ipo_oldsetName( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
return EXPP_setterWrapperTuple( (void *)self, args,
|
|
(setter)Ipo_setName );
|
|
}
|
|
|
|
static PyObject *Ipo_oldsetBlocktype( BPy_Ipo * self, PyObject * args )
|
|
{
|
|
return EXPP_setterWrapperTuple( (void *)self, args,
|
|
(setter)Ipo_setBlocktype );
|
|
}
|