use anyhow::{anyhow, Result}; use element::pane_axis; use gpui::{ div, point, size, AnyView, AnyWeakView, Axis, Bounds, Element as _, IntoElement, ParentElement as _, Pixels, Point, StyleRefinement, Styled as _, View, ViewContext, }; use parking_lot::Mutex; use serde::Deserialize; use std::sync::Arc; use ui::StyledExt as _; use super::{pane::Pane, workspace::Workspace}; pub const HANDLE_HITBOX_SIZE: f32 = 4.0; const HORIZONTAL_MIN_SIZE: f32 = 80.; const VERTICAL_MIN_SIZE: f32 = 100.; /// One or many panes, arranged in a horizontal or vertical axis due to a split. /// Panes have all their tabs and capabilities preserved, and can be split again or resized. /// Single-pane group is a regular pane. #[derive(Clone)] pub struct PaneGroup { pub(crate) root: Member, } impl PaneGroup { pub fn new(pane: View) -> Self { Self { root: Member::Pane(pane), } } pub fn split( &mut self, old_pane: &View, new_pane: &View, direction: SplitDirection, ) -> Result<()> { match &mut self.root { Member::Pane(pane) => { if pane == old_pane { self.root = Member::new_axis(old_pane.clone(), new_pane.clone(), direction); Ok(()) } else { Err(anyhow!("Pane not found")) } } Member::Axis(axis) => axis.split(old_pane, new_pane, direction), } } pub fn bounding_box_for_pane(&self, pane: &View) -> Option> { match &self.root { Member::Pane(_) => None, Member::Axis(axis) => axis.bounding_box_for_pane(pane), } } pub fn pane_at_pixel_position(&self, coordinate: Point) -> Option<&View> { match &self.root { Member::Pane(pane) => Some(pane), Member::Axis(axis) => axis.pane_at_pixel_position(coordinate), } } /// Returns: /// - Ok(true) if it found and removed a pane /// - Ok(false) if it found but did not remove the pane /// - Err(_) if it did not find the pane pub fn remove(&mut self, pane: &View) -> Result { match &mut self.root { Member::Pane(_) => Ok(false), Member::Axis(axis) => { if let Some(last_pane) = axis.remove(pane)? { self.root = last_pane; } Ok(true) } } } pub fn swap(&mut self, from: &View, to: &View) { match &mut self.root { Member::Pane(_) => {} Member::Axis(axis) => axis.swap(from, to), }; } #[allow(clippy::too_many_arguments)] pub(crate) fn render( &self, active_pane: &View, zoomed: Option<&AnyWeakView>, cx: &mut ViewContext, ) -> impl IntoElement { self.root.render(0, active_pane, zoomed, cx) } pub(crate) fn panes(&self) -> Vec<&View> { let mut panes = Vec::new(); self.root.collect_panes(&mut panes); panes } #[allow(unused)] pub(crate) fn first_pane(&self) -> View { self.root.first_pane() } } #[derive(Clone)] pub(crate) enum Member { Axis(PaneAxis), Pane(View), } impl Member { fn new_axis(old_pane: View, new_pane: View, direction: SplitDirection) -> Self { use Axis::*; use SplitDirection::*; let axis = match direction { Up | Down => Vertical, Left | Right => Horizontal, }; let members = match direction { Up | Left => vec![Member::Pane(new_pane), Member::Pane(old_pane)], Down | Right => vec![Member::Pane(old_pane), Member::Pane(new_pane)], }; Member::Axis(PaneAxis::new(axis, members)) } fn contains(&self, needle: &View) -> bool { match self { Member::Axis(axis) => axis.members.iter().any(|member| member.contains(needle)), Member::Pane(pane) => pane == needle, } } fn first_pane(&self) -> View { match self { Member::Axis(axis) => axis.members[0].first_pane(), Member::Pane(pane) => pane.clone(), } } #[allow(clippy::too_many_arguments)] pub fn render( &self, basis: usize, active_pane: &View, zoomed: Option<&AnyWeakView>, cx: &mut ViewContext, ) -> impl IntoElement { match self { Member::Pane(pane) => { if zoomed == Some(&pane.downgrade().into()) { return div().into_any(); } div() .relative() .flex_1() .size_full() .child( AnyView::from(pane.clone()) .cached(StyleRefinement::default().v_flex().size_full()), ) .into_any() } Member::Axis(axis) => axis.render(basis + 1, active_pane, zoomed, cx).into_any(), } } fn collect_panes<'a>(&'a self, panes: &mut Vec<&'a View>) { match self { Member::Axis(axis) => { for member in &axis.members { member.collect_panes(panes); } } Member::Pane(pane) => panes.push(pane), } } } #[derive(Clone)] pub(crate) struct PaneAxis { pub axis: Axis, pub members: Vec, pub flexes: Arc>>, pub bounding_boxes: Arc>>>>, } impl PaneAxis { pub fn new(axis: Axis, members: Vec) -> Self { let flexes = Arc::new(Mutex::new(vec![1.; members.len()])); let bounding_boxes = Arc::new(Mutex::new(vec![None; members.len()])); Self { axis, members, flexes, bounding_boxes, } } #[allow(unused)] pub fn load(axis: Axis, members: Vec, flexes: Option>) -> Self { let flexes = flexes.unwrap_or_else(|| vec![1.; members.len()]); debug_assert!(members.len() == flexes.len()); let flexes = Arc::new(Mutex::new(flexes)); let bounding_boxes = Arc::new(Mutex::new(vec![None; members.len()])); Self { axis, members, flexes, bounding_boxes, } } fn split( &mut self, old_pane: &View, new_pane: &View, direction: SplitDirection, ) -> Result<()> { for (mut idx, member) in self.members.iter_mut().enumerate() { match member { Member::Axis(axis) => { if axis.split(old_pane, new_pane, direction).is_ok() { return Ok(()); } } Member::Pane(pane) => { if pane == old_pane { if direction.axis() == self.axis { if direction.increasing() { idx += 1; } self.members.insert(idx, Member::Pane(new_pane.clone())); *self.flexes.lock() = vec![1.; self.members.len()]; } else { *member = Member::new_axis(old_pane.clone(), new_pane.clone(), direction); } return Ok(()); } } } } Err(anyhow!("Pane not found")) } fn remove(&mut self, pane_to_remove: &View) -> Result> { let mut found_pane = false; let mut remove_member = None; for (idx, member) in self.members.iter_mut().enumerate() { match member { Member::Axis(axis) => { if let Ok(last_pane) = axis.remove(pane_to_remove) { if let Some(last_pane) = last_pane { *member = last_pane; } found_pane = true; break; } } Member::Pane(pane) => { if pane == pane_to_remove { found_pane = true; remove_member = Some(idx); break; } } } } if found_pane { if let Some(idx) = remove_member { self.members.remove(idx); *self.flexes.lock() = vec![1.; self.members.len()]; } if self.members.len() == 1 { let result = self.members.pop(); *self.flexes.lock() = vec![1.; self.members.len()]; Ok(result) } else { Ok(None) } } else { Err(anyhow!("Pane not found")) } } fn swap(&mut self, from: &View, to: &View) { for member in self.members.iter_mut() { match member { Member::Axis(axis) => axis.swap(from, to), Member::Pane(pane) => { if pane == from { *member = Member::Pane(to.clone()); } else if pane == to { *member = Member::Pane(from.clone()) } } } } } fn bounding_box_for_pane(&self, pane: &View) -> Option> { debug_assert!(self.members.len() == self.bounding_boxes.lock().len()); for (idx, member) in self.members.iter().enumerate() { match member { Member::Pane(found) => { if pane == found { return self.bounding_boxes.lock()[idx]; } } Member::Axis(axis) => { if let Some(rect) = axis.bounding_box_for_pane(pane) { return Some(rect); } } } } None } fn pane_at_pixel_position(&self, coordinate: Point) -> Option<&View> { debug_assert!(self.members.len() == self.bounding_boxes.lock().len()); let bounding_boxes = self.bounding_boxes.lock(); for (idx, member) in self.members.iter().enumerate() { if let Some(coordinates) = bounding_boxes[idx] { if coordinates.contains(&coordinate) { return match member { Member::Pane(found) => Some(found), Member::Axis(axis) => axis.pane_at_pixel_position(coordinate), }; } } } None } #[allow(clippy::too_many_arguments)] fn render( &self, basis: usize, active_pane: &View, zoomed: Option<&AnyWeakView>, cx: &mut ViewContext, ) -> gpui::AnyElement { debug_assert!(self.members.len() == self.flexes.lock().len()); let mut active_pane_ix = None; pane_axis( self.axis, basis, self.flexes.clone(), self.bounding_boxes.clone(), cx.view().downgrade(), ) .children(self.members.iter().enumerate().map(|(ix, member)| { if member.contains(active_pane) { active_pane_ix = Some(ix); } member .render((basis + ix) * 10, active_pane, zoomed, cx) .into_any_element() })) .with_active_pane(active_pane_ix) .into_any_element() } } #[derive(Clone, Copy, Debug, Deserialize, PartialEq)] pub enum SplitDirection { Up, Down, Left, Right, } impl std::fmt::Display for SplitDirection { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { match self { SplitDirection::Up => write!(f, "up"), SplitDirection::Down => write!(f, "down"), SplitDirection::Left => write!(f, "left"), SplitDirection::Right => write!(f, "right"), } } } impl SplitDirection { pub fn all() -> [Self; 4] { [Self::Up, Self::Down, Self::Left, Self::Right] } pub fn edge(&self, rect: Bounds) -> Pixels { match self { Self::Up => rect.origin.y, Self::Down => rect.lower_left().y, Self::Left => rect.lower_left().x, Self::Right => rect.lower_right().x, } } pub fn along_edge(&self, bounds: Bounds, length: Pixels) -> Bounds { match self { Self::Up => Bounds { origin: bounds.origin, size: size(bounds.size.width, length), }, Self::Down => Bounds { origin: point(bounds.lower_left().x, bounds.lower_left().y - length), size: size(bounds.size.width, length), }, Self::Left => Bounds { origin: bounds.origin, size: size(length, bounds.size.height), }, Self::Right => Bounds { origin: point(bounds.lower_right().x - length, bounds.lower_left().y), size: size(length, bounds.size.height), }, } } pub fn axis(&self) -> Axis { match self { Self::Up | Self::Down => Axis::Vertical, Self::Left | Self::Right => Axis::Horizontal, } } pub fn increasing(&self) -> bool { match self { Self::Left | Self::Up => false, Self::Down | Self::Right => true, } } } mod element { use std::mem; use std::{cell::RefCell, iter, rc::Rc, sync::Arc}; use crate::util::ResultExt; use gpui::{ px, relative, Along, AnyElement, Axis, Bounds, Element, ElementId, GlobalElementId, IntoElement, MouseDownEvent, MouseMoveEvent, MouseUpEvent, ParentElement, Pixels, Point, Size, Style, WeakView, WindowContext, }; use gpui::{CursorStyle, Hitbox}; use parking_lot::Mutex; use smallvec::SmallVec; use ui::theme::ActiveTheme; use crate::Workspace; use super::{HANDLE_HITBOX_SIZE, HORIZONTAL_MIN_SIZE, VERTICAL_MIN_SIZE}; const DIVIDER_SIZE: f32 = 1.0; pub(super) fn pane_axis( axis: Axis, basis: usize, flexes: Arc>>, bounding_boxes: Arc>>>>, workspace: WeakView, ) -> PaneAxisElement { PaneAxisElement { axis, basis, flexes, bounding_boxes, children: SmallVec::new(), active_pane_ix: None, workspace, } } pub struct PaneAxisElement { axis: Axis, basis: usize, flexes: Arc>>, bounding_boxes: Arc>>>>, children: SmallVec<[AnyElement; 2]>, active_pane_ix: Option, workspace: WeakView, } pub struct PaneAxisLayout { dragged_handle: Rc>>, children: Vec, } struct PaneAxisChildLayout { bounds: Bounds, element: AnyElement, handle: Option, } struct PaneAxisHandleLayout { hitbox: Hitbox, divider_bounds: Bounds, } impl PaneAxisElement { pub fn with_active_pane(mut self, active_pane_ix: Option) -> Self { self.active_pane_ix = active_pane_ix; self } #[allow(clippy::too_many_arguments)] fn compute_resize( flexes: &Arc>>, e: &MouseMoveEvent, ix: usize, axis: Axis, child_start: Point, container_size: Size, workspace: WeakView, cx: &mut WindowContext, ) { let min_size = match axis { Axis::Horizontal => px(HORIZONTAL_MIN_SIZE), Axis::Vertical => px(VERTICAL_MIN_SIZE), }; let mut flexes = flexes.lock(); debug_assert!(flex_values_in_bounds(flexes.as_slice())); let size = move |ix, flexes: &[f32]| { container_size.along(axis) * (flexes[ix] / flexes.len() as f32) }; // Don't allow resizing to less than the minimum size, if elements are already too small if min_size - px(1.) > size(ix, flexes.as_slice()) { return; } let mut proposed_current_pixel_change = (e.position - child_start).along(axis) - size(ix, flexes.as_slice()); let flex_changes = |pixel_dx, target_ix, next: isize, flexes: &[f32]| { let flex_change = pixel_dx / container_size.along(axis); let current_target_flex = flexes[target_ix] + flex_change; let next_target_flex = flexes[(target_ix as isize + next) as usize] - flex_change; (current_target_flex, next_target_flex) }; let mut successors = iter::from_fn({ let forward = proposed_current_pixel_change > px(0.); let mut ix_offset = 0; let len = flexes.len(); move || { let result = if forward { (ix + 1 + ix_offset < len).then(|| ix + ix_offset) } else { (ix as isize - ix_offset as isize >= 0).then(|| ix - ix_offset) }; ix_offset += 1; result } }); while proposed_current_pixel_change.abs() > px(0.) { let Some(current_ix) = successors.next() else { break; }; let next_target_size = Pixels::max( size(current_ix + 1, flexes.as_slice()) - proposed_current_pixel_change, min_size, ); let current_target_size = Pixels::max( size(current_ix, flexes.as_slice()) + size(current_ix + 1, flexes.as_slice()) - next_target_size, min_size, ); let current_pixel_change = current_target_size - size(current_ix, flexes.as_slice()); let (current_target_flex, next_target_flex) = flex_changes(current_pixel_change, current_ix, 1, flexes.as_slice()); flexes[current_ix] = current_target_flex; flexes[current_ix + 1] = next_target_flex; proposed_current_pixel_change -= current_pixel_change; } workspace .update(cx, |this, cx| this.serialize_workspace(cx)) .log_err(); cx.stop_propagation(); cx.refresh(); } #[allow(clippy::too_many_arguments)] fn layout_handle( axis: Axis, pane_bounds: Bounds, cx: &mut WindowContext, ) -> PaneAxisHandleLayout { let handle_bounds = Bounds { origin: pane_bounds.origin.apply_along(axis, |origin| { origin + pane_bounds.size.along(axis) - px(HANDLE_HITBOX_SIZE / 2.) }), size: pane_bounds .size .apply_along(axis, |_| px(HANDLE_HITBOX_SIZE)), }; let divider_bounds = Bounds { origin: pane_bounds .origin .apply_along(axis, |origin| origin + pane_bounds.size.along(axis)), size: pane_bounds.size.apply_along(axis, |_| px(DIVIDER_SIZE)), }; PaneAxisHandleLayout { hitbox: cx.insert_hitbox(handle_bounds, true), divider_bounds, } } } impl IntoElement for PaneAxisElement { type Element = Self; fn into_element(self) -> Self::Element { self } } impl Element for PaneAxisElement { type RequestLayoutState = (); type PrepaintState = PaneAxisLayout; fn id(&self) -> Option { Some(self.basis.into()) } fn request_layout( &mut self, _global_id: Option<&GlobalElementId>, cx: &mut WindowContext, ) -> (gpui::LayoutId, Self::RequestLayoutState) { let mut style = Style::default(); style.flex_grow = 1.; style.flex_shrink = 1.; style.flex_basis = relative(0.).into(); style.size.width = relative(1.).into(); style.size.height = relative(1.).into(); (cx.request_layout(style, None), ()) } fn prepaint( &mut self, global_id: Option<&GlobalElementId>, bounds: Bounds, _state: &mut Self::RequestLayoutState, cx: &mut WindowContext, ) -> PaneAxisLayout { let dragged_handle = cx.with_element_state::>>, _>( global_id.unwrap(), |state, _cx| { let state = state.unwrap_or_else(|| Rc::new(RefCell::new(None))); (state.clone(), state) }, ); let flexes = self.flexes.lock().clone(); let len = self.children.len(); debug_assert!(flexes.len() == len); debug_assert!(flex_values_in_bounds(flexes.as_slice())); let magnification_value = 1.0; let active_pane_magnification = if magnification_value == 1. { None } else { Some(magnification_value) }; let total_flex = if let Some(flex) = active_pane_magnification { self.children.len() as f32 - 1. + flex } else { len as f32 }; let mut origin = bounds.origin; let space_per_flex = bounds.size.along(self.axis) / total_flex; let mut bounding_boxes = self.bounding_boxes.lock(); bounding_boxes.clear(); let mut layout = PaneAxisLayout { dragged_handle: dragged_handle.clone(), children: Vec::new(), }; for (ix, mut child) in mem::take(&mut self.children).into_iter().enumerate() { let child_flex = active_pane_magnification .map(|magnification| { if self.active_pane_ix == Some(ix) { magnification } else { 1. } }) .unwrap_or_else(|| flexes[ix]); let child_size = bounds .size .apply_along(self.axis, |_| space_per_flex * child_flex) .map(|d| d.round()); let child_bounds = Bounds { origin, size: child_size, }; bounding_boxes.push(Some(child_bounds)); child.layout_as_root(child_size.into(), cx); child.prepaint_at(origin, cx); origin = origin.apply_along(self.axis, |val| val + child_size.along(self.axis)); layout.children.push(PaneAxisChildLayout { bounds: child_bounds, element: child, handle: None, }) } for (ix, child_layout) in layout.children.iter_mut().enumerate() { if active_pane_magnification.is_none() { if ix < len - 1 { child_layout.handle = Some(Self::layout_handle(self.axis, child_layout.bounds, cx)); } } } layout } fn paint( &mut self, _id: Option<&GlobalElementId>, bounds: gpui::Bounds, _: &mut Self::RequestLayoutState, layout: &mut Self::PrepaintState, cx: &mut WindowContext, ) { for child in &mut layout.children { child.element.paint(cx); } for (ix, child) in &mut layout.children.iter_mut().enumerate() { if let Some(handle) = child.handle.as_mut() { let cursor_style = match self.axis { Axis::Vertical => CursorStyle::ResizeRow, Axis::Horizontal => CursorStyle::ResizeColumn, }; cx.set_cursor_style(cursor_style, &handle.hitbox); // Pane Group border cx.paint_quad(gpui::fill(handle.divider_bounds, cx.theme().border)); cx.on_mouse_event({ let dragged_handle = layout.dragged_handle.clone(); let flexes = self.flexes.clone(); let workspace = self.workspace.clone(); let handle_hitbox = handle.hitbox.clone(); move |e: &MouseDownEvent, phase, cx| { if phase.bubble() && handle_hitbox.is_hovered(cx) { dragged_handle.replace(Some(ix)); if e.click_count >= 2 { let mut borrow = flexes.lock(); *borrow = vec![1.; borrow.len()]; workspace .update(cx, |this, cx| this.serialize_workspace(cx)) .log_err(); cx.refresh(); } cx.stop_propagation(); } } }); cx.on_mouse_event({ let workspace = self.workspace.clone(); let dragged_handle = layout.dragged_handle.clone(); let flexes = self.flexes.clone(); let child_bounds = child.bounds; let axis = self.axis; move |e: &MouseMoveEvent, phase, cx| { let dragged_handle = dragged_handle.borrow(); if phase.bubble() { if *dragged_handle == Some(ix) { Self::compute_resize( &flexes, e, ix, axis, child_bounds.origin, bounds.size, workspace.clone(), cx, ) } } } }); } } cx.on_mouse_event({ let dragged_handle = layout.dragged_handle.clone(); move |_: &MouseUpEvent, phase, _cx| { if phase.bubble() { dragged_handle.replace(None); } } }); } } impl ParentElement for PaneAxisElement { fn extend(&mut self, elements: impl IntoIterator) { self.children.extend(elements) } } fn flex_values_in_bounds(flexes: &[f32]) -> bool { (flexes.iter().copied().sum::() - flexes.len() as f32).abs() < 0.001 } }