first commit
This commit is contained in:
Sander Hautvast
parent
ba33aba577
commit
ec636f160b
8 changed files with 711 additions and 0 deletions
3
.gitignore
vendored
Normal file
3
.gitignore
vendored
Normal file
|
|
@ -0,0 +1,3 @@
|
|||
/target
|
||||
/.DS_Store
|
||||
/.vscode
|
||||
331
Cargo.lock
generated
Normal file
331
Cargo.lock
generated
Normal file
|
|
@ -0,0 +1,331 @@
|
|||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 3
|
||||
|
||||
[[package]]
|
||||
name = "adler"
|
||||
version = "1.0.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "f26201604c87b1e01bd3d98f8d5d9a8fcbb815e8cedb41ffccbeb4bf593a35fe"
|
||||
|
||||
[[package]]
|
||||
name = "adler32"
|
||||
version = "1.2.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "aae1277d39aeec15cb388266ecc24b11c80469deae6067e17a1a7aa9e5c1f234"
|
||||
|
||||
[[package]]
|
||||
name = "autocfg"
|
||||
version = "1.0.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "cdb031dd78e28731d87d56cc8ffef4a8f36ca26c38fe2de700543e627f8a464a"
|
||||
|
||||
[[package]]
|
||||
name = "bitflags"
|
||||
version = "1.3.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
|
||||
|
||||
[[package]]
|
||||
name = "bytemuck"
|
||||
version = "1.7.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "439989e6b8c38d1b6570a384ef1e49c8848128f5a97f3914baef02920842712f"
|
||||
|
||||
[[package]]
|
||||
name = "byteorder"
|
||||
version = "1.4.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "14c189c53d098945499cdfa7ecc63567cf3886b3332b312a5b4585d8d3a6a610"
|
||||
|
||||
[[package]]
|
||||
name = "cfg-if"
|
||||
version = "1.0.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
|
||||
|
||||
[[package]]
|
||||
name = "color_quant"
|
||||
version = "1.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3d7b894f5411737b7867f4827955924d7c254fc9f4d91a6aad6b097804b1018b"
|
||||
|
||||
[[package]]
|
||||
name = "crc32fast"
|
||||
version = "1.3.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "738c290dfaea84fc1ca15ad9c168d083b05a714e1efddd8edaab678dc28d2836"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-channel"
|
||||
version = "0.5.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "06ed27e177f16d65f0f0c22a213e17c696ace5dd64b14258b52f9417ccb52db4"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"crossbeam-utils",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-deque"
|
||||
version = "0.8.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "6455c0ca19f0d2fbf751b908d5c55c1f5cbc65e03c4225427254b46890bdde1e"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"crossbeam-epoch",
|
||||
"crossbeam-utils",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-epoch"
|
||||
version = "0.9.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "4ec02e091aa634e2c3ada4a392989e7c3116673ef0ac5b72232439094d73b7fd"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"crossbeam-utils",
|
||||
"lazy_static",
|
||||
"memoffset",
|
||||
"scopeguard",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "crossbeam-utils"
|
||||
version = "0.8.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d82cfc11ce7f2c3faef78d8a684447b40d503d9681acebed6cb728d45940c4db"
|
||||
dependencies = [
|
||||
"cfg-if",
|
||||
"lazy_static",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "deflate"
|
||||
version = "0.8.6"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "73770f8e1fe7d64df17ca66ad28994a0a623ea497fa69486e14984e715c5d174"
|
||||
dependencies = [
|
||||
"adler32",
|
||||
"byteorder",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "either"
|
||||
version = "1.6.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e78d4f1cc4ae33bbfc157ed5d5a5ef3bc29227303d595861deb238fcec4e9457"
|
||||
|
||||
[[package]]
|
||||
name = "gif"
|
||||
version = "0.11.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c3a7187e78088aead22ceedeee99779455b23fc231fe13ec443f99bb71694e5b"
|
||||
dependencies = [
|
||||
"color_quant",
|
||||
"weezl",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "hermit-abi"
|
||||
version = "0.1.19"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "62b467343b94ba476dcb2500d242dadbb39557df889310ac77c5d99100aaac33"
|
||||
dependencies = [
|
||||
"libc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "image"
|
||||
version = "0.23.14"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "24ffcb7e7244a9bf19d35bf2883b9c080c4ced3c07a9895572178cdb8f13f6a1"
|
||||
dependencies = [
|
||||
"bytemuck",
|
||||
"byteorder",
|
||||
"color_quant",
|
||||
"gif",
|
||||
"jpeg-decoder",
|
||||
"num-iter",
|
||||
"num-rational",
|
||||
"num-traits",
|
||||
"png",
|
||||
"scoped_threadpool",
|
||||
"tiff",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "jpeg-decoder"
|
||||
version = "0.1.22"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "229d53d58899083193af11e15917b5640cd40b29ff475a1fe4ef725deb02d0f2"
|
||||
dependencies = [
|
||||
"rayon",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "lazy_static"
|
||||
version = "1.4.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
|
||||
|
||||
[[package]]
|
||||
name = "libc"
|
||||
version = "0.2.112"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1b03d17f364a3a042d5e5d46b053bbbf82c92c9430c592dd4c064dc6ee997125"
|
||||
|
||||
[[package]]
|
||||
name = "memoffset"
|
||||
version = "0.6.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "5aa361d4faea93603064a027415f07bd8e1d5c88c9fbf68bf56a285428fd79ce"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "miniz_oxide"
|
||||
version = "0.3.7"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "791daaae1ed6889560f8c4359194f56648355540573244a5448a83ba1ecc7435"
|
||||
dependencies = [
|
||||
"adler32",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "miniz_oxide"
|
||||
version = "0.4.4"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "a92518e98c078586bc6c934028adcca4c92a53d6a958196de835170a01d84e4b"
|
||||
dependencies = [
|
||||
"adler",
|
||||
"autocfg",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "num-integer"
|
||||
version = "0.1.44"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d2cc698a63b549a70bc047073d2949cce27cd1c7b0a4a862d08a8031bc2801db"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
"num-traits",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "num-iter"
|
||||
version = "0.1.42"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "b2021c8337a54d21aca0d59a92577a029af9431cb59b909b03252b9c164fad59"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
"num-integer",
|
||||
"num-traits",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "num-rational"
|
||||
version = "0.3.2"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "12ac428b1cb17fce6f731001d307d351ec70a6d202fc2e60f7d4c5e42d8f4f07"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
"num-integer",
|
||||
"num-traits",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "num-traits"
|
||||
version = "0.2.14"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9a64b1ec5cda2586e284722486d802acf1f7dbdc623e2bfc57e65ca1cd099290"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "num_cpus"
|
||||
version = "1.13.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "05499f3756671c15885fee9034446956fff3f243d6077b91e5767df161f766b3"
|
||||
dependencies = [
|
||||
"hermit-abi",
|
||||
"libc",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "octo"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"image",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "png"
|
||||
version = "0.16.8"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "3c3287920cb847dee3de33d301c463fba14dda99db24214ddf93f83d3021f4c6"
|
||||
dependencies = [
|
||||
"bitflags",
|
||||
"crc32fast",
|
||||
"deflate",
|
||||
"miniz_oxide 0.3.7",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rayon"
|
||||
version = "1.5.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "c06aca804d41dbc8ba42dfd964f0d01334eceb64314b9ecf7c5fad5188a06d90"
|
||||
dependencies = [
|
||||
"autocfg",
|
||||
"crossbeam-deque",
|
||||
"either",
|
||||
"rayon-core",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "rayon-core"
|
||||
version = "1.9.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d78120e2c850279833f1dd3582f730c4ab53ed95aeaaaa862a2a5c71b1656d8e"
|
||||
dependencies = [
|
||||
"crossbeam-channel",
|
||||
"crossbeam-deque",
|
||||
"crossbeam-utils",
|
||||
"lazy_static",
|
||||
"num_cpus",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "scoped_threadpool"
|
||||
version = "0.1.9"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "1d51f5df5af43ab3f1360b429fa5e0152ac5ce8c0bd6485cae490332e96846a8"
|
||||
|
||||
[[package]]
|
||||
name = "scopeguard"
|
||||
version = "1.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d29ab0c6d3fc0ee92fe66e2d99f700eab17a8d57d1c1d3b748380fb20baa78cd"
|
||||
|
||||
[[package]]
|
||||
name = "tiff"
|
||||
version = "0.6.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "9a53f4706d65497df0c4349241deddf35f84cee19c87ed86ea8ca590f4464437"
|
||||
dependencies = [
|
||||
"jpeg-decoder",
|
||||
"miniz_oxide 0.4.4",
|
||||
"weezl",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "weezl"
|
||||
version = "0.1.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "d8b77fdfd5a253be4ab714e4ffa3c49caf146b4de743e97510c0656cf90f1e8e"
|
||||
12
Cargo.toml
Normal file
12
Cargo.toml
Normal file
|
|
@ -0,0 +1,12 @@
|
|||
[package]
|
||||
name = "octo"
|
||||
version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
|
||||
|
||||
[dependencies]
|
||||
image="0.23.14"
|
||||
|
||||
[profile.release]
|
||||
debug = true
|
||||
6
README.md
Normal file
6
README.md
Normal file
|
|
@ -0,0 +1,6 @@
|
|||
Does quantization in pure rust on Rgba image, using https://crates.io/crates/image
|
||||
|
||||
algorithm:
|
||||
https://observablehq.com/@tmcw/octree-color-quantization
|
||||
|
||||
inspired by to https://github.com/objectProfessionals/movieMaps
|
||||
359
src/lib.rs
Normal file
359
src/lib.rs
Normal file
|
|
@ -0,0 +1,359 @@
|
|||
use std::{cell::RefCell, rc::Rc};
|
||||
|
||||
use image::{Rgba, RgbaImage};
|
||||
|
||||
const MAX_LEVEL: usize = 5;
|
||||
|
||||
pub fn quantize(image: &RgbaImage, num_colors:usize) -> RgbaImage{
|
||||
let mut quantizer=OctTreeQuantizer::new(num_colors);
|
||||
quantizer.quantize(image)
|
||||
}
|
||||
|
||||
struct OctTreeQuantizer {
|
||||
root: Rc<RefCell<OctTreeNode>>,
|
||||
reduce_colors: usize,
|
||||
maximum_colors: usize,
|
||||
colors: usize,
|
||||
color_list: Vec<Vec<Option<Rc<RefCell<OctTreeNode>>>>>,
|
||||
}
|
||||
|
||||
impl OctTreeQuantizer {
|
||||
fn new(num_colors: usize) -> Self {
|
||||
let mut new_quantizer = Self {
|
||||
root: Rc::new(RefCell::new(OctTreeNode::new())),
|
||||
reduce_colors: usize::max(512, num_colors * 2),
|
||||
maximum_colors: num_colors,
|
||||
color_list: vec![],
|
||||
colors: 0,
|
||||
};
|
||||
|
||||
for _ in 0..=MAX_LEVEL {
|
||||
new_quantizer.color_list.push(Vec::new());
|
||||
}
|
||||
new_quantizer
|
||||
}
|
||||
|
||||
pub fn quantize(&mut self, image: &RgbaImage) -> RgbaImage {
|
||||
for y in 0..image.height() {
|
||||
for x in 0..image.width() {
|
||||
self.insert_color(image.get_pixel(x, y), Rc::clone(&self.root));
|
||||
|
||||
if self.colors > self.reduce_colors {
|
||||
self.reduce_tree(self.reduce_colors);
|
||||
//reduce sets to None and the code below actually removes nodes from the list
|
||||
for level in &mut self.color_list {
|
||||
level.retain(|c| c.is_some());
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
let table = self.build_color_table();
|
||||
|
||||
let mut out = RgbaImage::new(image.width(), image.height());
|
||||
for y in 0..image.height() {
|
||||
for x in 0..image.width() {
|
||||
let pixel = image.get_pixel(x, y);
|
||||
if let Some(index) = self.get_index_for_color(pixel, &self.root) {
|
||||
let color = table.get(index).unwrap();
|
||||
if let Some(color) = color {
|
||||
out.put_pixel(x, y, *color);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
out
|
||||
}
|
||||
|
||||
fn get_index_for_color(
|
||||
&self,
|
||||
color: &Rgba<u8>,
|
||||
node: &Rc<RefCell<OctTreeNode>>,
|
||||
) -> Option<usize> {
|
||||
fn get_index_for_color(
|
||||
quantizer: &OctTreeQuantizer,
|
||||
color: &Rgba<u8>,
|
||||
level: usize,
|
||||
node: &Rc<RefCell<OctTreeNode>>,
|
||||
) -> Option<usize> {
|
||||
if level > MAX_LEVEL {
|
||||
return None;
|
||||
}
|
||||
let node = Rc::clone(node);
|
||||
let index = get_bitmask(color, &level);
|
||||
|
||||
let node_b = node.borrow();
|
||||
let child = &node_b.leaf[index];
|
||||
|
||||
if let Some(child) = child {
|
||||
let child_b = child.borrow();
|
||||
if child_b.is_leaf {
|
||||
return Some(child_b.index);
|
||||
} else {
|
||||
return get_index_for_color(quantizer, color, level + 1, child);
|
||||
}
|
||||
} else {
|
||||
return Some(node_b.index);
|
||||
}
|
||||
}
|
||||
|
||||
get_index_for_color(&self, color, 0, node)
|
||||
}
|
||||
|
||||
fn build_color_table(&mut self) -> Vec<Option<Rgba<u8>>> {
|
||||
//nested function that is called recursively
|
||||
fn build_color_table(
|
||||
quantizer: &mut OctTreeQuantizer,
|
||||
node: &Rc<RefCell<OctTreeNode>>,
|
||||
table: &mut Vec<Option<Rgba<u8>>>,
|
||||
index: usize,
|
||||
) -> usize {
|
||||
if quantizer.colors > quantizer.maximum_colors {
|
||||
quantizer.reduce_tree(quantizer.maximum_colors);
|
||||
}
|
||||
if node.borrow().is_leaf {
|
||||
{
|
||||
let node = node.borrow();
|
||||
let count = node.count;
|
||||
table[index] = Some(Rgba::from([
|
||||
(node.total_red / count as u32) as u8,
|
||||
(node.total_green / count as u32) as u8,
|
||||
(node.total_blue / count as u32) as u8,
|
||||
0xFF,
|
||||
]));
|
||||
}
|
||||
node.borrow_mut().index = index;
|
||||
|
||||
index + 1
|
||||
} else {
|
||||
let mut result = index;
|
||||
for i in 0..8 {
|
||||
// cannot iterate leaf, because that widens the scope of the borrow (of node)
|
||||
let mut node = node.borrow_mut();
|
||||
if let Some(leaf) = &node.leaf[i] {
|
||||
//could be immutable borrow
|
||||
let new_index = build_color_table(quantizer, leaf, table, result);
|
||||
node.index = index; //but also need mutable borrow here
|
||||
result = new_index;
|
||||
}
|
||||
}
|
||||
|
||||
result
|
||||
}
|
||||
}
|
||||
|
||||
let mut table: Vec<Option<Rgba<u8>>> = vec![None; self.colors];
|
||||
let node = Rc::clone(&self.root);
|
||||
build_color_table(self, &node, &mut table, 0);
|
||||
table
|
||||
}
|
||||
|
||||
fn insert_color(&mut self, rgb: &Rgba<u8>, node: Rc<RefCell<OctTreeNode>>) {
|
||||
//nested function that is called recursively
|
||||
fn insert_color(
|
||||
quantizer: &mut OctTreeQuantizer,
|
||||
color: &Rgba<u8>,
|
||||
level: usize,
|
||||
node: Rc<RefCell<OctTreeNode>>,
|
||||
) {
|
||||
if level > MAX_LEVEL {
|
||||
return;
|
||||
}
|
||||
|
||||
let index = get_bitmask(color, &level);
|
||||
|
||||
if node.borrow().leaf[index].is_none() {
|
||||
let mut child = OctTreeNode::new();
|
||||
child.parent = Some(Rc::clone(&node));
|
||||
child.p_index = quantizer.color_list[level].len();
|
||||
|
||||
if level == MAX_LEVEL {
|
||||
child.is_leaf = true;
|
||||
child.count = 1;
|
||||
child.total_red = color[0] as u32;
|
||||
child.total_green = color[1] as u32;
|
||||
child.total_blue = color[2] as u32;
|
||||
child.level = level;
|
||||
quantizer.colors += 1;
|
||||
}
|
||||
|
||||
let child = Rc::new(RefCell::new(child));
|
||||
quantizer.color_list[level].push(Some(Rc::clone(&child)));
|
||||
let clone = Rc::clone(&child);
|
||||
|
||||
{
|
||||
let mut mutnode = node.borrow_mut();
|
||||
mutnode.children += 1;
|
||||
mutnode.is_leaf = false;
|
||||
mutnode.leaf[index] = Some(child);
|
||||
}
|
||||
|
||||
if level < MAX_LEVEL {
|
||||
insert_color(quantizer, color, level + 1, clone);
|
||||
} else {
|
||||
return;
|
||||
}
|
||||
} else {
|
||||
if node
|
||||
.borrow()
|
||||
.leaf
|
||||
.get(index)
|
||||
.unwrap()
|
||||
.as_ref()
|
||||
.unwrap()
|
||||
.borrow()
|
||||
.is_leaf
|
||||
{
|
||||
let mut node = node.borrow_mut();
|
||||
let mut child = node
|
||||
.leaf
|
||||
.get_mut(index)
|
||||
.unwrap()
|
||||
.as_ref()
|
||||
.unwrap()
|
||||
.borrow_mut();
|
||||
child.count += 1;
|
||||
child.total_red += color[0] as u32;
|
||||
child.total_green += color[1] as u32;
|
||||
child.total_blue += color[2] as u32;
|
||||
return;
|
||||
} else {
|
||||
insert_color(
|
||||
quantizer,
|
||||
color,
|
||||
level + 1,
|
||||
Rc::clone(&(node.borrow().leaf[index]).as_ref().unwrap()),
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
insert_color(self, rgb, 0, node);
|
||||
}
|
||||
|
||||
fn reduce_tree(&mut self, num_colors: usize) {
|
||||
// Nested function that is called recursively
|
||||
fn reduce_tree(quantizer: &mut OctTreeQuantizer, num_colors: usize, level: isize) {
|
||||
if level < 0 {
|
||||
return;
|
||||
} else {
|
||||
let mut removals = Vec::new();
|
||||
let list = &quantizer.color_list[level as usize];
|
||||
for node in list {
|
||||
if let Some(node) = node {
|
||||
if node.borrow().children > 0 {
|
||||
for i in 0..8 {
|
||||
let mut color: Option<(usize, u32, u32, u32, usize)> = None;
|
||||
|
||||
if let Some(child) = node.borrow().leaf.get(i) {
|
||||
if let Some(child) = child {
|
||||
let child = child.borrow();
|
||||
color = Some((
|
||||
child.count,
|
||||
child.total_red,
|
||||
child.total_green,
|
||||
child.total_blue,
|
||||
child.p_index,
|
||||
));
|
||||
}
|
||||
}
|
||||
|
||||
// need to mutate node, which conflicts with previous borrow to retrieve the child
|
||||
if let Some(color) = color {
|
||||
let mut node = node.borrow_mut();
|
||||
node.count += color.0;
|
||||
node.total_red += color.1;
|
||||
node.total_green += color.2;
|
||||
node.total_blue += color.3;
|
||||
node.leaf[i] = None;
|
||||
node.children -= 1;
|
||||
quantizer.colors -= 1;
|
||||
|
||||
removals.push(color.4); //save for further processing outside loop (and borrow of colorlist)
|
||||
}
|
||||
}
|
||||
node.borrow_mut().is_leaf = true;
|
||||
quantizer.colors += 1;
|
||||
if quantizer.colors <= num_colors {
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
let color_list = &mut quantizer.color_list[(level as usize + 1)];
|
||||
for index in removals {
|
||||
color_list[index] = None; //set to None here, Option removed later
|
||||
}
|
||||
|
||||
reduce_tree(quantizer, num_colors, level - 1);
|
||||
}
|
||||
}
|
||||
|
||||
// call to nested function
|
||||
reduce_tree(self, num_colors, (MAX_LEVEL - 1) as isize);
|
||||
}
|
||||
}
|
||||
|
||||
struct OctTreeNode {
|
||||
children: usize,
|
||||
level: usize,
|
||||
parent: Option<Rc<RefCell<OctTreeNode>>>,
|
||||
leaf: Vec<Option<Rc<RefCell<OctTreeNode>>>>,
|
||||
is_leaf: bool,
|
||||
count: usize,
|
||||
total_red: u32,
|
||||
total_green: u32,
|
||||
total_blue: u32,
|
||||
index: usize,
|
||||
p_index: usize,
|
||||
}
|
||||
|
||||
impl OctTreeNode {
|
||||
fn new() -> Self {
|
||||
Self {
|
||||
children: 0,
|
||||
level: 0,
|
||||
parent: None,
|
||||
leaf: vec![None; 8],
|
||||
is_leaf: false,
|
||||
count: 0,
|
||||
total_red: 0,
|
||||
total_green: 0,
|
||||
total_blue: 0,
|
||||
index: 0,
|
||||
p_index: 0,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn get_bitmask(color: &Rgba<u8>, level: &usize) -> usize {
|
||||
let bit = 0x80 >> level;
|
||||
|
||||
let mut index = 0;
|
||||
if (color[0] & bit) != 0 {
|
||||
index += 4;
|
||||
}
|
||||
if (color[1] & bit) != 0 {
|
||||
index += 2;
|
||||
}
|
||||
if (color[2] & bit) != 0 {
|
||||
index += 1;
|
||||
}
|
||||
index
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
|
||||
use image::ImageError;
|
||||
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn test_big_image() -> Result<(), ImageError> {
|
||||
let src: RgbaImage = image::open("testdata/input.jpg").unwrap().into_rgba8();
|
||||
|
||||
let out = quantize(&src, 256);
|
||||
out.save_with_format("output.jpg", image::ImageFormat::Jpeg)?;
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
BIN
testdata/img.png
vendored
Normal file
BIN
testdata/img.png
vendored
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 4.1 MiB |
BIN
testdata/input.jpg
vendored
Normal file
BIN
testdata/input.jpg
vendored
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 2.1 MiB |
BIN
testdata/test.jpg
vendored
Normal file
BIN
testdata/test.jpg
vendored
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 3.7 KiB |
Loading…
Add table
Reference in a new issue