import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
+import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Locale;
-import java.util.concurrent.Callable;
+import java.util.Map;
import java.util.concurrent.Executors;
import javax.swing.JCheckBox;
import com.stevesoft.pat.Regex;
+import jalview.analysis.AlignmentUtils;
+import jalview.api.AlignmentViewPanel;
import jalview.api.structures.JalviewStructureDisplayI;
import jalview.bin.Cache;
import jalview.bin.Console;
import jalview.bin.Jalview;
+import jalview.datamodel.AlignmentAnnotation;
+import jalview.datamodel.AlignmentI;
import jalview.datamodel.PDBEntry;
+import jalview.datamodel.SequenceGroup;
import jalview.datamodel.SequenceI;
import jalview.ext.jmol.JmolParser;
import jalview.fts.api.FTSData;
import jalview.fts.core.FTSRestResponse;
import jalview.fts.service.pdb.PDBFTSRestClient;
import jalview.fts.service.threedbeacons.TDB_FTSData;
+import jalview.gui.StructureViewer.ViewerType;
import jalview.gui.structurechooser.PDBStructureChooserQuerySource;
import jalview.gui.structurechooser.StructureChooserQuerySource;
import jalview.gui.structurechooser.ThreeDBStructureChooserQuerySource;
};
// fetch db refs if OK pressed
- final Callable discoverCanonicalDBrefs = () -> {
+ final Runnable discoverCanonicalDBrefs = () -> {
btn_queryTDB.setEnabled(false);
populateSeqsWithoutSourceDBRef();
// call finished action directly
afterDbRefFetch.finished();
}
- return null;
};
- final Callable revertview = () -> {
+ final Runnable revertview = () -> {
if (lastSelected != null)
{
cmb_filterOption.setSelectedItem(lastSelected);
}
- return null;
};
int threshold = Cache.getDefault("UNIPROT_AUTOFETCH_THRESHOLD",
THRESHOLD_WARN_UNIPROT_FETCH_NEEDED);
}
else if (currentView == VIEWS_FROM_FILE)
{
- TFType tft = (TFType) StructureChooser.this.combo_tempFacAs
- .getSelectedItem();
- String paeFilename = StructureChooser.this.localPdbPaeMatrixFileName;
+ 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();
selectedSequence = userSelectedSeq;
String pdbFilename = selectedPdbFileName;
- PDBEntry fileEntry = new AssociatePdbFileWithSeq()
- .associatePdbWithSeq(pdbFilename, DataSourceType.FILE,
- selectedSequence, true, Desktop.instance, tft,
- paeFilename);
-
- /*
- SequenceI[] seqArray = new SequenceI[] { selectedSequence };
-
- StructureFile sf = ssm.computeMapping(true, seqArray, null,
- selectedPdbFileName, DataSourceType.FILE, null, tft,
- paeFilename);
- StructureMapping[] sm = ssm.getMapping(fileEntry.getFile());
- // DO SOMETHING WITH
- File paeFile = paeFilename == null ? null : new File(paeFilename);
- if (paeFilename != null && paeFile.exists())
- {
- AlignmentI al = StructureChooser.this.ap.getAlignment();
- try
- {
- EBIAlfaFold.importPaeJSONAsContactMatrixToSequence(al,
- paeFile, selectedSequence);
- } catch (IOException | ParseException e)
- {
- // TODO Auto-generated catch block
- e.printStackTrace();
- }
- }
- */
- sViewer = launchStructureViewer(ssm, new PDBEntry[] { fileEntry },
- ap, new SequenceI[]
- { selectedSequence });
+ StructureChooser.openStructureFileForSequence(ssm, sc, ap,
+ selectedSequence, true, pdbFilename, tft, paeFilename,
+ true);
}
SwingUtilities.invokeLater(new Runnable()
{
StructureSelectionManager ssm, final PDBEntry[] pdbEntriesToView,
final AlignmentPanel alignPanel, SequenceI[] sequences)
{
+ return launchStructureViewer(ssm, pdbEntriesToView, alignPanel,
+ sequences, null);
+ }
+
+ private StructureViewer launchStructureViewer(
+ StructureSelectionManager ssm, final PDBEntry[] pdbEntriesToView,
+ final AlignmentPanel alignPanel, SequenceI[] sequences,
+ ViewerType viewerType)
+ {
long progressId = sequences.hashCode();
setProgressBar(MessageManager
.getString("status.launching_3d_structure_viewer"), progressId);
for (SequenceI seq : sequences)
{
PDBEntry pdbe = pdbEntriesToView[p++];
- Console.debug(
- "##### pdbe=" + pdbe == null ? null : pdbe.toString());
- Console.debug("##### pdbe.getFile()=" + pdbe == null ? null
- : pdbe.getFile());
if (pdbe != null && pdbe.getFile() != null)
{
StructureMapping[] smm = ssm.getMapping(pdbe.getFile());
MessageManager.getString(
"status.fetching_3d_structures_for_selected_entries"),
progressId);
- theViewer.viewStructures(pdbEntriesToView, sequences, alignPanel);
+ theViewer.viewStructures(pdbEntriesToView, sequences, alignPanel,
+ viewerType);
}
else
{
"status.fetching_3d_structures_for",
pdbEntriesToView[0].getId()), progressId);
// Can we pass a pre-computeMappinged pdbFile?
- theViewer.viewStructures(pdbEntriesToView[0], sequences, alignPanel);
+ theViewer.viewStructures(pdbEntriesToView[0], sequences, alignPanel,
+ viewerType);
}
setProgressBar(null, progressId);
// remember the last viewer we used...
@Override
public void setProgressBar(String message, long id)
{
- if (!Platform.isHeadless())
+ if (!Platform.isHeadless() && progressBar != null)
progressBar.setProgressBar(message, id);
}
@Override
public void registerHandler(long id, IProgressIndicatorHandler handler)
{
- progressBar.registerHandler(id, handler);
+ if (progressBar != null)
+ progressBar.registerHandler(id, handler);
}
@Override
public boolean operationInProgress()
{
- return progressBar.operationInProgress();
+ return progressBar == null ? false : progressBar.operationInProgress();
}
public JalviewStructureDisplayI getOpenedStructureViewer()
/**
* 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,
+ boolean doXferSettings)
{
- // 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;
+ openStructureFileForSequence(ssm, sc, ap, seq, prompt, sFilename, tft,
+ paeFilename, false, true, doXferSettings, null);
+ }
+
+ public static StructureViewer openStructureFileForSequence(
+ StructureSelectionManager ssm, StructureChooser sc,
+ AlignmentPanel ap, SequenceI seq, boolean prompt,
+ String sFilename, TFType tft, String paeFilename,
+ boolean forceHeadless, boolean showRefAnnotations,
+ boolean doXferSettings, ViewerType viewerType)
+ {
+ StructureViewer sv = null;
+ boolean headless = forceHeadless;
+ if (sc == null)
+ {
+ // headless = true;
+ prompt = false;
+ sc = new StructureChooser(new SequenceI[] { seq }, seq, ap, false);
+ }
+ if (ssm == null)
+ {
+ ssm = ap.getStructureSelectionManager();
}
- StructureViewer sViewer = sc.launchStructureViewer(ssm,
- new PDBEntry[]
- { fileEntry }, ap, new SequenceI[] { seq });
+ PDBEntry fileEntry = new AssociatePdbFileWithSeq().associatePdbWithSeq(
+ sFilename, DataSourceType.FILE, seq, prompt, Desktop.instance,
+ tft, paeFilename, doXferSettings);
+
+ // if headless, "false" in the sc constructor above will avoid GUI behaviour
+ // in sc.launchStructureViewer()
+ if (!headless && !(viewerType == null))
+ {
+ sv = sc.launchStructureViewer(ssm, new PDBEntry[] { fileEntry }, ap,
+ new SequenceI[]
+ { seq }, viewerType);
+ }
sc.mainFrame.dispose();
+
+ if (showRefAnnotations)
+ showReferenceAnnotationsForSequence(ap.alignFrame, seq);
+
+ return sv;
+ }
+
+ public static void showReferenceAnnotationsForSequence(AlignFrame af,
+ SequenceI sequence)
+ {
+ AlignViewport av = af.getCurrentView();
+ AlignmentI al = av.getAlignment();
+
+ List<SequenceI> forSequences = new ArrayList<>();
+ forSequences.add(sequence);
+ final Map<SequenceI, List<AlignmentAnnotation>> candidates = new LinkedHashMap<>();
+ AlignmentUtils.findAddableReferenceAnnotations(forSequences, null,
+ candidates, al);
+ final SequenceGroup selectionGroup = av.getSelectionGroup();
+ AlignmentUtils.addReferenceAnnotations(candidates, al, selectionGroup);
+ for (AlignmentViewPanel ap : af.getAlignPanels())
+ {
+ // required to readjust the height and position of the PAE
+ // annotation
+ ap.adjustAnnotationHeight();
+ }
+
}
}