From 428574f8a44bdd21bbe408dde6d1df2b3db1efae Mon Sep 17 00:00:00 2001
From: Thog
Date: Tue, 6 Sep 2016 12:28:05 +0200
Subject: Generify ProcyonEntryFactory (Fix #18)

---
 .../cuchaz/enigma/analysis/SourceIndexBehaviorVisitor.java | 14 +++++---------
 1 file changed, 5 insertions(+), 9 deletions(-)

(limited to 'src/main/java/cuchaz/enigma/analysis')

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);
     }
-- 
cgit v1.2.3