import jinja2
+import transformation
+
ENV = jinja2.Environment(
autoescape=jinja2.select_autoescape([]),
loader=jinja2.FileSystemLoader('templates'),
trim_blocks=True,
)
-def generate_argument(c_string_literal):
- def c_escape(ch):
- return {
- '\n': r'\n',
- '"': r'\"',
- '\\': r'\\',
- }.get(ch, ch)
+def generate_integer_literal(c_integer_literal):
+ return 'integerLiteral({})'.format(c_integer_literal.value)
+
+def generate_string_literal(c_string_literal):
+ return 'stringLiteral(STRING_LITERAL_LIST[{}])'.format(c_string_literal.index)
+
+def generate_symbol_expression(symbol_expression):
+ return 'Environment_get(environment, SYMBOL_LIST[{}] /* symbol: {} */)'.format(
+ symbol_expression.symbol_list_index,
+ symbol_expression.symbol,
+ )
+
+def generate_variable_expression(expression):
+ return expression.variable
+
+def generate_expression(expression):
+ if isinstance(expression, transformation.CNegationExpression):
+ return generate_negation_expression(expression)
+
+ if isinstance(expression, transformation.CFunctionCallExpression):
+ return generate_function_call(expression)
+
+ LITERAL_TYPE_MAPPING = {
+ transformation.CIntegerLiteral: generate_integer_literal,
+ transformation.CStringLiteral: generate_string_literal,
+ transformation.CSymbolExpression: generate_symbol_expression,
+ }
+
+ if type(expression) in LITERAL_TYPE_MAPPING:
+ return LITERAL_TYPE_MAPPING[type(expression)](expression)
+
+ if isinstance(expression, transformation.CFunctionCallForFurInfixOperator):
+ return 'operator${}({}, {})'.format(
+ expression.name,
+ generate_expression(expression.left),
+ generate_expression(expression.right),
+ )
+
+ return {
+ transformation.CVariableExpression: generate_variable_expression,
+ }[type(expression)](expression)
+
+def generate_negation_expression(c_negation_expression):
+ return 'operator$negate({})'.format(
+ generate_expression(c_negation_expression.value)
+ )
+
+def generate_function_call(function_call):
+ # TODO This gets called twice, which is really inefficient--normalization would also allow other clauses besides a variable reference
+ # TODO This should no longer be called "name", as it can be an expression of a few types
+ # TODO Check the type of the things being called
+ get_closure_clause = generate_expression(function_call.name)
+ return '{}.instance.closure.call(environmentPool, {}.instance.closure.closed, {}, {})'.format(
+ get_closure_clause,
+ get_closure_clause,
+ function_call.argument_count,
+ # TODO This is just a single item containing a reference to the items list--make that clearer
+ generate_expression(function_call.argument_items),
+ )
+
+def generate_expression_statement(statement):
+ # TODO Do we need to garbage collect the results of arbitrary statements?
+ return '{};'.format(generate_expression(statement.expression))
+
+def generate_symbol_assignment_statement(statement):
+ return 'Environment_set(environment, SYMBOL_LIST[{}] /* symbol: {} */, {});'.format(
+ statement.target_symbol_list_index,
+ statement.target,
+ generate_expression(statement.expression),
+ )
+
+def generate_array_variable_initialization_statement(statement):
+ return 'Object {}[] = {{ {} }};'.format(
+ statement.variable,
+ ', '.join(generate_expression(i) for i in statement.items),
+ )
+
+def generate_variable_initialization_statement(statement):
+ return 'Object {} = {};'.format(
+ statement.variable,
+ generate_expression(statement.expression),
+ )
+
+def generate_variable_reassignment_statement(statement):
+ return '{} = {};'.format(
+ statement.variable,
+ generate_expression(statement.expression),
+ )
+
+
+def indent(s):
+ return '\n'.join(' ' * 2 + l for l in s.split('\n'))
- return '"{}"'.format(
- ''.join(c_escape(ch for ch in c_string_literal.value)),
+def generate_if_else_statement(statement):
+ # TODO Check that the argument is boolean
+ condition_expression = '{}.instance.boolean'.format(
+ generate_expression(statement.condition_expression),
)
-def generate_statement(c_function_call_statement):
- return '{}({});'.format(
- c_function_call_statement.name,
- ', '.join(generate_argument(argument) for argument in c_function_call_statement.arguments),
+ if len(statement.if_statements) == 0:
+ condition_expression = '!({})'.format(condition_expression)
+ if_statements = statement.else_statements
+ else_statements = ()
+ else:
+ if_statements = statement.if_statements
+ else_statements = statement.else_statements
+
+ generated_if_clause = 'if({})'.format(condition_expression)
+
+ if len(if_statements) == 0:
+ generated_if_statements = ';'
+ else:
+ generated_if_statements = indent('\n{{\n{}\n}}'.format(
+ indent('\n'.join(generate_statement(s) for s in if_statements)),
+ ))
+
+ if len(else_statements) == 0:
+ generated_else_statements = ''
+ else:
+ generated_else_statements = indent('\nelse\n{{\n{}\n}}'.format(
+ indent('\n'.join(generate_statement(s) for s in else_statements)),
+ ))
+
+ return generated_if_clause + generated_if_statements + generated_else_statements
+
+def generate_function_declaration(statement):
+ return 'Environment_set(environment, "{}", (Object){{ CLOSURE, (Instance)(Closure){{ environment, user${}$implementation }} }});'.format(statement.name, statement.name)
+
+def generate_statement(statement):
+ return {
+ transformation.CExpressionStatement: generate_expression_statement,
+ transformation.CFunctionDeclaration: generate_function_declaration,
+ transformation.CIfElseStatement: generate_if_else_statement,
+ transformation.CSymbolAssignmentStatement: generate_symbol_assignment_statement,
+ transformation.CArrayVariableInitializationStatement: generate_array_variable_initialization_statement,
+ transformation.CVariableInitializationStatement: generate_variable_initialization_statement,
+ transformation.CVariableReassignmentStatement: generate_variable_reassignment_statement,
+ }[type(statement)](statement)
+
+def generate_function_definition(definition):
+ template = ENV.get_template('function_definition.c')
+ return template.render(
+ name=definition.name,
+ argument_name_list=definition.argument_name_list,
+ statement_list=list(generate_statement(s) for s in definition.statement_list),
)
+ return definition
-def generate(c_program):
+def generate(program):
template = ENV.get_template('program.c')
return template.render(
- builtins=list(sorted(c_program.builtins)),
- statements=[generate_statement(statement) for statement in c_program.statements],
- standard_libraries=set(['stdio.h']),
+ builtins=tuple(sorted(program.builtin_set)),
+ function_definition_list=list(generate_function_definition(fd) for fd in program.function_definition_list),
+ infix_declarations=program.operator_declarations,
+ statements=list(generate_statement(s) for s in program.statements),
+ standard_libraries=list(sorted(program.standard_libraries)),
+ string_literal_list=program.string_literal_list,
+ symbol_list=program.symbol_list,
)
if __name__ == '__main__':
projects / fur / blobdiff