From 2b2249e873c4adfd2dd6e8f1f2489ccd9f6aa021 Mon Sep 17 00:00:00 2001
From: gegy1000
Date: Sat, 19 May 2018 17:02:46 +0200
Subject: Initial port to ASM

---
 .../java/cuchaz/enigma/analysis/ClassImplementationsTreeNode.java | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

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

diff --git a/src/main/java/cuchaz/enigma/analysis/ClassImplementationsTreeNode.java b/src/main/java/cuchaz/enigma/analysis/ClassImplementationsTreeNode.java
index f2fb2f8..d0e1ddb 100644
--- a/src/main/java/cuchaz/enigma/analysis/ClassImplementationsTreeNode.java
+++ b/src/main/java/cuchaz/enigma/analysis/ClassImplementationsTreeNode.java
@@ -21,8 +21,8 @@ import java.util.List;
 
 public class ClassImplementationsTreeNode extends DefaultMutableTreeNode {
 
-	private Translator deobfuscatingTranslator;
-	private ClassEntry entry;
+	private final Translator deobfuscatingTranslator;
+	private final ClassEntry entry;
 
 	public ClassImplementationsTreeNode(Translator deobfuscatingTranslator, ClassEntry entry) {
 		this.deobfuscatingTranslator = deobfuscatingTranslator;
@@ -31,7 +31,7 @@ public class ClassImplementationsTreeNode extends DefaultMutableTreeNode {
 
 	public static ClassImplementationsTreeNode findNode(ClassImplementationsTreeNode node, MethodEntry entry) {
 		// is this the node?
-		if (node.entry.equals(entry.getClassEntry())) {
+		if (node.entry.equals(entry.getOwnerClassEntry())) {
 			return node;
 		}
 
@@ -50,7 +50,7 @@ public class ClassImplementationsTreeNode extends DefaultMutableTreeNode {
 	}
 
 	public String getDeobfClassName() {
-		return this.deobfuscatingTranslator.translateClass(this.entry.getClassName());
+		return this.deobfuscatingTranslator.getTranslatedClass(entry).getClassName();
 	}
 
 	@Override
-- 
cgit v1.2.3