Files
dusk/tools/input/csv/__main__.py
T
2026-04-13 19:51:11 -05:00

59 lines
1.5 KiB
Python

import argparse
import csv
import os
parser = argparse.ArgumentParser(description="Input CSV to .h defines")
parser.add_argument("--csv", required=True, help="Path to Input CSV file")
parser.add_argument("--output", required=True, help="Path to output .h file")
args = parser.parse_args()
def id_enum(name):
return "INPUT_ACTION_" + name.upper()
# Load CSV
input_ids = []
with open(args.csv, newline="", encoding="utf-8") as f:
reader = csv.DictReader(f)
if "id" not in reader.fieldnames:
raise ValueError("CSV must have an 'id' column")
for row in reader:
input_id = row["id"]
if input_id not in input_ids:
input_ids.append(input_id)
# Assign enum values
id_values = {input_id: i + 1 for i, input_id in enumerate(input_ids)}
# Build output
out = [
"#pragma once",
'#include "dusk.h"',
"",
"typedef enum {",
" INPUT_ACTION_NULL = 0x0,",
"",
]
for input_id in input_ids:
out.append(f" {id_enum(input_id)} = 0x{id_values[input_id]:x},")
out += [
"",
f" INPUT_ACTION_COUNT = 0x{len(input_ids) + 1:x}",
"} inputaction_t;",
"",
"static const char_t *INPUT_ACTION_IDS[] = {",
]
for input_id in input_ids:
out.append(f" [{id_enum(input_id)}] = \"{input_id}\",")
out += [
"};",
"",
"static const char_t *INPUT_ACTION_SCRIPT =",
]
for input_id in input_ids:
out.append(f" \"{id_enum(input_id)} = {id_values[input_id]}\\n\"")
out += [";", ""]
os.makedirs(os.path.dirname(args.output), exist_ok=True)
with open(args.output, "w", encoding="utf-8") as f:
f.write("\n".join(out))