Compare commits

..

3 commits

Author SHA1 Message Date
bbaa094f00 fix: remove unnecessary rename_all clap parameter 2022-06-17 22:01:01 +02:00
65381889ed feat: allow multiple animations 2022-06-17 22:00:02 +02:00
75d1b93f7e fix: cleanup main() 2022-06-17 21:59:27 +02:00

View file

@ -12,7 +12,7 @@ use std::{
}; };
use bracket_color::prelude::{HSV, RGB}; use bracket_color::prelude::{HSV, RGB};
use clap::{Parser, Subcommand}; use clap::{Parser, Subcommand, ValueEnum};
use image::{imageops::FilterType, io::Reader as ImageReader, Pixel, Rgb, RgbImage}; use image::{imageops::FilterType, io::Reader as ImageReader, Pixel, Rgb, RgbImage};
use rand::Rng; use rand::Rng;
@ -35,7 +35,7 @@ struct Args {
remote_addr: SocketAddr, remote_addr: SocketAddr,
/// The action to perform /// The action to perform
#[clap(subcommand, rename_all = "kebab-case")] #[clap(subcommand)]
action: Action, action: Action,
} }
@ -80,12 +80,25 @@ impl From<Color> for Rgb<u8> {
#[derive(Clone, Debug, PartialEq, Eq, Subcommand)] #[derive(Clone, Debug, PartialEq, Eq, Subcommand)]
enum Action { enum Action {
Rainbow, Animation {
Solid { color: Color }, #[clap(value_enum)]
Image { path: PathBuf }, animation: Animation,
},
Solid {
color: Color,
},
Image {
path: PathBuf,
},
Clear, Clear,
} }
#[derive(Clone, Debug, PartialEq, Eq, ValueEnum)]
enum Animation {
Rainbow,
Bling,
}
fn bling(layout: Layout, frame: u32) -> RgbImage { fn bling(layout: Layout, frame: u32) -> RgbImage {
#![allow( #![allow(
clippy::cast_precision_loss, clippy::cast_precision_loss,
@ -181,30 +194,27 @@ fn main() -> anyhow::Result<()> {
first_strand_index: 8, first_strand_index: 8,
}; };
match args.action { let image = match args.action {
Action::Solid { color } => { Action::Solid { color } => {
let image = RgbImage::from_pixel(layout.width_px(), layout.height_px(), color.into()); RgbImage::from_pixel(layout.width_px(), layout.height_px(), color.into())
let frame_num: u32 = rand::thread_rng().gen();
send_frame(&socket, layout, frame_num, &image)?;
} }
Action::Clear => { Action::Clear => RgbImage::new(layout.width_px(), layout.height_px()),
let image = RgbImage::new(layout.width_px(), layout.height_px()); Action::Image { path } => ImageReader::open(path)?
let frame_num: u32 = rand::thread_rng().gen(); .decode()?
send_frame(&socket, layout, frame_num, &image)?; .resize_to_fill(layout.width_px(), layout.height_px(), FilterType::Gaussian)
} .into_rgb8(),
Action::Image { path } => { Action::Animation { animation } => {
let image = ImageReader::open(path)? let f = match animation {
.decode()? Animation::Rainbow => rainbow,
.resize_to_fill(layout.width_px(), layout.height_px(), FilterType::Gaussian) Animation::Bling => bling,
.into_rgb8(); };
let frame_num: u32 = rand::thread_rng().gen();
send_frame(&socket, layout, frame_num, &image)?;
}
Action::Rainbow => {
print!("{}", termion::clear::All); print!("{}", termion::clear::All);
for frame in 0.. { let mut frame = 0;
let image = rainbow(layout, frame); loop {
let start = Instant::now();
let image = f(layout, frame);
print_image(&image); print_image(&image);
@ -212,9 +222,13 @@ fn main() -> anyhow::Result<()> {
send_frame(&socket, layout, frame_num, &image)?; send_frame(&socket, layout, frame_num, &image)?;
sleep(Duration::from_millis(16)); sleep(Duration::from_millis(16));
frame += 1;
} }
} }
} };
let frame_num: u32 = rand::thread_rng().gen();
send_frame(&socket, layout, frame_num, &image)?;
Ok(()) Ok(())
} }