/* SPDX-License-Identifier: GPL-2.0-or-later */ #include "node_geometry_util.hh" #include "BKE_instances.hh" namespace blender::nodes::node_geo_input_instance_scale_cc { static void node_declare(NodeDeclarationBuilder &b) { b.add_output(N_("Scale")).field_source(); } class InstanceScaleFieldInput final : public bke::InstancesFieldInput { public: InstanceScaleFieldInput() : bke::InstancesFieldInput(CPPType::get(), "Scale") { } GVArray get_varray_for_context(const bke::Instances &instances, IndexMask /*mask*/) const final { auto scale_fn = [&](const int i) -> float3 { return instances.transforms()[i].scale(); }; return VArray::ForFunc(instances.instances_num(), scale_fn); } uint64_t hash() const override { return 8346343; } bool is_equal_to(const fn::FieldNode &other) const override { return dynamic_cast(&other) != nullptr; } }; static void node_geo_exec(GeoNodeExecParams params) { Field scale{std::make_shared()}; params.set_output("Scale", std::move(scale)); } } // namespace blender::nodes::node_geo_input_instance_scale_cc void register_node_type_geo_input_instance_scale() { namespace file_ns = blender::nodes::node_geo_input_instance_scale_cc; static bNodeType ntype; geo_node_type_base(&ntype, GEO_NODE_INPUT_INSTANCE_SCALE, "Instance Scale", NODE_CLASS_INPUT); ntype.geometry_node_execute = file_ns::node_geo_exec; ntype.declare = file_ns::node_declare; nodeRegisterType(&ntype); }