Differential D13337 Diff 45169 source/blender/nodes/geometry/nodes/node_geo_set_point_radius.cc

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_set_point_radius.cc

Show All 12 Lines
* along with this program; if not, write to the Free Software Foundation,		* along with this program; if not, write to the Free Software Foundation,
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.		* Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/		*/

#include "node_geometry_util.hh"		#include "node_geometry_util.hh"

namespace blender::nodes::node_geo_set_point_radius_cc {		namespace blender::nodes::node_geo_set_point_radius_cc {

static void geo_node_set_point_radius_declare(NodeDeclarationBuilder &b)		static void node_declare(NodeDeclarationBuilder &b)
{		{
b.add_input<decl::Geometry>(N_("Points")).supported_type(GEO_COMPONENT_TYPE_POINT_CLOUD);		b.add_input<decl::Geometry>(N_("Points")).supported_type(GEO_COMPONENT_TYPE_POINT_CLOUD);
b.add_input<decl::Bool>(N_("Selection")).default_value(true).hide_value().supports_field();		b.add_input<decl::Bool>(N_("Selection")).default_value(true).hide_value().supports_field();
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)
.supports_field()		.supports_field()
.subtype(PROP_DISTANCE);		.subtype(PROP_DISTANCE);
Show All 18 Lines	static void set_radius_in_component(GeometryComponent &component,
OutputAttribute_Typed<float> radii = component.attribute_try_get_for_output_only<float>(		OutputAttribute_Typed<float> radii = component.attribute_try_get_for_output_only<float>(
"radius", ATTR_DOMAIN_POINT);		"radius", ATTR_DOMAIN_POINT);
fn::FieldEvaluator radii_evaluator{field_context, &selection};		fn::FieldEvaluator radii_evaluator{field_context, &selection};
radii_evaluator.add_with_destination(radius_field, radii.varray());		radii_evaluator.add_with_destination(radius_field, radii.varray());
radii_evaluator.evaluate();		radii_evaluator.evaluate();
radii.save();		radii.save();
}		}

static void geo_node_set_point_radius_exec(GeoNodeExecParams params)		static void node_geo_exec(GeoNodeExecParams params)
{		{
GeometrySet geometry_set = params.extract_input<GeometrySet>("Points");		GeometrySet geometry_set = params.extract_input<GeometrySet>("Points");
Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");		Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");
Field<float> radii_field = params.extract_input<Field<float>>("Radius");		Field<float> radii_field = params.extract_input<Field<float>>("Radius");

geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {		geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
if (geometry_set.has_pointcloud()) {		if (geometry_set.has_pointcloud()) {
set_radius_in_component(geometry_set.get_component_for_write<PointCloudComponent>(),		set_radius_in_component(geometry_set.get_component_for_write<PointCloudComponent>(),
Show All 10 Lines
void register_node_type_geo_set_point_radius()		void register_node_type_geo_set_point_radius()
{		{
namespace file_ns = blender::nodes::node_geo_set_point_radius_cc;		namespace file_ns = blender::nodes::node_geo_set_point_radius_cc;

static bNodeType ntype;		static bNodeType ntype;

geo_node_type_base(		geo_node_type_base(
&ntype, GEO_NODE_SET_POINT_RADIUS, "Set Point Radius", NODE_CLASS_GEOMETRY, 0);		&ntype, GEO_NODE_SET_POINT_RADIUS, "Set Point Radius", NODE_CLASS_GEOMETRY, 0);
ntype.geometry_node_execute = file_ns::geo_node_set_point_radius_exec;		ntype.geometry_node_execute = file_ns::node_geo_exec;
ntype.declare = file_ns::geo_node_set_point_radius_declare;		ntype.declare = file_ns::node_declare;
nodeRegisterType(&ntype);		nodeRegisterType(&ntype);
}		}