photon_rs/
helpers.rs

1//! Helper functions for converting between various formats
2
3use crate::{PhotonImage, Rgb};
4use image::DynamicImage::ImageRgba8;
5use image::{DynamicImage, ImageBuffer};
6
7#[cfg(feature = "enable_wasm")]
8extern crate wasm_bindgen;
9
10/// Gets the square distance between two colours
11pub fn square_distance(color1: Rgb, color2: Rgb) -> i32 {
12    let (r1, g1, b1) = (color1.r as i32, color1.g as i32, color1.b as i32);
13    let (r2, g2, b2) = (color2.r as i32, color2.g as i32, color2.b as i32);
14    i32::pow(r1 - r2, 2) + i32::pow(g1 - g2, 2) + i32::pow(b1 - b2, 2)
15}
16
17// Read a DynamicImage from a given path.
18pub fn open_dyn_image(img_path: &'static str) -> DynamicImage {
19    image::open(img_path).unwrap()
20}
21
22/// Save a DynamicImage to a path.
23pub fn save_dyn_image(img: DynamicImage, filtered_img_path: &str) {
24    // let raw_pixels = img.raw_pixels;
25    // let width = img.width;
26    // let height = img.height;
27
28    // let img_buffer = ImageBuffer::from_vec(width, height, raw_pixels).unwrap();
29    // let dynimage = image::ImageRgba8(img_buffer);
30
31    img.save(filtered_img_path).unwrap();
32}
33
34/// Get raw pixels (as a vec of u8s) from a DynamicImage
35pub fn get_pixels(img: DynamicImage) -> Vec<u8> {
36    // get an image's raw pixels, and return as a vec of u8s
37    img.into_bytes()
38}
39
40/// Convert a PhotonImage to a DynamicImage type (struct used by the `image` crate)
41pub fn dyn_image_from_raw(photon_image: &PhotonImage) -> DynamicImage {
42    // convert a vec of raw pixels (as u8s) to a DynamicImage type
43    let _len_vec = photon_image.raw_pixels.len() as u128;
44    let raw_pixels = &photon_image.raw_pixels;
45    let img_buffer = ImageBuffer::from_vec(
46        photon_image.width,
47        photon_image.height,
48        raw_pixels.to_vec(),
49    )
50    .unwrap();
51    ImageRgba8(img_buffer)
52}
photon_rs/helpers.rs

photon_rs/
helpers.rs
