80 lines
2.5 KiB
Rust
80 lines
2.5 KiB
Rust
|
#![cfg_attr(docsrs, feature(doc_auto_cfg))]
|
||
|
|
#![warn(missing_docs, missing_debug_implementations, rust_2018_idioms)]
|
||
|
|
|
||
|
|
//! Cosmic theme library.
|
||
|
|
//!
|
||
|
|
//! Provides utilities for creating custom cosmic themes.
|
||
|
|
//!
|
||
|
|
|
||
|
|
#[cfg(feature = "contrast-derivation")]
|
||
|
|
pub use color_picker::*;
|
||
|
|
pub use config::*;
|
||
|
|
#[cfg(feature = "hex-color")]
|
||
|
|
pub use hex_color::*;
|
||
|
|
pub use model::*;
|
||
|
|
pub use output::*;
|
||
|
|
pub use theme_provider::*;
|
||
|
|
#[cfg(feature = "contrast-derivation")]
|
||
|
|
mod color_picker;
|
||
|
|
mod config;
|
||
|
|
#[cfg(feature = "hex-color")]
|
||
|
|
mod hex_color;
|
||
|
|
mod model;
|
||
|
|
mod output;
|
||
|
|
mod theme_provider;
|
||
|
|
/// utilities
|
||
|
|
pub mod util;
|
||
|
|
|
||
|
|
/// name of cosmic theme
|
||
|
|
pub const NAME: &'static str = "com.system76.CosmicTheme";
|
||
|
|
/// Name of the theme directory
|
||
|
|
pub const THEME_DIR: &str = "themes";
|
||
|
|
/// name of the palette directory
|
||
|
|
pub const PALETTE_DIR: &str = "palettes";
|
||
|
|
|
||
|
|
pub use palette;
|
||
|
|
|
||
|
|
/// theme derivation from an image
|
||
|
|
#[cfg(feature = "theme-from-image")]
|
||
|
|
pub mod theme_from_image {
|
||
|
|
use image::EncodableLayout;
|
||
|
|
use kmeans_colors::{get_kmeans_hamerly, Kmeans, Sort};
|
||
|
|
use palette::{rgb::Srgba, Pixel};
|
||
|
|
use palette::{IntoColor, Lab};
|
||
|
|
use std::path::Path;
|
||
|
|
|
||
|
|
/// Create a palette from an image
|
||
|
|
/// The palette is sorted by how often a color occurs in the image, most often first
|
||
|
|
pub fn theme_from_image<P: AsRef<Path>>(path: P) -> Option<Vec<Srgba>> {
|
||
|
|
// calculate kmeans colors from file
|
||
|
|
// let pixbuf = Pixbuf::from_file(path);
|
||
|
|
let img = image::open(path);
|
||
|
|
match img {
|
||
|
|
Ok(img) => {
|
||
|
|
let lab: Vec<Lab> = Srgba::from_raw_slice(img.to_rgba8().into_raw().as_bytes())
|
||
|
|
.iter()
|
||
|
|
.map(|x| x.color.into_format().into_color())
|
||
|
|
.collect();
|
||
|
|
|
||
|
|
let mut result = Kmeans::new();
|
||
|
|
|
||
|
|
// TODO random seed
|
||
|
|
for i in 0..2 {
|
||
|
|
let run_result = get_kmeans_hamerly(5, 20, 5.0, false, &lab, i as u64);
|
||
|
|
if run_result.score < result.score {
|
||
|
|
result = run_result;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
let mut res = Lab::sort_indexed_colors(&result.centroids, &result.indices);
|
||
|
|
res.sort_unstable_by(|a, b| (b.percentage).partial_cmp(&a.percentage).unwrap());
|
||
|
|
let colors: Vec<Srgba> = res.iter().map(|x| x.centroid.into_color()).collect();
|
||
|
|
Some(colors)
|
||
|
|
}
|
||
|
|
Err(err) => {
|
||
|
|
eprintln!("{}", err);
|
||
|
|
None
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|