blender
/
blender
136
407
Fork 589
Code Issues 6.3k Pull Requests 768 Projects 19 Wiki Activity

GPv3: Replace Material node #113818

Merged
Dalai Felinto merged 2 commits from dfelinto/blender:grease-nodes-replace-material into main 2023-10-17 18:51:56 +02:00
Conversation 0 Commits 2 Files Changed 1 +20 -6
1 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
source/blender/nodes/geometry/nodes/node_geo_material_replace.cc
View File

@ -10,18 +10,31 @@
#include "DNA_mesh_types.h"
#include "DNA_meshdata_types.h"
#include "BKE_grease_pencil.hh"
#include "BKE_material.h"
namespace blender::nodes::node_geo_material_replace_cc {
static void node_declare(NodeDeclarationBuilder &b)
{
b.add_input<decl::Geometry>("Geometry").supported_type(GeometryComponent::Type::Mesh);
b.add_input<decl::Geometry>("Geometry")
.supported_type({GeometryComponent::Type::Mesh, GeometryComponent::Type::GreasePencil});
b.add_input<decl::Material>("Old");
b.add_input<decl::Material>("New").translation_context(BLT_I18NCONTEXT_ID_MATERIAL);
b.add_output<decl::Geometry>("Geometry").propagate_all();
}
static void replace_materials(MutableSpan<Material *> materials,
Material *src_material,
Material *dst_material)
{
for (const int i : materials.index_range()) {
if (materials[i] == src_material) {
materials[i] = dst_material;
}
}
}
static void node_geo_exec(GeoNodeExecParams params)
{
Material *old_material = params.extract_input<Material *>("Old");
@ -31,11 +44,12 @@ static void node_geo_exec(GeoNodeExecParams params)
geometry_set.modify_geometry_sets([&](GeometrySet &geometry_set) {
if (Mesh *mesh = geometry_set.get_mesh_for_write()) {
for (const int i : IndexRange(mesh->totcol)) {
if (mesh->mat[i] == old_material) {
mesh->mat[i] = new_material;
}
}
replace_materials({mesh->mat, mesh->totcol}, old_material, new_material);
}
if (GreasePencil *grease_pencil = geometry_set.get_grease_pencil_for_write()) {
replace_materials({grease_pencil->material_array, grease_pencil->material_array_num},
old_material,
new_material);
}
});