Fixed duskdefs
This commit is contained in:
50
tools/csv_to_enum/__main__.py
Normal file
50
tools/csv_to_enum/__main__.py
Normal file
@@ -0,0 +1,50 @@
|
||||
import argparse
|
||||
import os
|
||||
import csv
|
||||
|
||||
parser = argparse.ArgumentParser(description="Convert CSV to .h defines")
|
||||
parser.add_argument("--csv", required=True, help="Path to CSV file")
|
||||
parser.add_argument("--output", required=True, help="Path to output .h file")
|
||||
parser.add_argument("--prefix", default="", help="Prefix for each define")
|
||||
parser.add_argument("--typedef", required=True, help="Typedef name for the enum")
|
||||
parser.add_argument("--take-column", required=True, help="Column name to take the values for the enum for")
|
||||
parser.add_argument("--null-entry", default=None, help="Optional name for a NULL/INVALID entry")
|
||||
parser.add_argument("--count-entry", default=None, help="Optional name for a COUNT entry")
|
||||
args = parser.parse_args()
|
||||
|
||||
outHeader = ""
|
||||
outHeader += "#include \"dusk.h\"\n\n"
|
||||
with open(args.csv, newline='') as csvfile:
|
||||
reader = csv.reader(csvfile)
|
||||
|
||||
headers = next(reader)
|
||||
if args.take_column not in headers:
|
||||
raise ValueError(f"Column '{args.take_column}' not found in CSV headers")
|
||||
take_column_index = headers.index(args.take_column)
|
||||
|
||||
# Extract entries
|
||||
entries = []
|
||||
for row in reader:
|
||||
value = row[take_column_index]
|
||||
clean_define_name = value.replace(" ", "_").upper()
|
||||
if clean_define_name in entries:
|
||||
continue # Avoid duplicates
|
||||
entries.append(clean_define_name)
|
||||
|
||||
# Gen enum.
|
||||
outHeader += f"typedef enum {{\n"
|
||||
if args.null_entry == "TRUE":
|
||||
outHeader += f" {args.prefix}_NULL = 0,\n\n"
|
||||
|
||||
for entry in entries:
|
||||
outHeader += f" {args.prefix}_{entry},\n"
|
||||
|
||||
if args.count_entry == "TRUE":
|
||||
outHeader += f"\n {args.prefix}_COUNT,\n"
|
||||
|
||||
outHeader += f"}} {args.typedef};\n\n"
|
||||
|
||||
# Write to output file (mkdirp if needed)
|
||||
os.makedirs(os.path.dirname(args.output), exist_ok=True)
|
||||
with open(args.output, 'w') as outFile:
|
||||
outFile.write(outHeader)
|
||||
Reference in New Issue
Block a user