diff options
Diffstat (limited to 'source/blender/nodes/geometry/nodes/legacy/node_geo_legacy_point_separate.cc')
-rw-r--r-- | source/blender/nodes/geometry/nodes/legacy/node_geo_legacy_point_separate.cc | 179 |
1 files changed, 179 insertions, 0 deletions
diff --git a/source/blender/nodes/geometry/nodes/legacy/node_geo_legacy_point_separate.cc b/source/blender/nodes/geometry/nodes/legacy/node_geo_legacy_point_separate.cc new file mode 100644 index 00000000000..9260928b311 --- /dev/null +++ b/source/blender/nodes/geometry/nodes/legacy/node_geo_legacy_point_separate.cc @@ -0,0 +1,179 @@ +/* + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License + * as published by the Free Software Foundation; either version 2 + * of the License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software Foundation, + * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + */ + +#include "BKE_attribute_math.hh" +#include "BKE_mesh.h" +#include "BKE_pointcloud.h" + +#include "DNA_mesh_types.h" +#include "DNA_pointcloud_types.h" + +#include "node_geometry_util.hh" + +namespace blender::nodes { + +template<typename T> +static void copy_data_based_on_mask(Span<T> data, + Span<bool> masks, + const bool invert, + MutableSpan<T> out_data) +{ + int offset = 0; + for (const int i : data.index_range()) { + if (masks[i] != invert) { + out_data[offset] = data[i]; + offset++; + } + } +} + +void copy_point_attributes_based_on_mask(const GeometryComponent &in_component, + GeometryComponent &result_component, + Span<bool> masks, + const bool invert) +{ + for (const AttributeIDRef &attribute_id : in_component.attribute_ids()) { + ReadAttributeLookup attribute = in_component.attribute_try_get_for_read(attribute_id); + const CustomDataType data_type = bke::cpp_type_to_custom_data_type(attribute.varray.type()); + + /* Only copy point attributes. Theoretically this could interpolate attributes on other + * domains to the point domain, but that would conflict with attributes that are built-in + * on other domains, which causes creating the attributes to fail. */ + if (attribute.domain != ATTR_DOMAIN_POINT) { + continue; + } + + OutputAttribute result_attribute = result_component.attribute_try_get_for_output_only( + attribute_id, ATTR_DOMAIN_POINT, data_type); + + attribute_math::convert_to_static_type(data_type, [&](auto dummy) { + using T = decltype(dummy); + VArray_Span span{attribute.varray.typed<T>()}; + MutableSpan<T> out_span = result_attribute.as_span<T>(); + copy_data_based_on_mask(span, masks, invert, out_span); + }); + + result_attribute.save(); + } +} + +} // namespace blender::nodes + +namespace blender::nodes::node_geo_legacy_point_separate_cc { + +static void node_declare(NodeDeclarationBuilder &b) +{ + b.add_input<decl::Geometry>(N_("Geometry")); + b.add_input<decl::String>(N_("Mask")); + b.add_output<decl::Geometry>(N_("Geometry 1")); + b.add_output<decl::Geometry>(N_("Geometry 2")); +} + +static void create_component_points(GeometryComponent &component, const int total) +{ + switch (component.type()) { + case GEO_COMPONENT_TYPE_MESH: + static_cast<MeshComponent &>(component).replace(BKE_mesh_new_nomain(total, 0, 0, 0, 0)); + break; + case GEO_COMPONENT_TYPE_POINT_CLOUD: + static_cast<PointCloudComponent &>(component).replace(BKE_pointcloud_new_nomain(total)); + break; + default: + BLI_assert(false); + break; + } +} + +static void separate_points_from_component(const GeometryComponent &in_component, + GeometryComponent &out_component, + const StringRef mask_name, + const bool invert) +{ + if (!in_component.attribute_domain_supported(ATTR_DOMAIN_POINT) || + in_component.attribute_domain_size(ATTR_DOMAIN_POINT) == 0) { + return; + } + + const VArray<bool> mask_attribute = in_component.attribute_get_for_read<bool>( + mask_name, ATTR_DOMAIN_POINT, false); + VArray_Span<bool> masks{mask_attribute}; + + const int total = masks.count(!invert); + if (total == 0) { + return; + } + + create_component_points(out_component, total); + + copy_point_attributes_based_on_mask(in_component, out_component, masks, invert); +} + +static GeometrySet separate_geometry_set(const GeometrySet &set_in, + const StringRef mask_name, + const bool invert) +{ + GeometrySet set_out; + for (const GeometryComponent *component : set_in.get_components_for_read()) { + if (component->type() == GEO_COMPONENT_TYPE_CURVE) { + /* Don't support the curve component for now, even though it has a point domain. */ + continue; + } + GeometryComponent &out_component = set_out.get_component_for_write(component->type()); + separate_points_from_component(*component, out_component, mask_name, invert); + } + return set_out; +} + +static void node_geo_exec(GeoNodeExecParams params) +{ + bool wait_for_inputs = false; + wait_for_inputs |= params.lazy_require_input("Geometry"); + wait_for_inputs |= params.lazy_require_input("Mask"); + if (wait_for_inputs) { + return; + } + const std::string mask_attribute_name = params.get_input<std::string>("Mask"); + GeometrySet geometry_set = params.get_input<GeometrySet>("Geometry"); + + /* TODO: This is not necessary-- the input geometry set can be read only, + * but it must be rewritten to handle instance groups. */ + geometry_set = geometry::realize_instances_legacy(geometry_set); + + if (params.lazy_output_is_required("Geometry 1")) { + params.set_output("Geometry 1", + separate_geometry_set(geometry_set, mask_attribute_name, true)); + } + if (params.lazy_output_is_required("Geometry 2")) { + params.set_output("Geometry 2", + separate_geometry_set(geometry_set, mask_attribute_name, false)); + } +} + +} // namespace blender::nodes::node_geo_legacy_point_separate_cc + +void register_node_type_geo_point_separate() +{ + namespace file_ns = blender::nodes::node_geo_legacy_point_separate_cc; + + static bNodeType ntype; + + geo_node_type_base( + &ntype, GEO_NODE_LEGACY_POINT_SEPARATE, "Point Separate", NODE_CLASS_GEOMETRY, 0); + ntype.declare = file_ns::node_declare; + ntype.geometry_node_execute = file_ns::node_geo_exec; + ntype.geometry_node_execute_supports_laziness = true; + nodeRegisterType(&ntype); +} |