2 * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3 * Copyright (C) $$Year-Rel$$ The Jalview Authors
5 * This file is part of Jalview.
7 * Jalview is free software: you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation, either version 3
10 * of the License, or (at your option) any later version.
12 * Jalview is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty
14 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
15 * PURPOSE. See the GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with Jalview. If not, see <http://www.gnu.org/licenses/>.
19 * The Jalview Authors are detailed in the 'AUTHORS' file.
23 import jalview.bin.Cache;
24 import jalview.util.MessageManager;
25 import jalview.ws.seqfetcher.DbSourceProxy;
27 import java.awt.BorderLayout;
28 import java.awt.Component;
29 import java.awt.Dimension;
30 import java.awt.FlowLayout;
31 import java.awt.GridLayout;
32 import java.awt.event.ActionEvent;
33 import java.awt.event.ActionListener;
34 import java.awt.event.KeyEvent;
35 import java.awt.event.KeyListener;
36 import java.awt.event.MouseAdapter;
37 import java.awt.event.MouseEvent;
38 import java.util.ArrayList;
39 import java.util.HashSet;
40 import java.util.Hashtable;
41 import java.util.List;
42 import java.util.Vector;
44 import javax.swing.JButton;
45 import javax.swing.JFrame;
46 import javax.swing.JLabel;
47 import javax.swing.JPanel;
48 import javax.swing.JScrollPane;
49 import javax.swing.JTree;
50 import javax.swing.ToolTipManager;
51 import javax.swing.event.TreeSelectionEvent;
52 import javax.swing.event.TreeSelectionListener;
53 import javax.swing.tree.DefaultMutableTreeNode;
54 import javax.swing.tree.DefaultTreeCellRenderer;
55 import javax.swing.tree.DefaultTreeModel;
56 import javax.swing.tree.TreeCellRenderer;
57 import javax.swing.tree.TreeNode;
58 import javax.swing.tree.TreePath;
59 import javax.swing.tree.TreeSelectionModel;
61 public class JDatabaseTree extends JalviewDialog implements KeyListener
63 boolean allowMultiSelections = false;
67 JButton getDatabaseSelectorButton()
69 final JButton viewdbs = new JButton(
70 MessageManager.getString("action.select_ddbb"));
71 viewdbs.addActionListener(new ActionListener()
75 public void actionPerformed(ActionEvent arg0)
87 private jalview.ws.SequenceFetcher sfetcher;
89 private JLabel dbstatus, dbstatex;
91 private JPanel mainPanel = new JPanel(new BorderLayout());
93 public JDatabaseTree(jalview.ws.SequenceFetcher sfetch)
96 initDialogFrame(mainPanel, true, false,
98 .getString("label.select_database_retrieval_source"),
101 * Dynamically generated database list will need a translation function from
102 * internal source to externally distinct names. UNIPROT and UP_NAME are
103 * identical DB sources, and should be collapsed.
105 DefaultMutableTreeNode tn = null, root = new DefaultMutableTreeNode();
106 Hashtable<String, DefaultMutableTreeNode> source = new Hashtable<String, DefaultMutableTreeNode>();
108 String dbs[] = sfetch.getSupportedDb();
109 Hashtable<String, String> ht = new Hashtable<String, String>();
110 for (int i = 0; i < dbs.length; i++)
112 tn = source.get(dbs[i]);
113 List<DbSourceProxy> srcs = sfetch.getSourceProxy(dbs[i]);
116 source.put(dbs[i], tn = new DefaultMutableTreeNode(dbs[i], true));
118 for (DbSourceProxy dbp : srcs)
120 if (ht.get(dbp.getDbName()) == null)
122 tn.add(new DefaultMutableTreeNode(dbp, false));
123 ht.put(dbp.getDbName(), dbp.getDbName());
127 System.err.println("dupe ig for : " + dbs[i] + " \t"
128 + dbp.getDbName() + " (" + dbp.getDbSource() + ")");
133 for (int i = 0; i < dbs.length; i++)
135 tn = source.get(dbs[i]);
140 if (tn.getChildCount() == 1)
142 DefaultMutableTreeNode ttn = (DefaultMutableTreeNode) tn
144 // remove nodes with only one child
145 tn.setUserObject(ttn.getUserObject());
146 tn.removeAllChildren();
147 source.put(dbs[i], tn);
148 tn.setAllowsChildren(false);
154 svp = new JScrollPane();
155 // svp.setAutoscrolls(true);
156 dbviews = new JTree(new DefaultTreeModel(root, false));
157 dbviews.setCellRenderer(new DbTreeRenderer(this));
159 dbviews.getSelectionModel().setSelectionMode(
160 TreeSelectionModel.SINGLE_TREE_SELECTION);
161 svp.getViewport().setView(dbviews);
162 // svp.getViewport().setMinimumSize(new Dimension(300,200));
163 // svp.setSize(300,250);
164 // JPanel panel=new JPanel();
165 // panel.setSize(new Dimension(350,220));
167 dbviews.addTreeSelectionListener(new TreeSelectionListener()
171 public void valueChanged(TreeSelectionEvent arg0)
173 _setSelectionState();
176 dbviews.addMouseListener(new MouseAdapter()
180 public void mousePressed(MouseEvent e)
182 if (e.getClickCount() == 2)
189 JPanel jc = new JPanel(new BorderLayout()), j = new JPanel(
191 jc.add(svp, BorderLayout.CENTER);
194 // TODO: make the panel stay a fixed size for longest dbname+example set.
195 JPanel dbstat = new JPanel(new GridLayout(2, 1));
196 dbstatus = new JLabel(" "); // set the height correctly for layout
197 dbstatus.setFont(f = JvSwingUtils.getLabelFont(false, true));
198 dbstatus.setSize(new Dimension(290, 50));
199 dbstatex = new JLabel(" ");
201 dbstatex.setSize(new Dimension(290, 50));
202 dbstat.add(dbstatus);
203 dbstat.add(dbstatex);
204 jc.add(dbstat, BorderLayout.SOUTH);
206 // j.setPreferredSize(new Dimension(300,50));
207 add(jc, BorderLayout.CENTER);
208 ok.setEnabled(false);
211 add(j, BorderLayout.SOUTH);
212 dbviews.addKeyListener(this);
216 private void sortTreeNodes(DefaultMutableTreeNode root)
218 if (root.getChildCount() == 0)
222 int count = root.getChildCount();
223 String[] names = new String[count];
224 DefaultMutableTreeNode[] nodes = new DefaultMutableTreeNode[count];
225 for (int i = 0; i < count; i++)
227 TreeNode node = root.getChildAt(i);
228 if (node instanceof DefaultMutableTreeNode)
230 DefaultMutableTreeNode child = (DefaultMutableTreeNode) node;
232 if (child.getUserObject() instanceof DbSourceProxy)
234 names[i] = ((DbSourceProxy) child.getUserObject()).getDbName()
239 names[i] = ((String) child.getUserObject()).toLowerCase();
240 sortTreeNodes(child);
247 .getString("error.implementation_error_cant_reorder_tree"));
250 jalview.util.QuickSort.sort(names, nodes);
251 root.removeAllChildren();
252 for (int i = count - 1; i >= 0; i--)
258 private class DbTreeRenderer extends DefaultTreeCellRenderer implements
263 public DbTreeRenderer(JDatabaseTree me)
266 ToolTipManager.sharedInstance().registerComponent(dbviews);
269 private Component returnLabel(String txt)
271 JLabel jl = new JLabel(txt);
272 jl.setFont(JvSwingUtils.getLabelFont());
277 public Component getTreeCellRendererComponent(JTree tree, Object value,
278 boolean selected, boolean expanded, boolean leaf, int row,
282 if (value != null && value instanceof DefaultMutableTreeNode)
284 DefaultMutableTreeNode vl = (DefaultMutableTreeNode) value;
285 value = vl.getUserObject();
286 if (value instanceof DbSourceProxy)
288 val = ((DbSourceProxy) value).getDbName();
289 if (((DbSourceProxy) value).getDescription() != null)
291 this.setToolTipText(((DbSourceProxy) value).getDescription());
296 if (value instanceof String)
298 val = (String) value;
306 return super.getTreeCellRendererComponent(tree, val, selected,
307 expanded, leaf, row, hasFocus);
312 List<DbSourceProxy> oldselection, selection = null;
314 TreePath[] tsel = null, oldtsel = null;
317 protected void raiseClosed()
319 for (ActionListener al : lstners)
321 al.actionPerformed(null);
326 protected void okPressed()
328 _setSelectionState();
332 protected void cancelPressed()
334 selection = oldselection;
336 _revertSelectionState();
342 oldselection = selection;
348 public boolean hasSelection()
350 return selection == null ? false : selection.size() == 0 ? false : true;
353 public List<DbSourceProxy> getSelectedSources()
359 * disable or enable selection handler
361 boolean handleSelections = true;
363 private void _setSelectionState()
365 if (!handleSelections)
369 ok.setEnabled(false);
370 if (dbviews.getSelectionCount() == 0)
375 tsel = dbviews.getSelectionPaths();
376 boolean forcedFirstChild = false;
377 List<DbSourceProxy> srcs = new ArrayList<DbSourceProxy>();
380 for (TreePath tp : tsel)
382 DefaultMutableTreeNode admt, dmt = (DefaultMutableTreeNode) tp
383 .getLastPathComponent();
384 if (dmt.getUserObject() != null)
387 * enable OK button once a selection has been made
390 if (dmt.getUserObject() instanceof DbSourceProxy)
392 srcs.add((DbSourceProxy) dmt.getUserObject());
396 if (allowMultiSelections)
398 srcs.addAll(sfetcher.getSourceProxy((String) dmt
404 .getSourceProxy((String) dmt.getUserObject()).get(0));
405 forcedFirstChild = true;
411 updateDbStatus(srcs, forcedFirstChild);
415 private void _revertSelectionState()
417 handleSelections = false;
418 if (selection == null || selection.size() == 0)
420 dbviews.clearSelection();
424 dbviews.setSelectionPaths(tsel);
426 handleSelections = true;
429 private void updateDbStatus(List<DbSourceProxy> srcs,
430 boolean forcedFirstChild)
433 String nm = "", qr = "";
434 for (DbSourceProxy dbs : srcs)
436 String tq = dbs.getTestQuery();
437 nm = dbs.getDbName();
438 if (tq != null && tq.trim().length() > 0 && dbs.isValidReference(tq))
445 dbstatex.setText(" ");
446 if (allowMultiSelections)
448 dbstatus.setText(MessageManager.formatMessage(
449 "label.selected_database_to_fetch_from", new String[] {
450 Integer.valueOf(srcs.size()).toString(),
451 (srcs.size() == 1 ? "" : "s"),
452 (srcs.size() > 0 ? " with " + x + " test quer"
453 + (x == 1 ? "y" : "ies") : ".") }));
459 dbstatus.setText(MessageManager.formatMessage(
460 "label.database_param", new String[] { nm }));
463 dbstatex.setText(MessageManager.formatMessage(
464 "label.example_param", new String[] { qr }));
469 dbstatus.setText(" ");
472 dbstatus.invalidate();
473 dbstatex.invalidate();
476 public String getSelectedItem()
480 return getSelectedSources().get(0).getDbName();
485 public String getExampleQueries()
491 StringBuffer sb = new StringBuffer();
492 HashSet<String> hs = new HashSet<String>();
493 for (DbSourceProxy dbs : getSelectedSources())
495 String tq = dbs.getTestQuery();
506 return sb.toString();
509 List<ActionListener> lstners = new Vector<ActionListener>();
511 public void addActionListener(ActionListener actionListener)
513 lstners.add(actionListener);
516 public void removeActionListener(ActionListener actionListener)
518 lstners.remove(actionListener);
521 public static void main(String args[])
523 Cache.getDasSourceRegistry();
524 JDatabaseTree jdt = new JDatabaseTree(new jalview.ws.SequenceFetcher());
525 JFrame foo = new JFrame();
526 foo.setLayout(new BorderLayout());
527 foo.add(jdt.getDatabaseSelectorButton(), BorderLayout.CENTER);
529 foo.setVisible(true);
531 final Thread us = Thread.currentThread();
532 jdt.addActionListener(new ActionListener()
536 public void actionPerformed(ActionEvent e)
546 } catch (InterruptedException x)
549 if (!jdt.hasSelection())
551 System.out.println("No Selection");
555 System.out.println("Selection: " + jdt.getSelectedItem());
557 for (DbSourceProxy pr : jdt.getSelectedSources())
559 System.out.println("Source " + s++ + ": " + pr.getDbName()
560 + " (" + pr.getDbSource() + ") Version "
561 + pr.getDbVersion() + ". Test:\t" + pr.getTestQuery());
563 System.out.println("Test queries: " + jdt.getExampleQueries());
566 } while (nultimes > 0 && foo.isVisible());
567 foo.setVisible(false);
571 public void keyPressed(KeyEvent arg0)
573 if (!arg0.isConsumed() && arg0.getKeyCode() == KeyEvent.VK_ENTER)
575 action = arg0.getKeyCode();
579 if (!arg0.isConsumed() && arg0.getKeyChar() == KeyEvent.VK_ESCAPE)
581 action = arg0.getKeyCode();
587 public void keyReleased(KeyEvent arg0)
589 // TODO Auto-generated method stub
594 public void keyTyped(KeyEvent arg0)
596 // TODO Auto-generated method stub