You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
121 lines
5.2 KiB
121 lines
5.2 KiB
#!/usr/bin/env python |
|
import csv |
|
import pathlib |
|
|
|
root = pathlib.Path(__file__).resolve().parent.parent |
|
translation_dummy_path = root.joinpath("Source/translation_dummy.cpp") |
|
|
|
base_paths = { |
|
"monstdat": root.joinpath("assets/txtdata/monsters/monstdat.tsv"), |
|
"unique_monstdat": root.joinpath("assets/txtdata/monsters/unique_monstdat.tsv"), |
|
"itemdat": root.joinpath("assets/txtdata/items/itemdat.tsv"), |
|
"unique_itemdat": root.joinpath("assets/txtdata/items/unique_itemdat.tsv"), |
|
"item_prefixes": root.joinpath("assets/txtdata/items/item_prefixes.tsv"), |
|
"item_suffixes": root.joinpath("assets/txtdata/items/item_suffixes.tsv"), |
|
"questdat": root.joinpath("assets/txtdata/quests/questdat.tsv"), |
|
"spelldat": root.joinpath("assets/txtdata/spells/spelldat.tsv"), |
|
"textdat": root.joinpath("assets/txtdata/text/textdat.tsv"), |
|
} |
|
|
|
hf_paths = { |
|
"monstdat": root.joinpath("mods/Hellfire/txtdata/monsters/monstdat.tsv"), |
|
"unique_itemdat": root.joinpath("mods/Hellfire/txtdata/items/unique_itemdat.tsv"), |
|
"item_prefixes": root.joinpath("mods/Hellfire/txtdata/items/item_prefixes.tsv"), |
|
"item_suffixes": root.joinpath("mods/Hellfire/txtdata/items/item_suffixes.tsv"), |
|
"spelldat": root.joinpath("mods/Hellfire/txtdata/spells/spelldat.tsv"), |
|
} |
|
|
|
seen_pairs = set() |
|
|
|
def write_entry(temp_source, var_name, context, string_value, use_p): |
|
if not string_value: |
|
return |
|
key = (context, string_value) |
|
if key in seen_pairs: |
|
return |
|
seen_pairs.add(key) |
|
if use_p: |
|
temp_source.write(f'const char *{var_name} = P_("{context}", "{string_value}");\n') |
|
else: |
|
temp_source.write(f'const char *{var_name} = N_("{string_value}");\n') |
|
|
|
def process_files(paths, temp_source): |
|
# Monsters |
|
with open(paths["monstdat"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for row in reader: |
|
var_name = row['_monster_id'] + "_NAME" |
|
write_entry(temp_source, var_name, "monster", row['name'], True) |
|
|
|
if "unique_monstdat" in paths: |
|
with open(paths["unique_monstdat"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for row in reader: |
|
var_name = row['name'].upper().replace(' ', '_').replace('-', '_') + "_NAME" |
|
write_entry(temp_source, var_name, "monster", row['name'], True) |
|
|
|
# Items |
|
if "itemdat" in paths: |
|
with open(paths["itemdat"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for i, row in enumerate(reader): |
|
name = row['name'] |
|
if name in ('Scroll of None', 'Non Item', 'Book of '): |
|
continue |
|
shortName = row['shortName'] |
|
var_name = row['id'] if row['id'] else f'ITEM_{i}' |
|
write_entry(temp_source, f'{var_name}_NAME', "default", name, False) |
|
if shortName: |
|
write_entry(temp_source, f'{var_name}_SHORT_NAME', "default", shortName, False) |
|
|
|
with open(paths["unique_itemdat"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for i, row in enumerate(reader): |
|
write_entry(temp_source, f'UNIQUE_ITEM_{i}_NAME', "default", row['name'], False) |
|
|
|
with open(paths["item_prefixes"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for i, row in enumerate(reader): |
|
write_entry(temp_source, f'ITEM_PREFIX_{i}_NAME', "default", row['name'], False) |
|
|
|
with open(paths["item_suffixes"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for i, row in enumerate(reader): |
|
write_entry(temp_source, f'ITEM_SUFFIX_{i}_NAME', "default", row['name'], False) |
|
|
|
# Quests |
|
if "questdat" in paths: |
|
with open(paths["questdat"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for i, row in enumerate(reader): |
|
var_name = 'QUEST_' + row['qlstr'].upper().replace(' ', '_').replace('-', '_') |
|
write_entry(temp_source, f'{var_name}_NAME', "default", row['qlstr'], False) |
|
|
|
# Spells |
|
with open(paths["spelldat"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for i, row in enumerate(reader): |
|
var_name = 'SPELL_' + row['name'].upper().replace(' ', '_').replace('-', '_') |
|
write_entry(temp_source, f'{var_name}_NAME', "spell", row['name'], True) |
|
|
|
# Text/Speeches |
|
if "textdat" in paths: |
|
with open(paths["textdat"], 'r') as tsv: |
|
reader = csv.DictReader(tsv, delimiter='\t') |
|
for i, row in enumerate(reader): |
|
write_entry(temp_source, f'TEXT_{i}', "default", row['txtstr'], False) |
|
|
|
with open(translation_dummy_path, 'w') as temp_source: |
|
temp_source.write(f'/**\n') |
|
temp_source.write(f' * @file translation_dummy.cpp\n') |
|
temp_source.write(f' *\n') |
|
temp_source.write(f' * Do not edit this file manually, it is automatically generated\n') |
|
temp_source.write(f' * and updated by the extract_translation_data.py script.\n') |
|
temp_source.write(f' */\n') |
|
temp_source.write(f'#include "utils/language.h"\n\n') |
|
temp_source.write(f'namespace {{\n\n') |
|
|
|
process_files(base_paths, temp_source) |
|
process_files(hf_paths, temp_source) |
|
|
|
temp_source.write(f'\n}} // namespace\n')
|
|
|