Add python script to generate 'struct value_string' from enum

The idea is that you have an input file that defines an enum with
symbolic names for numeric values.  The script then generates a 'struct
value_string' with corresponding strings.
This commit is contained in:
Harald Welte 2017-01-22 22:29:03 +01:00
parent 78e84df4dc
commit db5831c1c9
1 changed files with 60 additions and 0 deletions

60
tools/enum-create-valstr.py Executable file
View File

@ -0,0 +1,60 @@
#!/usr/bin/env python
#
# Code to generate a 'struct value string' from enums
# (C) 2017 by Harald Welte <laforge@gnumonks.org>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
from optparse import OptionParser
import pprint
from pycparser import c_parser, c_ast, parse_file
from value_string import export_value_str
class EnumValStrVisitor(c_ast.NodeVisitor):
"Generate a 'struct value_string' from an enum"
def visit_Enum(self, node):
name = "%s_vals" % node.name
elist = node.values
vdict = {}
for val in elist.enumerators:
vdict[val.name] = val.name
export_value_str(name, vdict, sort_key=None)
class EnumValuePrinter(c_ast.NodeVisitor):
def visit_Enum(self, node):
print("enum %s = {" % node.name)
elist = node.values
next_val = 0
for val in elist.enumerators:
if val.value:
numeric = int(val.value.value, 0)
next_val = numeric
else:
numeric = next_val
print("\t%s = %d" % (val.name, numeric))
next_val += 1
print("};")
parser = OptionParser()
(options, args) = parser.parse_args()
filename = args[0]
ast = parse_file(filename, use_cpp=True)
v = EnumValStrVisitor()
#v = EnumValuePrinter()
v.visit(ast)