heatman 0.1.1

Creating heatmap image from given CSV file
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

use crate::cli::Heatman;
use clap::Parser;

use std::path::Path;
use heatman::{Heatmap, Error, Data, Result};
use image::{ImageBuffer, Rgba};

mod cli;

fn output_image(data: Data<Rgba<u8>>, pixel: usize, dest: &Path) -> Result<()> {
    let context = Heatmap::new(data, pixel);
    let image: ImageBuffer<Rgba<u8>, Vec<u8>> = context.into();
    image.save(dest)
        .map_err(Error::Image)
}

fn generate_scaler(heatman: Heatman) -> Result<()> {
    let data = heatman::ScalerBuilder::build();
    let rgbdata: Data<Rgba<u8>> = data.into();
    output_image(rgbdata, heatman.pixel(), heatman.dest())
}

fn generate_heatmap(heatman: Heatman) -> Result<()> {
    let data = heatman.load_image()?;
    let order = heatman.order(&data)?;
    let order = order.apply_assistant_line(heatman.assistant_line_gap());
    let reordered_data = data.reorder(&order);
    let rgbdata: Data<Rgba<u8>> = reordered_data.into();
    output_image(rgbdata, heatman.pixel(), heatman.dest())
}

fn generate_items<F>(heatman: Heatman, mapper: F) -> Result<()>
where
        F: Fn(&Data<f64>) -> Vec<String> 
{
    let data = heatman.load_image()?;
    let order = heatman.order(&data)?;
    let order = order.apply_assistant_line(heatman.assistant_line_gap());
    let reordered_data = data.reorder(&order);
    let items = mapper(&reordered_data);
    for item in items {
        println!("{item}");
    }
    Ok(())
}

#[cfg(debug_assertions)]
mod gencomp;

fn rs_main(args: Vec<String>) -> Result<()> {
    let heatman: Heatman = Parser::try_parse_from(args)
        .map_err(Error::Clap)?;
    let heatman = heatman.validate()?;
    match heatman.mode() {
        cli::Mode::Scaler => generate_scaler(heatman),
        cli::Mode::Heatmap => generate_heatmap(heatman),
        cli::Mode::Rows => generate_items(heatman, |data| data.row_headers()),
        cli::Mode::Columns => generate_items(heatman, |data| data.col_headers()),
        #[cfg(debug_assertions)]
        cli::Mode::GenerateCompletions => {
            gencomp::generate("heatman", Path::new("completions"));
            Ok(())
        },
    }
}

fn main() {
    let args = std::env::args().collect::<Vec<String>>();
    match rs_main(args) {
        Ok(_) => {},
        Err(e) => {
            eprintln!("Error: {}", e);
            std::process::exit(1);
        }
    }
}