99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
import argparse
|
|
import os
|
|
import csv
|
|
from tools.util.type import detectType, stringToCType, typeToCType
|
|
|
|
parser = argparse.ArgumentParser(description="Item CSV to .h defines")
|
|
parser.add_argument("--csv", required=True, help="Path to item CSV file")
|
|
parser.add_argument("--output", required=True, help="Path to output .h file")
|
|
args = parser.parse_args()
|
|
|
|
def csvIdToEnumName(itemId):
|
|
return "ITEM_ID_" + itemId.upper()
|
|
|
|
# Load up CSV file.
|
|
outHeader = "#pragma once\n"
|
|
outHeader += '#include "dusk.h"\n\n'
|
|
|
|
itemIds = []
|
|
itemTypes = []
|
|
itemRowById = {}
|
|
|
|
with open(args.csv, newline="", encoding="utf-8") as csvfile:
|
|
reader = csv.DictReader(csvfile)
|
|
|
|
# CSV must have id and type columns.
|
|
if "id" not in reader.fieldnames or "type" not in reader.fieldnames:
|
|
raise Exception("CSV file must have 'id' and 'type' columns")
|
|
|
|
for row in reader:
|
|
itemId = row["id"]
|
|
itemType = row["type"]
|
|
|
|
if itemId not in itemIds:
|
|
itemIds.append(itemId)
|
|
|
|
if itemType not in itemTypes:
|
|
itemTypes.append(itemType)
|
|
|
|
itemRowById[itemId] = row
|
|
|
|
# Now Prep output
|
|
outHeader = "#pragma once\n"
|
|
outHeader += '#include "dusk.h"\n\n'
|
|
|
|
itemTypeValues = {}
|
|
itemIdValues = {}
|
|
count = 0
|
|
|
|
# Create enum for types and ids, include null and count.
|
|
outHeader += "typedef enum {\n"
|
|
outHeader += f" ITEM_TYPE_NULL = {count},\n"
|
|
count += 1
|
|
for itemType in itemTypes:
|
|
itemTypeValues[itemType] = count
|
|
outHeader += f" {csvIdToEnumName(itemType)} = {count},\n"
|
|
count += 1
|
|
outHeader += f" ITEM_TYPE_COUNT = {count}\n"
|
|
outHeader += "} itemtype_t;\n\n"
|
|
|
|
outHeader += "typedef enum {\n"
|
|
outHeader += f" ITEM_ID_NULL = {count},\n"
|
|
for itemId in itemIds:
|
|
itemIdValues[itemId] = count
|
|
outHeader += f" {csvIdToEnumName(itemId)} = {count},\n"
|
|
count += 1
|
|
outHeader += f" ITEM_ID_COUNT = {count}\n"
|
|
outHeader += "} itemid_t;\n\n"
|
|
|
|
# Create struct for item data.
|
|
outHeader += "typedef struct {\n"
|
|
outHeader += " itemid_t id;\n"
|
|
outHeader += " itemtype_t type;\n"
|
|
outHeader += " const char_t *name;\n"
|
|
outHeader += "} item_t;\n\n"
|
|
|
|
# Create array of item data.
|
|
outHeader += f"static const item_t ITEMS[] = {{\n"
|
|
for itemId in itemIds:
|
|
outHeader += f" [{csvIdToEnumName(itemId)}] = {{\n"
|
|
outHeader += f" .id = {csvIdToEnumName(itemId)},\n"
|
|
itemType = itemRowById[itemId]["type"]
|
|
outHeader += f" .type = {csvIdToEnumName(itemType)},\n"
|
|
itemName = itemRowById[itemId]["id"]
|
|
outHeader += f" .name = \"{itemName}\",\n"
|
|
outHeader += f" }},\n"
|
|
outHeader += f"}};\n\n"
|
|
|
|
# Create lua script defining items.
|
|
outHeader += f"static const char_t *ITEM_SCRIPT = \n"
|
|
for itemId in itemIds:
|
|
outHeader += f" \"{csvIdToEnumName(itemId)} = {itemIdValues[itemId]}\\n\"\n"
|
|
for itemType in itemTypes:
|
|
outHeader += f" \"{csvIdToEnumName(itemType)} = {itemTypeValues[itemType]}\\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) |