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.TreePath;
35 import javax.swing.tree.TreeSelectionModel;
37 public class JDatabaseTree extends JalviewDialog implements KeyListener
39 boolean allowMultiSelections = false;
41 JButton getDatabaseSelectorButton()
43 final JButton viewdbs = new JButton("--- Select Database ---");
44 viewdbs.addActionListener(new ActionListener()
48 public void actionPerformed(ActionEvent arg0)
60 private jalview.ws.SequenceFetcher sfetcher;
62 private JLabel dbstatus, dbstatex;
64 public JDatabaseTree(jalview.ws.SequenceFetcher sfetch)
66 initDialogFrame(this, true, false, "Select Database Retrieval Source",
69 * Dynamically generated database list will need a translation function from
70 * internal source to externally distinct names. UNIPROT and UP_NAME are
71 * identical DB sources, and should be collapsed.
73 DefaultMutableTreeNode tn = null, root = new DefaultMutableTreeNode();
74 Hashtable<String, DefaultMutableTreeNode> source = new Hashtable<String, DefaultMutableTreeNode>();
76 String dbs[] = sfetch.getSupportedDb();
77 Hashtable<String,String> ht = new Hashtable<String,String>();
78 for (int i = 0; i < dbs.length; i++)
80 tn = source.get(dbs[i]);
81 List<DbSourceProxy> srcs = sfetch.getSourceProxy(dbs[i]);
84 source.put(dbs[i], tn = new DefaultMutableTreeNode(dbs[i], true));
86 for (DbSourceProxy dbp : srcs)
88 if (ht.get(dbp.getDbName()) == null)
90 tn.add(new DefaultMutableTreeNode(dbp, false));
91 ht.put(dbp.getDbName(), dbp.getDbName());
95 System.err.println("dupe ig for : " + dbs[i] + " \t"
96 + dbp.getDbName() + " (" + dbp.getDbSource() + ")");
101 for (int i = 0; i < dbs.length; i++)
103 tn = source.get(dbs[i]);
108 if (tn.getChildCount() == 1)
110 DefaultMutableTreeNode ttn = (DefaultMutableTreeNode) tn
112 // remove nodes with only one child
113 tn.setUserObject(ttn.getUserObject());
114 tn.removeAllChildren();
115 source.put(dbs[i], tn);
116 tn.setAllowsChildren(false);
120 svp = new JScrollPane();
121 // svp.setAutoscrolls(true);
122 dbviews = new JTree(new DefaultTreeModel(root, false));
123 dbviews.setCellRenderer(new DbTreeRenderer(this));
125 dbviews.getSelectionModel().setSelectionMode(
126 TreeSelectionModel.SINGLE_TREE_SELECTION);
127 svp.getViewport().setView(dbviews);
128 // svp.getViewport().setMinimumSize(new Dimension(300,200));
129 // svp.setSize(300,250);
130 // JPanel panel=new JPanel();
131 // panel.setSize(new Dimension(350,220));
133 dbviews.addTreeSelectionListener(new TreeSelectionListener()
137 public void valueChanged(TreeSelectionEvent arg0)
139 _setSelectionState();
142 JPanel jc = new JPanel(new BorderLayout()), j = new JPanel(
144 jc.add(svp, BorderLayout.CENTER);
147 // TODO: make the panel stay a fixed size for longest dbname+example set.
148 JPanel dbstat = new JPanel(new GridLayout(2, 1));
149 dbstatus = new JLabel(" "); // set the height correctly for layout
150 dbstatus.setFont(f = JvSwingUtils.getLabelFont(false, true));
151 dbstatus.setSize(new Dimension(290, 50));
152 dbstatex = new JLabel(" ");
154 dbstatex.setSize(new Dimension(290, 50));
155 dbstat.add(dbstatus);
156 dbstat.add(dbstatex);
157 jc.add(dbstat, BorderLayout.SOUTH);
159 // j.setPreferredSize(new Dimension(300,50));
160 add(jc, BorderLayout.CENTER);
163 add(j, BorderLayout.SOUTH);
164 dbviews.addKeyListener(this);
168 private class DbTreeRenderer extends DefaultTreeCellRenderer implements
173 public DbTreeRenderer(JDatabaseTree me)
178 private Component returnLabel(String txt)
180 JLabel jl = new JLabel(txt);
181 jl.setFont(JvSwingUtils.getLabelFont());
186 public Component getTreeCellRendererComponent(JTree tree, Object value,
187 boolean selected, boolean expanded, boolean leaf, int row,
191 if (value != null && value instanceof DefaultMutableTreeNode)
193 DefaultMutableTreeNode vl = (DefaultMutableTreeNode) value;
194 value = vl.getUserObject();
195 if (value instanceof DbSourceProxy)
197 val = (((DbSourceProxy) value).getDbName());
201 if (value instanceof String)
203 val = ((String) value);
211 return super.getTreeCellRendererComponent(tree, val, selected,
212 expanded, leaf, row, hasFocus);
217 List<DbSourceProxy> oldselection, selection = null;
219 TreePath[] tsel = null, oldtsel = null;
222 protected void raiseClosed()
224 for (ActionListener al : lstners)
226 al.actionPerformed(null);
231 protected void okPressed()
233 _setSelectionState();
238 protected void cancelPressed()
240 selection = oldselection;
242 _revertSelectionState();
246 private void showDialog(Container parent)
248 oldselection = selection;
254 public boolean hasSelection()
256 return selection == null ? false : selection.size() == 0 ? false : true;
259 public List<DbSourceProxy> getSelectedSources()
265 * disable or enable selection handler
267 boolean handleSelections = true;
269 private void _setSelectionState()
271 if (!handleSelections)
275 if (dbviews.getSelectionCount() == 0)
279 tsel = dbviews.getSelectionPaths();
280 boolean forcedFirstChild = false;
281 List<DbSourceProxy> srcs = new ArrayList<DbSourceProxy>();
284 for (TreePath tp : tsel)
286 DefaultMutableTreeNode admt, dmt = (DefaultMutableTreeNode) tp
287 .getLastPathComponent();
288 if (dmt.getUserObject() != null)
290 if (dmt.getUserObject() instanceof DbSourceProxy)
292 srcs.add((DbSourceProxy) dmt.getUserObject());
296 if (allowMultiSelections)
298 srcs.addAll(sfetcher.getSourceProxy((String) dmt
303 if ((admt = dmt.getFirstLeaf()) != null
304 && admt.getUserObject() != null)
306 // modify db selection to just first leaf.
307 if (admt.getUserObject() instanceof DbSourceProxy)
309 srcs.add((DbSourceProxy) admt.getUserObject());
313 srcs.add(sfetcher.getSourceProxy(
314 (String) admt.getUserObject()).get(0));
316 forcedFirstChild = true;
323 updateDbStatus(srcs, forcedFirstChild);
327 private void _revertSelectionState()
329 handleSelections = false;
330 if (selection == null || selection.size() == 0)
332 dbviews.clearSelection();
336 dbviews.setSelectionPaths(tsel);
338 handleSelections = true;
341 private void updateDbStatus(List<DbSourceProxy> srcs,
342 boolean forcedFirstChild)
345 String nm = "", qr = "";
346 for (DbSourceProxy dbs : srcs)
348 String tq = dbs.getTestQuery();
349 nm = dbs.getDbName();
350 if (tq != null && tq.trim().length() > 0 && dbs.isValidReference(tq))
357 if (allowMultiSelections)
359 dbstatus.setText("Selected "
362 + (srcs.size() == 1 ? "" : "s")
364 + (srcs.size() > 0 ? " with " + x + " test quer"
365 + (x == 1 ? "y" : "ies") : "."));
366 dbstatex.setText(" ");
372 dbstatus.setText("Database: " + nm);
375 dbstatex.setText("Example: " + qr);
379 dbstatex.setText(" ");
384 dbstatus.setText(" ");
387 dbstatus.invalidate();
388 dbstatex.invalidate();
391 public String getSelectedItem()
395 return getSelectedSources().get(0).getDbName();
400 public String getExampleQueries()
406 StringBuffer sb = new StringBuffer();
407 HashSet<String> hs = new HashSet<String>();
408 for (DbSourceProxy dbs : getSelectedSources())
410 String tq = dbs.getTestQuery();
421 return sb.toString();
424 List<ActionListener> lstners = new Vector<ActionListener>();
426 public void addActionListener(ActionListener actionListener)
428 lstners.add(actionListener);
431 public void removeActionListener(ActionListener actionListener)
433 lstners.remove(actionListener);
436 public static void main(String args[])
438 Cache.getDasSourceRegistry();
439 JDatabaseTree jdt = new JDatabaseTree(new jalview.ws.SequenceFetcher());
440 JFrame foo = new JFrame();
441 foo.setLayout(new BorderLayout());
442 foo.add(jdt.getDatabaseSelectorButton(), BorderLayout.CENTER);
444 foo.setVisible(true);
446 final Thread us = Thread.currentThread();
447 jdt.addActionListener(new ActionListener()
451 public void actionPerformed(ActionEvent e)
461 } catch (InterruptedException x)
464 if (!jdt.hasSelection())
466 System.out.println("No Selection");
470 System.out.println("Selection: " + jdt.getSelectedItem());
472 for (DbSourceProxy pr : jdt.getSelectedSources())
474 System.out.println("Source " + s++ + ": " + pr.getDbName()
475 + " (" + pr.getDbSource() + ") Version "
476 + pr.getDbVersion() + ". Test:\t" + pr.getTestQuery());
478 System.out.println("Test queries: " + jdt.getExampleQueries());
481 } while (nultimes > 0 && foo.isVisible());
482 foo.setVisible(false);
486 public void keyPressed(KeyEvent arg0)
488 if (!arg0.isConsumed() && arg0.getKeyCode() == KeyEvent.VK_ENTER)
492 if (!arg0.isConsumed() && arg0.getKeyChar() == KeyEvent.VK_ESCAPE)
499 public void keyReleased(KeyEvent arg0)
501 // TODO Auto-generated method stub
506 public void keyTyped(KeyEvent arg0)
508 // TODO Auto-generated method stub