87 lines
2.5 KiB
Python
87 lines
2.5 KiB
Python
def RemoveAnnotation(strl: str) -> str:
|
|
idx = strl.find("//")
|
|
if idx == -1: return strl.strip()
|
|
|
|
return strl[:idx].strip()
|
|
|
|
def Underline2Camel(orig: str) -> str:
|
|
return ''.join(map(
|
|
lambda mixed: mixed[-1].upper() if mixed[0] == '_' else mixed[-1].lower(),
|
|
((('_' if idx == 0 else orig[idx - 1]), c) for idx, c in enumerate(orig))
|
|
)).replace('_', '')
|
|
|
|
class EnumEntry():
|
|
def __init__(self, name: str, val: str):
|
|
self.name: str = name
|
|
self.val: str = val
|
|
def __repr__(self) -> str:
|
|
return f'<{self.name}: {self.val}>'
|
|
|
|
|
|
class EnumBody():
|
|
def __init__(self, name: str, is_flag: bool):
|
|
self.name: str = name
|
|
self.is_flag: bool = is_flag
|
|
self.entries: list[EnumEntry] = []
|
|
def __repr__(self) -> str:
|
|
return self.entries.__repr__()
|
|
|
|
full_enum: list[EnumBody] = []
|
|
current_body: EnumEntry = None
|
|
with open('src/CKMISC.txt', 'r', encoding='utf-8') as fr:
|
|
while True:
|
|
ln = fr.readline()
|
|
if ln == '':
|
|
break
|
|
ln = RemoveAnnotation(ln)
|
|
if ln == '':
|
|
continue
|
|
|
|
if ln.startswith('enum'):
|
|
ln = ln[len('enum'):].strip()
|
|
|
|
is_flag = ln[0] == '!'
|
|
name = ln[1:] if is_flag else ln
|
|
|
|
if current_body:
|
|
full_enum.append(current_body)
|
|
current_body = EnumBody(name, is_flag)
|
|
|
|
else:
|
|
sp = ln.replace(',', '').split('=')
|
|
if len(sp) == 1:
|
|
entry = EnumEntry(sp[0].strip(), '')
|
|
else:
|
|
entry = EnumEntry(sp[0].strip(), sp[-1].strip())
|
|
current_body.entries.append(entry)
|
|
|
|
fr.close()
|
|
if current_body:
|
|
full_enum.append(current_body)
|
|
|
|
with open('dest/CKMISC.txt', 'w', encoding='utf-8') as fw:
|
|
# write define
|
|
for item in full_enum:
|
|
fw.write('enum class {} : int32_t {{\n'.format(item.name))
|
|
|
|
fw.write(',\n'.join(map(
|
|
lambda x: x.name if x.val == '' else (x.name + ' = ' + x.val),
|
|
item.entries
|
|
)))
|
|
fw.write('\n};\n')
|
|
|
|
fw.write('\n')
|
|
|
|
# write vector
|
|
for item in full_enum:
|
|
fw.write('static const std::array<std::pair<LibCmo::{}, const char*>, {}> _{} {{\n'.format(
|
|
item.name, len(item.entries), Underline2Camel(item.name)
|
|
))
|
|
fw.write(',\n'.join(map(
|
|
lambda x: '{{ LibCmo::{}, "{}" }}'.format(x.name, x.name),
|
|
item.entries
|
|
)))
|
|
fw.write('\n};\n')
|
|
|
|
fw.close()
|