about summary refs log tree commit diff
path: root/squash/src/main.rs
blob: d5cd2d8a75f8696582b781c0d801ad02bf813a74 (plain)
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
use colorsquash::Squasher;

use crate::cli::{InType, OutType};

mod cli;
mod image;

fn main() -> Result<(), anyhow::Error> {
	//gen: I should use clap or at least getopt, but this is fine.
	let cli = cli::get();

	let mut image = match cli.in_type {
		InType::Png => image::get_png(cli.input)?,
		InType::Jpeg => image::get_jpg(cli.input)?,
	};

	let mut squasher = Squasher::new(cli.color_count, &image.data);
	let size = squasher.map_over(&mut image.data);
	image.data.resize(size, 0);

	println!(
		"selected {} colours of max {}",
		squasher.palette().len(),
		cli.color_count
	);

	match cli.out_type {
		OutType::Png => image::save_png(image, squasher, cli.output),
		OutType::Gif => image::save_gif(image, squasher, cli.output),
	}
}