Blender  V3.3
node_geo_set_point_radius.cc
Go to the documentation of this file.
1 /* SPDX-License-Identifier: GPL-2.0-or-later */
2 
3 #include "node_geometry_util.hh"
4 
6 
8 {
9  b.add_input<decl::Geometry>(N_("Points")).supported_type(GEO_COMPONENT_TYPE_POINT_CLOUD);
10  b.add_input<decl::Bool>(N_("Selection")).default_value(true).hide_value().supports_field();
11  b.add_input<decl::Float>(N_("Radius"))
12  .default_value(0.05f)
13  .min(0.0f)
14  .supports_field()
15  .subtype(PROP_DISTANCE);
16  b.add_output<decl::Geometry>(N_("Points"));
17 }
18 
20  const Field<bool> &selection_field,
21  const Field<float> &radius_field)
22 {
23  const int domain_size = component.attribute_domain_size(ATTR_DOMAIN_POINT);
24  if (domain_size == 0) {
25  return;
26  }
27  MutableAttributeAccessor attributes = *component.attributes_for_write();
29 
30  AttributeWriter<float> radii = attributes.lookup_or_add_for_write<float>("radius",
32 
33  fn::FieldEvaluator evaluator{field_context, domain_size};
34  evaluator.set_selection(selection_field);
35  evaluator.add_with_destination(radius_field, radii.varray);
36  evaluator.evaluate();
37 
38  radii.finish();
39 }
40 
42 {
43  GeometrySet geometry_set = params.extract_input<GeometrySet>("Points");
44  Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");
45  Field<float> radii_field = params.extract_input<Field<float>>("Radius");
46 
47  geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
48  if (geometry_set.has_pointcloud()) {
50  selection_field,
51  radii_field);
52  }
53  });
54 
55  params.set_output("Points", std::move(geometry_set));
56 }
57 
58 } // namespace blender::nodes::node_geo_set_point_radius_cc
59 
61 {
63 
64  static bNodeType ntype;
65 
69  nodeRegisterType(&ntype);
70 }
@ ATTR_DOMAIN_POINT
Definition: BKE_attribute.h:27
@ GEO_COMPONENT_TYPE_POINT_CLOUD
#define GEO_NODE_SET_POINT_RADIUS
Definition: BKE_node.h:1459
#define NODE_CLASS_GEOMETRY
Definition: BKE_node.h:359
void nodeRegisterType(struct bNodeType *ntype)
Definition: node.cc:1357
static uint8 component(Color32 c, uint i)
Definition: ColorBlock.cpp:108
@ PROP_DISTANCE
Definition: RNA_types.h:149
GAttributeWriter lookup_or_add_for_write(const AttributeIDRef &attribute_id, const eAttrDomain domain, const eCustomDataType data_type, const AttributeInit &initializer=AttributeInitDefault())
void set_selection(Field< bool > selection)
Definition: FN_field.hh:366
uiWidgetBaseParameters params[MAX_WIDGET_BASE_BATCH]
static void node_geo_exec(GeoNodeExecParams params)
static void node_declare(NodeDeclarationBuilder &b)
static void set_radius_in_component(GeometryComponent &component, const Field< bool > &selection_field, const Field< float > &radius_field)
static const pxr::TfToken b("b", pxr::TfToken::Immortal)
MutableSpan< float > radii
void register_node_type_geo_set_point_radius()
void geo_node_type_base(bNodeType *ntype, int type, const char *name, short nclass)
GeometryComponent & get_component_for_write(GeometryComponentType component_type)
void modify_geometry_sets(ForeachSubGeometryCallback callback)
bool has_pointcloud() 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)