redstrate/physis
1
Fork 0
mirror of https://github.com/redstrate/Physis.git synced 2025-04-28 06:57:46 +00:00
Code Activity

Add support for writing the rest of the model header and metadata

Browse source
This commit is contained in:
Joshua Goins 2023-09-22 17:04:48 -04:00
parent 3d621d3588
commit 3a50523dd5
1 changed files with 32 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
src/model.rs
View file

@ -1,7 +1,7 @@
// SPDX-FileCopyrightText: 2023 Joshua Goins <josh@redstrate.com> // SPDX-FileCopyrightText: 2023 Joshua Goins <josh@redstrate.com>
// SPDX-License-Identifier: GPL-3.0-or-later // SPDX-License-Identifier: GPL-3.0-or-later
use std::io::{Cursor, Seek, SeekFrom}; use std::io::{Cursor, Seek, SeekFrom, Write};
use std::ptr::write; use std::ptr::write;
use binrw::{BinResult, binrw, BinWrite}; use binrw::{BinResult, binrw, BinWrite};
@ -301,7 +301,7 @@ struct VertexElement {
offset: u8, offset: u8,
vertex_type: VertexType, vertex_type: VertexType,
vertex_usage: VertexUsage, vertex_usage: VertexUsage,
#[br(pad_after = 3)] #[brw(pad_after = 3)]
usage_index: u8, usage_index: u8,
} }
@ -326,8 +326,15 @@ pub struct Lod {
pub parts: Vec<Part>, pub parts: Vec<Part>,
} }
#[derive(Clone)]
struct VertexDeclaration {
elements: Vec<VertexElement>,
}
pub struct MDL { pub struct MDL {
file_header: ModelFileHeader, file_header: ModelFileHeader,
vertex_declarations: Vec<VertexDeclaration>,
model_data: ModelData,
pub lods: Vec<Lod>, pub lods: Vec<Lod>,
pub affected_bone_names: Vec<String>, pub affected_bone_names: Vec<String>,
@ -339,11 +346,6 @@ impl MDL {
let mut cursor = Cursor::new(buffer); let mut cursor = Cursor::new(buffer);
let model_file_header = ModelFileHeader::read(&mut cursor).unwrap(); let model_file_header = ModelFileHeader::read(&mut cursor).unwrap();
#[derive(Clone)]
struct VertexDeclaration {
elements: Vec<VertexElement>,
}
let mut vertex_declarations: Vec<VertexDeclaration> = let mut vertex_declarations: Vec<VertexDeclaration> =
vec![ vec![
VertexDeclaration { elements: vec![] }; VertexDeclaration { elements: vec![] };
@ -370,7 +372,8 @@ impl MDL {
let mut affected_bone_names = vec![]; let mut affected_bone_names = vec![];
for mut offset in model.bone_name_offsets { for offset in &model.bone_name_offsets {
let mut offset = *offset;
let mut string = String::new(); let mut string = String::new();
let mut next_char = model.header.strings[offset as usize] as char; let mut next_char = model.header.strings[offset as usize] as char;
@ -385,7 +388,8 @@ impl MDL {
let mut material_names = vec![]; let mut material_names = vec![];
for mut offset in model.material_name_offsets { for offset in &model.material_name_offsets {
let mut offset = *offset;
let mut string = String::new(); let mut string = String::new();
let mut next_char = model.header.strings[offset as usize] as char; let mut next_char = model.header.strings[offset as usize] as char;
@ -525,6 +529,8 @@ impl MDL {
Some(MDL { Some(MDL {
file_header: model_file_header, file_header: model_file_header,
vertex_declarations,
model_data: model,
lods, lods,
affected_bone_names, affected_bone_names,
material_names material_names
@ -537,7 +543,23 @@ impl MDL {
{ {
let mut cursor = Cursor::new(&mut buffer); let mut cursor = Cursor::new(&mut buffer);
self.file_header.write(&mut cursor).unwrap(); // write file header
self.file_header.write(&mut cursor).ok()?;
// write vertex declarations
for declaration in &self.vertex_declarations {
for element in &declaration.elements {
element.write(&mut cursor).ok()?;
}
cursor.write_all(&[255u8]).ok()?;
// We don't have a +1 here like we do in read, because writing the EOF (255) pushes our cursor forward.
let to_seek = 17 * 8 - (declaration.elements.len()) * 8;
cursor.seek(SeekFrom::Current(to_seek as i64)).ok()?;
}
self.model_data.write(&mut cursor).ok()?;
} }
Some(buffer) Some(buffer)