use std::cell::RefCell;
use std::collections::HashMap;
use std::rc::Rc;
use sixtyfps_corelib::graphics::{SharedImageBuffer, Size};
#[cfg(target_arch = "wasm32")]
use sixtyfps_corelib::Property;
use sixtyfps_corelib::{items::ImageRendering, slice::Slice, ImageInner, SharedString};
use super::{CanvasRc, GLItemRenderer};
struct Texture {
id: femtovg::ImageId,
canvas: CanvasRc,
}
impl Texture {
fn size(&self) -> Option<Size> {
self.canvas
.borrow()
.image_info(self.id)
.map(|info| [info.width() as f32, info.height() as f32].into())
.ok()
}
}
impl Drop for Texture {
fn drop(&mut self) {
self.canvas.borrow_mut().delete_image(self.id);
}
}
#[cfg(target_arch = "wasm32")]
struct HTMLImage {
dom_element: web_sys::HtmlImageElement,
image_load_pending: core::pin::Pin<Rc<Property<bool>>>,
}
#[cfg(target_arch = "wasm32")]
impl HTMLImage {
fn new(url: &str) -> Self {
let dom_element = web_sys::HtmlImageElement::new().unwrap();
let image_load_pending = Rc::pin(Property::new(true));
let event_loop_proxy = crate::event_loop::with_window_target(|event_loop| {
event_loop.event_loop_proxy().clone()
});
dom_element.set_cross_origin(Some("anonymous"));
dom_element.set_onload(Some(
&wasm_bindgen::closure::Closure::once_into_js({
let image_load_pending = image_load_pending.clone();
move || {
image_load_pending.as_ref().set(false);
event_loop_proxy
.send_event(crate::event_loop::CustomEvent::RedrawAllWindows)
.ok();
}
})
.into(),
));
dom_element.set_src(&url);
Self { dom_element, image_load_pending }
}
fn size(&self) -> Option<Size> {
match self.image_load_pending.as_ref().get() {
true => None,
false => Some(Size::new(self.dom_element.width() as _, self.dom_element.height() as _)),
}
}
}
#[derive(derive_more::From)]
enum ImageData {
Texture(Texture),
DecodedImage(image::DynamicImage),
EmbeddedImage(SharedImageBuffer),
#[cfg(feature = "svg")]
Svg(usvg::Tree),
#[cfg(target_arch = "wasm32")]
HTMLImage(HTMLImage),
}
impl std::fmt::Debug for ImageData {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
use image::GenericImageView;
match self {
ImageData::Texture(t) => {
write!(f, "ImageData::Texture({:?})", t.id.0)
}
ImageData::DecodedImage(i) => {
write!(f, "ImageData::DecodedImage({}x{})", i.width(), i.height())
}
ImageData::EmbeddedImage(buffer) => {
write!(f, "ImageData::EmbeddedImage({}x{})", buffer.width(), buffer.height())
}
ImageData::Svg(_) => {
write!(f, "ImageData::SVG(...)")
}
#[cfg(target_arch = "wasm32")]
ImageData::HTMLImage(html_image) => {
write!(
f,
"ImageData::HTMLImage({}x{})",
html_image.dom_element.width(),
html_image.dom_element.height()
)
}
}
}
}
#[derive(Debug)]
pub(crate) struct CachedImage(RefCell<ImageData>);
impl CachedImage {
fn new_on_cpu(decoded_image: image::DynamicImage) -> Self {
Self(RefCell::new(ImageData::DecodedImage(decoded_image)))
}
pub fn new_on_gpu(canvas: &CanvasRc, image_id: femtovg::ImageId) -> Self {
Self(RefCell::new(Texture { id: image_id, canvas: canvas.clone() }.into()))
}
pub fn new_empty_on_gpu(canvas: &CanvasRc, width: usize, height: usize) -> Option<Self> {
if width == 0 || height == 0 {
return None;
}
let image_id = canvas
.borrow_mut()
.create_image_empty(
width,
height,
femtovg::PixelFormat::Rgba8,
femtovg::ImageFlags::PREMULTIPLIED | femtovg::ImageFlags::FLIP_Y,
)
.unwrap();
Self::new_on_gpu(canvas, image_id).into()
}
#[cfg(feature = "svg")]
fn new_on_cpu_svg(tree: usvg::Tree) -> Self {
Self(RefCell::new(ImageData::Svg(tree)))
}
pub fn new_from_resource(resource: &ImageInner) -> Option<Self> {
match resource {
ImageInner::None => None,
ImageInner::AbsoluteFilePath(path) => Self::new_from_path(path),
ImageInner::EmbeddedData { data, format } => Self::new_from_data(data, format),
ImageInner::EmbeddedImage(buffer) => {
Some(Self(RefCell::new(ImageData::EmbeddedImage(buffer.clone()))))
}
}
}
fn new_from_path(path: &SharedString) -> Option<Self> {
#[cfg(not(target_arch = "wasm32"))]
{
#[cfg(feature = "svg")]
if path.ends_with(".svg") {
return Some(Self::new_on_cpu_svg(
super::svg::load_from_path(std::path::Path::new(&path.as_str())).map_or_else(
|err| {
eprintln!("Error loading SVG from {}: {}", &path, err);
None
},
Some,
)?,
));
}
Some(Self::new_on_cpu(image::open(std::path::Path::new(&path.as_str())).map_or_else(
|decode_err| {
eprintln!("Error loading image from {}: {}", &path, decode_err);
None
},
Some,
)?))
}
#[cfg(target_arch = "wasm32")]
Some(Self(RefCell::new(ImageData::HTMLImage(HTMLImage::new(path)))))
}
fn new_from_data(data: &Slice<u8>, format: &Slice<u8>) -> Option<Self> {
#[cfg(feature = "svg")]
if format.as_slice() == b"svg" {
return Some(CachedImage::new_on_cpu_svg(
super::svg::load_from_data(data.as_slice()).map_or_else(
|svg_err| {
eprintln!("Error loading SVG: {}", svg_err);
None
},
Some,
)?,
));
}
let format = std::str::from_utf8(format.as_slice())
.ok()
.and_then(image::ImageFormat::from_extension);
let image = if let Some(format) = format {
image::load_from_memory_with_format(data.as_slice(), format)
} else {
image::load_from_memory(data.as_slice())
};
Some(CachedImage::new_on_cpu(image.map_or_else(
|decode_err| {
eprintln!("Error decoding image: {}", decode_err);
None
},
Some,
)?))
}
pub fn ensure_uploaded_to_gpu(
&self,
current_renderer: &GLItemRenderer,
scaling: Option<ImageRendering>,
) -> femtovg::ImageId {
use std::convert::TryFrom;
let canvas = ¤t_renderer.canvas;
let image_flags = match scaling.unwrap_or_default() {
ImageRendering::smooth => femtovg::ImageFlags::empty(),
ImageRendering::pixelated => femtovg::ImageFlags::NEAREST,
};
let img = &mut *self.0.borrow_mut();
if let ImageData::DecodedImage(decoded_image) = img {
let image_id = match femtovg::ImageSource::try_from(&*decoded_image) {
Ok(image_source) => canvas.borrow_mut().create_image(image_source, image_flags),
Err(_) => {
let converted = image::DynamicImage::ImageRgba8(decoded_image.to_rgba8());
let image_source = femtovg::ImageSource::try_from(&converted).unwrap();
canvas.borrow_mut().create_image(image_source, image_flags)
}
}
.unwrap();
*img = Texture { id: image_id, canvas: canvas.clone() }.into()
} else if let ImageData::EmbeddedImage(buffer) = img {
let (image_source, flags) = image_buffer_to_image_source(buffer);
let image_id =
canvas.borrow_mut().create_image(image_source, flags | image_flags).unwrap();
*img = Texture { id: image_id, canvas: canvas.clone() }.into()
}
#[cfg(target_arch = "wasm32")]
if let ImageData::HTMLImage(html_image) = img {
let image_id =
canvas.borrow_mut().create_image(&html_image.dom_element, image_flags).unwrap();
*img = Texture { id: image_id, canvas: canvas.clone() }.into()
}
match &img {
ImageData::Texture(Texture { id, .. }) => *id,
_ => unreachable!(),
}
}
pub fn upload_to_gpu(
&self,
current_renderer: &GLItemRenderer,
target_size: euclid::default::Size2D<u32>,
scaling: ImageRendering,
) -> Option<Self> {
use std::convert::TryFrom;
let canvas = ¤t_renderer.canvas;
let image_flags = match scaling {
ImageRendering::smooth => femtovg::ImageFlags::empty(),
ImageRendering::pixelated => femtovg::ImageFlags::NEAREST,
};
match &*self.0.borrow() {
ImageData::Texture(_) => None, ImageData::DecodedImage(decoded_image) => {
let image_id = match femtovg::ImageSource::try_from(&*decoded_image) {
Ok(image_source) => canvas.borrow_mut().create_image(image_source, image_flags),
Err(_) => {
let converted = image::DynamicImage::ImageRgba8(decoded_image.to_rgba8());
let image_source = femtovg::ImageSource::try_from(&converted).unwrap();
canvas.borrow_mut().create_image(image_source, image_flags)
}
}
.unwrap();
Some(Self::new_on_gpu(canvas, image_id))
}
ImageData::EmbeddedImage(_) => {
eprintln!("internal error: upload_to_gpu called on embedded image, which implies that the image was entered into the cache");
None
}
#[cfg(feature = "svg")]
ImageData::Svg(svg_tree) => match super::svg::render(svg_tree, target_size) {
Ok(rendered_svg_image) => Some(Self::new_on_cpu(rendered_svg_image)),
Err(err) => {
eprintln!("Error rendering SVG: {}", err);
None
}
},
#[cfg(target_arch = "wasm32")]
ImageData::HTMLImage(html_image) => html_image.size().map(|_| {
let image_id =
canvas.borrow_mut().create_image(&html_image.dom_element, image_flags).unwrap();
Self::new_on_gpu(canvas, image_id)
}),
}
}
pub fn size(&self) -> Option<Size> {
use image::GenericImageView;
match &*self.0.borrow() {
ImageData::Texture(texture) => texture.size(),
ImageData::DecodedImage(decoded_image) => {
let (width, height) = decoded_image.dimensions();
Some([width as f32, height as f32].into())
}
ImageData::EmbeddedImage(buffer) => {
Some([buffer.width() as f32, buffer.height() as f32].into())
}
#[cfg(feature = "svg")]
ImageData::Svg(tree) => {
let size = tree.svg_node().size.to_screen_size();
Some([size.width() as f32, size.height() as f32].into())
}
#[cfg(target_arch = "wasm32")]
ImageData::HTMLImage(html_image) => html_image.size(),
}
}
pub(crate) fn as_render_target(&self) -> femtovg::RenderTarget {
match &*self.0.borrow() {
ImageData::Texture(tex) => femtovg::RenderTarget::Image(tex.id),
_ => panic!(
"internal error: CachedImage::as_render_target() called on non-texture image"
),
}
}
pub(crate) fn filter(&self, canvas: &CanvasRc, filter: femtovg::ImageFilter) -> Self {
let (image_id, size) = match &*self.0.borrow() {
ImageData::Texture(texture) => texture.size().map(|size| (texture.id, size)),
_ => None,
}
.expect("internal error: Cannot filter non-GPU images");
let filtered_image = Self::new_empty_on_gpu(
canvas,
size.width.ceil() as usize,
size.height.ceil() as usize,
)
.expect(
"internal error: this can only fail if the filtered image was zero width or height",
);
let filtered_image_id = match &*filtered_image.0.borrow() {
ImageData::Texture(tex) => tex.id,
_ => panic!("internal error: CachedImage::new_empty_on_gpu did not return texture!"),
};
canvas.borrow_mut().filter_image(filtered_image_id, filter, image_id);
filtered_image
}
pub(crate) fn as_paint(&self) -> femtovg::Paint {
match &*self.0.borrow() {
ImageData::Texture(tex) => {
let size = tex
.size()
.expect("internal error: CachedImage::as_paint() called on zero-sized texture");
femtovg::Paint::image(tex.id, 0., 0., size.width, size.height, 0., 1.)
}
_ => panic!("internal error: CachedImage::as_paint() called on non-texture image"),
}
}
pub(crate) fn is_on_gpu(&self) -> bool {
matches!(&*self.0.borrow(), ImageData::Texture(_))
}
pub(crate) fn to_rgba(&self) -> Option<image::RgbaImage> {
if let ImageData::DecodedImage(img) = &*self.0.borrow() {
Some(img.to_rgba8())
} else {
None
}
}
}
#[derive(PartialEq, Eq, Hash, Debug, derive_more::From)]
enum CachedImageSourceKey {
Path(String),
EmbeddedData(by_address::ByAddress<&'static [u8]>),
}
#[derive(PartialEq, Eq, Hash, Debug)]
pub struct ImageCacheKey {
source_key: CachedImageSourceKey,
gpu_image_flags: ImageRendering,
}
impl ImageCacheKey {
pub fn new(resource: &ImageInner, gpu_image_flags: Option<ImageRendering>) -> Option<Self> {
Some(match resource {
ImageInner::None => return None,
ImageInner::AbsoluteFilePath(path) => {
if path.is_empty() {
return None;
}
Self {
source_key: path.to_string().into(),
gpu_image_flags: gpu_image_flags.unwrap_or_default(),
}
}
ImageInner::EmbeddedData { data, format: _ } => Self {
source_key: by_address::ByAddress(data.as_slice()).into(),
gpu_image_flags: gpu_image_flags.unwrap_or_default(),
},
ImageInner::EmbeddedImage { .. } => return None,
})
}
}
#[derive(Default)]
pub(crate) struct ImageCache(HashMap<ImageCacheKey, Rc<CachedImage>>);
impl ImageCache {
pub(crate) fn lookup_image_in_cache_or_create(
&mut self,
cache_key: ImageCacheKey,
image_create_fn: impl Fn() -> Option<Rc<CachedImage>>,
) -> Option<Rc<CachedImage>> {
Some(match self.0.entry(cache_key) {
std::collections::hash_map::Entry::Occupied(existing_entry) => {
existing_entry.get().clone()
}
std::collections::hash_map::Entry::Vacant(vacant_entry) => {
let new_image = image_create_fn()?;
vacant_entry.insert(new_image.clone());
new_image
}
})
}
pub(crate) fn load_image_resource(&mut self, resource: &ImageInner) -> Option<Rc<CachedImage>> {
ImageCacheKey::new(resource, None)
.and_then(|cache_key| {
self.lookup_image_in_cache_or_create(cache_key, || {
CachedImage::new_from_resource(resource).map(Rc::new)
})
})
.or_else(|| CachedImage::new_from_resource(resource).map(Rc::new))
}
pub(crate) fn drain(&mut self) {
self.0.retain(|_, cached_image| {
Rc::strong_count(cached_image) > 1 || cached_image.size().is_none()
});
}
pub(crate) fn remove_textures(&mut self) {
self.0.retain(|_, cached_image| !cached_image.is_on_gpu());
}
}
fn image_buffer_to_image_source(
buffer: &SharedImageBuffer,
) -> (femtovg::ImageSource<'_>, femtovg::ImageFlags) {
match buffer {
SharedImageBuffer::RGB8(buffer) => (
{ imgref::ImgRef::new(buffer.as_slice(), buffer.width(), buffer.height()).into() },
femtovg::ImageFlags::empty(),
),
SharedImageBuffer::RGBA8(buffer) => (
{ imgref::ImgRef::new(buffer.as_slice(), buffer.width(), buffer.height()).into() },
femtovg::ImageFlags::empty(),
),
SharedImageBuffer::RGBA8Premultiplied(buffer) => (
{ imgref::ImgRef::new(buffer.as_slice(), buffer.width(), buffer.height()).into() },
femtovg::ImageFlags::PREMULTIPLIED,
),
}
}