X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FStructureChooser.java;h=4807489bb965e7fc44f7f17b73ca9479fd9ec53b;hb=27dc4596d13971e00f04578d469c5e3eb604757f;hp=82ad03b953a7537698ead03ca6dea06e765b9bec;hpb=38301c257efe7d20d85ccded982904c67440be95;p=jalview.git diff --git a/src/jalview/gui/StructureChooser.java b/src/jalview/gui/StructureChooser.java index 82ad03b..4807489 100644 --- a/src/jalview/gui/StructureChooser.java +++ b/src/jalview/gui/StructureChooser.java @@ -21,17 +21,23 @@ package jalview.gui; +import java.awt.event.ActionEvent; +import java.awt.event.ActionListener; import java.awt.event.ItemEvent; import java.util.ArrayList; import java.util.Collection; import java.util.HashSet; import java.util.LinkedHashSet; import java.util.List; +import java.util.Locale; import java.util.concurrent.Executors; import javax.swing.JCheckBox; import javax.swing.JComboBox; import javax.swing.JLabel; +import javax.swing.JMenu; +import javax.swing.JMenuItem; +import javax.swing.JPopupMenu; import javax.swing.JTable; import javax.swing.SwingUtilities; import javax.swing.table.AbstractTableModel; @@ -48,6 +54,7 @@ import jalview.fts.core.FTSDataColumnPreferences; import jalview.fts.core.FTSRestRequest; import jalview.fts.core.FTSRestResponse; import jalview.fts.service.pdb.PDBFTSRestClient; +import jalview.fts.service.threedbeacons.TDB_FTSData; import jalview.gui.structurechooser.PDBStructureChooserQuerySource; import jalview.gui.structurechooser.StructureChooserQuerySource; import jalview.gui.structurechooser.ThreeDBStructureChooserQuerySource; @@ -58,6 +65,7 @@ import jalview.structure.StructureMapping; import jalview.structure.StructureSelectionManager; import jalview.util.MessageManager; import jalview.ws.DBRefFetcher; +import jalview.ws.DBRefFetcher.FetchFinishedListenerI; import jalview.ws.seqfetcher.DbSourceProxy; import jalview.ws.sifts.SiftsSettings; @@ -73,6 +81,11 @@ public class StructureChooser extends GStructureChooser { private static final String AUTOSUPERIMPOSE = "AUTOSUPERIMPOSE"; + /** + * transient combo box choice for initiating 3db fetch + */ + private static final String VIEWS_QUERYING_TDB = "QUERY_3DB"; + private SequenceI selectedSequence; private SequenceI[] selectedSequences; @@ -97,25 +110,75 @@ public class StructureChooser extends GStructureChooser private Collection lastDiscoveredStructuresSet; + private boolean canQueryTDB = false; + + private boolean notQueriedTDBYet = true; + + List seqsWithoutSourceDBRef = null; + private static StructureViewer lastTargetedView = null; public StructureChooser(SequenceI[] selectedSeqs, SequenceI selectedSeq, AlignmentPanel ap) { // which FTS engine to use - data = StructureChooserQuerySource - .getQuerySourceFor(selectedSeqs); + data = StructureChooserQuerySource.getQuerySourceFor(selectedSeqs); initDialog(); - + this.ap = ap; this.selectedSequence = selectedSeq; this.selectedSequences = selectedSeqs; this.progressIndicator = (ap == null) ? null : ap.alignFrame; init(); - + } /** + * sets canQueryTDB if protein sequences without a canonical uniprot ref or at + * least one structure are discovered. + */ + private void populateSeqsWithoutSourceDBRef() + { + seqsWithoutSourceDBRef = new ArrayList(); + boolean needCanonical = false; + for (SequenceI seq : selectedSequences) + { + if (seq.isProtein()) + { + int dbRef = ThreeDBStructureChooserQuerySource + .checkUniprotRefs(seq.getDBRefs()); + if (dbRef < 0) + { + if (dbRef == -1) + { + // need to retrieve canonicals + needCanonical = true; + seqsWithoutSourceDBRef.add(seq); + } + else + { + // could be a sequence with pdb ref + if (seq.getAllPDBEntries() == null + || seq.getAllPDBEntries().size() == 0) + { + seqsWithoutSourceDBRef.add(seq); + } + } + } + } + } + // retrieve database refs for protein sequences + if (!seqsWithoutSourceDBRef.isEmpty()) + { + canQueryTDB = true; + if (needCanonical) + { + notQueriedTDBYet = false; + } + } + }; + + /** * Initializes parameters used by the Structure Chooser Panel */ protected void init() @@ -127,115 +190,138 @@ public class StructureChooser extends GStructureChooser chk_superpose.setSelected(Cache.getDefault(AUTOSUPERIMPOSE, true)); - final Runnable discoverPDBStructures = new Runnable() + Executors.defaultThreadFactory().newThread(new Runnable() { - @Override public void run() { - // check which FTS engine to use - data = StructureChooserQuerySource - .getQuerySourceFor(selectedSequences); + populateSeqsWithoutSourceDBRef(); + initialStructureDiscovery(); + } - // ensure a filter option is in force for search - populateFilterComboBox(true, cachedPDBExists); + }).start(); - // looks for any existing structures already loaded - // for the sequences (the cached ones) - // then queries the StructureChooserQuerySource to - // discover more structures. - // - // Possible optimisation is to only begin querying - // the structure chooser if there are no cached structures. + } - long startTime = System.currentTimeMillis(); - updateProgressIndicator(MessageManager - .getString("status.loading_cached_pdb_entries"), startTime); - loadLocalCachedPDBEntries(); - updateProgressIndicator(null, startTime); - updateProgressIndicator(MessageManager.getString( - "status.searching_for_pdb_structures"), startTime); - fetchStructuresMetaData(); - // revise filter options if no results were found - populateFilterComboBox(isStructuresDiscovered(), cachedPDBExists); - discoverStructureViews(); - updateProgressIndicator(null, startTime); - mainFrame.setVisible(true); - updateCurrentView(); - } - }; - final List seqsWithoutSourceDBRef = new ArrayList(); + // called by init + private void initialStructureDiscovery() + { + // check which FTS engine to use + data = StructureChooserQuerySource.getQuerySourceFor(selectedSequences); + + // ensure a filter option is in force for search + populateFilterComboBox(true, cachedPDBExists); + + // looks for any existing structures already loaded + // for the sequences (the cached ones) + // then queries the StructureChooserQuerySource to + // discover more structures. + // + // Possible optimisation is to only begin querying + // the structure chooser if there are no cached structures. + + long startTime = System.currentTimeMillis(); + updateProgressIndicator( + MessageManager.getString("status.loading_cached_pdb_entries"), + startTime); + loadLocalCachedPDBEntries(); + updateProgressIndicator(null, startTime); + updateProgressIndicator( + MessageManager.getString("status.searching_for_pdb_structures"), + startTime); + fetchStructuresMetaData(); + // revise filter options if no results were found + populateFilterComboBox(isStructuresDiscovered(), cachedPDBExists); + discoverStructureViews(); + updateProgressIndicator(null, startTime); + mainFrame.setVisible(true); + updateCurrentView(); + } + + private void promptForTDBFetch() + { + final long progressId = System.currentTimeMillis(); - final Runnable discoverCanonicalDBrefs = new Runnable() + // final action after prompting and discovering db refs + final Runnable strucDiscovery = new Runnable() { @Override public void run() { - long progressId = System.currentTimeMillis(); - - int y = seqsWithoutSourceDBRef.size(); - - setProgressBar(MessageManager.formatMessage( - "status.fetching_dbrefs_for_sequences_without_valid_refs", - y), progressId); - SequenceI[] seqWithoutSrcDBRef = seqsWithoutSourceDBRef - .toArray(new SequenceI[y]); + // TODO: warn if no accessions discovered + populateSeqsWithoutSourceDBRef(); + // redo initial discovery - this time with 3d beacons + // Executors. + previousWantedFields=null; - DBRefFetcher dbRefFetcher = new DBRefFetcher(seqWithoutSrcDBRef, - progressBar, new DbSourceProxy[] - { new jalview.ws.dbsources.Uniprot() }, - null, false); - - dbRefFetcher.fetchDBRefs(true); - - setProgressBar("Fetch complete.", progressId); // todo i18n - - Executors.defaultThreadFactory().newThread(discoverPDBStructures).start(); + initialStructureDiscovery(); } }; - Executors.defaultThreadFactory().newThread(new Runnable() + final FetchFinishedListenerI afterDbRefFetch = new FetchFinishedListenerI() + { + + @Override + public void finished() + { + // filter has been selected, so we set flag to remove ourselves + notQueriedTDBYet = false; + // new thread to discover structures - via 3d beacons + Executors.defaultThreadFactory().newThread(strucDiscovery).start(); + + } + }; + + // fetch db refs if OK pressed + final Runnable discoverCanonicalDBrefs = new Runnable() { + @Override public void run() { + populateSeqsWithoutSourceDBRef(); - for (SequenceI seq : selectedSequences) + final int y = seqsWithoutSourceDBRef.size(); + if (y > 0) { - if (seq.isProtein()) - { - int dbRef = ThreeDBStructureChooserQuerySource - .checkUniprotRefs(seq.getDBRefs()); - if (dbRef < 0) - { - if (seq.getAllPDBEntries()==null && seq.getAllPDBEntries().size()==0) - { - seqsWithoutSourceDBRef.add(seq); - } - } - } - } - // retrieve database refs for protein sequences - if (!seqsWithoutSourceDBRef.isEmpty()) - { - // need cancel and no to result in the discoverPDB action - mocked is 'cancel' - JvOptionPane.newOptionDialog(Desktop.getDesktop()) - .setResponseHandler(JvOptionPane.OK_OPTION, discoverCanonicalDBrefs) - .setResponseHandler(JvOptionPane.CANCEL_OPTION, discoverPDBStructures) - .setResponseHandler(JvOptionPane.NO_OPTION, discoverPDBStructures) - .showDialog(MessageManager.formatMessage("label.fetch_references_for", - seqsWithoutSourceDBRef.size()), MessageManager.getString( - "label.fetch_uniprot_references"), - JvOptionPane.YES_NO_OPTION, - JvOptionPane.PLAIN_MESSAGE, null, new Object[] - { MessageManager.getString("action.ok"), - MessageManager.getString("action.cancel") }, - MessageManager.getString("action.ok")); + final SequenceI[] seqWithoutSrcDBRef = seqsWithoutSourceDBRef + .toArray(new SequenceI[y]); + DBRefFetcher dbRefFetcher = new DBRefFetcher(seqWithoutSrcDBRef, + progressBar, new DbSourceProxy[] + { new jalview.ws.dbsources.Uniprot() }, null, false); + dbRefFetcher.addListener(afterDbRefFetch); + // ideally this would also gracefully run with callbacks + dbRefFetcher.fetchDBRefs(true); } else { - // get structures directly - Executors.defaultThreadFactory().newThread(discoverPDBStructures).start(); + // call finished action directly + afterDbRefFetch.finished(); } - }; - }).start();; + } + }; + final Runnable revertview = new Runnable() { + public void run() { + if (lastSelected!=null) { + cmb_filterOption.setSelectedItem(lastSelected); + } + }; + }; + // need cancel and no to result in the discoverPDB action - mocked is + // 'cancel' + JvOptionPane.newOptionDialog(this) + .setResponseHandler(JvOptionPane.OK_OPTION, + discoverCanonicalDBrefs) + .setResponseHandler(JvOptionPane.CANCEL_OPTION, revertview) + .setResponseHandler(JvOptionPane.NO_OPTION, revertview) + .showDialog( + MessageManager.formatMessage( + "label.fetch_references_for_3dbeacons", + seqsWithoutSourceDBRef.size()), + MessageManager + .getString("label.3dbeacons"), + JvOptionPane.YES_NO_OPTION, JvOptionPane.PLAIN_MESSAGE, + null, new Object[] + { MessageManager.getString("action.ok"), + MessageManager.getString("action.cancel") }, + MessageManager.getString("action.ok")); } /** @@ -331,8 +417,9 @@ public class StructureChooser extends GStructureChooser resultList = data.fetchStructuresMetaData(seq, wantedFields, selectedFilterOpt, !chk_invertFilter.isSelected()); // null response means the FTSengine didn't yield a query for this - // consider designing a special exception if we really wanted to be OOCrazy - if (resultList==null) + // consider designing a special exception if we really wanted to be + // OOCrazy + if (resultList == null) { continue; } @@ -357,9 +444,9 @@ public class StructureChooser extends GStructureChooser { getResultTable() .setModel(data.getTableModel(discoveredStructuresSet)); - + noOfStructuresFound = discoveredStructuresSet.size(); - lastDiscoveredStructuresSet=discoveredStructuresSet; + lastDiscoveredStructuresSet = discoveredStructuresSet; mainFrame.setTitle(MessageManager.formatMessage( "label.structure_chooser_no_of_structures", noOfStructuresFound, totalTime)); @@ -432,8 +519,9 @@ public class StructureChooser extends GStructureChooser FTSRestResponse resultList; try { - resultList = data.selectFirstRankedQuery(seq, discoveredStructuresSet,wantedFields, - fieldToFilterBy, !chk_invertFilter.isSelected()); + resultList = data.selectFirstRankedQuery(seq, + discoveredStructuresSet, wantedFields, fieldToFilterBy, + !chk_invertFilter.isSelected()); } catch (Exception e) { @@ -458,7 +546,7 @@ public class StructureChooser extends GStructureChooser reorderedStructuresSet.addAll(discoveredStructuresSet); getResultTable() .setModel(data.getTableModel(reorderedStructuresSet)); - + FTSRestResponse.configureTableColumn(getResultTable(), wantedFields, tempUserPrefs); getResultTable().getColumn("Ref Sequence").setPreferredWidth(120); @@ -532,6 +620,7 @@ public class StructureChooser extends GStructureChooser { populateFilterComboBox(haveData, cachedPDBExist, null); } + /** * Populates the filter combo-box options dynamically depending on discovered * structures @@ -539,45 +628,57 @@ public class StructureChooser extends GStructureChooser protected void populateFilterComboBox(boolean haveData, boolean cachedPDBExist, FilterOption lastSel) { - + /* * temporarily suspend the change listener behaviour */ cmb_filterOption.removeItemListener(this); - int selSet=-1; + int selSet = -1; cmb_filterOption.removeAllItems(); if (haveData) { - List filters = data.getAvailableFilterOptions(VIEWS_FILTER); - data.updateAvailableFilterOptions(VIEWS_FILTER, filters, lastDiscoveredStructuresSet); - int p=0; - for (FilterOption filter:filters) + List filters = data + .getAvailableFilterOptions(VIEWS_FILTER); + data.updateAvailableFilterOptions(VIEWS_FILTER, filters, + lastDiscoveredStructuresSet); + int p = 0; + for (FilterOption filter : filters) { - if (lastSel!=null && filter.equals(lastSel)) { - selSet=p; + if (lastSel != null && filter.equals(lastSel)) + { + selSet = p; } p++; cmb_filterOption.addItem(filter); } } + cmb_filterOption.addItem( new FilterOption(MessageManager.getString("label.enter_pdb_id"), - "-", VIEWS_ENTER_ID, false,null)); + "-", VIEWS_ENTER_ID, false, null)); cmb_filterOption.addItem( new FilterOption(MessageManager.getString("label.from_file"), - "-", VIEWS_FROM_FILE, false,null)); + "-", VIEWS_FROM_FILE, false, null)); + if (canQueryTDB && notQueriedTDBYet) + { + FilterOption queryTDBOption = new FilterOption( + MessageManager.getString("label.search_3dbeacons"), "-", + VIEWS_QUERYING_TDB, false, null); + cmb_filterOption.addItem(queryTDBOption); + } if (cachedPDBExist) { FilterOption cachedOption = new FilterOption( MessageManager.getString("label.cached_structures"), "-", - VIEWS_LOCAL_PDB, false,null); + VIEWS_LOCAL_PDB, false, null); cmb_filterOption.addItem(cachedOption); - if (selSet==-1) { + if (selSet == -1) + { cmb_filterOption.setSelectedItem(cachedOption); } } - if (selSet>-1) + if (selSet > -1) { cmb_filterOption.setSelectedIndex(selSet); } @@ -591,24 +692,45 @@ public class StructureChooser extends GStructureChooser { FilterOption selectedFilterOpt = ((FilterOption) cmb_filterOption .getSelectedItem()); + + // first check if we need to rebuild dialog + if (selectedFilterOpt.getView() == VIEWS_QUERYING_TDB) + { + promptForTDBFetch(); + return; + } + if (lastSelected == selectedFilterOpt) + { + // don't need to do anything, probably + return; + } + // otherwise, record selection + // and update the layout and dialog accordingly + lastSelected = selectedFilterOpt; + layout_switchableViews.show(pnl_switchableViews, selectedFilterOpt.getView()); String filterTitle = mainFrame.getTitle(); mainFrame.setTitle(frameTitle); chk_invertFilter.setVisible(false); + if (selectedFilterOpt.getView() == VIEWS_FILTER) { mainFrame.setTitle(filterTitle); // TDB Query has no invert as yet - chk_invertFilter.setVisible(selectedFilterOpt.getQuerySource() instanceof PDBStructureChooserQuerySource); - - if (data!=selectedFilterOpt.getQuerySource() || data.needsRefetch(selectedFilterOpt)) + chk_invertFilter.setVisible(selectedFilterOpt + .getQuerySource() instanceof PDBStructureChooserQuerySource); + + if (data != selectedFilterOpt.getQuerySource() + || data.needsRefetch(selectedFilterOpt)) { data = selectedFilterOpt.getQuerySource(); // rebuild the views completely, since prefs will also change tabRefresh(); return; - } else { + } + else + { filterResultSet(selectedFilterOpt.getValue()); } } @@ -676,6 +798,38 @@ public class StructureChooser extends GStructureChooser .setEnabled(selectedCount > 1 || targetView.getItemCount() > 0); } + @Override + protected boolean showPopupFor(int selectedRow, int x, int y) + { + FilterOption selectedFilterOpt = ((FilterOption) cmb_filterOption + .getSelectedItem()); + String currentView = selectedFilterOpt.getView(); + + if (currentView == VIEWS_FILTER && data instanceof ThreeDBStructureChooserQuerySource) + { + + TDB_FTSData row=((ThreeDBStructureChooserQuerySource)data).getFTSDataFor(getResultTable(), selectedRow, discoveredStructuresSet); + String pageUrl = row.getModelViewUrl(); + JPopupMenu popup = new JPopupMenu("3D Beacons"); + JMenuItem viewUrl = new JMenuItem("View model web page"); + viewUrl.addActionListener( + new ActionListener() { + @Override + public void actionPerformed(ActionEvent e) + { + Desktop.showUrl(pageUrl); + } + } + ); + popup.add(viewUrl); + SwingUtilities.invokeLater(new Runnable() { + public void run() { popup.show(getResultTable(), x, y); } + }); + return true; + } + // event not handled by us + return false; + } /** * Validates inputs from the Manual PDB entry panel */ @@ -747,7 +901,7 @@ public class StructureChooser extends GStructureChooser { validateSelections(); } - + private FilterOption lastSelected=null; /** * Handles the state change event for the 'filter' combo-box and 'invert' * check-box @@ -855,7 +1009,8 @@ public class StructureChooser extends GStructureChooser int[] selectedRows = restable.getSelectedRows(); PDBEntry[] pdbEntriesToView = new PDBEntry[selectedRows.length]; List selectedSeqsToView = new ArrayList<>(); - pdbEntriesToView = data.collectSelectedRows(restable,selectedRows,selectedSeqsToView); + pdbEntriesToView = data.collectSelectedRows(restable, + selectedRows, selectedSeqsToView); SequenceI[] selectedSeqs = selectedSeqsToView .toArray(new SequenceI[selectedSeqsToView.size()]); @@ -874,8 +1029,9 @@ public class StructureChooser extends GStructureChooser List selectedSeqsToView = new ArrayList<>(); for (int row : selectedRows) { - PDBEntry pdbEntry = ((PDBEntryTableModel) tbl_local_pdb.getModel()).getPDBEntryAt(row).getPdbEntry(); - + PDBEntry pdbEntry = ((PDBEntryTableModel) tbl_local_pdb + .getModel()).getPDBEntryAt(row).getPdbEntry(); + pdbEntriesToView[count++] = pdbEntry; SequenceI selectedSeq = (SequenceI) tbl_local_pdb .getValueAt(row, refSeqColIndex); @@ -902,7 +1058,7 @@ public class StructureChooser extends GStructureChooser if (pdbIdStr.split(":").length > 1) { pdbEntry.setId(pdbIdStr.split(":")[0]); - pdbEntry.setChainCode(pdbIdStr.split(":")[1].toUpperCase()); + pdbEntry.setChainCode(pdbIdStr.split(":")[1].toUpperCase(Locale.ROOT)); } else { @@ -1129,7 +1285,7 @@ public class StructureChooser extends GStructureChooser // TODO move this pdb id search into the PDB specific // FTSSearchEngine // for moment, it will work fine as is because it is self-contained - String searchTerm = text.toLowerCase(); + String searchTerm = text.toLowerCase(Locale.ROOT); searchTerm = searchTerm.split(":")[0]; // System.out.println(">>>>> search term : " + searchTerm); List wantedFields = new ArrayList<>(); @@ -1176,7 +1332,7 @@ public class StructureChooser extends GStructureChooser public void run() { fetchStructuresMetaData(); - //populateFilterComboBox(true, cachedPDBExists); + // populateFilterComboBox(true, cachedPDBExists); filterResultSet( ((FilterOption) cmb_filterOption.getSelectedItem()) @@ -1279,8 +1435,8 @@ public class StructureChooser extends GStructureChooser { if (pdbEntry.hasProvider()) { - return pdbEntry.getProvider()+":"+pdbEntry.getId(); - } + return pdbEntry.getProvider() + ":" + pdbEntry.getId(); + } return pdbEntry.toString(); } @@ -1325,12 +1481,13 @@ public class StructureChooser extends GStructureChooser protected void setFTSDocFieldPrefs(FTSDataColumnPreferences newPrefs) { data.setDocFieldPrefs(newPrefs); - + } /** * - * @return true when all initialisation threads have finished and dialog is visible + * @return true when all initialisation threads have finished and dialog is + * visible */ public boolean isDialogVisible() {