Compare commits

..

11 Commits

Author SHA1 Message Date
LeovanGit a6118a7def fix vertex colors export 2023-11-22 21:14:38 -06:00
PrismaticFlower be09e10db5 update version number 2023-11-22 21:14:38 -06:00
PrismaticFlower ceb8cd79c3 Skip adding color attributes when unneeded
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.
2023-11-22 21:14:38 -06:00
LeovanGit 13a6511f23 Add vertex colors to blender + fix unpack_color() 2023-11-22 21:14:38 -06:00
itdominator a62f56d461 Partial Revert: Animation Track patch cleanup 2023-11-22 21:14:38 -06:00
itdominator 8a7d9b0958 Animation Track patch cleanup 2023-11-22 21:14:38 -06:00
itdominator 62206e8dbc Fixed import of animations to allow for bulk import 2023-11-22 21:14:38 -06:00
William Herald Snyder cc4a1b0e04 Create sv faces from half-edge list. Current method doesn't check for duplicates, but Blender seems to filter them anyhow. 2023-01-08 22:15:19 -05:00
William Herald Snyder ab253f0acc Ignore swap files 2023-01-08 20:52:23 -05:00
William Herald Snyder 582ed1ace5 Create mesh from shadow geometry with verts, edges, but no faces 2023-01-07 06:51:54 -05:00
William Herald Snyder 63f9e43e17 Read SHDW chunks 2023-01-06 21:30:45 -05:00
5 changed files with 164 additions and 10 deletions

2
.gitignore vendored
View File

@ -1,6 +1,8 @@
.DS_Store .DS_Store
*.msh *.msh
*.swp
# Created by https://www.gitignore.io/api/python,visualstudiocode # Created by https://www.gitignore.io/api/python,visualstudiocode
# Edit at https://www.gitignore.io/?templates=python,visualstudiocode # Edit at https://www.gitignore.io/?templates=python,visualstudiocode

View File

