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.event.ActionEvent;
12 import java.awt.event.ActionListener;
13 import java.util.ArrayList;
14 import java.util.Hashtable;
15 import java.util.List;
16 import java.util.Vector;
18 import javax.swing.JButton;
19 import javax.swing.JFrame;
20 import javax.swing.JLabel;
21 import javax.swing.JPanel;
22 import javax.swing.JScrollPane;
23 import javax.swing.JTree;
24 import javax.swing.JViewport;
25 import javax.swing.event.TreeSelectionEvent;
26 import javax.swing.event.TreeSelectionListener;
27 import javax.swing.tree.DefaultMutableTreeNode;
28 import javax.swing.tree.DefaultTreeCellRenderer;
29 import javax.swing.tree.DefaultTreeModel;
30 import javax.swing.tree.MutableTreeNode;
31 import javax.swing.tree.TreeCellRenderer;
32 import javax.swing.tree.TreeModel;
33 import javax.swing.tree.TreePath;
34 import javax.swing.tree.TreeSelectionModel;
36 import com.sun.java.help.search.DBPartParameters;
38 public class JDatabaseTree extends JalviewDialog
40 JButton getDatabaseSelectorButton()
42 final JButton viewdbs = new JButton("--- Select Database ---");
43 viewdbs.addActionListener(new ActionListener()
47 public void actionPerformed(ActionEvent arg0)
59 private jalview.ws.SequenceFetcher sfetcher;
61 private JLabel dbstatus;
63 public JDatabaseTree(jalview.ws.SequenceFetcher sfetch)
65 initDialogFrame(this, true, false, "Select Database Retrieval Source",
68 * Dynamically generated database list will need a translation function from
69 * internal source to externally distinct names. UNIPROT and UP_NAME are
70 * identical DB sources, and should be collapsed.
72 DefaultMutableTreeNode tn = null, root = new DefaultMutableTreeNode();
73 Hashtable<DbSourceProxy, DefaultMutableTreeNode> tns = new Hashtable<DbSourceProxy, DefaultMutableTreeNode>();
74 Hashtable<String, DefaultMutableTreeNode> source = new Hashtable<String, DefaultMutableTreeNode>();
76 String dbs[] = sfetch.getSupportedDb();
77 Hashtable ht = new Hashtable();
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"
100 for (int i = 0; i < dbs.length; i++)
102 tn = source.get(dbs[i]);
107 if (tn.getChildCount() == 1)
109 // remove nodes with only one child
110 source.put(dbs[i], tn = (DefaultMutableTreeNode) tn.getChildAt(0));
114 svp = new JScrollPane();
115 // svp.setAutoscrolls(true);
116 dbviews = new JTree(new DefaultTreeModel(root, false));
117 dbviews.setCellRenderer(new DbTreeRenderer(this));
119 dbviews.getSelectionModel().setSelectionMode(
120 TreeSelectionModel.SINGLE_TREE_SELECTION);
121 svp.getViewport().setView(dbviews);
122 // svp.getViewport().setMinimumSize(new Dimension(300,200));
123 // svp.setSize(300,250);
124 // JPanel panel=new JPanel();
125 // panel.setSize(new Dimension(350,220));
127 dbviews.addTreeSelectionListener(new TreeSelectionListener()
131 public void valueChanged(TreeSelectionEvent arg0)
133 _setSelectionState();
136 JPanel jc = new JPanel(new BorderLayout()), j = new JPanel(
138 jc.add(svp, BorderLayout.CENTER);
141 // TODO: make the panel resize and fit this in
142 dbstatus = new JLabel(" "); // set the height correctly for layout
143 dbstatus.setFont(f = JvSwingUtils.getLabelFont(false, true));
144 dbstatus.setSize(new Dimension(290, 50));
145 jc.add(dbstatus, BorderLayout.SOUTH);
147 // j.setPreferredSize(new Dimension(300,50));
148 add(jc, BorderLayout.CENTER);
151 add(j, BorderLayout.SOUTH);
155 private class DbTreeRenderer extends DefaultTreeCellRenderer implements
160 public DbTreeRenderer(JDatabaseTree me)
165 private Component returnLabel(String txt)
167 JLabel jl = new JLabel(txt);
168 jl.setFont(JvSwingUtils.getLabelFont());
173 public Component getTreeCellRendererComponent(JTree tree, Object value,
174 boolean selected, boolean expanded, boolean leaf, int row,
178 if (value != null && value instanceof DefaultMutableTreeNode)
180 DefaultMutableTreeNode vl = (DefaultMutableTreeNode) value;
181 value = vl.getUserObject();
182 if (value instanceof DbSourceProxy)
184 val = (((DbSourceProxy) value).getDbName());
188 if (value instanceof String)
190 val = ((String) value);
198 return super.getTreeCellRendererComponent(tree, val, selected,
199 expanded, leaf, row, hasFocus);
204 List<DbSourceProxy> selection = null;
207 protected void raiseClosed()
209 for (ActionListener al : lstners)
211 al.actionPerformed(null);
216 protected void okPressed()
218 _setSelectionState();
223 protected void cancelPressed()
228 private void showDialog(Container parent)
234 public boolean hasSelection()
236 return selection == null ? false : selection.size() == 0 ? false : true;
239 public List<DbSourceProxy> getSelectedSources()
244 private void _setSelectionState()
246 if (dbviews.getSelectionCount() == 0)
250 List<DbSourceProxy> srcs = new ArrayList<DbSourceProxy>();
251 for (TreePath tp : dbviews.getSelectionPaths())
253 DefaultMutableTreeNode dmt = (DefaultMutableTreeNode) tp
254 .getLastPathComponent();
255 if (dmt.getUserObject() != null)
257 if (dmt.getUserObject() instanceof DbSourceProxy)
259 srcs.add((DbSourceProxy) dmt.getUserObject());
263 srcs.addAll(sfetcher.getSourceProxy((String) dmt.getUserObject()));
267 updateDbStatus(srcs);
271 private void updateDbStatus(List<DbSourceProxy> srcs)
274 for (DbSourceProxy dbs : srcs)
276 String tq = dbs.getTestQuery();
277 if (tq != null && tq.trim().length()>0 && dbs.isValidReference(tq))
283 dbstatus.setText("Selected "
286 + (srcs.size() == 1 ? "" : "s")
288 + (srcs.size() > 0 ? " with " + x + " test quer"
289 + (x == 1 ? "y" : "ies") : "."));
290 dbstatus.invalidate();
293 public String getSelectedItem()
297 return getSelectedSources().get(0).getDbName();
302 public String getExampleQueries()
308 StringBuffer sb = new StringBuffer();
309 for (DbSourceProxy dbs : getSelectedSources())
315 sb.append(dbs.getTestQuery());
317 return sb.toString();
320 List<ActionListener> lstners = new Vector<ActionListener>();
322 public void addActionListener(ActionListener actionListener)
324 lstners.add(actionListener);
327 public void removeActionListener(ActionListener actionListener)
329 lstners.remove(actionListener);
332 public static void main(String args[])
334 Cache.getDasSourceRegistry();
335 JDatabaseTree jdt = new JDatabaseTree(new jalview.ws.SequenceFetcher());
336 JFrame foo = new JFrame();
337 foo.setLayout(new BorderLayout());
338 foo.add(jdt.getDatabaseSelectorButton(), BorderLayout.CENTER);
340 foo.setVisible(true);
342 final Thread us = Thread.currentThread();
343 jdt.addActionListener(new ActionListener()
347 public void actionPerformed(ActionEvent e)
357 } catch (InterruptedException x)
360 if (!jdt.hasSelection())
362 System.out.println("No Selection");
366 System.out.println("Selection: " + jdt.getSelectedItem());
368 for (DbSourceProxy pr : jdt.getSelectedSources())
370 System.out.println("Source " + s++ + ": " + pr.getDbName()
371 + " (" + pr.getDbSource() + ") Version "
372 + pr.getDbVersion() + ". Test:\t" + pr.getTestQuery());
374 System.out.println("Test queries: " + jdt.getExampleQueries());
377 } while (nultimes > 0 && foo.isVisible());
378 foo.setVisible(false);