61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
import argparse
|
|
import os
|
|
import csv
|
|
from tools.util.type import detectType, stringToCType, typeToCType
|
|
|
|
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 csvIdToEnumName(inputId):
|
|
return "INPUT_ACTION_" + inputId.upper()
|
|
|
|
# Load up CSV file.
|
|
outHeader = "#pragma once\n"
|
|
outHeader += '#include "dusk.h"\n\n'
|
|
|
|
with open(args.csv, newline="", encoding="utf-8") as csvfile:
|
|
reader = csv.DictReader(csvfile)
|
|
|
|
# CSV must have id column.
|
|
if "id" not in reader.fieldnames:
|
|
raise Exception("CSV file must have 'id' column")
|
|
|
|
# For each ID
|
|
inputIds = []
|
|
inputIdValues = {}
|
|
for row in reader:
|
|
inputId = row["id"]
|
|
if inputId not in inputIds:
|
|
inputIds.append(inputId)
|
|
|
|
# For each ID, create enum entry.
|
|
count = 0
|
|
outHeader += "typedef enum {\n"
|
|
outHeader += f" INPUT_ACTION_NULL = {count},\n\n"
|
|
count += 1
|
|
for inputId in inputIds:
|
|
inputIdValues[inputId] = count
|
|
outHeader += f" {csvIdToEnumName(inputId)} = {count},\n"
|
|
count += 1
|
|
outHeader += f"\n INPUT_ACTION_COUNT = {count}\n"
|
|
outHeader += "} inputaction_t;\n\n"
|
|
|
|
# Write IDs to char array.
|
|
outHeader += f"static const char_t* INPUT_ACTION_IDS[] = {{\n"
|
|
for inputId in inputIds:
|
|
outHeader += f" [{csvIdToEnumName(inputId)}] = \"{inputId}\",\n"
|
|
outHeader += f"}};\n\n"
|
|
|
|
# Lua Script
|
|
outHeader += f"static const char_t *INPUT_ACTION_SCRIPT = \n"
|
|
for inputId in inputIds:
|
|
# Reference the enum
|
|
outHeader += f" \"{csvIdToEnumName(inputId)} = {inputIdValues[inputId]}\\n\"\n"
|
|
outHeader += f";\n\n"
|
|
|
|
# Write to output file.
|
|
os.makedirs(os.path.dirname(args.output), exist_ok=True)
|
|
with open(args.output, "w", encoding="utf-8") as outFile:
|
|
outFile.write(outHeader) |