Files
test2/source/blender/python/bmesh/bmesh_py_utils.cc

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

992 lines
30 KiB
C++
Raw Permalink Normal View History

/* SPDX-FileCopyrightText: 2012 Blender Authors
*
* SPDX-License-Identifier: GPL-2.0-or-later */
/** \file
* \ingroup pybmesh
*
* This file defines the 'bmesh.utils' module.
* Utility functions for operating on 'bmesh.types'
*/
#include <Python.h>
#include "BLI_math_base.h"
#include "BLI_utildefines.h"
#include "MEM_guardedalloc.h"
#include "../mathutils/mathutils.hh"
#include "bmesh.hh"
#include "bmesh_py_types.hh"
#include "bmesh_py_utils.hh" /* own include */
#include "../generic/py_capi_utils.hh"
#include "../generic/python_compat.hh"
#include "../generic/python_utildefines.hh"
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_vert_collapse_edge_doc,
".. method:: vert_collapse_edge(vert, edge)\n"
"\n"
" Collapse a vertex into an edge.\n"
"\n"
" :arg vert: The vert that will be collapsed.\n"
" :type vert: :class:`bmesh.types.BMVert`\n"
" :arg edge: The edge to collapse into.\n"
" :type edge: :class:`bmesh.types.BMEdge`\n"
" :return: The resulting edge from the collapse operation.\n"
" :rtype: :class:`bmesh.types.BMEdge`\n");
static PyObject *bpy_bm_utils_vert_collapse_edge(PyObject * /*self*/, PyObject *args)
{
BPy_BMEdge *py_edge;
BPy_BMVert *py_vert;
BMesh *bm;
BMEdge *e_new = nullptr;
if (!PyArg_ParseTuple(
args, "O!O!:vert_collapse_edge", &BPy_BMVert_Type, &py_vert, &BPy_BMEdge_Type, &py_edge))
{
return nullptr;
}
BPY_BM_CHECK_OBJ(py_edge);
BPY_BM_CHECK_OBJ(py_vert);
/* this doubles for checking that the verts are in the same mesh */
if (!(py_edge->e->v1 == py_vert->v || py_edge->e->v2 == py_vert->v)) {
PyErr_SetString(PyExc_ValueError,
"vert_collapse_edge(vert, edge): the vertex is not found in the edge");
return nullptr;
}
if (BM_vert_edge_count_is_over(py_vert->v, 2)) {
PyErr_SetString(PyExc_ValueError,
"vert_collapse_edge(vert, edge): vert has more than 2 connected edges");
return nullptr;
}
bm = py_edge->bm;
e_new = BM_vert_collapse_edge(bm, py_edge->e, py_vert->v, true, true, true);
if (e_new) {
return BPy_BMEdge_CreatePyObject(bm, e_new);
}
PyErr_SetString(PyExc_ValueError,
"vert_collapse_edge(vert, edge): no new edge created, internal error");
return nullptr;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_vert_collapse_faces_doc,
".. method:: vert_collapse_faces(vert, edge, fac, join_faces)\n"
"\n"
" Collapses a vertex that has only two manifold edges onto a vertex it shares an "
"edge with.\n"
"\n"
" :arg vert: The vert that will be collapsed.\n"
" :type vert: :class:`bmesh.types.BMVert`\n"
" :arg edge: The edge to collapse into.\n"
" :type edge: :class:`bmesh.types.BMEdge`\n"
" :arg fac: The factor to use when merging customdata [0 - 1].\n"
" :type fac: float\n"
" :arg join_faces: When true the faces around the vertex will be joined otherwise "
"collapse the vertex by merging the 2 edges this vertex connects to into one.\n"
" :type join_faces: bool\n"
" :return: The resulting edge from the collapse operation.\n"
" :rtype: :class:`bmesh.types.BMEdge`\n");
static PyObject *bpy_bm_utils_vert_collapse_faces(PyObject * /*self*/, PyObject *args)
{
BPy_BMEdge *py_edge;
BPy_BMVert *py_vert;
float fac;
int do_join_faces;
BMesh *bm;
BMEdge *e_new = nullptr;
if (!PyArg_ParseTuple(args,
"O!O!fi:vert_collapse_faces",
&BPy_BMVert_Type,
&py_vert,
&BPy_BMEdge_Type,
&py_edge,
&fac,
&do_join_faces))
{
return nullptr;
}
BPY_BM_CHECK_OBJ(py_edge);
BPY_BM_CHECK_OBJ(py_vert);
/* this doubles for checking that the verts are in the same mesh */
if (!(py_edge->e->v1 == py_vert->v || py_edge->e->v2 == py_vert->v)) {
PyErr_SetString(PyExc_ValueError,
"vert_collapse_faces(vert, edge): the vertex is not found in the edge");
return nullptr;
}
if (BM_vert_edge_count_is_over(py_vert->v, 2)) {
PyErr_SetString(PyExc_ValueError,
"vert_collapse_faces(vert, edge): vert has more than 2 connected edges");
return nullptr;
}
bm = py_edge->bm;
e_new = BM_vert_collapse_faces(
bm, py_edge->e, py_vert->v, clamp_f(fac, 0.0f, 1.0f), true, do_join_faces, true, true);
if (e_new) {
return BPy_BMEdge_CreatePyObject(bm, e_new);
}
PyErr_SetString(PyExc_ValueError,
"vert_collapse_faces(vert, edge): no new edge created, internal error");
return nullptr;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_vert_dissolve_doc,
".. method:: vert_dissolve(vert)\n"
"\n"
" Dissolve this vertex (will be removed).\n"
"\n"
" :arg vert: The vert to be dissolved.\n"
" :type vert: :class:`bmesh.types.BMVert`\n"
" :return: True when the vertex dissolve is successful.\n"
" :rtype: bool\n");
static PyObject *bpy_bm_utils_vert_dissolve(PyObject * /*self*/, PyObject *args)
{
BPy_BMVert *py_vert;
BMesh *bm;
if (!PyArg_ParseTuple(args, "O!:vert_dissolve", &BPy_BMVert_Type, &py_vert)) {
return nullptr;
}
BPY_BM_CHECK_OBJ(py_vert);
bm = py_vert->bm;
2021-07-16 11:45:52 +10:00
return PyBool_FromLong(BM_vert_dissolve(bm, py_vert->v));
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_vert_splice_doc,
".. method:: vert_splice(vert, vert_target)\n"
"\n"
" Splice vert into vert_target.\n"
"\n"
" :arg vert: The vertex to be removed.\n"
" :type vert: :class:`bmesh.types.BMVert`\n"
" :arg vert_target: The vertex to use.\n"
" :type vert_target: :class:`bmesh.types.BMVert`\n"
"\n"
" .. note:: The verts mustn't share an edge or face.\n");
static PyObject *bpy_bm_utils_vert_splice(PyObject * /*self*/, PyObject *args)
{
const char *error_prefix = "vert_splice(...)";
BPy_BMVert *py_vert;
BPy_BMVert *py_vert_target;
BMesh *bm;
bool ok;
if (!PyArg_ParseTuple(
args, "O!O!:vert_splice", &BPy_BMVert_Type, &py_vert, &BPy_BMVert_Type, &py_vert_target))
{
return nullptr;
}
BPY_BM_CHECK_OBJ(py_vert);
BPY_BM_CHECK_OBJ(py_vert_target);
bm = py_vert->bm;
BPY_BM_CHECK_SOURCE_OBJ(bm, error_prefix, py_vert_target);
if (py_vert->v == py_vert_target->v) {
PyErr_Format(PyExc_ValueError, "%s: vert arguments match", error_prefix);
return nullptr;
}
if (BM_edge_exists(py_vert->v, py_vert_target->v)) {
PyErr_Format(PyExc_ValueError, "%s: verts cannot share an edge", error_prefix);
return nullptr;
}
if (BM_vert_pair_share_face_check(py_vert->v, py_vert_target->v)) {
PyErr_Format(PyExc_ValueError, "%s: verts cannot share a face", error_prefix);
return nullptr;
}
/* should always succeed */
ok = BM_vert_splice(bm, py_vert_target->v, py_vert->v);
BLI_assert(ok == true);
UNUSED_VARS_NDEBUG(ok);
Py_RETURN_NONE;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_vert_separate_doc,
".. method:: vert_separate(vert, edges)\n"
"\n"
" Separate this vertex at every edge.\n"
"\n"
" :arg vert: The vert to be separated.\n"
" :type vert: :class:`bmesh.types.BMVert`\n"
" :arg edges: The edges to separated.\n"
" :type edges: :class:`bmesh.types.BMEdge`\n"
" :return: The newly separated verts (including the vertex passed).\n"
" :rtype: tuple[:class:`bmesh.types.BMVert`, ...]\n");
static PyObject *bpy_bm_utils_vert_separate(PyObject * /*self*/, PyObject *args)
{
const char *error_prefix = "vert_separate(...)";
BPy_BMVert *py_vert;
PyObject *edge_seq;
BMesh *bm;
BMVert **elem;
int elem_len;
PyObject *ret;
if (!PyArg_ParseTuple(args, "O!O:vert_separate", &BPy_BMVert_Type, &py_vert, &edge_seq)) {
return nullptr;
}
BPY_BM_CHECK_OBJ(py_vert);
bm = py_vert->bm;
/* Edges to split. */
Py_ssize_t edge_array_num;
BMEdge **edge_array = BPy_BMEdge_PySeq_As_Array(
&bm, edge_seq, 0, PY_SSIZE_T_MAX, &edge_array_num, true, true, error_prefix);
if (edge_array == nullptr) {
return nullptr;
}
BM_vert_separate(bm, py_vert->v, edge_array, edge_array_num, false, &elem, &elem_len);
/* return collected verts */
ret = BPy_BMVert_Array_As_Tuple(bm, elem, elem_len);
MEM_freeN(elem);
PyMem_FREE(edge_array);
return ret;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_edge_split_doc,
".. method:: edge_split(edge, vert, fac)\n"
"\n"
" Split an edge, return the newly created data.\n"
"\n"
" :arg edge: The edge to split.\n"
" :type edge: :class:`bmesh.types.BMEdge`\n"
" :arg vert: One of the verts on the edge, defines the split direction.\n"
" :type vert: :class:`bmesh.types.BMVert`\n"
" :arg fac: The point on the edge where the new vert will be created [0 - 1].\n"
" :type fac: float\n"
" :return: The newly created (edge, vert) pair.\n"
" :rtype: tuple[:class:`bmesh.types.BMEdge`, :class:`bmesh.types.BMVert`]\n");
static PyObject *bpy_bm_utils_edge_split(PyObject * /*self*/, PyObject *args)
{
BPy_BMEdge *py_edge;
BPy_BMVert *py_vert;
float fac;
BMesh *bm;
BMVert *v_new = nullptr;
BMEdge *e_new = nullptr;
if (!PyArg_ParseTuple(
args, "O!O!f:edge_split", &BPy_BMEdge_Type, &py_edge, &BPy_BMVert_Type, &py_vert, &fac))
{
return nullptr;
}
BPY_BM_CHECK_OBJ(py_edge);
BPY_BM_CHECK_OBJ(py_vert);
/* this doubles for checking that the verts are in the same mesh */
if (!(py_edge->e->v1 == py_vert->v || py_edge->e->v2 == py_vert->v)) {
PyErr_SetString(PyExc_ValueError,
"edge_split(edge, vert): the vertex is not found in the edge");
return nullptr;
}
bm = py_edge->bm;
v_new = BM_edge_split(bm, py_edge->e, py_vert->v, &e_new, clamp_f(fac, 0.0f, 1.0f));
if (v_new && e_new) {
PyObject *ret = PyTuple_New(2);
PyTuple_SET_ITEMS(
ret, BPy_BMEdge_CreatePyObject(bm, e_new), BPy_BMVert_CreatePyObject(bm, v_new));
return ret;
}
PyErr_SetString(PyExc_ValueError,
"edge_split(edge, vert): couldn't split the edge, internal error");
return nullptr;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_edge_rotate_doc,
".. method:: edge_rotate(edge, ccw=False)\n"
"\n"
" Rotate the edge and return the newly created edge.\n"
" If rotating the edge fails, None will be returned.\n"
"\n"
" :arg edge: The edge to rotate.\n"
" :type edge: :class:`bmesh.types.BMEdge`\n"
" :arg ccw: When True the edge will be rotated counter clockwise.\n"
" :type ccw: bool\n"
" :return: The newly rotated edge.\n"
" :rtype: :class:`bmesh.types.BMEdge`\n");
static PyObject *bpy_bm_utils_edge_rotate(PyObject * /*self*/, PyObject *args)
{
BPy_BMEdge *py_edge;
bool do_ccw = false;
BMesh *bm;
BMEdge *e_new = nullptr;
if (!PyArg_ParseTuple(
args, "O!|O&:edge_rotate", &BPy_BMEdge_Type, &py_edge, PyC_ParseBool, &do_ccw))
{
return nullptr;
}
BPY_BM_CHECK_OBJ(py_edge);
bm = py_edge->bm;
e_new = BM_edge_rotate(bm, py_edge->e, do_ccw, 0);
if (e_new) {
return BPy_BMEdge_CreatePyObject(bm, e_new);
}
Py_RETURN_NONE;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_face_split_doc,
".. method:: face_split(face, vert_a, vert_b, *, coords=(), use_exist=True, example=None)\n"
"\n"
" Face split with optional intermediate points.\n"
"\n"
" :arg face: The face to cut.\n"
" :type face: :class:`bmesh.types.BMFace`\n"
" :arg vert_a: First vertex to cut in the face (face must contain the vert).\n"
" :type vert_a: :class:`bmesh.types.BMVert`\n"
" :arg vert_b: Second vertex to cut in the face (face must contain the vert).\n"
" :type vert_b: :class:`bmesh.types.BMVert`\n"
" :arg coords: Optional sequence of 3D points in between *vert_a* and *vert_b*.\n"
" :type coords: Sequence[Sequence[float]]\n"
" :arg use_exist: .Use an existing edge if it exists (Only used when *coords* argument is "
"empty or omitted)\n"
" :type use_exist: bool\n"
" :arg example: Newly created edge will copy settings from this one.\n"
" :type example: :class:`bmesh.types.BMEdge`\n"
" :return: The newly created face or None on failure.\n"
" :rtype: tuple[:class:`bmesh.types.BMFace`, :class:`bmesh.types.BMLoop`]\n");
static PyObject *bpy_bm_utils_face_split(PyObject * /*self*/, PyObject *args, PyObject *kw)
{
static const char *kwlist[] = {
"face", "vert_a", "vert_b", "coords", "use_exist", "example", nullptr};
BPy_BMFace *py_face;
BPy_BMVert *py_vert_a;
BPy_BMVert *py_vert_b;
/* optional */
PyObject *py_coords = nullptr;
bool edge_exists = true;
BPy_BMEdge *py_edge_example = nullptr;
float *coords;
int ncoords = 0;
BMesh *bm;
BMFace *f_new = nullptr;
BMLoop *l_new = nullptr;
BMLoop *l_a, *l_b;
if (!PyArg_ParseTupleAndKeywords(args,
kw,
PyAPI: use keyword only arguments Use keyword only arguments for the following functions. - addon_utils.module_bl_info 2nd arg `info_basis`. - addon_utils.modules 1st `module_cache`, 2nd arg `refresh`. - addon_utils.modules_refresh 1st arg `module_cache`. - bl_app_template_utils.activate 1nd arg `template_id`. - bl_app_template_utils.import_from_id 2nd arg `ignore_not_found`. - bl_app_template_utils.import_from_path 2nd arg `ignore_not_found`. - bl_keymap_utils.keymap_from_toolbar.generate 2nd & 3rd args `use_fallback_keys` & `use_reset`. - bl_keymap_utils.platform_helpers.keyconfig_data_oskey_from_ctrl 2nd arg `filter_fn`. - bl_ui_utils.bug_report_url.url_prefill_from_blender 1st arg `addon_info`. - bmesh.types.BMFace.copy 1st & 2nd args `verts`, `edges`. - bmesh.types.BMesh.calc_volume 1st arg `signed`. - bmesh.types.BMesh.from_mesh 2nd..4th args `face_normals`, `use_shape_key`, `shape_key_index`. - bmesh.types.BMesh.from_object 3rd & 4th args `cage`, `face_normals`. - bmesh.types.BMesh.transform 2nd arg `filter`. - bmesh.types.BMesh.update_edit_mesh 2nd & 3rd args `loop_triangles`, `destructive`. - bmesh.types.{BMVertSeq,BMEdgeSeq,BMFaceSeq}.sort 1st & 2nd arg `key`, `reverse`. - bmesh.utils.face_split 4th..6th args `coords`, `use_exist`, `example`. - bpy.data.libraries.load 2nd..4th args `link`, `relative`, `assets_only`. - bpy.data.user_map 1st..3rd args `subset`, `key_types, `value_types`. - bpy.msgbus.subscribe_rna 5th arg `options`. - bpy.path.abspath 2nd & 3rd args `start` & `library`. - bpy.path.clean_name 2nd arg `replace`. - bpy.path.ensure_ext 3rd arg `case_sensitive`. - bpy.path.module_names 2nd arg `recursive`. - bpy.path.relpath 2nd arg `start`. - bpy.types.EditBone.transform 2nd & 3rd arg `scale`, `roll`. - bpy.types.Operator.as_keywords 1st arg `ignore`. - bpy.types.Struct.{keyframe_insert,keyframe_delete} 2nd..5th args `index`, `frame`, `group`, `options`. - bpy.types.WindowManager.popup_menu 2nd & 3rd arg `title`, `icon`. - bpy.types.WindowManager.popup_menu_pie 3rd & 4th arg `title`, `icon`. - bpy.utils.app_template_paths 1st arg `subdir`. - bpy.utils.app_template_paths 1st arg `subdir`. - bpy.utils.blend_paths 1st..3rd args `absolute`, `packed`, `local`. - bpy.utils.execfile 2nd arg `mod`. - bpy.utils.keyconfig_set 2nd arg `report`. - bpy.utils.load_scripts 1st & 2nd `reload_scripts` & `refresh_scripts`. - bpy.utils.preset_find 3rd & 4th args `display_name`, `ext`. - bpy.utils.resource_path 2nd & 3rd arg `major`, `minor`. - bpy.utils.script_paths 1st..4th args `subdir`, `user_pref`, `check_all`, `use_user`. - bpy.utils.smpte_from_frame 2nd & 3rd args `fps`, `fps_base`. - bpy.utils.smpte_from_seconds 2nd & 3rd args `fps`, `fps_base`. - bpy.utils.system_resource 2nd arg `subdir`. - bpy.utils.time_from_frame 2nd & 3rd args `fps`, `fps_base`. - bpy.utils.time_to_frame 2nd & 3rd args `fps`, `fps_base`. - bpy.utils.units.to_string 4th..6th `precision`, `split_unit`, `compatible_unit`. - bpy.utils.units.to_value 4th arg `str_ref_unit`. - bpy.utils.user_resource 2nd & 3rd args `subdir`, `create` - bpy_extras.view3d_utils.location_3d_to_region_2d 4th arg `default`. - bpy_extras.view3d_utils.region_2d_to_origin_3d 4th arg `clamp`. - gpu.offscreen.unbind 1st arg `restore`. - gpu_extras.batch.batch_for_shader 4th arg `indices`. - gpu_extras.batch.presets.draw_circle_2d 4th arg `segments`. - gpu_extras.presets.draw_circle_2d 4th arg `segments`. - imbuf.types.ImBuf.resize 2nd arg `resize`. - imbuf.write 2nd arg `filepath`. - mathutils.kdtree.KDTree.find 2nd arg `filter`. - nodeitems_utils.NodeCategory 3rd & 4th arg `descriptions`, `items`. - nodeitems_utils.NodeItem 2nd..4th args `label`, `settings`, `poll`. - nodeitems_utils.NodeItemCustom 1st & 2nd arg `poll`, `draw`. - rna_prop_ui.draw 5th arg `use_edit`. - rna_prop_ui.rna_idprop_ui_get 2nd arg `create`. - rna_prop_ui.rna_idprop_ui_prop_clear 3rd arg `remove`. - rna_prop_ui.rna_idprop_ui_prop_get 3rd arg `create`. - rna_xml.xml2rna 2nd arg `root_rna`. - rna_xml.xml_file_write 4th arg `skip_typemap`.
2021-06-08 18:03:14 +10:00
"O!O!O!|$OO&O!:face_split",
(char **)kwlist,
&BPy_BMFace_Type,
&py_face,
&BPy_BMVert_Type,
&py_vert_a,
&BPy_BMVert_Type,
&py_vert_b,
&py_coords,
PyC_ParseBool,
&edge_exists,
&BPy_BMEdge_Type,
&py_edge_example))
{
return nullptr;
}
BPY_BM_CHECK_OBJ(py_face);
BPY_BM_CHECK_OBJ(py_vert_a);
BPY_BM_CHECK_OBJ(py_vert_b);
if (py_edge_example) {
BPY_BM_CHECK_OBJ(py_edge_example);
}
/* this doubles for checking that the verts are in the same mesh */
if ((l_a = BM_face_vert_share_loop(py_face->f, py_vert_a->v)) &&
(l_b = BM_face_vert_share_loop(py_face->f, py_vert_b->v)))
{
/* pass */
}
else {
PyErr_SetString(PyExc_ValueError,
"face_split(...): one of the verts passed is not found in the face");
return nullptr;
}
if (py_vert_a->v == py_vert_b->v) {
PyErr_SetString(PyExc_ValueError, "face_split(...): vert arguments must differ");
return nullptr;
}
if (py_coords) {
ncoords = mathutils_array_parse_alloc_v(&coords, 3, py_coords, "face_split(...): ");
if (ncoords == -1) {
return nullptr;
}
}
else {
if (BM_loop_is_adjacent(l_a, l_b)) {
PyErr_SetString(PyExc_ValueError, "face_split(...): verts are adjacent in the face");
return nullptr;
}
}
/* --- main function body --- */
bm = py_face->bm;
if (ncoords) {
f_new = BM_face_split_n(bm,
py_face->f,
l_a,
l_b,
(float (*)[3])coords,
ncoords,
&l_new,
py_edge_example ? py_edge_example->e : nullptr);
PyMem_Free(coords);
}
else {
f_new = BM_face_split(bm,
py_face->f,
l_a,
l_b,
&l_new,
py_edge_example ? py_edge_example->e : nullptr,
edge_exists);
}
if (f_new && l_new) {
PyObject *ret = PyTuple_New(2);
PyTuple_SET_ITEMS(
ret, BPy_BMFace_CreatePyObject(bm, f_new), BPy_BMLoop_CreatePyObject(bm, l_new));
return ret;
}
PyErr_SetString(PyExc_ValueError, "face_split(...): couldn't split the face, internal error");
return nullptr;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_face_split_edgenet_doc,
".. method:: face_split_edgenet(face, edgenet)\n"
"\n"
" Splits a face into any number of regions defined by an edgenet.\n"
"\n"
" :arg face: The face to split.\n"
" :type face: :class:`bmesh.types.BMFace`\n"
" :arg face: The face to split.\n"
" :type face: :class:`bmesh.types.BMFace`\n"
" :arg edgenet: Sequence of edges.\n"
" :type edgenet: Sequence[:class:`bmesh.types.BMEdge`]\n"
" :return: The newly created faces.\n"
" :rtype: tuple[:class:`bmesh.types.BMFace`, ...]\n"
"\n"
" .. note::\n"
"\n"
" Regions defined by edges need to connect to the face, otherwise they're "
"ignored as loose edges.\n");
static PyObject *bpy_bm_utils_face_split_edgenet(PyObject * /*self*/, PyObject *args, PyObject *kw)
{
const char *error_prefix = "face_split_edgenet(...)";
static const char *kwlist[] = {"face", "edgenet", nullptr};
BPy_BMFace *py_face;
PyObject *edge_seq;
BMesh *bm;
bool ok;
if (!PyArg_ParseTupleAndKeywords(args,
kw,
"O!O:face_split_edgenet",
(char **)kwlist,
&BPy_BMFace_Type,
&py_face,
&edge_seq))
{
return nullptr;
}
BPY_BM_CHECK_OBJ(py_face);
bm = py_face->bm;
Py_ssize_t edge_array_num;
BMEdge **edge_array = BPy_BMEdge_PySeq_As_Array(
&bm, edge_seq, 1, PY_SSIZE_T_MAX, &edge_array_num, true, true, error_prefix);
if (edge_array == nullptr) {
return nullptr;
}
/* --- main function body --- */
blender::Vector<BMFace *> face_arr;
ok = BM_face_split_edgenet(bm, py_face->f, edge_array, edge_array_num, &face_arr);
PyMem_FREE(edge_array);
if (ok) {
PyObject *ret = BPy_BMFace_Array_As_Tuple(bm, face_arr.data(), face_arr.size());
return ret;
}
PyErr_SetString(PyExc_ValueError,
"face_split_edgenet(...): couldn't split the face, internal error");
return nullptr;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_face_join_doc,
".. method:: face_join(faces, remove=True)\n"
"\n"
" Joins a sequence of faces.\n"
"\n"
" :arg faces: Sequence of faces.\n"
" :type faces: :class:`bmesh.types.BMFace`\n"
" :arg remove: Remove the edges and vertices between the faces.\n"
" :type remove: bool\n"
" :return: The newly created face or None on failure.\n"
" :rtype: :class:`bmesh.types.BMFace`\n");
static PyObject *bpy_bm_utils_face_join(PyObject * /*self*/, PyObject *args)
{
const char *error_prefix = "face_join(...)";
BMesh *bm = nullptr;
PyObject *py_face_array;
BMFace *f_new;
bool do_remove = true;
if (!PyArg_ParseTuple(args, "O|O&:face_join", &py_face_array, PyC_ParseBool, &do_remove)) {
return nullptr;
}
Py_ssize_t face_seq_len = 0;
BMFace **face_array = BPy_BMFace_PySeq_As_Array(
&bm, py_face_array, 2, PY_SSIZE_T_MAX, &face_seq_len, true, true, error_prefix);
if (face_array == nullptr) {
return nullptr; /* error will be set */
}
/* Go ahead and join the face!
* --------------------------- */
BMFace *f_double;
f_new = BM_faces_join(bm, face_array, int(face_seq_len), do_remove, &f_double);
/* See #BM_faces_join note on callers asserting when `r_double` is non-null. */
BLI_assert_msg(f_double == nullptr,
"Doubled face detected at " AT ". Resulting mesh may be corrupt.");
PyMem_FREE(face_array);
if (f_new) {
return BPy_BMFace_CreatePyObject(bm, f_new);
}
Py_RETURN_NONE;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_face_vert_separate_doc,
".. method:: face_vert_separate(face, vert)\n"
"\n"
" Rip a vertex in a face away and add a new vertex.\n"
"\n"
" :arg face: The face to separate.\n"
" :type face: :class:`bmesh.types.BMFace`\n"
" :arg vert: A vertex in the face to separate.\n"
" :type vert: :class:`bmesh.types.BMVert`\n"
2015-04-30 09:55:54 +10:00
" :return vert: The newly created vertex or None on failure.\n"
" :rtype vert: :class:`bmesh.types.BMVert`\n"
"\n"
" .. note::\n"
"\n"
" This is the same as loop_separate, and has only been added for convenience.\n");
static PyObject *bpy_bm_utils_face_vert_separate(PyObject * /*self*/, PyObject *args)
{
const char *error_prefix = "face_vert_separate()";
BPy_BMFace *py_face;
BPy_BMVert *py_vert;
BMesh *bm;
BMLoop *l;
BMVert *v_old, *v_new;
if (!PyArg_ParseTuple(
args, "O!O!:face_vert_separate", &BPy_BMFace_Type, &py_face, &BPy_BMVert_Type, &py_vert))
{
return nullptr;
}
bm = py_face->bm;
BPY_BM_CHECK_OBJ(py_face);
BPY_BM_CHECK_SOURCE_OBJ(bm, error_prefix, py_vert);
l = BM_face_vert_share_loop(py_face->f, py_vert->v);
if (l == nullptr) {
PyErr_Format(PyExc_ValueError, "%s: vertex not found in face", error_prefix);
return nullptr;
}
v_old = l->v;
v_new = BM_face_loop_separate(bm, l);
if (v_new != v_old) {
return BPy_BMVert_CreatePyObject(bm, v_new);
}
Py_RETURN_NONE;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_face_flip_doc,
".. method:: face_flip(faces)\n"
"\n"
" Flip the faces direction.\n"
"\n"
" :arg face: Face to flip.\n"
" :type face: :class:`bmesh.types.BMFace`\n");
static PyObject *bpy_bm_utils_face_flip(PyObject * /*self*/, BPy_BMFace *value)
2012-02-24 11:04:09 +00:00
{
if (!BPy_BMFace_Check(value)) {
PyErr_Format(PyExc_TypeError,
"face_flip(face): BMFace expected, not '%.200s'",
Py_TYPE(value)->tp_name);
return nullptr;
2012-02-24 11:04:09 +00:00
}
BPY_BM_CHECK_OBJ(value);
BM_face_normal_flip(value->bm, value->f);
Py_RETURN_NONE;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_loop_separate_doc,
".. method:: loop_separate(loop)\n"
"\n"
" Rip a vertex in a face away and add a new vertex.\n"
"\n"
" :arg loop: The loop to separate.\n"
" :type loop: :class:`bmesh.types.BMLoop`\n"
" :return vert: The newly created vertex or None on failure.\n"
" :rtype vert: :class:`bmesh.types.BMVert`\n");
static PyObject *bpy_bm_utils_loop_separate(PyObject * /*self*/, BPy_BMLoop *value)
{
BMesh *bm;
BMLoop *l;
BMVert *v_old, *v_new;
if (!BPy_BMLoop_Check(value)) {
PyErr_Format(PyExc_TypeError,
"loop_separate(loop): BMLoop expected, not '%.200s'",
Py_TYPE(value)->tp_name);
return nullptr;
}
BPY_BM_CHECK_OBJ(value);
bm = value->bm;
l = value->l;
v_old = l->v;
v_new = BM_face_loop_separate(bm, l);
if (v_new != v_old) {
return BPy_BMVert_CreatePyObject(bm, v_new);
}
Py_RETURN_NONE;
}
PyDoc_STRVAR(
/* Wrap. */
bpy_bm_utils_uv_select_check_doc,
".. method:: uv_select_check(bm, /, *, sync=True, flush=False, contiguous=False)\n"
"\n"
" Split an edge, return the newly created data.\n"
"\n"
" :arg sync: Check the data is properly synchronized between UV's and the underlying mesh. "
"Failure to synchronize with the mesh selection may cause tools not to behave properly.\n"
" :type sync: bool\n"
" :arg flush: Check the selection has been properly flushed between elements "
"(based on the current :class:`BMesh.select_mode`).\n"
" :type flush: bool\n"
" :arg contiguous: Check connected UV's and edges have a matching selection state.\n"
" :type contiguous: bool\n"
" :return: An error dictionary or None when there are no errors found.\n"
" :rtype: dict[str, int] | None\n");
static PyObject *bpy_bm_utils_uv_select_check(PyObject * /*self*/, PyObject *args, PyObject *kwds)
{
const char *error_prefix = "uv_select_check(...)";
BPy_BMesh *py_bm;
bool check_sync = true;
bool check_contiguous = false;
bool check_flush = false;
static const char *_keywords[] = {
"",
"sync",
"flush",
"contiguous",
nullptr,
};
static _PyArg_Parser _parser = {
PY_ARG_PARSER_HEAD_COMPAT()
"O!" /* `bm` */
"|$" /* Optional keyword only arguments. */
"O&" /* `sync` */
"O&" /* `flush` */
"O&" /* `contiguous` */
":uv_select_check",
_keywords,
nullptr,
};
if (!_PyArg_ParseTupleAndKeywordsFast(args,
kwds,
&_parser,
&BPy_BMesh_Type,
&py_bm,
PyC_ParseBool,
&check_sync,
PyC_ParseBool,
&check_flush,
PyC_ParseBool,
&check_contiguous))
{
return nullptr;
}
BPY_BM_CHECK_OBJ(py_bm);
BMesh *bm = py_bm->bm;
if (check_sync) {
if (bpy_bm_check_uv_select_sync_valid(bm, error_prefix) == -1) {
return nullptr;
}
}
const int cd_loop_uv_offset = check_contiguous ?
CustomData_get_offset(&bm->ldata, CD_PROP_FLOAT2) :
-1;
if (check_contiguous) {
if (cd_loop_uv_offset == -1) {
PyErr_SetString(PyExc_ValueError, "contiguous=True for a mesh without UV coordinates");
return nullptr;
}
}
UVSelectValidateInfo info = {};
const bool is_valid = BM_mesh_uvselect_is_valid(
bm, cd_loop_uv_offset, check_sync, check_flush, check_contiguous, &info);
if (is_valid) {
Py_RETURN_NONE;
}
PyObject *result = PyDict_New();
#define DICT_ADD_INT_MEMBER(info_struct, member) \
PyDict_SetItemString(result, STRINGIFY(member), PyLong_FromLong(info_struct.member))
{
UVSelectValidateInfo_Sync &info_sub = info.sync;
DICT_ADD_INT_MEMBER(info_sub, count_uv_vert_any_selected_with_vert_unselected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_vert_none_selected_with_vert_selected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_edge_any_selected_with_edge_unselected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_edge_none_selected_with_edge_selected);
}
if (check_flush) {
UVSelectValidateInfo_Flush &info_sub = info.flush;
DICT_ADD_INT_MEMBER(info_sub, count_uv_edge_selected_with_any_verts_unselected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_edge_unselected_with_all_verts_selected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_face_selected_with_any_verts_unselected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_face_unselected_with_all_verts_selected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_face_selected_with_any_edges_unselected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_face_unselected_with_all_edges_selected);
}
if (check_contiguous) {
UVSelectValidateInfo_Contiguous &info_sub = info.contiguous;
DICT_ADD_INT_MEMBER(info_sub, count_uv_vert_non_contiguous_selected);
DICT_ADD_INT_MEMBER(info_sub, count_uv_edge_non_contiguous_selected);
}
if (check_flush && check_contiguous) {
UVSelectValidateInfo_FlushAndContiguous &info_sub = info.flush_contiguous;
DICT_ADD_INT_MEMBER(info_sub, count_uv_vert_isolated_in_edge_or_face_mode);
DICT_ADD_INT_MEMBER(info_sub, count_uv_vert_isolated_in_face_mode);
DICT_ADD_INT_MEMBER(info_sub, count_uv_edge_isolated_in_face_mode);
}
#undef DICT_ADD_INT_MEMBER
return result;
}
#ifdef __GNUC__
# ifdef __clang__
# pragma clang diagnostic push
# pragma clang diagnostic ignored "-Wcast-function-type"
# else
# pragma GCC diagnostic push
# pragma GCC diagnostic ignored "-Wcast-function-type"
# endif
#endif
static PyMethodDef BPy_BM_utils_methods[] = {
2012-08-26 11:35:43 +00:00
{"vert_collapse_edge",
(PyCFunction)bpy_bm_utils_vert_collapse_edge,
METH_VARARGS,
bpy_bm_utils_vert_collapse_edge_doc},
{"vert_collapse_faces",
(PyCFunction)bpy_bm_utils_vert_collapse_faces,
METH_VARARGS,
bpy_bm_utils_vert_collapse_faces_doc},
{"vert_dissolve",
(PyCFunction)bpy_bm_utils_vert_dissolve,
METH_VARARGS,
bpy_bm_utils_vert_dissolve_doc}, /* could use METH_O */
{"vert_splice",
(PyCFunction)bpy_bm_utils_vert_splice,
METH_VARARGS,
bpy_bm_utils_vert_splice_doc},
2012-08-26 11:35:43 +00:00
{"vert_separate",
(PyCFunction)bpy_bm_utils_vert_separate,
METH_VARARGS,
bpy_bm_utils_vert_separate_doc},
{"edge_split",
(PyCFunction)bpy_bm_utils_edge_split,
METH_VARARGS,
bpy_bm_utils_edge_split_doc},
{"edge_rotate",
(PyCFunction)bpy_bm_utils_edge_rotate,
METH_VARARGS,
bpy_bm_utils_edge_rotate_doc},
{"face_split",
(PyCFunction)bpy_bm_utils_face_split,
METH_VARARGS | METH_KEYWORDS,
bpy_bm_utils_face_split_doc},
{"face_split_edgenet",
(PyCFunction)bpy_bm_utils_face_split_edgenet,
2012-08-26 11:35:43 +00:00
METH_VARARGS | METH_KEYWORDS,
bpy_bm_utils_face_split_edgenet_doc},
2012-08-26 11:35:43 +00:00
{"face_join", (PyCFunction)bpy_bm_utils_face_join, METH_VARARGS, bpy_bm_utils_face_join_doc},
{"face_vert_separate",
(PyCFunction)bpy_bm_utils_face_vert_separate,
METH_VARARGS,
bpy_bm_utils_face_vert_separate_doc},
{"face_flip", (PyCFunction)bpy_bm_utils_face_flip, METH_O, bpy_bm_utils_face_flip_doc},
{"loop_separate",
(PyCFunction)bpy_bm_utils_loop_separate,
METH_O,
bpy_bm_utils_loop_separate_doc},
{"uv_select_check",
(PyCFunction)bpy_bm_utils_uv_select_check,
METH_VARARGS | METH_KEYWORDS,
bpy_bm_utils_uv_select_check_doc},
{nullptr, nullptr, 0, nullptr},
};
#ifdef __GNUC__
# ifdef __clang__
# pragma clang diagnostic pop
# else
# pragma GCC diagnostic pop
# endif
#endif
PyDoc_STRVAR(
/* Wrap. */
BPy_BM_utils_doc,
"This module provides access to blenders bmesh data structures.");
static PyModuleDef BPy_BM_utils_module_def = {
/*m_base*/ PyModuleDef_HEAD_INIT,
/*m_name*/ "bmesh.utils",
/*m_doc*/ BPy_BM_utils_doc,
/*m_size*/ 0,
/*m_methods*/ BPy_BM_utils_methods,
/*m_slots*/ nullptr,
/*m_traverse*/ nullptr,
/*m_clear*/ nullptr,
/*m_free*/ nullptr,
};
PyObject *BPyInit_bmesh_utils()
{
PyObject *submodule;
submodule = PyModule_Create(&BPy_BM_utils_module_def);
return submodule;
}