103 lines
2.6 KiB
Rust
103 lines
2.6 KiB
Rust
use crate::colour::Colour;
|
|
use crate::{from_base36, ToBase36};
|
|
|
|
#[derive(Debug, Eq, PartialEq)]
|
|
pub struct Palette {
|
|
pub id: u64,
|
|
pub name: Option<String>,
|
|
pub colours: Vec<Colour>,
|
|
}
|
|
|
|
impl From<String> for Palette {
|
|
fn from(string: String) -> Palette {
|
|
let lines: Vec<&str> = string.lines().collect();
|
|
|
|
let id = from_base36(&lines[0].replace("PAL ", ""));
|
|
|
|
let name = match lines[1].starts_with("NAME") {
|
|
true => Some(lines[1].replace("NAME ", "").to_string()),
|
|
false => None,
|
|
};
|
|
|
|
let colour_start_index = if name.is_some() { 2 } else { 1 };
|
|
|
|
let colours = lines[colour_start_index..].iter().map(|&line| {
|
|
Colour::from(line.to_string())
|
|
}).collect();
|
|
|
|
Palette { id, name, colours }
|
|
}
|
|
}
|
|
|
|
impl ToString for Palette {
|
|
#[inline]
|
|
fn to_string(&self) -> String {
|
|
let name = if self.name.as_ref().is_some() {
|
|
format!("NAME {}\n", self.name.as_ref().unwrap())
|
|
} else {
|
|
"".to_string()
|
|
};
|
|
|
|
let mut colours = String::new();
|
|
for colour in &self.colours {
|
|
colours.push_str(&format!("{}\n", colour.to_string()));
|
|
}
|
|
colours.pop();
|
|
|
|
format!("PAL {}\n{}{}", self.id.to_base36(), name, colours)
|
|
}
|
|
}
|
|
|
|
#[test]
|
|
fn test_palette_from_string() {
|
|
let output = Palette::from(
|
|
"PAL 1\nNAME lamplight\n45,45,59\n66,60,39\n140,94,1".to_string()
|
|
);
|
|
|
|
let expected = Palette {
|
|
id: 1,
|
|
name: Some("lamplight".to_string()),
|
|
colours: vec![
|
|
Colour {red: 45, green: 45, blue: 59},
|
|
Colour {red: 66, green: 60, blue: 39},
|
|
Colour {red: 140, green: 94, blue: 1 },
|
|
],
|
|
};
|
|
|
|
assert_eq!(output, expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_palette_from_string_no_name() {
|
|
let output = Palette::from(
|
|
"PAL 9\n45,45,59\n66,60,39\n140,94,1".to_string()
|
|
);
|
|
|
|
let expected = Palette {
|
|
id: 9,
|
|
name: None,
|
|
colours: vec![
|
|
Colour {red: 45, green: 45, blue: 59},
|
|
Colour {red: 66, green: 60, blue: 39},
|
|
Colour {red: 140, green: 94, blue: 1 },
|
|
],
|
|
};
|
|
|
|
assert_eq!(output, expected);
|
|
}
|
|
|
|
#[test]
|
|
fn test_palette_to_string() {
|
|
let output = Palette {
|
|
id: 16,
|
|
name: Some("moss".to_string()),
|
|
colours: vec![
|
|
Colour {red: 1, green: 2, blue: 3 },
|
|
Colour {red: 255, green: 254, blue: 253},
|
|
Colour {red: 126, green: 127, blue: 128},
|
|
]
|
|
}.to_string();
|
|
let expected = "PAL g\nNAME moss\n1,2,3\n255,254,253\n126,127,128".to_string();
|
|
assert_eq!(output, expected);
|
|
}
|