mirror of
https://github.com/SPLURT-Station/S.P.L.U.R.T-Station-13.git
synced 2025-12-10 18:02:57 +00:00
95 lines
3.5 KiB
Python
95 lines
3.5 KiB
Python
#!/usr/bin/env python3
|
|
import shutil
|
|
from collections import defaultdict
|
|
from . import frontend
|
|
from .dmm import *
|
|
|
|
def merge_map(new_map, old_map, delete_unused=False):
|
|
if new_map.key_length != old_map.key_length:
|
|
print("Warning: Key lengths differ, taking new map")
|
|
print(f" Old: {old_map.key_length}")
|
|
print(f" New: {new_map.key_length}")
|
|
return new_map
|
|
|
|
if new_map.size != old_map.size:
|
|
print("Warning: Map dimensions differ, taking new map")
|
|
print(f" Old: {old_map.size}")
|
|
print(f" New: {new_map.size}")
|
|
return new_map
|
|
|
|
key_length, size = old_map.key_length, old_map.size
|
|
merged = DMM(key_length, size)
|
|
merged.dictionary = old_map.dictionary.copy()
|
|
|
|
known_keys = dict() # mapping fron 'new' key to 'merged' key
|
|
unused_keys = set(old_map.dictionary.keys()) # keys going unused
|
|
|
|
# step one: parse the new version, compare it to the old version, merge both
|
|
for z, y, x in new_map.coords_zyx:
|
|
new_key = new_map.grid[x, y, z]
|
|
# if this key has been processed before, it can immediately be merged
|
|
try:
|
|
merged.grid[x, y, z] = known_keys[new_key]
|
|
continue
|
|
except KeyError:
|
|
pass
|
|
|
|
def select_key(assigned):
|
|
merged.grid[x, y, z] = known_keys[new_key] = assigned
|
|
|
|
old_key = old_map.grid[x, y, z]
|
|
old_tile = old_map.dictionary[old_key]
|
|
new_tile = new_map.dictionary[new_key]
|
|
|
|
# this tile is the exact same as before, so the old key is used
|
|
if new_tile == old_tile:
|
|
select_key(old_key)
|
|
unused_keys.remove(old_key)
|
|
|
|
# the tile is different here, but if it exists in the merged dictionary, that key can be used
|
|
elif new_tile in merged.dictionary.inv:
|
|
newold_key = merged.dictionary.inv[new_tile]
|
|
select_key(newold_key)
|
|
unused_keys.remove(newold_key)
|
|
|
|
# the tile is brand new and it needs a new key, but if the old key isn't being used any longer it can be used instead
|
|
elif old_tile not in new_map.dictionary.inv and old_key in unused_keys:
|
|
merged.dictionary[old_key] = new_tile
|
|
select_key(old_key)
|
|
unused_keys.remove(old_key)
|
|
|
|
# all other options ruled out, a brand new key is generated for the brand new tile
|
|
else:
|
|
fresh_key = merged.generate_new_key()
|
|
merged.dictionary[fresh_key] = new_tile
|
|
select_key(fresh_key)
|
|
|
|
# step two: delete unused keys
|
|
if unused_keys:
|
|
#print(f"Notice: Trimming {len(unused_keys)} unused dictionary keys.")
|
|
for key in unused_keys:
|
|
del merged.dictionary[key]
|
|
|
|
# sanity check: that the merged map equals the new map
|
|
for z, y, x in new_map.coords_zyx:
|
|
new_tile = new_map.dictionary[new_map.grid[x, y, z]]
|
|
merged_tile = merged.dictionary[merged.grid[x, y, z]]
|
|
if new_tile != merged_tile:
|
|
print(f"Error: the map has been mangled! This is a mapmerge bug!")
|
|
print(f"At {x},{y},{z}.")
|
|
print(f"Should be {new_tile}")
|
|
print(f"Instead is {merged_tile}")
|
|
raise RuntimeError()
|
|
|
|
return merged
|
|
|
|
def main(settings):
|
|
for fname in frontend.process(settings, "merge", backup=True):
|
|
shutil.copyfile(fname, fname + ".before")
|
|
old_map = DMM.from_file(fname + ".backup")
|
|
new_map = DMM.from_file(fname)
|
|
merge_map(new_map, old_map).to_file(fname, settings.tgm)
|
|
|
|
if __name__ == '__main__':
|
|
main(frontend.read_settings())
|