import jalview.datamodel.SequenceI;
import jalview.jbgui.GStructureChooser;
import jalview.jbgui.PDBDocFieldPreferences;
+import jalview.structure.StructureSelectionManager;
import jalview.util.MessageManager;
import jalview.ws.dbsources.PDBRestClient;
import jalview.ws.dbsources.PDBRestClient.PDBDocField;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
+import java.util.Hashtable;
import java.util.LinkedHashSet;
import java.util.List;
-import java.util.Vector;
import javax.swing.JCheckBox;
import javax.swing.JComboBox;
import javax.swing.JLabel;
+import javax.swing.table.DefaultTableModel;
+
/**
* Provides the behaviors for the Structure chooser Panel
private boolean isValidPBDEntry;
+ private static Hashtable<String, PDBEntry> cachedEntryMap;
+
public StructureChooser(SequenceI[] selectedSeqs, SequenceI selectedSeq,
AlignmentPanel ap)
{
long startTime = System.currentTimeMillis();
String msg = MessageManager.getString("status.fetching_db_refs");
updateProgressIndicator(msg, startTime);
+ loadLocalCachedPDBEntries();
fetchStructuresMetaData();
populateFilterComboBox();
updateProgressIndicator(null, startTime);
+ " Found (" + totalTime + ")");
}
+ public void loadLocalCachedPDBEntries()
+ {
+ DefaultTableModel tableModel = new DefaultTableModel();
+ tableModel.addColumn("Sequence");
+ tableModel.addColumn("PDB Id");
+ tableModel.addColumn("Type");
+ tableModel.addColumn("File");
+ cachedEntryMap = new Hashtable<String, PDBEntry>();
+ for (SequenceI seq : selectedSequences)
+ {
+ if (seq.getDatasetSequence() != null
+ && seq.getDatasetSequence().getPDBId() != null)
+ {
+ for (PDBEntry pdbEntry : seq.getDatasetSequence().getPDBId())
+ {
+ String[] pdbEntryRowData = new String[]
+ { seq.getDisplayId(false), pdbEntry.getId(), pdbEntry.getType(),
+ pdbEntry.getFile() };
+ tableModel.addRow(pdbEntryRowData);
+ cachedEntryMap.put(seq.getDisplayId(false) + pdbEntry.getId(),
+ pdbEntry);
+ }
+ }
+ }
+ tbl_local_pdb.setModel(tableModel);
+ }
+
/**
* Update the DBRef entry for a given sequence with values retrieved from
* PDBResponseSummary
* the sequences to build a query for
* @return the built query string
*/
- @SuppressWarnings("unchecked")
+
public static String buildQuery(SequenceI seq)
{
- String query = seq.getName();
- StringBuilder queryBuilder = new StringBuilder();
- int count = 0;
+ HashSet<String> seqRefs = new LinkedHashSet<String>();
+ String seqName = seq.getName();
+ String[] names = seqName.toLowerCase().split("\\|");
+ for (String name : names)
+ {
+ System.out.println("Found name : " + name);
+ name.trim();
+ if (isValidSeqName(name))
+ {
+ seqRefs.add(name);
+ }
+ }
if (seq.getPDBId() != null)
{
- for (PDBEntry entry : (Vector<PDBEntry>) seq.getPDBId())
+ for (PDBEntry entry : seq.getPDBId())
{
- queryBuilder.append("text:").append(entry.getId()).append(" OR ");
+ seqRefs.add(entry.getId());
}
}
if (seq.getDBRef() != null && seq.getDBRef().length != 0)
{
+ int count = 0;
for (DBRefEntry dbRef : seq.getDBRef())
{
- queryBuilder.append("text:")
- .append(dbRef.getAccessionId().replaceAll("GO:", ""))
- .append(" OR ");
+ seqRefs.add(getDBRefId(dbRef));
++count;
if (count > 10)
{
break;
}
}
- int endIndex = queryBuilder.lastIndexOf(" OR ");
- query = queryBuilder.toString().substring(5, endIndex);
}
+
+ StringBuilder queryBuilder = new StringBuilder();
+ for (String seqRef : seqRefs)
+ {
+ queryBuilder.append("text:").append(seqRef).append(" OR ");
+ }
+ int endIndex = queryBuilder.lastIndexOf(" OR ");
+ String query = queryBuilder.toString().substring(5, endIndex);
return query;
}
/**
+ * Ensures sequence ref names are not less than 3 characters and does not
+ * contain a database name
+ *
+ * @param seqName
+ * @return
+ */
+ public static boolean isValidSeqName(String seqName)
+ {
+ String ignoreList = "pdb,uniprot";
+ if (seqName.length() < 3)
+ {
+ return false;
+ }
+ for (String ignoredEntry : ignoreList.split(","))
+ {
+ if (seqName.equalsIgnoreCase(ignoredEntry))
+ {
+ return false;
+ }
+ }
+ return true;
+ }
+
+ public static String getDBRefId(DBRefEntry dbRef)
+ {
+ String ref = dbRef.getAccessionId().replaceAll("GO:", "");
+ return ref;
+ }
+
+ /**
* Filters a given list of discovered structures based on supplied argument
*
* @param fieldToFilterBy
VIEWS_ENTER_ID));
cmb_filterOption.addItem(new FilterOption("From File", "-",
VIEWS_FROM_FILE));
+ cmb_filterOption.addItem(new FilterOption("Cached PDB Entries", "-",
+ VIEWS_LOCAL_PDB));
}
/**
chk_invertFilter.setVisible(true);
filterResultSet(selectedFilterOpt.getValue());
}
- else
+ else if (selectedFilterOpt.getView() == VIEWS_ENTER_ID
+ || selectedFilterOpt.getView() == VIEWS_FROM_FILE)
{
idInputAssSeqPanel.loadCmbAssSeq();
fileChooserAssSeqPanel.loadCmbAssSeq();
btn_view.setEnabled(true);
}
}
+ else if (currentView == VIEWS_LOCAL_PDB)
+ {
+ if (tbl_local_pdb.getSelectedRows().length > 0)
+ {
+ btn_view.setEnabled(true);
+ }
+ }
else if (currentView == VIEWS_ENTER_ID)
{
validateAssociationEnterPdb();
pdbEntry.setType("PDB");
pdbEntriesToView[count++] = pdbEntry;
}
- new StructureViewer(ap.getStructureSelectionManager())
- .viewStructures(ap, pdbEntriesToView,
- ap.av.collateForPDB(pdbEntriesToView));
+
+ launchStructureViewer(ap.getStructureSelectionManager(),
+ pdbEntriesToView, ap, selectedSequences);
+ }
+ else if(currentView == VIEWS_LOCAL_PDB){
+ int[] selectedRows = tbl_local_pdb.getSelectedRows();
+ PDBEntry[] pdbEntriesToView = new PDBEntry[selectedRows.length];
+ int count = 0;
+ for (int row : selectedRows)
+ {
+ String entryKey = tbl_local_pdb.getValueAt(row, 0).toString() + tbl_local_pdb.getValueAt(row, 1).toString();
+ pdbEntriesToView[count++] = cachedEntryMap.get(entryKey);
+ }
+ launchStructureViewer(ap.getStructureSelectionManager(),
+ pdbEntriesToView, ap, selectedSequences);
}
else if (currentView == VIEWS_ENTER_ID)
{
- selectedSequence = ((AssociateSeqOptions) idInputAssSeqPanel
+ SequenceI userSelectedSeq = ((AssociateSeqOptions) idInputAssSeqPanel
.getCmb_assSeq().getSelectedItem()).getSequence();
+ if (userSelectedSeq != null)
+ {
+ selectedSequence = userSelectedSeq;
+ }
PDBEntry pdbEntry = new PDBEntry();
pdbEntry.setId(txt_search.getText());
pdbEntry.setType("PDB");
selectedSequence.getDatasetSequence().addPDBId(pdbEntry);
PDBEntry[] pdbEntriesToView = new PDBEntry[]
{ pdbEntry };
- new StructureViewer(ap.getStructureSelectionManager())
- .viewStructures(ap, pdbEntriesToView,
- ap.av.collateForPDB(pdbEntriesToView));
+ launchStructureViewer(ap.getStructureSelectionManager(),
+ pdbEntriesToView, ap, new SequenceI[]
+ { selectedSequence });
}
else if (currentView == VIEWS_FROM_FILE)
{
- selectedSequence = ((AssociateSeqOptions) fileChooserAssSeqPanel
+ SequenceI userSelectedSeq = ((AssociateSeqOptions) fileChooserAssSeqPanel
.getCmb_assSeq().getSelectedItem()).getSequence();
- new AssociatePdbFileWithSeq().associatePdbWithSeq(
+ if (userSelectedSeq != null)
+ {
+ selectedSequence = userSelectedSeq;
+ }
+ PDBEntry fileEntry = new AssociatePdbFileWithSeq()
+ .associatePdbWithSeq(
selectedPdbFileName, jalview.io.AppletFormatAdapter.FILE,
selectedSequence, true, Desktop.instance);
+
+ launchStructureViewer(ap.getStructureSelectionManager(),
+ new PDBEntry[]
+ { fileEntry }, ap, new SequenceI[]
+ { selectedSequence });
}
mainFrame.dispose();
}
+ private void launchStructureViewer(StructureSelectionManager ssm,
+ PDBEntry[] pdbEntriesToView, AlignmentPanel alignPanel,
+ SequenceI[] selectedSequences)
+ {
+ StructureViewer sViewer = new StructureViewer(ssm);
+ if (pdbEntriesToView.length > 1)
+ {
+ sViewer.viewStructures(alignPanel, pdbEntriesToView,
+ alignPanel.av.collateForPDB(pdbEntriesToView));
+ }
+ else
+ {
+ sViewer.viewStructures(pdbEntriesToView[0], selectedSequences, null,
+ alignPanel);
+ }
+ }
+
/**
* Populates the combo-box used in associating manually fetched structures to
* a unique sequence when more than one sequence selection is made.
git://source.jalview.org / jalview.git / blobdiff