Generify ProcyonEntryFactory (Fix #18)

1 files changed, 5 insertions, 9 deletions

diff --git a/src/main/java/cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java b/src/main/java/cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java
index 2c84cd6..9a98b91 100644
--- a/src/main/java/cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java
+++ b/src/main/java/cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java
@@ -96,15 +96,11 @@ public class SourceIndexBehaviorVisitor extends SourceIndexVisitor {
     @Override
     public Void visitParameterDeclaration(ParameterDeclaration node, SourceIndex index) {
         ParameterDefinition def = node.getUserData(Keys.PARAMETER_DEFINITION);
-        if (def.getMethod() instanceof MemberReference) {
-            BehaviorEntry behaviorEntry;
-            if (def.getMethod() instanceof MethodDefinition)
-                behaviorEntry = ProcyonEntryFactory.getBehaviorEntry((MethodDefinition) def.getMethod());
-            else
-                behaviorEntry = ProcyonEntryFactory.getMethodEntry((MemberReference) def.getMethod()); // Generic definition
-            ArgumentEntry argumentEntry = new ArgumentEntry(behaviorEntry, argumentPosition++, node.getName());
-            index.addDeclaration(node.getNameToken(), argumentEntry);
-        }
+        if (def.getMethod() instanceof MemberReference)
+            if (def.getMethod() instanceof MethodReference)
+                index.addDeclaration(node.getNameToken(),
+                        new ArgumentEntry(ProcyonEntryFactory.getBehaviorEntry((MethodReference) def.getMethod()),
+                                argumentPosition++, node.getName()));
 
         return recurse(node, index);
     }