use freya_engine::prelude::*;
use freya_native_core::real_dom::NodeImmutable;
use freya_node_state::{
AspectRatio,
ReferencesState,
StyleState,
TransformState,
};
use super::utils::ElementUtils;
use crate::dom::DioxusNode;
pub struct ImageElement;
impl ElementUtils for ImageElement {
fn render(
self,
layout_node: &torin::prelude::LayoutNode,
node_ref: &DioxusNode,
canvas: &Canvas,
_font_collection: &mut FontCollection,
_font_manager: &FontMgr,
_default_fonts: &[String],
_scale_factor: f32,
) {
let area = layout_node.visible_area();
let node_style = node_ref.get::<StyleState>().unwrap();
let node_references = node_ref.get::<ReferencesState>().unwrap();
let node_transform = node_ref.get::<TransformState>().unwrap();
let draw_img = |bytes: &[u8]| {
let pic = Image::from_encoded(unsafe { Data::new_bytes(bytes) });
if let Some(pic) = pic {
let mut paint = Paint::default();
paint.set_anti_alias(true);
let width_ratio = area.width() / pic.width() as f32;
let height_ratio = area.height() / pic.height() as f32;
let (width, height) = match node_transform.aspect_ratio {
AspectRatio::Max => {
let ratio = width_ratio.max(height_ratio);
(pic.width() as f32 * ratio, pic.height() as f32 * ratio)
}
AspectRatio::Min => {
let ratio = width_ratio.min(height_ratio);
(pic.width() as f32 * ratio, pic.height() as f32 * ratio)
}
AspectRatio::None => (area.width(), area.height()),
};
let rect = Rect::new(
area.min_x(),
area.min_y(),
area.min_x() + width,
area.min_y() + height,
);
canvas.draw_image_rect(pic, None, rect, &paint);
}
};
if let Some(image_ref) = &node_references.image_ref {
let image_data = image_ref.0.lock().unwrap();
if let Some(image_data) = image_data.as_ref() {
draw_img(image_data)
}
} else if let Some(image_data) = &node_style.image_data {
draw_img(image_data.as_slice())
}
}
}