Compare commits

1 Commits

Author SHA1 Message Date
dee88c916e wip 2025-11-19 09:41:55 -07:00
14 changed files with 263 additions and 549 deletions

View File

@@ -1,21 +0,0 @@
name: Test Rust project
on: [push]
jobs:
test:
runs-on: ${{ matrix.os }}
strategy:
matrix:
os: [ubuntu-latest]
rust: [stable]
steps:
- uses: https://git.oliveratkinson.net/Oliver/setup-rust-action@master
with:
rust-version: ${{ matrix.rust }}
- uses: actions/checkout@master
- name: Run tests
run: cargo test --verbose
- name: Clippy
run: cargo clippy

View File

@@ -2,12 +2,8 @@
*New* Spreadsheet Calculator Improved *New* Spreadsheet Calculator Improved
![Screenshot](readme/screenshot.png)
Based loosely off sc-im (spreadsheet calculator improvised), which has dumb keybinds and not many features. Based loosely off sc-im (spreadsheet calculator improvised), which has dumb keybinds and not many features.
[Checkout a demo file](readme/demo.nscim), download, then open with `neoscim demo.nscim`.
## Keybinds ## Keybinds
### Normal mode ### Normal mode
@@ -18,13 +14,8 @@ Based loosely off sc-im (spreadsheet calculator improvised), which has dumb keyb
| `k` | Move up | | `k` | Move up |
| `h` | Move left | | `h` | Move left |
| `l` | Move right | | `l` | Move right |
| `0` | Go to beginning of the row | | `0` | Go to beginning of row |
| `g0` | Go to beginning of the visual row | | `gg` | Go to beginning of column |
| `$` | Go to end of the row |
| `g$` | Go to end of the visual row |
| `gg` | Go to beginning of the column |
| `G` | Go to end of column |
| `gG` | Go to end of the the visual column |
| `i`/`a` | Enter insert mode on current cell | | `i`/`a` | Enter insert mode on current cell |
| `r` | Enter insert mode on current cell, deleting contents | | `r` | Enter insert mode on current cell, deleting contents |
| `v` | Enter visual mode | | `v` | Enter visual mode |
@@ -60,16 +51,6 @@ Based loosely off sc-im (spreadsheet calculator improvised), which has dumb keyb
| `:q` | Quit program | | `:q` | Quit program |
| `:q!` | Quit program, even if the file isn't saved | | `:q!` | Quit program, even if the file isn't saved |
#### Visual mode commands
These commands operate on selections.
| Commmand | Description |
| - | - |
| `:plot` | Plots the current selection to `plot.png` using gnuplot |
| `:export <filename>` | Exports the current selection to a new file |
| `:fill <value>` | (Aliased as `f`) Fill the selection with `<value>`, special variables `x`,`y`,`xi`, and `yi` are avaliable. Variables `x` and `y` are the global coordinates of the cell, and `xi` and `yi` are the local coordinates of the selection. |
## Math / Functions ## Math / Functions
### Math ### Math
@@ -87,11 +68,11 @@ These commands operate on selections.
| Identifier | Arg Qty | Argument Types | Description | | Identifier | Arg Qty | Argument Types | Description |
| - | - | - | - | | - | - | - | - |
| `avg` | 1 | Numeric | Returns the average of the arguments | | `avg` | >= 1 | Numeric | Returns the average of the arguments |
| `sum` | 1 | Numbers | Returns the sum of the arguments | | `sum` | >= 1 | Numbers | Returns the sum of the arguments |
| `xlookup` | 3 | Number/String, Range, Range | Searches for the number/string in the first range, returning the item at the same index in the second range | | `xlookup` | 3 | Range, Number/String, Range | Searches for the number/string in the first range, returning the item at the same index in the second range |
| `min` | 1 | Numeric | Returns the minimum of the arguments | | `min` | >= 1 | Numeric | Returns the minimum of the arguments |
| `max` | 1 | Numeric | Returns the maximum of the arguments | | `max` | >= 1 | Numeric | Returns the maximum of the arguments |
| `len` | 1 | String/Tuple | Returns the character length of a string, or the amount of elements in a tuple (not recursively) | | `len` | 1 | String/Tuple | Returns the character length of a string, or the amount of elements in a tuple (not recursively) |
| `floor` | 1 | Numeric | Returns the largest integer less than or equal to a number | | `floor` | 1 | Numeric | Returns the largest integer less than or equal to a number |
| `round` | 1 | Numeric | Returns the nearest integer to a number. Rounds half-way cases away from 0.0 | | `round` | 1 | Numeric | Returns the nearest integer to a number. Rounds half-way cases away from 0.0 |
@@ -131,7 +112,7 @@ These commands operate on selections.
| `str::to_lowercase` | 1 | String | Returns the lower-case version of the string | | `str::to_lowercase` | 1 | String | Returns the lower-case version of the string |
| `str::to_uppercase` | 1 | String | Returns the upper-case version of the string | | `str::to_uppercase` | 1 | String | Returns the upper-case version of the string |
| `str::trim` | 1 | String | Strips whitespace from the start and the end of the string | | `str::trim` | 1 | String | Strips whitespace from the start and the end of the string |
| `str::from` | 0 | Any | Returns passed value as string | | `str::from` | >= 0 | Any | Returns passed value as string |
| `str::substring` | 3 | String, Int, Int | Returns a substring of the first argument, starting at the second argument and ending at the third argument. If the last argument is omitted, the substring extends to the end of the string | | `str::substring` | 3 | String, Int, Int | Returns a substring of the first argument, starting at the second argument and ending at the third argument. If the last argument is omitted, the substring extends to the end of the string |
| `bitand` | 2 | Int | Computes the bitwise and of the given integers | | `bitand` | 2 | Int | Computes the bitwise and of the given integers |
| `bitor` | 2 | Int | Computes the bitwise or of the given integers | | `bitor` | 2 | Int | Computes the bitwise or of the given integers |
@@ -151,8 +132,8 @@ If the cell can be a number (parsed into a float) it will be, if not, then if th
* Cell references move with copy / paste * Cell references move with copy / paste
* =A0: Will translate * =A0: Will translate
* =$A0: Will translate Y only (numbers) * =$A0: Will translate Y only
* =A$0: Will translate X only (letters) * =$A0: Will translate X only
* =$A$0: No translation * =$A$0: No translation
* Keybinds are closer to vim keying (i/a start inserting into a cell) * Keybinds are closer to vim keying (i/a start inserting into a cell)

View File

@@ -1,21 +0,0 @@
,,,
Most of vim's movement works here as well,,,
,,,
,i,,
j,Movement,l,
,j,,
,,,
,"insert text with r, i, or a",,
,,,
1,,,
1,Text can overflow cells,,
1,But may get cut off if it's too long,,10
1,,,If it reaches another cell
1,,,
Total:,=sum(A:A),Sum A:A,
,=math::log2(B14),Do math on the output of another function,
,,,
,,,
,,11,Number.
,,=C18+1,Referencing the number and adding 1
,,,Copying the cell down will translate the reference

Binary file not shown.

Before

Width:  |  Height:  |  Size: 21 KiB

View File

