itdominator/SWBF-msh-Blender-IO
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Skip adding color attributes when unneeded

Browse Source 
This is a very small change to skip adding the vertex colours to the Blender mesh if no segment of the geometry being loaded has vertex colours.
This commit is contained in:
PrismaticFlower
2023-10-29 11:13:12 +13:00
parent 0ac921d855
commit ae42cda6ab
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
addons/io_scene_swbf_msh/msh_mesh_to_blend.py
View File

@@ -59,6 +59,7 @@ def model_to_mesh_object(model: Model, scene : Scene, materials_map : Dict[str,
if model.geometry: if model.geometry:
geometry_has_colors = any(segment.colors for segment in model.geometry)
for segment in model.geometry: for segment in model.geometry:
@@ -79,7 +80,7 @@ def model_to_mesh_object(model: Model, scene : Scene, materials_map : Dict[str,
if segment.colors: if segment.colors:
vertex_colors.extend(segment.colors) vertex_colors.extend(segment.colors)
else: elif geometry_has_colors:
[vertex_colors.extend([0.0, 0.0, 0.0, 1.0]) for _ in range(len(segment.positions))] [vertex_colors.extend([0.0, 0.0, 0.0, 1.0]) for _ in range(len(segment.positions))]
if segment.weights: if segment.weights:
@@ -135,6 +136,7 @@ def model_to_mesh_object(model: Model, scene : Scene, materials_map : Dict[str,
blender_mesh.uv_layers[0].data.foreach_set("uv", [component for i in flat_indices for component in vertex_uvs[i]]) blender_mesh.uv_layers[0].data.foreach_set("uv", [component for i in flat_indices for component in vertex_uvs[i]])
# Colors # Colors
if geometry_has_colors:
blender_mesh.color_attributes.new("COLOR0", "FLOAT_COLOR", "POINT") blender_mesh.color_attributes.new("COLOR0", "FLOAT_COLOR", "POINT")
blender_mesh.color_attributes[0].data.foreach_set("color", vertex_colors) blender_mesh.color_attributes[0].data.foreach_set("color", vertex_colors)