import javax.swing.SwingUtilities;
import javax.swing.table.AbstractTableModel;
+import com.stevesoft.pat.Regex;
+
import jalview.api.structures.JalviewStructureDisplayI;
import jalview.bin.Cache;
import jalview.bin.Console;
import jalview.bin.Jalview;
import jalview.datamodel.PDBEntry;
import jalview.datamodel.SequenceI;
+import jalview.ext.jmol.JmolParser;
import jalview.fts.api.FTSData;
import jalview.fts.api.FTSDataColumnI;
import jalview.fts.api.FTSRestClientI;
import jalview.gui.structurechooser.StructureChooserQuerySource;
import jalview.gui.structurechooser.ThreeDBStructureChooserQuerySource;
import jalview.io.DataSourceType;
+import jalview.io.FileFormatException;
+import jalview.io.JalviewFileChooser;
+import jalview.io.JalviewFileView;
import jalview.jbgui.FilterOption;
import jalview.jbgui.GStructureChooser;
+import jalview.structure.StructureImportSettings.TFType;
import jalview.structure.StructureMapping;
import jalview.structure.StructureSelectionManager;
import jalview.util.MessageManager;
import jalview.util.Platform;
+import jalview.util.StringUtils;
import jalview.ws.DBRefFetcher;
import jalview.ws.DBRefFetcher.FetchFinishedListenerI;
+import jalview.ws.datamodel.alphafold.PAEContactMatrix;
+import jalview.ws.dbsources.EBIAlfaFold;
import jalview.ws.seqfetcher.DbSourceProxy;
import jalview.ws.sifts.SiftsSettings;
private String selectedPdbFileName;
+ private TFType localPdbTempfacType;
+
+ private String localPdbPaeMatrixFileName;
+
private boolean isValidPBDEntry;
private boolean cachedPDBExists;
// TODO: JAL-3048 not needed for Jalview-JS until JSmol dep and
// StructureChooser
// works
- jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(
+ JalviewFileChooser chooser = new JalviewFileChooser(
Cache.getProperty("LAST_DIRECTORY"));
- chooser.setFileView(new jalview.io.JalviewFileView());
+ chooser.setFileView(new JalviewFileView());
chooser.setDialogTitle(
MessageManager.formatMessage("label.select_pdb_file_for",
selectedSequence.getDisplayId(false)));
selectedSequence.getDisplayId(false)));
int value = chooser.showOpenDialog(null);
- if (value == jalview.io.JalviewFileChooser.APPROVE_OPTION)
+ if (value == JalviewFileChooser.APPROVE_OPTION)
{
selectedPdbFileName = chooser.getSelectedFile().getPath();
Cache.setProperty("LAST_DIRECTORY", selectedPdbFileName);
+ boolean guessTFType = localPdbPaeMatrixFileName == null;
+ localPdbPaeMatrixFileName = guessPAEFilename();
+ guessTFType |= localPdbPaeMatrixFileName != null;
+ Regex alphaFold = JmolParser.getNewAlphafoldValidator();
+ if (guessTFType
+ && alphaFold.search(new File(selectedPdbFileName).getName())
+ && !tempFacAsChanged)
+ {
+ // localPdbPaeMatrixFileName was null and now isn't and filename could
+ // well be AlphaFold and user hasn't adjusted the tempFacType
+ combo_tempFacAs.setSelectedItem(TFType.PLDDT);
+ }
validateSelections();
}
}
/**
+ * Handles action event for btn_paeMatrixFile
+ */
+ @Override
+ protected void paeMatrixFile_actionPerformed()
+ {
+ File pdbFile = new File(selectedPdbFileName);
+ String setFile = Cache.getProperty("LAST_DIRECTORY");
+ if (localPdbPaeMatrixFileName != null)
+ {
+ File paeFile = new File(localPdbPaeMatrixFileName);
+ if (paeFile.exists())
+ setFile = paeFile.getAbsolutePath();
+ else if (paeFile.getParentFile().exists())
+ setFile = paeFile.getParentFile().getAbsolutePath();
+ }
+ else
+ {
+ String guess = guessPAEFilename();
+ if (guess != null)
+ setFile = guess;
+ }
+ JalviewFileChooser chooser = new JalviewFileChooser(setFile);
+ chooser.setFileView(new JalviewFileView());
+ chooser.setDialogTitle(MessageManager.formatMessage(
+ "label.select_pae_matrix_file_for", pdbFile.getName()));
+ chooser.setToolTipText(MessageManager.formatMessage(
+ "label.load_pae_matrix_file_associate_with_structure",
+ pdbFile.getName()));
+
+ // TODO convert to Callable/Promise
+ int value = chooser.showOpenDialog(null);
+ if (value == JalviewFileChooser.APPROVE_OPTION)
+ {
+ String fileName = chooser.getSelectedFile().getPath();
+ try {
+ PAEContactMatrix.validateContactMatrixFile(fileName);
+ } catch (Exception thr)
+ {
+ JvOptionPane.showInternalMessageDialog(this, MessageManager
+ .formatMessage("label.couldnt_load_file", new Object[]
+ { fileName})+"<br>"+thr.getLocalizedMessage(),
+ MessageManager.getString("label.error_loading_file"),
+ JvOptionPane.WARNING_MESSAGE);
+ Console.error("Couldn't import "+fileName+" as a PAE matrix",thr);
+ return;
+ }
+ localPdbPaeMatrixFileName = fileName;
+ Cache.setProperty("LAST_DIRECTORY", localPdbPaeMatrixFileName);
+ }
+ validateAssociationFromFile();
+ }
+
+ private String guessPAEFilename()
+ {
+ if (selectedPdbFileName.toLowerCase(Locale.ROOT).endsWith(".pdb")
+ || selectedPdbFileName.toLowerCase(Locale.ROOT)
+ .endsWith(".cif"))
+ {
+ String jsonExt = selectedPdbFileName.substring(0,
+ selectedPdbFileName.length() - 4) + ".json";
+ // AlphaFold naming scheme
+ String guessFile1 = StringUtils.replaceLast(jsonExt, "model",
+ "predicted_aligned_error");
+ // nf-core mode naming scheme
+ String guessFile2 = StringUtils.replaceLast(jsonExt, ".json",
+ "_scores.json");
+ if (new File(guessFile1).exists())
+ {
+ return guessFile1;
+ }
+ else if (new File(jsonExt).exists())
+ {
+ return jsonExt;
+ }
+ else if (new File(guessFile2).exists())
+ {
+ return guessFile2;
+ }
+ }
+ return null;
+ }
+
+ /**
* Populates the filter combo-box options dynamically depending on discovered
* structures
*/
{
AssociateSeqOptions assSeqOpt = (AssociateSeqOptions) fileChooserAssSeqPanel
.getCmb_assSeq().getSelectedItem();
- lbl_fromFileStatus.setIcon(errorImage);
+ // lbl_fromFileStatus.setIcon(errorImage);
+ String pdbFileString = "";
+ String pdbFileTooltip = "";
if (selectedSequences.length == 1 || (assSeqOpt != null && !assSeqOpt
.getName().equalsIgnoreCase("-Select Associated Seq-")))
{
if (selectedPdbFileName != null && selectedPdbFileName.length() > 0)
{
btn_add.setEnabled(true);
- lbl_fromFileStatus.setIcon(goodImage);
+ // lbl_fromFileStatus.setIcon(goodImage);
+ pdbFileString = new File(selectedPdbFileName).getName();
+ pdbFileTooltip = new File(selectedPdbFileName).getAbsolutePath();
+ setPdbOptionsEnabled(true);
+ }
+ else
+ {
+ pdbFileString = MessageManager.getString("label.none");
+ pdbFileTooltip = MessageManager.getString("label.nothing_selected");
+ setPdbOptionsEnabled(false);
}
}
else
{
btn_pdbFromFile.setEnabled(false);
- lbl_fromFileStatus.setIcon(errorImage);
+ setPdbOptionsEnabled(false);
+ // lbl_fromFileStatus.setIcon(errorImage);
+ pdbFileString = MessageManager.getString("label.none");
+ pdbFileTooltip = MessageManager.getString("label.nothing_selected");
+ }
+ lbl_pdbFile.setText(pdbFileString);
+ lbl_pdbFile.setToolTipText(pdbFileTooltip);
+
+ // PAE file choice
+ String paeFileString = "";
+ String paeFileTooltip = "";
+ if (localPdbPaeMatrixFileName != null
+ && localPdbPaeMatrixFileName.length() > 0)
+ {
+ paeFileString = new File(localPdbPaeMatrixFileName).getName();
+ paeFileTooltip = new File(localPdbPaeMatrixFileName)
+ .getAbsolutePath();
+ }
+ else
+ {
+ paeFileString = MessageManager.getString("label.none");
+ paeFileTooltip = MessageManager.getString("label.nothing_selected");
}
+ lbl_paeFile.setText(paeFileString);
+ lbl_paeFile.setToolTipText(paeFileTooltip);
}
@Override
final StructureSelectionManager ssm = ap.getStructureSelectionManager();
final int preferredHeight = pnl_filter.getHeight();
-
+ btn_add.setEnabled(false);
+ btn_newView.setEnabled(false);
Runnable viewStruc = new Runnable()
{
@Override
}
else if (currentView == VIEWS_FROM_FILE)
{
- SequenceI userSelectedSeq = ((AssociateSeqOptions) fileChooserAssSeqPanel
- .getCmb_assSeq().getSelectedItem()).getSequence();
+ StructureChooser sc = StructureChooser.this;
+ TFType tft = (TFType) sc.combo_tempFacAs.getSelectedItem();
+ String paeFilename = sc.localPdbPaeMatrixFileName;
+ AssociateSeqOptions assSeqOpt = (AssociateSeqOptions) fileChooserAssSeqPanel
+ .getCmb_assSeq().getSelectedItem();
+ SequenceI userSelectedSeq = assSeqOpt.getSequence();
if (userSelectedSeq != null)
- {
selectedSequence = userSelectedSeq;
- }
- PDBEntry fileEntry = new AssociatePdbFileWithSeq()
- .associatePdbWithSeq(selectedPdbFileName,
- DataSourceType.FILE, selectedSequence, true,
- Desktop.instance);
+ String pdbFilename = selectedPdbFileName;
- sViewer = launchStructureViewer(ssm, new PDBEntry[] { fileEntry },
- ap, new SequenceI[]
- { selectedSequence });
+ StructureChooser.openStructureFileForSequence(ssm, sc, ap,
+ selectedSequence, true, pdbFilename, tft, paeFilename);
}
SwingUtilities.invokeLater(new Runnable()
{
setProgressBar(MessageManager.formatMessage(
"status.fetching_3d_structures_for",
pdbEntriesToView[0].getId()), progressId);
+ // Can we pass a pre-computeMappinged pdbFile?
theViewer.viewStructures(pdbEntriesToView[0], sequences, alignPanel);
}
setProgressBar(null, progressId);
/**
* Open a single structure file for a given sequence
*/
- public static void openStructureFileForSequence(AlignmentPanel ap,
- SequenceI seq, File sFile)
+ public static void openStructureFileForSequence(
+ StructureSelectionManager ssm, StructureChooser sc,
+ AlignmentPanel ap, SequenceI seq, boolean prompt,
+ String sFilename, TFType tft, String paeFilename)
{
- //Open the chooser headlessly. Not sure this is actually needed ?
- StructureChooser sc = new StructureChooser(new SequenceI[] { seq }, seq,
- ap, false);
- StructureSelectionManager ssm = ap.getStructureSelectionManager();
- PDBEntry fileEntry = null;
- try
- {
- fileEntry = new AssociatePdbFileWithSeq().associatePdbWithSeq(
- sFile.getAbsolutePath(), DataSourceType.FILE, seq, true,
- Desktop.instance);
- } catch (Exception e)
- {
- Console.error("Could not open structure file '"
- + sFile.getAbsolutePath() + "'");
- return;
+ boolean headless = false;
+ if (sc == null)
+ {
+ headless = true;
+ sc = new StructureChooser(new SequenceI[] { seq }, seq, ap, false);
}
+ if (ssm == null)
+ ssm = ap.getStructureSelectionManager();
+
+ PDBEntry fileEntry = new AssociatePdbFileWithSeq().associatePdbWithSeq(
+ sFilename, DataSourceType.FILE, seq, prompt, Desktop.instance,
+ tft, paeFilename);
StructureViewer sViewer = sc.launchStructureViewer(ssm,
new PDBEntry[]
{ fileEntry }, ap, new SequenceI[] { seq });
- sc.mainFrame.dispose();
+ if (headless)
+ sc.mainFrame.dispose();
}
}
git://source.jalview.org / jalview.git / blobdiff