521 lines
16 KiB
Rust
521 lines
16 KiB
Rust
pub use cosmic_bg_config::{Color, Config, Entry, Gradient, ScalingMode, Source};
|
|
use eyre::{eyre, OptionExt};
|
|
use fast_image_resize::SrcCropping;
|
|
use futures_lite::Stream;
|
|
use image::imageops::FilterType;
|
|
use image::{DynamicImage, ImageBuffer, Rgba, RgbaImage};
|
|
use jxl_oxide::{EnumColourEncoding, JxlImage, PixelFormat};
|
|
use std::{
|
|
borrow::Cow,
|
|
collections::{hash_map::DefaultHasher, BTreeSet, HashMap},
|
|
hash::{Hash, Hasher},
|
|
io::Read,
|
|
path::{Path, PathBuf},
|
|
pin::Pin,
|
|
};
|
|
|
|
pub const DEFAULT_COLORS: &[Color] = &[
|
|
Color::Single([0.580, 0.922, 0.922]),
|
|
Color::Single([0.000, 0.286, 0.427]),
|
|
Color::Single([1.000, 0.678, 0.000]),
|
|
Color::Single([0.282, 0.725, 0.78]),
|
|
Color::Single([0.333, 0.278, 0.259]),
|
|
Color::Single([0.969, 0.878, 0.384]),
|
|
Color::Single([0.063, 0.165, 0.298]),
|
|
Color::Single([1.000, 0.843, 0.631]),
|
|
Color::Single([0.976, 0.227, 0.514]),
|
|
Color::Single([1.000, 0.612, 0.867]),
|
|
Color::Single([0.812, 0.490, 1.000]),
|
|
Color::Single([0.835, 0.549, 1.000]),
|
|
Color::Single([0.243, 0.533, 1.000]),
|
|
Color::Single([0.584, 0.769, 0.988]),
|
|
Color::Gradient(Gradient {
|
|
colors: Cow::Borrowed(&[[1.000, 0.678, 0.000], [0.282, 0.725, 0.78]]),
|
|
radius: 180.0,
|
|
}),
|
|
Color::Gradient(Gradient {
|
|
colors: Cow::Borrowed(&[[1.000, 0.843, 0.631], [0.58, 0.922, 0.922]]),
|
|
radius: 180.0,
|
|
}),
|
|
Color::Gradient(Gradient {
|
|
colors: Cow::Borrowed(&[[1.000, 0.612, 0.867], [0.976, 0.29, 0.514]]),
|
|
radius: 180.0,
|
|
}),
|
|
Color::Gradient(Gradient {
|
|
colors: Cow::Borrowed(&[[0.584, 0.769, 0.988], [0.063, 0.165, 0.298]]),
|
|
radius: 180.0,
|
|
}),
|
|
Color::Gradient(Gradient {
|
|
colors: Cow::Borrowed(&[[0.969, 0.878, 0.384], [0.333, 0.278, 0.259]]),
|
|
radius: 180.0,
|
|
}),
|
|
];
|
|
|
|
pub async fn config() -> (Config, HashMap<String, (String, (u32, u32))>) {
|
|
let mut displays = HashMap::new();
|
|
|
|
if let Ok(list) = cosmic_randr_shell::list().await {
|
|
for (_key, output) in list.outputs {
|
|
displays.insert(output.name, (output.model, output.physical));
|
|
}
|
|
}
|
|
|
|
let context = cosmic_bg_config::context().expect("failed to get helper for cosmic bg config");
|
|
|
|
let config = match Config::load(&context) {
|
|
Ok(conf) => conf,
|
|
Err(why) => {
|
|
tracing::warn!(?why, "Config file error, falling back to defaults");
|
|
Config::default()
|
|
}
|
|
};
|
|
|
|
(config, displays)
|
|
}
|
|
|
|
pub fn set(config: &mut Config, entry: Entry) {
|
|
if let Ok(context) = cosmic_bg_config::context() {
|
|
let _res = context.set_same_on_all(config.same_on_all);
|
|
|
|
if let Err(why) = config.set_entry(&context, entry) {
|
|
tracing::error!(?why, "failed to set background");
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Path to directory where wallpaper thumbnails are stored.
|
|
#[must_use]
|
|
pub fn cache_dir() -> Option<PathBuf> {
|
|
dirs::cache_dir().map(|path| {
|
|
let cache = path.join("cosmic-settings/wallpapers");
|
|
let _res = std::fs::create_dir_all(&cache);
|
|
cache
|
|
})
|
|
}
|
|
|
|
/// Loads wallpapers in parallel by spawning tasks with a rayon thread pool.
|
|
#[must_use]
|
|
pub async fn load_each_from_path(
|
|
path: PathBuf,
|
|
recurse: bool,
|
|
) -> Pin<Box<dyn Send + Stream<Item = (PathBuf, RgbaImage, RgbaImage)>>> {
|
|
let wallpapers = tokio::task::spawn_blocking(move || {
|
|
// Directories to search recursively.
|
|
let mut paths = vec![path];
|
|
// Discovered image files that will be loaded as wallpapers.
|
|
let mut wallpapers = BTreeSet::new();
|
|
|
|
while let Some(path) = paths.pop() {
|
|
if let Ok(dir) = path.read_dir() {
|
|
for entry in dir.filter_map(Result::ok) {
|
|
let Ok(file_type) = entry.file_type() else {
|
|
continue;
|
|
};
|
|
|
|
let path = entry.path();
|
|
|
|
// Recursively search directories, while storing only image files.
|
|
if recurse && file_type.is_dir() {
|
|
paths.push(path);
|
|
} else if file_type.is_file() {
|
|
let path = if path.extension().map_or(false, |ext| ext == "jxl") {
|
|
path
|
|
} else if let Ok(Some(kind)) = infer::get_from_path(&path) {
|
|
if infer::MatcherType::Image == kind.matcher_type() {
|
|
path
|
|
} else {
|
|
continue;
|
|
}
|
|
} else {
|
|
continue;
|
|
};
|
|
|
|
wallpapers.insert(path);
|
|
|
|
if wallpapers.len() > 99 {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
wallpapers
|
|
});
|
|
|
|
if let Ok(wallpapers) = wallpapers.await {
|
|
use futures_util::StreamExt;
|
|
let future = futures_util::stream::iter(wallpapers)
|
|
.map(|path| tokio::task::spawn_blocking(|| load_image_with_thumbnail(path)))
|
|
.buffered(4)
|
|
.filter_map(|value| async { value.ok()? });
|
|
|
|
Box::pin(future)
|
|
} else {
|
|
Box::pin(futures_lite::stream::empty())
|
|
}
|
|
}
|
|
|
|
#[must_use]
|
|
pub fn load_image_with_thumbnail(
|
|
path: PathBuf,
|
|
) -> Option<(
|
|
PathBuf,
|
|
ImageBuffer<Rgba<u8>, Vec<u8>>,
|
|
ImageBuffer<Rgba<u8>, Vec<u8>>,
|
|
)> {
|
|
let cache_dir = cache_dir();
|
|
let image_operation = load_thumbnail(&mut Vec::new(), cache_dir.as_deref(), &path);
|
|
|
|
if let Some(image_operation) = image_operation {
|
|
let tokio_handle = tokio::runtime::Handle::current();
|
|
|
|
let display_thumbnail = match image_operation {
|
|
ImageOperation::Cached(thumbnail) => thumbnail.to_rgba8(),
|
|
|
|
ImageOperation::GenerateThumbnail { path, image } => {
|
|
let image = resize_thumbnail(&image, 300, 169).to_rgba8();
|
|
|
|
if let Some(path) = path {
|
|
// Save thumbnail to disk without blocking.
|
|
tokio_handle.spawn_blocking({
|
|
let image = image.clone();
|
|
move || {
|
|
if let Err(why) = image.save(&path) {
|
|
tracing::error!(?path, ?why, "failed to save image thumbnail");
|
|
|
|
let _res = std::fs::remove_file(&path);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
image
|
|
}
|
|
};
|
|
|
|
let mut selection_thumbnail = image::imageops::resize(
|
|
&display_thumbnail,
|
|
158,
|
|
105,
|
|
image::imageops::FilterType::Lanczos3,
|
|
);
|
|
|
|
round(&mut selection_thumbnail, [8, 8, 8, 8]);
|
|
|
|
Some((path, display_thumbnail, selection_thumbnail))
|
|
} else {
|
|
None
|
|
}
|
|
}
|
|
|
|
enum ImageOperation {
|
|
GenerateThumbnail {
|
|
path: Option<PathBuf>,
|
|
image: DynamicImage,
|
|
},
|
|
Cached(DynamicImage),
|
|
}
|
|
|
|
/// Loads wallpaper thumbnails, or defines what needs to be done to create them.
|
|
///
|
|
///
|
|
/// Caching reduces time required to load a wallpaper by 99%.
|
|
#[must_use]
|
|
fn load_thumbnail(
|
|
input_buffer: &mut Vec<u8>,
|
|
cache_dir: Option<&Path>,
|
|
path: &Path,
|
|
) -> Option<ImageOperation> {
|
|
if let Some(cache_dir) = cache_dir {
|
|
if let Ok(ctime) = path.metadata().and_then(|meta| meta.created()) {
|
|
// Search for thumbnail by a unique hash string.
|
|
let mut hasher = DefaultHasher::new();
|
|
path.hash(&mut hasher);
|
|
ctime.hash(&mut hasher);
|
|
let hash = hasher.finish();
|
|
|
|
let thumbnail_path = cache_dir.join(format!("{hash:x}.png"));
|
|
|
|
if thumbnail_path.exists() {
|
|
if let Some(image) = open_image(input_buffer, &thumbnail_path) {
|
|
return Some(ImageOperation::Cached(image));
|
|
}
|
|
|
|
let _res = std::fs::remove_file(&thumbnail_path);
|
|
}
|
|
|
|
return Some(ImageOperation::GenerateThumbnail {
|
|
path: Some(thumbnail_path),
|
|
image: open_image(input_buffer, path)?,
|
|
});
|
|
}
|
|
}
|
|
|
|
if let Some(image) = open_image(input_buffer, path) {
|
|
return Some(ImageOperation::GenerateThumbnail { path: None, image });
|
|
}
|
|
|
|
None
|
|
}
|
|
|
|
fn open_image(input_buffer: &mut Vec<u8>, path: &Path) -> Option<DynamicImage> {
|
|
if path.extension().map_or(false, |ext| ext == "jxl") {
|
|
return match decode_jpegxl(path) {
|
|
Ok(image) => Some(image),
|
|
Err(why) => {
|
|
tracing::error!(?path, ?why, "image decode failed");
|
|
None
|
|
}
|
|
};
|
|
}
|
|
|
|
let capacity = match path.metadata() {
|
|
Ok(metadata) => metadata.len() as usize,
|
|
Err(why) => {
|
|
tracing::error!(?path, ?why, "error loading image metadata");
|
|
return None;
|
|
}
|
|
};
|
|
|
|
input_buffer.clear();
|
|
input_buffer.reserve_exact(capacity);
|
|
|
|
let mut file = match std::fs::File::open(path) {
|
|
Ok(file) => file,
|
|
Err(why) => {
|
|
tracing::error!(?path, ?why, "error opening image");
|
|
return None;
|
|
}
|
|
};
|
|
|
|
if let Err(why) = file.read_to_end(input_buffer) {
|
|
tracing::error!(?path, ?why, "error reading image");
|
|
}
|
|
|
|
let input_cursor = std::io::Cursor::new(input_buffer);
|
|
let mut image_decoder = image::ImageReader::new(input_cursor);
|
|
|
|
image_decoder = if let Ok(decoder) = image_decoder.with_guessed_format() {
|
|
decoder
|
|
} else {
|
|
tracing::error!(?path, "unsupported image format");
|
|
return None;
|
|
};
|
|
|
|
match image_decoder.decode() {
|
|
Ok(image) => Some(image),
|
|
Err(why) => {
|
|
tracing::error!(?path, ?why, "image decode failed");
|
|
None
|
|
}
|
|
}
|
|
}
|
|
|
|
// https://users.rust-lang.org/t/how-to-trim-image-to-circle-image-without-jaggy/70374/2
|
|
fn round(img: &mut image::ImageBuffer<image::Rgba<u8>, Vec<u8>>, radius: [u32; 4]) {
|
|
let (width, height) = img.dimensions();
|
|
assert!(radius[0] + radius[1] <= width);
|
|
assert!(radius[3] + radius[2] <= width);
|
|
assert!(radius[0] + radius[3] <= height);
|
|
assert!(radius[1] + radius[2] <= height);
|
|
|
|
// top left
|
|
border_radius(img, radius[0], |x, y| (x - 1, y - 1));
|
|
// top right
|
|
border_radius(img, radius[1], |x, y| (width - x, y - 1));
|
|
// bottom right
|
|
border_radius(img, radius[2], |x, y| (width - x, height - y));
|
|
// bottom left
|
|
border_radius(img, radius[3], |x, y| (x - 1, height - y));
|
|
}
|
|
|
|
fn border_radius(
|
|
img: &mut image::ImageBuffer<image::Rgba<u8>, Vec<u8>>,
|
|
r: u32,
|
|
coordinates: impl Fn(u32, u32) -> (u32, u32),
|
|
) {
|
|
if r == 0 {
|
|
return;
|
|
}
|
|
let r0 = r;
|
|
|
|
// 16x antialiasing: 16x16 grid creates 256 possible shades, great for u8!
|
|
let r = 16 * r;
|
|
|
|
let mut x = 0;
|
|
let mut y = r - 1;
|
|
let mut p: i32 = 2 - r as i32;
|
|
|
|
// ...
|
|
|
|
let mut alpha: u16 = 0;
|
|
let mut skip_draw = true;
|
|
|
|
let draw = |img: &mut RgbaImage, alpha, x, y| {
|
|
debug_assert!((1..=256).contains(&alpha));
|
|
let pixel_alpha = &mut img[coordinates(r0 - x, r0 - y)].0[3];
|
|
*pixel_alpha = ((alpha * *pixel_alpha as u16 + 128) / 256) as u8;
|
|
};
|
|
|
|
'l: loop {
|
|
// (comments for bottom_right case:)
|
|
// remove contents below current position
|
|
{
|
|
let i = x / 16;
|
|
for j in y / 16 + 1..r0 {
|
|
img[coordinates(r0 - i, r0 - j)].0[3] = 0;
|
|
}
|
|
}
|
|
// remove contents right of current position mirrored
|
|
{
|
|
let j = x / 16;
|
|
for i in y / 16 + 1..r0 {
|
|
img[coordinates(r0 - i, r0 - j)].0[3] = 0;
|
|
}
|
|
}
|
|
|
|
// draw when moving to next pixel in x-direction
|
|
if !skip_draw {
|
|
draw(img, alpha, x / 16 - 1, y / 16);
|
|
draw(img, alpha, y / 16, x / 16 - 1);
|
|
alpha = 0;
|
|
}
|
|
|
|
for _ in 0..16 {
|
|
skip_draw = false;
|
|
|
|
if x >= y {
|
|
break 'l;
|
|
}
|
|
|
|
alpha += y as u16 % 16 + 1;
|
|
if p < 0 {
|
|
x += 1;
|
|
p += (2 * x + 2) as i32;
|
|
} else {
|
|
// draw when moving to next pixel in y-direction
|
|
if y % 16 == 0 {
|
|
draw(img, alpha, x / 16, y / 16);
|
|
draw(img, alpha, y / 16, x / 16);
|
|
skip_draw = true;
|
|
alpha = (x + 1) as u16 % 16 * 16;
|
|
}
|
|
|
|
x += 1;
|
|
p -= (2 * (y - x) + 2) as i32;
|
|
y -= 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
// one corner pixel left
|
|
if x / 16 == y / 16 {
|
|
// column under current position possibly not yet accounted
|
|
if x == y {
|
|
alpha += y as u16 % 16 + 1;
|
|
}
|
|
let s = y as u16 % 16 + 1;
|
|
let alpha = 2 * alpha - s * s;
|
|
draw(img, alpha, x / 16, y / 16);
|
|
}
|
|
|
|
// remove remaining square of content in the corner
|
|
let range = y / 16 + 1..r0;
|
|
for i in range.clone() {
|
|
for j in range.clone() {
|
|
img[coordinates(r0 - i, r0 - j)].0[3] = 0;
|
|
}
|
|
}
|
|
}
|
|
|
|
/// Decodes JPEG XL image files into `image::DynamicImage` via `jxl-oxide`.
|
|
pub fn decode_jpegxl(path: &std::path::Path) -> eyre::Result<DynamicImage> {
|
|
let mut image = JxlImage::builder()
|
|
.open(path)
|
|
.map_err(|why| eyre!("failed to read image header: {why}"))?;
|
|
image.request_color_encoding(EnumColourEncoding::srgb(
|
|
jxl_oxide::RenderingIntent::Relative,
|
|
));
|
|
let render = image
|
|
.render_frame(0)
|
|
.map_err(|why| eyre!("failed to render image frame: {why}"))?;
|
|
|
|
let framebuffer = render.image_all_channels();
|
|
match image.pixel_format() {
|
|
PixelFormat::Graya => image::GrayAlphaImage::from_raw(
|
|
framebuffer.width() as u32,
|
|
framebuffer.height() as u32,
|
|
framebuffer
|
|
.buf()
|
|
.iter()
|
|
.map(|x| x * 255. + 0.5)
|
|
.map(|x| x as u8)
|
|
.collect::<Vec<_>>(),
|
|
)
|
|
.map(DynamicImage::ImageLumaA8)
|
|
.ok_or_eyre("Can't decode gray alpha buffer"),
|
|
PixelFormat::Gray => image::GrayImage::from_raw(
|
|
framebuffer.width() as u32,
|
|
framebuffer.height() as u32,
|
|
framebuffer
|
|
.buf()
|
|
.iter()
|
|
.map(|x| x * 255. + 0.5)
|
|
.map(|x| x as u8)
|
|
.collect::<Vec<_>>(),
|
|
)
|
|
.map(DynamicImage::ImageLuma8)
|
|
.ok_or_eyre("Can't decode gray buffer"),
|
|
PixelFormat::Rgba => image::RgbaImage::from_raw(
|
|
framebuffer.width() as u32,
|
|
framebuffer.height() as u32,
|
|
framebuffer
|
|
.buf()
|
|
.iter()
|
|
.map(|x| x * 255. + 0.5)
|
|
.map(|x| x as u8)
|
|
.collect::<Vec<_>>(),
|
|
)
|
|
.map(DynamicImage::ImageRgba8)
|
|
.ok_or_eyre("Can't decode rgba buffer"),
|
|
PixelFormat::Rgb => image::RgbImage::from_raw(
|
|
framebuffer.width() as u32,
|
|
framebuffer.height() as u32,
|
|
framebuffer
|
|
.buf()
|
|
.iter()
|
|
.map(|x| x * 255. + 0.5)
|
|
.map(|x| x as u8)
|
|
.collect::<Vec<_>>(),
|
|
)
|
|
.map(DynamicImage::ImageRgb8)
|
|
.ok_or_eyre("Can't decode rgb buffer"),
|
|
//TODO: handle this
|
|
PixelFormat::Cmyk => Err(eyre!("unsupported pixel format: CMYK")),
|
|
PixelFormat::Cmyka => Err(eyre!("unsupported pixel format: CMYKA")),
|
|
}
|
|
}
|
|
|
|
/// Use `fast-image-resize` crate for faster thumbnail generation.
|
|
fn resize_thumbnail(
|
|
img: &image::DynamicImage,
|
|
new_width: u32,
|
|
new_height: u32,
|
|
) -> image::DynamicImage {
|
|
let mut resizer = fast_image_resize::Resizer::new();
|
|
let options = fast_image_resize::ResizeOptions {
|
|
algorithm: fast_image_resize::ResizeAlg::Convolution(
|
|
fast_image_resize::FilterType::Lanczos3,
|
|
),
|
|
cropping: SrcCropping::FitIntoDestination((new_width as f64, new_height as f64)),
|
|
..Default::default()
|
|
};
|
|
let mut new_image = image::DynamicImage::new(new_width, new_height, img.color());
|
|
if let Err(err) = resizer.resize(img, &mut new_image, &options) {
|
|
tracing::warn!(?err, "Failed to use `fast_image_resize`. Falling back.");
|
|
new_image =
|
|
image::imageops::resize(img, new_width, new_height, FilterType::Lanczos3).into();
|
|
}
|
|
new_image
|
|
}
|