mirror of
https://github.com/HarbourMasters/Starship.git
synced 2025-01-23 21:45:00 +03:00
c69978c5eb
* it begins * stuff * remove extra * matched * forgot one * small things * more matches * filenames
100 lines
3.4 KiB
Python
100 lines
3.4 KiB
Python
import argparse
|
|
|
|
char_code = ["END","NWL","NP2","NP3","NP4","NP5","NP6","NP7","NP8","NP9","NPA","NPB","SPC","HSP","QSP","NPF","CLF","CUP","CRT","CDN","AUP","ALF","ADN","ART","L_A","L_B","L_C","L_D","L_E","L_F","L_G","L_H","L_I","L_J","L_K","L_L","L_M","L_N","L_O","L_P","L_Q","L_R","L_S","L_T","L_U","L_V","L_W","L_X","L_Y","L_Z","l_a","l_b","l_c","l_d","l_e","l_f","l_g","l_h","l_i","l_j","l_k","l_l","l_m","l_n","l_o","l_p","l_q","l_r","l_s","l_t","l_u","l_v","l_w","l_x","l_y","l_z","EXM","QST","DSH","CMA","PRD","N_0","N_1","N_2","N_3","N_4","N_5","N_6","N_7","N_8","N_9","APS","LPR","RPR","CLN","PIP"]
|
|
|
|
def convert_line(line):
|
|
if 'x' in line or '{' in line or '}' in line:
|
|
return None
|
|
|
|
chars = line.split(',')
|
|
|
|
if len(chars) <= 1:
|
|
return None
|
|
|
|
newchars = [char_code[int(x)] for x in chars[0:-1]]
|
|
|
|
return ','.join(newchars) + ',\n'
|
|
|
|
|
|
def convert_file(filepath):
|
|
with open(filepath, 'r') as file:
|
|
file_lines = file.readlines()
|
|
for i, line in enumerate(file_lines):
|
|
newchars = convert_line(line)
|
|
if newchars:
|
|
file_lines[i] = newchars
|
|
|
|
with open(filepath, 'w') as file:
|
|
file.writelines(file_lines)
|
|
|
|
return
|
|
|
|
def convert_names(filepath):
|
|
with open(filepath, 'r') as file:
|
|
file_text = file.read()
|
|
startidx = file_text.find('gMsgLookup[780] = {\n') + len('gMsgLookup[780] = {\n')
|
|
names = file_text[startidx:-3].split(',')
|
|
names = [x.strip(' \n}{') for x in names]
|
|
namepairs = zip(names[::2], names[1::2])
|
|
|
|
for pair in namepairs:
|
|
tname = 'sMsg_ID_' + pair[0]
|
|
file_text = file_text.replace(pair[1], tname)
|
|
with open(filepath, 'w') as file:
|
|
file.write(file_text)
|
|
return
|
|
|
|
code_to_text = {'CLF':'(C<)', 'CUP':'(C^)', 'CRT':'(C>)', 'CDN':'(Cv)', 'AUP':'^', 'ALF':'<', 'ADN':'v', 'ART':'>', 'EXM':'!', 'QST':'?', 'DSH':'-', 'CMA':',', 'PRD':'.', 'APS':"'", 'LPR':'(', 'RPR':')', 'CLN':':', 'PIP':'|'}
|
|
|
|
def parse_line(line):
|
|
codes = [x.strip() for x in line.split(',')]
|
|
text = ''
|
|
for i, code in enumerate(codes):
|
|
if code.startswith(('NP', 'PRI')) or code == '':
|
|
continue
|
|
elif 'SP' in code:
|
|
if(text and text[-1] != ' '):
|
|
text += ' '
|
|
elif 'NWL' in code:
|
|
if text:
|
|
text += '\n// '
|
|
elif code.startswith('_'):
|
|
text += code.strip('_')
|
|
else:
|
|
text += code_to_text[code]
|
|
return text
|
|
|
|
|
|
def add_comments(filepath):
|
|
with open(filepath, 'r') as file:
|
|
file_lines = file.readlines()
|
|
found_text = False
|
|
start_line = 0
|
|
for i, line in enumerate(file_lines):
|
|
if not found_text:
|
|
found_text = line.startswith('u16 gMsg_ID_')
|
|
out_text = '// '
|
|
start_line = i
|
|
elif 'END' in line:
|
|
if out_text.endswith('\n// '):
|
|
out_text = out_text[:-4]
|
|
file_lines[start_line] = out_text + '\n' + file_lines[start_line]
|
|
print(file_lines[start_line])
|
|
found_text = False
|
|
else:
|
|
out_text += parse_line(line)
|
|
with open(filepath, 'w') as file:
|
|
file.writelines(file_lines)
|
|
return
|
|
|
|
|
|
parser = argparse.ArgumentParser(description='Convert StarFox 64 dialog characters')
|
|
parser.add_argument('infile', help="SF64 text file to convert")
|
|
|
|
if __name__ == '__main__':
|
|
args = parser.parse_args()
|
|
|
|
# convert_file(args.infile)
|
|
# convert_names(args.infile)
|
|
add_comments(args.infile)
|