import collections
+import normalization
+
+CPSBuiltinExpression = collections.namedtuple(
+ 'CPSBuiltinExpression',
+ (
+ 'symbol',
+ ),
+)
+
+CPSFunctionCallExpression = collections.namedtuple(
+ 'CPSFunctionCallExpression',
+ (
+ 'metadata',
+ 'function_expression',
+ 'argument_count',
+ ),
+)
+
+CPSIntegerLiteralExpression = collections.namedtuple(
+ 'CPSIntegerLiteralExpression',
+ (
+ 'integer',
+ ),
+)
+
+CPSLambdaExpression = collections.namedtuple(
+ 'CPSLambdaExpression',
+ (
+ 'name',
+ 'argument_name_list',
+ 'statement_list',
+ ),
+)
+
+CPSListConstructExpression = collections.namedtuple(
+ 'CPSListConstructExpression',
+ (
+ 'allocate',
+ ),
+)
+
+CPSStringLiteralExpression = collections.namedtuple(
+ 'CPSStringLiteralExpression',
+ (
+ 'string',
+ ),
+)
+
+CPSStructureLiteralExpression = collections.namedtuple(
+ 'CPSStructureLiteralExpression',
+ (
+ 'field_count',
+ ),
+)
+
+CPSSymbolExpression = collections.namedtuple(
+ 'CPSSymbolExpression',
+ (
+ 'symbol',
+ ),
+)
+
+CPSSymbolLiteralExpression = collections.namedtuple(
+ 'CPSSymbolLiteralExpression',
+ (
+ 'symbol',
+ ),
+)
+
+CPSVariableExpression = collections.namedtuple(
+ 'CPSVariableExpression',
+ (
+ 'variable',
+ ),
+)
+
+CPSArrayVariableInitializationStatement = collections.namedtuple(
+ 'CPSArrayVariableInitializationStatement',
+ (
+ 'variable',
+ 'items',
+ ),
+)
+
+CPSAssignmentStatement = collections.namedtuple(
+ 'CPSAssignmentStatement',
+ (
+ 'target',
+ 'expression',
+ ),
+)
+
+CPSExpressionStatement = collections.namedtuple(
+ 'CPSExpressionStatement',
+ (
+ 'expression',
+ ),
+)
+
+CPSIfElseExpression = collections.namedtuple(
+ 'CPSIfElseExpression',
+ (
+ 'condition_expression',
+ 'if_statement_list',
+ 'else_statement_list',
+ ),
+)
+
+CPSPushStatement = collections.namedtuple(
+ 'CPSPushStatement',
+ (
+ 'expression',
+ ),
+)
+
+CPSVariableInitializationStatement = collections.namedtuple(
+ 'CPSVariableInitializationStatement',
+ (
+ 'variable',
+ 'expression',
+ ),
+)
+
+CPSSymbolArrayVariableInitializationStatement = collections.namedtuple(
+ 'CPSSymbolArrayVariableInitializationStatement',
+ (
+ 'variable',
+ 'symbol_list',
+ ),
+)
+
CPSProgram = collections.namedtuple(
'CPSProgram',
(
),
)
+def convert_builtin_expression(expression):
+ return CPSBuiltinExpression(symbol=expression.symbol)
+
+def convert_function_call_expression(expression):
+ return CPSFunctionCallExpression(
+ metadata=expression.metadata,
+ function_expression=convert_expression(expression.function_expression),
+ argument_count=expression.argument_count,
+ )
+
+def convert_integer_literal_expression(expression):
+ return CPSIntegerLiteralExpression(integer=expression.integer)
+
+def convert_lambda_expression(expression):
+ return CPSLambdaExpression(
+ name=expression.name,
+ argument_name_list=expression.argument_name_list,
+ statement_list=tuple(convert_statement(s) for s in expression.statement_list),
+ )
+
+def convert_list_construct_expression(expression):
+ return CPSListConstructExpression(allocate=expression.allocate)
+
+def convert_string_literal_expression(expression):
+ return CPSStringLiteralExpression(string=expression.string)
+
+def convert_structure_literal_expression(expression):
+ return CPSStructureLiteralExpression(
+ field_count=expression.field_count,
+ )
+
+def convert_symbol_expression(expression):
+ return CPSSymbolExpression(symbol=expression.symbol)
+
+def convert_symbol_literal_expression(expression):
+ return CPSSymbolLiteralExpression(symbol=expression.symbol)
+
+def convert_variable_expression(expression):
+ return CPSVariableExpression(variable=expression.variable)
+
+def convert_expression(expression):
+ return {
+ normalization.NormalBuiltinExpression: convert_builtin_expression,
+ normalization.NormalFunctionCallExpression: convert_function_call_expression,
+ normalization.NormalIfElseExpression: convert_if_else_expression,
+ normalization.NormalIntegerLiteralExpression: convert_integer_literal_expression,
+ normalization.NormalLambdaExpression: convert_lambda_expression,
+ normalization.NormalListConstructExpression: convert_list_construct_expression,
+ normalization.NormalStringLiteralExpression: convert_string_literal_expression,
+ normalization.NormalStructureLiteralExpression: convert_structure_literal_expression,
+ normalization.NormalSymbolExpression: convert_symbol_expression,
+ normalization.NormalSymbolLiteralExpression: convert_symbol_literal_expression,
+ normalization.NormalVariableExpression: convert_variable_expression,
+ }[type(expression)](expression)
+
+def convert_assignment_statement(statement):
+ return CPSAssignmentStatement(
+ target=statement.target,
+ expression=convert_expression(statement.expression),
+ )
+
+def convert_expression_statement(statement):
+ return CPSExpressionStatement(
+ expression=convert_expression(statement.expression),
+ )
+
+def convert_if_else_expression(statement):
+ if_statement_list=tuple(convert_statement(s) for s in statement.if_statement_list)
+ else_statement_list=tuple(convert_statement(s) for s in statement.else_statement_list)
+
+ return CPSIfElseExpression(
+ condition_expression=convert_expression(statement.condition_expression),
+ if_statement_list=if_statement_list,
+ else_statement_list=else_statement_list,
+ )
+
+def convert_push_statement(statement):
+ return CPSPushStatement(
+ expression=convert_expression(statement.expression),
+ )
+
+def convert_variable_initialization_statement(statement):
+ return CPSVariableInitializationStatement(
+ variable=statement.variable,
+ expression=convert_expression(statement.expression),
+ )
+
+def convert_statement(statement):
+ return {
+ normalization.NormalAssignmentStatement: convert_assignment_statement,
+ normalization.NormalExpressionStatement: convert_expression_statement,
+ normalization.NormalPushStatement: convert_push_statement,
+ normalization.NormalVariableInitializationStatement: convert_variable_initialization_statement,
+ }[type(statement)](statement)
+
+def convert_statement_list(statement_list):
+ return tuple(convert_statement(s) for s in statement_list)
+
+
def convert(program):
return CPSProgram(
- statement_list=program.statement_list,
+ statement_list=convert_statement_list(program.statement_list),
)