Commit | Line | Data |
---|---|---|
3cdf3fd8 NR |
1 | package be.nikiroo.fanfix_swing.gui.browser; |
2 | ||
3 | import java.awt.BorderLayout; | |
4 | import java.awt.Component; | |
5 | import java.awt.event.ActionEvent; | |
6 | import java.awt.event.ActionListener; | |
7 | import java.util.ArrayList; | |
8 | import java.util.List; | |
9 | import java.util.Map; | |
10 | ||
3cdf3fd8 NR |
11 | import javax.swing.JTree; |
12 | import javax.swing.SwingWorker; | |
3cdf3fd8 NR |
13 | import javax.swing.event.TreeSelectionEvent; |
14 | import javax.swing.event.TreeSelectionListener; | |
3cdf3fd8 NR |
15 | import javax.swing.plaf.basic.BasicTreeUI; |
16 | import javax.swing.tree.DefaultMutableTreeNode; | |
17 | import javax.swing.tree.DefaultTreeCellRenderer; | |
18 | import javax.swing.tree.DefaultTreeModel; | |
19 | import javax.swing.tree.TreeCellRenderer; | |
20 | import javax.swing.tree.TreePath; | |
21 | ||
3cdf3fd8 | 22 | import be.nikiroo.fanfix_swing.gui.SearchBar; |
2a03ecc0 | 23 | import be.nikiroo.fanfix_swing.gui.utils.ListenerPanel; |
3cdf3fd8 | 24 | import be.nikiroo.fanfix_swing.gui.utils.TreeCellSpanner; |
2a03ecc0 | 25 | import be.nikiroo.fanfix_swing.gui.utils.TreeSnapshot; |
3cdf3fd8 NR |
26 | import be.nikiroo.fanfix_swing.gui.utils.UiHelper; |
27 | import be.nikiroo.fanfix_swing.images.IconGenerator; | |
28 | import be.nikiroo.fanfix_swing.images.IconGenerator.Icon; | |
29 | import be.nikiroo.fanfix_swing.images.IconGenerator.Size; | |
30 | ||
2a03ecc0 | 31 | public abstract class BasicTab<T> extends ListenerPanel { |
3cdf3fd8 NR |
32 | private int totalCount = 0; |
33 | private List<String> selectedElements = new ArrayList<String>(); | |
34 | private T data; | |
35 | private String baseTitle; | |
36 | private String listenerCommand; | |
37 | private int index; | |
38 | ||
39 | private JTree tree; | |
2a03ecc0 | 40 | private DefaultMutableTreeNode root; |
3cdf3fd8 NR |
41 | private SearchBar searchBar; |
42 | ||
43 | public BasicTab(int index, String listenerCommand) { | |
44 | setLayout(new BorderLayout()); | |
45 | ||
46 | this.index = index; | |
47 | this.listenerCommand = listenerCommand; | |
48 | ||
49 | data = createEmptyData(); | |
50 | totalCount = 0; | |
51 | ||
2a03ecc0 | 52 | root = new DefaultMutableTreeNode(); |
3cdf3fd8 NR |
53 | |
54 | tree = new JTree(root); | |
55 | tree.setUI(new BasicTreeUI()); | |
32ed6089 NR |
56 | TreeCellSpanner spanner = new TreeCellSpanner(tree, |
57 | generateCellRenderer()); | |
3cdf3fd8 NR |
58 | tree.setCellRenderer(spanner); |
59 | tree.setRootVisible(false); | |
60 | tree.setShowsRootHandles(false); | |
61 | ||
62 | tree.addTreeSelectionListener(new TreeSelectionListener() { | |
63 | @Override | |
64 | public void valueChanged(TreeSelectionEvent e) { | |
65 | List<String> elements = new ArrayList<String>(); | |
66 | TreePath[] paths = tree.getSelectionPaths(); | |
67 | if (paths != null) { | |
68 | for (TreePath path : paths) { | |
69 | String key = path.getLastPathComponent().toString(); | |
70 | elements.add(keyToElement(key)); | |
71 | } | |
72 | } | |
73 | ||
74 | List<String> selectedElements = new ArrayList<String>(); | |
75 | for (String element : elements) { | |
76 | if (!selectedElements.contains(element)) { | |
77 | selectedElements.add(element); | |
78 | } | |
79 | } | |
80 | ||
81 | BasicTab.this.selectedElements = selectedElements; | |
82 | ||
2a03ecc0 | 83 | fireActionPerformed(BasicTab.this.listenerCommand); |
3cdf3fd8 NR |
84 | } |
85 | }); | |
86 | ||
87 | add(UiHelper.scroll(tree), BorderLayout.CENTER); | |
88 | ||
89 | searchBar = new SearchBar(); | |
90 | add(searchBar, BorderLayout.NORTH); | |
91 | searchBar.addActionListener(new ActionListener() { | |
92 | @Override | |
93 | public void actionPerformed(ActionEvent e) { | |
2a03ecc0 | 94 | reloadData(); |
3cdf3fd8 NR |
95 | } |
96 | }); | |
97 | ||
2a03ecc0 NR |
98 | reloadData(); |
99 | } | |
100 | ||
101 | public void reloadData() { | |
77e5ecd4 NR |
102 | final TreeSnapshot snapshot = new TreeSnapshot(tree) { |
103 | @Override | |
104 | protected boolean isSamePath(TreePath oldPath, TreePath newPath) { | |
105 | String oldString = oldPath.toString(); | |
106 | if (oldString.endsWith("/]")) | |
32ed6089 NR |
107 | oldString = oldString.substring(0, oldString.length() - 2) |
108 | + "]"; | |
77e5ecd4 NR |
109 | |
110 | String newString = newPath.toString(); | |
111 | if (newString.endsWith("/]")) | |
32ed6089 NR |
112 | newString = newString.substring(0, newString.length() - 2) |
113 | + "]"; | |
77e5ecd4 NR |
114 | |
115 | return oldString.equals(newString); | |
116 | } | |
117 | }; | |
3cdf3fd8 NR |
118 | SwingWorker<Map<String, List<String>>, Integer> worker = new SwingWorker<Map<String, List<String>>, Integer>() { |
119 | @Override | |
32ed6089 NR |
120 | protected Map<String, List<String>> doInBackground() |
121 | throws Exception { | |
2a03ecc0 NR |
122 | fillData(data); |
123 | return null; | |
3cdf3fd8 NR |
124 | } |
125 | ||
126 | @Override | |
127 | protected void done() { | |
2a03ecc0 NR |
128 | try { |
129 | get(); | |
130 | } catch (Exception e) { | |
131 | // TODO: error | |
132 | } | |
133 | ||
3cdf3fd8 NR |
134 | root.removeAllChildren(); |
135 | totalCount = loadData(root, data, searchBar.getText()); | |
136 | ((DefaultTreeModel) tree.getModel()).reload(); | |
137 | ||
2a03ecc0 NR |
138 | snapshot.apply(); |
139 | ||
140 | fireActionPerformed(listenerCommand); | |
3cdf3fd8 NR |
141 | } |
142 | }; | |
143 | worker.execute(); | |
144 | } | |
145 | ||
146 | /** | |
147 | * The currently selected elements, or an empty list. | |
148 | * | |
149 | * @return the sources (cannot be NULL) | |
150 | */ | |
151 | public List<String> getSelectedElements() { | |
152 | return selectedElements; | |
153 | } | |
154 | ||
155 | public int getTotalCount() { | |
156 | return totalCount; | |
157 | } | |
158 | ||
159 | public String getBaseTitle() { | |
160 | return baseTitle; | |
161 | } | |
162 | ||
163 | public void setBaseTitle(String baseTitle) { | |
164 | this.baseTitle = baseTitle; | |
165 | } | |
166 | ||
167 | public String getTitle() { | |
168 | String title = getBaseTitle(); | |
169 | String count = ""; | |
170 | if (totalCount > 0) { | |
171 | int selected = selectedElements.size(); | |
32ed6089 NR |
172 | count = " (" + (selected > 0 ? selected + "/" : "") + totalCount |
173 | + ")"; | |
3cdf3fd8 NR |
174 | } |
175 | ||
176 | return title + count; | |
177 | } | |
178 | ||
179 | public int getIndex() { | |
180 | return index; | |
181 | } | |
182 | ||
183 | public void unselect() { | |
184 | tree.clearSelection(); | |
185 | } | |
186 | ||
3cdf3fd8 | 187 | protected boolean checkFilter(String filter, String value) { |
32ed6089 NR |
188 | return (filter == null || filter.isEmpty() |
189 | || value.toLowerCase().contains(filter.toLowerCase())); | |
3cdf3fd8 NR |
190 | } |
191 | ||
192 | protected boolean checkFilter(String filter, List<String> list) { | |
193 | for (String value : list) { | |
194 | if (checkFilter(filter, value)) | |
195 | return true; | |
196 | } | |
197 | return false; | |
198 | } | |
199 | ||
200 | protected abstract T createEmptyData(); | |
201 | ||
202 | protected abstract void fillData(T data); | |
203 | ||
204 | protected abstract String keyToElement(String key); | |
205 | ||
206 | protected abstract String keyToDisplay(String key); | |
207 | ||
32ed6089 NR |
208 | protected abstract int loadData(DefaultMutableTreeNode root, T data, |
209 | String filter); | |
3cdf3fd8 NR |
210 | |
211 | private TreeCellRenderer generateCellRenderer() { | |
212 | DefaultTreeCellRenderer renderer = new DefaultTreeCellRenderer() { | |
213 | @Override | |
32ed6089 NR |
214 | public Component getTreeCellRendererComponent(JTree tree, |
215 | Object value, boolean selected, boolean expanded, | |
3cdf3fd8 NR |
216 | boolean leaf, int row, boolean hasFocus) { |
217 | if (value instanceof DefaultMutableTreeNode) { | |
218 | if (((DefaultMutableTreeNode) value).getLevel() > 1) { | |
219 | setLeafIcon(null); | |
220 | setLeafIcon(IconGenerator.get(Icon.empty, Size.x4)); | |
221 | } else { | |
222 | setLeafIcon(IconGenerator.get(Icon.empty, Size.x16)); | |
223 | } | |
224 | } | |
225 | ||
226 | String display = value == null ? "" : value.toString(); | |
77e5ecd4 | 227 | display = keyToDisplay(display); |
3cdf3fd8 | 228 | |
32ed6089 NR |
229 | return super.getTreeCellRendererComponent(tree, display, |
230 | selected, expanded, leaf, row, hasFocus); | |
3cdf3fd8 NR |
231 | } |
232 | }; | |
233 | ||
234 | renderer.setClosedIcon(IconGenerator.get(Icon.arrow_right, Size.x16)); | |
235 | renderer.setOpenIcon(IconGenerator.get(Icon.arrow_down, Size.x16)); | |
236 | renderer.setLeafIcon(IconGenerator.get(Icon.empty, Size.x16)); | |
237 | ||
238 | return renderer; | |
239 | } | |
240 | } |