blob: 56711880ff797569f13dee07c209d0f6d8f4722f (
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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
package cuchaz.enigma.gui.elements;
import java.awt.BorderLayout;
import java.awt.event.MouseEvent;
import java.util.Collection;
import java.util.Vector;
import javax.swing.JList;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JSplitPane;
import javax.swing.JTree;
import javax.swing.ListSelectionModel;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreeNode;
import javax.swing.tree.TreePath;
import cuchaz.enigma.analysis.ReferenceTreeNode;
import cuchaz.enigma.gui.Gui;
import cuchaz.enigma.gui.TokenListCellRenderer;
import cuchaz.enigma.gui.renderer.CallsTreeCellRenderer;
import cuchaz.enigma.gui.util.GuiUtil;
import cuchaz.enigma.gui.util.ScaleUtil;
import cuchaz.enigma.gui.util.SingleTreeSelectionModel;
import cuchaz.enigma.source.Token;
import cuchaz.enigma.translation.representation.entry.ClassEntry;
import cuchaz.enigma.translation.representation.entry.Entry;
import cuchaz.enigma.translation.representation.entry.FieldEntry;
import cuchaz.enigma.translation.representation.entry.MethodEntry;
public class CallsTree {
private final JPanel panel = new JPanel(new BorderLayout());
private final JTree callsTree = new JTree();
private final JList<Token> tokens = new JList<>();
private final Gui gui;
public CallsTree(Gui gui) {
this.gui = gui;
this.callsTree.setModel(null);
this.callsTree.setCellRenderer(new CallsTreeCellRenderer(gui));
this.callsTree.setSelectionModel(new SingleTreeSelectionModel());
this.callsTree.setShowsRootHandles(true);
this.callsTree.addMouseListener(GuiUtil.onMouseClick(this::onTreeClicked));
this.tokens.setCellRenderer(new TokenListCellRenderer(gui.getController()));
this.tokens.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
this.tokens.setLayoutOrientation(JList.VERTICAL);
this.tokens.addMouseListener(GuiUtil.onMouseClick(this::onTokenClicked));
this.tokens.setPreferredSize(ScaleUtil.getDimension(0, 200));
this.tokens.setMinimumSize(ScaleUtil.getDimension(0, 200));
JSplitPane contentPane = new JSplitPane(JSplitPane.VERTICAL_SPLIT, true, new JScrollPane(this.callsTree), new JScrollPane(this.tokens));
contentPane.setResizeWeight(1); // let the top side take all the slack
contentPane.resetToPreferredSizes();
this.panel.add(contentPane, BorderLayout.CENTER);
}
public void showCalls(Entry<?> entry, boolean recurse) {
TreeNode node = null;
if (entry instanceof ClassEntry classEntry) {
node = this.gui.getController().getClassReferences(classEntry);
} else if (entry instanceof FieldEntry fieldEntry) {
node = this.gui.getController().getFieldReferences(fieldEntry);
} else if (entry instanceof MethodEntry methodEntry) {
node = this.gui.getController().getMethodReferences(methodEntry, recurse);
}
this.callsTree.setModel(new DefaultTreeModel(node));
this.panel.show();
}
public void showTokens(Collection<Token> tokens) {
this.tokens.setListData(new Vector<>(tokens));
this.tokens.setSelectedIndex(0);
}
public void clearTokens() {
this.tokens.setListData(new Vector<>());
}
@SuppressWarnings("unchecked")
private void onTreeClicked(MouseEvent event) {
if (event.getClickCount() >= 2 && event.getButton() == MouseEvent.BUTTON1) {
// get the selected node
TreePath path = this.callsTree.getSelectionPath();
if (path == null) {
return;
}
Object node = path.getLastPathComponent();
if (node instanceof ReferenceTreeNode referenceNode) {
if (referenceNode.getReference() != null) {
this.gui.getController().navigateTo(referenceNode.getReference());
} else {
this.gui.getController().navigateTo(referenceNode.getEntry());
}
}
}
}
private void onTokenClicked(MouseEvent event) {
if (event.getClickCount() == 2) {
Token selected = this.tokens.getSelectedValue();
if (selected != null) {
this.gui.openClass(this.gui.getController().getTokenHandle().getRef()).navigateToToken(selected);
}
}
}
public void retranslateUi() {
}
public JPanel getPanel() {
return this.panel;
}
}
|
