aboutsummaryrefslogtreecommitdiffstats
path: root/src/input.rs
blob: 2d982e591f17fd2bed9e499c7e5ceb3f58f9a10f (plain) (blame)
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
use crossterm::event::{KeyCode, KeyEvent, KeyModifiers};

use crate::{Calculator, Constant, Function, Operator, Signal};

impl Calculator {
    pub fn handle_input(&mut self, key: KeyEvent) -> Signal {
        match key.modifiers {
            KeyModifiers::CONTROL => match key.code {
                KeyCode::Char('c') => {
                    return Signal::Exit;
                }
                _ => {}
            },
            KeyModifiers::SHIFT => match key.code {
                KeyCode::Char('Q') => {
                    return Signal::Exit;
                }
                KeyCode::Char('J' | 'K') => self.swap(),
                KeyCode::Char('D') => {
                    self.input_buffer = String::new();
                }
                KeyCode::Char('C') => self.clear(),
                KeyCode::Char('A') => {
                    self.angle_mode = self.angle_mode.toggle();
                }
                KeyCode::Char('N') => {
                    if self.input_buffer.len() > 0 {
                        if let Ok(f) = self.input_buffer.parse::<f64>() {
                            self.input_buffer = format!("{}", -f);
                        }
                    } else if let Some(v) = self.stack.pop() {
                        self.stack.push(-v);
                    }
                }
                _ => {}
            },
            KeyModifiers::NONE => match key.code {
                KeyCode::Char(c) => self.push_to_buffer(c),
                KeyCode::Backspace => {
                    self.input_buffer.pop();
                }
                KeyCode::Enter => {
                    if let Ok(func) = Function::parse(&self.input_buffer) {
                        if let Some(val) = self.stack.pop() {
                            self.stack.push(func.call(val, self.angle_mode));
                        }
                    } else if let Ok(c) = Constant::parse(&self.input_buffer) {
                        self.stack.push(c.value());
                    } else if let Ok(bu) = self.input_buffer.parse::<f64>() {
                        self.stack.push(bu);
                    }
                    self.input_buffer = String::new();
                }
                _ => {}
            },
            _ => {}
        }
        return Signal::None;
    }

    fn push_to_buffer(&mut self, c: char) {
        if c == '.' && !self.input_buffer.contains('.') {
            if self.input_buffer.len() == 0 {
                self.input_buffer.push('0');
            }
            self.input_buffer.push(c);
        } else if let Ok(op) = Operator::parse(c) {
            if let Ok(c) = Constant::parse(&self.input_buffer) {
                self.stack.push(c.value());
            } else if let Ok(f) = self.input_buffer.parse::<f64>() {
                self.stack.push(f);
            }
            self.input_buffer = String::new();
            self.perform_operation(op);
        } else {
            self.input_buffer.push(c);
        }
    }

    fn swap(&mut self) {
        if let Some(st) = self.stack.pop() {
            if let Ok(bu) = self.input_buffer.parse::<f64>() {
                self.stack.push(bu);
            }
            self.input_buffer = format!("{}", st);
        }
    }

    // Clear stack and input buffer.
    fn clear(&mut self) {
        self.input_buffer = String::new();
        self.stack = Vec::new();
    }
}