#![warn(missing_docs)]
#![doc = include_str!("../README.md")]
use std::f32::consts::PI;
use bevy::camera::{CameraUpdateSystems, RenderTarget};
use bevy::input::gestures::PinchGesture;
use bevy::input::mouse::MouseWheel;
use bevy::prelude::*;
use bevy::transform::TransformSystems;
use bevy::window::{PrimaryWindow, WindowRef};
#[cfg(feature = "bevy_egui")]
use bevy_egui::EguiPreUpdateSet;
#[cfg(feature = "bevy_egui")]
pub use crate::egui::{EguiFocusIncludesHover, EguiWantsFocus};
use crate::input::{button_zoom_just_pressed, mouse_key_tracker, MouseKeyTracker};
pub use crate::touch::TouchControls;
use crate::touch::{touch_tracker, TouchGestures, TouchTracker};
use crate::traits::OptionalClamp;
#[cfg(feature = "bevy_egui")]
mod egui;
mod input;
mod touch;
mod traits;
mod util;
pub struct PanOrbitCameraPlugin;
impl Plugin for PanOrbitCameraPlugin {
fn build(&self, app: &mut App) {
app.init_resource::<ActiveCameraData>()
.init_resource::<MouseKeyTracker>()
.init_resource::<TouchTracker>()
.add_systems(
PostUpdate,
(
(
active_viewport_data
.run_if(|active_cam: Res<ActiveCameraData>| !active_cam.manual),
mouse_key_tracker,
touch_tracker,
),
pan_orbit_camera,
)
.chain()
.in_set(PanOrbitCameraSystemSet)
.before(TransformSystems::Propagate)
.before(CameraUpdateSystems),
);
#[cfg(feature = "bevy_egui")]
{
app.init_resource::<EguiWantsFocus>()
.init_resource::<EguiFocusIncludesHover>()
.add_systems(
PostUpdate,
egui::check_egui_wants_focus
.after(EguiPreUpdateSet::InitContexts)
.before(PanOrbitCameraSystemSet),
);
}
}
}
#[derive(SystemSet, Debug, Hash, PartialEq, Eq, Clone)]
pub struct PanOrbitCameraSystemSet;
#[derive(Component, Reflect, Copy, Clone, Debug, PartialEq)]
#[require(Camera3d)]
pub struct PanOrbitCamera {
pub focus: Vec3,
pub radius: Option<f32>,
pub yaw: Option<f32>,
pub pitch: Option<f32>,
pub target_focus: Vec3,
pub target_yaw: f32,
pub target_pitch: f32,
pub target_radius: f32,
pub yaw_upper_limit: Option<f32>,
pub yaw_lower_limit: Option<f32>,
pub pitch_upper_limit: Option<f32>,
pub pitch_lower_limit: Option<f32>,
pub focus_bounds_origin: Vec3,
pub focus_bounds_shape: Option<FocusBoundsShape>,
pub zoom_upper_limit: Option<f32>,
pub zoom_lower_limit: f32,
pub orbit_sensitivity: f32,
pub orbit_smoothness: f32,
pub pan_sensitivity: f32,
pub pan_smoothness: f32,
pub zoom_sensitivity: f32,
pub zoom_smoothness: f32,
pub button_orbit: MouseButton,
pub button_pan: MouseButton,
pub button_zoom: Option<MouseButton>,
pub button_zoom_axis: ButtonZoomAxis,
pub modifier_orbit: Option<KeyCode>,
pub modifier_pan: Option<KeyCode>,
pub touch_enabled: bool,
pub touch_controls: TouchControls,
pub trackpad_behavior: TrackpadBehavior,
pub trackpad_pinch_to_zoom_enabled: bool,
pub trackpad_sensitivity: f32,
pub reversed_zoom: bool,
pub is_upside_down: bool,
pub allow_upside_down: bool,
pub enabled: bool,
pub initialized: bool,
pub force_update: bool,
pub axis: [Vec3; 3],
pub use_real_time: bool,
}
impl Default for PanOrbitCamera {
fn default() -> Self {
PanOrbitCamera {
focus: Vec3::ZERO,
target_focus: Vec3::ZERO,
radius: None,
is_upside_down: false,
allow_upside_down: false,
orbit_sensitivity: 1.0,
orbit_smoothness: 0.1,
pan_sensitivity: 1.0,
pan_smoothness: 0.02,
zoom_sensitivity: 1.0,
zoom_smoothness: 0.1,
button_orbit: MouseButton::Left,
button_pan: MouseButton::Right,
button_zoom: None,
button_zoom_axis: ButtonZoomAxis::Y,
modifier_orbit: None,
modifier_pan: None,
touch_enabled: true,
touch_controls: TouchControls::OneFingerOrbit,
trackpad_behavior: TrackpadBehavior::Default,
trackpad_pinch_to_zoom_enabled: false,
trackpad_sensitivity: 1.0,
reversed_zoom: false,
enabled: true,
yaw: None,
pitch: None,
target_yaw: 0.0,
target_pitch: 0.0,
target_radius: 1.0,
initialized: false,
yaw_upper_limit: None,
yaw_lower_limit: None,
pitch_upper_limit: None,
pitch_lower_limit: None,
focus_bounds_origin: Vec3::ZERO,
focus_bounds_shape: None,
zoom_upper_limit: None,
zoom_lower_limit: 0.05,
force_update: false,
axis: [Vec3::X, Vec3::Y, Vec3::Z],
use_real_time: false,
}
}
}
#[derive(Resource, Default, Debug, PartialEq)]
pub struct ActiveCameraData {
pub entity: Option<Entity>,
pub viewport_size: Option<Vec2>,
pub window_size: Option<Vec2>,
pub manual: bool,
}
#[derive(Clone, PartialEq, Debug, Reflect, Copy)]
pub enum FocusBoundsShape {
Sphere(Sphere),
Cuboid(Cuboid),
}
#[derive(Clone, PartialEq, Debug, Reflect, Copy)]
pub enum ButtonZoomAxis {
X,
Y,
XY,
}
impl From<Sphere> for FocusBoundsShape {
fn from(value: Sphere) -> Self {
Self::Sphere(value)
}
}
impl From<Cuboid> for FocusBoundsShape {
fn from(value: Cuboid) -> Self {
Self::Cuboid(value)
}
}
#[derive(Clone, PartialEq, Debug, Reflect, Copy)]
pub enum TrackpadBehavior {
Default,
BlenderLike {
modifier_pan: Option<KeyCode>,
modifier_zoom: Option<KeyCode>,
},
}
impl TrackpadBehavior {
pub fn blender_default() -> Self {
Self::BlenderLike {
modifier_pan: Some(KeyCode::ShiftLeft),
modifier_zoom: Some(KeyCode::ControlLeft),
}
}
}
#[allow(clippy::too_many_arguments)]
fn active_viewport_data(
mut active_cam: ResMut<ActiveCameraData>,
mouse_input: Res<ButtonInput<MouseButton>>,
key_input: Res<ButtonInput<KeyCode>>,
pinch_events: MessageReader<PinchGesture>,
scroll_events: MessageReader<MouseWheel>,
touches: Res<Touches>,
primary_windows: Query<&Window, With<PrimaryWindow>>,
other_windows: Query<&Window, Without<PrimaryWindow>>,
orbit_cameras: Query<(Entity, &Camera, &PanOrbitCamera)>,
#[cfg(feature = "bevy_egui")] egui_wants_focus: Res<EguiWantsFocus>,
) {
let mut new_resource = ActiveCameraData::default();
let mut max_cam_order = 0;
let mut has_input = false;
for (entity, camera, pan_orbit) in orbit_cameras.iter() {
let input_just_activated = input::orbit_just_pressed(pan_orbit, &mouse_input, &key_input)
|| input::pan_just_pressed(pan_orbit, &mouse_input, &key_input)
|| !pinch_events.is_empty()
|| !scroll_events.is_empty()
|| button_zoom_just_pressed(pan_orbit, &mouse_input)
|| (touches.iter_just_pressed().count() > 0
&& touches.iter_just_pressed().count() == touches.iter().count());
if input_just_activated {
has_input = true;
#[allow(unused_mut, unused_assignments)]
let mut should_get_input = true;
#[cfg(feature = "bevy_egui")]
{
should_get_input = !egui_wants_focus.prev && !egui_wants_focus.curr;
}
if should_get_input {
if let RenderTarget::Window(win_ref) = camera.target {
let Some(window) = (match win_ref {
WindowRef::Primary => primary_windows.single().ok(),
WindowRef::Entity(entity) => other_windows.get(entity).ok(),
}) else {
continue;
};
if let Some(input_position) = window.cursor_position().or(touches
.iter_just_pressed()
.collect::<Vec<_>>()
.first()
.map(|touch| touch.position()))
{
if let Some(Rect { min, max }) = camera.logical_viewport_rect() {
let cursor_in_vp = input_position.x > min.x
&& input_position.x < max.x
&& input_position.y > min.y
&& input_position.y < max.y;
if cursor_in_vp && camera.order >= max_cam_order {
new_resource = ActiveCameraData {
entity: Some(entity),
viewport_size: camera.logical_viewport_size(),
window_size: Some(Vec2::new(window.width(), window.height())),
manual: false,
};
max_cam_order = camera.order;
}
}
}
}
}
}
}
if has_input {
active_cam.set_if_neq(new_resource);
}
}
fn pan_orbit_camera(
active_cam: Res<ActiveCameraData>,
mouse_key_tracker: Res<MouseKeyTracker>,
touch_tracker: Res<TouchTracker>,
mut orbit_cameras: Query<(Entity, &mut PanOrbitCamera, &mut Transform, &mut Projection)>,
time_real: Res<Time<Real>>,
time_virt: Res<Time<Virtual>>,
) {
for (entity, mut pan_orbit, mut transform, mut projection) in orbit_cameras.iter_mut() {
let apply_zoom_limits = {
let zoom_upper_limit = pan_orbit.zoom_upper_limit;
let zoom_lower_limit = pan_orbit.zoom_lower_limit;
move |zoom: f32| zoom.clamp_optional(Some(zoom_lower_limit), zoom_upper_limit)
};
let apply_yaw_limits = {
let yaw_upper_limit = pan_orbit.yaw_upper_limit;
let yaw_lower_limit = pan_orbit.yaw_lower_limit;
move |yaw: f32| yaw.clamp_optional(yaw_lower_limit, yaw_upper_limit)
};
let apply_pitch_limits = {
let pitch_upper_limit = pan_orbit.pitch_upper_limit;
let pitch_lower_limit = pan_orbit.pitch_lower_limit;
move |pitch: f32| pitch.clamp_optional(pitch_lower_limit, pitch_upper_limit)
};
let apply_focus_limits = {
let origin = pan_orbit.focus_bounds_origin;
let shape = pan_orbit.focus_bounds_shape;
move |focus: Vec3| {
let Some(shape) = shape else {
return focus;
};
match shape {
FocusBoundsShape::Cuboid(shape) => shape.closest_point(focus - origin) + origin,
FocusBoundsShape::Sphere(shape) => shape.closest_point(focus - origin) + origin,
}
}
};
if !pan_orbit.initialized {
let (yaw, pitch, radius) = util::calculate_from_translation_and_focus(
transform.translation,
pan_orbit.focus,
pan_orbit.axis,
);
let &mut mut yaw = pan_orbit.yaw.get_or_insert(yaw);
let &mut mut pitch = pan_orbit.pitch.get_or_insert(pitch);
let &mut mut radius = pan_orbit.radius.get_or_insert(radius);
let mut focus = pan_orbit.focus;
yaw = apply_yaw_limits(yaw);
pitch = apply_pitch_limits(pitch);
radius = apply_zoom_limits(radius);
focus = apply_focus_limits(focus);
pan_orbit.yaw = Some(yaw);
pan_orbit.pitch = Some(pitch);
pan_orbit.radius = Some(radius);
pan_orbit.target_yaw = yaw;
pan_orbit.target_pitch = pitch;
pan_orbit.target_radius = radius;
pan_orbit.target_focus = focus;
util::update_orbit_transform(
yaw,
pitch,
radius,
focus,
&mut transform,
&mut projection,
pan_orbit.axis,
);
pan_orbit.initialized = true;
}
let mut orbit = Vec2::ZERO;
let mut pan = Vec2::ZERO;
let mut scroll_line = 0.0;
let mut scroll_pixel = 0.0;
let mut orbit_button_changed = false;
if pan_orbit.enabled && active_cam.entity == Some(entity) {
let zoom_direction = match pan_orbit.reversed_zoom {
true => -1.0,
false => 1.0,
};
orbit = mouse_key_tracker.orbit * pan_orbit.orbit_sensitivity;
pan = mouse_key_tracker.pan * pan_orbit.pan_sensitivity;
scroll_line =
mouse_key_tracker.scroll_line * zoom_direction * pan_orbit.zoom_sensitivity;
scroll_pixel =
mouse_key_tracker.scroll_pixel * zoom_direction * pan_orbit.zoom_sensitivity;
orbit_button_changed = mouse_key_tracker.orbit_button_changed;
if pan_orbit.touch_enabled {
let (touch_orbit, touch_pan, touch_zoom_pixel) = match pan_orbit.touch_controls {
TouchControls::OneFingerOrbit => match touch_tracker.get_touch_gestures() {
TouchGestures::None => (Vec2::ZERO, Vec2::ZERO, 0.0),
TouchGestures::OneFinger(one_finger_gestures) => {
(one_finger_gestures.motion, Vec2::ZERO, 0.0)
}
TouchGestures::TwoFinger(two_finger_gestures) => (
Vec2::ZERO,
two_finger_gestures.motion,
two_finger_gestures.pinch * 0.015,
),
},
TouchControls::TwoFingerOrbit => match touch_tracker.get_touch_gestures() {
TouchGestures::None => (Vec2::ZERO, Vec2::ZERO, 0.0),
TouchGestures::OneFinger(one_finger_gestures) => {
(Vec2::ZERO, one_finger_gestures.motion, 0.0)
}
TouchGestures::TwoFinger(two_finger_gestures) => (
two_finger_gestures.motion,
Vec2::ZERO,
two_finger_gestures.pinch * 0.015,
),
},
};
orbit += touch_orbit * pan_orbit.orbit_sensitivity;
pan += touch_pan * pan_orbit.pan_sensitivity;
scroll_pixel += touch_zoom_pixel * zoom_direction * pan_orbit.zoom_sensitivity;
}
}
if orbit_button_changed {
let world_up = pan_orbit.axis[1];
pan_orbit.is_upside_down = transform.up().dot(world_up) < 0.0;
}
let mut has_moved = false;
if orbit.length_squared() > 0.0 {
if let Some(win_size) = active_cam.window_size {
let delta_x = {
let delta = orbit.x / win_size.x * PI * 2.0;
if pan_orbit.is_upside_down {
-delta
} else {
delta
}
};
let delta_y = orbit.y / win_size.y * PI;
pan_orbit.target_yaw -= delta_x;
pan_orbit.target_pitch += delta_y;
has_moved = true;
}
}
if pan.length_squared() > 0.0 {
if let Some(vp_size) = active_cam.viewport_size {
let mut multiplier = 1.0;
match *projection {
Projection::Perspective(ref p) => {
pan *= Vec2::new(p.fov * p.aspect_ratio, p.fov) / vp_size;
if let Some(radius) = pan_orbit.radius {
multiplier = radius;
}
}
Projection::Orthographic(ref p) => {
pan *= Vec2::new(p.area.width(), p.area.height()) / vp_size;
}
Projection::Custom(_) => todo!(),
}
let right = transform.rotation * pan_orbit.axis[0] * -pan.x;
let up = transform.rotation * pan_orbit.axis[1] * pan.y;
let translation = (right + up) * multiplier;
pan_orbit.target_focus += translation;
has_moved = true;
}
}
if (scroll_line + scroll_pixel).abs() > 0.0 {
let line_delta = -scroll_line * (pan_orbit.target_radius) * 0.2;
let pixel_delta = -scroll_pixel * (pan_orbit.target_radius) * 0.2;
pan_orbit.target_radius += line_delta + pixel_delta;
pan_orbit.radius = pan_orbit
.radius
.map(|value| apply_zoom_limits(value + pixel_delta));
has_moved = true;
}
pan_orbit.target_yaw = apply_yaw_limits(pan_orbit.target_yaw);
pan_orbit.target_pitch = apply_pitch_limits(pan_orbit.target_pitch);
pan_orbit.target_radius = apply_zoom_limits(pan_orbit.target_radius);
pan_orbit.target_focus = apply_focus_limits(pan_orbit.target_focus);
if !pan_orbit.allow_upside_down {
pan_orbit.target_pitch = pan_orbit.target_pitch.clamp(-PI / 2.0, PI / 2.0);
}
let delta = if pan_orbit.use_real_time {
time_real.delta_secs()
} else {
time_virt.delta_secs()
};
if let (Some(yaw), Some(pitch), Some(radius)) =
(pan_orbit.yaw, pan_orbit.pitch, pan_orbit.radius)
{
if has_moved
|| pan_orbit.target_yaw != yaw
|| pan_orbit.target_pitch != pitch
|| pan_orbit.target_radius != radius
|| pan_orbit.target_focus != pan_orbit.focus
|| pan_orbit.force_update
{
let new_yaw = util::lerp_and_snap_f32(
yaw,
pan_orbit.target_yaw,
pan_orbit.orbit_smoothness,
delta,
);
let new_pitch = util::lerp_and_snap_f32(
pitch,
pan_orbit.target_pitch,
pan_orbit.orbit_smoothness,
delta,
);
let new_radius = util::lerp_and_snap_f32(
radius,
pan_orbit.target_radius,
pan_orbit.zoom_smoothness,
delta,
);
let new_focus = util::lerp_and_snap_vec3(
pan_orbit.focus,
pan_orbit.target_focus,
pan_orbit.pan_smoothness,
delta,
);
util::update_orbit_transform(
new_yaw,
new_pitch,
new_radius,
new_focus,
&mut transform,
&mut projection,
pan_orbit.axis,
);
pan_orbit.yaw = Some(new_yaw);
pan_orbit.pitch = Some(new_pitch);
pan_orbit.radius = Some(new_radius);
pan_orbit.focus = new_focus;
pan_orbit.force_update = false;
}
}
}
}