steffo/lihzahrd
steffo/lihzahrd
1
Fork 0
mirror of https://github.com/Steffo99/lihzahrd.git synced 2024-11-21 23:54:23 +00:00
Code Activity

Optimize struct.unpack calls

Browse source
This commit is contained in:
Steffo 2023-03-06 02:36:15 +01:00
parent d451fd3731
commit 58d84d26f0
1 changed files with 35 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
lihzahrd/fileutils/filereader.py
View file

@ -28,45 +28,67 @@ class FileReader:
def __init__(self, file: typing.IO): def __init__(self, file: typing.IO):
self.file: typing.IO = file self.file: typing.IO = file
_bool = struct.Struct("?").unpack
def bool(self) -> bool: def bool(self) -> bool:
return struct.unpack("?", self.file.read(1))[0] return self._bool(self.file.read(1))[0]
_int1 = struct.Struct("B").unpack
def int1(self) -> int: def int1(self) -> int:
return struct.unpack("B", self.file.read(1))[0] return self._int1(self.file.read(1))[0]
def uint1(self) -> int: def uint1(self) -> int:
return struct.unpack("B", self.file.read(1))[0] return self._int1(self.file.read(1))[0]
_int2 = struct.Struct("h").unpack
def int2(self) -> int: def int2(self) -> int:
return struct.unpack("h", self.file.read(2))[0] return self._int2(self.file.read(2))[0]
_uint2 = struct.Struct("H").unpack
def uint2(self) -> int: def uint2(self) -> int:
return struct.unpack("H", self.file.read(2))[0] return self._uint2(self.file.read(2))[0]
_int4 = struct.Struct("i").unpack
def int4(self) -> int: def int4(self) -> int:
return struct.unpack("i", self.file.read(4))[0] return self._int4(self.file.read(4))[0]
_uint4 = struct.Struct("i").unpack
def uint4(self) -> int: def uint4(self) -> int:
return struct.unpack("I", self.file.read(4))[0] return self._uint4(self.file.read(4))[0]
_int8 = struct.Struct("q").unpack
def int8(self) -> int: def int8(self) -> int:
return struct.unpack("q", self.file.read(8))[0] return self._int8(self.file.read(8))[0]
_uint8 = struct.Struct("Q").unpack
def uint8(self) -> int: def uint8(self) -> int:
return struct.unpack("Q", self.file.read(8))[0] return self._uint8(self.file.read(8))[0]
_single = struct.Struct("f").unpack
def single(self) -> float: def single(self) -> float:
return struct.unpack("f", self.file.read(4))[0] return self._single(self.file.read(4))[0]
_double = struct.Struct("d").unpack
def double(self) -> float: def double(self) -> float:
return struct.unpack("d", self.file.read(8))[0] return self._double(self.file.read(8))[0]
def bits(self) -> typing.Tuple[bool, bool, bool, bool, bool, bool, bool, bool]: def bits(self) -> typing.Tuple[bool, bool, bool, bool, bool, bool, bool, bool]:
data = struct.unpack("B", self.file.read(1))[0] data = self._int1(self.file.read(1))[0]
return INT_TO_BITS_CACHE[data] return INT_TO_BITS_CACHE[data]
_rect = struct.Struct("iiii").unpack
def rect(self) -> Rect: def rect(self) -> Rect:
left, right, top, bottom = struct.unpack("iiii", self.file.read(16)) left, right, top, bottom = self._rect(self.file.read(16))
return Rect(left, right, top, bottom) return Rect(left, right, top, bottom)
def uleb128(self) -> int: def uleb128(self) -> int: