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

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_set_material_index.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_material_index_cc {		namespace blender::nodes::node_geo_set_material_index_cc {

static void geo_node_set_material_index_declare(NodeDeclarationBuilder &b)		static void node_declare(NodeDeclarationBuilder &b)
{		{
b.add_input<decl::Geometry>(N_("Geometry")).supported_type(GEO_COMPONENT_TYPE_MESH);		b.add_input<decl::Geometry>(N_("Geometry")).supported_type(GEO_COMPONENT_TYPE_MESH);
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::Int>(N_("Material Index")).supports_field().min(0);		b.add_input<decl::Int>(N_("Material Index")).supports_field().min(0);
b.add_output<decl::Geometry>(N_("Geometry"));		b.add_output<decl::Geometry>(N_("Geometry"));
}		}

static void set_material_index_in_component(GeometryComponent &component,		static void set_material_index_in_component(GeometryComponent &component,
Show All 14 Lines	static void set_material_index_in_component(GeometryComponent &component,
OutputAttribute_Typed<int> indices = component.attribute_try_get_for_output_only<int>(		OutputAttribute_Typed<int> indices = component.attribute_try_get_for_output_only<int>(
"material_index", ATTR_DOMAIN_FACE);		"material_index", ATTR_DOMAIN_FACE);
fn::FieldEvaluator material_evaluator{field_context, &selection};		fn::FieldEvaluator material_evaluator{field_context, &selection};
material_evaluator.add_with_destination(index_field, indices.varray());		material_evaluator.add_with_destination(index_field, indices.varray());
material_evaluator.evaluate();		material_evaluator.evaluate();
indices.save();		indices.save();
}		}

static void geo_node_set_material_index_exec(GeoNodeExecParams params)		static void node_geo_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<int> index_field = params.extract_input<Field<int>>("Material Index");		Field<int> index_field = params.extract_input<Field<int>>("Material Index");

geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {		geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
if (geometry_set.has_mesh()) {		if (geometry_set.has_mesh()) {
set_material_index_in_component(		set_material_index_in_component(
geometry_set.get_component_for_write<MeshComponent>(), selection_field, index_field);		geometry_set.get_component_for_write<MeshComponent>(), selection_field, index_field);
}		}
});		});
params.set_output("Geometry", std::move(geometry_set));		params.set_output("Geometry", std::move(geometry_set));
}		}

} // namespace blender::nodes::node_geo_set_material_index_cc		} // namespace blender::nodes::node_geo_set_material_index_cc

void register_node_type_geo_set_material_index()		void register_node_type_geo_set_material_index()
{		{
namespace file_ns = blender::nodes::node_geo_set_material_index_cc;		namespace file_ns = blender::nodes::node_geo_set_material_index_cc;

static bNodeType ntype;		static bNodeType ntype;

geo_node_type_base(		geo_node_type_base(
&ntype, GEO_NODE_SET_MATERIAL_INDEX, "Set Material Index", NODE_CLASS_GEOMETRY, 0);		&ntype, GEO_NODE_SET_MATERIAL_INDEX, "Set Material Index", NODE_CLASS_GEOMETRY, 0);
ntype.geometry_node_execute = file_ns::geo_node_set_material_index_exec;		ntype.geometry_node_execute = file_ns::node_geo_exec;
ntype.declare = file_ns::geo_node_set_material_index_declare;		ntype.declare = file_ns::node_declare;
nodeRegisterType(&ntype);		nodeRegisterType(&ntype);
}		}