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

Implement bit unpacking in a more efficient way

Browse source 
Co-authored-by: "rodrigo a. araujo" <rodjjo@gmail.com>
This commit is contained in:
Steffo 2023-03-06 00:04:51 +01:00
parent 1060930d66
commit 518d273066
1 changed files with 16 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

31
lihzahrd/fileutils/filereader.py
View file

@ -6,6 +6,21 @@ import functools
from .rect import Rect from .rect import Rect
INT_TO_BITS_CACHE = {
i: (
bool(i & 0b0000_0001),
bool(i & 0b0000_0010),
bool(i & 0b0000_0100),
bool(i & 0b0000_1000),
bool(i & 0b0001_0000),
bool(i & 0b0010_0000),
bool(i & 0b0100_0000),
bool(i & 0b1000_0000)
)
for i in range(256)
}
class FileReader: class FileReader:
"""Helper class for deserializing a Terraria world file.""" """Helper class for deserializing a Terraria world file."""
@ -47,23 +62,9 @@ class FileReader:
def double(self) -> float: def double(self) -> float:
return struct.unpack("d", self.file.read(8))[0] return struct.unpack("d", self.file.read(8))[0]
@staticmethod
@functools.lru_cache(256)
def _bitify(data) -> typing.Tuple[bool, bool, bool, bool, bool, bool, bool, bool]:
return (
bool(data & 0b0000_0001),
bool(data & 0b0000_0010),
bool(data & 0b0000_0100),
bool(data & 0b0000_1000),
bool(data & 0b0001_0000),
bool(data & 0b0010_0000),
bool(data & 0b0100_0000),
bool(data & 0b1000_0000),
)
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 = struct.unpack("B", self.file.read(1))[0]
return self._bitify(data) return INT_TO_BITS_CACHE[data]
def rect(self) -> Rect: def rect(self) -> Rect:
left, right, top, bottom = struct.unpack("iiii", self.file.read(16)) left, right, top, bottom = struct.unpack("iiii", self.file.read(16))