Josh McKinney 6dcd53bc6b
feat: add ergonomic methods for layouting Rects (#1909)
This commit introduces new methods for the `Rect` struct that simplify
the process of splitting a `Rect` into sub-rects according to a given
`Layout`. By putting these methods on the `Rect` struct, we make it a
bit more natural that a layout is applied to the `Rect` itself, rather
than passing a `Rect` to the `Layout` struct to be split.

Adds:
- `Rect::layout` and `Rect::try_layout` methods that allow splitting a
  `Rect` into an array of sub-rects according to a given `Layout`.
- `Rect::layout_vec` method that returns a `Vec` of sub-rects.
- `Layout::try_areas` method that returns an array of sub-rects, with
  compile-time checks for the number of constraints. This is added
  mainly for consistency with the new `Rect` methods.

```rust
use ratatui_core::layout::{Layout, Constraint, Rect};
let area = Rect::new(0, 0, 10, 10);
let layout = Layout::vertical([Constraint::Fill(1); 2]);

// Rect::layout() infers the number of constraints at compile time:
let [top, main] = area.layout(&layout);

// Rect::try_layout() and Layout::try_areas() do the same, but return a
// Result:
let [top, main] = area.try_layout(&layout)?;
let [top, main] = layout.try_areas(area)?;

// Rect::layout_vec() returns a Vec of sub-rects:
let areas_vec = area.layout_vec(&layout);

// you can also explicitly specify the number of constraints:
let areas = area.layout::<2>(&layout);
let areas = area.try_layout::<2>(&layout)?;
let areas = layout.try_areas::<2>(area)?;
```
2025-06-28 01:23:34 -07:00

127 lines
4.4 KiB
Rust

//! # [Ratatui] `Scrollbar` example
//!
//! The latest version of this example is available in the [widget examples] folder in the
//! repository.
//!
//! Please note that the examples are designed to be run against the `main` branch of the Github
//! repository. This means that you may not be able to compile with the latest release version on
//! crates.io, or the one that you have installed locally.
//!
//! See the [examples readme] for more information on finding examples that match the version of the
//! library you are using.
//!
//! [Ratatui]: https://github.com/ratatui/ratatui
//! [widget examples]: https://github.com/ratatui/ratatui/blob/main/ratatui-widgets/examples
//! [examples readme]: https://github.com/ratatui/ratatui/blob/main/examples/README.md
use color_eyre::Result;
use crossterm::event::{self, KeyCode};
use ratatui::Frame;
use ratatui::layout::{Constraint, Layout, Margin, Rect};
use ratatui::style::{Color, Stylize};
use ratatui::symbols::scrollbar::Set;
use ratatui::text::{Line, Span};
use ratatui::widgets::{Paragraph, Scrollbar, ScrollbarOrientation, ScrollbarState};
fn main() -> Result<()> {
color_eyre::install()?;
let mut vertical = ScrollbarState::new(100);
let mut horizontal = ScrollbarState::new(100);
ratatui::run(|terminal| {
loop {
terminal.draw(|frame| render(frame, &mut vertical, &mut horizontal))?;
if let Some(key) = event::read()?.as_key_press_event() {
match key.code {
KeyCode::Char('q') | KeyCode::Esc => break Ok(()),
KeyCode::Char('j') | KeyCode::Down => vertical.next(),
KeyCode::Char('k') | KeyCode::Up => vertical.prev(),
KeyCode::Char('l') | KeyCode::Right => horizontal.next(),
KeyCode::Char('h') | KeyCode::Left => horizontal.prev(),
_ => {}
}
}
}
})
}
/// Render the UI with vertical/horizontal scrollbars.
fn render(frame: &mut Frame, vertical: &mut ScrollbarState, horizontal: &mut ScrollbarState) {
let layout = Layout::vertical([Constraint::Length(1), Constraint::Fill(1)]).spacing(1);
let [top, main] = frame.area().layout(&layout);
let title = Line::from_iter([
Span::from("Scrollbar Widget").bold(),
Span::from(" (Press 'q' to quit, arrow keys to scroll)"),
]);
frame.render_widget(title.centered(), top);
render_content(frame, main, vertical, horizontal);
render_vertical_scrollbar(frame, main, vertical);
render_horizontal_scrollbar(frame, main, horizontal);
}
/// Render a vertical scrollbar on the right side of the area.
pub fn render_vertical_scrollbar(frame: &mut Frame, area: Rect, vertical: &mut ScrollbarState) {
let scrollbar = Scrollbar::new(ScrollbarOrientation::VerticalRight);
frame.render_stateful_widget(
scrollbar,
area.inner(Margin {
vertical: 1,
horizontal: 0,
}),
vertical,
);
}
/// Render a horizontal scrollbar at the bottom of the area.
pub fn render_horizontal_scrollbar(frame: &mut Frame, area: Rect, horizontal: &mut ScrollbarState) {
let scrollbar = Scrollbar::new(ScrollbarOrientation::HorizontalBottom)
.symbols(Set {
track: "-",
thumb: "",
begin: "<",
end: ">",
})
.track_style(Color::Yellow)
.begin_style(Color::Green)
.end_style(Color::Red);
frame.render_stateful_widget(
scrollbar,
area.inner(Margin {
vertical: 0,
horizontal: 1,
}),
horizontal,
);
}
/// Render some content.
fn render_content(
frame: &mut Frame,
area: Rect,
vertical: &ScrollbarState,
horizontal: &ScrollbarState,
) {
let content = vec![
Line::from("This is a paragraph with a vertical and horizontal scrollbar."),
Line::from_iter(["Lorem ipsum dolor sit amet, consectetur adipiscing elit.".repeat(10)]),
Line::from_iter([
"Horizontal: ".bold(),
horizontal.get_position().to_string().yellow(),
]),
Line::from_iter([
"Vertical: ".bold(),
vertical.get_position().to_string().yellow(),
]),
];
frame.render_widget(
Paragraph::new(content).scroll((
vertical.get_position() as u16,
horizontal.get_position() as u16,
)),
area,
);
}