150 lines
4.2 KiB
Rust
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(Clone, Debug, Eq, PartialEq)]
|
|
pub struct Sprite {
|
|
pub id: String,
|
|
pub name: Option<String>,
|
|
pub animation_frames: Vec<Image>,
|
|
pub dialogue_id: Option<String>,
|
|
pub room_id: Option<String>,
|
|
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.as_ref().unwrap(),
|
|
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 = lines[0].replace("SPR ", "");
|
|
let mut name = None;
|
|
let mut dialogue_id: Option<String> = None;
|
|
let mut room_id: Option<String> = 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(room_position[0].to_string());
|
|
|
|
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,
|
|
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());
|
|
}
|
|
}
|