From 5e01b55caf49ddbed0f9abdf8ca6592303001985 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Sat, 31 Oct 2015 11:14:31 +0000 Subject: Add evaluation of some int operations --- .../couscous/backends/python/PrimitiveMethods.java | 35 +++++++++++++++++++++- .../backends/python/PythonCodeGenerator.java | 13 +++----- 2 files changed, 38 insertions(+), 10 deletions(-) (limited to 'src/main/java/org/zwobble/couscous/backends/python') diff --git a/src/main/java/org/zwobble/couscous/backends/python/PrimitiveMethods.java b/src/main/java/org/zwobble/couscous/backends/python/PrimitiveMethods.java index 02db3b2..1da18f9 100644 --- a/src/main/java/org/zwobble/couscous/backends/python/PrimitiveMethods.java +++ b/src/main/java/org/zwobble/couscous/backends/python/PrimitiveMethods.java @@ -6,11 +6,14 @@ import java.util.Optional; import org.zwobble.couscous.ast.TypeName; import org.zwobble.couscous.backends.python.ast.PythonExpressionNode; +import org.zwobble.couscous.values.IntegerValue; import org.zwobble.couscous.values.StringValue; import com.google.common.collect.ImmutableMap; +import com.google.common.collect.ImmutableMap.Builder; import static java.util.Arrays.asList; +import static org.zwobble.couscous.backends.python.ast.PythonAttributeAccessNode.pythonAttributeAccess; import static org.zwobble.couscous.backends.python.ast.PythonCallNode.pythonCall; import static org.zwobble.couscous.backends.python.ast.PythonGetSliceNode.pythonGetSlice; import static org.zwobble.couscous.backends.python.ast.PythonVariableReferenceNode.pythonVariableReference; @@ -29,15 +32,45 @@ public class PrimitiveMethods { .build(); + private static final Map INT_METHODS; + + static { + ImmutableMap.Builder methods = ImmutableMap.builder(); + + addMethod(methods, "add", "__add__"); + addMethod(methods, "subtract", "__sub__"); + addMethod(methods, "multiply", "__mul__"); + addMethod(methods, "divide", "__floordiv__"); + addMethod(methods, "mod", "__mod__"); + + addMethod(methods, "greaterThan", "__gt__"); + addMethod(methods, "greaterThanOrEqual", "__ge__"); + addMethod(methods, "lessThan", "__lt__"); + addMethod(methods, "lessThanOrEqual", "__le__"); + + INT_METHODS = methods.build(); + } + + private static void addMethod( + Builder methods, + String methodName, + String pythonMethodName) { + methods.put(methodName, (receiver, arguments) -> + pythonCall( + pythonAttributeAccess(receiver, pythonMethodName), + arguments)); + } + private static final Map> METHODS = ImmutableMap.>builder() .put(StringValue.REF, STRING_METHODS) + .put(IntegerValue.REF, INT_METHODS) .build(); public static boolean isPrimitive(TypeName type) { return METHODS.containsKey(type); } - + public static Optional getPrimitiveMethod(TypeName type, String methodName) { return Optional.ofNullable(METHODS.get(type)) .flatMap(methodsForType -> Optional.ofNullable(methodsForType.get(methodName))); diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java index 85858ad..a10c54d 100644 --- a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java +++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java @@ -28,6 +28,7 @@ import org.zwobble.couscous.ast.visitors.ExpressionNodeMapper; import org.zwobble.couscous.ast.visitors.NodeVisitorWithEmptyDefaults; import org.zwobble.couscous.ast.visitors.NodeVisitors; import org.zwobble.couscous.ast.visitors.StatementNodeMapper; +import org.zwobble.couscous.backends.python.PrimitiveMethods.PrimitiveMethodGenerator; import org.zwobble.couscous.backends.python.ast.PythonBlock; import org.zwobble.couscous.backends.python.ast.PythonClassNode; import org.zwobble.couscous.backends.python.ast.PythonExpressionNode; @@ -65,8 +66,6 @@ import static org.zwobble.couscous.backends.python.ast.PythonStringLiteralNode.p import static org.zwobble.couscous.backends.python.ast.PythonVariableReferenceNode.pythonVariableReference; public class PythonCodeGenerator { - - public static PythonModuleNode generateCode(ClassNode classNode) { Iterator imports = generateImports(classNode).iterator(); PythonFunctionDefinitionNode constructor = generateConstructor(classNode.getConstructor()); @@ -155,8 +154,6 @@ public class PythonCodeGenerator { } private static class StatementGenerator implements StatementNodeMapper { - - @Override public PythonStatementNode visit(ReturnNode returnNode) { return pythonReturn(generateExpression(returnNode.getValue())); @@ -188,8 +185,6 @@ public class PythonCodeGenerator { private static final ExpressionGenerator EXPRESSION_GENERATOR = new ExpressionGenerator(); private static class ExpressionGenerator implements ExpressionNodeMapper { - - @Override public PythonExpressionNode visit(LiteralNode literal) { return generateCode(literal.getValue()); @@ -217,10 +212,10 @@ public class PythonCodeGenerator { @Override public PythonExpressionNode visit(MethodCallNode methodCall) { - final org.zwobble.couscous.backends.python.ast.PythonExpressionNode receiver = generateExpression(methodCall.getReceiver()); - final java.util.List arguments = generateExpressions(methodCall.getArguments()); + PythonExpressionNode receiver = generateExpression(methodCall.getReceiver()); + List arguments = generateExpressions(methodCall.getArguments()); if (isPrimitive(methodCall.getReceiver())) { - final org.zwobble.couscous.backends.python.PrimitiveMethods.PrimitiveMethodGenerator primitiveMethodGenerator = PrimitiveMethods.getPrimitiveMethod(methodCall.getReceiver().getType(), methodCall.getMethodName()).get(); + PrimitiveMethodGenerator primitiveMethodGenerator = PrimitiveMethods.getPrimitiveMethod(methodCall.getReceiver().getType(), methodCall.getMethodName()).get(); return primitiveMethodGenerator.generate(receiver, arguments); } else { return pythonCall(pythonAttributeAccess(receiver, methodCall.getMethodName()), arguments); -- cgit v1.2.3