Differential D12656 Diff 42521 source/blender/nodes/geometry/nodes/node_geo_material_replace.cc

Changeset View

Standalone View

source/blender/nodes/geometry/nodes/node_geo_material_replace.cc

	Show All 34 Lines
	}			}

	static void geo_node_material_replace_exec(GeoNodeExecParams params)			static void geo_node_material_replace_exec(GeoNodeExecParams params)
	{			{
	Material old_material = params.extract_input<Material >("Old");			Material old_material = params.extract_input<Material >("Old");
	Material new_material = params.extract_input<Material >("New");			Material new_material = params.extract_input<Material >("New");

	GeometrySet geometry_set = params.extract_input<GeometrySet>("Geometry");			GeometrySet geometry_set = params.extract_input<GeometrySet>("Geometry");
	geometry_set = geometry_set_realize_instances(geometry_set);

	if (geometry_set.has<MeshComponent>()) {			geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
	MeshComponent &mesh_component = geometry_set.get_component_for_write<MeshComponent>();			Mesh *mesh = geometry_set.get_mesh_for_write();
	Mesh *mesh = mesh_component.get_for_write();
	if (mesh != nullptr) {			if (mesh != nullptr) {
	for (const int i : IndexRange(mesh->totcol)) {			for (const int i : IndexRange(mesh->totcol)) {
	if (mesh->mat[i] == old_material) {			if (mesh->mat[i] == old_material) {
	mesh->mat[i] = new_material;			mesh->mat[i] = new_material;
	}			}
	}			}
	}			}
	}			});

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

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

	void register_node_type_geo_material_replace()			void register_node_type_geo_material_replace()
	{			{
	static bNodeType ntype;			static bNodeType ntype;

	geo_node_type_base(			geo_node_type_base(
	&ntype, GEO_NODE_MATERIAL_REPLACE, "Material Replace", NODE_CLASS_GEOMETRY, 0);			&ntype, GEO_NODE_MATERIAL_REPLACE, "Material Replace", NODE_CLASS_GEOMETRY, 0);
	ntype.declare = blender::nodes::geo_node_material_replace_declare;			ntype.declare = blender::nodes::geo_node_material_replace_declare;
	ntype.geometry_node_execute = blender::nodes::geo_node_material_replace_exec;			ntype.geometry_node_execute = blender::nodes::geo_node_material_replace_exec;
	nodeRegisterType(&ntype);			nodeRegisterType(&ntype);
	}			}