redstrate/physis
1
Fork 0
mirror of https://github.com/redstrate/Physis.git synced 2025-04-22 04:37:46 +00:00
Code Activity

Reorganize some functions in EXD

Browse source
This commit is contained in:
Joshua Goins 2024-04-14 13:28:43 -04:00
parent 6f12227134
commit 90bd2d334d
1 changed files with 56 additions and 59 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

115
src/exd.rs
View file

@ -70,27 +70,72 @@ pub struct ExcelRow {
} }
impl EXD { impl EXD {
pub fn from_existing(exh: &EXH, buffer: ByteSpan) -> Option<EXD> {
let mut cursor = Cursor::new(buffer);
let mut exd = EXD::read(&mut cursor).ok()?;
for i in 0..exh.header.row_count {
for offset in &exd.data_offsets {
if offset.row_id == i {
cursor.seek(SeekFrom::Start(offset.offset.into())).ok()?;
let row_header = ExcelDataRowHeader::read(&mut cursor).ok()?;
let header_offset = offset.offset + 6;// std::mem::size_of::<ExcelDataRowHeader>() as u32;
let mut read_row = |row_offset: u32| -> Option<ExcelRow> {
let mut subrow = ExcelRow {
data: Vec::with_capacity(exh.column_definitions.len()),
};
for column in &exh.column_definitions {
cursor
.seek(SeekFrom::Start((row_offset + column.offset as u32).into()))
.ok()?;
subrow
.data
.push(Self::read_column(&mut cursor, exh, row_offset, column).unwrap());
}
Some(subrow)
};
if row_header.row_count > 1 {
for i in 0..row_header.row_count {
let subrow_offset =
header_offset + (i * exh.header.data_offset + 2 * (i + 1)) as u32;
exd.rows.push(read_row(subrow_offset).unwrap());
}
} else {
exd.rows.push(read_row(header_offset).unwrap());
}
}
}
}
Some(exd)
}
fn read_data_raw<Z: BinRead<Args<'static> = ()>>(cursor: &mut Cursor<ByteSpan>) -> Option<Z> fn read_data_raw<Z: BinRead<Args<'static> = ()>>(cursor: &mut Cursor<ByteSpan>) -> Option<Z>
{ {
Some( Z::read_options(
Z::read_options( cursor,
cursor, Endian::Big,
Endian::Big, (),
(), ).ok()
)
.unwrap(),
)
} }
fn read_column( fn read_column(
cursor: &mut Cursor<ByteSpan>, cursor: &mut Cursor<ByteSpan>,
exh: &EXH, exh: &EXH,
offset: u32, row_offset: u32,
column: &ExcelColumnDefinition, column: &ExcelColumnDefinition,
) -> Option<ColumnData> { ) -> Option<ColumnData> {
let mut read_packed_bool = |shift: i32| -> bool { let mut read_packed_bool = |shift: i32| -> bool {
let bit = 1 << shift; let bit = 1 << shift;
let bool_data: i32 = Self::read_data_raw(cursor).unwrap(); let bool_data: i32 = Self::read_data_raw(cursor).unwrap_or(0);
(bool_data & bit) == bit (bool_data & bit) == bit
}; };
@ -101,7 +146,7 @@ impl EXD {
cursor cursor
.seek(SeekFrom::Start( .seek(SeekFrom::Start(
(offset + exh.header.data_offset as u32 + string_offset).into(), (row_offset + exh.header.data_offset as u32 + string_offset).into(),
)) ))
.ok()?; .ok()?;
@ -149,54 +194,6 @@ impl EXD {
} }
} }
pub fn from_existing(exh: &EXH, buffer: ByteSpan) -> Option<EXD> {
let mut cursor = Cursor::new(buffer);
let mut exd = EXD::read(&mut cursor).ok()?;
for i in 0..exh.header.row_count {
for offset in &exd.data_offsets {
if offset.row_id == i {
cursor.seek(SeekFrom::Start(offset.offset.into())).ok()?;
let row_header = ExcelDataRowHeader::read(&mut cursor).ok()?;
let header_offset = offset.offset + 6;
let mut read_row = |offset: u32| -> Option<ExcelRow> {
let mut subrow = ExcelRow {
data: Vec::with_capacity(exh.column_definitions.len()),
};
for column in &exh.column_definitions {
cursor
.seek(SeekFrom::Start((offset + column.offset as u32).into()))
.ok()?;
subrow
.data
.push(Self::read_column(&mut cursor, exh, offset, column).unwrap());
}
Some(subrow)
};
if row_header.row_count > 1 {
for i in 0..row_header.row_count {
let subrow_offset =
header_offset + (i * exh.header.data_offset + 2 * (i + 1)) as u32;
exd.rows.push(read_row(subrow_offset).unwrap());
}
} else {
exd.rows.push(read_row(header_offset).unwrap());
}
}
}
}
Some(exd)
}
pub fn calculate_filename( pub fn calculate_filename(
name: &str, name: &str,
language: Language, language: Language,