bitsy-parser/src/sprite.rs

150 lines
4.2 KiB
Rust

use crate::{from_base36, optional_data_line, AnimationFrames, Image, Position, ToBase36};
use crate::image::animation_frames_from_string;
use std::str::FromStr;
#[derive(Debug, Eq, PartialEq)]
pub struct Sprite {
pub id: u64,
pub name: Option<String>,
pub animation_frames: Vec<Image>,
pub dialogue_id: Option<String>,
pub room_id: Option<u64>,
pub position: Option<Position>,
pub colour_id: Option<u64>,
pub items: Vec<String>,
}
impl Sprite {
#[inline]
fn name_line(&self) -> String {
optional_data_line("NAME", self.name.as_ref())
}
#[inline]
fn dialogue_line(&self) -> String {
optional_data_line("DLG", self.dialogue_id.as_ref())
}
#[inline]
fn room_position_line(&self) -> String {
if self.room_id.is_some() && self.position.is_some() {
format!(
"\nPOS {} {}",
self.room_id.unwrap().to_base36(),
self.position.as_ref().unwrap().to_string()
)
} else {
"".to_string()
}
}
#[inline]
fn colour_line(&self) -> String {
optional_data_line("COL", self.colour_id.as_ref())
}
#[inline]
fn item_lines(&self) -> String {
if self.items.len() == 0 {
"".to_string()
} else {
let lines: Vec<String> = self.items.iter().map(|item| format!("ITM {}", item)).collect();
format!("\n{}", lines.join("\n"))
}
}
}
impl From<String> for Sprite {
#[inline]
fn from(string: String) -> Sprite {
let mut lines: Vec<&str> = string.lines().collect();
let id = from_base36(&lines[0].replace("SPR ", ""));
let mut name = None;
let mut dialogue_id: Option<String> = None;
let mut room_id: Option<u64> = None;
let mut position: Option<Position> = None;
let mut colour_id: Option<u64> = None;
let mut items: Vec<String> = Vec::new();
loop {
let last_line = lines.pop().unwrap();
if last_line.starts_with("NAME") {
name = Some(last_line.replace("NAME ", "").to_string());
} else if last_line.starts_with("DLG") {
dialogue_id = Some(last_line.replace("DLG ", "").to_string());
} else if last_line.starts_with("POS") {
let last_line = last_line.replace("POS ", "");
let room_position: Vec<&str> = last_line.split(' ').collect();
room_id = Some(from_base36(&room_position[0]));
if room_position.len() < 2 {
panic!("Bad room/position for sprite: {}", string);
}
position = Some(Position::from_str(room_position[1]).unwrap());
} else if last_line.starts_with("COL") {
colour_id = Some(last_line.replace("COL ", "").parse().unwrap());
} else if last_line.starts_with("ITM") {
items.push(last_line.replace("ITM ", ""));
} else {
lines.push(last_line);
break;
}
}
items.reverse();
let animation_frames = animation_frames_from_string(
lines[1..].join("\n")
);
Sprite {
id,
name,
animation_frames,
dialogue_id,
room_id,
position,
colour_id,
items
}
}
}
impl ToString for Sprite {
#[inline]
fn to_string(&self) -> String {
format!(
"SPR {}\n{}{}{}{}{}{}",
self.id.to_base36(),
self.animation_frames.to_string(),
self.name_line(),
self.dialogue_line(),
self.room_position_line(),
self.colour_line(),
self.item_lines(),
)
}
}
#[cfg(test)]
mod test {
use crate::mock;
use crate::sprite::Sprite;
#[test]
fn test_sprite_from_string() {
let output = Sprite::from(include_str!("test-resources/sprite").to_string());
let expected = mock::sprite();
assert_eq!(output, expected);
}
#[test]
fn test_sprite_to_string() {
assert_eq!(mock::sprite().to_string(), include_str!("test-resources/sprite").to_string());
}
}