@ -32,7 +32,111 @@ def validate_segment_geometry(segment : GeometrySegment):
return True return True
def model_to_mesh_object(model: Model, scene : Scene, materials_map : Dict[str, bpy.types.Material]) -> bpy.types.Object: def get_shadow_geometry(model: Model):
for segment in model.geometry:
if segment.shadow_geometry is not None:
return segment.shadow_geometry
return None
# SHDW mesh info is of a different form from
# normal segment geometry
def model_to_shadow_mesh(model: Model, shadow_geometry : ShadowGeometry):
blender_mesh = bpy.data.meshes.new(model.name)
# As is the case with normal geometry processing,
# these will contain flattened lists
vertex_positions = [convert_vector_space(position) for position in shadow_geometry.positions]
# Vertices
blender_mesh.vertices.add(len(vertex_positions))
blender_mesh.vertices.foreach_set("co", [component for vertex_position in vertex_positions for component in vertex_position])
def faces_from_half_edges(half_edges : List[Tuple[int,int,int,int]]) -> List[List[int]]:
faces = []
visited_edges = [False] * len(half_edges)
for i in range(len(half_edges)):
if visited_edges[i]:
continue
curr_edge = half_edges[i]
curr_index = curr_edge[0]
starting_index = curr_index
face_length = 0
face_temp = [0] * 5
while True:
if face_length + 1> len(face_temp):
face_temp.append(curr_index)
else:
face_temp[face_length] = curr_index
face_length += 1
curr_edge = half_edges[curr_edge[1]]
curr_index = curr_edge[0]
if (curr_index == starting_index):
break
#print(f"Added a face of length: {face_length}")
faces.append(face_temp[0:face_length])
return faces
polygons = faces_from_half_edges(shadow_geometry.edges)
# LOOPS
flat_indices = [index for polygon in polygons for index in polygon]
blender_mesh.loops.add(len(flat_indices))
# Position indices
blender_mesh.loops.foreach_set("vertex_index", flat_indices)
# POLYGONS/FACES
blender_mesh.polygons.add(len(polygons))
# Indices of starting loop for each polygon
polygon_loop_start_indices = [0] * len(polygons)
current_polygon_start_index = 0
# Number of loops in this polygon. Polygon i will use
# loops from polygon_loop_start_indices[i] to
# polygon_loop_start_indices[i] + polygon_loop_totals[i]
polygon_loop_totals = [0] * len(polygons)
for i,polygon in enumerate(polygons):
polygon_loop_start_indices[i] = current_polygon_start_index
current_polygon_length = len(polygon)
current_polygon_start_index += current_polygon_length
polygon_loop_totals[i] = current_polygon_length
blender_mesh.polygons.foreach_set("loop_start", polygon_loop_start_indices)
blender_mesh.polygons.foreach_set("loop_total", polygon_loop_totals)
blender_mesh.validate(clean_customdata=False)
blender_mesh.update()
#sv_name = model.name if model.name.startswith("sv_") else "sv_" + model.name
blender_mesh_object = bpy.data.objects.new(model.name, blender_mesh)
return blender_mesh_object
def model_to_mesh(model: Model, scene: Scene, materials_map : Dict[str, bpy.types.Material]) -> bpy.types.Object:
blender_mesh = bpy.data.meshes.new(model.name) blender_mesh = bpy.data.meshes.new(model.name)
@ -142,25 +246,24 @@ def model_to_mesh_object(model: Model, scene : Scene, materials_map : Dict[str,
# POLYGONS/FACES # POLYGONS/FACES
blender_mesh.polygons.add(len(polygons)) blender_mesh.polygons.add(len(polygons))
# Indices of starting loop for each polygon # Indices of starting loop for each polygon
polygon_loop_start_indices = [] polygon_loop_start_indices = [0] * len(polygons)
current_polygon_start_index = 0 current_polygon_start_index = 0
# Number of loops in this polygon. Polygon i will use # Number of loops in this polygon. Polygon i will use
# loops from polygon_loop_start_indices[i] to # loops from polygon_loop_start_indices[i] to
# polygon_loop_start_indices[i] + polygon_loop_totals[i] # polygon_loop_start_indices[i] + polygon_loop_totals[i]
polygon_loop_totals = [] polygon_loop_totals = [0] * len(polygons)
for polygon in polygons: for i,polygon in enumerate(polygons):
polygon_loop_start_indices.append(current_polygon_start_index) polygon_loop_start_indices[i] = current_polygon_start_index
current_polygon_length = len(polygon) current_polygon_length = len(polygon)
current_polygon_start_index += current_polygon_length current_polygon_start_index += current_polygon_length
polygon_loop_totals.append(current_polygon_length) polygon_loop_totals[i] = current_polygon_length
blender_mesh.polygons.foreach_set("loop_start", polygon_loop_start_indices) blender_mesh.polygons.foreach_set("loop_start", polygon_loop_start_indices)
blender_mesh.polygons.foreach_set("loop_total", polygon_loop_totals) blender_mesh.polygons.foreach_set("loop_total", polygon_loop_totals)
@ -199,3 +302,16 @@ def model_to_mesh_object(model: Model, scene : Scene, materials_map : Dict[str,
return blender_mesh_object return blender_mesh_object
def model_to_mesh_object(model: Model, scene : Scene, materials_map : Dict[str, bpy.types.Material]) -> bpy.types.Object:
shadow_geometry = get_shadow_geometry(model)
if shadow_geometry is not None:
return model_to_shadow_mesh(model, shadow_geometry)
else:
return model_to_mesh(model, scene, materials_map)

View File

@ -39,6 +39,20 @@ class VertexWeight:
weight: float = 1.0 weight: float = 1.0
bone: int = 0 bone: int = 0
@dataclass
class ShadowGeometry:
""" Class representing 'SHDW' chunks. """
# Perhaps I could just use the positions list in the segment
# class, but I don't know if SHDW info can coexist with
# a normal geometry segment...
positions: List[Vector] = field(default_factory=list)
# The second two entries may not be necessary...
edges: List[Tuple[int,int,int,int]] = field(default_factory=list)
@dataclass @dataclass
class GeometrySegment: class GeometrySegment:
""" Class representing a 'SEGM' section in a .msh file. """ """ Class representing a 'SEGM' section in a .msh file. """
@ -56,6 +70,7 @@ class GeometrySegment:
triangles: List[List[int]] = field(default_factory=list) triangles: List[List[int]] = field(default_factory=list)
triangle_strips: List[List[int]] = None triangle_strips: List[List[int]] = None
shadow_geometry: ShadowGeometry = None
@dataclass @dataclass
class CollisionPrimitive: class CollisionPrimitive:

View File

@ -388,6 +388,28 @@ def _read_segm(segm: Reader, materials_list: List[Material]) -> GeometrySegment:
#if segm.read_u16 != 0: #if segm.read_u16 != 0:
# segm.skip_bytes(-2) # segm.skip_bytes(-2)
elif next_header == "SHDW":
shadow_geometry = ShadowGeometry()
with segm.read_child() as shdw:
#print("Found shadow chunk")
num_positions = shdw.read_u32()
#print(f" Num verts in shadow mesh: {num_positions}")
shadow_geometry.positions = [shdw.read_vec() for _ in range(num_positions)]
num_edges = shdw.read_u32()
#print(f" Num edges in shadow mesh: {num_edges}")
edges = []
for i in range(num_edges):
edges.append(tuple(shdw.read_u16(4)))
#print(" " + str(edges[-1]))
shadow_geometry.edges = edges
geometry_seg.shadow_geometry = shadow_geometry
elif next_header == "WGHT": elif next_header == "WGHT":
with segm.read_child() as wght: with segm.read_child() as wght:

View File

@ -44,10 +44,9 @@ def extract_models(scene: Scene, materials_map : Dict[str, bpy.types.Material])
new_obj = bpy.data.objects.new(model.name, None) new_obj = bpy.data.objects.new(model.name, None)
new_obj.empty_display_size = 1 new_obj.empty_display_size = 1
new_obj.empty_display_type = 'PLAIN_AXES' new_obj.empty_display_type = 'PLAIN_AXES'
new_obj.name = model.name
model_map[model.name] = new_obj model_map[model.name] = new_obj
new_obj.name = model.name
if model.parent: if model.parent:
new_obj.parent = model_map[model.parent] new_obj.parent = model_map[model.parent]