65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
import zipfile
|
|
import argparse
|
|
import logging
|
|
import struct
|
|
|
|
|
|
def main():
|
|
objects = []
|
|
_ch = logging.StreamHandler()
|
|
_ch.setLevel("DEBUG")
|
|
_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
_ch.setFormatter(_formatter)
|
|
logger = logging.getLogger(__name__)
|
|
logger.addHandler(_ch)
|
|
logger.setLevel("DEBUG")
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("map_file")
|
|
args = parser.parse_args()
|
|
|
|
map_file = args.map_file
|
|
logger.info("Opening %s", map_file)
|
|
zip_file = zipfile.ZipFile(map_file)
|
|
files = zip_file.namelist()
|
|
bin_filename = None
|
|
for file in files:
|
|
logger.info("Found file %s", file)
|
|
if file.endswith(".bin"):
|
|
bin_filename = file
|
|
logger.info("Found BIN file %s", bin_filename)
|
|
break
|
|
if not bin_filename:
|
|
logger.error("FATAL: BIN file not found")
|
|
zip_file.close()
|
|
exit(1)
|
|
|
|
logger.info("Opening %s", bin_filename)
|
|
with zip_file.open(bin_filename) as bin_file:
|
|
byte = bin_file.read(1)
|
|
while byte != b"":
|
|
if byte == b'\x2a': # start object
|
|
# object is 20 bytes long
|
|
model = struct.unpack("<L", bin_file.read(4))[0]
|
|
x = struct.unpack("<f", bin_file.read(4))[0]
|
|
y = struct.unpack("<f", bin_file.read(4))[0]
|
|
z = struct.unpack("<f", bin_file.read(4))[0]
|
|
angle = struct.unpack("<f", bin_file.read(4))[0]
|
|
objects.append({"model": model, "x": x, "y": y, "z": z, "angle": angle})
|
|
logger.info("Object model=%s x=%s y=%s z=%s angle=%s", model, x, y, z, angle)
|
|
byte = bin_file.read(1)
|
|
zip_file.close()
|
|
|
|
logger.info("Number of objects: %s", len(objects))
|
|
models_count = {}
|
|
for game_object in objects:
|
|
if not game_object["model"] in models_count:
|
|
models_count[game_object["model"]] = 0
|
|
models_count[game_object["model"]] += 1
|
|
models_count = sorted(models_count.items(), key=lambda kv: kv[1], reverse=True)
|
|
for top_model in models_count:
|
|
logger.info("Model %s was found %s times", top_model[0], top_model[1])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|