wipe/src/sampler.rs

100 lines
2.5 KiB
Rust
Raw Normal View History

2022-04-08 18:05:00 +00:00
use crossterm::style::Color;
use crate::animation::Animation;
use crate::char::CharSampler;
use crate::color::ColorSampler;
use crate::fill::FillMode;
use crate::vec::Vector;
pub enum Sample {
Keep,
Draw { char: char, color: Color },
Clear,
}
pub trait Sampler {
fn sample(&self, step: f32, pos: Vector) -> Sample;
}
pub struct ComposedSampler {
animation: Box<dyn Animation>,
fill: Box<dyn FillMode>,
color: Box<dyn ColorSampler>,
char: Box<dyn CharSampler>,
}
impl ComposedSampler {
pub fn new(animation: Box<dyn Animation>,
fill: Box<dyn FillMode>,
color: Box<dyn ColorSampler>,
char: Box<dyn CharSampler>) -> Self {
Self { animation, fill, color, char }
}
}
impl Sampler for ComposedSampler {
fn sample(&self, step: f32, pos: Vector) -> Sample {
let level = self.animation.sample(step, pos);
if level > 1.0 {
Sample::Keep
} else if level > 0.0 {
let char = self.char.sample(level);
let fill = self.fill.sample(level, pos);
let color = self.color.sample(fill);
Sample::Draw { char, color }
} else {
Sample::Clear
}
}
}
#[cfg(test)]
mod test {
use crate::char::SimpleCharSampler;
use super::*;
struct MockAnimation;
struct MockFillMode;
struct MockColorSampler;
impl Animation for MockAnimation {
fn sample(&self, step: f32, pos: Vector) -> f32 {
step + pos.x + pos.y
}
}
impl FillMode for MockFillMode {
fn sample(&self, level: f32, pos: Vector) -> f32 {
level + pos.x + pos.y
}
}
impl ColorSampler for MockColorSampler {
fn sample(&self, _: f32) -> Color {
Color::Green
}
}
fn create_sampler() -> ComposedSampler {
let anim = Box::new(MockAnimation { });
let fill = Box::new(MockFillMode { });
let color = Box::new(MockColorSampler { });
let char = Box::new(SimpleCharSampler::new("0123456789".to_string()));
ComposedSampler::new(anim, fill, color, char)
}
#[test]
fn sample_keep() {
assert!(matches!(create_sampler().sample(0.7, Vector::new(0.3, 0.1)), Sample::Keep));
}
#[test]
fn sample_draw() {
assert!(matches!(create_sampler().sample(0.4, Vector::new(0.1, 0.1)), Sample::Draw { char: '6', color: Color::Green }));
}
#[test]
fn sample_clear() {
assert!(matches!(create_sampler().sample(-1.0, Vector::new(0.4, 0.5)), Sample::Clear));
}
}