Differential D15131 Diff 52297 source/blender/nodes/geometry/nodes/node_geo_input_instance_rotation.cc

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_input_instance_rotation.cc

This file was added.

				/* SPDX-License-Identifier: GPL-2.0-or-later */

				#include "node_geometry_util.hh"

				namespace blender::nodes::node_geo_input_instance_rotation_cc {

				static void node_declare(NodeDeclarationBuilder &b)
				{
				b.add_output<decl::Vector>(N_("Rotation")).field_source();
				}

				class VectorFieldInput final : public GeometryFieldInput {
				public:
				VectorFieldInput() : GeometryFieldInput(CPPType::get<float3>(), "Rotation")
				{
				}

				GVArray get_varray_for_context(const GeometryComponent &component,
				const eAttrDomain domain,
				IndexMask mask) const final
				{
				if (component.type() != GEO_COMPONENT_TYPE_INSTANCES) {
				return {};
				}

				const InstancesComponent &instance_component = static_cast<const InstancesComponent &>(
				component);

				auto rotation_fn = [&](const int i) -> float3 {
				return instance_component.instance_transforms()[i].to_euler();
				};

				return component.attribute_try_adapt_domain<float3>(
				VArray<float3>::ForFunc(instance_component.instances_num(), rotation_fn),
				ATTR_DOMAIN_INSTANCE,
				domain);
				}

				uint64_t hash() const override
				{
				return 63563636;
				}

				bool is_equal_to(const fn::FieldNode &other) const override
				{
				return dynamic_cast<const VectorFieldInput *>(&other) != nullptr;
				}
				};

				static void node_geo_exec(GeoNodeExecParams params)
				{
				Field<float3> rotation{std::make_shared<VectorFieldInput>()};
				params.set_output("Rotation", std::move(rotation));
				}

				} // namespace blender::nodes::node_geo_input_instance_rotation_cc

				void register_node_type_geo_input_instance_rotation()
				{
				namespace file_ns = blender::nodes::node_geo_input_instance_rotation_cc;

				static bNodeType ntype;
				geo_node_type_base(
				&ntype, GEO_NODE_INPUT_INSTANCE_ROTATION, "Instance Rotation", NODE_CLASS_INPUT);
				ntype.geometry_node_execute = file_ns::node_geo_exec;
				ntype.declare = file_ns::node_declare;
				nodeRegisterType(&ntype);
				}