use anyhow::{anyhow, Error}; use bitflags::bitflags; use serde::de::{self, Deserialize, Deserializer}; use std::{ cmp::{max, min}, str::FromStr, }; #[derive(Debug, Clone, Copy, PartialEq)] /// UNSTABLE pub enum CursorKind { /// █ Block, /// | Bar, /// _ Underline, /// Hidden cursor, can set cursor position with this to let IME have correct cursor position. Hidden, } impl Default for CursorKind { fn default() -> Self { Self::Block } } impl FromStr for CursorKind { type Err = Error; fn from_str(s: &str) -> Result { match s { "bar" => Ok(Self::Bar), "block" => Ok(Self::Block), "underline" => Ok(Self::Underline), _ => Err(anyhow!("Invalid cursor '{}'", s)), } } } // toml deserializer doesn't seem to recognize string as enum impl<'de> Deserialize<'de> for CursorKind { fn deserialize(deserializer: D) -> Result where D: Deserializer<'de>, { let s = String::deserialize(deserializer)?; s.parse().map_err(de::Error::custom) } } #[derive(Debug, Clone, PartialEq, Eq, Hash)] pub struct Margin { pub vertical: u16, pub horizontal: u16, } /// A simple rectangle used in the computation of the layout and to give widgets an hint about the /// area they are supposed to render to. (x, y) = (0, 0) is at the top left corner of the screen. #[derive(Debug, Clone, Copy, Hash, PartialEq, Eq)] pub struct Rect { pub x: u16, pub y: u16, pub width: u16, pub height: u16, } impl Default for Rect { fn default() -> Rect { Rect { x: 0, y: 0, width: 0, height: 0, } } } impl Rect { /// Creates a new rect, with width and height limited to keep the area under max u16. /// If clipped, aspect ratio will be preserved. pub fn new(x: u16, y: u16, width: u16, height: u16) -> Rect { let max_area = u16::max_value(); let (clipped_width, clipped_height) = if u32::from(width) * u32::from(height) > u32::from(max_area) { let aspect_ratio = f64::from(width) / f64::from(height); let max_area_f = f64::from(max_area); let height_f = (max_area_f / aspect_ratio).sqrt(); let width_f = height_f * aspect_ratio; (width_f as u16, height_f as u16) } else { (width, height) }; Rect { x, y, width: clipped_width, height: clipped_height, } } #[inline] pub fn area(self) -> u16 { self.width * self.height } #[inline] pub fn left(self) -> u16 { self.x } #[inline] pub fn right(self) -> u16 { self.x.saturating_add(self.width) } #[inline] pub fn top(self) -> u16 { self.y } #[inline] pub fn bottom(self) -> u16 { self.y.saturating_add(self.height) } // Returns a new Rect with width reduced from the left side. // This changes the `x` coordinate and clamps it to the right // edge of the original Rect. pub fn clip_left(self, width: u16) -> Rect { let width = std::cmp::min(width, self.width); Rect { x: self.x.saturating_add(width), width: self.width.saturating_sub(width), ..self } } // Returns a new Rect with width reduced from the right side. // This does _not_ change the `x` coordinate. pub fn clip_right(self, width: u16) -> Rect { Rect { width: self.width.saturating_sub(width), ..self } } // Returns a new Rect with height reduced from the top. // This changes the `y` coordinate and clamps it to the bottom // edge of the original Rect. pub fn clip_top(self, height: u16) -> Rect { let height = std::cmp::min(height, self.height); Rect { y: self.y.saturating_add(height), height: self.height.saturating_sub(height), ..self } } // Returns a new Rect with height reduced from the bottom. // This does _not_ change the `y` coordinate. pub fn clip_bottom(self, height: u16) -> Rect { Rect { height: self.height.saturating_sub(height), ..self } } pub fn with_height(self, height: u16) -> Rect { // new height may make area > u16::max_value, so use new() Self::new(self.x, self.y, self.width, height) } pub fn with_width(self, width: u16) -> Rect { Self::new(self.x, self.y, width, self.height) } pub fn inner(self, margin: &Margin) -> Rect { if self.width < 2 * margin.horizontal || self.height < 2 * margin.vertical { Rect::default() } else { Rect { x: self.x + margin.horizontal, y: self.y + margin.vertical, width: self.width - 2 * margin.horizontal, height: self.height - 2 * margin.vertical, } } } /// Calculate the union between two [`Rect`]s. pub fn union(self, other: Rect) -> Rect { // Example: // // If `Rect` A is positioned at `(0, 0)` with a width and height of `5`, // and `Rect` B is positioned at `(5, 0)` with a width and height of `2`, // then this is the resulting union: // // x1 = min(0, 5) => x1 = 0 // y1 = min(0, 0) => y1 = 0 // x2 = max(0 + 5, 5 + 2) => x2 = 7 // y2 = max(0 + 5, 0 + 2) => y2 = 5 let x1 = min(self.x, other.x); let y1 = min(self.y, other.y); let x2 = max(self.x + self.width, other.x + other.width); let y2 = max(self.y + self.height, other.y + other.height); Rect { x: x1, y: y1, width: x2 - x1, height: y2 - y1, } } /// Calculate the intersection between two [`Rect`]s. pub fn intersection(self, other: Rect) -> Rect { // Example: // // If `Rect` A is positioned at `(0, 0)` with a width and height of `5`, // and `Rect` B is positioned at `(5, 0)` with a width and height of `2`, // then this is the resulting intersection: // // x1 = max(0, 5) => x1 = 5 // y1 = max(0, 0) => y1 = 0 // x2 = min(0 + 5, 5 + 2) => x2 = 5 // y2 = min(0 + 5, 0 + 2) => y2 = 2 let x1 = max(self.x, other.x); let y1 = max(self.y, other.y); let x2 = min(self.x + self.width, other.x + other.width); let y2 = min(self.y + self.height, other.y + other.height); Rect { x: x1, y: y1, width: x2 - x1, height: y2 - y1, } } pub fn intersects(self, other: Rect) -> bool { self.x < other.x + other.width && self.x + self.width > other.x && self.y < other.y + other.height && self.y + self.height > other.y } } #[derive(Debug, Clone, Copy, PartialEq)] #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))] pub enum Color { Reset, Black, Red, Green, Yellow, Blue, Magenta, Cyan, Gray, LightRed, LightGreen, LightYellow, LightBlue, LightMagenta, LightCyan, LightGray, White, Rgb(u8, u8, u8), Indexed(u8), } #[cfg(feature = "term")] impl From for crossterm::style::Color { fn from(color: Color) -> Self { use crossterm::style::Color as CColor; match color { Color::Reset => CColor::Reset, Color::Black => CColor::Black, Color::Red => CColor::DarkRed, Color::Green => CColor::DarkGreen, Color::Yellow => CColor::DarkYellow, Color::Blue => CColor::DarkBlue, Color::Magenta => CColor::DarkMagenta, Color::Cyan => CColor::DarkCyan, Color::Gray => CColor::DarkGrey, Color::LightRed => CColor::Red, Color::LightGreen => CColor::Green, Color::LightBlue => CColor::Blue, Color::LightYellow => CColor::Yellow, Color::LightMagenta => CColor::Magenta, Color::LightCyan => CColor::Cyan, Color::LightGray => CColor::Grey, Color::White => CColor::White, Color::Indexed(i) => CColor::AnsiValue(i), Color::Rgb(r, g, b) => CColor::Rgb { r, g, b }, } } } bitflags! { /// Modifier changes the way a piece of text is displayed. /// /// They are bitflags so they can easily be composed. /// /// ## Examples /// /// ```rust /// # use helix_view::graphics::Modifier; /// /// let m = Modifier::BOLD | Modifier::ITALIC; /// ``` #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))] pub struct Modifier: u16 { const BOLD = 0b0000_0000_0001; const DIM = 0b0000_0000_0010; const ITALIC = 0b0000_0000_0100; const UNDERLINED = 0b0000_0000_1000; const SLOW_BLINK = 0b0000_0001_0000; const RAPID_BLINK = 0b0000_0010_0000; const REVERSED = 0b0000_0100_0000; const HIDDEN = 0b0000_1000_0000; const CROSSED_OUT = 0b0001_0000_0000; } } impl FromStr for Modifier { type Err = &'static str; fn from_str(modifier: &str) -> Result { match modifier { "bold" => Ok(Self::BOLD), "dim" => Ok(Self::DIM), "italic" => Ok(Self::ITALIC), "underlined" => Ok(Self::UNDERLINED), "slow_blink" => Ok(Self::SLOW_BLINK), "rapid_blink" => Ok(Self::RAPID_BLINK), "reversed" => Ok(Self::REVERSED), "hidden" => Ok(Self::HIDDEN), "crossed_out" => Ok(Self::CROSSED_OUT), _ => Err("Invalid modifier"), } } } /// Style let you control the main characteristics of the displayed elements. /// /// ```rust /// # use helix_view::graphics::{Color, Modifier, Style}; /// Style::default() /// .fg(Color::Black) /// .bg(Color::Green) /// .add_modifier(Modifier::ITALIC | Modifier::BOLD); /// ``` /// /// It represents an incremental change. If you apply the styles S1, S2, S3 to a cell of the /// terminal buffer, the style of this cell will be the result of the merge of S1, S2 and S3, not /// just S3. /// /// ```rust /// # use helix_view::graphics::{Rect, Color, Modifier, Style}; /// # use helix_tui::buffer::Buffer; /// let styles = [ /// Style::default().fg(Color::Blue).add_modifier(Modifier::BOLD | Modifier::ITALIC), /// Style::default().bg(Color::Red), /// Style::default().fg(Color::Yellow).remove_modifier(Modifier::ITALIC), /// ]; /// let mut buffer = Buffer::empty(Rect::new(0, 0, 1, 1)); /// for style in &styles { /// buffer.get_mut(0, 0).set_style(*style); /// } /// assert_eq!( /// Style { /// fg: Some(Color::Yellow), /// bg: Some(Color::Red), /// add_modifier: Modifier::BOLD, /// sub_modifier: Modifier::empty(), /// }, /// buffer.get(0, 0).style(), /// ); /// ``` /// /// The default implementation returns a `Style` that does not modify anything. If you wish to /// reset all properties until that point use [`Style::reset`]. /// /// ``` /// # use helix_view::graphics::{Rect, Color, Modifier, Style}; /// # use helix_tui::buffer::Buffer; /// let styles = [ /// Style::default().fg(Color::Blue).add_modifier(Modifier::BOLD | Modifier::ITALIC), /// Style::reset().fg(Color::Yellow), /// ]; /// let mut buffer = Buffer::empty(Rect::new(0, 0, 1, 1)); /// for style in &styles { /// buffer.get_mut(0, 0).set_style(*style); /// } /// assert_eq!( /// Style { /// fg: Some(Color::Yellow), /// bg: Some(Color::Reset), /// add_modifier: Modifier::empty(), /// sub_modifier: Modifier::empty(), /// }, /// buffer.get(0, 0).style(), /// ); /// ``` #[derive(Debug, Clone, Copy, PartialEq)] #[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))] pub struct Style { pub fg: Option, pub bg: Option, pub add_modifier: Modifier, pub sub_modifier: Modifier, } impl Default for Style { fn default() -> Style { Style { fg: None, bg: None, add_modifier: Modifier::empty(), sub_modifier: Modifier::empty(), } } } impl Style { /// Returns a `Style` resetting all properties. pub fn reset() -> Style { Style { fg: Some(Color::Reset), bg: Some(Color::Reset), add_modifier: Modifier::empty(), sub_modifier: Modifier::all(), } } /// Changes the foreground color. /// /// ## Examples /// /// ```rust /// # use helix_view::graphics::{Color, Style}; /// let style = Style::default().fg(Color::Blue); /// let diff = Style::default().fg(Color::Red); /// assert_eq!(style.patch(diff), Style::default().fg(Color::Red)); /// ``` pub fn fg(mut self, color: Color) -> Style { self.fg = Some(color); self } /// Changes the background color. /// /// ## Examples /// /// ```rust /// # use helix_view::graphics::{Color, Style}; /// let style = Style::default().bg(Color::Blue); /// let diff = Style::default().bg(Color::Red); /// assert_eq!(style.patch(diff), Style::default().bg(Color::Red)); /// ``` pub fn bg(mut self, color: Color) -> Style { self.bg = Some(color); self } /// Changes the text emphasis. /// /// When applied, it adds the given modifier to the `Style` modifiers. /// /// ## Examples /// /// ```rust /// # use helix_view::graphics::{Color, Modifier, Style}; /// let style = Style::default().add_modifier(Modifier::BOLD); /// let diff = Style::default().add_modifier(Modifier::ITALIC); /// let patched = style.patch(diff); /// assert_eq!(patched.add_modifier, Modifier::BOLD | Modifier::ITALIC); /// assert_eq!(patched.sub_modifier, Modifier::empty()); /// ``` pub fn add_modifier(mut self, modifier: Modifier) -> Style { self.sub_modifier.remove(modifier); self.add_modifier.insert(modifier); self } /// Changes the text emphasis. /// /// When applied, it removes the given modifier from the `Style` modifiers. /// /// ## Examples /// /// ```rust /// # use helix_view::graphics::{Color, Modifier, Style}; /// let style = Style::default().add_modifier(Modifier::BOLD | Modifier::ITALIC); /// let diff = Style::default().remove_modifier(Modifier::ITALIC); /// let patched = style.patch(diff); /// assert_eq!(patched.add_modifier, Modifier::BOLD); /// assert_eq!(patched.sub_modifier, Modifier::ITALIC); /// ``` pub fn remove_modifier(mut self, modifier: Modifier) -> Style { self.add_modifier.remove(modifier); self.sub_modifier.insert(modifier); self } /// Results in a combined style that is equivalent to applying the two individual styles to /// a style one after the other. /// /// ## Examples /// ``` /// # use helix_view::graphics::{Color, Modifier, Style}; /// let style_1 = Style::default().fg(Color::Yellow); /// let style_2 = Style::default().bg(Color::Red); /// let combined = style_1.patch(style_2); /// assert_eq!( /// Style::default().patch(style_1).patch(style_2), /// Style::default().patch(combined)); /// ``` pub fn patch(mut self, other: Style) -> Style { self.fg = other.fg.or(self.fg); self.bg = other.bg.or(self.bg); self.add_modifier.remove(other.sub_modifier); self.add_modifier.insert(other.add_modifier); self.sub_modifier.remove(other.add_modifier); self.sub_modifier.insert(other.sub_modifier); self } } #[cfg(test)] mod tests { use super::*; #[test] fn test_rect_size_truncation() { for width in 256u16..300u16 { for height in 256u16..300u16 { let rect = Rect::new(0, 0, width, height); rect.area(); // Should not panic. assert!(rect.width < width || rect.height < height); // The target dimensions are rounded down so the math will not be too precise // but let's make sure the ratios don't diverge crazily. assert!( (f64::from(rect.width) / f64::from(rect.height) - f64::from(width) / f64::from(height)) .abs() < 1.0 ) } } // One dimension below 255, one above. Area above max u16. let width = 900; let height = 100; let rect = Rect::new(0, 0, width, height); assert_ne!(rect.width, 900); assert_ne!(rect.height, 100); assert!(rect.width < width || rect.height < height); } #[test] fn test_rect_size_preservation() { for width in 0..256u16 { for height in 0..256u16 { let rect = Rect::new(0, 0, width, height); rect.area(); // Should not panic. assert_eq!(rect.width, width); assert_eq!(rect.height, height); } } // One dimension below 255, one above. Area below max u16. let rect = Rect::new(0, 0, 300, 100); assert_eq!(rect.width, 300); assert_eq!(rect.height, 100); } #[test] fn test_rect_chop_from_left() { let rect = Rect::new(0, 0, 20, 30); assert_eq!(Rect::new(10, 0, 10, 30), rect.clip_left(10)); assert_eq!( Rect::new(20, 0, 0, 30), rect.clip_left(40), "x should be clamped to original width if new width is bigger" ); } #[test] fn test_rect_chop_from_right() { let rect = Rect::new(0, 0, 20, 30); assert_eq!(Rect::new(0, 0, 10, 30), rect.clip_right(10)); } #[test] fn test_rect_chop_from_top() { let rect = Rect::new(0, 0, 20, 30); assert_eq!(Rect::new(0, 10, 20, 20), rect.clip_top(10)); assert_eq!( Rect::new(0, 30, 20, 0), rect.clip_top(50), "y should be clamped to original height if new height is bigger" ); } #[test] fn test_rect_chop_from_bottom() { let rect = Rect::new(0, 0, 20, 30); assert_eq!(Rect::new(0, 0, 20, 20), rect.clip_bottom(10)); } fn styles() -> Vec