Differential D13337 Diff 45164 source/blender/nodes/geometry/nodes/node_geo_set_spline_cyclic.cc

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_set_spline_cyclic.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_spline_cyclic_cc {		namespace blender::nodes::node_geo_set_spline_cyclic_cc {

static void geo_node_set_spline_cyclic_declare(NodeDeclarationBuilder &b)		static void node_declare(NodeDeclarationBuilder &b)
{		{
b.add_input<decl::Geometry>(N_("Geometry")).supported_type(GEO_COMPONENT_TYPE_CURVE);		b.add_input<decl::Geometry>(N_("Geometry")).supported_type(GEO_COMPONENT_TYPE_CURVE);
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::Bool>(N_("Cyclic")).supports_field();		b.add_input<decl::Bool>(N_("Cyclic")).supports_field();
b.add_output<decl::Geometry>(N_("Geometry"));		b.add_output<decl::Geometry>(N_("Geometry"));
}		}

static void set_cyclic_in_component(GeometryComponent &component,		static void set_cyclic_in_component(GeometryComponent &component,
Show All 14 Lines	static void set_cyclic_in_component(GeometryComponent &component,
OutputAttribute_Typed<bool> cyclics = component.attribute_try_get_for_output_only<bool>(		OutputAttribute_Typed<bool> cyclics = component.attribute_try_get_for_output_only<bool>(
"cyclic", ATTR_DOMAIN_CURVE);		"cyclic", ATTR_DOMAIN_CURVE);
fn::FieldEvaluator cyclic_evaluator{field_context, &selection};		fn::FieldEvaluator cyclic_evaluator{field_context, &selection};
cyclic_evaluator.add_with_destination(cyclic_field, cyclics.varray());		cyclic_evaluator.add_with_destination(cyclic_field, cyclics.varray());
cyclic_evaluator.evaluate();		cyclic_evaluator.evaluate();
cyclics.save();		cyclics.save();
}		}

static void geo_node_set_spline_cyclic_exec(GeoNodeExecParams params)		static void node_exec(GeoNodeExecParams params)
{		{
GeometrySet geometry_set = params.extract_input<GeometrySet>("Geometry");		GeometrySet geometry_set = params.extract_input<GeometrySet>("Geometry");
Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");		Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");
Field<bool> cyclic_field = params.extract_input<Field<bool>>("Cyclic");		Field<bool> cyclic_field = params.extract_input<Field<bool>>("Cyclic");

geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {		geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
if (geometry_set.has_curve()) {		if (geometry_set.has_curve()) {
set_cyclic_in_component(		set_cyclic_in_component(
Show All 9 Lines
void register_node_type_geo_set_spline_cyclic()		void register_node_type_geo_set_spline_cyclic()
{		{
namespace file_ns = blender::nodes::node_geo_set_spline_cyclic_cc;		namespace file_ns = blender::nodes::node_geo_set_spline_cyclic_cc;

static bNodeType ntype;		static bNodeType ntype;

geo_node_type_base(		geo_node_type_base(
&ntype, GEO_NODE_SET_SPLINE_CYCLIC, "Set Spline Cyclic", NODE_CLASS_GEOMETRY, 0);		&ntype, GEO_NODE_SET_SPLINE_CYCLIC, "Set Spline Cyclic", NODE_CLASS_GEOMETRY, 0);
ntype.geometry_node_execute = file_ns::geo_node_set_spline_cyclic_exec;		ntype.geometry_node_execute = file_ns::node_exec;
ntype.declare = file_ns::geo_node_set_spline_cyclic_declare;		ntype.declare = file_ns::node_declare;
nodeRegisterType(&ntype);		nodeRegisterType(&ntype);
}		}