3 import jalview.bin.Cache;
4 import jalview.ws.seqfetcher.DbSourceProxy;
6 import java.awt.BorderLayout;
7 import java.awt.Component;
8 import java.awt.Container;
9 import java.awt.Dimension;
10 import java.awt.FlowLayout;
11 import java.awt.GridLayout;
12 import java.awt.event.ActionEvent;
13 import java.awt.event.ActionListener;
14 import java.awt.event.KeyEvent;
15 import java.awt.event.KeyListener;
16 import java.util.ArrayList;
17 import java.util.HashSet;
18 import java.util.Hashtable;
19 import java.util.List;
20 import java.util.Vector;
22 import javax.swing.JButton;
23 import javax.swing.JFrame;
24 import javax.swing.JLabel;
25 import javax.swing.JPanel;
26 import javax.swing.JScrollPane;
27 import javax.swing.JTree;
28 import javax.swing.event.TreeSelectionEvent;
29 import javax.swing.event.TreeSelectionListener;
30 import javax.swing.tree.DefaultMutableTreeNode;
31 import javax.swing.tree.DefaultTreeCellRenderer;
32 import javax.swing.tree.DefaultTreeModel;
33 import javax.swing.tree.TreeCellRenderer;
34 import javax.swing.tree.TreeNode;
35 import javax.swing.tree.TreePath;
36 import javax.swing.tree.TreeSelectionModel;
38 public class JDatabaseTree extends JalviewDialog implements KeyListener
40 boolean allowMultiSelections = false;
42 JButton getDatabaseSelectorButton()
44 final JButton viewdbs = new JButton("--- Select Database ---");
45 viewdbs.addActionListener(new ActionListener()
49 public void actionPerformed(ActionEvent arg0)
61 private jalview.ws.SequenceFetcher sfetcher;
63 private JLabel dbstatus, dbstatex;
65 public JDatabaseTree(jalview.ws.SequenceFetcher sfetch)
67 initDialogFrame(this, true, false, "Select Database Retrieval Source",
70 * Dynamically generated database list will need a translation function from
71 * internal source to externally distinct names. UNIPROT and UP_NAME are
72 * identical DB sources, and should be collapsed.
74 DefaultMutableTreeNode tn = null, root = new DefaultMutableTreeNode();
75 Hashtable<String, DefaultMutableTreeNode> source = new Hashtable<String, DefaultMutableTreeNode>();
77 String dbs[] = sfetch.getSupportedDb();
78 Hashtable<String, String> ht = new Hashtable<String, String>();
79 for (int i = 0; i < dbs.length; i++)
81 tn = source.get(dbs[i]);
82 List<DbSourceProxy> srcs = sfetch.getSourceProxy(dbs[i]);
85 source.put(dbs[i], tn = new DefaultMutableTreeNode(dbs[i], true));
87 for (DbSourceProxy dbp : srcs)
89 if (ht.get(dbp.getDbName()) == null)
91 tn.add(new DefaultMutableTreeNode(dbp, false));
92 ht.put(dbp.getDbName(), dbp.getDbName());
96 System.err.println("dupe ig for : " + dbs[i] + " \t"
97 + dbp.getDbName() + " (" + dbp.getDbSource() + ")");
102 for (int i = 0; i < dbs.length; i++)
104 tn = source.get(dbs[i]);
109 if (tn.getChildCount() == 1)
111 DefaultMutableTreeNode ttn = (DefaultMutableTreeNode) tn
113 // remove nodes with only one child
114 tn.setUserObject(ttn.getUserObject());
115 tn.removeAllChildren();
116 source.put(dbs[i], tn);
117 tn.setAllowsChildren(false);
123 svp = new JScrollPane();
124 // svp.setAutoscrolls(true);
125 dbviews = new JTree(new DefaultTreeModel(root, false));
126 dbviews.setCellRenderer(new DbTreeRenderer(this));
128 dbviews.getSelectionModel().setSelectionMode(
129 TreeSelectionModel.SINGLE_TREE_SELECTION);
130 svp.getViewport().setView(dbviews);
131 // svp.getViewport().setMinimumSize(new Dimension(300,200));
132 // svp.setSize(300,250);
133 // JPanel panel=new JPanel();
134 // panel.setSize(new Dimension(350,220));
136 dbviews.addTreeSelectionListener(new TreeSelectionListener()
140 public void valueChanged(TreeSelectionEvent arg0)
142 _setSelectionState();
145 JPanel jc = new JPanel(new BorderLayout()), j = new JPanel(
147 jc.add(svp, BorderLayout.CENTER);
150 // TODO: make the panel stay a fixed size for longest dbname+example set.
151 JPanel dbstat = new JPanel(new GridLayout(2, 1));
152 dbstatus = new JLabel(" "); // set the height correctly for layout
153 dbstatus.setFont(f = JvSwingUtils.getLabelFont(false, true));
154 dbstatus.setSize(new Dimension(290, 50));
155 dbstatex = new JLabel(" ");
157 dbstatex.setSize(new Dimension(290, 50));
158 dbstat.add(dbstatus);
159 dbstat.add(dbstatex);
160 jc.add(dbstat, BorderLayout.SOUTH);
162 // j.setPreferredSize(new Dimension(300,50));
163 add(jc, BorderLayout.CENTER);
166 add(j, BorderLayout.SOUTH);
167 dbviews.addKeyListener(this);
171 private void sortTreeNodes(DefaultMutableTreeNode root)
173 if (root.getChildCount() == 0)
177 int count = root.getChildCount();
178 String[] names = new String[count];
179 DefaultMutableTreeNode[] nodes = new DefaultMutableTreeNode[count];
180 for (int i = 0; i < count; i++)
182 TreeNode node = root.getChildAt(i);
183 if (node instanceof DefaultMutableTreeNode)
185 DefaultMutableTreeNode child = (DefaultMutableTreeNode) node;
187 if (child.getUserObject() instanceof DbSourceProxy)
189 names[i] = ((DbSourceProxy) child.getUserObject()).getDbName()
194 names[i] = ((String) child.getUserObject()).toLowerCase();
195 sortTreeNodes(child);
201 "Implementation Error: Can't reorder this tree. Not DefaultMutableTreeNode.");
204 jalview.util.QuickSort.sort(names, nodes);
205 root.removeAllChildren();
206 for (int i = count - 1; i >= 0; i--)
212 private class DbTreeRenderer extends DefaultTreeCellRenderer implements
217 public DbTreeRenderer(JDatabaseTree me)
222 private Component returnLabel(String txt)
224 JLabel jl = new JLabel(txt);
225 jl.setFont(JvSwingUtils.getLabelFont());
230 public Component getTreeCellRendererComponent(JTree tree, Object value,
231 boolean selected, boolean expanded, boolean leaf, int row,
235 if (value != null && value instanceof DefaultMutableTreeNode)
237 DefaultMutableTreeNode vl = (DefaultMutableTreeNode) value;
238 value = vl.getUserObject();
239 if (value instanceof DbSourceProxy)
241 val = (((DbSourceProxy) value).getDbName());
245 if (value instanceof String)
247 val = ((String) value);
255 return super.getTreeCellRendererComponent(tree, val, selected,
256 expanded, leaf, row, hasFocus);
261 List<DbSourceProxy> oldselection, selection = null;
263 TreePath[] tsel = null, oldtsel = null;
266 protected void raiseClosed()
268 for (ActionListener al : lstners)
270 al.actionPerformed(null);
275 protected void okPressed()
277 _setSelectionState();
282 protected void cancelPressed()
284 selection = oldselection;
286 _revertSelectionState();
290 private void showDialog(Container parent)
292 oldselection = selection;
298 public boolean hasSelection()
300 return selection == null ? false : selection.size() == 0 ? false : true;
303 public List<DbSourceProxy> getSelectedSources()
309 * disable or enable selection handler
311 boolean handleSelections = true;
313 private void _setSelectionState()
315 if (!handleSelections)
319 if (dbviews.getSelectionCount() == 0)
323 tsel = dbviews.getSelectionPaths();
324 boolean forcedFirstChild = false;
325 List<DbSourceProxy> srcs = new ArrayList<DbSourceProxy>();
328 for (TreePath tp : tsel)
330 DefaultMutableTreeNode admt, dmt = (DefaultMutableTreeNode) tp
331 .getLastPathComponent();
332 if (dmt.getUserObject() != null)
334 if (dmt.getUserObject() instanceof DbSourceProxy)
336 srcs.add((DbSourceProxy) dmt.getUserObject());
340 if (allowMultiSelections)
342 srcs.addAll(sfetcher.getSourceProxy((String) dmt
347 if ((admt = dmt.getFirstLeaf()) != null
348 && admt.getUserObject() != null)
350 // modify db selection to just first leaf.
351 if (admt.getUserObject() instanceof DbSourceProxy)
353 srcs.add((DbSourceProxy) admt.getUserObject());
357 srcs.add(sfetcher.getSourceProxy(
358 (String) admt.getUserObject()).get(0));
360 forcedFirstChild = true;
367 updateDbStatus(srcs, forcedFirstChild);
371 private void _revertSelectionState()
373 handleSelections = false;
374 if (selection == null || selection.size() == 0)
376 dbviews.clearSelection();
380 dbviews.setSelectionPaths(tsel);
382 handleSelections = true;
385 private void updateDbStatus(List<DbSourceProxy> srcs,
386 boolean forcedFirstChild)
389 String nm = "", qr = "";
390 for (DbSourceProxy dbs : srcs)
392 String tq = dbs.getTestQuery();
393 nm = dbs.getDbName();
394 if (tq != null && tq.trim().length() > 0 && dbs.isValidReference(tq))
401 if (allowMultiSelections)
403 dbstatus.setText("Selected "
406 + (srcs.size() == 1 ? "" : "s")
408 + (srcs.size() > 0 ? " with " + x + " test quer"
409 + (x == 1 ? "y" : "ies") : "."));
410 dbstatex.setText(" ");
416 dbstatus.setText("Database: " + nm);
419 dbstatex.setText("Example: " + qr);
423 dbstatex.setText(" ");
428 dbstatus.setText(" ");
431 dbstatus.invalidate();
432 dbstatex.invalidate();
435 public String getSelectedItem()
439 return getSelectedSources().get(0).getDbName();
444 public String getExampleQueries()
450 StringBuffer sb = new StringBuffer();
451 HashSet<String> hs = new HashSet<String>();
452 for (DbSourceProxy dbs : getSelectedSources())
454 String tq = dbs.getTestQuery();
465 return sb.toString();
468 List<ActionListener> lstners = new Vector<ActionListener>();
470 public void addActionListener(ActionListener actionListener)
472 lstners.add(actionListener);
475 public void removeActionListener(ActionListener actionListener)
477 lstners.remove(actionListener);
480 public static void main(String args[])
482 Cache.getDasSourceRegistry();
483 JDatabaseTree jdt = new JDatabaseTree(new jalview.ws.SequenceFetcher());
484 JFrame foo = new JFrame();
485 foo.setLayout(new BorderLayout());
486 foo.add(jdt.getDatabaseSelectorButton(), BorderLayout.CENTER);
488 foo.setVisible(true);
490 final Thread us = Thread.currentThread();
491 jdt.addActionListener(new ActionListener()
495 public void actionPerformed(ActionEvent e)
505 } catch (InterruptedException x)
508 if (!jdt.hasSelection())
510 System.out.println("No Selection");
514 System.out.println("Selection: " + jdt.getSelectedItem());
516 for (DbSourceProxy pr : jdt.getSelectedSources())
518 System.out.println("Source " + s++ + ": " + pr.getDbName()
519 + " (" + pr.getDbSource() + ") Version "
520 + pr.getDbVersion() + ". Test:\t" + pr.getTestQuery());
522 System.out.println("Test queries: " + jdt.getExampleQueries());
525 } while (nultimes > 0 && foo.isVisible());
526 foo.setVisible(false);
530 public void keyPressed(KeyEvent arg0)
532 if (!arg0.isConsumed() && arg0.getKeyCode() == KeyEvent.VK_ENTER)
536 if (!arg0.isConsumed() && arg0.getKeyChar() == KeyEvent.VK_ESCAPE)
543 public void keyReleased(KeyEvent arg0)
545 // TODO Auto-generated method stub
550 public void keyTyped(KeyEvent arg0)
552 // TODO Auto-generated method stub