Virtual-World-RE/NeoGF
4
0
Fork 0
mirror of https://github.com/Virtual-World-RE/NeoGF.git synced 2024-11-15 08:45:33 +01:00
Code Issues Releases Wiki Activity

Update pzztool.py

Browse Source 
[*] programmer le TEST et vérifier tous les headers du data_afs pour voir si seul les TPLs ont le magicfile
[*] programmer le TEST et vérifier le bon fonctionnement du test pack/unpack sans décompression automatique
[*] f"{mavar}" <- voir si ça marche avec l'arithmétique et l'utiliser pour plus d'élégance

améliorations diverses sur des trucs qui ne servaient à rien
This commit is contained in:
tmpz23 2021-11-19 14:05:40 +01:00 committed by GitHub
parent 2dad1cd5b8
commit 2fc1f210b7
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
pzztool.py
View File

@ -3,7 +3,7 @@ from math import ceil
from struct import unpack
from pathlib import Path
from os import listdir, path
__version__ = "1.3.8"
__version__ = "1.3.9"
__author__ = "rigodron, algoflash, GGLinnk"
__OriginalAutor__ = "infval"
__license__ = "MIT"
@ -144,7 +144,7 @@ def pzz_compress(b):
def pzz_unpack(pzz_path, dest_folder):
if pzz_path.suffix != ".pzz":
print("WARNING - Unpack : Invalid file format '" + pzz_path.suffix + "'; it should be .pzz file format")
print(f"WARNING - Unpack : Invalid file format '{pzz_path.suffix}'; it should be .pzz file format")
if dest_folder != Path('.'):
unpacked_pzz_path = dest_folder
@ -160,7 +160,7 @@ def pzz_unpack(pzz_path, dest_folder):
# files_descriptors reçoit un tuple avec l'ensemble des descripteurs de fichiers (groupes d'uint32 big-endian)
files_descriptors = unpack(f">{file_count}I", pzz_file.read(file_count * 4))
print("File count:", file_count)
print(f"File count:{file_count}")
offset = CHUNK_SIZE
# on parcours le tuple de descripteurs de fichiers
@ -182,10 +182,10 @@ def pzz_unpack(pzz_path, dest_folder):
file_len = file_descriptor * CHUNK_SIZE
# On forme le nom du nouveau fichier que l'on va extraire
filename = "{:03}{}_{}".format(index, compression_status, pzz_path.stem)
filename = f"{index:03}{compression_status}_{pzz_path.stem}"
file_path = (unpacked_pzz_path / filename).with_suffix(".dat")
print("Offset: {:010} - {}".format(offset, file_path.stem))
print(f"Offset: {offset:010} - {file_path.stem}")
# Si la taille est nulle, on créé un fichier vide et on passe au descripteur de fichier suivant
if file_len == 0:
@ -219,7 +219,7 @@ def pzz_pack(src_path, dest_file):
else:
pzz_path = src_path.with_suffix(".pzz")
print(str(file_count) + " files to pack in " + str(pzz_path))
print(f"{file_count} files to pack in {pzz_path}")
with pzz_path.open("wb") as pzz_file:
# On écrit file_count au début de header
@ -252,16 +252,16 @@ def pzz_pack(src_path, dest_file):
with (src_path / src_file_name).open("rb") as src_file:
# Le fichier doit être compressé avant d'être pack
if compression_status == 'C' and is_compressed is False:
if compression_status == 'C' and not is_compressed:
pzz_file.write(pzz_compress(src_file.read()))
# Le fichier doit être décompressé avant d'être pack
elif compression_status == 'U' and is_compressed is True:
elif compression_status == 'U' and is_compressed:
pzz_file.write(pzz_decompress(src_file.read()))
else:
pzz_file.write(src_file.read())
# Si le fichier n'est pas compressé, on ajoute le padding pour correspondre à un multiple de CHUNK_SIZE
if compression_status == 'C' and (pzz_file.tell() % CHUNK_SIZE) > 0:
if compression_status == 'U' and (pzz_file.tell() % CHUNK_SIZE) > 0:
pzz_file.write(b"\x00" * (CHUNK_SIZE - (pzz_file.tell() % CHUNK_SIZE)))
@ -302,10 +302,10 @@ if __name__ == '__main__':
p_output.mkdir(exist_ok=True)
for filename in listdir(p_input):
if (not args.disable_ignore) and ("_uncompressed" in filename):
if (not args.disable_ignore) and not ("_compressed" in filename):
if args.verbose:
print(f"Compressing {filename}")
recomp_filename = filename.replace("_uncompressed", "_recompressed")
recomp_filename = Path(filename).stem + "_compressed" + Path(filename).suffix
uncompressed = open(path.join(p_input, filename), 'rb')
recompressed = open(path.join(p_output, filename), 'wb')
@ -321,7 +321,7 @@ if __name__ == '__main__':
for filename in listdir(p_input):
if (not args.disable_ignore) and ("_compressed" in filename):
print(f"Decompressing {filename}")
uncomp_filename = filename.replace("_compressed", "_uncompressed")
uncomp_filename = filename.replace("_compressed", "")
compressed = open(path.join(p_input, filename), 'rb')
uncompressed = open(path.join(p_output, uncomp_filename), 'wb')