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
|
use mlua::{Table, UserData, Value};
use super::{cell::Cell, loader::Loader, luaref::SheetLuaRef, Sheet};
static mut REGISTER: Vec<Sheet> = Vec::new();
pub type SheetId = usize;
#[derive(Debug)]
pub struct Register;
impl Register {
pub fn create(width: usize, height: usize) -> SheetId {
unsafe {
let id = REGISTER.len();
let sheet = Sheet::new(width, height, id);
REGISTER.push(sheet);
id
}
}
pub fn create_from_table(table: Table) -> mlua::Result<SheetId> {
let id = unsafe { 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)
}
}
unsafe { REGISTER.push(sheet) };
Ok(id)
}
pub fn get(id: SheetId) -> Option<&'static mut Sheet> {
unsafe { REGISTER.get_mut(id) }
}
}
impl UserData for Register {
fn add_methods<'lua, M: mlua::UserDataMethods<'lua, Self>>(methods: &mut M) {
methods.add_function("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("open", |lua, path: String| {
if let Some(id) = Loader::get().load_sheet(path, lua) {
let luaref = SheetLuaRef::new(id);
if let Ok(ud) = lua.create_userdata(luaref) {
return Ok(Value::UserData(ud));
}
}
Ok(Value::Nil)
});
methods.add_function("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)
}
})
}
}
|
