Blender  V3.3
node_geo_edge_split.cc
Go to the documentation of this file.
1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 
3 #include "BKE_mesh.h"
4 #include "BKE_mesh_runtime.h"
5 
6 #include "bmesh.h"
7 #include "bmesh_tools.h"
8 
9 #include "node_geometry_util.hh"
10 
12 
14 {
15  b.add_input<decl::Geometry>(N_("Mesh")).supported_type(GEO_COMPONENT_TYPE_MESH);
16  b.add_input<decl::Bool>(N_("Selection")).default_value(true).hide_value().supports_field();
17  b.add_output<decl::Geometry>(N_("Mesh"));
18 }
19 
20 static Mesh *mesh_edge_split(const Mesh &mesh, const IndexMask selection)
21 {
22  BMeshCreateParams bmesh_create_params{};
23  bmesh_create_params.use_toolflags = true;
24  const BMAllocTemplate allocsize = {0, 0, 0, 0};
25  BMesh *bm = BM_mesh_create(&allocsize, &bmesh_create_params);
26 
27  BMeshFromMeshParams bmesh_from_mesh_params{};
28  bmesh_from_mesh_params.cd_mask_extra.vmask = CD_MASK_ORIGINDEX;
29  bmesh_from_mesh_params.cd_mask_extra.emask = CD_MASK_ORIGINDEX;
30  bmesh_from_mesh_params.cd_mask_extra.pmask = CD_MASK_ORIGINDEX;
31  BM_mesh_bm_from_me(bm, &mesh, &bmesh_from_mesh_params);
32 
34  for (const int i : selection) {
35  BMEdge *edge = BM_edge_at_index(bm, i);
37  }
38 
39  BM_mesh_edgesplit(bm, false, true, false);
40 
43 
44  return result;
45 }
46 
48 {
49  GeometrySet geometry_set = params.extract_input<GeometrySet>("Mesh");
50 
51  const Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");
52 
53  geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
54  if (!geometry_set.has_mesh()) {
55  return;
56  }
57 
58  const MeshComponent &mesh_component = *geometry_set.get_component_for_read<MeshComponent>();
59  GeometryComponentFieldContext field_context{mesh_component, ATTR_DOMAIN_EDGE};
60  const int domain_size = mesh_component.attribute_domain_size(ATTR_DOMAIN_EDGE);
61  fn::FieldEvaluator selection_evaluator{field_context, domain_size};
62  selection_evaluator.add(selection_field);
63  selection_evaluator.evaluate();
64  const IndexMask selection = selection_evaluator.get_evaluated_as_mask(0);
65 
66  geometry_set.replace_mesh(mesh_edge_split(*mesh_component.get_for_read(), selection));
67  });
68 
69  params.set_output("Mesh", std::move(geometry_set));
70 }
71 
72 } // namespace blender::nodes::node_geo_edge_split_cc
73 
75 {
76  namespace file_ns = blender::nodes::node_geo_edge_split_cc;
77 
78  static bNodeType ntype;
79 
83  nodeRegisterType(&ntype);
84 }
@ ATTR_DOMAIN_EDGE
Definition: BKE_attribute.h:28
@ GEO_COMPONENT_TYPE_MESH
struct Mesh * BKE_mesh_from_bmesh_for_eval_nomain(struct BMesh *bm, const struct CustomData_MeshMasks *cd_mask_extra, const struct Mesh *me_settings)
#define NODE_CLASS_GEOMETRY
Definition: BKE_node.h:359
#define GEO_NODE_SPLIT_EDGES
Definition: BKE_node.h:1465
void nodeRegisterType(struct bNodeType *ntype)
Definition: node.cc:1357
#define CD_MASK_ORIGINDEX
@ BM_EDGE
Definition: bmesh_class.h:384
@ BM_ELEM_TAG
Definition: bmesh_class.h:484
void BM_mesh_edgesplit(BMesh *bm, const bool use_verts, const bool tag_only, const bool copy_select)
#define BM_elem_flag_enable(ele, hflag)
Definition: bmesh_inline.h:14
ATTR_WARN_UNUSED_RESULT BMesh * bm
void BM_mesh_free(BMesh *bm)
BMesh Free Mesh.
Definition: bmesh_mesh.cc:258
BMesh * BM_mesh_create(const BMAllocTemplate *allocsize, const struct BMeshCreateParams *params)
Definition: bmesh_mesh.cc:125
void BM_mesh_elem_table_ensure(BMesh *bm, const char htype)
Definition: bmesh_mesh.cc:558
BLI_INLINE BMEdge * BM_edge_at_index(BMesh *bm, const int index)
Definition: bmesh_mesh.h:109
void BM_mesh_bm_from_me(BMesh *bm, const Mesh *me, const struct BMeshFromMeshParams *params)
int attribute_domain_size(eAttrDomain domain) const
Definition: geometry_set.cc:63
const Mesh * get_for_read() const
int add(GField field, GVArray *varray_ptr)
Definition: field.cc:731
uiWidgetBaseParameters params[MAX_WIDGET_BASE_BATCH]
static void node_geo_exec(GeoNodeExecParams params)
static void node_declare(NodeDeclarationBuilder &b)
static Mesh * mesh_edge_split(const Mesh &mesh, const IndexMask selection)
static const pxr::TfToken b("b", pxr::TfToken::Immortal)
void register_node_type_geo_edge_split()
void geo_node_type_base(bNodeType *ntype, int type, const char *name, short nclass)
struct CustomData_MeshMasks cd_mask_extra
void replace_mesh(Mesh *mesh, GeometryOwnershipType ownership=GeometryOwnershipType::Owned)
const GeometryComponent * get_component_for_read(GeometryComponentType component_type) const
void modify_geometry_sets(ForeachSubGeometryCallback callback)
bool has_mesh() const
Defines a node type.
Definition: BKE_node.h:226
NodeGeometryExecFunction geometry_node_execute
Definition: BKE_node.h:316
NodeDeclareFunction declare
Definition: BKE_node.h:324
#define N_(msgid)