Commit | Line | Data |
---|---|---|
3cdf3fd8 NR |
1 | package be.nikiroo.fanfix_swing.gui.browser; |
2 | ||
3 | import java.util.HashMap; | |
4 | import java.util.List; | |
5 | import java.util.Map; | |
6 | ||
7 | import javax.swing.tree.DefaultMutableTreeNode; | |
8 | ||
9 | import be.nikiroo.fanfix.Instance; | |
10 | ||
11 | public class SourceTab extends BasicTab<Map<String, List<String>>> { | |
12 | public SourceTab(int index, String listenerCommand) { | |
13 | super(index, listenerCommand); | |
14 | } | |
15 | ||
16 | @Override | |
17 | protected Map<String, List<String>> createEmptyData() { | |
18 | return new HashMap<String, List<String>>(); | |
19 | } | |
20 | ||
21 | @Override | |
22 | protected void fillData(Map<String, List<String>> data) { | |
23 | try { | |
32ed6089 NR |
24 | Map<String, List<String>> sourcesGrouped = Instance.getInstance() |
25 | .getLibrary().getSourcesGrouped(); | |
3cdf3fd8 NR |
26 | for (String group : sourcesGrouped.keySet()) { |
27 | data.put(group, sourcesGrouped.get(group)); | |
28 | } | |
29 | } catch (Exception e) { | |
30 | // TODO | |
31 | e.printStackTrace(); | |
32 | } | |
33 | } | |
34 | ||
35 | @Override | |
36 | protected String keyToElement(String key) { | |
37 | return key.substring(1); | |
38 | } | |
39 | ||
40 | @Override | |
41 | protected String keyToDisplay(String key) { | |
77e5ecd4 NR |
42 | if (key.trim().isEmpty()) { |
43 | return "[*]"; // Root node | |
44 | } | |
45 | ||
3cdf3fd8 NR |
46 | // Get and remove type |
47 | String type = key.substring(0, 1); | |
48 | key = key.substring(1); | |
49 | ||
50 | if (!type.equals(">")) { | |
51 | // Only display the final name | |
52 | int pos = key.toString().lastIndexOf("/"); | |
53 | if (pos >= 0) { | |
54 | key = key.toString().substring(pos + 1); | |
55 | } | |
56 | } | |
57 | ||
58 | if (key.toString().isEmpty()) { | |
59 | key = " "; | |
60 | } | |
61 | ||
62 | return key; | |
63 | } | |
64 | ||
65 | @Override | |
32ed6089 NR |
66 | protected int loadData(DefaultMutableTreeNode root, |
67 | Map<String, List<String>> sourcesGrouped, String filter) { | |
3cdf3fd8 NR |
68 | int count = 0; |
69 | for (String source : sourcesGrouped.keySet()) { | |
32ed6089 NR |
70 | if (checkFilter(filter, source) |
71 | || checkFilter(filter, sourcesGrouped.get(source))) { | |
d6c8579c NR |
72 | List<String> children = sourcesGrouped.get(source); |
73 | boolean hasChildren = (children.size() > 1) | |
32ed6089 NR |
74 | || (children.size() == 1 |
75 | && !children.get(0).trim().isEmpty()); | |
76 | DefaultMutableTreeNode sourceNode = new DefaultMutableTreeNode( | |
77 | ">" + source + (hasChildren ? "/" : "")); | |
3cdf3fd8 | 78 | root.add(sourceNode); |
d6c8579c | 79 | for (String subSource : children) { |
32ed6089 NR |
80 | if (checkFilter(filter, source) |
81 | || checkFilter(filter, subSource)) { | |
3cdf3fd8 | 82 | count = count + 1; |
32ed6089 NR |
83 | if (subSource.isEmpty() |
84 | && sourcesGrouped.get(source).size() > 1) { | |
85 | sourceNode.add( | |
86 | new DefaultMutableTreeNode(" " + source)); | |
3cdf3fd8 | 87 | } else if (!subSource.isEmpty()) { |
32ed6089 NR |
88 | sourceNode.add(new DefaultMutableTreeNode( |
89 | " " + source + "/" + subSource)); | |
3cdf3fd8 NR |
90 | } |
91 | } | |
92 | } | |
93 | } | |
94 | } | |
95 | ||
96 | return count; | |
97 | } | |
98 | } |