From c0fc10f59a858a417dd5efc61db339671f97c968 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 29 Aug 2016 16:20:06 +0100
Subject: Desugar all types at once

---
 .../couscous/backends/python/PythonBackend.java    | 23 ++++++++++++++++++----
 .../backends/python/PythonCodeGenerator.java       | 15 +-------------
 2 files changed, 20 insertions(+), 18 deletions(-)

(limited to 'src/main/java/org/zwobble/couscous/backends/python')

diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonBackend.java b/src/main/java/org/zwobble/couscous/backends/python/PythonBackend.java
index 2a3701e..3df2d98 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonBackend.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonBackend.java
@@ -4,6 +4,10 @@ import com.google.common.collect.ImmutableList;
 import com.google.common.io.Resources;
 import org.zwobble.couscous.Backend;
 import org.zwobble.couscous.ast.TypeNode;
+import org.zwobble.couscous.ast.visitors.NodeTransformer;
+import org.zwobble.couscous.transforms.DesugarForEachToFor;
+import org.zwobble.couscous.transforms.DesugarForToWhile;
+import org.zwobble.couscous.transforms.DesugarSwitchToIfElse;
 
 import java.io.File;
 import java.io.IOException;
@@ -34,8 +38,8 @@ public class PythonBackend implements Backend {
     
     @Override
     public void compile(List<TypeNode> classes) throws IOException {
-        for (TypeNode classNode : classes) {
-            compileClass(classNode);
+        for (TypeNode classNode : desugar(classes)) {
+            writeClass(classNode);
         }
         for (String runtimeFile :  RUNTIME_FILES) {
             String path = relativePathForModule(runtimeFile);
@@ -44,7 +48,18 @@ public class PythonBackend implements Backend {
                 Resources.toString(Resources.getResource("org/zwobble/couscous/backends/python/runtime/" + path), StandardCharsets.UTF_8));
         }
     }
-    
+
+    private List<TypeNode> desugar(List<TypeNode> classes) {
+        return NodeTransformer.applyAll(
+            list(
+                DesugarSwitchToIfElse.transformer(),
+                DesugarForEachToFor.transformer(),
+                DesugarForToWhile.transformer()
+            ),
+            classes
+        );
+    }
+
     private Path destinationPathForModule(String moduleName) {
         return root.resolve(packageName).resolve(relativePathForModule(moduleName));
     }
@@ -53,7 +68,7 @@ public class PythonBackend implements Backend {
         return moduleName.replace(".", File.separator) + ".py";
     }
     
-    private void compileClass(TypeNode classNode) throws IOException {
+    private void writeClass(TypeNode classNode) throws IOException {
         writeModule(
             classNode.getName().getQualifiedName(),
             serialize(generateCode(classNode)));
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 2190f66..0ce02e3 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
@@ -7,12 +7,8 @@ import com.google.common.collect.Iterators;
 import org.zwobble.couscous.ast.*;
 import org.zwobble.couscous.ast.structure.NodeStructure;
 import org.zwobble.couscous.ast.visitors.DynamicNodeMapper;
-import org.zwobble.couscous.ast.visitors.NodeTransformer;
 import org.zwobble.couscous.backends.naming.Names;
 import org.zwobble.couscous.backends.python.ast.*;
-import org.zwobble.couscous.transforms.DesugarForEachToFor;
-import org.zwobble.couscous.transforms.DesugarForToWhile;
-import org.zwobble.couscous.transforms.DesugarSwitchToIfElse;
 import org.zwobble.couscous.types.ScalarType;
 import org.zwobble.couscous.types.Type;
 import org.zwobble.couscous.types.Types;
@@ -60,16 +56,7 @@ import static org.zwobble.couscous.util.ExtraIterables.*;
 import static org.zwobble.couscous.util.ExtraLists.*;
 
 public class PythonCodeGenerator {
-    public static PythonModuleNode generateCode(TypeNode sugaredTypeNode) {
-        NodeTransformer switchTransformer = DesugarSwitchToIfElse.transformer();
-        NodeTransformer forTransformer = DesugarForToWhile.transformer();
-        NodeTransformer forEachTransformer = DesugarForEachToFor.transformer();
-        TypeNode typeNode = forTransformer.transformTypeDeclaration(
-            forEachTransformer.transformTypeDeclaration(
-                switchTransformer.transformTypeDeclaration(sugaredTypeNode)
-            )
-        );
-
+    public static PythonModuleNode generateCode(TypeNode typeNode) {
         Iterator<PythonImportNode> imports = generateImports(typeNode).iterator();
 
         PythonImportNode internalsImport = pythonImport(
-- 
cgit v1.2.3