70 lines
1.3 KiB
Rust
70 lines
1.3 KiB
Rust
use std::{collections::HashSet, fmt::Display};
|
|
|
|
#[derive(Debug, Hash, PartialEq, Eq)]
|
|
pub struct Point(pub isize, pub isize);
|
|
|
|
#[derive(Debug)]
|
|
pub enum Fold {
|
|
Horizontal(isize),
|
|
Vertical(isize),
|
|
}
|
|
|
|
#[derive(Debug)]
|
|
pub struct Canvas {
|
|
pub width: isize,
|
|
pub height: isize,
|
|
pub points: HashSet<Point>,
|
|
}
|
|
|
|
impl Canvas {
|
|
pub fn fold(self, fold: &Fold) -> Self {
|
|
let (width, height) = match fold {
|
|
Fold::Horizontal(amount) => (self.width, *amount),
|
|
Fold::Vertical(amount) => (*amount, self.height),
|
|
};
|
|
|
|
let mut points = HashSet::new();
|
|
for Point(x, y) in self.points {
|
|
if y > height {
|
|
points.insert(Point(x, height - (y - height)));
|
|
continue;
|
|
}
|
|
|
|
if x > width {
|
|
points.insert(Point(width - (x - width), y));
|
|
continue;
|
|
}
|
|
|
|
points.insert(Point(x, y));
|
|
}
|
|
|
|
Self {
|
|
width,
|
|
height,
|
|
points,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Display for Canvas {
|
|
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
|
for y in 0..self.height {
|
|
for x in 0..self.width {
|
|
if x % (self.width / 8) == 0 {
|
|
write!(f, " ")?;
|
|
}
|
|
|
|
if self.points.contains(&Point(x, y)) {
|
|
write!(f, "█")?;
|
|
} else {
|
|
write!(f, " ")?;
|
|
}
|
|
}
|
|
|
|
writeln!(f)?;
|
|
}
|
|
|
|
Ok(())
|
|
}
|
|
}
|