From 0970f5a5074da34fffaea70f3b42d012edb501c5 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Thu, 14 Jul 2016 22:50:55 +0100
Subject: Simplify dynamic visitor creation

---
 .../couscous/backends/python/PythonCodeGenerator.java        | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

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

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 be3b81f..993b6aa 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
@@ -4,7 +4,6 @@ import com.google.common.base.Strings;
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.Iterables;
 import com.google.common.collect.Iterators;
-import net.bytebuddy.description.type.TypeDescription;
 import org.zwobble.couscous.ast.*;
 import org.zwobble.couscous.ast.structure.NodeStructure;
 import org.zwobble.couscous.ast.visitors.DynamicNodeMapper;
@@ -114,17 +113,14 @@ public class PythonCodeGenerator {
 
     private static Set<ScalarType> findReferencedClasses(TypeNode classNode) {
         return NodeStructure.descendantNodes(classNode)
-            .flatMap(findDirectlyReferencedClasses)
+            .flatMap(FindDirectlyReferencedClasses.VISITOR)
             .collect(Collectors.toSet());
     }
 
-    private static final Function<Node, Stream<ScalarType>> findDirectlyReferencedClasses =
-        DynamicNodeMapper.<FindDirectlyReferencedClasses, Stream<ScalarType>>build(
-            FindDirectlyReferencedClasses.class,
-            new TypeDescription.ForLoadedType(Stream.class),
-            "visit"
-        ).instantiate(new FindDirectlyReferencedClasses());
     public static class FindDirectlyReferencedClasses {
+        private static final Function<Node, Stream<ScalarType>> VISITOR =
+            DynamicNodeMapper.instantiate(new FindDirectlyReferencedClasses(), "visit");
+
         public Stream<ScalarType> visit(Node node) {
             return Stream.empty();
         }
-- 
cgit v1.2.3