feat: allow smaller images

src/main.rs

@@ -5,6 +5,7 @@ use std::{
     io::stdout,
     net::{Ipv4Addr, SocketAddr, UdpSocket},
     num::ParseIntError,
+    path::PathBuf,
     str::FromStr,
     thread::sleep,
     time::Duration,
@@ -12,7 +13,7 @@ use std::{
 
 use bracket_color::prelude::{HSV, RGB};
 use clap::{Parser, Subcommand};
-use image::{Pixel, Rgb, RgbImage};
+use image::{imageops::FilterType, io::Reader as ImageReader, Pixel, Rgb, RgbImage};
 use rand::Rng;
 
 use splink_client::{send_frame, Layout};
@@ -81,10 +82,11 @@ impl From<Color> for Rgb<u8> {
 enum Action {
     Rainbow,
     Solid { color: Color },
+    Image { path: PathBuf },
     Clear,
 }
 
-fn get_frame(layout: Layout, frame: u32) -> RgbImage {
+fn bling(layout: Layout, frame: u32) -> RgbImage {
     #![allow(
         clippy::cast_precision_loss,
         clippy::cast_lossless,
@@ -190,6 +192,14 @@ fn main() -> anyhow::Result<()> {
             let frame_num: u32 = rand::thread_rng().gen();
             send_frame(&socket, layout, frame_num, &image)?;
         }
+        Action::Image { path } => {
+            let image = ImageReader::open(path)?
+                .decode()?
+                .resize_to_fill(layout.width_px(), layout.height_px(), FilterType::Gaussian)
+                .into_rgb8();
+            let frame_num: u32 = rand::thread_rng().gen();
+            send_frame(&socket, layout, frame_num, &image)?;
+        }
         Action::Rainbow => {
             print!("{}", termion::clear::All);
 

src/strandifier.rs

@@ -23,6 +23,9 @@ pub struct Strandifier<'a> {
     pixels_remaining: u32,
     next_x: u32,
     next_y: u32,
+
+    offset_x: u32,
+    offset_y: u32,
 }
 
 impl<'a> Strandifier<'a> {
@@ -41,7 +44,7 @@ impl<'a> Strandifier<'a> {
         image: &'a RgbImage,
         strand_num: u32,
     ) -> Result<Self, StrandifierError> {
-        if layout.width_px() != image.width() || layout.height_px() != image.height() {
+        if layout.width_px() < image.width() || layout.height_px() < image.height() {
             return Err(StrandifierError::WrongDimensions);
         }
 
@@ -55,6 +58,9 @@ impl<'a> Strandifier<'a> {
         let first_x = panel_x * layout.gang_len;
         let first_y = panel_y * layout.num_gangs;
 
+        let offset_x = (layout.width_px() - image.width()) / 2;
+        let offset_y = (layout.height_px() - image.height()) / 2;
+
         Ok(Self {
             layout,
             image,
@@ -62,6 +68,9 @@ impl<'a> Strandifier<'a> {
             pixels_remaining: layout.strand_len(),
             next_x: first_x,
             next_y: first_y,
+
+            offset_x,
+            offset_y,
         })
     }
 }
@@ -98,7 +107,7 @@ impl<'a> Iterator for Strandifier<'a> {
         }
         self.pixels_remaining -= 1;
 
-        Some(*self.image.get_pixel(x, y))
+        Some(*self.image.get_pixel(x + self.offset_x, y + self.offset_y))
     }
 
     fn size_hint(&self) -> (usize, Option<usize>) {