mcq 0.1.0

A Rust port of Java implementation of Median Cut Quantization algorithm.
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

#[macro_use]
extern crate image;
extern crate mcq;

use std::fs::*;
use std::path::*;
use std::io::BufReader;

use mcq::MMCQ;

const COLOR_HEIGHT: u32 = 64;
const QUANT_SIZE: u32 = 16;

fn main() {
    let paths = read_dir("./examples/res");

    for path in paths.unwrap() {
        let path = path.unwrap().path();
        let path = path.to_str().unwrap();
        process_image(path);
    }

    println!("\nPlease visit 'target' folder for the results");
}

fn process_image(file: &str) {
    println!("Reading image {}", file);

    let mcq = {
        let img = image::load(BufReader::new(File::open(file).unwrap()), image::JPEG).unwrap().to_rgba();
        let data = img.into_vec();

        // Here we extract the quantized colors from the image.
        // We need no more than 16 colors (QUANT_SIZE).
        MMCQ::from_pixels_u8_rgba(data.as_slice(), QUANT_SIZE)
    };

    // A `Vec` of colors, descendantely sorted by usage frequency
    let qc = mcq.get_quantized_colors();
    // println!("Quantized {:?}", qc);

    // =============================================================================================
    // Here we will demonstrate the extracted colors by generating the image
    // that consists of both original image and a resulted palette.
    // =============================================================================================

    let img = image::load(BufReader::new(File::open(file).unwrap()), image::JPEG).unwrap().to_rgba();
    let (ix, iy) = img.dimensions();

    let mut imgbuf = image::ImageBuffer::new(ix, iy + COLOR_HEIGHT);

    for x in 0..ix {
        for y in 0..iy {
            imgbuf.put_pixel(x, y, img.get_pixel(x, y).clone());
        }
    }

    let color_width = ix / QUANT_SIZE;

    for y in (iy + 1)..(iy + COLOR_HEIGHT) {
        for x0 in 0..QUANT_SIZE {
            let x1 = x0 * color_width;
            let q = qc[x0 as usize];

            for x2 in 0..color_width {
                imgbuf.put_pixel(x1 + x2, y, image::Rgba([q.red, q.grn, q.blu, 0xff]));
            }
        }
    }

    let ref mut fout = File::create(format!("./target/{}.png",
                                            Path::new(file).file_name().unwrap().to_str().unwrap())
            .as_str())
        .unwrap();

    let _ = image::ImageRgba8(imgbuf).save(fout, image::PNG);
}