java-decompiler: post-import cleanup (classes moved)

1 files changed, 41 insertions, 0 deletions

diff --git a/src/org/jetbrains/java/decompiler/modules/renamer/ClassWrapperNode.java b/src/org/jetbrains/java/decompiler/modules/renamer/ClassWrapperNode.java
new file mode 100644
index 0000000..a3015a8
--- /dev/null
+++ b/src/org/jetbrains/java/decompiler/modules/renamer/ClassWrapperNode.java
@@ -0,0 +1,41 @@
+package org.jetbrains.java.decompiler.modules.renamer;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.jetbrains.java.decompiler.struct.StructClass;
+
+public class ClassWrapperNode {
+
+	private StructClass classStruct;
+	
+	private ClassWrapperNode superclass;
+	
+	private List<ClassWrapperNode> subclasses = new ArrayList<ClassWrapperNode>();
+	
+	public ClassWrapperNode(StructClass cl) {
+		this.classStruct = cl;
+	}
+	
+	public void addSubclass(ClassWrapperNode node) {
+		node.setSuperclass(this);
+		subclasses.add(node);
+	}
+
+	public StructClass getClassStruct() {
+		return classStruct;
+	}
+
+	public List<ClassWrapperNode> getSubclasses() {
+		return subclasses;
+	}
+
+	public ClassWrapperNode getSuperclass() {
+		return superclass;
+	}
+
+	public void setSuperclass(ClassWrapperNode superclass) {
+		this.superclass = superclass;
+	}
+
+}