From 90814bea8ff04fa30c4950a12404df57baa8bbd6 Mon Sep 17 00:00:00 2001 From: Juergen Stuber Date: Sun, 3 Nov 2024 19:25:14 +0100 Subject: [PATCH] Add primes --- Cargo.toml | 1 + README.txt | 13 ++- src/bin/primes/main.rs | 216 +++++++++++++++++++++++++++++++++++++++++ 3 files changed, 225 insertions(+), 5 deletions(-) create mode 100644 src/bin/primes/main.rs diff --git a/Cargo.toml b/Cargo.toml index 9c4b07a..bca5b06 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -14,3 +14,4 @@ edition = "2021" rand = "0.8.4" chrono = "0.4.6" lowdim = { version = "0.6.0", features = ["random"] } +primal = "0.3.3" diff --git a/README.txt b/README.txt index a7006fb..5ae7ddd 100644 --- a/README.txt +++ b/README.txt @@ -8,22 +8,25 @@ bimood - color cycling mood lamp with two complementary colors parameter is the seconds for one cycle through the colors predprey - grass-prey-predator simulation - parameter is number of actions per 40ms frame + parameter is the number of actions per 40ms frame maze - draws a maze by drawing corridors - parameter is number of actions per 40ms frame + parameter is the number of actions per 40ms frame dualmaze - draws a maze by drawing walls - parameter is number of actions per 40ms frame + parameter is the number of actions per 40ms frame life - game of life - parameter is number of actions per second + parameter is the number of actions per second matrix-code - the falling code animation from Matrix parameter is not used yet lorenz - a Lorenz attractor - parameter is number of actions per 40ms frame + parameter is the number of actions per 40ms frame + +primes - display primes as rod numerals (hommage to Rune Mields) + parameter is the number of pixels scrolled per second Compile with "cargo build --release". diff --git a/src/bin/primes/main.rs b/src/bin/primes/main.rs new file mode 100644 index 0000000..4e6054b --- /dev/null +++ b/src/bin/primes/main.rs @@ -0,0 +1,216 @@ +use std::env::args; +use std::io::stdout; +use std::io::Write; +use std::thread::sleep; +use std::time::Duration; + +use lowdim::bb2d; +use lowdim::p2d; +use lowdim::v2d; +use lowdim::Array2d; +use lowdim::BBox2d; +use lowdim::Point2d; + +const COLORS: usize = 4; + +#[derive(Clone, Copy, Debug, PartialEq, Eq)] +enum Pixel { + Background, + Foreground, +} + +#[derive(Clone, Debug)] +struct Frame { + pixels: Array2d, +} +impl Frame { + pub fn new(bbox: BBox2d) -> Frame { + let pixels = Array2d::with(bbox, |_| Pixel::Background); + Frame { pixels } + } + pub fn bbox(&self) -> BBox2d { + self.pixels.bbox() + } + pub fn set(&mut self, pos: Point2d, pixel: Pixel) { + if self.bbox().contains(&pos) { + self.pixels[pos] = pixel; + } + } +} + +fn send(w: &mut T, frame: &Frame) -> std::io::Result<()> { + for y in frame.bbox().y_range().rev() { + for x in frame.bbox().x_range() { + let pixel = frame.pixels[p2d(x, y)]; + let white = match pixel { + Pixel::Background => 170, + Pixel::Foreground => 0, + }; + let rgbw: &[u8] = &[0, 0, 0, white]; + w.write_all(rgbw)?; + } + } + w.flush() +} + +fn encode_digit(d: u32) -> (i64, bool) { + if d <= 5 { + (i64::from(d), false) + } else { + (i64::from(d) - 5, true) + } +} + +fn render_zero(frame: &mut Frame, pos: &mut Point2d) { + for v in &[ + v2d(1, 0), + v2d(2, 0), + v2d(3, 0), + v2d(1, 4), + v2d(2, 4), + v2d(3, 4), + v2d(0, 1), + v2d(0, 2), + v2d(0, 3), + v2d(4, 1), + v2d(4, 2), + v2d(4, 3), + ] { + frame.set(*pos + v, Pixel::Foreground); + } + *pos += v2d(5, 0); +} + +const STROKE_LEN: i64 = 11; +const STROKE_MID: i64 = 5; + +fn render_vertical(frame: &mut Frame, pos: &mut Point2d, digit: u32) { + if digit == 0 { + render_zero(frame, pos); + } else { + let (ones, five) = encode_digit(digit); + for i in 0..ones { + for j in 0..STROKE_LEN { + frame.set(*pos + v2d(2 * i, j), Pixel::Foreground); + } + } + if five { + for j in 0..STROKE_LEN { + frame.set(*pos + v2d(ones - 1 - STROKE_MID + j, 11), Pixel::Foreground); + } + } + *pos += v2d(2 * ones - 1, 0); + } +} + +fn render_horizontal(frame: &mut Frame, pos: &mut Point2d, digit: u32) { + if digit == 0 { + render_zero(frame, pos); + } else { + let (ones, five) = encode_digit(digit); + for i in 0..ones { + for j in 0..STROKE_LEN { + frame.set(*pos + v2d(j, 2 * i), Pixel::Foreground); + } + } + if five { + for j in (2 * ones - 1)..STROKE_LEN { + frame.set(*pos + v2d(STROKE_MID, j), Pixel::Foreground); + } + } + *pos += v2d(STROKE_LEN, 0); + } +} + +fn render(frame: &mut Frame, pos: &mut Point2d, n: usize) { + let s = n.to_string(); + let mut vertical = s.len() % 2 != 0; + for c in s.chars() { + let digit = c.to_digit(10).unwrap(); + if vertical { + render_vertical(frame, pos, digit); + } else { + render_horizontal(frame, pos, digit); + } + vertical = !vertical; + } +} + +const DEFAULT_ARG: u64 = 4; + +fn main() -> std::io::Result<()> { + let args = args().collect::>(); + eprintln!("executing {}", args[0]); + + let x_size = args[1].parse::().unwrap(); + let y_size = args[2].parse::().unwrap(); + let arg = if let Some(s) = args.get(3) { + s.parse::().unwrap_or(DEFAULT_ARG) + } else { + DEFAULT_ARG + }; + eprintln!("screen size {}x{}, arg {}", x_size, y_size, arg); + + let bbox = bb2d(0..x_size as i64, 0..y_size as i64); + + let mut frame = Frame::new(bbox); + + let t_frame = (1000 / arg).max(50); // s + let delay = Duration::from_millis(t_frame); + + // Start position for rendering (lower left pixel). + let y_mid = (bbox.y_min() + bbox.y_max()) / 2; + let mut start_pos = p2d(bbox.x_end(), y_mid - 6); + + let mut primes_iter = primal::Primes::all(); + let mut visible_primes = Vec::new(); + + let space = v2d(2, 0); + loop { + let mut new_frame = Frame::new(bbox); + let mut new_visible_primes = Vec::new(); + + let mut pos = start_pos; + + // Render the primes that are already visible. + for p in visible_primes { + render(&mut new_frame, &mut pos, p); + if pos.x() + STROKE_MID < frame.bbox().x_start() { + // The prime is not visible any more, + // not even some overhang from a five bar. + // Omit it from the visible primes and move the start position + // to the start of the next number. + start_pos = pos + space; + } else { + // Keep the prime for displaying it the next time. + new_visible_primes.push(p); + } + + // Render a space between numbers. + pos += space; + } + + // Fill up the visible primes when necessary. + while pos.x() < frame.bbox().x_end() + STROKE_MID { + let p = primes_iter.next().unwrap(); + eprintln!("{}", p); + + // Keep the prime for displaying it the next time. + new_visible_primes.push(p); + + render(&mut new_frame, &mut pos, p); + } + + // Scroll one pixel to the left. + start_pos -= v2d(1, 0); + + frame = new_frame; + visible_primes = new_visible_primes; + + let mut buf = Vec::with_capacity(x_size * y_size * COLORS); + send(&mut buf, &frame)?; + stdout().write_all(&buf)?; + stdout().flush()?; + sleep(delay); + } +}