code.kerkeslager.com Git - fur/blob - generation.py

   1 import jinja2
   2
   3 import transformation
   4
   5 ENV = jinja2.Environment(
   6     autoescape=jinja2.select_autoescape([]),
   7     loader=jinja2.FileSystemLoader('templates'),
   8     trim_blocks=True,
   9 )
  10
  11 def generate_integer_literal(c_integer_literal):
  12     return 'integerLiteral({})'.format(c_integer_literal.value)
  13
  14 def generate_string_literal(c_string_literal):
  15     def c_escape(ch):
  16         return {
  17             '\n': r'\n',
  18             '"': r'\"',
  19             '\\': r'\\',
  20         }.get(ch, ch)
  21
  22     return 'stringLiteral(runtime, "{}")'.format(
  23         ''.join(c_escape(ch for ch in c_string_literal.value)),
  24     )
  25
  26 def generate_argument(c_argument):
  27     if isinstance(c_argument, transformation.CFunctionCallExpression):
  28         return generate_function_call(c_argument)
  29
  30     LITERAL_TYPE_MAPPING = {
  31         transformation.CIntegerLiteral: generate_integer_literal,
  32         transformation.CStringLiteral: generate_string_literal,
  33     }
  34
  35     if type(c_argument) in LITERAL_TYPE_MAPPING:
  36         return LITERAL_TYPE_MAPPING[type(c_argument)](c_argument)
  37
  38     INFIX_TYPE_MAPPING = {
  39         transformation.CAdditionExpression: 'add',
  40         transformation.CSubtractionExpression: 'subtract',
  41         transformation.CMultiplicationExpression: 'multiply',
  42         transformation.CIntegerDivisionExpression: 'integerDivide',
  43         transformation.CModularDivisionExpression: 'modularDivide',
  44     }
  45
  46     return 'builtin${}({}, {})'.format(
  47         INFIX_TYPE_MAPPING[type(c_argument)],
  48         generate_argument(c_argument.left),
  49         generate_argument(c_argument.right),
  50     )
  51
  52 def generate_function_call(c_function_call):
  53     return '{}({})'.format(
  54         c_function_call.name,
  55         ', '.join(generate_argument(argument) for argument in c_function_call.arguments),
  56     )
  57
  58 def generate_statement(c_function_call_statement):
  59     return '{};'.format(generate_function_call(c_function_call_statement))
  60
  61 def generate(c_program):
  62     template = ENV.get_template('program.c')
  63     return template.render(
  64         builtins=list(sorted(c_program.builtins)),
  65         statements=[generate_statement(statement) for statement in c_program.statements],
  66         standard_libraries=list(sorted(c_program.standard_libraries)),
  67     )
  68
  69 if __name__ == '__main__':
  70     import unittest
  71
  72     unittest.main()