use bevy::prelude::*;
pub trait Lens<T> {
fn lerp(&mut self, target: &mut T, ratio: f32);
}
#[cfg(feature = "bevy_text")]
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct TextColorLens {
pub start: Color,
pub end: Color,
pub section: usize,
}
#[cfg(feature = "bevy_text")]
impl Lens<Text> for TextColorLens {
fn lerp(&mut self, target: &mut Text, ratio: f32) {
let start: Vec4 = self.start.into();
let end: Vec4 = self.end.into();
let value = start.lerp(end, ratio);
target.sections[self.section].style.color = value.into();
}
}
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct TransformPositionLens {
pub start: Vec3,
pub end: Vec3,
}
impl Lens<Transform> for TransformPositionLens {
fn lerp(&mut self, target: &mut Transform, ratio: f32) {
let value = self.start + (self.end - self.start) * ratio;
target.translation = value;
}
}
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct TransformRotationLens {
pub start: Quat,
pub end: Quat,
}
impl Lens<Transform> for TransformRotationLens {
fn lerp(&mut self, target: &mut Transform, ratio: f32) {
target.rotation = self.start.slerp(self.end, ratio);
}
}
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct TransformRotateXLens {
pub start: f32,
pub end: f32,
}
impl Lens<Transform> for TransformRotateXLens {
fn lerp(&mut self, target: &mut Transform, ratio: f32) {
let angle = (self.end - self.start).mul_add(ratio, self.start);
target.rotation = Quat::from_rotation_x(angle);
}
}
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct TransformRotateYLens {
pub start: f32,
pub end: f32,
}
impl Lens<Transform> for TransformRotateYLens {
fn lerp(&mut self, target: &mut Transform, ratio: f32) {
let angle = (self.end - self.start).mul_add(ratio, self.start);
target.rotation = Quat::from_rotation_y(angle);
}
}
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct TransformRotateZLens {
pub start: f32,
pub end: f32,
}
impl Lens<Transform> for TransformRotateZLens {
fn lerp(&mut self, target: &mut Transform, ratio: f32) {
let angle = (self.end - self.start).mul_add(ratio, self.start);
target.rotation = Quat::from_rotation_z(angle);
}
}
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct TransformRotateAxisLens {
pub axis: Vec3,
pub start: f32,
pub end: f32,
}
impl Lens<Transform> for TransformRotateAxisLens {
fn lerp(&mut self, target: &mut Transform, ratio: f32) {
let angle = (self.end - self.start).mul_add(ratio, self.start);
target.rotation = Quat::from_axis_angle(self.axis, angle);
}
}
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct TransformScaleLens {
pub start: Vec3,
pub end: Vec3,
}
impl Lens<Transform> for TransformScaleLens {
fn lerp(&mut self, target: &mut Transform, ratio: f32) {
let value = self.start + (self.end - self.start) * ratio;
target.scale = value;
}
}
#[cfg(feature = "bevy_ui")]
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct UiPositionLens {
pub start: UiRect,
pub end: UiRect,
}
#[cfg(feature = "bevy_ui")]
fn lerp_val(start: &Val, end: &Val, ratio: f32) -> Val {
match (start, end) {
(Val::Percent(start), Val::Percent(end)) => {
Val::Percent((end - start).mul_add(ratio, *start))
}
(Val::Px(start), Val::Px(end)) => Val::Px((end - start).mul_add(ratio, *start)),
_ => *start,
}
}
#[cfg(feature = "bevy_ui")]
impl Lens<Style> for UiPositionLens {
fn lerp(&mut self, target: &mut Style, ratio: f32) {
target.position = UiRect {
left: lerp_val(&self.start.left, &self.end.left, ratio),
right: lerp_val(&self.start.right, &self.end.right, ratio),
top: lerp_val(&self.start.top, &self.end.top, ratio),
bottom: lerp_val(&self.start.bottom, &self.end.bottom, ratio),
};
}
}
#[cfg(feature = "bevy_sprite")]
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct ColorMaterialColorLens {
pub start: Color,
pub end: Color,
}
#[cfg(feature = "bevy_sprite")]
impl Lens<ColorMaterial> for ColorMaterialColorLens {
fn lerp(&mut self, target: &mut ColorMaterial, ratio: f32) {
let start: Vec4 = self.start.into();
let end: Vec4 = self.end.into();
let value = start.lerp(end, ratio);
target.color = value.into();
}
}
#[cfg(feature = "bevy_sprite")]
#[derive(Debug, Copy, Clone, PartialEq)]
pub struct SpriteColorLens {
pub start: Color,
pub end: Color,
}
#[cfg(feature = "bevy_sprite")]
impl Lens<Sprite> for SpriteColorLens {
fn lerp(&mut self, target: &mut Sprite, ratio: f32) {
let start: Vec4 = self.start.into();
let end: Vec4 = self.end.into();
let value = start.lerp(end, ratio);
target.color = value.into();
}
}
#[cfg(test)]
mod tests {
use std::f32::consts::TAU;
use super::*;
#[cfg(feature = "bevy_text")]
#[test]
fn text_color() {
let mut lens = TextColorLens {
start: Color::RED,
end: Color::BLUE,
section: 0,
};
let mut text = Text::from_section("", default());
lens.lerp(&mut text, 0.);
assert_eq!(text.sections[0].style.color, Color::RED);
lens.lerp(&mut text, 1.);
assert_eq!(text.sections[0].style.color, Color::BLUE);
lens.lerp(&mut text, 0.3);
assert_eq!(text.sections[0].style.color, Color::rgba(0.7, 0., 0.3, 1.0));
}
#[test]
fn transform_position() {
let mut lens = TransformPositionLens {
start: Vec3::ZERO,
end: Vec3::new(1., 2., -4.),
};
let mut transform = Transform::default();
lens.lerp(&mut transform, 0.);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
lens.lerp(&mut transform, 1.);
assert!(transform
.translation
.abs_diff_eq(Vec3::new(1., 2., -4.), 1e-5));
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
lens.lerp(&mut transform, 0.3);
assert!(transform
.translation
.abs_diff_eq(Vec3::new(0.3, 0.6, -1.2), 1e-5));
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
#[test]
fn transform_rotation() {
let mut lens = TransformRotationLens {
start: Quat::IDENTITY,
end: Quat::from_rotation_z(100_f32.to_radians()),
};
let mut transform = Transform::default();
lens.lerp(&mut transform, 0.);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
lens.lerp(&mut transform, 1.);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform
.rotation
.abs_diff_eq(Quat::from_rotation_z(100_f32.to_radians()), 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
lens.lerp(&mut transform, 0.3);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform
.rotation
.abs_diff_eq(Quat::from_rotation_z(30_f32.to_radians()), 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
#[test]
fn transform_rotate_x() {
let mut lens = TransformRotateXLens {
start: 0.,
end: 1440_f32.to_radians(), };
let mut transform = Transform::default();
for (index, ratio) in [0., 0.25, 0.5, 0.75, 1.].iter().enumerate() {
lens.lerp(&mut transform, *ratio);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
if index == 1 || index == 3 {
assert!(transform
.rotation
.abs_diff_eq(Quat::from_xyzw(0., 0., 0., -1.), 1e-5));
} else {
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
}
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
lens.lerp(&mut transform, 0.1);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform
.rotation
.abs_diff_eq(Quat::from_rotation_x(0.1 * (4. * TAU)), 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
#[test]
fn transform_rotate_y() {
let mut lens = TransformRotateYLens {
start: 0.,
end: 1440_f32.to_radians(), };
let mut transform = Transform::default();
for (index, ratio) in [0., 0.25, 0.5, 0.75, 1.].iter().enumerate() {
lens.lerp(&mut transform, *ratio);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
if index == 1 || index == 3 {
assert!(transform
.rotation
.abs_diff_eq(Quat::from_xyzw(0., 0., 0., -1.), 1e-5));
} else {
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
}
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
lens.lerp(&mut transform, 0.1);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform
.rotation
.abs_diff_eq(Quat::from_rotation_y(0.1 * (4. * TAU)), 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
#[test]
fn transform_rotate_z() {
let mut lens = TransformRotateZLens {
start: 0.,
end: 1440_f32.to_radians(), };
let mut transform = Transform::default();
for (index, ratio) in [0., 0.25, 0.5, 0.75, 1.].iter().enumerate() {
lens.lerp(&mut transform, *ratio);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
if index == 1 || index == 3 {
assert!(transform
.rotation
.abs_diff_eq(Quat::from_xyzw(0., 0., 0., -1.), 1e-5));
} else {
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
}
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
lens.lerp(&mut transform, 0.1);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform
.rotation
.abs_diff_eq(Quat::from_rotation_z(0.1 * (4. * TAU)), 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
#[test]
fn transform_rotate_axis() {
let axis = Vec3::ONE.normalize();
let mut lens = TransformRotateAxisLens {
axis,
start: 0.,
end: 1440_f32.to_radians(), };
let mut transform = Transform::default();
for (index, ratio) in [0., 0.25, 0.5, 0.75, 1.].iter().enumerate() {
lens.lerp(&mut transform, *ratio);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
if index == 1 || index == 3 {
assert!(transform
.rotation
.abs_diff_eq(Quat::from_xyzw(0., 0., 0., -1.), 1e-5));
} else {
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
}
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
lens.lerp(&mut transform, 0.1);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform
.rotation
.abs_diff_eq(Quat::from_axis_angle(axis, 0.1 * (4. * TAU)), 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ONE, 1e-5));
}
#[test]
fn transform_scale() {
let mut lens = TransformScaleLens {
start: Vec3::ZERO,
end: Vec3::new(1., 2., -4.),
};
let mut transform = Transform::default();
lens.lerp(&mut transform, 0.);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::ZERO, 1e-5));
lens.lerp(&mut transform, 1.);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::new(1., 2., -4.), 1e-5));
lens.lerp(&mut transform, 0.3);
assert!(transform.translation.abs_diff_eq(Vec3::ZERO, 1e-5));
assert!(transform.rotation.abs_diff_eq(Quat::IDENTITY, 1e-5));
assert!(transform.scale.abs_diff_eq(Vec3::new(0.3, 0.6, -1.2), 1e-5));
}
#[cfg(feature = "bevy_ui")]
#[test]
fn ui_position() {
let mut lens = UiPositionLens {
start: UiRect {
left: Val::Px(0.),
top: Val::Px(0.),
right: Val::Auto,
bottom: Val::Percent(25.),
},
end: UiRect {
left: Val::Px(1.),
top: Val::Px(5.),
right: Val::Auto,
bottom: Val::Percent(45.),
},
};
let mut style = Style::default();
lens.lerp(&mut style, 0.);
assert_eq!(style.position.left, Val::Px(0.));
assert_eq!(style.position.top, Val::Px(0.));
assert_eq!(style.position.right, Val::Auto);
assert_eq!(style.position.bottom, Val::Percent(25.));
lens.lerp(&mut style, 1.);
assert_eq!(style.position.left, Val::Px(1.));
assert_eq!(style.position.top, Val::Px(5.));
assert_eq!(style.position.right, Val::Auto);
assert_eq!(style.position.bottom, Val::Percent(45.));
lens.lerp(&mut style, 0.3);
assert_eq!(style.position.left, Val::Px(0.3));
assert_eq!(style.position.top, Val::Px(1.5));
assert_eq!(style.position.right, Val::Auto);
assert_eq!(style.position.bottom, Val::Percent(31.));
}
#[cfg(feature = "bevy_sprite")]
#[test]
fn colormaterial_color() {
let mut lens = ColorMaterialColorLens {
start: Color::RED,
end: Color::BLUE,
};
let mut mat = ColorMaterial {
color: Color::WHITE,
texture: None,
};
lens.lerp(&mut mat, 0.);
assert_eq!(mat.color, Color::RED);
lens.lerp(&mut mat, 1.);
assert_eq!(mat.color, Color::BLUE);
lens.lerp(&mut mat, 0.3);
assert_eq!(mat.color, Color::rgba(0.7, 0., 0.3, 1.0));
}
#[cfg(feature = "bevy_sprite")]
#[test]
fn sprite_color() {
let mut lens = SpriteColorLens {
start: Color::RED,
end: Color::BLUE,
};
let mut sprite = Sprite {
color: Color::WHITE,
..default()
};
lens.lerp(&mut sprite, 0.);
assert_eq!(sprite.color, Color::RED);
lens.lerp(&mut sprite, 1.);
assert_eq!(sprite.color, Color::BLUE);
lens.lerp(&mut sprite, 0.3);
assert_eq!(sprite.color, Color::rgba(0.7, 0., 0.3, 1.0));
}
}