3bcfb151c1
Replace plain-text type information with the type syntax used for Python's type annotations as it's more concise, especially for callbacks which often didn't include useful type information. Note that this change only applies to inline doc-strings, generated doc-strings from RNA need to be updated separately. Details: - Many minor corrections were made when "list" was incorrectly used instead of "sequence". - Some type information wasn't defined in the doc-strings and has been added. - Verbose type info would benefit from support for type aliases.
120 lines
2.8 KiB
C++
120 lines
2.8 KiB
C++
/* SPDX-FileCopyrightText: 2023 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
/** \file
|
|
* \ingroup pymathutils
|
|
*/
|
|
|
|
#include <Python.h>
|
|
|
|
#include "mathutils.hh"
|
|
#include "mathutils_interpolate.hh"
|
|
|
|
#include "BLI_math_geom.h"
|
|
#include "BLI_utildefines.h"
|
|
|
|
#ifndef MATH_STANDALONE /* define when building outside blender */
|
|
# include "MEM_guardedalloc.h"
|
|
#endif
|
|
|
|
/*-------------------------DOC STRINGS ---------------------------*/
|
|
PyDoc_STRVAR(
|
|
/* Wrap. */
|
|
M_Interpolate_doc,
|
|
"The Blender interpolate module");
|
|
|
|
/* ---------------------------------WEIGHT CALCULATION ----------------------- */
|
|
|
|
#ifndef MATH_STANDALONE
|
|
|
|
PyDoc_STRVAR(
|
|
/* Wrap. */
|
|
M_Interpolate_poly_3d_calc_doc,
|
|
".. function:: poly_3d_calc(veclist, pt)\n"
|
|
"\n"
|
|
" Calculate barycentric weights for a point on a polygon.\n"
|
|
"\n"
|
|
" :arg veclist: Sequence of 3D positions.\n"
|
|
" :type veclist: Sequence[Sequence[float]]\n"
|
|
" :arg pt: 2D or 3D position."
|
|
" :type pt: Sequence[float]"
|
|
" :return: list of per-vector weights.\n"
|
|
" :rtype: list[float]\n");
|
|
static PyObject *M_Interpolate_poly_3d_calc(PyObject * /*self*/, PyObject *args)
|
|
{
|
|
float fp[3];
|
|
float(*vecs)[3];
|
|
Py_ssize_t len;
|
|
|
|
PyObject *point, *veclist, *ret;
|
|
int i;
|
|
|
|
if (!PyArg_ParseTuple(args, "OO:poly_3d_calc", &veclist, &point)) {
|
|
return nullptr;
|
|
}
|
|
|
|
if (mathutils_array_parse(
|
|
fp, 2, 3 | MU_ARRAY_ZERO, point, "pt must be a 2-3 dimensional vector") == -1)
|
|
{
|
|
return nullptr;
|
|
}
|
|
|
|
len = mathutils_array_parse_alloc_v(((float **)&vecs), 3, veclist, __func__);
|
|
if (len == -1) {
|
|
return nullptr;
|
|
}
|
|
|
|
if (len) {
|
|
float *weights = static_cast<float *>(MEM_mallocN(sizeof(float) * len, __func__));
|
|
|
|
interp_weights_poly_v3(weights, vecs, len, fp);
|
|
|
|
ret = PyList_New(len);
|
|
for (i = 0; i < len; i++) {
|
|
PyList_SET_ITEM(ret, i, PyFloat_FromDouble(weights[i]));
|
|
}
|
|
|
|
MEM_freeN(weights);
|
|
|
|
PyMem_Free(vecs);
|
|
}
|
|
else {
|
|
ret = PyList_New(0);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
#endif /* !MATH_STANDALONE */
|
|
|
|
static PyMethodDef M_Interpolate_methods[] = {
|
|
#ifndef MATH_STANDALONE
|
|
{"poly_3d_calc",
|
|
(PyCFunction)M_Interpolate_poly_3d_calc,
|
|
METH_VARARGS,
|
|
M_Interpolate_poly_3d_calc_doc},
|
|
#endif
|
|
{nullptr, nullptr, 0, nullptr},
|
|
};
|
|
|
|
static PyModuleDef M_Interpolate_module_def = {
|
|
/*m_base*/ PyModuleDef_HEAD_INIT,
|
|
/*m_name*/ "mathutils.interpolate",
|
|
/*m_doc*/ M_Interpolate_doc,
|
|
/*m_size*/ 0,
|
|
/*m_methods*/ M_Interpolate_methods,
|
|
/*m_slots*/ nullptr,
|
|
/*m_traverse*/ nullptr,
|
|
/*m_clear*/ nullptr,
|
|
/*m_free*/ nullptr,
|
|
};
|
|
|
|
/*----------------------------MODULE INIT-------------------------*/
|
|
|
|
PyMODINIT_FUNC PyInit_mathutils_interpolate()
|
|
{
|
|
PyObject *submodule = PyModule_Create(&M_Interpolate_module_def);
|
|
return submodule;
|
|
}
|