LEGO_Universe/utils
1
0
Fork 0
mirror of https://github.com/lcdr/utils.git synced 2024-08-30 17:32:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Changed decompress_sd0 to operate on bytes instead of files.

Browse Source
This commit is contained in:
lcdr 2015-09-26 16:29:40 +02:00
parent e2e7ff5183
commit 3e10476cd6
1 changed files with 13 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
decompress_sd0.py
View File

@ -2,18 +2,16 @@ import argparse
import os.path
import zlib
def decompress(in_path, out_path):
with open(in_path, "rb") as in_file:
data = in_file.read()
def decompress(data):
assert data[:5] == b"sd0\x01\xff"
pos = 5
with open(out_path, "wb") as out_file:
while pos < len(data):
length = int.from_bytes(data[pos:pos+4], "little")
pos += 4
out_file.write(zlib.decompress(data[pos:pos+length]))
pos += length
out = b""
while pos < len(data):
length = int.from_bytes(data[pos:pos+4], "little")
pos += 4
out += zlib.decompress(data[pos:pos+length])
pos += length
return out
if __name__ == "__main__":
parser = argparse.ArgumentParser()
@ -24,6 +22,10 @@ if __name__ == "__main__":
filename, ext = os.path.splitext(os.path.basename(args.in_path))
args.out_path = filename+"_decompressed"+ext
decompress(args.in_path, args.out_path)
with open(args.in_path, "rb") as file:
data = in_file.read()
with open(args.out_path, "wb") as file:
file.write(decompress(data))
print("Decompressed file:", args.out_path)