1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
use crossterm::{
execute,
terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen},
};
use std::io;
use tui::{
backend::{Backend, CrosstermBackend},
layout::{Alignment, Constraint, Direction, Layout, Rect},
style::{Modifier, Style},
text::{Span, Spans},
widgets::{Block, Borders, List, ListItem, Paragraph, Widget},
Frame, Terminal,
};
use crate::{AngleMode, Calculator, Constant, Function};
const WIDTH: u16 = 32;
impl Calculator {
pub fn draw<B: Backend>(&self, f: &mut Frame<B>) {
let chunks = layout(self.stack.len(), f.size());
f.render_widget(version_number_widget(), chunks[0]);
f.render_widget(angle_mode_widget(self.angle_mode), chunks[2]);
f.render_widget(stack_widget(&self.stack), chunks[3]);
f.render_widget(input_buffer_widget(&self.input_buffer), chunks[4]);
}
}
pub fn init_terminal() -> Result<Terminal<CrosstermBackend<io::Stdout>>, io::Error> {
enable_raw_mode()?;
let mut stdout = io::stdout();
execute!(stdout, EnterAlternateScreen)?;
let backend = CrosstermBackend::new(stdout);
Terminal::new(backend)
}
pub fn cleanup_terminal<B>(mut terminal: Terminal<B>) -> Result<(), io::Error>
where
B: Backend + io::Write,
{
disable_raw_mode()?;
execute!(terminal.backend_mut(), LeaveAlternateScreen,)?;
terminal.show_cursor()?;
Ok(())
}
fn layout(stack_size: usize, frame_size: Rect) -> Vec<Rect> {
let columns = Layout::default()
.direction(Direction::Horizontal)
.constraints([Constraint::Length(WIDTH), Constraint::Max(u16::MAX)].as_ref())
.split(frame_size);
Layout::default()
.direction(Direction::Vertical)
.constraints(
[
Constraint::Length(1), // Version number
Constraint::Max(u16::MAX), // Fill
Constraint::Length(1), // Angle mode
Constraint::Length(stack_size as u16 + 2), // Stack
Constraint::Length(3), // Input buffer
]
.as_ref(),
)
.split(columns[0])
}
fn angle_mode_widget(angle_mode: AngleMode) -> impl Widget {
Paragraph::new(format!("{}", angle_mode)).alignment(Alignment::Right)
}
fn stack_widget(stack: &Vec<f64>) -> impl Widget {
List::new(
stack
.iter()
.map(|f| ListItem::new(format!(" {}", f)))
.collect::<Vec<ListItem>>(),
)
.block(Block::default().borders(Borders::ALL))
}
fn input_buffer_widget(input_buffer: &str) -> impl Widget {
Paragraph::new(Spans::from(vec![
Span::raw(">"),
Span::styled(
format!(" {}", input_buffer),
input_buffer_style(input_buffer),
),
]))
.block(Block::default().borders(Borders::ALL))
}
fn version_number_widget() -> impl Widget {
Paragraph::new(format!("pfc-{}", option_env!("CARGO_PKG_VERSION").unwrap()))
.alignment(Alignment::Center)
}
fn input_buffer_style(input_buffer: &str) -> Style {
if let Ok(_) = Function::parse(&input_buffer) {
Style::default().add_modifier(Modifier::BOLD)
} else if let Ok(_) = Constant::parse(&input_buffer) {
Style::default().add_modifier(Modifier::BOLD)
} else {
Style::default()
}
}
|