prog

tools/display/color/csv/__main__.py

@@ -0,0 +1,64 @@
+import argparse
+import os
+import csv
+from tools.util.type import detectType, stringToCType, typeToCType
+
+parser = argparse.ArgumentParser(description="Color CSV to .h defines")
+parser.add_argument("--csv", required=True, help="Path to color 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'
+
+outHeader += f"typedef float_t colorchannelf_t;\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)