81 lines
2.6 KiB
Rust
81 lines
2.6 KiB
Rust
use image::{FilterType, GenericImage, ImageError, ImageFormat};
|
|
use image::open as image_open;
|
|
use rexif::{self, ExifData};
|
|
use rphotos::models::Photo;
|
|
use std::{fs, io};
|
|
use std::ffi::OsStr;
|
|
use std::path::{Path, PathBuf};
|
|
|
|
pub struct PhotosDir {
|
|
basedir: PathBuf,
|
|
}
|
|
|
|
impl PhotosDir {
|
|
pub fn new(basedir: PathBuf) -> Self {
|
|
PhotosDir { basedir: basedir }
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub fn get_scaled_image(&self,
|
|
photo: Photo,
|
|
width: u32,
|
|
height: u32)
|
|
-> Result<Vec<u8>, ImageError> {
|
|
let path = self.basedir.join(photo.path);
|
|
info!("Should open {:?}", path);
|
|
let img = try!(image_open(path));
|
|
let img = if width < img.width() || height < img.height() {
|
|
img.resize(width, height, FilterType::CatmullRom)
|
|
} else {
|
|
img
|
|
};
|
|
let img = match photo.rotation {
|
|
_x @ 0...44 => img,
|
|
_x @ 45...134 => img.rotate90(),
|
|
_x @ 135...224 => img.rotate180(),
|
|
_x @ 225...314 => img.rotate270(),
|
|
_x @ 315...360 => img,
|
|
x => {
|
|
warn!("Should rotate photo {} deg, which is unsupported", x);
|
|
img
|
|
}
|
|
};
|
|
// TODO Put the icon in some kind of cache!
|
|
let mut buf = Vec::new();
|
|
try!(img.save(&mut buf, ImageFormat::JPEG));
|
|
Ok(buf)
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub fn has_file<S: AsRef<OsStr> + ?Sized>(&self, path: &S) -> bool {
|
|
self.basedir.join(Path::new(path)).is_file()
|
|
}
|
|
|
|
#[allow(dead_code)]
|
|
pub fn find_files(&self,
|
|
dir: &Path,
|
|
cb: &Fn(&str, &ExifData))
|
|
-> io::Result<()> {
|
|
let absdir = self.basedir.join(dir);
|
|
if try!(fs::metadata(&absdir)).is_dir() {
|
|
let bl = self.basedir.to_str().unwrap().len() + 1;
|
|
debug!("Should look in {:?}", absdir);
|
|
for entry in try!(fs::read_dir(absdir)) {
|
|
let entry = try!(entry);
|
|
if try!(fs::metadata(entry.path())).is_dir() {
|
|
try!(self.find_files(&entry.path(), cb));
|
|
} else {
|
|
let p1 = entry.path();
|
|
if let Ok(exif) = rexif::parse_file(&p1.to_str().unwrap()) {
|
|
let path = p1.to_str().unwrap();
|
|
cb(&path[bl..], &exif);
|
|
} else {
|
|
debug!("'{:?}' has no exif data.", p1);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|