@@ -8,18 +8,19 @@ use std::{
use ratatui::{ use ratatui::{
DefaultTerminal, Frame, DefaultTerminal, Frame,
crossterm::event, crossterm::event::{self, KeyCode},
layout::{self, Constraint, Layout, Rect}, layout::{self, Constraint, Layout, Margin, Rect},
prelude, prelude,
style::{Color, Modifier, Style}, style::{Color, Modifier, Style},
widgets::{Paragraph, Widget}, widgets::{Block, Borders, Paragraph, Widget},
}; };
use crate::app::{ use crate::app::{
clipboard::Clipboard, clipboard::Clipboard,
error_msg::StatusMessage, error_msg::StatusMessage,
logic::{calc::{Grid, get_header_size}, cell::CellType}, logic::{calc::Grid, cell::CellType},
mode::Mode, mode::Mode,
plot::Plot,
screen::ScreenSpace, screen::ScreenSpace,
}; };
@@ -35,6 +36,7 @@ pub struct App {
// this could probably be a normal array // this could probably be a normal array
pub marks: HashMap<char, (usize, usize)>, pub marks: HashMap<char, (usize, usize)>,
pub clipboard: Clipboard, pub clipboard: Clipboard,
pub plot_popup: Option<Plot>,
} }
impl Widget for &App { impl Widget for &App {
@@ -63,12 +65,8 @@ impl Widget for &App {
for x in 0..x_max { for x in 0..x_max {
for y in 0..y_max { for y in 0..y_max {
let mut display = String::new(); let mut display = String::new();
let mut style = Style::new(); let mut style = Style::new().fg(Color::White);
// Custom width for the header of each row
let row_header_width = get_header_size() as u16;
// ^^ Feels like it oculd be static but evaluating string lens doesn't work at
// compile time. Thus cannot be static.
let cell_width = self.screen.get_cell_width(&self.vars) as u16; let cell_width = self.screen.get_cell_width(&self.vars) as u16;
let cell_height = self.screen.get_cell_height(&self.vars) as u16; let cell_height = self.screen.get_cell_height(&self.vars) as u16;
@@ -90,15 +88,6 @@ impl Widget for &App {
let mut should_render = true; let mut should_render = true;
let mut suggest_upper_bound = None; let mut suggest_upper_bound = None;
/// Center the text "99 " -> " 99 "
fn center_text(text: &str, avaliable_space: i32) -> String {
let margin = avaliable_space - text.len() as i32;
let margin = margin/2;
let l_margin = (0..margin).into_iter().map(|_| ' ').collect::<String>();
let r_margin = (0..(margin-(l_margin.len() as i32))).into_iter().map(|_| ' ').collect::<String>();
format!("{l_margin}{text}{r_margin}")
}
match (x == 0, y == 0) { match (x == 0, y == 0) {
// 0,0 vi mode // 0,0 vi mode
(true, true) => { (true, true) => {
@@ -107,7 +96,7 @@ impl Widget for &App {
} }
// row names // row names
(true, false) => { (true, false) => {
display = center_text(&y_idx.to_string(), row_header_width as i32); display = y_idx.to_string();
let bg = if y_idx == self.grid.cursor().1 { let bg = if y_idx == self.grid.cursor().1 {
Color::DarkGray Color::DarkGray
@@ -120,7 +109,7 @@ impl Widget for &App {
} }
// column names // column names
(false, true) => { (false, true) => {
display = center_text(&Grid::num_to_char(x_idx), cell_width as i32); display = Grid::num_to_char(x_idx);
let bg = if x_idx == self.grid.cursor().0 { let bg = if x_idx == self.grid.cursor().0 {
Color::DarkGray Color::DarkGray
@@ -136,7 +125,6 @@ impl Widget for &App {
(false, false) => { (false, false) => {
match self.grid.get_cell_raw(x_idx, y_idx) { match self.grid.get_cell_raw(x_idx, y_idx) {
Some(cell) => { Some(cell) => {
// Render in different colors based on type of contents
match cell { match cell {
CellType::Number(c) => display = c.to_string(), CellType::Number(c) => display = c.to_string(),
CellType::String(s) => { CellType::String(s) => {
@@ -148,9 +136,6 @@ impl Widget for &App {
Ok(val) => { Ok(val) => {
display = val.to_string(); display = val.to_string();
style = Style::new() style = Style::new()
.fg(Color::White)
// TODO This breaks dumb terminals like the windows
// terminal
.underline_color(Color::DarkGray) .underline_color(Color::DarkGray)
.add_modifier(Modifier::UNDERLINED); .add_modifier(Modifier::UNDERLINED);
} }
@@ -166,7 +151,6 @@ impl Widget for &App {
} }
} }
// Allow for text in one cell to visually overflow into empty cells
suggest_upper_bound = Some(display.len() as u16); suggest_upper_bound = Some(display.len() as u16);
// check for cells to the right, see if we should truncate the cell width // check for cells to the right, see if we should truncate the cell width
for i in 1..(display.len() as f32 / cell_width as f32).ceil() as usize { for i in 1..(display.len() as f32 / cell_width as f32).ceil() as usize {
@@ -183,14 +167,11 @@ impl Widget for &App {
} }
} }
} }
// Don't render blank cells
None => should_render = false, None => should_render = false,
} }
if is_selected(x.into(), y.into()) { if is_selected(x.into(), y.into()) {
style = style.bg(Color::Blue); style = style.bg(Color::Blue);
// Make it so that cells render when selected. This fixes issue #32
should_render = true;
} }
if (x_idx, y_idx) == self.grid.cursor() { if (x_idx, y_idx) == self.grid.cursor() {
should_render = true; should_render = true;
@@ -202,21 +183,11 @@ impl Widget for &App {
} }
} }
} }
if should_render { if should_render {
let mut x_off = area.x + (x * cell_width); let x_off = area.x + (x * cell_width);
let y_off = area.y + (y * cell_height); let y_off = area.y + (y * cell_height);
// Adjust for the fact that the first column let area = if let Some(suggestion) = suggest_upper_bound {
// is smaller, since it is just headers
if x > 0 {
x_off = x_off - (cell_width - row_header_width);
}
// If this is the row header column
let area = if x==0 && y != 0 {
Rect::new(x_off, y_off, row_header_width, cell_height)
} else if let Some(suggestion) = suggest_upper_bound {
let max_available_width = area.width - x_off; let max_available_width = area.width - x_off;
// draw the biggest cell possible, without going OOB off the screen // draw the biggest cell possible, without going OOB off the screen
let width = min(max_available_width, suggestion as u16); let width = min(max_available_width, suggestion as u16);
@@ -248,6 +219,7 @@ impl App {
marks: HashMap::new(), marks: HashMap::new(),
clipboard: Clipboard::new(), clipboard: Clipboard::new(),
file_modified_date: SystemTime::now(), file_modified_date: SystemTime::now(),
plot_popup: None,
} }
} }
@@ -346,9 +318,46 @@ impl App {
)), )),
cmd_line_debug, cmd_line_debug,
); );
// popups
if let Some(plot) = &self.plot_popup {
let block = Block::default()
.title("Plot Editor")
.title_alignment(layout::Alignment::Center)
.border_type(ratatui::widgets::BorderType::Rounded)
.borders(Borders::all())
.style(Style::default().fg(Color::White));
let popup_y = 20;
let popup_x = 40;
let a = frame.area();
let xpos = (a.width / 2) - (popup_x / 2);
let ypos = (a.height / 2) - (popup_y / 2);
let area = Rect::new(xpos, ypos, popup_x, popup_y);
frame.render_widget(ratatui::widgets::Clear, area);
frame.render_widget(block, area);
let area = area.inner(Margin::new(1, 1));
frame.render_widget(plot, area);
}
} }
fn handle_events(&mut self) -> io::Result<()> { fn handle_events(&mut self) -> io::Result<()> {
if let Some(plot) = &mut self.plot_popup {
if let event::Event::Key(key) = event::read()? {
match key.code {
KeyCode::Esc => self.plot_popup = None,
KeyCode::PrintScreen | KeyCode::Enter => unimplemented!("Generate plot data"),
KeyCode::Delete => plot.del_column(),
KeyCode::Insert => plot.add_column(),
KeyCode::Char(c) => plot.process_key(c),
_ => {}
}
}
return Ok(());
}
match &mut self.mode { match &mut self.mode {
Mode::VisualCmd(pos, chord) => match event::read()? { Mode::VisualCmd(pos, chord) => match event::read()? {
event::Event::Key(key) => match key.code { event::Event::Key(key) => match key.code {

View File

@@ -49,7 +49,6 @@ impl Clipboard {
// cursor // cursor
let (cx, cy) = into.cursor(); let (cx, cy) = into.cursor();
// iterate thru the clipbaord's cells
for (x, row) in self.clipboard.iter().enumerate() { for (x, row) in self.clipboard.iter().enumerate() {
for (y, cell) in row.iter().enumerate() { for (y, cell) in row.iter().enumerate() {
let idx = (x + cx, y + cy); let idx = (x + cx, y + cy);
@@ -59,7 +58,7 @@ impl Clipboard {
let trans = cell.translate_cell(self.source_cell, into.cursor()); let trans = cell.translate_cell(self.source_cell, into.cursor());
into.set_cell_raw(idx, Some(trans)); into.set_cell_raw(idx, Some(trans));
} else { } else {
// The cell at this location doesn't exist (empty) // cell doesn't exist, no need to translate
into.set_cell_raw::<CellType>(idx, None); into.set_cell_raw::<CellType>(idx, None);
} }
} else { } else {
@@ -351,48 +350,3 @@ fn copy_paste_y_locked_var() {
let c = app.grid.get_cell("B2").as_ref().expect("Just set it"); let c = app.grid.get_cell("B2").as_ref().expect("Just set it");
assert_eq!(c.to_string(), "=B$0"); assert_eq!(c.to_string(), "=B$0");
} }
#[test]
fn copy_paste_var_in_function() {
let mut app = App::new();
app.grid.set_cell("A0", 4.to_string());
Mode::process_key(&mut app, 'j');
app.grid.set_cell("A1", "=math::log2(A0)".to_string());
app.mode = super::mode::Mode::Chord(Chord::new('y'));
Mode::process_key(&mut app, 'y');
Mode::process_key(&mut app, 'j');
Mode::process_key(&mut app, 'p');
let a = app.grid.get_cell("A2").as_ref().expect("Should've been set by paste");
assert_eq!(a.to_string(), "=math::log2(A1)");
}
#[test]
fn copy_paste_range_in_function() {
let mut app = App::new();
app.grid.set_cell("A0", 1.to_string());
app.grid.set_cell("A1", 1.to_string());
app.grid.set_cell("A2", 1.to_string());
app.grid.set_cell("B0", "=sum(A:A)".to_string());
app.grid.mv_cursor_to(1, 0);
app.mode = super::mode::Mode::Chord(Chord::new('y'));
Mode::process_key(&mut app, 'y');
Mode::process_key(&mut app, 'l');
Mode::process_key(&mut app, 'p');
let a = app.grid.get_cell("C0").as_ref().expect("Should've been set by paste");
assert_eq!(a.to_string(), "=sum(B:B)");
// now copy the range the other direction
app.grid.mv_cursor_to(2, 0);
app.mode = super::mode::Mode::Chord(Chord::new('y'));
Mode::process_key(&mut app, 'y');
app.grid.mv_cursor_to(1, 1);
Mode::process_key(&mut app, 'p');
let a = app.grid.get_cell("B1").as_ref().expect("Should've been set by paste");
assert_eq!(a.to_string(), "=sum(A:A)");
}

View File

@@ -17,12 +17,7 @@ use crate::app::{
#[cfg(test)] #[cfg(test)]
use crate::app::app::App; use crate::app::app::App;
pub fn get_header_size() -> usize { pub const LEN: usize = 1000;
let row_header_width = LEN.to_string().len();
row_header_width
}
pub const LEN: usize = 1001;
pub const CSV_EXT: &str = "csv"; pub const CSV_EXT: &str = "csv";
pub const CUSTOM_EXT: &str = "nscim"; pub const CUSTOM_EXT: &str = "nscim";
@@ -86,7 +81,7 @@ impl Grid {
/// Save file to `path` as a csv. Path with have `csv` appended to it if it /// Save file to `path` as a csv. Path with have `csv` appended to it if it
/// does not already have the extension. /// does not already have the extension.
pub fn save_to(&mut self, path: impl Into<PathBuf>) -> std::io::Result<()> { pub fn save_to(&mut self, path: impl Into<PathBuf>) -> std::io::Result<()> {
let mut path = path.into(); let path = path.into();
let resolve_values; let resolve_values;
@@ -99,15 +94,13 @@ impl Grid {
resolve_values = false; resolve_values = false;
} }
_ => { _ => {
// File as an extension but isn't ours. resolve_values = false;
// Save as csv-like // path.add_extension(CUSTOM_EXT);
resolve_values = true;
} }
}, },
None => { None => {
// File has no extension. Save it as our file type
resolve_values = false; resolve_values = false;
path.add_extension(CUSTOM_EXT); // path.add_extension(CUSTOM_EXT);
} }
} }
@@ -270,11 +263,9 @@ impl Grid {
} }
} }
} }
pub fn insert_row_below(&mut self, (x, y): (usize, usize)) { pub fn insert_row_below(&mut self, (x, y): (usize, usize)) {
self.insert_row_above((x,y+1)); self.insert_row_above((x,y+1));
} }
pub fn insert_column_before(&mut self, (insertion_x, _y): (usize, usize)) { pub fn insert_column_before(&mut self, (insertion_x, _y): (usize, usize)) {
let mut v = Vec::with_capacity(LEN); let mut v = Vec::with_capacity(LEN);
for _ in 0..LEN { for _ in 0..LEN {
@@ -291,7 +282,6 @@ impl Grid {
// add 1 because of the insertion // add 1 because of the insertion
if arg_x < insertion_x { rolling.to_owned() } else { rolling.replace(old, new) } if arg_x < insertion_x { rolling.to_owned() } else { rolling.replace(old, new) }
} else { } else {
// FIXME could be a range
unimplemented!("Invalid variable wanted to be translated") unimplemented!("Invalid variable wanted to be translated")
} }
}) })
@@ -301,7 +291,6 @@ impl Grid {
} }
} }
} }
pub fn insert_column_after(&mut self, (x, y): (usize, usize)) { pub fn insert_column_after(&mut self, (x, y): (usize, usize)) {
self.insert_column_before((x + 1, y)); self.insert_column_before((x + 1, y));
} }
@@ -350,7 +339,7 @@ impl Grid {
/// Only evaluates equations, such as `=10` or `=A1/C2`, not /// Only evaluates equations, such as `=10` or `=A1/C2`, not
/// strings or numbers. /// strings or numbers.
pub fn evaluate(&self, mut eq: &str) -> Result<CellType, String> { pub fn evaluate(&self, mut eq: &str) -> Result<f64, String> {
if eq.starts_with('=') { if eq.starts_with('=') {
eq = &eq[1..]; eq = &eq[1..];
} else { } else {
@@ -364,15 +353,11 @@ impl Grid {
if v.is_number() { if v.is_number() {
if v.is_float() { if v.is_float() {
let val = v.as_float().expect("Value lied about being a float"); let val = v.as_float().expect("Value lied about being a float");
return Ok(CellType::Number(val)); return Ok(val);
} else if v.is_int() { } else if v.is_int() {
let val = v.as_int().expect("Value lied about being an int"); let i = v.as_int().expect("Value lied about being an int");
return Ok(CellType::Number(val as f64)); return Ok(i as f64);
} }
} else if v.is_string() {
// ^^ This allows for functions to return a string
let s = v.as_string().expect("Value lied about being a String");
return Ok(CellType::String(s));
} }
return Err("Result is NaN".to_string()); return Err("Result is NaN".to_string());
}; };
@@ -397,36 +382,12 @@ impl Grid {
} }
} }
pub fn range_as_indices(range: &str) -> Option<(usize, usize)> {
let v = range.split(':').collect::<Vec<&str>>();
if v.len() == 2 {
let start_col = v[0];
let end_col = v[1];
let start_idx = Grid::char_to_idx(start_col);
let end_idx = Grid::char_to_idx(end_col);
return Some((start_idx, end_idx))
}
None
}
pub fn char_to_idx(i: &str) -> usize {
let x_idx = i
.chars()
.filter(|f| f.is_alphabetic())
.enumerate()
.map(|(idx, c)| ((c.to_ascii_lowercase() as usize).saturating_sub(97)) + (26 * idx))
.fold(0, |a, b| a + b);
x_idx
}
/// Parse values in the format of A0, C10 ZZ99, etc, and /// Parse values in the format of A0, C10 ZZ99, etc, and
/// turn them into an X,Y index. /// turn them into an X,Y index.
pub fn parse_to_idx(i: &str) -> Option<(usize, usize)> { pub fn parse_to_idx(i: &str) -> Option<(usize, usize)> {
let i = i.replace('$', ""); let i = i.replace('$', "");
let chars = i.chars().take_while(|c| c.is_alphabetic()).collect::<String>(); let chars = i.chars().take_while(|c| c.is_alphabetic()).collect::<Vec<char>>();
let nums = i.chars().skip(chars.len()).take_while(|c| c.is_numeric()).collect::<String>(); let nums = i.chars().skip(chars.len()).take_while(|c| c.is_numeric()).collect::<String>();
// At least half the arguments are gone // At least half the arguments are gone
@@ -435,7 +396,11 @@ impl Grid {
} }
// get the x index from the chars // get the x index from the chars
let x_idx = Self::char_to_idx(&chars); let x_idx = chars
.iter()
.enumerate()
.map(|(idx, c)| (c.to_ascii_lowercase() as usize - 97) + (26 * idx))
.fold(0, |a, b| a + b);
// get the y index from the numbers // get the y index from the numbers
if let Ok(y_idx) = nums.parse::<usize>() { if let Ok(y_idx) = nums.parse::<usize>() {
@@ -527,11 +492,11 @@ fn saving_csv() {
// insure that the cells are there // insure that the cells are there
let cell = app.grid.get_cell_raw(0, 10).as_ref().expect("Should've been set"); let cell = app.grid.get_cell_raw(0, 10).as_ref().expect("Should've been set");
let res = app.grid.evaluate(&cell.to_string()).expect("Should evaluate"); let res = app.grid.evaluate(&cell.to_string()).expect("Should evaluate");
assert_eq!(res, (11.0).into()); assert_eq!(res, 11.0);
assert_eq!(cell.escaped_csv_string(), "=A9+A$0"); assert_eq!(cell.escaped_csv_string(), "=A9+A$0");
let cell = app.grid.get_cell_raw(1, 10).as_ref().expect("Should've been set"); let cell = app.grid.get_cell_raw(1, 10).as_ref().expect("Should've been set");
let res = app.grid.evaluate(&cell.to_string()).expect("Should evaluate"); let res = app.grid.evaluate(&cell.to_string()).expect("Should evaluate");
assert_eq!(res, (121.0).into()); assert_eq!(res, 121.0);
assert_eq!(cell.escaped_csv_string(), "=A10^2"); assert_eq!(cell.escaped_csv_string(), "=A10^2");
// set saving the file // set saving the file
@@ -570,16 +535,16 @@ fn saving_neoscim() {
// insure that the cells are there // insure that the cells are there
let cell = app.grid.get_cell_raw(0, 10).as_ref().expect("Should've been set"); let cell = app.grid.get_cell_raw(0, 10).as_ref().expect("Should've been set");
let res = app.grid.evaluate(&cell.to_string()).expect("Should evaluate"); let res = app.grid.evaluate(&cell.to_string()).expect("Should evaluate");
assert_eq!(res, (11.0).into()); assert_eq!(res, 11.0);
assert_eq!(cell.escaped_csv_string(), "=A9+A$0"); assert_eq!(cell.escaped_csv_string(), "=A9+A$0");
let cell = app.grid.get_cell_raw(1, 10).as_ref().expect("Should've been set"); let cell = app.grid.get_cell_raw(1, 10).as_ref().expect("Should've been set");
let res = app.grid.evaluate(&cell.to_string()).expect("Should evaluate"); let res = app.grid.evaluate(&cell.to_string()).expect("Should evaluate");
assert_eq!(res, (121.0).into()); assert_eq!(res, 121.0);
assert_eq!(cell.escaped_csv_string(), "=A10^2"); assert_eq!(cell.escaped_csv_string(), "=A10^2");
// set saving the file // set saving the file
let filename = format!("/tmp/file.{CUSTOM_EXT}"); let filename= "/tmp/file.neoscim";
app.grid.save_to(&filename).expect("This will only work on linux systems"); app.grid.save_to(filename).expect("This will only work on linux systems");
let mut file = fs::OpenOptions::new().read(true).open(filename).expect("Just wrote the file"); let mut file = fs::OpenOptions::new().read(true).open(filename).expect("Just wrote the file");
let mut buf = String::new(); let mut buf = String::new();
file.read_to_string(&mut buf).expect("Just opened the file"); file.read_to_string(&mut buf).expect("Just opened the file");
@@ -615,7 +580,6 @@ fn alphanumeric_indexing() {
assert_eq!(Grid::parse_to_idx("A"), None); assert_eq!(Grid::parse_to_idx("A"), None);
assert_eq!(Grid::parse_to_idx(":"), None); assert_eq!(Grid::parse_to_idx(":"), None);
assert_eq!(Grid::parse_to_idx("="), None); assert_eq!(Grid::parse_to_idx("="), None);
assert_eq!(Grid::parse_to_idx("A:A"), None);
assert_eq!(Grid::num_to_char(0).trim(), "A"); assert_eq!(Grid::num_to_char(0).trim(), "A");
assert_eq!(Grid::num_to_char(25).trim(), "Z"); assert_eq!(Grid::num_to_char(25).trim(), "Z");
@@ -636,31 +600,31 @@ fn valid_equations() {
// cell math // cell math
let cell = grid.get_cell("C0").as_ref().expect("Just set it"); let cell = grid.get_cell("C0").as_ref().expect("Just set it");
let res = grid.evaluate(&cell.to_string()).expect("Should evaluate."); let res = grid.evaluate(&cell.to_string()).expect("Should evaluate.");
assert_eq!(res, (3.).into()); assert_eq!(res, 3.);
// divide floats // divide floats
grid.set_cell("D0", "=5./2.".to_string()); grid.set_cell("D0", "=5./2.".to_string());
let cell = grid.get_cell("D0").as_ref().expect("I just set this"); let cell = grid.get_cell("D0").as_ref().expect("I just set this");
let res = grid.evaluate(&cell.to_string()).expect("Should be ok"); let res = grid.evaluate(&cell.to_string()).expect("Should be ok");
assert_eq!(res, (2.5).into()); assert_eq!(res, 2.5);
// Float / Int mix // Float / Int mix
grid.set_cell("D0", "=5./2".to_string()); grid.set_cell("D0", "=5./2".to_string());
let cell = grid.get_cell("D0").as_ref().expect("I just set this"); let cell = grid.get_cell("D0").as_ref().expect("I just set this");
let res = grid.evaluate(&cell.to_string()).expect("Should be ok"); let res = grid.evaluate(&cell.to_string()).expect("Should be ok");
assert_eq!(res, (2.5).into()); assert_eq!(res, 2.5);
// divide "ints" (should become floats) // divide "ints" (should become floats)
grid.set_cell("D0", "=5/2".to_string()); grid.set_cell("D0", "=5/2".to_string());
let cell = grid.get_cell("D0").as_ref().expect("I just set this"); let cell = grid.get_cell("D0").as_ref().expect("I just set this");
let res = grid.evaluate(&cell.to_string()).expect("Should be ok"); let res = grid.evaluate(&cell.to_string()).expect("Should be ok");
assert_eq!(res, (2.5).into()); assert_eq!(res, 2.5);
// Non-equation that should still be valid // Non-equation that should still be valid
grid.set_cell("D0", "=10".to_string()); grid.set_cell("D0", "=10".to_string());
let cell = grid.get_cell("D0").as_ref().expect("I just set this"); let cell = grid.get_cell("D0").as_ref().expect("I just set this");
let res = grid.evaluate(&cell.to_string()).expect("Should be ok"); let res = grid.evaluate(&cell.to_string()).expect("Should be ok");
assert_eq!(res, (10.).into()); assert_eq!(res, 10.);
} }
// Cell = output of Cell = value of Cells. // Cell = output of Cell = value of Cells.
@@ -677,7 +641,7 @@ fn fn_of_fn() {
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok()); assert!(res.is_ok());
assert_eq!(res.unwrap(), (6.).into()); assert_eq!(res.unwrap(), 6.);
return; return;
} }
panic!("Cell not found"); panic!("Cell not found");
@@ -713,7 +677,7 @@ fn invalid_equations() {
let cell = grid.get_cell("B0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("B0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok()); assert!(res.is_ok());
assert!(res.is_ok_and(|v| v == (10.).into())); assert!(res.is_ok_and(|v| v == 10.));
// Trailing comma in function call // Trailing comma in function call
grid.set_cell("A0", 5.); grid.set_cell("A0", 5.);
@@ -721,7 +685,7 @@ fn invalid_equations() {
grid.set_cell("B0", "=avg(A0,A1,)".to_string()); grid.set_cell("B0", "=avg(A0,A1,)".to_string());
let cell = grid.get_cell("B0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("B0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert_eq!(res.unwrap(), (7.5).into()); assert_eq!(res.unwrap(), 7.5);
} }
#[test] #[test]
@@ -752,19 +716,19 @@ fn avg_function() {
let cell = grid.get_cell("A0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("A0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok()); assert!(res.is_ok());
assert_eq!(res.unwrap(), (5.).into()); assert_eq!(res.unwrap(), 5.);
grid.set_cell("A0", "=avg(5,10)".to_string()); grid.set_cell("A0", "=avg(5,10)".to_string());
let cell = grid.get_cell("A0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("A0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok()); assert!(res.is_ok());
assert_eq!(res.unwrap(), (7.5).into()); assert_eq!(res.unwrap(), 7.5);
grid.set_cell("A0", "=avg(5,10,15)".to_string()); grid.set_cell("A0", "=avg(5,10,15)".to_string());
let cell = grid.get_cell("A0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("A0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok()); assert!(res.is_ok());
assert_eq!(res.unwrap(), (10.).into()); assert_eq!(res.unwrap(), 10.);
grid.set_cell("A0", "=avg(foo)".to_string()); grid.set_cell("A0", "=avg(foo)".to_string());
let cell = grid.get_cell("A0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("A0").as_ref().expect("Just set the cell");
@@ -790,13 +754,13 @@ fn sum_function() {
let cell = grid.get_cell("A0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("A0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok()); assert!(res.is_ok());
assert_eq!(res.unwrap(), (5.).into()); assert_eq!(res.unwrap(), 5.);
grid.set_cell("A0", "=sum(5,10)".to_string()); grid.set_cell("A0", "=sum(5,10)".to_string());
let cell = grid.get_cell("A0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("A0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok()); assert!(res.is_ok());
assert_eq!(res.unwrap(), (15.).into()); assert_eq!(res.unwrap(), 15.);
grid.set_cell("A0", "=sum(foo)".to_string()); grid.set_cell("A0", "=sum(foo)".to_string());
let cell = grid.get_cell("A0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("A0").as_ref().expect("Just set the cell");
@@ -822,11 +786,11 @@ fn xlookup_function() {
grid.set_cell("A1", "Sarah".to_string()); grid.set_cell("A1", "Sarah".to_string());
grid.set_cell("C0", 31.); grid.set_cell("C0", 31.);
grid.set_cell("C1", 41.); grid.set_cell("C1", 41.);
grid.set_cell("B0", "=xlookup(\"Bobby\",A:A,C:C)".to_string()); grid.set_cell("B0", "=xlookup(A:A,\"Bobby\",C:C)".to_string());
let cell = grid.get_cell("B0").as_ref().expect("Just set the cell"); let cell = grid.get_cell("B0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string()); let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok()); assert!(res.is_ok());
assert_eq!(res.unwrap(), (31.).into()); assert_eq!(res.unwrap(), 31.);
} }
#[test] #[test]
@@ -880,19 +844,6 @@ fn parse_csv() {
); );
} }
#[test]
fn invalid_ranges() {
let mut grid = Grid::new();
grid.set_cell("A0", 2.);
grid.set_cell("A1", 1.);
// ASCII to number conversion needs to not overflow
grid.set_cell("B0", "=sum($:A)".to_string());
let cell = grid.get_cell("B0").as_ref().expect("Just set it");
let _ = grid.evaluate(&cell.to_string()).expect("Should evaluate.");
}
#[test] #[test]
fn ranges() { fn ranges() {
let mut grid = Grid::new(); let mut grid = Grid::new();
@@ -904,13 +855,13 @@ fn ranges() {
// range with numbers // range with numbers
let cell = grid.get_cell("B0").as_ref().expect("Just set it"); let cell = grid.get_cell("B0").as_ref().expect("Just set it");
let res = grid.evaluate(&cell.to_string()).expect("Should evaluate."); let res = grid.evaluate(&cell.to_string()).expect("Should evaluate.");
assert_eq!(res, (3.).into()); assert_eq!(res, 3.);
// use range output as input for other function // use range output as input for other function
grid.set_cell("B1", "=B0*2".to_string()); grid.set_cell("B1", "=B0*2".to_string());
let cell = grid.get_cell("B1").as_ref().expect("Just set it"); let cell = grid.get_cell("B1").as_ref().expect("Just set it");
let res = grid.evaluate(&cell.to_string()).expect("Should evaluate."); let res = grid.evaluate(&cell.to_string()).expect("Should evaluate.");
assert_eq!(res, (6.).into()); assert_eq!(res, 6.);
// use equation outputs as range input // use equation outputs as range input
grid.set_cell("A2", "=C0+1".to_string()); grid.set_cell("A2", "=C0+1".to_string());
@@ -918,11 +869,11 @@ fn ranges() {
let cell = grid.get_cell("A2").as_ref().expect("Just set it"); let cell = grid.get_cell("A2").as_ref().expect("Just set it");
let res = grid.evaluate(&cell.to_string()).expect("Should evaluate."); let res = grid.evaluate(&cell.to_string()).expect("Should evaluate.");
assert_eq!(res, (6.).into()); assert_eq!(res, 6.);
let cell = grid.get_cell("B0").as_ref().expect("Just set it"); let cell = grid.get_cell("B0").as_ref().expect("Just set it");
let res = grid.evaluate(&cell.to_string()).expect("Should evaluate."); let res = grid.evaluate(&cell.to_string()).expect("Should evaluate.");
assert_eq!(res, (9.).into()); assert_eq!(res, 9.);
// use function outputs as range input // use function outputs as range input
grid.set_cell("B1", 2.); grid.set_cell("B1", 2.);
@@ -931,7 +882,7 @@ fn ranges() {
let cell = grid.get_cell("C0").as_ref().expect("Just set it"); let cell = grid.get_cell("C0").as_ref().expect("Just set it");
let res = grid.evaluate(&cell.to_string()).expect("Should evaluate."); let res = grid.evaluate(&cell.to_string()).expect("Should evaluate.");
assert_eq!(res, (5.).into()); assert_eq!(res, 5.);
// use range outputs as range input // use range outputs as range input
grid.set_cell("D0", "=sum(C:C)".to_string()); grid.set_cell("D0", "=sum(C:C)".to_string());
@@ -939,7 +890,7 @@ fn ranges() {
let cell = grid.get_cell("D0").as_ref().expect("Just set it"); let cell = grid.get_cell("D0").as_ref().expect("Just set it");
let res = grid.evaluate(&cell.to_string()).expect("Should evaluate."); let res = grid.evaluate(&cell.to_string()).expect("Should evaluate.");
assert_eq!(res, (6.).into()); assert_eq!(res, 6.);
} }
#[test] #[test]
@@ -1024,36 +975,6 @@ fn insert_col_before_3() {
assert_eq!(cell.to_string(), "=B0*B1"); assert_eq!(cell.to_string(), "=B0*B1");
} }
#[test]
fn insert_col_before_static_range() {
let mut grid = Grid::new();
grid.set_cell("A0", 2.);
grid.set_cell("A1", 2.);
grid.set_cell("B0", "=sum(A:A)".to_string());
grid.mv_cursor_to(0, 1);
grid.insert_column_before(grid.cursor());
let cell = grid.get_cell("C0").as_ref().expect("Just set it");
assert_eq!(cell.to_string(), "=sum(A:A)");
}
#[test]
fn insert_col_before_move_range() {
let mut grid = Grid::new();
grid.set_cell("B0", 2.);
grid.set_cell("B1", 2.);
grid.set_cell("A0", "=sum(B:B)".to_string());
grid.mv_cursor_to(0, 0);
grid.insert_column_after(grid.cursor());
let cell = grid.get_cell("A0").as_ref().expect("Just set it");
assert_eq!(cell.to_string(), "=sum(C:C)");
}
#[test] #[test]
fn insert_row_above_1() { fn insert_row_above_1() {
let mut grid = Grid::new(); let mut grid = Grid::new();
@@ -1139,35 +1060,6 @@ fn cell_eval_depth() {
assert_eq!(c.to_string(), "=A5+$A$0"); assert_eq!(c.to_string(), "=A5+$A$0");
let res = app.grid.evaluate(&c.to_string()).expect("Should evaluate"); let res = app.grid.evaluate(&c.to_string()).expect("Should evaluate");
assert_eq!(res, (7.).into()); assert_eq!(res, 7.);
}
#[test]
fn return_string_from_fn() {
let mut grid = Grid::new();
grid.set_cell("A0", "=if(2>1, \"A\", \"B\")".to_string()); // true, A
grid.set_cell("A1", "=if(1>2, \"A\", \"B\")".to_string()); // false, B
let cell = grid.get_cell("A0").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok());
if let Ok(f) = res {
if let CellType::String(s) = f {
assert_eq!("A", s);
} else {
unreachable!();
}
}
let cell = grid.get_cell("A1").as_ref().expect("Just set the cell");
let res = grid.evaluate(&cell.to_string());
assert!(res.is_ok());
if let Ok(f) = res {
if let CellType::String(s) = f {
assert_eq!("B", s);
} else {
unreachable!();
}
}
} }

View File

@@ -54,46 +54,42 @@ impl CellType {
} }
} }
/// `replace_fn` takes the string, the old value, and then the new value.
/// It can be thought of as `echo $1 | sed s/$2/$3/g`
pub fn custom_translate_cell(&self, from: (usize, usize), to: (usize, usize), replace_fn: impl Fn(&str, &str, &str) -> String) -> CellType { pub fn custom_translate_cell(&self, from: (usize, usize), to: (usize, usize), replace_fn: impl Fn(&str, &str, &str) -> String) -> CellType {
match self { match self {
// don't translate non-equations // don't translate non-equations
CellType::Number(_) | CellType::String(_) => return self.clone(), CellType::Number(_) | CellType::String(_) => return self.clone(),
CellType::Equation(eq) => { CellType::Equation(eq) => {
// Populate the context
let ctx = ExtractionContext::new();
let _ = eval_with_context(&eq[1..], &ctx);
let mut equation = eq.clone();
// translate standard vars A0 -> A1
// extract all the variables // extract all the variables
let ctx = ExtractionContext::new();
let _ = eval_with_context(eq, &ctx);
let mut rolling = eq.clone();
// translate standard vars A0 -> A1
for old_var in ctx.dump_vars() { for old_var in ctx.dump_vars() {
let mut lock_x = false; let mut lock_x = false;
let mut lock_y = false; let mut lock_y = false;
if old_var.contains('$') { if old_var.contains('$') {
let locations = old_var let locations = old_var.char_indices().filter(|(_, c)| *c == '$').map(|(i, _)| i).collect::<Vec<usize>>();
.char_indices()
.filter(|(_, c)| *c == '$').map(|(i, _)| i)
.collect::<Vec<usize>>();
match locations.len() { match locations.len() {
1 => { 1 => {
if locations[0] == 0 { if locations[0] == 0 {
// locking the X axis (A,B,C...) // locking the X axis (A,B,C...)
lock_x = true; lock_x = true;
} else { } else if locations[0] < old_var.len() {
// inside the string somewhere, gonna assume this means to lock Y (1,2,3...) // inside the string somewhere, gonna assume this means to lock Y (1,2,3...)
lock_y = true; lock_y = true;
} else {
// where tf is this dollar sign?
// (It's somewhere malformed, like A0$ or something)
} }
} }
2 => { 2 => {
// Ignore this variable all together, effectively lockng X & Y // YOLO, lock both X & Y
continue; continue; // just pretend you never even saw this var
} }
_ => { _ => {
// There are 0 or >2 "$" in this string.
//
// Could probably optimize the code or something so you only go over the string // Could probably optimize the code or something so you only go over the string
// once, instead of contains() then getting the indexes of where it is. // once, instead of contains() then getting the indexes of where it is.
// You could then put your no-$ code here. // You could then put your no-$ code here.
@@ -102,7 +98,6 @@ impl CellType {
} }
if let Some((src_x, src_y)) = Grid::parse_to_idx(&old_var) { if let Some((src_x, src_y)) = Grid::parse_to_idx(&old_var) {
// Use i32s instead of usize in case of negative numbers
let (x1, y1) = from; let (x1, y1) = from;
let x1 = x1 as i32; let x1 = x1 as i32;
let y1 = y1 as i32; let y1 = y1 as i32;
@@ -125,7 +120,6 @@ impl CellType {
let alpha = Grid::num_to_char(dest_x); let alpha = Grid::num_to_char(dest_x);
let alpha = alpha.trim(); let alpha = alpha.trim();
// Persist the "$" locking
let new_var = if lock_x { let new_var = if lock_x {
format!("${alpha}{dest_y}") format!("${alpha}{dest_y}")
} else if lock_y { } else if lock_y {
@@ -134,48 +128,15 @@ impl CellType {
format!("{alpha}{dest_y}") format!("{alpha}{dest_y}")
}; };
// swap out vars // swap out vars
equation = replace_fn(&equation, &old_var, &new_var); rolling = replace_fn(&rolling, &old_var, &new_var);
// rolling = rolling.replace(&old_var, &new_var); // rolling = rolling.replace(&old_var, &new_var);
} else { } else {
// why you coping invalid stuff, nerd? // why you coping invalid stuff, nerd?
//
// could be copying a range
if old_var.contains(':') {
let parts = old_var.split(':').collect::<Vec<&str>>();
// This means the var was formatted as X:X
if parts.len() == 2 {
// how far is the movement?
let dx = to.0 as i32 - from.0 as i32;
let range_start = parts[0];
let range_end = parts[1];
// get the letters as numbers
let xs = Grid::char_to_idx(range_start) as i32;
let xe = Grid::char_to_idx(range_end) as i32;
// apply movement
let mut new_range_start = xs+dx;
let mut new_range_end = xe+dx;
// bottom out at 0
if new_range_start < 0 {
new_range_start = 0;
}
if new_range_end < 0 {
new_range_end = 0;
}
// convert the index back into a letter and then submit it
let start = Grid::num_to_char(new_range_start as usize);
let end = Grid::num_to_char(new_range_end as usize);
equation = replace_fn(&equation, &old_var, &format!("{}:{}", start.trim(), end.trim()));
} }
} }
} return rolling.into();
}
return equation.into();
} }
} }
} }
@@ -195,15 +156,3 @@ impl Display for CellType {
write!(f, "{d}") write!(f, "{d}")
} }
} }
impl PartialEq for CellType {
fn eq(&self, other: &Self) -> bool {
match (self, other) {
(Self::Number(left), Self::Number(right)) => left == right,
(Self::String(left), Self::String(right)) => left == right,
(Self::Equation(left), Self::Equation(right)) => left == right,
_ => false,
}
}
}

View File

@@ -16,10 +16,24 @@ pub struct CallbackContext<'a> {
impl<'a> CallbackContext<'a> { impl<'a> CallbackContext<'a> {
fn expand_range(&self, range: &str) -> Option<Vec<&CellType>> { fn expand_range(&self, range: &str) -> Option<Vec<&CellType>> {
if let Some((start, end)) = Grid::range_as_indices(range) { let v = range.split(':').collect::<Vec<&str>>();
if v.len() == 2 {
let start_col = v[0];
let end_col = v[1];
let as_index = |s: &str| {
s.char_indices()
// .filter(|f| f.1 as u8 >= 97) // prevent sub with overflow errors
.map(|(idx, c)| (c.to_ascii_lowercase() as usize - 97) + (26 * idx))
.fold(0, |a, b| a + b)
};
let start_idx = as_index(start_col);
let end_idx = as_index(end_col);
let mut buf = Vec::new(); let mut buf = Vec::new();
for x in start..=end { for x in start_idx..=end_idx {
for y in 0..=self.variables.max_y_at_x(x) { for y in 0..=self.variables.max_y_at_x(x) {
if let Some(s) = self.variables.get_cell_raw(x, y) { if let Some(s) = self.variables.get_cell_raw(x, y) {
buf.push(s); buf.push(s);
@@ -90,13 +104,13 @@ impl<'a> CallbackContext<'a> {
functions.insert( functions.insert(
"xlookup".to_string(), "xlookup".to_string(),
Function::new(|arg| { Function::new(|arg| {
let expected = vec![ValueType::String, ValueType::Tuple, ValueType::Tuple]; let expected = vec![ValueType::Tuple, ValueType::String, ValueType::Tuple];
if arg.is_tuple() { if arg.is_tuple() {
let args = arg.as_tuple()?; let args = arg.as_tuple()?;
if args.len() == 3 { if args.len() == 3 {
let lookup_value = &args[0]; let lookup_array = &args[0];
let lookup_array = &args[1]; let lookup_value = &args[1];
let return_array = &args[2]; let return_array = &args[2];
if lookup_array.is_tuple() && return_array.is_tuple() { if lookup_array.is_tuple() && return_array.is_tuple() {
@@ -148,8 +162,8 @@ impl<'a> Context for CallbackContext<'a> {
} }
if let Ok(mut trail) = self.eval_breadcrumbs.write() { if let Ok(mut trail) = self.eval_breadcrumbs.write() {
let find = trail.iter().filter(|id| *id == identifier).count(); let find = trail.iter().filter(|id| *id == identifier).collect::<Vec<&String>>();
if find > 0 { if find.len() > 0 {
// recursion detected // recursion detected
return None; return None;
} else { } else {
@@ -181,7 +195,7 @@ impl<'a> Context for CallbackContext<'a> {
}, },
Err(e) => { Err(e) => {
match e { match e {
EvalexprError::VariableIdentifierNotFound(_err) => { EvalexprError::VariableIdentifierNotFound(_) => {
// If the variable isn't found, that's ~~probably~~ because // If the variable isn't found, that's ~~probably~~ because
// of recursive reference, considering all references // of recursive reference, considering all references
// are grabbed straight from the table. // are grabbed straight from the table.
@@ -203,14 +217,8 @@ impl<'a> Context for CallbackContext<'a> {
CellType::Number(e) => vals.push(Value::Float(*e)), CellType::Number(e) => vals.push(Value::Float(*e)),
CellType::String(s) => vals.push(Value::String(s.to_owned())), CellType::String(s) => vals.push(Value::String(s.to_owned())),
CellType::Equation(eq) => { CellType::Equation(eq) => {
match eval_with_context(&eq[1..], self) { if let Ok(val) = eval_with_context(&eq[1..], self) {
Ok(val) => vals.push(val), vals.push(val);
Err(_err) => {
// At this point we are getting an error because
// recursion protection made this equation return
// None. We now don't get any evaluation.
return None
},
} }
} }
} }
@@ -262,19 +270,10 @@ impl ExtractionContext {
} }
} }
pub fn dump_vars(&self) -> Vec<String> { pub fn dump_vars(&self) -> Vec<String> {
if let Ok(r) = self.var_registry.read() { if let Ok(r) = self.var_registry.read() { r.clone() } else { Vec::new() }
r.clone()
} else {
Vec::new()
} }
}
#[allow(dead_code)]
pub fn dump_fns(&self) -> Vec<String> { pub fn dump_fns(&self) -> Vec<String> {
if let Ok(r) = self.fn_registry.read() { if let Ok(r) = self.fn_registry.read() { r.clone() } else { Vec::new() }
r.clone()
} else {
Vec::new()
}
} }
} }
@@ -302,7 +301,8 @@ impl Context for ExtractionContext {
} else { } else {
panic!("The RwLock should always be write-able") panic!("The RwLock should always be write-able")
} }
Ok(Value::Int(1)) // Ok(Value::Int(1))
unimplemented!("Extracting function identifier not implemented yet")
} }
fn are_builtin_functions_disabled(&self) -> bool { fn are_builtin_functions_disabled(&self) -> bool {

View File

@@ -4,3 +4,4 @@ mod error_msg;
mod screen; mod screen;
mod logic; mod logic;
mod clipboard; mod clipboard;
mod plot;

View File

@@ -15,9 +15,9 @@ use crate::app::{
app::App, app::App,
error_msg::StatusMessage, error_msg::StatusMessage,
logic::{ logic::{
calc::{Grid, LEN}, calc::{CSV_EXT, CUSTOM_EXT, Grid, LEN},
cell::CellType, cell::CellType,
}, }, plot::Plot,
}; };
pub enum Mode { pub enum Mode {
@@ -70,11 +70,25 @@ impl Mode {
"w" => { "w" => {
// first try the passed argument as file // first try the passed argument as file
if let Some(arg) = args.get(1) { if let Some(arg) = args.get(1) {
let path: PathBuf = arg.into(); let mut path: PathBuf = arg.into();
match path.extension() {
Some(s) => {
match s.to_str() {
// leave the file alone, it already has
// a valid extension
Some(CSV_EXT) | Some(CUSTOM_EXT) => {}
_ => {
path.add_extension(CUSTOM_EXT);
}
}
}
None => {
path.add_extension(CUSTOM_EXT);
}
};
// TODO Check if the file we are writing to exists, since // TODO Check if the file exists, but the program wasn't opened with it. We might be accidentally overwriting something else.
// this code path already knows that we are writing to a new file. // let mut file = fs::OpenOptions::new().write(true).append(false).truncate(true).create(true).open(path)?;
// We might be accidentally overwriting something else.
if let Err(e) = app.grid.save_to(&path) { if let Err(e) = app.grid.save_to(&path) {
app.msg = StatusMessage::error(format!("{e}")); app.msg = StatusMessage::error(format!("{e}"));
@@ -175,21 +189,6 @@ impl Mode {
}; };
match args[0] { match args[0] {
"f" | "fill" => {
for (i, x) in (low_x..=hi_x).enumerate() {
for (j, y) in (low_y..=hi_y).enumerate() {
let arg = args.get(1)
.map(|s| s.replace("xi", &i.to_string()))
.map(|s| s.replace("yi", &j.to_string()))
.map(|s| s.replace("x", &x.to_string()))
.map(|s| s.replace("y", &y.to_string()))
;
app.grid.set_cell_raw((x,y), arg);
}
}
app.mode = Mode::Normal
}
"export" => { "export" => {
if let Some(arg1) = args.get(1) { if let Some(arg1) = args.get(1) {
save_range(&arg1); save_range(&arg1);
@@ -199,6 +198,8 @@ impl Mode {
app.mode = Mode::Normal app.mode = Mode::Normal
} }
"plot" => { "plot" => {
app.plot_popup = Some(Plot::new(0, 1));
return;
// Use gnuplot to plot the selected data. // Use gnuplot to plot the selected data.
// * Temp data will be stored in /tmp/ // * Temp data will be stored in /tmp/
// * Output will either be plot.png or a name that you pass in // * Output will either be plot.png or a name that you pass in
@@ -217,16 +218,10 @@ impl Mode {
let s = s.replace("$OUTPUT", "/tmp/output.png"); let s = s.replace("$OUTPUT", "/tmp/output.png");
let _ = fs::write("/tmp/plot.p", s); let _ = fs::write("/tmp/plot.p", s);
let cmd_res= Command::new("gnuplot").arg("/tmp/plot.p").output(); let _ = Command::new("gnuplot").arg("/tmp/plot.p").output();
if let Err(err) = cmd_res {
match err.kind() {
std::io::ErrorKind::NotFound => app.msg = StatusMessage::error("Error - Is gnuplot installed?"),
_ => app.msg = StatusMessage::error(format!("{err}")),
};
} else {
let _ = fs::copy("/tmp/output.png", output_filename); let _ = fs::copy("/tmp/output.png", output_filename);
app.msg = StatusMessage::info(format!("Created {output_filename}. Artifacts are in /tmp"));
} app.msg = StatusMessage::info("Wrote gnuplot data to /tmp");
app.mode = Mode::Normal app.mode = Mode::Normal
} }
_ => {} _ => {}
@@ -267,18 +262,6 @@ impl Mode {
app.grid.mv_cursor_to(0, y); app.grid.mv_cursor_to(0, y);
return; return;
} }
// Go to end of row
'$' => {
let (_, y) = app.grid.cursor();
app.grid.mv_cursor_to(super::logic::calc::LEN, y);
return;
}
// Go to bottom of column
'G' => {
let (x, _) = app.grid.cursor();
app.grid.mv_cursor_to(x, super::logic::calc::LEN,);
return;
}
// edit cell // edit cell
'i' | 'a' => { 'i' | 'a' => {
let (x, y) = app.grid.cursor(); let (x, y) = app.grid.cursor();
@@ -392,34 +375,6 @@ impl Mode {
app.grid.mv_cursor_to(x, 0); app.grid.mv_cursor_to(x, 0);
app.mode = Mode::Normal; app.mode = Mode::Normal;
} }
// Go to the bottom of the current window
("g", 'G') => {
let (x, _) = app.grid.cursor();
let (_, y_height) = app.screen.get_screen_size(&app.vars);
let y_origin = app.screen.scroll_y();
app.grid.mv_cursor_to(x, y_origin+y_height);
app.mode = Mode::Normal;
return;
}
// Go to the right edge of the current window
("g", '$') => {
let (_, y) = app.grid.cursor();
let (x_width, _) = app.screen.get_screen_size(&app.vars);
let x_origin = app.screen.scroll_x();
app.grid.mv_cursor_to(x_origin+x_width, y);
app.mode = Mode::Normal;
}
// Go to the left edge of the current window
("g", '0') => {
let (_, y) = app.grid.cursor();
let x_origin = app.screen.scroll_x();
app.grid.mv_cursor_to(x_origin, y);
app.mode = Mode::Normal;
return;
}
// center screen to cursor // center screen to cursor
("z", 'z') => { ("z", 'z') => {
app.screen.center_x(app.grid.cursor(), &app.vars); app.screen.center_x(app.grid.cursor(), &app.vars);

45
src/app/plot.rs Normal file
View File

@@ -0,0 +1,45 @@
use ratatui::{layout::{Constraint, Direction, Layout}, widgets::{Paragraph, Widget}};
pub struct Plot {
x: usize,
y: Vec<usize>,
}
impl Plot {
pub fn new(x: usize, y: usize) -> Self {
Self {
x,
y: vec![y],
}
}
pub fn add_column(&mut self) {
self.y.push(1)
}
pub fn del_column(&mut self) {
self.y.pop();
}
pub fn process_key(&mut self, c: char) {
}
}
impl Widget for &Plot {
fn render(self, area: ratatui::prelude::Rect, buf: &mut ratatui::prelude::Buffer) {
// plus 1 for x
let columns = self.y.len() + 1;
let mut constraints = Vec::new();
for _ in 0..=columns {
constraints.push(Constraint::Min(1));
}
Paragraph::new("Foobar").render(area, buf);
let layout = Layout::default()
.direction(Direction::Vertical)
.constraints(constraints)
.split(area);
let x_space = layout[0];
}
}

View File

@@ -5,12 +5,9 @@ use ratatui::prelude;
use crate::app::logic::calc::LEN; use crate::app::logic::calc::LEN;
pub struct ScreenSpace { pub struct ScreenSpace {
/// This is measured in cells. /// This is measured in cells
/// This is the top-left cell index.
scroll: (usize, usize), scroll: (usize, usize),
/// In chars
default_cell_len: usize, default_cell_len: usize,
/// In chars
default_cell_hight: usize, default_cell_hight: usize,
/// This is measured in chars /// This is measured in chars
last_seen_screen_size: RwLock<(usize, usize)> last_seen_screen_size: RwLock<(usize, usize)>
@@ -20,49 +17,37 @@ impl ScreenSpace {
pub fn new() -> Self { pub fn new() -> Self {
Self { Self {
scroll: (0, 0), scroll: (0, 0),
default_cell_len: 9, default_cell_len: 10,
default_cell_hight: 1, default_cell_hight: 1,
last_seen_screen_size: RwLock::new((0,0)) last_seen_screen_size: RwLock::new((0,0))
} }
} }
pub fn center_x(&mut self, (cursor_x, _): (usize, usize), vars: &HashMap<String, String>) { pub fn center_x(&mut self, (cursor_x, _): (usize, usize), vars: &HashMap<String, String>) {
let (x_cells, _) = self.get_screen_size(vars); if let Ok(screen_size) = self.last_seen_screen_size.read() {
let x_cells = (screen_size.0 / self.get_cell_width(vars) as usize) -2;
let x_center = self.scroll_x() + (x_cells/2); let x_center = self.scroll_x() + (x_cells/2);
let delta = cursor_x as isize - x_center as isize; let delta = cursor_x as isize - x_center as isize;
self.scroll.0 = self.scroll.0.saturating_add_signed(delta); self.scroll.0 = self.scroll.0.saturating_add_signed(delta);
} }
}
pub fn center_y(&mut self, (_, cursor_y): (usize, usize), vars: &HashMap<String, String>) { pub fn center_y(&mut self, (_, cursor_y): (usize, usize), vars: &HashMap<String, String>) {
let (_, y_cells) = self.get_screen_size(vars); if let Ok(screen_size) = self.last_seen_screen_size.read() {
let y_cells = (screen_size.1 / self.get_cell_height(vars) as usize) -2;
let y_center = self.scroll_y() + (y_cells/2); let y_center = self.scroll_y() + (y_cells/2);
let delta = cursor_y as isize - y_center as isize; let delta = cursor_y as isize - y_center as isize;
self.scroll.1 = self.scroll.1.saturating_add_signed(delta); self.scroll.1 = self.scroll.1.saturating_add_signed(delta);
} }
}
/// In chars pub fn scroll_based_on_cursor_location(&mut self, (cursor_x, cursor_y): (usize, usize), vars: &HashMap<String, String>) {
pub fn get_screen_size(&self, vars: &HashMap<String, String>) -> (usize, usize) {
if let Ok(screen_size) = self.last_seen_screen_size.read() { if let Ok(screen_size) = self.last_seen_screen_size.read() {
// ======= X ======= // ======= X =======
// screen seems to be 2 cells smaller than it should be // screen seems to be 2 cells smaller than it should be
// this is probably related to issue #6 // this is probably related to issue #6
let x_cells = (screen_size.0 / self.get_cell_width(vars) as usize) -2; let x_cells = (screen_size.0 / self.get_cell_width(vars) as usize) -2;
// ======= Y =======
// screen seems to be 2 cells smaller than it should be
// this is probably related to issue #6
let y_cells = (screen_size.1 / self.get_cell_height(vars) as usize) -2;
(x_cells,y_cells)
} else {
(0,0)
}
}
pub fn scroll_based_on_cursor_location(&mut self, (cursor_x, cursor_y): (usize, usize), vars: &HashMap<String, String>) {
let (x_cells, y_cells) = self.get_screen_size(vars);
let lower_x = self.scroll_x(); let lower_x = self.scroll_x();
let upper_x = self.scroll_x() + x_cells; let upper_x = self.scroll_x() + x_cells;
@@ -75,7 +60,10 @@ impl ScreenSpace {
self.scroll.0 = self.scroll.0.saturating_add(delta); self.scroll.0 = self.scroll.0.saturating_add(delta);
} }
// ======= Y =======
// screen seems to be 2 cells smaller than it should be
// this is probably related to issue #6
let y_cells = (screen_size.1 / self.get_cell_height(vars) as usize) -2;
let lower_y = self.scroll_y(); let lower_y = self.scroll_y();
let upper_y = self.scroll_y() + y_cells; let upper_y = self.scroll_y() + y_cells;
@@ -89,15 +77,14 @@ impl ScreenSpace {
self.scroll.1 = self.scroll.1.saturating_add(delta); self.scroll.1 = self.scroll.1.saturating_add(delta);
} }
} }
}
pub fn scroll_x(&self) -> usize { pub fn scroll_x(&self) -> usize {
self.scroll.0 self.scroll.0
} }
pub fn scroll_y(&self) -> usize{ pub fn scroll_y(&self) -> usize{
self.scroll.1 self.scroll.1
} }
pub fn get_cell_height(&self, vars: &HashMap<String, String>) -> usize { pub fn get_cell_height(&self, vars: &HashMap<String, String>) -> usize {
if let Some(h) = vars.get("height") { if let Some(h) = vars.get("height") {
if let Ok(p) = h.parse::<usize>() { if let Ok(p) = h.parse::<usize>() {
@@ -106,7 +93,6 @@ impl ScreenSpace {
} }
return self.default_cell_hight return self.default_cell_hight
} }
pub fn get_cell_width(&self, vars: &HashMap<String, String>) -> usize { pub fn get_cell_width(&self, vars: &HashMap<String, String>) -> usize {
if let Some(h) = vars.get("length") { if let Some(h) = vars.get("length") {
if let Ok(p) = h.parse::<usize>() { if let Ok(p) = h.parse::<usize>() {
@@ -114,34 +100,19 @@ impl ScreenSpace {
} }
} }
self.default_cell_len self.default_cell_len
} }
pub fn how_many_cells_fit_in(&self, area: &prelude::Rect, vars: &HashMap<String, String>) -> (u16, u16) { pub fn how_many_cells_fit_in(&self, area: &prelude::Rect, vars: &HashMap<String, String>) -> (u16, u16) {
if let Ok(mut l) = self.last_seen_screen_size.write() { if let Ok(mut l) = self.last_seen_screen_size.write() {
l.0 = area.width as usize; l.0 = area.width as usize;
l.1 = area.height as usize; l.1 = area.height as usize;
} }
// let width = (area.width as usize + calc::get_header_size() -1) / self.get_cell_width(vars);
let width = area.width as usize / self.get_cell_width(vars);
let height = area.height as usize / self.get_cell_height(vars);
let x_max = let x_max =
if width > LEN { if area.width as usize / self.get_cell_width(vars) > LEN { LEN - 1 } else { area.width as usize / self.get_cell_width(vars)};
LEN - 1
} else {
width
};
let y_max = let y_max =
if height > LEN { if area.height as usize / self.get_cell_height(vars) > LEN { LEN - 1 } else { area.height as usize / self.get_cell_height(vars)};
LEN - 1
} else {
height
};
(x_max as u16, y_max as u16) (x_max as u16, y_max as u16)
} }
} }
#[test] #[test]

View File

@@ -14,5 +14,4 @@ set grid
set term png size 1280, 720 set term png size 1280, 720
set output '$OUTPUT' set output '$OUTPUT'
plot datafile using 1:2 with linespoints linestyle 1 plot datafile using 1:2 with linespoints linestyle 1
# plot datafile using 1:2 with linespoints linestyle 1, datafile using 1:3, etc...
replot replot