eRgo35/ascii
1
0
Fork 0
mirror of https://github.com/eRgo35/ascii.git synced 2026-02-04 04:46:09 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

optimized code a bit

Browse Source
This commit is contained in:
Michał Czyż
2024-07-07 14:13:23 +02:00
parent 0d7e950503
commit 274ceb0b83
1 changed files with 31 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

117
src/libs/ascii.rs
View File

@@ -3,86 +3,6 @@ use colored::Colorize;
use image; use image;
use image::GenericImageView; use image::GenericImageView;
fn grayscale_ascii(
img: &image::DynamicImage,
width: u32,
height: u32,
invert: bool,
pixel_size: usize,
) -> String {
let mut ascii_img = String::new();
for y in 0..height {
for x in 0..width {
let pixel = img.get_pixel(x, y);
let red = pixel[0];
let green = pixel[1];
let blue = pixel[2];
let pixel_iterator: Vec<u16> = vec![red.into(), green.into(), blue.into()];
let darkest_color: &u16 = pixel_iterator.iter().min().unwrap();
let brightest_color: &u16 = pixel_iterator.iter().max().unwrap();
let mut brightness = ((darkest_color + brightest_color) / 2) as u8;
if invert {
brightness = 255 - brightness;
}
let char_pixel = select_char(&brightness).repeat(pixel_size);
ascii_img.push_str(&char_pixel);
}
ascii_img.push('\n');
}
ascii_img
}
fn colorful_ascii(
img: &image::DynamicImage,
width: u32,
height: u32,
invert: bool,
pixel_size: usize,
) -> String {
let mut ascii_img = String::new();
for y in 0..height {
for x in 0..width {
let pixel = img.get_pixel(x, y);
let mut red = pixel[0];
let mut green = pixel[1];
let mut blue = pixel[2];
let pixel_iterator: Vec<u16> = vec![red.into(), green.into(), blue.into()];
let darkest_color: &u16 = pixel_iterator.iter().min().unwrap();
let brightest_color: &u16 = pixel_iterator.iter().max().unwrap();
let mut brightness = ((darkest_color + brightest_color) / 2) as u8;
if invert {
red = 255 - red;
green = 255 - green;
blue = 255 - blue;
brightness = 255 - brightness;
}
let mut char_pixel = select_char(&brightness).repeat(pixel_size);
char_pixel = select_dominant_color((red, green, blue), char_pixel);
ascii_img.push_str(&char_pixel);
}
ascii_img.push('\n');
}
ascii_img
}
fn select_char(brightness: &u8) -> String { fn select_char(brightness: &u8) -> String {
let char = match brightness { let char = match brightness {
0..=25 => " ", 0..=25 => " ",
@@ -103,17 +23,42 @@ fn select_char(brightness: &u8) -> String {
fn select_dominant_color(pixel: (u8, u8, u8), char_pixel: String) -> String { fn select_dominant_color(pixel: (u8, u8, u8), char_pixel: String) -> String {
let (red, green, blue) = pixel; let (red, green, blue) = pixel;
let char_pixel = char_pixel.truecolor(red, green, blue).to_string(); char_pixel.truecolor(red, green, blue).to_string()
char_pixel
} }
pub fn to_ascii(img: &image::DynamicImage, args: &args::Args) -> String { pub fn to_ascii(img: &image::DynamicImage, args: &args::Args) -> String {
let mut ascii_img = String::new();
let (width, height) = img.dimensions(); let (width, height) = img.dimensions();
if args.colorful { for y in 0..height {
return colorful_ascii(img, width, height, args.invert, args.pixel); for x in 0..width {
let pixel = img.get_pixel(x, y);
let (mut red, mut green, mut blue) = (pixel[0], pixel[1], pixel[2]);
let pixel_iterator: Vec<u16> = vec![red.into(), green.into(), blue.into()];
let darkest_color: &u16 = pixel_iterator.iter().min().unwrap();
let brightest_color: &u16 = pixel_iterator.iter().max().unwrap();
let mut brightness = ((darkest_color + brightest_color) / 2) as u8;
if args.invert {
red = 255 - red;
green = 255 - green;
blue = 255 - blue;
brightness = 255 - brightness;
}
let mut char_pixel = select_char(&brightness).repeat(args.pixel);
if args.colorful {
char_pixel = select_dominant_color((red, green, blue), char_pixel);
}
ascii_img.push_str(&char_pixel);
}
ascii_img.push('\n');
} }
grayscale_ascii(img, width, height, args.invert, args.pixel) ascii_img
} }