use crate::prelude::*;
use crate::util::init::vec_with_capacity;
use crate::wad::build::builder::DataBuilder;
use crate::wad::elem::GMElement;
use crate::wad::parse::reader::DataReader;
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CodeLocal {
pub name: GMRef<String>,
pub variables: Vec<LocalVariable>,
}
impl GMElement for CodeLocal {
fn deserialize(reader: &mut DataReader) -> Result<Self> {
let local_variables_count = reader.read_u32()?;
let name: GMRef<String> = reader.read_gm_string()?;
let mut variables: Vec<LocalVariable> = vec_with_capacity(local_variables_count)?;
for _ in 0..local_variables_count {
variables.push(LocalVariable::deserialize(reader)?);
}
Ok(Self { name, variables })
}
fn serialize(&self, builder: &mut DataBuilder) -> Result<()> {
builder.write_usize(self.variables.len())?;
builder.write_gm_string(self.name)?;
for local_var in &self.variables {
local_var.serialize(builder)?;
}
Ok(())
}
}
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct LocalVariable {
pub weird_index: u32,
pub name: GMRef<String>,
}
impl GMElement for LocalVariable {
fn deserialize(reader: &mut DataReader) -> Result<Self> {
let weird_index = reader.read_u32()?;
let name: GMRef<String> = reader.read_gm_string()?;
Ok(Self { weird_index, name })
}
fn serialize(&self, builder: &mut DataBuilder) -> Result<()> {
builder.write_u32(self.weird_index);
builder.write_gm_string(self.name)?;
Ok(())
}
}