sm64pc/tools/texrename.py

72 lines
2.1 KiB
Python

#!/usr/bin/env python3
import sys
import os
import shutil
if len(sys.argv) < 3:
print("usage: texrename <in_dir> <out_dir> [<crcmap_file>]")
sys.exit(1)
inpath = sys.argv[1]
outpath = sys.argv[2]
mapfname = "crcmap.txt"
if len(sys.argv) > 3:
mapfname = sys.argv[3]
# catalog the original texture pack
texmap = dict()
imgexts = frozenset(['.png', '.bmp', '.jpg', '.tga', '.gif'])
try:
for root, dirs, files in os.walk(inpath):
for f in files:
ffull = os.path.join(root, f)
[fpath, fname] = os.path.split(f)
ext = os.path.splitext(fname)[1].lower()
if fname[0] == '.' or not (ext in imgexts):
continue
crc = 0
try:
if '#' in fname: # rice pack format: "GAME NAME#hash#whatever"
crc = int(fname.split('#')[1], 16)
else: # just the crc probably
crc = int(os.path.splitext(fname)[0], 16)
except ValueError:
print('unknown filename format: {0}'.format(ffull))
continue
texmap[crc] = ffull
except OSError as e:
print('error opening {0}: {1}'.format(inpath, e))
sys.exit(2)
# load the CRC map
crcmap = list()
try:
with open(mapfname, 'r') as f:
for line in f:
line = line.strip()
if line == '' or line[0] == '#':
continue
tok = line.split(',')
crcstr = tok[0].strip()
if crcstr.startswith('0x'):
crc = int(crcstr[2:], 16)
else:
crc = int(crcstr)
crcmap.append((crc, os.path.join(outpath, tok[1].strip())))
except OSError as e:
print('could not open {0}: {1}'.format(mapfname, e))
except ValueError as e:
print('invalid integer in {0}: {1}'.format(mapfname, e))
sys.exit(3)
# copy the files to the correct locations
for (crc, path) in crcmap:
if not (crc in texmap):
print('unmatched CRC: {0} ({1})'.format(crc, path))
else:
[fpath, fname] = os.path.split(path)
if not os.path.exists(fpath):
os.makedirs(fpath)
shutil.copy2(texmap[crc], path)