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
|
use std::sync::{Arc, RwLock};
use mlua::{Table, UserData, Value};
use super::{cell::Cell, Sheet, SheetLuaRef};
static REGISTER: RwLock<Vec<Arc<RwLock<Sheet>>>> = RwLock::new(Vec::new());
pub type SheetId = usize;
#[derive(Debug)]
pub struct Register;
impl Register {
pub fn create(width: usize, height: usize) -> SheetId {
let mut register = REGISTER.write().unwrap();
let id = register.len();
let sheet = Sheet::new(width, height, id);
register.push(Arc::new(RwLock::new(sheet)));
id
}
pub fn create_from_table(table: Table) -> mlua::Result<SheetId> {
let mut register = REGISTER.write().unwrap();
let id = register.len();
let height = table.len()? as usize;
let mut width: usize = 0;
for column in table.clone().pairs::<usize, Table>() {
width = width.max(column?.1.len()? as usize)
}
let mut sheet = Sheet::new(width, height, id);
for row_res in table.pairs::<usize, Table>() {
let (row, c) = row_res?;
for column_res in c.pairs::<usize, Cell>() {
let (column, cell) = column_res?;
sheet.set_cell(row, column, cell)
}
}
register.push(Arc::new(RwLock::new(sheet)));
Ok(id)
}
pub fn get(id: SheetId) -> Option<Arc<RwLock<Sheet>>> {
let register = REGISTER.read().unwrap();
if id < register.len() {
Some(Arc::clone(®ister[id]))
} else {
None
}
}
}
impl UserData for Register {
fn add_methods<'lua, M: mlua::UserDataMethods<'lua, Self>>(methods: &mut M) {
methods.add_function_mut("create", |lua, table: Table| {
let id = Register::create_from_table(table)?;
let luaref = SheetLuaRef::new(id);
if let Ok(ud) = lua.create_userdata(luaref) {
Ok(Value::UserData(ud))
} else {
Ok(Value::Nil)
}
});
methods.add_function_mut("get", |lua, id: SheetId| {
if Register::get(id).is_some() {
let luaref = SheetLuaRef::new(id);
if let Ok(ud) = lua.create_userdata(luaref) {
Ok(Value::UserData(ud))
} else {
Ok(Value::Nil)
}
} else {
Ok(Value::Nil)
}
})
}
}
|
