wbrawner/reone
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Change filelists strategy to search by Speaker, not VO_ResRef

Browse source
This commit is contained in:
Vsevolod Kremianskii 2021-04-15 10:04:30 +07:00
parent 6ddda21f45
commit 6efd9951c7
1 changed files with 8 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
scripts/filelists.py
View file

@ -6,8 +6,8 @@ import os
import random
import sys
extract_dir = r"D:\OpenKotOR\Extract\KotOR"
wav_dir = r"D:\OpenKotOR\Extract\KotOR\voices"
extract_dir = r"D:\OpenKotOR\Extract\TSL"
wav_dir = r"D:\OpenKotOR\Extract\TSL\voices"
if not os.path.exists(extract_dir):
@ -17,10 +17,6 @@ if not os.path.exists(wav_dir):
raise RuntimeError("WAV directory does not exist")
def is_trainable_text(text):
return not (text.startswith("[") and text.endswith("]"))
def index_or_negative_one(string, substr, beg=0):
try:
return string.index(substr, beg)
@ -55,15 +51,16 @@ def get_lines_from_dlg(obj, speaker, tlk_strings):
if "EntryList|15" in obj:
uniq_sound = set()
for entry in obj["EntryList|15"]:
if "VO_ResRef|11" in entry:
if "Speaker|10" in entry and "VO_ResRef|11" in entry:
voresref = entry["VO_ResRef|11"].lower()
textstrref = int(entry["Text|12"].split("|")[0])
if textstrref != -1:
text = tlk_strings[textstrref][1]
if voresref and (not voresref in uniq_sound) and is_trainable_text(text):
entry_speaker = entry["Speaker|10"].lower()
if voresref and (not voresref in uniq_sound) and textstrref != -1 and speaker in entry_speaker:
text = clear_text(tlk_strings[textstrref][1])
if text:
wav_filename = os.path.join(wav_dir, voresref + ".wav")
if os.path.exists(wav_filename):
lines.append("{}|{}|0\n".format(wav_filename, clear_text(text)))
lines.append("{}|{}|0\n".format(wav_filename, text))
uniq_sound.add(voresref)
return lines