slsDetectorPackage/python/scripts/generate_enums.py

import re

def comment_remover(text):
    def replacer(match):
        s = match.group(0)
        if s.startswith('/'):
            return " " # note: a space and not an empty string
        else:
            return s
    pattern = re.compile(
        r'//.*?$|/\*.*?\*/|\'(?:\\.|[^\\\'])*\'|"(?:\\.|[^\\"])*"',
        re.DOTALL | re.MULTILINE
    )
    return re.sub(pattern, replacer, text)

def extract_enums(lines):
    line_iter = iter(lines)
    enums = {}
    for line in line_iter:
        m = re.search("(?<=enum )\w+(?= {)", line)
        if m:
            enum_name = m.group()
            # print(enum_name)
            fields = []
            while True:
                l  = next(line_iter)
                if '};' in l:
                    break
                m = re.search("\w+", l)
                try:
                    # print('\t', m.group())
                    fields.append(m.group())

                except:
                    pass
            enums[enum_name] = fields
    return enums

def generate_enum_string(enums):
    data = []
    for key, value in enums.items():
        data.append(f'py::enum_<slsDetectorDefs::{key}>(Defs, "{key}")\n')
        for v in value:
            data.append(f'\t.value("{v}", slsDetectorDefs::{key}::{v})\n')
        data.append('.export_values();\n\n')
    return ''.join(data)

with open('../../slsSupportLib/include/sls_detector_defs.h') as f:
    data = f.read()

data = comment_remover(data)
data = data.splitlines()
enums = extract_enums(data)
s = generate_enum_string(enums)

with open('../src/enums_in.cpp') as f:
    data = f.read()

text = data.replace('[[ENUMS]]', s)
warning = '/* WARINING This file is auto generated any edits might be overwritten without warning */\n\n'
with open('../src/enums.cpp', 'w') as f:
    f.write(warning)
    f.write(text)