chuda 0.2.2

CUDA-accelerated high quality truecolour ANSI image renderer
Documentation 
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

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

use anyhow::{Context, Result, bail};
use chuda::{Backend, RenderOptions, RenderRequest, Renderer, SourceImage};
use clap::Parser;

#[derive(Debug, Parser)]
#[command(version, about)]
struct Args {
    input: PathBuf,
    #[arg(short = 's', long = "size", value_name = "WIDTH")]
    width: u32,
    #[arg(short, long, value_name = "DIR")]
    output: Option<PathBuf>,
    #[arg(long, default_value_t = 2.0)]
    font_ratio: f32,
    #[arg(long, default_value_t = 0.10)]
    transparent_threshold: f32,
    #[arg(long, default_value = "auto")]
    backend: Backend,
}

fn main() -> Result<()> {
    let args = Args::parse();
    let options = RenderOptions {
        font_ratio: args.font_ratio,
        transparent_threshold: args.transparent_threshold,
    };
    let renderer = Renderer::new(args.backend);
    if args.input.is_file() {
        let image = SourceImage::open(&args.input)?;
        let frame = renderer.render(RenderRequest::new(&image, args.width, options))?;
        warn_fallback(&renderer);
        std::io::stdout().lock().write_all(&frame.to_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(&renderer, &args.input, output, args.width, options)
}

fn warn_fallback(renderer: &Renderer) {
    if let Some(warning) = renderer.take_fallback_warning() {
        eprintln!("warning: {warning}");
    }
}

fn render_tree(
    renderer: &Renderer,
    root: &Path,
    output: &Path,
    width: u32,
    options: RenderOptions,
) -> 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(|ext| ext.eq_ignore_ascii_case("png"))
            {
                files.push(path);
            }
        }
    }
    files.sort();
    for input in files {
        let image =
            SourceImage::open(&input).with_context(|| format!("rendering {}", input.display()))?;
        let frame = renderer.render(RenderRequest::new(&image, width, options))?;
        warn_fallback(renderer);
        let mut target = output.join(input.strip_prefix(root)?);
        target.set_extension("ansi");
        if let Some(parent) = target.parent() {
            fs::create_dir_all(parent)?;
        }
        fs::write(&target, frame.to_ansi())
            .with_context(|| format!("writing {}", target.display()))?;
    }
    Ok(())
}