lutgen 0.15.0

A blazingly fast interpolated LUT utility for arbitrary and popular color palettes.
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

//! Interpolated remapping algorithms.

use std::sync::atomic::AtomicBool;
use std::sync::Arc;

pub use gaussian_blur::GaussianBlurRemapper;
pub use gaussian_sample::GaussianSamplingRemapper;
use image::Rgba;
use kiddo::float::kdtree::KdTree;
pub use nearest_neighbor::NearestNeighborRemapper;
#[cfg(feature = "rayon")]
use rayon::prelude::*;
pub use rbf::{GaussianRemapper, LinearRemapper, ShepardRemapper};

use crate::RgbaImage;

mod gaussian_blur;
mod gaussian_sample;
mod nearest_neighbor;
mod rbf;

/// Interpolated Remapper. Implements an algorithm with some initialization parameters.
pub trait InterpolatedRemapper<'a>: Sync {
    /// Remap a single pixel in place
    fn remap_pixel(&self, pixel: &mut Rgba<u8>);

    /// Remap an image in place. Default implementation uses `rayon` to iterate in parallel over
    /// the pixels.
    fn remap_image(&self, image: &mut RgbaImage) {
        image.pixels_mut().for_each(|pixel| {
            self.remap_pixel(pixel);
        });
    }

    /// Remap an image in place, aborting if the given atomic boolean is true.
    fn remap_image_with_interrupt(&self, image: &mut RgbaImage, abort: Arc<AtomicBool>) {
        image.pixels_mut().for_each(|pixel| {
            if !abort.load(std::sync::atomic::Ordering::Relaxed) {
                self.remap_pixel(pixel);
            }
        });
    }

    /// Rayon version
    #[cfg(feature = "rayon")]
    fn par_remap_image(&self, image: &mut RgbaImage) {
        image.par_pixels_mut().for_each(|pixel| {
            self.remap_pixel(pixel);
        });
    }

    /// Rayon version
    #[cfg(feature = "rayon")]
    fn par_remap_image_with_interrupt(&self, image: &mut RgbaImage, abort: Arc<AtomicBool>) {
        image.par_pixels_mut().for_each(|pixel| {
            if !abort.load(std::sync::atomic::Ordering::Relaxed) {
                self.remap_pixel(pixel);
            }
        });
    }
}

/// Type alias for our internal color tree for NN lookups
type ColorTree = KdTree<f64, u32, 3, 4, u32>;