Directional navigation now considers UiTransform rotation

crates/bevy_input_focus/src/directional_navigation.rs

@@ -66,7 +66,7 @@ use bevy_ecs::{
     prelude::*,
     system::SystemParam,
 };
-use bevy_math::{CompassOctant, Dir2, Rect, Vec2};
+use bevy_math::{ops, CompassOctant, Dir2, Rect, Vec2};
 use bevy_ui::{ComputedNode, ComputedUiTargetCamera, UiGlobalTransform};
 use thiserror::Error;
 
@@ -402,7 +402,17 @@ impl DirectionalNavigationMap {
         self.neighbors.get(&entity)
     }
 }
-
+fn get_rotated_bounds(size: Vec2, rotation: f32) -> Vec2 {
+    if rotation == 0.0 {
+        return size;
+    }
+    let cos_r = ops::cos(rotation).abs();
+    let sin_r = ops::sin(rotation).abs();
+    Vec2::new(
+        size.x * cos_r + size.y * sin_r,
+        size.x * sin_r + size.y * cos_r,
+    )
+}
 /// A system parameter for navigating between focusable entities in a directional way.
 #[derive(SystemParam, Debug)]
 pub struct DirectionalNavigation<'w, 's> {
@@ -492,12 +502,13 @@ impl<'w, 's> DirectionalNavigation<'w, 's> {
                     if let Some(tc) = computed_target_camera.get()
                         && tc == target_camera
                     {
-                        let (_scale, _rotation, translation) =
-                            transform.to_scale_angle_translation();
+                        let (scale, rotation, translation) = transform.to_scale_angle_translation();
+                        let scaled_size = computed.size() * computed.inverse_scale_factor() * scale;
+                        let rotated_size = get_rotated_bounds(scaled_size, rotation);
                         Some(FocusableArea {
                             entity,
                             position: translation * computed.inverse_scale_factor(),
-                            size: computed.size() * computed.inverse_scale_factor(),
+                            size: rotated_size,
                         })
                     } else {
                         // The node either does not have a target camera or it is not the same as the desired one.
@@ -520,13 +531,13 @@ impl<'w, 's> DirectionalNavigation<'w, 's> {
             None,
             |(entity, computed_target_camera, computed, transform)| {
                 if let Some(target_camera) = computed_target_camera.get() {
-                    let (_scale, _rotation, translation) = transform.to_scale_angle_translation();
+                    let (scale, _rotation, translation) = transform.to_scale_angle_translation();
                     Some((
                         target_camera,
                         FocusableArea {
                             entity,
                             position: translation * computed.inverse_scale_factor(),
-                            size: computed.size() * computed.inverse_scale_factor(),
+                            size: computed.size() * computed.inverse_scale_factor() * scale,
                         },
                     ))
                 } else {

examples/ui/auto_directional_navigation.rs

@@ -227,6 +227,8 @@ fn setup_scattered_ui(mut commands: Commands, mut input_focus: ResMut<InputFocus
                     border_radius: BorderRadius::all(px(12)),
                     ..default()
                 },
+                // Apply a scale transform to demonstrate that navigation handles transforms
+                Transform::from_scale(Vec3::splat(2.0)),
                 // This is the key: just add this component for automatic navigation!
                 AutoDirectionalNavigation::default(),
                 ResetTimer::default(),