blob: 3f1625de3dc5aeaf88a6c03e331fef5fb5f821c1 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
package cuchaz.enigma.gui.elements;
import java.awt.BorderLayout;
import java.awt.event.MouseEvent;
import javax.annotation.Nullable;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreeCellRenderer;
import javax.swing.tree.TreeNode;
import javax.swing.tree.TreePath;
import cuchaz.enigma.analysis.ClassInheritanceTreeNode;
import cuchaz.enigma.analysis.MethodInheritanceTreeNode;
import cuchaz.enigma.gui.Gui;
import cuchaz.enigma.gui.util.GuiUtil;
import cuchaz.enigma.gui.util.SingleTreeSelectionModel;
import cuchaz.enigma.translation.representation.entry.ClassEntry;
import cuchaz.enigma.translation.representation.entry.Entry;
public abstract class AbstractInheritanceTree {
private final JPanel panel = new JPanel(new BorderLayout());
private final JTree tree = new JTree();
protected final Gui gui;
public AbstractInheritanceTree(Gui gui, TreeCellRenderer cellRenderer) {
this.gui = gui;
this.tree.setModel(null);
this.tree.setCellRenderer(cellRenderer);
this.tree.setSelectionModel(new SingleTreeSelectionModel());
this.tree.setShowsRootHandles(true);
this.tree.addMouseListener(GuiUtil.onMouseClick(this::onClick));
this.panel.add(new JScrollPane(this.tree));
}
private void onClick(MouseEvent event) {
if (event.getClickCount() >= 2 && event.getButton() == MouseEvent.BUTTON1) {
// get the selected node
TreePath path = tree.getSelectionPath();
if (path == null) {
return;
}
Object node = path.getLastPathComponent();
if (node instanceof ClassInheritanceTreeNode classNode) {
gui.getController().navigateTo(new ClassEntry(classNode.getObfClassName()));
} else if (node instanceof MethodInheritanceTreeNode methodNode) {
if (methodNode.isImplemented()) {
gui.getController().navigateTo(methodNode.getMethodEntry());
}
}
}
}
public void display(Entry<?> entry) {
this.tree.setModel(null);
DefaultMutableTreeNode node = this.getNodeFor(entry);
if (node != null) {
// show the tree at the root
TreePath path = GuiUtil.getPathToRoot(node);
this.tree.setModel(new DefaultTreeModel((TreeNode) path.getPathComponent(0)));
this.tree.expandPath(path);
this.tree.setSelectionRow(this.tree.getRowForPath(path));
}
this.panel.show();
}
public void retranslateUi() {
}
@Nullable
protected abstract DefaultMutableTreeNode getNodeFor(Entry<?> entry);
protected abstract String getPanelName();
public JPanel getPanel() {
return this.panel;
}
}
|
