9cabf31ebc
This was a problem with the BezTriple type. Write access to BezTriple via 'pt' member did not work. Preferred method to access BPy type members, especially for write access, is via get*/set* methods. BezTriple.setPoints() will accept x,y coordinates as either a tuple or a list. Updated BezTriple section of Ipo module doc.
371 lines
13 KiB
C
371 lines
13 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
|
|
*
|
|
* ***** END GPL/BL DUAL LICENSE BLOCK *****
|
|
*/
|
|
|
|
#include "BezTriple.h"
|
|
|
|
#include <Python.h>
|
|
|
|
#include <BKE_main.h>
|
|
#include <BKE_global.h>
|
|
#include <BKE_object.h>
|
|
#include <BKE_library.h>
|
|
#include <BLI_blenlib.h>
|
|
#include <DNA_ipo_types.h>
|
|
|
|
#include "constant.h"
|
|
#include "gen_utils.h"
|
|
#include "modules.h"
|
|
|
|
/*****************************************************************************/
|
|
/* Python API function prototypes for the BezTriple module. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_BezTriple_New (PyObject * self, PyObject * args);
|
|
static PyObject *M_BezTriple_Get (PyObject * self, PyObject * args);
|
|
|
|
/*****************************************************************************/
|
|
/* Python C_BezTriple instance methods declarations: */
|
|
/*****************************************************************************/
|
|
static PyObject *BezTriple_setPoints (C_BezTriple * self, PyObject * args);
|
|
static PyObject *BezTriple_getPoints (C_BezTriple * self);
|
|
static PyObject *BezTriple_getTriple (C_BezTriple * self);
|
|
|
|
/*****************************************************************************/
|
|
/* Python BezTriple_Type callback function prototypes: */
|
|
/*****************************************************************************/
|
|
static void BezTripleDeAlloc (C_BezTriple * self);
|
|
static int BezTripleSetAttr (C_BezTriple * self, char *name, PyObject * v);
|
|
static PyObject *BezTripleGetAttr (C_BezTriple * self, char *name);
|
|
static PyObject *BezTripleRepr (C_BezTriple * self);
|
|
|
|
/*****************************************************************************/
|
|
/* Python method structure definition for Blender.BezTriple module: */
|
|
/*****************************************************************************/
|
|
|
|
struct PyMethodDef M_BezTriple_methods[] = {
|
|
{"New", (PyCFunction) M_BezTriple_New, METH_VARARGS | METH_KEYWORDS, 0},
|
|
{"Get", M_BezTriple_Get, METH_VARARGS, 0},
|
|
{"get", M_BezTriple_Get, METH_VARARGS, 0},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
/*****************************************************************************/
|
|
/* Python C_BezTriple methods table: */
|
|
/*****************************************************************************/
|
|
static PyMethodDef C_BezTriple_methods[] = {
|
|
/* name, method, flags, doc */
|
|
{"setPoints", (PyCFunction) BezTriple_setPoints, METH_VARARGS,
|
|
"(str) - Change BezTriple point coordinates"},
|
|
{"getPoints", (PyCFunction) BezTriple_getPoints, METH_NOARGS,
|
|
"() - return BezTriple knot point x and y coordinates"},
|
|
{"getTriple", (PyCFunction) BezTriple_getTriple, METH_NOARGS,
|
|
"() - return list of 3 floating point triplets. order is H1, knot, H2"},
|
|
{NULL, NULL, 0, NULL}
|
|
};
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Python BezTriple_Type structure definition: */
|
|
/*****************************************************************************/
|
|
PyTypeObject BezTriple_Type = {
|
|
PyObject_HEAD_INIT (NULL) 0, /* ob_size */
|
|
"BezTriple", /* tp_name */
|
|
sizeof (C_BezTriple), /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
/* methods */
|
|
(destructor) BezTripleDeAlloc, /* tp_dealloc */
|
|
0, /* tp_print */
|
|
(getattrfunc) BezTripleGetAttr, /* tp_getattr */
|
|
(setattrfunc) BezTripleSetAttr, /* tp_setattr */
|
|
0, /* tp_compare */
|
|
(reprfunc) BezTripleRepr, /* tp_repr */
|
|
0, /* tp_as_number */
|
|
0, /* tp_as_sequence */
|
|
0, /* tp_as_mapping */
|
|
0, /* tp_as_hash */
|
|
0, 0, 0, 0, 0, 0,
|
|
0, /* tp_doc */
|
|
0, 0, 0, 0, 0, 0,
|
|
C_BezTriple_methods, /* tp_methods */
|
|
0, /* tp_members */
|
|
0, /* tm_getset */
|
|
0
|
|
};
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_BezTriple_New */
|
|
/* Python equivalent: Blender.BezTriple.New */
|
|
/*****************************************************************************/
|
|
static PyObject *
|
|
M_BezTriple_New (PyObject * self, PyObject * args)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_BezTriple_Get */
|
|
/* Python equivalent: Blender.BezTriple.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_BezTriple_Get (PyObject * self, PyObject * args)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: BezTripleDeAlloc */
|
|
/* Description: This is a callback function for the C_BezTriple type. It is */
|
|
/* the destructor function. */
|
|
/*****************************************************************************/
|
|
static void
|
|
BezTripleDeAlloc (C_BezTriple * self)
|
|
{
|
|
PyObject_DEL (self);
|
|
}
|
|
|
|
static PyObject *
|
|
BezTriple_getPoints (C_BezTriple * self)
|
|
{
|
|
struct BezTriple *bezt = self->beztriple;
|
|
PyObject *l = PyList_New (0);
|
|
int i;
|
|
for (i = 0; i < 2; i++)
|
|
{
|
|
PyList_Append (l, PyFloat_FromDouble (bezt->vec[1][i]));
|
|
}
|
|
return l;
|
|
}
|
|
|
|
|
|
/*
|
|
* BezTriple_getTriple
|
|
*
|
|
* get the coordinate data for a BezTriple.
|
|
* returns a list of 3 points.
|
|
* list order is handle1, knot, handle2.
|
|
* each point consists of a list of x,y,z float values.
|
|
*/
|
|
|
|
static PyObject *
|
|
BezTriple_getTriple (C_BezTriple * self)
|
|
{
|
|
int i;
|
|
struct BezTriple *bezt = self->beztriple;
|
|
PyObject *retlist = PyList_New (0);
|
|
PyObject *point;
|
|
|
|
for (i = 0; i < 3; i++)
|
|
{
|
|
point = Py_BuildValue ("[fff]",
|
|
bezt->vec[i][0],
|
|
bezt->vec[i][1], bezt->vec[i][2]);
|
|
|
|
PyList_Append (retlist, point);
|
|
}
|
|
|
|
return retlist;
|
|
}
|
|
|
|
|
|
static PyObject *
|
|
BezTriple_setPoints (C_BezTriple * self, PyObject * args)
|
|
{
|
|
|
|
int i;
|
|
struct BezTriple *bezt = self->beztriple;
|
|
PyObject *popo = 0;
|
|
|
|
if (!PyArg_ParseTuple (args, "O", &popo))
|
|
return (EXPP_ReturnPyObjError
|
|
(PyExc_TypeError, "expected sequence argument"));
|
|
|
|
if (PySequence_Check (popo) == 0)
|
|
{
|
|
puts ("error in BezTriple_setPoints - expected sequence");
|
|
Py_INCREF (Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
{
|
|
/*
|
|
some debug stuff
|
|
this will become an overloaded args check
|
|
*/
|
|
int size = PySequence_Size (popo);
|
|
printf ("\n dbg: sequence size is %d\n", size);
|
|
}
|
|
|
|
for (i = 0; i < 2; i++)
|
|
{
|
|
PyObject *o = PySequence_GetItem (popo, i);
|
|
if (!o)
|
|
printf ("\n bad o. o no!\n");
|
|
|
|
/* bezt->vec[1][i] = PyFloat_AsDouble (PyTuple_GetItem (popo, i)); */
|
|
bezt->vec[1][i] = PyFloat_AsDouble (o);
|
|
bezt->vec[0][i] = bezt->vec[1][i] - 1;
|
|
bezt->vec[2][i] = bezt->vec[1][i] + 1;
|
|
}
|
|
|
|
/* experimental fussing with handles - ipo.c: calchandles_ipocurve */
|
|
if (bezt->vec[0][0] > bezt->vec[1][0])
|
|
bezt->vec[0][0] = bezt->vec[1][0];
|
|
|
|
if (bezt->vec[2][0] < bezt->vec[1][0])
|
|
bezt->vec[2][0] = bezt->vec[1][0];
|
|
|
|
Py_INCREF (Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function: BezTripleGetAttr */
|
|
/* Description: This is a callback function for the C_BezTriple type. It is */
|
|
/* the function that accesses C_BezTriple "member variables" and */
|
|
/* methods. */
|
|
/*****************************************************************************/
|
|
static PyObject *
|
|
BezTripleGetAttr (C_BezTriple * self, char *name)
|
|
{
|
|
if (strcmp (name, "pt") == 0)
|
|
return BezTriple_getPoints (self);
|
|
else if (strcmp (name, "vec") == 0)
|
|
return BezTriple_getTriple (self);
|
|
|
|
/* look for default methods */
|
|
return Py_FindMethod (C_BezTriple_methods, (PyObject *) self, name);
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: BezTripleSetAttr */
|
|
/* Description: This is a callback function for the C_BezTriple type. It is the */
|
|
/* function that sets BezTriple Data attributes (member variables).*/
|
|
/*****************************************************************************/
|
|
static int
|
|
BezTripleSetAttr (C_BezTriple * self, char *name, PyObject * value)
|
|
{
|
|
#if 0
|
|
/*
|
|
this does not work at the moment: Wed Apr 7 2004
|
|
when the necessary code to make pt act like a sequence is
|
|
available, it will be reenabled
|
|
*/
|
|
|
|
if (strcmp (name, "pt") == 0)
|
|
BezTriple_setPoints (self, value);
|
|
|
|
return 0; /* normal exit */
|
|
#endif
|
|
|
|
return (EXPP_ReturnIntError (PyExc_AttributeError,
|
|
"cannot set a read-only attribute"));
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: BezTripleRepr */
|
|
/* Description: This is a callback function for the C_BezTriple type. It */
|
|
/* builds a meaninful string to represent BezTriple objects. */
|
|
/*****************************************************************************/
|
|
static PyObject *
|
|
BezTripleRepr (C_BezTriple * self)
|
|
{
|
|
/* float vec[3][3];
|
|
float alfa;
|
|
short s[3][2];
|
|
short h1, h2;
|
|
char f1, f2, f3, hide;
|
|
*/
|
|
char str[1000];
|
|
sprintf (str,
|
|
"BezTriple %f %f %f %f %f %f %f %f %f %f\n %d %d %d %d %d %d %d %d %d %d %d %d\n",
|
|
self->beztriple->vec[0][0], self->beztriple->vec[0][1],
|
|
self->beztriple->vec[0][2], self->beztriple->vec[1][0],
|
|
self->beztriple->vec[1][1], self->beztriple->vec[1][2],
|
|
self->beztriple->vec[2][0], self->beztriple->vec[2][1],
|
|
self->beztriple->vec[2][2], self->beztriple->alfa,
|
|
self->beztriple->s[0][0], self->beztriple->s[0][1],
|
|
self->beztriple->s[1][0], self->beztriple->s[1][1],
|
|
self->beztriple->s[2][0], self->beztriple->s[1][1],
|
|
self->beztriple->h1, self->beztriple->h2, self->beztriple->f1,
|
|
self->beztriple->f2, self->beztriple->f3, self->beztriple->hide);
|
|
return PyString_FromString (str);
|
|
}
|
|
|
|
/* Three Python BezTriple_Type helper functions needed by the Object module: */
|
|
|
|
/*****************************************************************************/
|
|
/* Function: BezTriple_CreatePyObject */
|
|
/* Description: This function will create a new C_BezTriple from an existing */
|
|
/* Blender ipo structure. */
|
|
/*****************************************************************************/
|
|
PyObject *
|
|
BezTriple_CreatePyObject (BezTriple * bzt)
|
|
{
|
|
C_BezTriple *pybeztriple;
|
|
|
|
pybeztriple = (C_BezTriple *) PyObject_NEW (C_BezTriple, &BezTriple_Type);
|
|
|
|
if (!pybeztriple)
|
|
return EXPP_ReturnPyObjError (PyExc_MemoryError,
|
|
"couldn't create C_BezTriple object");
|
|
|
|
pybeztriple->beztriple = bzt;
|
|
|
|
return (PyObject *) pybeztriple;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: BezTriple_CheckPyObject */
|
|
/* Description: This function returns true when the given PyObject is of the */
|
|
/* type BezTriple. Otherwise it will return false. */
|
|
/*****************************************************************************/
|
|
int
|
|
BezTriple_CheckPyObject (PyObject * pyobj)
|
|
{
|
|
return (pyobj->ob_type == &BezTriple_Type);
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: BezTriple_FromPyObject */
|
|
/* Description: This function returns the Blender beztriple from the given */
|
|
/* PyObject. */
|
|
/*****************************************************************************/
|
|
BezTriple *
|
|
BezTriple_FromPyObject (PyObject * pyobj)
|
|
{
|
|
return ((C_BezTriple *) pyobj)->beztriple;
|
|
}
|