import hashlib
import os
import struct
import wave

SPEECH_TABLES = {
    # US norm.exe
    "c40f3d32d1d4e11c97f8c960e6011495566fb9f7e61b437b82dbe49194edbb7a": 0xbd978,
    # UK norm.exe, from https://archive.org/details/NormalityUKDOS
    "49dcc2085369106113cabe1319046272827817bedb30b1765d42be8e5d1ce3f7": 0xbd8d0
}
LINE_COUNT = 6501

def hashfile(filename):
    with open(filename, 'rb') as f:
        m = hashlib.sha256()
        m.update(f.read())
        return m.hexdigest()

def dumpspeech(exefilename, speechfilename, outputdir):
    os.makedirs(outputdir, exist_ok=True)
    table_offset = SPEECH_TABLES.get(hashfile(exefilename))
    if table_offset is None:
        raise Exception("Unrecognized norm.exe, sorry!")
    with open(exefilename, 'rb') as f:
        with open(speechfilename, 'rb') as s:
            f.seek(table_offset)
            for _ in range(LINE_COUNT):
                (line, offset, size) = struct.unpack("<iii", f.read(12))
                if line >= 0 and offset >= 0 and size > 1:
                    s.seek(offset)
                    with wave.open(os.path.join(outputdir, f'{line}.wav'), 'wb') as w:
                        w.setnchannels(1)
                        w.setsampwidth(1)
                        w.setframerate(11025)
                        w.writeframes(s.read(size))

dumpspeech("norm.exe.us", "sound.raw.us", "lines-us")
dumpspeech("norm.exe.uk", "sound.raw.uk", "lines-uk")