use crate::{
props::Props,
widget::{
node::WidgetNode,
unit::WidgetUnitData,
utils::{Color, Rect, Transform, Vec2},
WidgetId,
},
PrefabValue, Scalar,
};
use serde::{Deserialize, Serialize};
use std::{collections::HashMap, convert::TryFrom};
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub struct ImageBoxFrame {
#[serde(default)]
pub source: Rect,
#[serde(default)]
pub destination: Rect,
#[serde(default)]
pub frame_only: bool,
#[serde(default)]
pub frame_keep_aspect_ratio: bool,
}
impl From<Scalar> for ImageBoxFrame {
fn from(v: Scalar) -> Self {
Self {
source: v.into(),
destination: v.into(),
frame_only: false,
frame_keep_aspect_ratio: false,
}
}
}
impl From<(Scalar, bool)> for ImageBoxFrame {
fn from((v, fo): (Scalar, bool)) -> Self {
Self {
source: v.into(),
destination: v.into(),
frame_only: fo,
frame_keep_aspect_ratio: false,
}
}
}
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub enum ImageBoxImageScaling {
#[default]
Stretch,
Frame(ImageBoxFrame),
}
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub struct ImageBoxColor {
#[serde(default)]
pub color: Color,
#[serde(default)]
pub scaling: ImageBoxImageScaling,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ImageBoxImage {
#[serde(default)]
pub id: String,
#[serde(default)]
#[serde(skip_serializing_if = "Option::is_none")]
pub source_rect: Option<Rect>,
#[serde(default)]
pub scaling: ImageBoxImageScaling,
#[serde(default = "ImageBoxImage::default_tint")]
pub tint: Color,
}
impl Default for ImageBoxImage {
fn default() -> Self {
Self {
id: Default::default(),
source_rect: Default::default(),
scaling: Default::default(),
tint: Self::default_tint(),
}
}
}
impl ImageBoxImage {
fn default_tint() -> Color {
Color {
r: 1.0,
g: 1.0,
b: 1.0,
a: 1.0,
}
}
}
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub struct ImageBoxProceduralVertex {
#[serde(default)]
pub position: Vec2,
#[serde(default)]
pub page: Scalar,
#[serde(default)]
pub tex_coord: Vec2,
#[serde(default)]
pub color: Color,
}
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub struct ImageBoxProcedural {
#[serde(default)]
pub id: String,
#[serde(default)]
#[serde(skip_serializing_if = "HashMap::is_empty")]
pub parameters: HashMap<String, Scalar>,
#[serde(default)]
#[serde(skip_serializing_if = "Vec::is_empty")]
pub images: Vec<String>,
#[serde(default)]
#[serde(skip_serializing_if = "Vec::is_empty")]
pub vertices: Vec<ImageBoxProceduralVertex>,
#[serde(default)]
#[serde(skip_serializing_if = "Vec::is_empty")]
pub triangles: Vec<[u32; 3]>,
#[serde(default)]
pub fit_to_rect: bool,
}
impl ImageBoxProcedural {
pub fn new(id: impl ToString) -> Self {
Self {
id: id.to_string(),
parameters: Default::default(),
images: Default::default(),
vertices: Default::default(),
triangles: Default::default(),
fit_to_rect: false,
}
}
pub fn param(mut self, id: impl ToString, value: Scalar) -> Self {
self.parameters.insert(id.to_string(), value);
self
}
pub fn image(mut self, id: impl ToString) -> Self {
self.images.push(id.to_string());
self
}
pub fn triangle(mut self, vertices: [ImageBoxProceduralVertex; 3]) -> Self {
let count = self.vertices.len() as u32;
self.vertices.extend(vertices);
self.triangles.push([count, count + 1, count + 2]);
self
}
pub fn quad(mut self, vertices: [ImageBoxProceduralVertex; 4]) -> Self {
let count = self.vertices.len() as u32;
self.vertices.extend(vertices);
self.triangles.push([count, count + 1, count + 2]);
self.triangles.push([count + 2, count + 3, count]);
self
}
pub fn fit_to_rect(mut self, value: bool) -> Self {
self.fit_to_rect = value;
self
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum ImageBoxMaterial {
Color(ImageBoxColor),
Image(ImageBoxImage),
Procedural(ImageBoxProcedural),
}
impl Default for ImageBoxMaterial {
fn default() -> Self {
Self::Color(Default::default())
}
}
#[derive(Debug, Default, Copy, Clone, Serialize, Deserialize)]
pub enum ImageBoxSizeValue {
#[default]
Fill,
Exact(Scalar),
}
#[derive(Debug, Default, Copy, Clone, Serialize, Deserialize)]
pub struct ImageBoxAspectRatio {
#[serde(default)]
pub horizontal_alignment: Scalar,
#[serde(default)]
pub vertical_alignment: Scalar,
#[serde(default)]
pub outside: bool,
}
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub struct ImageBox {
#[serde(default)]
pub id: WidgetId,
#[serde(default)]
pub width: ImageBoxSizeValue,
#[serde(default)]
pub height: ImageBoxSizeValue,
#[serde(default)]
#[serde(skip_serializing_if = "Option::is_none")]
pub content_keep_aspect_ratio: Option<ImageBoxAspectRatio>,
#[serde(default)]
pub material: ImageBoxMaterial,
#[serde(default)]
pub transform: Transform,
}
impl WidgetUnitData for ImageBox {
fn id(&self) -> &WidgetId {
&self.id
}
}
impl TryFrom<ImageBoxNode> for ImageBox {
type Error = ();
fn try_from(node: ImageBoxNode) -> Result<Self, Self::Error> {
let ImageBoxNode {
id,
width,
height,
content_keep_aspect_ratio,
material,
transform,
..
} = node;
Ok(Self {
id,
width,
height,
content_keep_aspect_ratio,
material,
transform,
})
}
}
#[derive(Debug, Default, Clone)]
pub struct ImageBoxNode {
pub id: WidgetId,
pub props: Props,
pub width: ImageBoxSizeValue,
pub height: ImageBoxSizeValue,
pub content_keep_aspect_ratio: Option<ImageBoxAspectRatio>,
pub material: ImageBoxMaterial,
pub transform: Transform,
}
impl ImageBoxNode {
pub fn remap_props<F>(&mut self, mut f: F)
where
F: FnMut(Props) -> Props,
{
let props = std::mem::take(&mut self.props);
self.props = (f)(props);
}
}
impl From<ImageBoxNode> for WidgetNode {
fn from(data: ImageBoxNode) -> Self {
Self::Unit(data.into())
}
}
#[derive(Debug, Default, Clone, Serialize, Deserialize)]
pub(crate) struct ImageBoxNodePrefab {
#[serde(default)]
pub id: WidgetId,
#[serde(default)]
pub props: PrefabValue,
#[serde(default)]
pub width: ImageBoxSizeValue,
#[serde(default)]
pub height: ImageBoxSizeValue,
#[serde(default)]
#[serde(skip_serializing_if = "Option::is_none")]
pub content_keep_aspect_ratio: Option<ImageBoxAspectRatio>,
#[serde(default)]
pub material: ImageBoxMaterial,
#[serde(default)]
pub transform: Transform,
}