Move thumbnail generation to thumbnailer crate
Signed-off-by: trivernis <trivernis@protonmail.com>pull/4/head
parent
31d8c7d6fd
commit
2735033859
@ -1,58 +0,0 @@
|
|||||||
use crate::error::RepoResult;
|
|
||||||
use image::imageops::FilterType;
|
|
||||||
use image::io::Reader as ImageReader;
|
|
||||||
use image::{DynamicImage, ImageOutputFormat};
|
|
||||||
use std::io::Cursor;
|
|
||||||
use std::path::PathBuf;
|
|
||||||
use tokio::fs::OpenOptions;
|
|
||||||
use tokio::io::{AsyncRead, AsyncReadExt, BufReader};
|
|
||||||
|
|
||||||
/// Represents the different sizes of a thumbnail
|
|
||||||
#[derive(Clone, Copy, Debug)]
|
|
||||||
pub enum ThumbnailSize {
|
|
||||||
Small,
|
|
||||||
Medium,
|
|
||||||
Large,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ThumbnailSize {
|
|
||||||
pub fn dimensions(&self) -> (u32, u32) {
|
|
||||||
match self {
|
|
||||||
ThumbnailSize::Small => (128, 128),
|
|
||||||
ThumbnailSize::Medium => (256, 256),
|
|
||||||
ThumbnailSize::Large => (512, 512),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Reads an image from a path
|
|
||||||
pub async fn read_image_from_path(path: &PathBuf) -> RepoResult<DynamicImage> {
|
|
||||||
let file = OpenOptions::new().read(true).open(path).await?;
|
|
||||||
let mut reader = BufReader::new(file);
|
|
||||||
read_image(&mut reader).await
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Reads an image from a reader
|
|
||||||
pub async fn read_image<R: AsyncRead + Unpin>(reader: &mut R) -> RepoResult<DynamicImage> {
|
|
||||||
let mut buf = Vec::new();
|
|
||||||
reader.read_to_end(&mut buf).await?;
|
|
||||||
let image = ImageReader::new(Cursor::new(buf))
|
|
||||||
.with_guessed_format()?
|
|
||||||
.decode()?;
|
|
||||||
|
|
||||||
Ok(image)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Returns the bytes of an image in the png format
|
|
||||||
pub fn get_image_bytes_png(image: DynamicImage) -> RepoResult<Vec<u8>> {
|
|
||||||
let mut buf = Vec::new();
|
|
||||||
image.write_to(&mut buf, ImageOutputFormat::Png)?;
|
|
||||||
|
|
||||||
Ok(buf)
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Creates a thumbnail with the defined thumbnail size
|
|
||||||
pub fn create_thumbnail(image: DynamicImage, size: ThumbnailSize) -> DynamicImage {
|
|
||||||
let (height, width) = size.dimensions();
|
|
||||||
image.resize(height, width, FilterType::Nearest)
|
|
||||||
}
|
|
||||||
@ -1,11 +1,10 @@
|
|||||||
pub mod context;
|
pub mod context;
|
||||||
pub mod error;
|
pub mod error;
|
||||||
pub mod file_hash_store;
|
pub mod file_hash_store;
|
||||||
pub mod image_processing;
|
|
||||||
pub mod settings;
|
pub mod settings;
|
||||||
pub mod type_keys;
|
pub mod type_keys;
|
||||||
pub mod utils;
|
pub mod utils;
|
||||||
|
|
||||||
pub use futures;
|
pub use futures;
|
||||||
pub use image;
|
|
||||||
pub use rmp_ipc;
|
pub use rmp_ipc;
|
||||||
|
pub use thumbnailer;
|
||||||
|
|||||||
Loading…
Reference in New Issue