mirror of
https://github.com/pragma-/pbot.git
synced 2024-11-27 06:19:25 +01:00
74 lines
2.4 KiB
Python
Executable File
74 lines
2.4 KiB
Python
Executable File
#!/usr/bin/env python
|
|
from __future__ import print_function
|
|
|
|
import sys
|
|
|
|
from pycparser import c_parser, c_generator, c_ast, plyparser
|
|
from pycparser.ply import yacc
|
|
|
|
with open("paren/stddef") as f:
|
|
STDDEF = f.read()
|
|
|
|
class CParser(c_parser.CParser):
|
|
def __init__(self, *a, **kw):
|
|
super(CParser, self).__init__(*a, **kw)
|
|
self.p_expression_hack = self._expression_hack
|
|
self.cparser = yacc.yacc(
|
|
module=self,
|
|
start='expression_hack',
|
|
debug=kw.get('yacc_debug', False),
|
|
errorlog=yacc.NullLogger(),
|
|
optimize=kw.get('yacc_optimize', True),
|
|
tabmodule=kw.get('yacctab', 'yacctab'))
|
|
|
|
def parse(self, text, filename='', debuglevel=0):
|
|
self.clex.filename = filename
|
|
self.clex.reset_lineno()
|
|
self._scope_stack = [dict()]
|
|
self._last_yielded_token = None
|
|
for name in STDDEF.split('\n'):
|
|
if name:
|
|
self._add_typedef_name(name, None)
|
|
return self.cparser.parse(
|
|
input=text,
|
|
lexer=self.clex,
|
|
debug=debuglevel)
|
|
|
|
def _expression_hack(self, p):
|
|
""" expression_hack : translation_unit expression
|
|
| expression
|
|
"""
|
|
p[0] = c_ast.ExprList([p[2] if len(p) == 3 else p[1]])
|
|
|
|
|
|
class CGenerator(c_generator.CGenerator):
|
|
def visit_UnaryOp(self, n):
|
|
# don't parenthesize an operand to sizeof if it's not a type
|
|
if n.op == 'sizeof':
|
|
if isinstance(n.expr, c_ast.Typename):
|
|
return 'sizeof (%s)' % self.visit(n.expr)
|
|
else:
|
|
return 'sizeof %s' % self._parenthesize_unless_simple(n.expr)
|
|
return super(CGenerator, self).visit_UnaryOp(n)
|
|
|
|
def visit_Assignment(self, n):
|
|
return '%s %s %s' % (self.visit(n.lvalue), n.op, self._parenthesize_unless_simple(n.rvalue))
|
|
|
|
def translate_to_c(input):
|
|
parser = CParser(yacc_optimize=False)
|
|
try:
|
|
ast = parser.parse(input, '<input>')
|
|
except plyparser.ParseError as e:
|
|
print(sys.argv[1] + ': ' + "Error: {0}".format(e.args[0]))
|
|
return
|
|
generator = CGenerator()
|
|
print(sys.argv[1] + ': ' + generator.visit(ast))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) > 2:
|
|
translate_to_c(' '.join(sys.argv[2:]))
|
|
else:
|
|
print(sys.argv[1] + ': ' + "Usage: paren <expression>")
|
|
|