Merge branch 'blender2.7'
[blender.git] / source / blender / python / mathutils / mathutils_interpolate.c
1 /*
2  * This program is free software; you can redistribute it and/or
3  * modify it under the terms of the GNU General Public License
4  * as published by the Free Software Foundation; either version 2
5  * of the License, or (at your option) any later version.
6  *
7  * This program is distributed in the hope that it will be useful,
8  * but WITHOUT ANY WARRANTY; without even the implied warranty of
9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  * GNU General Public License for more details.
11  *
12  * You should have received a copy of the GNU General Public License
13  * along with this program; if not, write to the Free Software Foundation,
14  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
15  */
16
17 /** \file
18  * \ingroup pymathutils
19  */
20
21
22 #include <Python.h>
23
24 #include "mathutils.h"
25 #include "mathutils_interpolate.h"
26
27 #include "BLI_math.h"
28 #include "BLI_utildefines.h"
29
30 #ifndef MATH_STANDALONE /* define when building outside blender */
31 #  include "MEM_guardedalloc.h"
32 #endif
33
34 /*-------------------------DOC STRINGS ---------------------------*/
35 PyDoc_STRVAR(M_Interpolate_doc,
36 "The Blender interpolate module"
37 );
38
39 /* ---------------------------------WEIGHT CALCULATION ----------------------- */
40
41 #ifndef MATH_STANDALONE
42
43 PyDoc_STRVAR(M_Interpolate_poly_3d_calc_doc,
44 ".. function:: poly_3d_calc(veclist, pt)\n"
45 "\n"
46 "   Calculate barycentric weights for a point on a polygon.\n"
47 "\n"
48 "   :arg veclist: list of vectors\n"
49 "   :arg pt: point"
50 "   :rtype: list of per-vector weights\n"
51 );
52 static PyObject *M_Interpolate_poly_3d_calc(PyObject *UNUSED(self), PyObject *args)
53 {
54         float fp[3];
55         float (*vecs)[3];
56         Py_ssize_t len;
57
58         PyObject *point, *veclist, *ret;
59         int i;
60
61         if (!PyArg_ParseTuple(
62                 args, "OO!:poly_3d_calc",
63                 &veclist,
64                 &vector_Type, &point))
65         {
66                 return NULL;
67         }
68
69         if (BaseMath_ReadCallback((VectorObject *)point) == -1)
70                 return NULL;
71
72         fp[0] = ((VectorObject *)point)->vec[0];
73         fp[1] = ((VectorObject *)point)->vec[1];
74         if (((VectorObject *)point)->size > 2)
75                 fp[2] = ((VectorObject *)point)->vec[2];
76         else
77                 fp[2] = 0.0f;  /* if its a 2d vector then set the z to be zero */
78
79         len = mathutils_array_parse_alloc_v(((float **)&vecs), 3, veclist, __func__);
80         if (len == -1) {
81                 return NULL;
82         }
83
84         if (len) {
85                 float *weights = MEM_mallocN(sizeof(float) * len, __func__);
86
87                 interp_weights_poly_v3(weights, vecs, len, fp);
88
89                 ret = PyList_New(len);
90                 for (i = 0; i < len; i++) {
91                         PyList_SET_ITEM(ret, i, PyFloat_FromDouble(weights[i]));
92                 }
93
94                 MEM_freeN(weights);
95
96                 PyMem_Free(vecs);
97         }
98         else {
99                 ret = PyList_New(0);
100         }
101
102         return ret;
103 }
104
105 #endif /* MATH_STANDALONE */
106
107
108 static PyMethodDef M_Interpolate_methods[] = {
109 #ifndef MATH_STANDALONE
110         {"poly_3d_calc", (PyCFunction) M_Interpolate_poly_3d_calc, METH_VARARGS, M_Interpolate_poly_3d_calc_doc},
111 #endif
112         {NULL, NULL, 0, NULL},
113 };
114
115 static struct PyModuleDef M_Interpolate_module_def = {
116         PyModuleDef_HEAD_INIT,
117         "mathutils.interpolate",  /* m_name */
118         M_Interpolate_doc,  /* m_doc */
119         0,  /* m_size */
120         M_Interpolate_methods,  /* m_methods */
121         NULL,  /* m_reload */
122         NULL,  /* m_traverse */
123         NULL,  /* m_clear */
124         NULL,  /* m_free */
125 };
126
127 /*----------------------------MODULE INIT-------------------------*/
128 PyMODINIT_FUNC PyInit_mathutils_interpolate(void)
129 {
130         PyObject *submodule = PyModule_Create(&M_Interpolate_module_def);
131         return submodule;
132 }