Page MenuHome
Differential D16858 Diff 59011 source/blender/nodes/geometry/nodes/node_geo_mesh_to_points.cc

Changeset View

Standalone View

View Options

source/blender/nodes/geometry/nodes/node_geo_mesh_to_points.cc

Show All 17 Lines
namespace blender::nodes::node_geo_mesh_to_points_cc { namespace blender::nodes::node_geo_mesh_to_points_cc {
NODE_STORAGE_FUNCS(NodeGeometryMeshToPoints) NODE_STORAGE_FUNCS(NodeGeometryMeshToPoints)
static void node_declare(NodeDeclarationBuilder &b) static void node_declare(NodeDeclarationBuilder &b)
{ {
b.add_input<decl::Geometry>(N_("Mesh")).supported_type(GEO_COMPONENT_TYPE_MESH); b.add_input<decl::Geometry>(N_("Mesh")).supported_type(GEO_COMPONENT_TYPE_MESH);
b.add_input<decl::Bool>(N_("Selection")).default_value(true).supports_field().hide_value(); b.add_input<decl::Bool>(N_("Selection")).default_value(true).field_on_all().hide_value();
b.add_input<decl::Vector>(N_("Position")).implicit_field(implicit_field_inputs::position); b.add_input<decl::Vector>(N_("Position")).implicit_field(implicit_field_inputs::position);
b.add_input<decl::Float>(N_("Radius")) b.add_input<decl::Float>(N_("Radius"))
.default_value(0.05f) .default_value(0.05f)
.min(0.0f) .min(0.0f)
.subtype(PROP_DISTANCE) .subtype(PROP_DISTANCE)
.supports_field(); .field_on_all();
b.add_output<decl::Geometry>(N_("Points")); b.add_output<decl::Geometry>(N_("Points")).propagate_all();
} }
static void node_layout(uiLayout *layout, bContext * /*C*/, PointerRNA *ptr) static void node_layout(uiLayout *layout, bContext * /*C*/, PointerRNA *ptr)
{ {
uiItemR(layout, ptr, "mode", 0, "", ICON_NONE); uiItemR(layout, ptr, "mode", 0, "", ICON_NONE);
} }
static void node_init(bNodeTree * /*tree*/, bNode *node) static void node_init(bNodeTree * /*tree*/, bNode *node)
{ {
NodeGeometryMeshToPoints *data = MEM_cnew<NodeGeometryMeshToPoints>(__func__); NodeGeometryMeshToPoints *data = MEM_cnew<NodeGeometryMeshToPoints>(__func__);
data->mode = GEO_NODE_MESH_TO_POINTS_VERTICES; data->mode = GEO_NODE_MESH_TO_POINTS_VERTICES;
node->storage = data; node->storage = data;
} }
static void geometry_set_mesh_to_points(GeometrySet &geometry_set, static void geometry_set_mesh_to_points(GeometrySet &geometry_set,
Field<float3> &position_field, Field<float3> &position_field,
Field<float> &radius_field, Field<float> &radius_field,
Field<bool> &selection_field, Field<bool> &selection_field,
const eAttrDomain domain) const eAttrDomain domain,
const AnonymousAttributePropagationInfo &propagation_info)
{ {
const Mesh *mesh = geometry_set.get_mesh_for_read(); const Mesh *mesh = geometry_set.get_mesh_for_read();
if (mesh == nullptr) { if (mesh == nullptr) {
geometry_set.remove_geometry_during_modify(); geometry_set.remove_geometry_during_modify();
return; return;
} }
const int domain_size = mesh->attributes().domain_size(domain); const int domain_size = mesh->attributes().domain_size(domain);
if (domain_size == 0) { if (domain_size == 0) {
Show All 21 Linesstatic void geometry_set_mesh_to_points(GeometrySet &geometry_set,
position.finish(); position.finish();
GSpanAttributeWriter radius = dst_attributes.lookup_or_add_for_write_only_span( GSpanAttributeWriter radius = dst_attributes.lookup_or_add_for_write_only_span(
"radius", ATTR_DOMAIN_POINT, CD_PROP_FLOAT); "radius", ATTR_DOMAIN_POINT, CD_PROP_FLOAT);
array_utils::gather(evaluator.get_evaluated(1), selection, radius.span); array_utils::gather(evaluator.get_evaluated(1), selection, radius.span);
radius.finish(); radius.finish();
Map<AttributeIDRef, AttributeKind> attributes; Map<AttributeIDRef, AttributeKind> attributes;
geometry_set.gather_attributes_for_propagation( geometry_set.gather_attributes_for_propagation({GEO_COMPONENT_TYPE_MESH},
{GEO_COMPONENT_TYPE_MESH}, GEO_COMPONENT_TYPE_POINT_CLOUD, false, attributes); GEO_COMPONENT_TYPE_POINT_CLOUD,
false,
propagation_info,
attributes);
attributes.remove("position"); attributes.remove("position");
const AttributeAccessor src_attributes = mesh->attributes(); const AttributeAccessor src_attributes = mesh->attributes();
for (Map<AttributeIDRef, AttributeKind>::Item entry : attributes.items()) { for (Map<AttributeIDRef, AttributeKind>::Item entry : attributes.items()) {
const AttributeIDRef attribute_id = entry.key; const AttributeIDRef attribute_id = entry.key;
const eCustomDataType data_type = entry.value.data_type; const eCustomDataType data_type = entry.value.data_type;
GVArray src = src_attributes.lookup_or_default(attribute_id, domain, data_type); GVArray src = src_attributes.lookup_or_default(attribute_id, domain, data_type);
Show All 23 Lines static fn::CustomMF_SI_SO<float, float> max_zero_fn(
fn::CustomMF_presets::AllSpanOrSingle()); fn::CustomMF_presets::AllSpanOrSingle());
auto max_zero_op = std::make_shared<FieldOperation>( auto max_zero_op = std::make_shared<FieldOperation>(
FieldOperation(max_zero_fn, {std::move(radius)})); FieldOperation(max_zero_fn, {std::move(radius)}));
Field<float> positive_radius(std::move(max_zero_op), 0); Field<float> positive_radius(std::move(max_zero_op), 0);
const NodeGeometryMeshToPoints &storage = node_storage(params.node()); const NodeGeometryMeshToPoints &storage = node_storage(params.node());
const GeometryNodeMeshToPointsMode mode = (GeometryNodeMeshToPointsMode)storage.mode; const GeometryNodeMeshToPointsMode mode = (GeometryNodeMeshToPointsMode)storage.mode;
const AnonymousAttributePropagationInfo &propagation_info = params.get_output_propagation_info(
"Points");
geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) { geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
switch (mode) { switch (mode) {
case GEO_NODE_MESH_TO_POINTS_VERTICES: case GEO_NODE_MESH_TO_POINTS_VERTICES:
geometry_set_mesh_to_points( geometry_set_mesh_to_points(geometry_set,
geometry_set, position, positive_radius, selection, ATTR_DOMAIN_POINT); position,
positive_radius,
selection,
ATTR_DOMAIN_POINT,
propagation_info);
break; break;
case GEO_NODE_MESH_TO_POINTS_EDGES: case GEO_NODE_MESH_TO_POINTS_EDGES:
geometry_set_mesh_to_points( geometry_set_mesh_to_points(geometry_set,
geometry_set, position, positive_radius, selection, ATTR_DOMAIN_EDGE); position,
positive_radius,
selection,
ATTR_DOMAIN_EDGE,
propagation_info);
break; break;
case GEO_NODE_MESH_TO_POINTS_FACES: case GEO_NODE_MESH_TO_POINTS_FACES:
geometry_set_mesh_to_points( geometry_set_mesh_to_points(geometry_set,
geometry_set, position, positive_radius, selection, ATTR_DOMAIN_FACE); position,
positive_radius,
selection,
ATTR_DOMAIN_FACE,
propagation_info);
break; break;
case GEO_NODE_MESH_TO_POINTS_CORNERS: case GEO_NODE_MESH_TO_POINTS_CORNERS:
geometry_set_mesh_to_points( geometry_set_mesh_to_points(geometry_set,
geometry_set, position, positive_radius, selection, ATTR_DOMAIN_CORNER); position,
positive_radius,
selection,
ATTR_DOMAIN_CORNER,
propagation_info);
break; break;
} }
}); });
params.set_output("Points", std::move(geometry_set)); params.set_output("Points", std::move(geometry_set));
} }
} // namespace blender::nodes::node_geo_mesh_to_points_cc } // namespace blender::nodes::node_geo_mesh_to_points_cc
Show All 16 Lines