diff --git a/Cargo.lock b/Cargo.lock index b4926fe..f8a30c6 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -42,6 +42,7 @@ dependencies = [ "cortex-m 0.7.4", "cortex-m-rt", "embedded-graphics", + "embedded-time", "panic-halt", "panic-rtt-target", "rtt-target 0.2.2", diff --git a/Cargo.toml b/Cargo.toml index ad1d892..cc74cea 100644 --- a/Cargo.toml +++ b/Cargo.toml @@ -14,4 +14,5 @@ stm32f3xx-hal = { version = "0.9.0", features = ["ld", "rt", "stm32f303xe"] } rtt-target = { version = "0.2.2", features = ["cortex-m"] } panic-rtt-target = { version = "0.1.1", features = ["cortex-m"] } ssd1306 = "0.7.0" -embedded-graphics = "0.7.1" \ No newline at end of file +embedded-graphics = "0.7.1" +embedded-time = "0.12.1" \ No newline at end of file diff --git a/README.org b/README.org index 1b71774..9c6a084 100644 --- a/README.org +++ b/README.org @@ -1,11 +1,29 @@ #+TITLE: bad-apple-embedded-rs +#+html:

+ +Why? Because I wanted to and because Rust rocks! + * Features +This project was built using an =STM32-F303ZE= with a simple =ssd1306= OLED display. +It draws 21x10 with (almost perfectly) stable 8 frames per second and +was designed and built to play Bad Apple, but anything else would also work. +Frame timings and length of the video can be adjusted, too high frames per second can lead to slowed down playback due to the render time being larger than the desired one. +Faster render times are being adjusted for. + +* How to use +1. Convert a video to an ASCII file, using =video-to-ascii.py= +2. Adjust frame timing in the main program (draw time per picture, width, height, total frames) +3. Flash and enjoy! + +* Input format +The screen renders video through ASCII characters. It takes in a long text file, containing all the images in a contiguous stream, which is then read in =IMAGE_LEN=-sized (height * width + newline chars) chunks in the main program. +If these parameters don't match up, the end result won't either. * Dependencies -The video-to-ascii conversion script uses [[https://github.com/ivanl-exe/image-to-ascii/][image-to-ascii]] for the conversion in the background. +The =video-to-ascii.py= conversion script uses [[https://github.com/ivanl-exe/image-to-ascii/][image-to-ascii]] for the conversion in the background. It is assumed to be located on the same height as this project and that it is build in release-mode (due to performance). * TODO -- [ ] Add README Stuff -- [ ] Rewrite ascii conversion in Rust (with an own impl of ascii-to-text) +- [ ] Fix timing in draw +- [ ] Rewrite ascii conversion in Rust or Haskell (with an own implementation of =ascii-to-text=) diff --git a/assets/bad_apple.jpg b/assets/bad_apple.jpg new file mode 100644 index 0000000..f0365d9 Binary files /dev/null and b/assets/bad_apple.jpg differ diff --git a/src/main.rs b/src/main.rs index 013e771..4d35a64 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,42 +1,47 @@ #![no_std] #![no_main] -use cortex_m::prelude::*; +use cortex_m::delay::Delay; use cortex_m_rt::entry; +use embedded_graphics::mono_font::MonoTextStyleBuilder; use embedded_graphics::{ mono_font::ascii::FONT_4X6, pixelcolor::BinaryColor, prelude::*, text::{Baseline, Text}, }; -use panic_halt as _; +use embedded_time::fixed_point::FixedPoint as _; +use hal::{i2c::I2c, timer::MonoTimer}; +use pac::{CorePeripherals, Peripherals}; +use panic_rtt_target as _; use rtt_target::{rprintln, rtt_init_print}; use ssd1306::{prelude::*, I2CDisplayInterface, Ssd1306}; -use stm32f3xx_hal::{self as hal, pac, prelude::*}; use stm32f3xx_hal::prelude::_embedded_hal_digital_InputPin; +use stm32f3xx_hal::{self as hal, pac, prelude::*}; -const IMAGE_LEN: usize = 220; // size of a single ascii image on screen -const IMAGE_END: usize = 1093 * 220; // position of the last ascii char +const IMAGE_WIDTH: usize = 21; +const IMAGE_HEIGHT: usize = 10; +const TOTAL_FRAMES: usize = 1749; +const DRAW_TIME: usize = 125; // ms for a single screen draw + +const IMAGE_LEN: usize = IMAGE_HEIGHT * (IMAGE_WIDTH + 1); // adjust for newline char +const IMAGE_END: usize = TOTAL_FRAMES * IMAGE_LEN; // position of the last ascii char #[entry] fn main() -> ! { rtt_init_print!(); - let peripherals = pac::Peripherals::take().unwrap(); - let mut core_peripherals = pac::CorePeripherals::take().unwrap(); + let peripherals = Peripherals::take().unwrap(); + let mut core_peripherals = CorePeripherals::take().unwrap(); let mut rcc = peripherals.RCC.constrain(); let mut flash = peripherals.FLASH.constrain(); let clocks = rcc.cfgr.freeze(&mut flash.acr); let mut gpiob = peripherals.GPIOB.split(&mut rcc.ahb); let mut gpioc = peripherals.GPIOC.split(&mut rcc.ahb); - let monotimer = hal::timer::MonoTimer::new( - core_peripherals.DWT, - clocks, - &mut core_peripherals.DCB); - // TODO: dont use hardcoded val here - let mut delay = cortex_m::delay::Delay::new(core_peripherals.SYST, 8000000); + let monotimer = MonoTimer::new(core_peripherals.DWT, clocks, &mut core_peripherals.DCB); + let mut delay_provider = Delay::new(core_peripherals.SYST, clocks.hclk().integer()); - let mut button1 = gpioc + let button1 = gpioc .pc13 .into_pull_down_input(&mut gpioc.moder, &mut gpioc.pupdr); @@ -47,7 +52,7 @@ fn main() -> ! { .pb9 .into_af_open_drain(&mut gpiob.moder, &mut gpiob.otyper, &mut gpiob.afrh); - let i2c = hal::i2c::I2c::new( + let i2c = I2c::new( peripherals.I2C1, (scl, sda), 1000.kHz().try_into().unwrap(), @@ -60,40 +65,47 @@ fn main() -> ! { .into_buffered_graphics_mode(); display.init().unwrap(); - let text_style = embedded_graphics::mono_font::MonoTextStyleBuilder::new() + let text_style = MonoTextStyleBuilder::new() .font(&FONT_4X6) .text_color(BinaryColor::On) .build(); - let bad_apple = core::str::from_utf8(include_bytes!("../assets/gen.txt")).unwrap(); + let ascii_txt = core::str::from_utf8(include_bytes!("../assets/ascii.txt")).unwrap(); // start indexing at 0, draw IMAGE_LEN ascii chars to display let mut index: usize = 0; loop { - let instant = monotimer.now(); + let monotimer_instant = monotimer.now(); + display.clear(); - rprintln!("draw: {} to {}", index, index + IMAGE_LEN); let text = Text::with_baseline( - &bad_apple[index..index + IMAGE_LEN], + &ascii_txt[index..index + IMAGE_LEN], Point::new(0, 0), text_style, Baseline::Top, ); text.draw(&mut display).unwrap(); + // draw out to physical screen display.flush().unwrap(); // go to next frame or reset to start index = (index + IMAGE_LEN) % IMAGE_END; - let elapsed = instant.elapsed(); - rprintln!("draw took: {}", elapsed); - rprintln!("freq: {}", monotimer.frequency()); - delay.delay_ms(100); + // adjust for desired framerate + // TODO: chip is too fast for some reason + let freq = monotimer.frequency().0; + // maybe use float calc here? + let ms_per_draw = 1000 / (freq / monotimer_instant.elapsed()); + rprintln!("Draw took {}ms", ms_per_draw); + match DRAW_TIME.checked_sub(ms_per_draw as usize) { + Some(result) => delay_provider.delay_ms(result as u32), + None => (), + }; - // kill switch + // reset if button1.is_high().unwrap() { - loop {} + index = 0 } } } diff --git a/video-to-ascii.py b/video-to-ascii.py index bcdf4bc..73200a1 100644 --- a/video-to-ascii.py +++ b/video-to-ascii.py @@ -1,15 +1,28 @@ """ -Convert a video to a txt in a format, which i need +Convert a video to a textfile in a format, which i need: + +ASCII_TXT: +--- +IMAGE +IMAGE +IMAGE +... +--- + Using: https://github.com/ivanl-exe/image-to-ascii/ """ + import cv2 import os import shutil import argparse + OUTPUT = 'assets/output' ASCII_TXT = 'assets/ascii.txt' +TOOL_PATH = '../image-to-ascii/target/release/image_to_ascii' + def parse_args(): """ @@ -22,6 +35,7 @@ def parse_args(): return parser.parse_args() + def main(): """ Convert frames to jpgs, convert each to ascii, and append it to the ascii.txt @@ -59,11 +73,21 @@ def main(): for f in os.listdir(OUTPUT): filename = os.path.abspath(OUTPUT + '/' + f) print(f'Processing: {filename}') - os.system(f'../image-to-ascii/target/release/image_to_ascii {args.width} {args.height} true {filename} >> {ASCII_TXT}') + os.system(f'{TOOL_PATH} {args.width} {args.height} true {filename} >> {ASCII_TXT}') # cleanup video.release() cv2.destroyAllWindows() + +def check_deps(): + """ + Checks if the deps, used in this script, are present + """ + if not os.path.isfile(TOOL_PATH): + print("Missing image-to-ascii at expected location!") + exit(1) + if __name__ == '__main__': + check_deps() main()