Add shift transform

This commit is contained in:
Rico Riedel 2022-08-03 18:00:26 +02:00
parent 32cd72552e
commit 74e83cd543
No known key found for this signature in database
GPG Key ID: 75AC868575DE7B18
3 changed files with 122 additions and 60 deletions

View File

@ -1,5 +1,6 @@
pub mod convert;
pub mod pattern;
pub mod transform;
mod error;
mod printer;
@ -17,6 +18,7 @@ pub use vec::*;
use crate::convert::*;
use crate::pattern::*;
use crate::transform::*;
use clap::{Parser, ValueEnum};
use crossterm::style::Color;
use crossterm::style::Color::*;
@ -43,6 +45,8 @@ struct Args {
colors: Vec<PalletEnum>,
#[clap(long, value_enum)]
color_pattern: Vec<PatternEnum>,
#[clap(long)]
color_shift: Option<bool>,
}
#[derive(ValueEnum, Clone)]
@ -81,13 +85,100 @@ enum PatternEnum {
Wheel,
}
#[derive(derive_more::Constructor)]
struct PatternConfig<'a> {
patterns: &'a Vec<PatternEnum>,
shift: Option<bool>,
}
impl Args {
fn char_config(&self) -> PatternConfig {
PatternConfig::new(&self.char_pattern, Some(true))
}
fn color_config(&self) -> PatternConfig {
PatternConfig::new(&self.color_pattern, self.color_shift)
}
fn pallet(&self, rand: &mut impl Rng) -> Vec<Color> {
match choose(&self.colors, rand) {
PalletEnum::Red => vec![DarkRed, Red, White],
PalletEnum::Yellow => vec![DarkYellow, Yellow, White],
PalletEnum::Green => vec![DarkGreen, Green, White],
PalletEnum::Blue => vec![DarkBlue, Blue, White],
PalletEnum::Magenta => vec![DarkMagenta, Magenta, White],
PalletEnum::Cyan => vec![DarkCyan, Cyan, White],
PalletEnum::Rainbow => vec![Red, Yellow, Green, Blue, Cyan, Magenta],
PalletEnum::DarkRed => vec![Black, DarkRed, Red],
PalletEnum::DarkYellow => vec![Black, DarkYellow, Yellow],
PalletEnum::DarkGreen => vec![Black, DarkGreen, Green],
PalletEnum::DarkBlue => vec![Black, DarkBlue, Blue],
PalletEnum::DarkMagenta => vec![Black, DarkMagenta, Magenta],
PalletEnum::DarkCyan => vec![Black, DarkCyan, Cyan],
PalletEnum::DarkRainbow => vec![
DarkRed,
DarkYellow,
DarkGreen,
DarkBlue,
DarkCyan,
DarkMagenta,
],
PalletEnum::RedYellow => vec![Red, DarkRed, DarkYellow, Yellow],
PalletEnum::YellowGreen => vec![Yellow, DarkYellow, DarkGreen, Green],
PalletEnum::GreenBlue => vec![Green, DarkGreen, DarkBlue, Blue],
PalletEnum::BlueCyan => vec![Blue, DarkBlue, DarkCyan, Cyan],
PalletEnum::CyanMagenta => vec![Cyan, DarkCyan, DarkMagenta, Magenta],
PalletEnum::MagentaRed => vec![Magenta, DarkMagenta, DarkRed, Red],
PalletEnum::Gray => vec![Black, DarkGrey, Grey, White],
}
}
}
impl<'a> PatternConfig<'a> {
fn create_base(&self, rand: &mut impl Rng) -> Box<dyn PatternFactory> {
match choose(self.patterns, rand) {
PatternEnum::Circle => Box::new(CircleFactory::default()),
PatternEnum::Line => Box::new(LineFactory::default()),
PatternEnum::Rhombus => Box::new(RhombusFactory::default()),
PatternEnum::Wheel => Box::new(WheelFactory::default()),
}
}
fn create(&self, rand: &mut impl Rng) -> Box<dyn PatternFactory> {
let mut pattern = self.create_base(rand);
if self.shift.unwrap_or(rand.gen()) {
pattern = Box::new(ShiftFactory::new(pattern))
}
pattern
}
}
fn choose<TValue: ValueEnum + Clone, TRand: Rng>(
options: &Vec<TValue>,
rand: &mut TRand,
) -> TValue {
if options.is_empty() {
TValue::value_variants()
.iter()
.choose(rand)
.unwrap()
.clone()
} else {
options.iter().choose(rand).unwrap().clone()
}
}
fn main() -> Result<(), Error> {
let args = Args::parse();
let rand = &mut thread_rng();
let char: Box<dyn PatternFactory> = create_pattern(choose(args.char_pattern, rand));
let color: Box<dyn PatternFactory> = create_pattern(choose(args.color_pattern, rand));
let pallet = create_pallet(choose(args.colors, rand));
let char = args.char_config().create(rand);
let color = args.color_config().create(rand);
let pallet = args.pallet(rand);
let sampler = SamplerFactoryImpl::new(char, color);
let char_converter = CharConverterImpl::new(args.chars);
@ -104,60 +195,3 @@ fn main() -> Result<(), Error> {
timer.run(renderer)
}
fn choose<TValue: ValueEnum + Clone, TRand: Rng>(options: Vec<TValue>, rand: &mut TRand) -> TValue {
if options.is_empty() {
TValue::value_variants()
.iter()
.choose(rand)
.unwrap()
.clone()
} else {
options.iter().choose(rand).unwrap().clone()
}
}
fn create_pattern(pattern: PatternEnum) -> Box<dyn PatternFactory> {
match pattern {
PatternEnum::Circle => Box::new(CircleFactory::default()),
PatternEnum::Line => Box::new(LineFactory::default()),
PatternEnum::Rhombus => Box::new(RhombusFactory::default()),
PatternEnum::Wheel => Box::new(WheelFactory::default()),
}
}
fn create_pallet(colors: PalletEnum) -> Vec<Color> {
match colors {
PalletEnum::Red => vec![DarkRed, Red, White],
PalletEnum::Yellow => vec![DarkYellow, Yellow, White],
PalletEnum::Green => vec![DarkGreen, Green, White],
PalletEnum::Blue => vec![DarkBlue, Blue, White],
PalletEnum::Magenta => vec![DarkMagenta, Magenta, White],
PalletEnum::Cyan => vec![DarkCyan, Cyan, White],
PalletEnum::Rainbow => vec![Red, Yellow, Green, Blue, Cyan, Magenta],
PalletEnum::DarkRed => vec![Black, DarkRed, Red],
PalletEnum::DarkYellow => vec![Black, DarkYellow, Yellow],
PalletEnum::DarkGreen => vec![Black, DarkGreen, Green],
PalletEnum::DarkBlue => vec![Black, DarkBlue, Blue],
PalletEnum::DarkMagenta => vec![Black, DarkMagenta, Magenta],
PalletEnum::DarkCyan => vec![Black, DarkCyan, Cyan],
PalletEnum::DarkRainbow => vec![
DarkRed,
DarkYellow,
DarkGreen,
DarkBlue,
DarkCyan,
DarkMagenta,
],
PalletEnum::RedYellow => vec![Red, DarkRed, DarkYellow, Yellow],
PalletEnum::YellowGreen => vec![Yellow, DarkYellow, DarkGreen, Green],
PalletEnum::GreenBlue => vec![Green, DarkGreen, DarkBlue, Blue],
PalletEnum::BlueCyan => vec![Blue, DarkBlue, DarkCyan, Cyan],
PalletEnum::CyanMagenta => vec![Cyan, DarkCyan, DarkMagenta, Magenta],
PalletEnum::MagentaRed => vec![Magenta, DarkMagenta, DarkRed, Red],
PalletEnum::Gray => vec![Black, DarkGrey, Grey, White],
}
}

3
src/transform/mod.rs Normal file
View File

@ -0,0 +1,3 @@
mod shift;
pub use shift::*;

25
src/transform/shift.rs Normal file
View File

@ -0,0 +1,25 @@
use crate::pattern::*;
use crate::Vector;
#[derive(derive_more::Constructor)]
pub struct ShiftFactory {
child: Box<dyn PatternFactory>,
}
#[derive(derive_more::Constructor)]
pub struct Shift {
child: Box<dyn Pattern>,
shift: f32,
}
impl PatternFactory for ShiftFactory {
fn create(&self, config: &Config) -> Box<dyn Pattern> {
Box::new(Shift::new(self.child.create(config), config.step))
}
}
impl Pattern for Shift {
fn sample(&self, pos: Vector) -> f32 {
self.child.sample(pos) + 1.0 - 2.0 * self.shift
}
}