generateResourcesFile.py

#!/usr/bin/env python3

import os
import sys
import itertools
from enum import Enum
import shlex
import re

def convert(name):
    s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', name)
    return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).upper()

def main():
    if len(sys.argv) < 3:
        return
    with open(sys.argv[1], 'r') as resources_file:
        args = shlex.split(resources_file.read())
        with open(sys.argv[2], 'w') as out_file:
            # Add include guard
            out_file.write("#ifndef " + convert(sys.argv[3]) + "_RESOURCES_HPP_\n")
            out_file.write("#define " + convert(sys.argv[3]) + "_RESOURCES_HPP_\n\n")
            
            # Add header includes=
            out_file.write("#include <Util/ResourceDefs.hpp>\n")
            out_file.write("\n")
            out_file.write("#include <cstddef>\n")
            out_file.write("#include <utility>\n\n")
            
            # Add namespace declaration
            out_file.write("namespace " + sys.argv[3] + " {\n\n")
            
            # Declare ResourceID enum
            out_file.write("enum class ResourceID : size_t {\n")
            num = 0
            for i in args[0::4]:
                out_file.write("\t\t" + i + " = " + str(num) + ",\n")
                num = num + 1
            
            out_file.write("\n};\n\n")
            
            # Add Resource<ResourceID> array declaration
            out_file.write("constexpr Resource<ResourceID> Resources[] = {\n")
            
            class State(Enum):
                ID = 1
                TYPE = 2
                PATH = 3
                MAIN_FILE = 4
            
            # Add array contents
            # ID first, Type second, Path third
            first_in_list = True
            state = State.ID
            for i in args:
                if state == State.ID:
                    if not first_in_list:
                        out_file.write(",\n")
                    first_in_list = False
                    out_file.write("\t\t{ResourceID::" + i + ", ")
                    state = State.TYPE
                    continue
                elif state == State.TYPE:
                    out_file.write("ResourceType::" + i + ", ")
                    state = State.PATH
                    continue
                elif state == State.PATH:
                    out_file.write("\"" + i + "\", ")
                    state = State.MAIN_FILE
                    continue
                elif state == State.MAIN_FILE:
                    out_file.write("\"" + i + "\"}")
                    state = State.ID
                    continue
            out_file.write("\n};\n\n")
            
            # Add getResource convenience function
            out_file.write("constexpr inline Resource<ResourceID> getResource(ResourceID id) {\n")
            out_file.write("\treturn Resources[static_cast<size_t>(id)];\n")
            out_file.write("}\n\n")
            out_file.write("} /* namespace " + sys.argv[3] + " */\n\n\n")
            
            out_file.write("#endif /* " + convert(sys.argv[3]) + "_RESOURCES_HPP_ */\n")
if __name__ == '__main__':
    main()