750 lines
23 KiB
C
750 lines
23 KiB
C
/*
|
|
*
|
|
* ***** 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 "Metaball.h"
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Python Metaball_Type structure definition: */
|
|
/*****************************************************************************/
|
|
PyTypeObject Metaball_Type =
|
|
{
|
|
PyObject_HEAD_INIT(NULL)
|
|
0, /* ob_size */
|
|
"Metaball", /* tp_name */
|
|
sizeof (BPy_Metaball), /* tp_basicsize */
|
|
0, /* tp_itemsize */
|
|
/* methods */
|
|
(destructor)MetaballDeAlloc, /* tp_dealloc */
|
|
0, /* tp_print */
|
|
(getattrfunc)MetaballGetAttr, /* tp_getattr */
|
|
(setattrfunc)MetaballSetAttr, /* tp_setattr */
|
|
0, /* tp_compare */
|
|
(reprfunc)MetaballRepr, /* 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,
|
|
BPy_Metaball_methods, /* tp_methods */
|
|
0, /* tp_members */
|
|
};
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Metaball_New */
|
|
/* Python equivalent: Blender.Metaball.New */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Metaball_New(PyObject *self, PyObject *args)
|
|
{
|
|
char*name = 0;
|
|
BPy_Metaball *pymball; /* for Data object wrapper in Python */
|
|
MetaBall *blmball; /* for actual Data we create in Blender */
|
|
char buf[21];
|
|
if (!PyArg_ParseTuple(args, "|s", &name))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError,
|
|
"expected string argument (or nothing)"));
|
|
|
|
blmball = add_mball(); /* first create the MetaBall Data in Blender */
|
|
|
|
if (blmball){
|
|
/* return user count to zero since add_mball() incref'ed it */
|
|
blmball->id.us = 0;
|
|
/* now create the wrapper obj in Python */
|
|
pymball = (BPy_Metaball *)PyObject_NEW(BPy_Metaball, &Metaball_Type);
|
|
}
|
|
else
|
|
return (EXPP_ReturnPyObjError (PyExc_RuntimeError,
|
|
"couldn't create MetaBall Data in Blender"));
|
|
|
|
if (pymball == NULL)
|
|
return (EXPP_ReturnPyObjError (PyExc_MemoryError,
|
|
"couldn't create MetaBall Data object"));
|
|
|
|
pymball->metaball = blmball;
|
|
/*link Python mballer wrapper to Blender MetaBall */
|
|
if(name) { /* user gave us a name for the metaball, use it */
|
|
PyOS_snprintf(buf, sizeof(buf), "%s", name);
|
|
rename_id(&blmball->id, buf);
|
|
}
|
|
return (PyObject *)pymball;
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: M_Metaball_Get */
|
|
/* Python equivalent: Blender.Metaball.Get */
|
|
/* Description: Receives a string and returns the metaball data obj */
|
|
/* whose name matches the string. If no argument is */
|
|
/* passed in, a list of all metaball data names in the */
|
|
/* current scene is returned. */
|
|
/*****************************************************************************/
|
|
static PyObject *M_Metaball_Get(PyObject *self, PyObject *args)
|
|
{
|
|
char error_msg[64];
|
|
char *name = NULL;
|
|
MetaBall *mball_iter;
|
|
|
|
if (!PyArg_ParseTuple(args, "|s", &name))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError,
|
|
"expected string argument (or nothing)"));
|
|
|
|
mball_iter = G.main->mball.first;
|
|
|
|
if (name) { /* (name) - Search mball by name */
|
|
|
|
BPy_Metaball *wanted_mball = NULL;
|
|
|
|
while ((mball_iter) && (wanted_mball == NULL)) {
|
|
if (strcmp (name, mball_iter->id.name+2) == 0) {
|
|
wanted_mball=(BPy_Metaball*)PyObject_NEW(BPy_Metaball,&Metaball_Type);
|
|
if (wanted_mball)
|
|
wanted_mball->metaball = mball_iter;
|
|
}
|
|
mball_iter = mball_iter->id.next;
|
|
}
|
|
|
|
if (wanted_mball == NULL) { /* Requested mball doesn't exist */
|
|
PyOS_snprintf(error_msg, sizeof(error_msg),
|
|
"MetaBall \"%s\" not found", name);
|
|
return (EXPP_ReturnPyObjError (PyExc_NameError, error_msg));
|
|
}
|
|
|
|
return (PyObject *)wanted_mball;
|
|
}
|
|
|
|
else { /* () - return a list of all mballs in the scene */
|
|
PyObject *mballlist;
|
|
|
|
mballlist = PyList_New (0);
|
|
|
|
if (mballlist == NULL)
|
|
return (PythonReturnErrorObject (PyExc_MemoryError,
|
|
"couldn't create PyList"));
|
|
|
|
while (mball_iter) {
|
|
BPy_Metaball *found_mball=(BPy_Metaball*)PyObject_NEW(BPy_Metaball,&Metaball_Type);
|
|
found_mball->metaball = mball_iter;
|
|
PyList_Append (mballlist, (PyObject *)found_mball);
|
|
mball_iter = mball_iter->id.next;
|
|
}
|
|
|
|
return (mballlist);
|
|
}
|
|
|
|
}
|
|
|
|
/******************************************************************************/
|
|
/* Function: Metaball_Init */
|
|
/******************************************************************************/
|
|
PyObject *Metaball_Init (void)
|
|
{
|
|
PyObject *submodule;
|
|
|
|
Metaball_Type.ob_type = &PyType_Type;
|
|
|
|
submodule = Py_InitModule3("Blender.Metaball",
|
|
M_Metaball_methods, M_Metaball_doc);
|
|
|
|
return (submodule);
|
|
}
|
|
|
|
/*******************************************************************************/
|
|
/* Python BPy_Metaball methods: */
|
|
/*******************************************************************************/
|
|
static PyObject *Metaball_getName(BPy_Metaball *self)
|
|
{
|
|
|
|
PyObject *attr = PyString_FromString(self->metaball->id.name+2);
|
|
|
|
if (attr) return attr;
|
|
|
|
return (EXPP_ReturnPyObjError (PyExc_RuntimeError,
|
|
"couldn't get Metaball.name attribute"));
|
|
}
|
|
|
|
|
|
|
|
static PyObject *Metaball_setName(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
char *name = NULL;
|
|
char buf[20];
|
|
|
|
|
|
if (!PyArg_ParseTuple(args, "s", &name))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError,
|
|
"expected string argument"));
|
|
PyOS_snprintf(buf, sizeof(buf), "%s", name);
|
|
rename_id(&self->metaball->id, buf);
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
|
|
static PyObject *Metaball_getBbox(BPy_Metaball *self)
|
|
{
|
|
int i,j;
|
|
PyObject* ll;
|
|
PyObject* l = PyList_New(0);
|
|
if (self->metaball->bb == NULL) {Py_INCREF(Py_None);return Py_None;}
|
|
for(i = 0;i<8;i++)
|
|
{
|
|
ll = PyList_New(0);
|
|
for(j = 0;j<3;j++)
|
|
PyList_Append( ll, PyFloat_FromDouble(self->metaball->bb->vec[i][j]));
|
|
PyList_Append( l, ll);
|
|
}
|
|
|
|
return l;
|
|
}
|
|
|
|
static PyObject *Metaball_getNMetaElems(BPy_Metaball *self)
|
|
{
|
|
int i = 0;
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
if(!ptr) return (PyInt_FromLong(0) );
|
|
while(ptr)
|
|
{
|
|
i++;
|
|
ptr = ptr->next;
|
|
}
|
|
return (PyInt_FromLong(i) );
|
|
}
|
|
|
|
|
|
|
|
static PyObject *Metaball_getloc(BPy_Metaball *self)
|
|
{
|
|
PyObject* l = PyList_New(0);
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->loc[0]));
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->loc[1]));
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->loc[2]));
|
|
return l;
|
|
}
|
|
|
|
static PyObject *Metaball_setloc(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
PyObject *listargs=0;
|
|
int i;
|
|
if (!PyArg_ParseTuple(args, "O", &listargs))
|
|
return (EXPP_ReturnPyObjError(PyExc_TypeError,"expected a list"));
|
|
if (!PyList_Check(listargs))
|
|
return (EXPP_ReturnPyObjError(PyExc_TypeError,"expected a list"));
|
|
for(i=0;i<3;i++){
|
|
PyObject * xx = PyList_GetItem(listargs,i);
|
|
self->metaball->loc[i] = PyFloat_AsDouble(xx);
|
|
}
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyObject *Metaball_getrot(BPy_Metaball *self)
|
|
{
|
|
PyObject* l = PyList_New(0);
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->rot[0]));
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->rot[1]));
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->rot[2]));
|
|
return l;
|
|
}
|
|
|
|
static PyObject *Metaball_setrot(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
PyObject *listargs=0;
|
|
int i;
|
|
if (!PyArg_ParseTuple(args, "O", &listargs))
|
|
return (EXPP_ReturnPyObjError(PyExc_TypeError,"expected a list"));
|
|
if (!PyList_Check(listargs))
|
|
return (EXPP_ReturnPyObjError(PyExc_TypeError,"expected a list"));
|
|
for(i=0;i<3;i++){
|
|
PyObject * xx = PyList_GetItem(listargs,i);
|
|
self->metaball->rot[i] = PyFloat_AsDouble(xx);
|
|
}
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyObject *Metaball_getsize(BPy_Metaball *self)
|
|
{
|
|
PyObject* l = PyList_New(0);
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->size[0]));
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->size[1]));
|
|
PyList_Append( l, PyFloat_FromDouble(self->metaball->size[2]));
|
|
return l;
|
|
}
|
|
|
|
static PyObject *Metaball_setsize(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
PyObject *listargs=0;
|
|
int i;
|
|
if (!PyArg_ParseTuple(args, "O", &listargs))
|
|
return (EXPP_ReturnPyObjError(PyExc_TypeError,"expected a list"));
|
|
if (!PyList_Check(listargs))
|
|
return (EXPP_ReturnPyObjError(PyExc_TypeError,"expected a list"));
|
|
for(i=0;i<3;i++){
|
|
PyObject * xx = PyList_GetItem(listargs,i);
|
|
self->metaball->size[i] = PyFloat_AsDouble(xx);
|
|
}
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
}
|
|
|
|
static PyObject *Metaball_getWiresize(BPy_Metaball *self)
|
|
{
|
|
return PyFloat_FromDouble(self->metaball->wiresize);
|
|
}
|
|
|
|
static PyObject *Metaball_setWiresize(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
|
|
float val;
|
|
if (!PyArg_ParseTuple(args, "f", &val))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError,"expected float args"));
|
|
|
|
self->metaball->wiresize = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
static PyObject *Metaball_getRendersize(BPy_Metaball *self)
|
|
{
|
|
return PyFloat_FromDouble(self->metaball->rendersize);
|
|
}
|
|
|
|
static PyObject *Metaball_setRendersize(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
|
|
float val;
|
|
if (!PyArg_ParseTuple(args, "f", &val))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError,"expected float args"));
|
|
|
|
self->metaball->rendersize = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
static PyObject *Metaball_getThresh(BPy_Metaball *self)
|
|
{
|
|
return PyFloat_FromDouble(self->metaball->thresh);
|
|
}
|
|
|
|
static PyObject *Metaball_setThresh(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
|
|
float val;
|
|
if (!PyArg_ParseTuple(args, "f", &val))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError,"expected float args"));
|
|
|
|
self->metaball->thresh = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
|
/*******************************************************************************/
|
|
/* get/set metaelems data, */
|
|
/*******************************************************************************/
|
|
|
|
static PyObject *Metaball_getMetadata(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
int num;
|
|
int i = 0;
|
|
char*name = NULL;
|
|
MetaElem *ptr;
|
|
|
|
if (!PyArg_ParseTuple(args, "si", &name,&num))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, \
|
|
"expected (string int) argument"));
|
|
/*jump to the num-th MetaElem*/
|
|
ptr = self->metaball->elems.first;
|
|
if(!ptr)
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "no metaelem found"));
|
|
for(i = 0;i<num;i++)
|
|
{
|
|
if(!ptr)
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "metaelem not found"));
|
|
ptr = ptr->next;
|
|
}
|
|
if(!strcmp(name,"type"))
|
|
return (PyInt_FromLong(ptr->type));
|
|
if(!strcmp(name,"x"))
|
|
return (PyFloat_FromDouble(ptr->x));
|
|
if(!strcmp(name,"y"))
|
|
return (PyFloat_FromDouble(ptr->y));
|
|
if(!strcmp(name,"z"))
|
|
return (PyFloat_FromDouble(ptr->z));
|
|
if(!strcmp(name,"expx"))
|
|
return (PyFloat_FromDouble(ptr->expx));
|
|
if(!strcmp(name,"expy"))
|
|
return (PyFloat_FromDouble(ptr->expy));
|
|
if(!strcmp(name,"expz"))
|
|
return (PyFloat_FromDouble(ptr->expz));
|
|
if(!strcmp(name,"rad"))
|
|
return (PyFloat_FromDouble(ptr->rad));
|
|
if(!strcmp(name,"rad2"))
|
|
return (PyFloat_FromDouble(ptr->rad2));
|
|
if(!strcmp(name,"s"))
|
|
return (PyFloat_FromDouble(ptr->s));
|
|
if(!strcmp(name,"len"))
|
|
return (PyFloat_FromDouble(ptr->len));
|
|
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "unknown name "));
|
|
}
|
|
|
|
|
|
|
|
static PyObject *Metaball_setMetadata(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
int num;
|
|
int i = 0;
|
|
char*name = NULL;
|
|
int intval=-1;
|
|
float floatval=0;
|
|
MetaElem *ptr;
|
|
|
|
if (!PyArg_ParseTuple(args, "sif", &name,&num,&floatval))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, \
|
|
"expected string,int,int float arguments"));
|
|
intval = (int)floatval;
|
|
/*jump to the num-th MetaElem*/
|
|
ptr = self->metaball->elems.first;
|
|
if(!ptr)
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "metaelem not found"));
|
|
for(i = 0;i<num;i++)
|
|
{
|
|
if(!ptr)
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "metaelem not found"));
|
|
ptr = ptr->next;
|
|
}
|
|
if(!strcmp(name,"type"))
|
|
{ptr->type=intval;return (PyInt_FromLong(intval));}
|
|
{ptr->pad=intval;return (PyInt_FromLong(intval));}
|
|
if(!strcmp(name,"x"))
|
|
{ptr->x=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"y"))
|
|
{ptr->y=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"z"))
|
|
{ptr->z=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"expx"))
|
|
{ptr->expx=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"expy"))
|
|
{ptr->expy=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"expz"))
|
|
{ptr->expz=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"rad"))
|
|
{ptr->rad=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"rad2"))
|
|
{ptr->rad2=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"s"))
|
|
{ptr->s=floatval;return (PyFloat_FromDouble(floatval));}
|
|
if(!strcmp(name,"len"))
|
|
{ptr->len=floatval;return (PyFloat_FromDouble(floatval));}
|
|
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "unknown field "));
|
|
}
|
|
|
|
static PyObject *Metaball_getMetatype(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
int num;
|
|
int i = 0;
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
if (!PyArg_ParseTuple(args, "i", &num))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, \
|
|
"expected int argument"));
|
|
if(!ptr) return (PyInt_FromLong(0));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
return (PyInt_FromLong(ptr->type));
|
|
}
|
|
|
|
|
|
|
|
static PyObject *Metaball_setMetatype(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
int num,val, i = 0;
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
if (!PyArg_ParseTuple(args, "ii", &num,&val))
|
|
return (EXPP_ReturnPyObjError(PyExc_TypeError,"expected int int arguments"));
|
|
if(!ptr) return (EXPP_ReturnPyObjError (PyExc_TypeError, "No MetaElem"));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
ptr->type = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
|
static PyObject *Metaball_getMetax(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
int num;
|
|
int i = 0;
|
|
if (!PyArg_ParseTuple(args, "i", &num))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int argument"));
|
|
if(!ptr) return (PyFloat_FromDouble(0));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
return (PyFloat_FromDouble(ptr->x));
|
|
}
|
|
|
|
|
|
|
|
static PyObject *Metaball_setMetax(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
int num, i = 0;
|
|
float val;
|
|
if (!PyArg_ParseTuple(args, "if", &num,&val))
|
|
return (EXPP_ReturnPyObjError(PyExc_TypeError,"expected int float args"));
|
|
if(!ptr) return (EXPP_ReturnPyObjError (PyExc_TypeError, "No MetaElem"));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
ptr->x = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
static PyObject *Metaball_getMetay(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
int num;
|
|
int i = 0;
|
|
if (!PyArg_ParseTuple(args, "i", &num))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int argument"));
|
|
if(!ptr) return (PyFloat_FromDouble(0));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
return (PyFloat_FromDouble(ptr->y));
|
|
}
|
|
|
|
|
|
|
|
static PyObject *Metaball_setMetay(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
int num, i = 0;
|
|
float val;
|
|
if (!PyArg_ParseTuple(args, "if", &num,&val))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int float args"));
|
|
if(!ptr) return (EXPP_ReturnPyObjError (PyExc_TypeError, "No MetaElem"));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
ptr->y = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
|
static PyObject *Metaball_getMetaz(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
int num;
|
|
int i = 0;
|
|
if (!PyArg_ParseTuple(args, "i", &num))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int argument"));
|
|
if(!ptr) return (PyFloat_FromDouble(0));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
return (PyFloat_FromDouble(ptr->z));
|
|
}
|
|
|
|
static PyObject *Metaball_setMetaz(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
int num, i = 0;
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
float val;
|
|
if (!PyArg_ParseTuple(args, "if", &num,&val))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int float args"));
|
|
if(!ptr) return (EXPP_ReturnPyObjError (PyExc_TypeError, "No MetaElem"));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
ptr->z = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
|
static PyObject *Metaball_getMetas(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
int num;
|
|
int i = 0;
|
|
if (!PyArg_ParseTuple(args, "i", &num))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int argument"));
|
|
if(!ptr) return (PyFloat_FromDouble(0));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
return (PyFloat_FromDouble(ptr->s));
|
|
}
|
|
|
|
static PyObject *Metaball_setMetas(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
int num, i = 0;
|
|
MetaElem*ptr = self->metaball->elems.first;
|
|
float val;
|
|
if (!PyArg_ParseTuple(args, "if", &num,&val))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int float args"));
|
|
if(!ptr) return (EXPP_ReturnPyObjError (PyExc_TypeError, "No MetaElem"));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
ptr->s = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
static PyObject *Metaball_getMetalen(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
int num;
|
|
int i = 0;
|
|
MetaElem *ptr;
|
|
|
|
if (!PyArg_ParseTuple(args, "i", &num))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int argument"));
|
|
ptr = self->metaball->elems.first;
|
|
if(!ptr) return (PyFloat_FromDouble(0));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
return (PyFloat_FromDouble(ptr->len));
|
|
}
|
|
|
|
static PyObject *Metaball_setMetalen(BPy_Metaball *self,PyObject*args)
|
|
{
|
|
int num, i = 0;
|
|
float val;
|
|
MetaElem *ptr;
|
|
|
|
if (!PyArg_ParseTuple(args, "if", &num,&val))
|
|
return (EXPP_ReturnPyObjError (PyExc_TypeError, "expected int float args"));
|
|
ptr = self->metaball->elems.first;
|
|
if(!ptr) return (EXPP_ReturnPyObjError (PyExc_TypeError, "No MetaElem"));
|
|
for(i = 0;i<num;i++){ptr = ptr->next;}
|
|
ptr->len = val;
|
|
|
|
Py_INCREF(Py_None);
|
|
return Py_None;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/*****************************************************************************/
|
|
/* Function: MetaballDeAlloc */
|
|
/* Description: This is a callback function for the BPy_Metaball type. It is */
|
|
/* the destructor function. */
|
|
/*****************************************************************************/
|
|
static void MetaballDeAlloc (BPy_Metaball *self)
|
|
{
|
|
PyObject_DEL (self);
|
|
}
|
|
/*
|
|
static int MetaballPrint (BPy_Metaball *self, FILE *fp, int flags)
|
|
{
|
|
fprintf(fp, "[MetaBall \"%s\"]", self->metaball->id.name+2);
|
|
return 0;
|
|
|
|
}
|
|
*/
|
|
/*****************************************************************************/
|
|
/* Function: MetaballGetAttr */
|
|
/* Description: This is a callback function for the BPy_Metaball type. It is */
|
|
/* the function that accesses BPy_Metaball "member variables" and */
|
|
/* methods. */
|
|
/*****************************************************************************/
|
|
static PyObject *MetaballGetAttr (BPy_Metaball *self, char *name)
|
|
{
|
|
|
|
if (strcmp (name, "name") == 0)return Metaball_getName (self);
|
|
if (strcmp (name, "rot") == 0)return Metaball_getrot (self);
|
|
if (strcmp (name, "loc") == 0)return Metaball_getloc (self);
|
|
if (strcmp (name, "size") == 0)return Metaball_getsize (self);
|
|
return Py_FindMethod(BPy_Metaball_methods, (PyObject *)self, name);
|
|
}
|
|
|
|
/*******************************************************************************/
|
|
/* Function: MetaballSetAttr */
|
|
/* Description: This is a callback function for the BPy_Metaball type. It is the */
|
|
/* function that sets Metaball Data attributes (member variables).*/
|
|
/*******************************************************************************/
|
|
static int MetaballSetAttr (BPy_Metaball *self, char *name, PyObject *value)
|
|
{
|
|
PyObject *valtuple = Py_BuildValue("(O)", value);
|
|
|
|
if (!valtuple)
|
|
return EXPP_ReturnIntError(PyExc_MemoryError,
|
|
"MetaballSetAttr: couldn't create PyTuple");
|
|
|
|
if (strcmp (name, "name") == 0)
|
|
{
|
|
Metaball_setName (self, valtuple);
|
|
return 0;
|
|
}
|
|
|
|
if (strcmp (name, "rot") == 0)
|
|
{
|
|
Metaball_setrot (self, valtuple);
|
|
return 0;
|
|
}
|
|
if (strcmp (name, "loc") == 0)
|
|
{
|
|
Metaball_setloc (self, valtuple);
|
|
return 0;
|
|
}
|
|
|
|
if (strcmp (name, "size") == 0)
|
|
{
|
|
Metaball_setsize (self, valtuple);
|
|
return 0;
|
|
}
|
|
return (EXPP_ReturnIntError (PyExc_KeyError,"attribute not found"));
|
|
}
|
|
|
|
/*****************************************************************************/
|
|
/* Function: MetaballRepr */
|
|
/* Description: This is a callback function for the BPy_Metaball type. It */
|
|
/* builds a meaninful string to represent metaball objects. */
|
|
/*****************************************************************************/
|
|
static PyObject *MetaballRepr (BPy_Metaball *self)
|
|
{
|
|
return PyString_FromFormat("[Metaball \"%s\"]", self->metaball->id.name+2);
|
|
}
|
|
|