josh/TCPlayerbotCore
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Misc: support enum class in enumutils_describe.py

Browse Source
This commit is contained in:
Shauren
2022-02-05 20:55:14 +01:00
parent 26f3c25dae
commit a5c95bd01e
1 changed files with 14 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
contrib/enumutils_describe.py
+14 -10
View File
@@ -25,7 +25,7 @@ if not getcwd().endswith('src'):
print('(Invoke as \'python ../contrib/enumutils_describe.py\')') print('(Invoke as \'python ../contrib/enumutils_describe.py\')')
exit(1) exit(1)
EnumPattern = compile(r'//\s*EnumUtils: DESCRIBE THIS(?:\s*\(in ([^\)]+)\))?\s+enum\s+([0-9A-Za-z]+)[^\n]*\s*{([^}]+)};') EnumPattern = compile(r'//\s*EnumUtils: DESCRIBE THIS(?:\s*\(in ([^\)]+)\))?\s+enum\s+(class\s+)?([0-9A-Za-z]+)[^\n]*\s*{([^}]+)};')
EnumValuesPattern = compile(r'\s+\S.+?(,|$)[^\n]*') EnumValuesPattern = compile(r'\s+\S.+?(,|$)[^\n]*')
EnumValueNamePattern = compile(r'^\s*([a-zA-Z0-9_]+)', flags=MULTILINE) EnumValueNamePattern = compile(r'^\s*([a-zA-Z0-9_]+)', flags=MULTILINE)
EnumValueSkipLinePattern = compile(r'^\s*//') EnumValueSkipLinePattern = compile(r'^\s*//')
@@ -53,9 +53,13 @@ def processFile(path, filename):
enums = [] enums = []
for enum in EnumPattern.finditer(file): for enum in EnumPattern.finditer(file):
prefix = enum.group(1) or '' prefix = enum.group(1) or ''
name = enum.group(2) name = enum.group(3)
values = [] values = []
for value in EnumValuesPattern.finditer(enum.group(3)): # support enum class
valuePrefix = ''
if enum.group(2) is not None:
valuePrefix = name + '::'
for value in EnumValuesPattern.finditer(enum.group(4)):
valueData = value.group(0) valueData = value.group(0)
valueNameMatch = EnumValueNamePattern.search(valueData) valueNameMatch = EnumValueNamePattern.search(valueData)
@@ -88,7 +92,7 @@ def processFile(path, filename):
if valueDescription is None: if valueDescription is None:
valueDescription = '' valueDescription = ''
values.append((valueName, valueTitle, valueDescription)) values.append((valueName, valueTitle, valueDescription, valuePrefix))
enums.append((prefix + name, prefix, values)) enums.append((prefix + name, prefix, values))
print('%s.h: Enum %s parsed with %d values' % (filename, name, len(values))) print('%s.h: Enum %s parsed with %d values' % (filename, name, len(values)))
@@ -122,8 +126,8 @@ def processFile(path, filename):
output.write('{\n') output.write('{\n')
output.write(' switch (value)\n') output.write(' switch (value)\n')
output.write(' {\n') output.write(' {\n')
for label, title, description in values: for label, title, description, labelPrefix in values:
output.write(' case %s: return { %s, %s, %s };\n' % (prefix + label, strescape(label), strescape(title), strescape(description))) output.write(' case %s: return { %s, %s, %s };\n' % (prefix + labelPrefix + label, strescape(label), strescape(title), strescape(description)))
output.write(' default: throw std::out_of_range("value");\n') output.write(' default: throw std::out_of_range("value");\n')
output.write(' }\n') output.write(' }\n')
output.write('}\n') output.write('}\n')
@@ -136,8 +140,8 @@ def processFile(path, filename):
output.write('{\n') output.write('{\n')
output.write(' switch (index)\n') output.write(' switch (index)\n')
output.write(' {\n') output.write(' {\n')
for (i, (label, title, description)) in enumerate(values): for (i, (label, title, description, labelPrefix)) in enumerate(values):
output.write(' case %d: return %s;\n' % (i, prefix + label)) output.write(' case %d: return %s;\n' % (i, prefix + labelPrefix + label))
output.write(' default: throw std::out_of_range("index");\n') output.write(' default: throw std::out_of_range("index");\n')
output.write(' }\n') output.write(' }\n')
output.write('}\n') output.write('}\n')
@@ -147,8 +151,8 @@ def processFile(path, filename):
output.write('{\n') output.write('{\n')
output.write(' switch (value)\n') output.write(' switch (value)\n')
output.write(' {\n') output.write(' {\n')
for (i, (label, title, description)) in enumerate(values): for (i, (label, title, description, labelPrefix)) in enumerate(values):
output.write(' case %s: return %d;\n' % (prefix + label, i)) output.write(' case %s: return %d;\n' % (prefix + labelPrefix + label, i))
output.write(' default: throw std::out_of_range("value");\n') output.write(' default: throw std::out_of_range("value");\n')
output.write(' }\n') output.write(' }\n')
output.write('}\n') output.write('}\n')