use crate::tab::{MIN_TERMINAL_HEIGHT, MIN_TERMINAL_WIDTH}; use crate::{panes::PaneId, tab::Pane}; use std::cmp::Ordering; use std::collections::HashMap; use zellij_utils::data::{Direction, ResizeStrategy}; use zellij_utils::errors::prelude::*; use zellij_utils::pane_size::{Dimension, PaneGeom, Size, Viewport}; use std::cell::RefCell; use std::rc::Rc; const MOVE_INCREMENT_HORIZONTAL: usize = 10; const MOVE_INCREMENT_VERTICAL: usize = 5; const MAX_PANES: usize = 100; // For error reporting fn no_pane_id(pane_id: &PaneId) -> String { format!("no floating pane with ID {:?} found", pane_id) } pub struct FloatingPaneGrid<'a> { panes: Rc>>>, desired_pane_positions: Rc>>, display_area: Size, // includes all panes (including eg. the status bar and tab bar in the default layout) viewport: Viewport, // includes all non-UI panes } impl<'a> FloatingPaneGrid<'a> { pub fn new( panes: impl IntoIterator)>, desired_pane_positions: &'a mut HashMap, display_area: Size, viewport: Viewport, ) -> Self { let panes: HashMap<_, _> = panes.into_iter().map(|(p_id, p)| (*p_id, p)).collect(); FloatingPaneGrid { panes: Rc::new(RefCell::new(panes)), desired_pane_positions: Rc::new(RefCell::new(desired_pane_positions)), display_area, viewport, } } pub fn move_pane_by(&mut self, pane_id: PaneId, x: isize, y: isize) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} by ({x}, {y})"); // true => succeeded to move, false => failed to move let new_pane_position = { let mut panes = self.panes.borrow_mut(); let pane = panes .iter_mut() .find(|(p_id, _p)| **p_id == pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)? .1; let mut new_pane_position = pane.position_and_size(); let min_x = self.viewport.x as isize; let min_y = self.viewport.y as isize; let max_x = (self.viewport.cols + self.viewport.x) .saturating_sub(new_pane_position.cols.as_usize()); let max_y = (self.viewport.rows + self.viewport.y) .saturating_sub(new_pane_position.rows.as_usize()); let new_x = std::cmp::max(min_x, new_pane_position.x as isize + x); let new_x = std::cmp::min(new_x, max_x as isize); let new_y = std::cmp::max(min_y, new_pane_position.y as isize + y); let new_y = std::cmp::min(new_y, max_y as isize); new_pane_position.x = new_x as usize; new_pane_position.y = new_y as usize; new_pane_position }; self.set_pane_geom(pane_id, new_pane_position) .with_context(err_context) } fn set_pane_geom(&mut self, pane_id: PaneId, new_pane_geom: PaneGeom) -> Result<()> { let err_context = || { format!( "failed to set pane {pane_id:?} geometry to {:?}", new_pane_geom ) }; let mut panes = self.panes.borrow_mut(); let pane = panes .iter_mut() .find(|(p_id, _p)| **p_id == pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)? .1; pane.set_geom(new_pane_geom); let mut desired_pane_positions = self.desired_pane_positions.borrow_mut(); desired_pane_positions.insert(pane_id, new_pane_geom); Ok(()) } pub fn resize(&mut self, space: Size) -> Result<()> { let err_context = || { format!( "failed to resize from {:?} to {:?}", self.display_area, space ) }; let mut panes = self.panes.borrow_mut(); let desired_pane_positions = self.desired_pane_positions.borrow(); // account for the difference between the viewport (including non-ui pane items which we // do not want to override) and the display_area, which is the area we can go over let display_size_row_difference = self.display_area.rows.saturating_sub(self.viewport.rows); let display_size_column_difference = self.display_area.cols.saturating_sub(self.viewport.cols); let mut new_viewport = self.viewport; new_viewport.cols = space.cols.saturating_sub(display_size_column_difference); new_viewport.rows = space.rows.saturating_sub(display_size_row_difference); for (pane_id, pane) in panes.iter_mut() { let mut new_pane_geom = pane.current_geom(); let desired_pane_geom = desired_pane_positions .get(pane_id) .with_context(|| { format!( "failed to acquire desired pane geometry for pane {:?}", pane_id ) }) .with_context(err_context)?; let desired_pane_geom_is_inside_viewport = pane_geom_is_inside_viewport(&new_viewport, desired_pane_geom); let pane_is_in_desired_position = new_pane_geom == *desired_pane_geom; if pane_is_in_desired_position && desired_pane_geom_is_inside_viewport { continue; } else if desired_pane_geom_is_inside_viewport { pane.set_geom(*desired_pane_geom); } else { let pane_right_side = new_pane_geom.x + new_pane_geom.cols.as_usize(); let pane_bottom_side = new_pane_geom.y + new_pane_geom.rows.as_usize(); let viewport_right_side = new_viewport.x + new_viewport.cols; let viewport_bottom_side = new_viewport.y + new_viewport.rows; let excess_width = pane_right_side.saturating_sub(viewport_right_side); let excess_height = pane_bottom_side.saturating_sub(viewport_bottom_side); let extra_width = viewport_right_side.saturating_sub(pane_right_side); let extra_height = viewport_bottom_side.saturating_sub(pane_bottom_side); // handle shrink width if excess_width > 0 && new_pane_geom.x.saturating_sub(excess_width) > new_viewport.x { new_pane_geom.x = new_pane_geom.x.saturating_sub(excess_width); } else if excess_width > 0 && new_pane_geom.cols.as_usize().saturating_sub(excess_width) > MIN_TERMINAL_WIDTH { new_pane_geom .cols .set_inner(new_pane_geom.cols.as_usize().saturating_sub(excess_width)); } else if excess_width > 0 { let reduce_x_by = new_pane_geom.x.saturating_sub(new_viewport.x); let reduced_width = new_pane_geom .cols .as_usize() .saturating_sub(excess_width.saturating_sub(reduce_x_by)); new_pane_geom.x = new_viewport.x; new_pane_geom .cols .set_inner(std::cmp::max(reduced_width, MIN_TERMINAL_WIDTH)); } // handle shrink height if excess_height > 0 && new_pane_geom.y.saturating_sub(excess_height) > new_viewport.y { new_pane_geom.y = new_pane_geom.y.saturating_sub(excess_height); } else if excess_height > 0 && new_pane_geom.rows.as_usize().saturating_sub(excess_height) > MIN_TERMINAL_HEIGHT { new_pane_geom .rows .set_inner(new_pane_geom.rows.as_usize().saturating_sub(excess_height)); } else if excess_height > 0 { let reduce_y_by = new_pane_geom.y.saturating_sub(new_viewport.y); let reduced_height = new_pane_geom .rows .as_usize() .saturating_sub(excess_height.saturating_sub(reduce_y_by)); new_pane_geom.y = new_viewport.y; new_pane_geom .rows .set_inner(std::cmp::max(reduced_height, MIN_TERMINAL_HEIGHT)); } // handle expand width if extra_width > 0 { let max_right_coords = new_viewport.x + new_viewport.cols; if new_pane_geom.x < desired_pane_geom.x { if desired_pane_geom.x + new_pane_geom.cols.as_usize() <= max_right_coords { new_pane_geom.x = desired_pane_geom.x } else if new_pane_geom.x + new_pane_geom.cols.as_usize() + extra_width < max_right_coords { new_pane_geom.x += extra_width; } else { new_pane_geom.x = max_right_coords.saturating_sub(new_pane_geom.cols.as_usize()); } } if new_pane_geom.cols.as_usize() < desired_pane_geom.cols.as_usize() { if new_pane_geom.x + desired_pane_geom.cols.as_usize() <= max_right_coords { new_pane_geom .cols .set_inner(desired_pane_geom.cols.as_usize()); } else if new_pane_geom.x + new_pane_geom.cols.as_usize() + extra_width < max_right_coords { new_pane_geom .cols .set_inner(new_pane_geom.cols.as_usize() + extra_width); } else { new_pane_geom.cols.set_inner( new_pane_geom.cols.as_usize() + (max_right_coords - (new_pane_geom.x + new_pane_geom.cols.as_usize())), ); } } } // handle expand height if extra_height > 0 { let max_bottom_coords = new_viewport.y + new_viewport.rows; if new_pane_geom.y < desired_pane_geom.y { if desired_pane_geom.y + new_pane_geom.rows.as_usize() <= max_bottom_coords { new_pane_geom.y = desired_pane_geom.y } else if new_pane_geom.y + new_pane_geom.rows.as_usize() + extra_height < max_bottom_coords { new_pane_geom.y += extra_height; } else { new_pane_geom.y = max_bottom_coords.saturating_sub(new_pane_geom.rows.as_usize()); } } if new_pane_geom.rows.as_usize() < desired_pane_geom.rows.as_usize() { if new_pane_geom.y + desired_pane_geom.rows.as_usize() <= max_bottom_coords { new_pane_geom .rows .set_inner(desired_pane_geom.rows.as_usize()); } else if new_pane_geom.y + new_pane_geom.rows.as_usize() + extra_height < max_bottom_coords { new_pane_geom .rows .set_inner(new_pane_geom.rows.as_usize() + extra_height); } else { new_pane_geom.rows.set_inner( new_pane_geom.rows.as_usize() + (max_bottom_coords - (new_pane_geom.y + new_pane_geom.rows.as_usize())), ); } } } pane.set_geom(new_pane_geom); } } Ok(()) } pub fn move_pane_left(&mut self, pane_id: &PaneId) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} left"); if let Some(move_by) = self .can_move_pane_left(pane_id, MOVE_INCREMENT_HORIZONTAL) .with_context(err_context)? { self.move_pane_position_left(pane_id, move_by) .with_context(err_context)?; } Ok(()) } pub fn move_pane_right(&mut self, pane_id: &PaneId) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} right"); if let Some(move_by) = self .can_move_pane_right(pane_id, MOVE_INCREMENT_HORIZONTAL) .with_context(err_context)? { self.move_pane_position_right(pane_id, move_by) .with_context(err_context)?; } Ok(()) } pub fn move_pane_down(&mut self, pane_id: &PaneId) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} down"); if let Some(move_by) = self .can_move_pane_down(pane_id, MOVE_INCREMENT_VERTICAL) .with_context(err_context)? { self.move_pane_position_down(pane_id, move_by) .with_context(err_context)?; } Ok(()) } pub fn move_pane_up(&mut self, pane_id: &PaneId) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} up"); if let Some(move_by) = self .can_move_pane_up(pane_id, MOVE_INCREMENT_VERTICAL) .with_context(err_context)? { self.move_pane_position_up(pane_id, move_by) .with_context(err_context)?; } Ok(()) } fn can_move_pane_left(&self, pane_id: &PaneId, move_by: usize) -> Result> { let err_context = || { format!( "failed to determine if pane {pane_id:?} can be moved left by {move_by} columns" ) }; let panes = self.panes.borrow(); let pane = panes .get(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)?; let space_until_left_screen_edge = pane.x().saturating_sub(self.viewport.x); Ok(if space_until_left_screen_edge >= move_by { Some(move_by) } else if space_until_left_screen_edge > 0 { Some(space_until_left_screen_edge) } else { None }) } fn can_move_pane_right(&self, pane_id: &PaneId, move_by: usize) -> Result> { let err_context = || { format!( "failed to determine if pane {pane_id:?} can be moved right by {move_by} columns" ) }; let panes = self.panes.borrow(); let pane = panes .get(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)?; let space_until_right_screen_edge = (self.viewport.x + self.viewport.cols).saturating_sub(pane.x() + pane.cols()); Ok(if space_until_right_screen_edge >= move_by { Some(move_by) } else if space_until_right_screen_edge > 0 { Some(space_until_right_screen_edge) } else { None }) } fn can_move_pane_up(&self, pane_id: &PaneId, move_by: usize) -> Result> { let err_context = || format!("failed to determine if pane {pane_id:?} can be moved up by {move_by} rows"); let panes = self.panes.borrow(); let pane = panes .get(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)?; let space_until_top_screen_edge = pane.y().saturating_sub(self.viewport.y); Ok(if space_until_top_screen_edge >= move_by { Some(move_by) } else if space_until_top_screen_edge > 0 { Some(space_until_top_screen_edge) } else { None }) } fn can_move_pane_down(&self, pane_id: &PaneId, move_by: usize) -> Result> { let err_context = || { format!("failed to determine if pane {pane_id:?} can be moved down by {move_by} rows") }; let panes = self.panes.borrow(); let pane = panes .get(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)?; let space_until_bottom_screen_edge = (self.viewport.y + self.viewport.rows).saturating_sub(pane.y() + pane.rows()); Ok(if space_until_bottom_screen_edge >= move_by { Some(move_by) } else if space_until_bottom_screen_edge > 0 { Some(space_until_bottom_screen_edge) } else { None }) } fn move_pane_position_left(&mut self, pane_id: &PaneId, move_by: usize) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} left by {move_by}"); let new_pane_geom = { let mut panes = self.panes.borrow_mut(); let pane = panes .get_mut(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)?; let mut current_geom = pane.position_and_size(); current_geom.x -= move_by; current_geom }; self.set_pane_geom(*pane_id, new_pane_geom) .with_context(err_context) } fn move_pane_position_right(&mut self, pane_id: &PaneId, move_by: usize) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} right by {move_by}"); let new_pane_geom = { let mut panes = self.panes.borrow_mut(); let pane = panes .get_mut(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)?; let mut current_geom = pane.position_and_size(); current_geom.x += move_by; current_geom }; self.set_pane_geom(*pane_id, new_pane_geom) .with_context(err_context) } fn move_pane_position_down(&mut self, pane_id: &PaneId, move_by: usize) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} down by {move_by}"); let new_pane_geom = { let mut panes = self.panes.borrow_mut(); let pane = panes .get_mut(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)?; let mut current_geom = pane.position_and_size(); current_geom.y += move_by; current_geom }; self.set_pane_geom(*pane_id, new_pane_geom) .with_context(err_context) } fn move_pane_position_up(&mut self, pane_id: &PaneId, move_by: usize) -> Result<()> { let err_context = || format!("failed to move pane {pane_id:?} up by {move_by}"); let new_pane_geom = { let mut panes = self.panes.borrow_mut(); let pane = panes .get_mut(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)?; let mut current_geom = pane.position_and_size(); current_geom.y -= move_by; current_geom }; self.set_pane_geom(*pane_id, new_pane_geom) .with_context(err_context) } pub fn change_pane_size( &mut self, pane_id: &PaneId, strategy: &ResizeStrategy, change_by: (usize, usize), // (x, y) ) -> Result<()> { let err_context = || format!("failed to {strategy} for pane {pane_id:?}"); let mut geometry = self .panes .borrow() .get(pane_id) .with_context(|| no_pane_id(&pane_id)) .with_context(err_context)? .position_and_size(); let change_by = if strategy.direction.is_none() { (change_by.0 / 2, change_by.1 / 2) } else { change_by }; // Move left border if strategy.move_left_border_left() || strategy.move_all_borders_out() { let increment = std::cmp::min(geometry.x.saturating_sub(self.viewport.x), change_by.0); // Invert if on boundary already if increment == 0 && strategy.direction.is_some() { return self.change_pane_size(pane_id, &strategy.invert(), change_by); } geometry.x -= increment; geometry .cols .set_inner(geometry.cols.as_usize() + increment); } else if strategy.move_left_border_right() || strategy.move_all_borders_in() { let increment = std::cmp::min( geometry.cols.as_usize().saturating_sub(MIN_TERMINAL_WIDTH), change_by.0, ); geometry.x += increment; geometry .cols .set_inner(geometry.cols.as_usize() - increment); }; // Move right border if strategy.move_right_border_right() || strategy.move_all_borders_out() { let increment = std::cmp::min( (self.viewport.x + self.viewport.cols) .saturating_sub(geometry.x + geometry.cols.as_usize()), change_by.0, ); // Invert if on boundary already if increment == 0 && strategy.direction.is_some() { return self.change_pane_size(pane_id, &strategy.invert(), change_by); } geometry .cols .set_inner(geometry.cols.as_usize() + increment); } else if strategy.move_right_border_left() || strategy.move_all_borders_in() { let increment = std::cmp::min( geometry.cols.as_usize().saturating_sub(MIN_TERMINAL_WIDTH), change_by.0, ); geometry .cols .set_inner(geometry.cols.as_usize() - increment); }; // Move upper border if strategy.move_upper_border_up() || strategy.move_all_borders_out() { let increment = std::cmp::min(geometry.y.saturating_sub(self.viewport.y), change_by.1); // Invert if on boundary already if increment == 0 && strategy.direction.is_some() { return self.change_pane_size(pane_id, &strategy.invert(), change_by); } geometry.y -= increment; geometry .rows .set_inner(geometry.rows.as_usize() + increment); } else if strategy.move_upper_border_down() || strategy.move_all_borders_in() { let increment = std::cmp::min( geometry.rows.as_usize().saturating_sub(MIN_TERMINAL_HEIGHT), change_by.1, ); geometry.y += increment; geometry .rows .set_inner(geometry.rows.as_usize() - increment); } // Move lower border if strategy.move_lower_border_down() || strategy.move_all_borders_out() { let increment = std::cmp::min( (self.viewport.y + self.viewport.rows) .saturating_sub(geometry.y + geometry.rows.as_usize()), change_by.1, ); // Invert if on boundary already if increment == 0 && strategy.direction.is_some() { return self.change_pane_size(pane_id, &strategy.invert(), change_by); } geometry .rows .set_inner(geometry.rows.as_usize() + increment); } else if strategy.move_lower_border_up() || strategy.move_all_borders_in() { let increment = std::cmp::min( geometry.rows.as_usize().saturating_sub(MIN_TERMINAL_HEIGHT), change_by.1, ); geometry .rows .set_inner(geometry.rows.as_usize() - increment); } self.set_pane_geom(*pane_id, geometry) .with_context(err_context) } pub fn next_selectable_pane_id_to_the_left(&self, current_pane_id: &PaneId) -> Option { let panes = self.panes.borrow(); let current_pane = panes.get(current_pane_id)?; let panes: Vec<(PaneId, &&mut Box)> = panes .iter() .filter(|(_, p)| p.selectable()) .map(|(p_id, p)| (*p_id, p)) .collect(); let next_index = panes .iter() .enumerate() .filter(|(_, (_, c))| { c.is_left_of(Box::as_ref(current_pane)) && c.horizontally_overlaps_with(Box::as_ref(current_pane)) }) .max_by(|(_, (_, a)), (_, (_, b))| { let x_comparison = a.x().cmp(&b.x()); match x_comparison { Ordering::Equal => a.y().cmp(&b.y()), _ => x_comparison, } }) .map(|(_, (pid, _))| pid) .copied(); next_index } pub fn pane_id_on_edge(&self, direction: Direction) -> Option { let panes = self.panes.borrow(); let panes: Vec<(PaneId, &&mut Box)> = panes .iter() .filter(|(_, p)| p.selectable()) .map(|(p_id, p)| (*p_id, p)) .collect(); let next_index = panes .iter() .enumerate() .max_by(|(_, (_, a)), (_, (_, b))| match direction { Direction::Left => { let x_comparison = a.x().cmp(&b.x()); match x_comparison { Ordering::Equal => a.y().cmp(&b.y()), _ => x_comparison, } }, Direction::Right => { let x_comparison = b.x().cmp(&a.x()); match x_comparison { Ordering::Equal => a.y().cmp(&b.y()), _ => x_comparison, } }, Direction::Up => { let y_comparison = a.y().cmp(&b.y()); match y_comparison { Ordering::Equal => a.x().cmp(&b.x()), _ => y_comparison, } }, Direction::Down => { let y_comparison = b.y().cmp(&a.y()); match y_comparison { Ordering::Equal => b.x().cmp(&a.x()), _ => y_comparison, } }, }) .map(|(_, (pid, _))| pid) .copied(); next_index } pub fn next_selectable_pane_id_below(&self, current_pane_id: &PaneId) -> Option { let panes = self.panes.borrow(); let current_pane = panes.get(current_pane_id)?; let panes: Vec<(PaneId, &&mut Box)> = panes .iter() .filter(|(_, p)| p.selectable()) .map(|(p_id, p)| (*p_id, p)) .collect(); let next_index = panes .iter() .enumerate() .filter(|(_, (_, c))| { c.is_below(Box::as_ref(current_pane)) && c.vertically_overlaps_with(Box::as_ref(current_pane)) }) .min_by(|(_, (_, a)), (_, (_, b))| { let y_comparison = a.y().cmp(&b.y()); match y_comparison { Ordering::Equal => b.x().cmp(&a.x()), _ => y_comparison, } }) .map(|(_, (pid, _))| pid) .copied(); next_index } pub fn next_selectable_pane_id_above(&self, current_pane_id: &PaneId) -> Option { let panes = self.panes.borrow(); let current_pane = panes.get(current_pane_id)?; let panes: Vec<(PaneId, &&mut Box)> = panes .iter() .filter(|(_, p)| p.selectable()) .map(|(p_id, p)| (*p_id, p)) .collect(); let next_index = panes .iter() .enumerate() .filter(|(_, (_, c))| { c.is_above(Box::as_ref(current_pane)) && c.vertically_overlaps_with(Box::as_ref(current_pane)) }) .max_by(|(_, (_, a)), (_, (_, b))| { let y_comparison = a.y().cmp(&b.y()); match y_comparison { Ordering::Equal => b.x().cmp(&a.x()), _ => y_comparison, } }) .map(|(_, (pid, _))| pid) .copied(); next_index } pub fn next_selectable_pane_id_to_the_right(&self, current_pane_id: &PaneId) -> Option { let panes = self.panes.borrow(); let current_pane = panes.get(current_pane_id)?; let panes: Vec<(PaneId, &&mut Box)> = panes .iter() .filter(|(_, p)| p.selectable()) .map(|(p_id, p)| (*p_id, p)) .collect(); let next_index = panes .iter() .enumerate() .filter(|(_, (_, c))| { c.is_right_of(Box::as_ref(current_pane)) && c.horizontally_overlaps_with(Box::as_ref(current_pane)) }) .min_by(|(_, (_, a)), (_, (_, b))| { let x_comparison = a.x().cmp(&b.x()); match x_comparison { Ordering::Equal => a.y().cmp(&b.y()), _ => x_comparison, } }) .map(|(_, (pid, _))| pid) .copied(); next_index } pub fn next_selectable_pane_id(&self, current_pane_id: &PaneId) -> Option { let panes = self.panes.borrow(); let mut panes: Vec<(PaneId, &&mut Box)> = panes .iter() .filter(|(_, p)| p.selectable()) .map(|(p_id, p)| (*p_id, p)) .collect(); panes.sort_by(|(_a_id, a_pane), (_b_id, b_pane)| { if a_pane.y() == b_pane.y() { a_pane.x().cmp(&b_pane.x()) } else { a_pane.y().cmp(&b_pane.y()) } }); let active_pane_position = panes .iter() .position(|(id, _)| id == current_pane_id) .unwrap(); let next_active_pane_id = panes .get(active_pane_position + 1) .or_else(|| panes.get(0)) .map(|p| p.0) .unwrap(); Some(next_active_pane_id) } pub fn previous_selectable_pane_id(&self, current_pane_id: &PaneId) -> Option { let panes = self.panes.borrow(); let mut panes: Vec<(PaneId, &&mut Box)> = panes .iter() .filter(|(_, p)| p.selectable()) .map(|(p_id, p)| (*p_id, p)) .collect(); panes.sort_by(|(_a_id, a_pane), (_b_id, b_pane)| { if a_pane.y() == b_pane.y() { a_pane.x().cmp(&b_pane.x()) } else { a_pane.y().cmp(&b_pane.y()) } }); let active_pane_position = panes.iter().position(|(id, _)| id == current_pane_id)?; let last_pane = panes.last()?; let previous_active_pane_id = if active_pane_position == 0 { last_pane.0 } else { panes.get(active_pane_position - 1)?.0 }; Some(previous_active_pane_id) } pub fn find_room_for_new_pane(&self) -> Option { let panes = self.panes.borrow(); let pane_geoms: Vec = panes.values().map(|p| p.position_and_size()).collect(); macro_rules! find_unoccupied_offset { ($get_geom_with_offset:expr, $viewport:expr, $other_geoms:expr) => { let mut offset = 0; loop { let geom_with_current_offset = $get_geom_with_offset(offset); if pane_geom_is_big_enough(&geom_with_current_offset) && pane_geom_is_unoccupied_and_inside_viewport( $viewport, &geom_with_current_offset, $other_geoms, ) { return Some(geom_with_current_offset); } else if !pane_geom_is_inside_viewport($viewport, &geom_with_current_offset) { break; } else if offset > MAX_PANES { // this is mostly to kill the loop no matter what break; } else { offset += 2; } } }; } find_unoccupied_offset!( |offset| half_size_middle_geom(&self.viewport, offset), &self.viewport, &pane_geoms ); find_unoccupied_offset!( |offset| half_size_top_left_geom(&self.viewport, offset), &self.viewport, &pane_geoms ); find_unoccupied_offset!( |offset| half_size_top_right_geom(&self.viewport, offset), &self.viewport, &pane_geoms ); find_unoccupied_offset!( |offset| half_size_bottom_left_geom(&self.viewport, offset), &self.viewport, &pane_geoms ); find_unoccupied_offset!( |offset| half_size_bottom_right_geom(&self.viewport, offset), &self.viewport, &pane_geoms ); return None; } } fn half_size_middle_geom(space: &Viewport, offset: usize) -> PaneGeom { let mut geom = PaneGeom { x: space.x + (space.cols as f64 / 4.0).round() as usize + offset, y: space.y + (space.rows as f64 / 4.0).round() as usize + offset, cols: Dimension::fixed(space.cols / 2), rows: Dimension::fixed(space.rows / 2), is_stacked: false, }; geom.cols.set_inner(space.cols / 2); geom.rows.set_inner(space.rows / 2); geom } fn half_size_top_left_geom(space: &Viewport, offset: usize) -> PaneGeom { let mut geom = PaneGeom { x: space.x + 2 + offset, y: space.y + 2 + offset, cols: Dimension::fixed(space.cols / 3), rows: Dimension::fixed(space.rows / 3), is_stacked: false, }; geom.cols.set_inner(space.cols / 3); geom.rows.set_inner(space.rows / 3); geom } fn half_size_top_right_geom(space: &Viewport, offset: usize) -> PaneGeom { let mut geom = PaneGeom { x: ((space.x + space.cols) - (space.cols / 3) - 2).saturating_sub(offset), y: space.y + 2 + offset, cols: Dimension::fixed(space.cols / 3), rows: Dimension::fixed(space.rows / 3), is_stacked: false, }; geom.cols.set_inner(space.cols / 3); geom.rows.set_inner(space.rows / 3); geom } fn half_size_bottom_left_geom(space: &Viewport, offset: usize) -> PaneGeom { let mut geom = PaneGeom { x: space.x + 2 + offset, y: ((space.y + space.rows) - (space.rows / 3) - 2).saturating_sub(offset), cols: Dimension::fixed(space.cols / 3), rows: Dimension::fixed(space.rows / 3), is_stacked: false, }; geom.cols.set_inner(space.cols / 3); geom.rows.set_inner(space.rows / 3); geom } fn half_size_bottom_right_geom(space: &Viewport, offset: usize) -> PaneGeom { let mut geom = PaneGeom { x: ((space.x + space.cols) - (space.cols / 3) - 2).saturating_sub(offset), y: ((space.y + space.rows) - (space.rows / 3) - 2).saturating_sub(offset), cols: Dimension::fixed(space.cols / 3), rows: Dimension::fixed(space.rows / 3), is_stacked: false, }; geom.cols.set_inner(space.cols / 3); geom.rows.set_inner(space.rows / 3); geom } fn pane_geom_is_inside_viewport(viewport: &Viewport, geom: &PaneGeom) -> bool { geom.y >= viewport.y && geom.y + geom.rows.as_usize() <= viewport.y + viewport.rows && geom.x >= viewport.x && geom.x + geom.cols.as_usize() <= viewport.x + viewport.cols } fn pane_geom_is_big_enough(geom: &PaneGeom) -> bool { geom.rows.as_usize() >= MIN_TERMINAL_HEIGHT && geom.cols.as_usize() >= MIN_TERMINAL_WIDTH } fn pane_geom_is_unoccupied_and_inside_viewport( viewport: &Viewport, geom: &PaneGeom, existing_geoms: &[PaneGeom], ) -> bool { pane_geom_is_inside_viewport(viewport, geom) && !existing_geoms.iter().any(|p| p == geom) }