chuda 0.1.1

CUDA-accelerated high quality truecolour ANSI image renderer
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

mod gpu;
mod render;
mod symbols;

use std::{
    fs,
    path::{Path, PathBuf},
};

use anyhow::{Context, Result, bail};
use clap::Parser;

#[derive(Debug, Parser)]
#[command(version, about)]
struct Args {
    /// PNG file or a directory tree containing PNG files.
    input: PathBuf,

    /// Output cell width. Height is derived from image and terminal aspect ratios.
    #[arg(short = 's', long = "size", value_name = "WIDTH")]
    width: u32,

    /// Required for directory input. The input tree is mirrored here as .ansi files.
    #[arg(short, long, value_name = "DIR")]
    output: Option<PathBuf>,

    /// Cell pixel aspect ratio (terminal cells are normally twice as tall as wide).
    #[arg(long, default_value_t = 2.0)]
    font_ratio: f32,

    /// Bias toward detailed opaque cells. Edge transparency must improve the
    /// fit by roughly this fraction of an 8x8 cell to win.
    #[arg(long, default_value_t = 0.10)]
    transparent_threshold: f32,
}

fn main() -> Result<()> {
    let args = Args::parse();
    if args.width == 0 {
        bail!("--size must be greater than zero");
    }
    if !(args.font_ratio.is_finite() && args.font_ratio > 0.0) {
        bail!("--font-ratio must be a positive finite number");
    }
    if !(0.0..=1.0).contains(&args.transparent_threshold) {
        bail!("--transparent-threshold must be between 0 and 1");
    }

    if args.input.is_file() {
        let ansi = render::render_png(
            &args.input,
            args.width,
            args.font_ratio,
            args.transparent_threshold,
        )?;
        use std::io::Write;
        std::io::stdout().lock().write_all(&ansi)?;
        return Ok(());
    }
    if !args.input.is_dir() {
        bail!("input does not exist: {}", args.input.display());
    }
    let output = args
        .output
        .as_ref()
        .context("--output is required for directory input")?;
    render_tree(
        &args.input,
        output,
        args.width,
        args.font_ratio,
        args.transparent_threshold,
    )
}

fn render_tree(
    root: &Path,
    output: &Path,
    width: u32,
    font_ratio: f32,
    transparent_threshold: f32,
) -> Result<()> {
    let mut pending = vec![root.to_path_buf()];
    let mut files = Vec::new();
    while let Some(dir) = pending.pop() {
        for entry in fs::read_dir(&dir).with_context(|| format!("reading {}", dir.display()))? {
            let path = entry?.path();
            if path.is_dir() {
                pending.push(path);
            } else if path
                .extension()
                .is_some_and(|e| e.eq_ignore_ascii_case("png"))
            {
                files.push(path);
            }
        }
    }
    files.sort();
    for input in files {
        let relative = input.strip_prefix(root)?;
        let mut target = output.join(relative);
        target.set_extension("ansi");
        if let Some(parent) = target.parent() {
            fs::create_dir_all(parent)?;
        }
        let ansi = render::render_png(&input, width, font_ratio, transparent_threshold)
            .with_context(|| format!("rendering {}", input.display()))?;
        fs::write(&target, ansi).with_context(|| format!("writing {}", target.display()))?;
    }
    Ok(())
}