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

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_set_curve_tilt.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_curve_tilt_cc {		namespace blender::nodes::node_geo_set_curve_tilt_cc {

static void geo_node_set_curve_tilt_declare(NodeDeclarationBuilder &b)		static void node_declare(NodeDeclarationBuilder &b)
{		{
b.add_input<decl::Geometry>(N_("Curve")).supported_type(GEO_COMPONENT_TYPE_CURVE);		b.add_input<decl::Geometry>(N_("Curve")).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::Float>(N_("Tilt")).subtype(PROP_ANGLE).supports_field();		b.add_input<decl::Float>(N_("Tilt")).subtype(PROP_ANGLE).supports_field();
b.add_output<decl::Geometry>(N_("Curve"));		b.add_output<decl::Geometry>(N_("Curve"));
}		}

static void set_tilt_in_component(GeometryComponent &component,		static void set_tilt_in_component(GeometryComponent &component,
Show All 14 Lines	static void set_tilt_in_component(GeometryComponent &component,
OutputAttribute_Typed<float> tilts = component.attribute_try_get_for_output_only<float>(		OutputAttribute_Typed<float> tilts = component.attribute_try_get_for_output_only<float>(
"tilt", ATTR_DOMAIN_POINT);		"tilt", ATTR_DOMAIN_POINT);
fn::FieldEvaluator tilt_evaluator{field_context, &selection};		fn::FieldEvaluator tilt_evaluator{field_context, &selection};
tilt_evaluator.add_with_destination(tilt_field, tilts.varray());		tilt_evaluator.add_with_destination(tilt_field, tilts.varray());
tilt_evaluator.evaluate();		tilt_evaluator.evaluate();
tilts.save();		tilts.save();
}		}

static void geo_node_set_curve_tilt_exec(GeoNodeExecParams params)		static void node_geo_exec(GeoNodeExecParams params)
{		{
GeometrySet geometry_set = params.extract_input<GeometrySet>("Curve");		GeometrySet geometry_set = params.extract_input<GeometrySet>("Curve");
Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");		Field<bool> selection_field = params.extract_input<Field<bool>>("Selection");
Field<float> tilt_field = params.extract_input<Field<float>>("Tilt");		Field<float> tilt_field = params.extract_input<Field<float>>("Tilt");

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_tilt_in_component(		set_tilt_in_component(
geometry_set.get_component_for_write<CurveComponent>(), selection_field, tilt_field);		geometry_set.get_component_for_write<CurveComponent>(), selection_field, tilt_field);
}		}
});		});

params.set_output("Curve", std::move(geometry_set));		params.set_output("Curve", std::move(geometry_set));
}		}

} // namespace blender::nodes::node_geo_set_curve_tilt_cc		} // namespace blender::nodes::node_geo_set_curve_tilt_cc

void register_node_type_geo_set_curve_tilt()		void register_node_type_geo_set_curve_tilt()
{		{
namespace file_ns = blender::nodes::node_geo_set_curve_tilt_cc;		namespace file_ns = blender::nodes::node_geo_set_curve_tilt_cc;

static bNodeType ntype;		static bNodeType ntype;

geo_node_type_base(&ntype, GEO_NODE_SET_CURVE_TILT, "Set Curve Tilt", NODE_CLASS_GEOMETRY, 0);		geo_node_type_base(&ntype, GEO_NODE_SET_CURVE_TILT, "Set Curve Tilt", NODE_CLASS_GEOMETRY, 0);
ntype.geometry_node_execute = file_ns::geo_node_set_curve_tilt_exec;		ntype.geometry_node_execute = file_ns::node_geo_exec;
ntype.declare = file_ns::geo_node_set_curve_tilt_declare;		ntype.declare = file_ns::node_declare;
nodeRegisterType(&ntype);		nodeRegisterType(&ntype);
}		}