X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fstructures%2Fmodels%2FAAStructureBindingModel.java;h=44d698dafe9f27282aead9cf00cab8af2be94fce;hb=57738a1f3c19b1c3a00bd3ac5108f8cd0af32f99;hp=4dfdc2a4e2ad1d4bb0079f023ec1019446f6feb0;hpb=9c1a9d682a2664d525bfd0f38bae861292dc3921;p=jalview.git diff --git a/src/jalview/structures/models/AAStructureBindingModel.java b/src/jalview/structures/models/AAStructureBindingModel.java index 4dfdc2a..44d698d 100644 --- a/src/jalview/structures/models/AAStructureBindingModel.java +++ b/src/jalview/structures/models/AAStructureBindingModel.java @@ -20,47 +20,59 @@ */ package jalview.structures.models; +import java.util.Locale; + +import java.awt.Color; +import java.io.File; +import java.io.IOException; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.BitSet; +import java.util.HashMap; +import java.util.LinkedHashMap; +import java.util.List; +import java.util.Map; + +import javax.swing.SwingUtilities; + import jalview.api.AlignViewportI; import jalview.api.AlignmentViewPanel; import jalview.api.FeatureRenderer; import jalview.api.SequenceRenderer; import jalview.api.StructureSelectionManagerProvider; import jalview.api.structures.JalviewStructureDisplayI; +import jalview.bin.Console; import jalview.datamodel.AlignmentI; import jalview.datamodel.HiddenColumns; +import jalview.datamodel.MappedFeatures; import jalview.datamodel.PDBEntry; +import jalview.datamodel.SequenceFeature; import jalview.datamodel.SequenceI; +import jalview.ext.rbvi.chimera.JalviewChimeraBinding; +import jalview.gui.AlignmentPanel; +import jalview.gui.Desktop; import jalview.gui.StructureViewer.ViewerType; import jalview.io.DataSourceType; +import jalview.io.StructureFile; import jalview.renderer.seqfeatures.FeatureColourFinder; import jalview.schemes.ColourSchemeI; import jalview.schemes.ResidueProperties; import jalview.structure.AtomSpec; import jalview.structure.AtomSpecModel; +import jalview.structure.StructureCommandI; import jalview.structure.StructureCommandsI; -import jalview.structure.StructureCommandsI.SuperposeData; +import jalview.structure.StructureCommandsI.AtomSpecType; import jalview.structure.StructureListener; import jalview.structure.StructureMapping; import jalview.structure.StructureSelectionManager; import jalview.util.Comparison; import jalview.util.MessageManager; -import java.awt.Color; -import java.util.ArrayList; -import java.util.Arrays; -import java.util.BitSet; -import java.util.HashMap; -import java.util.LinkedHashMap; -import java.util.List; -import java.util.Map; - -import javax.swing.SwingUtilities; - /** * - * A base class to hold common function for protein structure model binding. - * Initial version created by refactoring JMol and Chimera binding models, but - * other structure viewers could in principle be accommodated in future. + * A base class to hold common function for 3D structure model binding. Initial + * version created by refactoring JMol and Chimera binding models, but other + * structure viewers could in principle be accommodated in future. * * @author gmcarstairs * @@ -69,6 +81,45 @@ public abstract class AAStructureBindingModel extends SequenceStructureBindingModel implements StructureListener, StructureSelectionManagerProvider { + /** + * Data bean class to simplify parameterisation in superposeStructures + */ + public static class SuperposeData + { + public String filename; + + public String pdbId; + + public String chain = ""; + + /** + * is the mapped sequence not protein ? + */ + public boolean isRna; + + /* + * The pdb residue number (if any) mapped to columns of the alignment + */ + public int[] pdbResNo; // or use SparseIntArray? + + public String modelId; + + /** + * Constructor + * + * @param width + * width of alignment (number of columns that may potentially + * participate in superposition) + * @param model + * structure viewer model number + */ + public SuperposeData(int width, String model) + { + pdbResNo = new int[width]; + modelId = model; + } + } + private static final int MIN_POS_TO_SUPERPOSE = 4; private static final String COLOURING_STRUCTURES = MessageManager @@ -120,17 +171,22 @@ public abstract class AAStructureBindingModel protected boolean colourBySequence = true; + /** + * true if all sequences appear to be nucleotide + */ private boolean nucleotide; private boolean finishedInit = false; /** - * current set of model filenames loaded in the Jmol instance + * current set of model filenames loaded in the viewer */ protected String[] modelFileNames = null; public String fileLoadingError; + protected Thread externalViewerMonitor; + /** * Constructor * @@ -216,6 +272,7 @@ public abstract class AAStructureBindingModel chains = newchains; return chainmaps > 0; } + public StructureSelectionManager getSsm() { return ssm; @@ -613,7 +670,8 @@ public abstract class AAStructureBindingModel * @return */ protected int findSuperposableResidues(AlignmentI alignment, - BitSet matched, SuperposeData[] structures) + BitSet matched, + AAStructureBindingModel.SuperposeData[] structures) { int refStructure = -1; String[] files = getStructureFiles(); @@ -664,7 +722,7 @@ public abstract class AAStructureBindingModel structures[pdbfnum].chain = chain; } structures[pdbfnum].pdbId = mapping.getPdbId(); - structures[pdbfnum].isRna = theSequence.getRNA() != null; + structures[pdbfnum].isRna = !theSequence.isProtein(); /* * move on to next pdb file (ignore sequences for other chains @@ -828,19 +886,19 @@ public abstract class AAStructureBindingModel } } - SuperposeData[] structures = new SuperposeData[files.length]; + AAStructureBindingModel.SuperposeData[] structures = new AAStructureBindingModel.SuperposeData[files.length]; for (int f = 0; f < files.length; f++) { - structures[f] = new SuperposeData(width, - f + commandGenerator.getModelStartNo()); + structures[f] = new AAStructureBindingModel.SuperposeData(width, + getModelIdForFile(files[f])); } /* * Calculate the superposable alignment columns ('matched'), and the * corresponding structure residue positions (structures.pdbResNo) */ - int refStructure = findSuperposableResidues(alignment, - matched, structures); + int refStructure = findSuperposableResidues(alignment, matched, + structures); /* * require at least 4 positions to be able to execute superposition @@ -848,8 +906,8 @@ public abstract class AAStructureBindingModel int nmatched = matched.cardinality(); if (nmatched < MIN_POS_TO_SUPERPOSE) { - String msg = MessageManager.formatMessage("label.insufficient_residues", - nmatched); + String msg = MessageManager + .formatMessage("label.insufficient_residues", nmatched); error += view.getViewName() + ": " + msg + "; "; continue; } @@ -864,23 +922,28 @@ public abstract class AAStructureBindingModel * Show all as backbone before doing superposition(s) * (residues used for matching will be shown as ribbon) */ - executeCommand(commandGenerator.showBackbone(), false); + // todo better way to ensure synchronous than setting getReply true!! + executeCommands(commandGenerator.showBackbone(), true, null); + AtomSpecType backbone = structures[refStructure].isRna + ? AtomSpecType.PHOSPHATE + : AtomSpecType.ALPHA; /* - * superpose each (other) sequence to it in turn + * superpose each (other) structure to the reference in turn */ for (int i = 0; i < structures.length; i++) { if (i != refStructure) { AtomSpecModel atomSpec = getAtomSpec(structures[i], matched); - String commands = commandGenerator.superposeStructures(refAtoms, - atomSpec); - List replies = executeCommands(true, commands); + List commands = commandGenerator + .superposeStructures(refAtoms, atomSpec, backbone); + List replies = executeCommands(commands, true, null); for (String reply : replies) { // return this error (Chimera only) to the user - if (reply.toLowerCase().contains("unequal numbers of atoms")) + if (reply.toLowerCase(Locale.ROOT) + .contains("unequal numbers of atoms")) { error += "; " + reply; } @@ -892,7 +955,8 @@ public abstract class AAStructureBindingModel return error; } - private AtomSpecModel getAtomSpec(SuperposeData superposeData, + private AtomSpecModel getAtomSpec( + AAStructureBindingModel.SuperposeData superposeData, BitSet matched) { AtomSpecModel model = new AtomSpecModel(); @@ -900,7 +964,7 @@ public abstract class AAStructureBindingModel while (nextColumnMatch != -1) { int pdbResNum = superposeData.pdbResNo[nextColumnMatch]; - model.addRange(superposeData.modelNo, pdbResNum, pdbResNum, + model.addRange(superposeData.modelId, pdbResNum, pdbResNum, superposeData.chain); nextColumnMatch = matched.nextSetBit(nextColumnMatch + 1); } @@ -929,8 +993,8 @@ public abstract class AAStructureBindingModel // TODO: JAL-628 colour chains distinctly across all visible models - executeCommand(commandGenerator.colourByChain(), false, - COLOURING_STRUCTURES); + executeCommand(false, COLOURING_STRUCTURES, + commandGenerator.colourByChain()); } /** @@ -941,7 +1005,7 @@ public abstract class AAStructureBindingModel { colourBySequence = false; - executeCommand(commandGenerator.colourByCharge(), false, + executeCommands(commandGenerator.colourByCharge(), false, COLOURING_STRUCTURES); } @@ -961,7 +1025,7 @@ public abstract class AAStructureBindingModel { return; } - + /* * build a map of {Residue3LetterCode, Color} */ @@ -980,143 +1044,139 @@ public abstract class AAStructureBindingModel /* * pass to the command constructor, and send the command */ - String cmd = commandGenerator.colourByResidues(colours); - executeCommand(cmd, false, COLOURING_STRUCTURES); + List cmd = commandGenerator + .colourByResidues(colours); + executeCommands(cmd, false, COLOURING_STRUCTURES); } public void setBackgroundColour(Color col) { - String cmd = commandGenerator.setBackgroundColour(col); - executeCommand(cmd, false, null); + StructureCommandI cmd = commandGenerator.setBackgroundColour(col); + executeCommand(false, null, cmd); } /** - * Sends one command to the structure viewer. If {@code getReply} is true, the - * command is sent synchronously, otherwise in a deferred thread. - *

- * If a progress message is supplied, this is displayed before command - * execution, and removed afterwards. + * Execute one structure viewer command. If {@code getReply} is true, may + * optionally return one or more reply messages, else returns null. * * @param cmd * @param getReply + */ + protected abstract List executeCommand(StructureCommandI cmd, + boolean getReply); + + /** + * Executes one or more structure viewer commands + * + * @param commands + * @param getReply * @param msg - * @return */ - private List executeCommand(String cmd, boolean getReply, - String msg) + protected List executeCommands(List commands, + boolean getReply, String msg) { - if (getReply) - { - return executeSynchronous(cmd, msg, getReply); - } - else - { - executeAsynchronous(cmd, msg); - return null; - } + return executeCommand(getReply, msg, + commands.toArray(new StructureCommandI[commands.size()])); } /** - * Sends the command in the current thread. If a message is supplied, this is - * shown before the thread is started, and removed when it completes. May - * return a reply to the command if requested. + * Executes one or more structure viewer commands, optionally returning the + * reply, and optionally showing a status message while the command is being + * executed. + *

+ * If a reply is wanted, the execution is done synchronously (waits), + * otherwise it is done in a separate thread (doesn't wait). WARNING: if you + * are sending commands that need to execute before later calls to + * executeCommand (e.g. mouseovers, which clean up after previous ones) then + * set getReply true to ensure that commands are not executed out of order. * - * @param cmd - * @param msg * @param getReply + * @param msg + * @param cmds * @return */ - private List executeSynchronous(String cmd, String msg, boolean getReply) + protected List executeCommand(boolean getReply, String msg, + StructureCommandI... cmds) { - final JalviewStructureDisplayI theViewer = getViewer(); + JalviewStructureDisplayI theViewer = getViewer(); final long handle = msg == null ? 0 : theViewer.startProgressBar(msg); - try - { - return executeCommand(cmd, getReply); - } finally + + if (getReply) { - if (msg != null) + /* + * execute and wait for reply + */ + List response = new ArrayList<>(); + try { - theViewer.stopProgressBar(null, handle); + for (StructureCommandI cmd : cmds) + { + List replies = executeCommand(cmd, true); + if (replies != null) + { + response.addAll(replies); + } + } + return response; + } finally + { + if (msg != null) + { + theViewer.stopProgressBar(null, handle); + } } } - } - /** - * Sends the command in a separate thread. If a message is supplied, this is - * shown before the thread is started, and removed when it completes. No value - * is returned. - * - * @param cmd - * @param msg - */ - private void executeAsynchronous(String cmd, String msg) - { - final JalviewStructureDisplayI theViewer = getViewer(); - final long handle = msg == null ? 0 : theViewer.startProgressBar(msg); - - SwingUtilities.invokeLater(new Runnable() + /* + * fire and forget + */ + String threadName = msg == null ? "StructureCommand" : msg; + new Thread(new Runnable() { @Override public void run() { try { - executeCommand(cmd, false); + for (StructureCommandI cmd : cmds) + { + executeCommand(cmd, false); + } } finally { if (msg != null) { - theViewer.stopProgressBar(null, handle); + SwingUtilities.invokeLater(new Runnable() + { + @Override + public void run() + { + theViewer.stopProgressBar(null, handle); + } + }); } } } - }); - } - - protected abstract List executeCommand(String command, - boolean getReply); - - protected List executeCommands(boolean getReply, - String... commands) - { - // todo: tidy this up - List response = getReply ? new ArrayList<>() : null; - for (String cmd : commands) - { - List replies = executeCommand(cmd, getReply); - if (getReply && replies != null) - { - response.addAll(replies); - } - } - return response; + }, threadName).start(); + return null; } /** - * colour any structures associated with sequences in the given alignment - * using the getFeatureRenderer() and getSequenceRenderer() renderers but only - * if colourBySequence is enabled. + * Colours any structures associated with sequences in the given alignment as + * coloured in the alignment view, provided colourBySequence is enabled */ public void colourBySequence(AlignmentViewPanel alignmentv) { - if (!colourBySequence || !isLoadingFinished()) - { - return; - } - if (getSsm() == null) + if (!colourBySequence || !isLoadingFinished() || getSsm() == null) { return; } - String[] files = getStructureFiles(); - - SequenceRenderer sr = getSequenceRenderer(alignmentv); - Map colourMap = buildColoursMap(ssm, files, - sequence, sr, alignmentv); + Map colourMap = buildColoursMap(ssm, sequence, + alignmentv); - String[] colourBySequenceCommands = commandGenerator + List colourBySequenceCommands = commandGenerator .colourBySequence(colourMap); - executeCommands(false, colourBySequenceCommands); + executeCommands(colourBySequenceCommands, false, COLOURING_STRUCTURES); } /** @@ -1124,7 +1184,7 @@ public abstract class AAStructureBindingModel */ public void focusView() { - executeCommand(commandGenerator.focusView(), false); + executeCommand(false, null, commandGenerator.focusView()); } /** @@ -1150,22 +1210,21 @@ public abstract class AAStructureBindingModel if (tokens.length == 2) { String pdbFile = getFileForChain(chainId); - int modelNo = getModelNoForFile(pdbFile); - String model = modelNo == -1 ? "" : String.valueOf(modelNo); + String model = getModelIdForFile(pdbFile); showThese.add(model + ":" + tokens[1]); } } - executeCommand(commandGenerator.showChains(showThese), false); + executeCommands(commandGenerator.showChains(showThese), false, null); } /** - * Answers the structure viewer's model number given a PDB file name. Returns - * -1 if model number is not found. + * Answers the structure viewer's model id given a PDB file name. Returns an + * empty string if model id is not found. * * @param chainId * @return */ - protected abstract int getModelNoForFile(String chainId); + protected abstract String getModelIdForFile(String chainId); public boolean hasFileLoadingError() { @@ -1173,8 +1232,7 @@ public abstract class AAStructureBindingModel } /** - * Returns the FeatureRenderer for the given alignment view, or null if - * feature display is turned off in the view. + * Returns the FeatureRenderer for the given alignment view * * @param avp * @return @@ -1183,9 +1241,11 @@ public abstract class AAStructureBindingModel { AlignmentViewPanel ap = (avp == null) ? getViewer().getAlignmentPanel() : avp; - return ap.getAlignViewport().isShowSequenceFeatures() - ? ap.getFeatureRenderer() - : null; + if (ap == null) + { + return null; + } + return ap.getFeatureRenderer(); } protected void setStructureCommands(StructureCommandsI cmd) @@ -1240,60 +1300,31 @@ public abstract class AAStructureBindingModel protected abstract ViewerType getViewerType(); /** - * Send a structure viewer command asynchronously in a new thread. If the - * progress message is not null, display this message while the command is - * executing. + * Builds a data structure which records mapped structure residues for each + * colour. From this we can easily generate the viewer commands for colour by + * sequence. Constructs and returns a map of {@code Color} to + * {@code AtomSpecModel}, where the atomspec model holds * - * @param command - * @param progressMsg - */ - protected void sendAsynchronousCommand(String command, String progressMsg) - { - final JalviewStructureDisplayI theViewer = getViewer(); - final long handle = progressMsg == null ? 0 - : theViewer.startProgressBar(progressMsg); - SwingUtilities.invokeLater(new Runnable() - { - @Override - public void run() - { - try - { - executeCommand(command, false); - } finally - { - if (progressMsg != null) - { - theViewer.stopProgressBar(null, handle); - } - } - } - }); - - } - - /** * 

-   * Build a data structure which records residues for each colour. 
-   * From this we can easily generate the viewer command for colour by sequence.
-   * Color
-   *     Model number
-   *         Chain
-   *             Residue positions
-   * Ordering is by order of addition (for colours and positions), natural ordering (for models and chains)
+   *   Model ids
+   *     Chains
+   *       Residue positions
    *
* + * Ordering is by order of addition (for colours), natural ordering (for + * models and chains) + * * @param ssm - * @param files * @param sequence - * @param sr * @param viewPanel * @return */ protected Map buildColoursMap( - StructureSelectionManager ssm, String[] files, - SequenceI[][] sequence, SequenceRenderer sr, AlignmentViewPanel viewPanel) + StructureSelectionManager ssm, SequenceI[][] sequence, + AlignmentViewPanel viewPanel) { + String[] files = getStructureFiles(); + SequenceRenderer sr = getSequenceRenderer(viewPanel); FeatureRenderer fr = viewPanel.getFeatureRenderer(); FeatureColourFinder finder = new FeatureColourFinder(fr); AlignViewportI viewport = viewPanel.getAlignViewport(); @@ -1301,17 +1332,17 @@ public abstract class AAStructureBindingModel AlignmentI al = viewport.getAlignment(); Map colourMap = new LinkedHashMap<>(); Color lastColour = null; - + for (int pdbfnum = 0; pdbfnum < files.length; pdbfnum++) { - final int modelNumber = pdbfnum + commandGenerator.getModelStartNo(); + final String modelId = getModelIdForFile(files[pdbfnum]); StructureMapping[] mapping = ssm.getMapping(files[pdbfnum]); - + if (mapping == null || mapping.length < 1) { continue; } - + int startPos = -1, lastPos = -1; String lastChain = ""; for (int s = 0; s < sequence[pdbfnum].length; s++) @@ -1331,14 +1362,14 @@ public abstract class AAStructureBindingModel continue; } int pos = mapping[m].getPDBResNum(asp.findPosition(r)); - + if (pos < 1 || pos == lastPos) { continue; } - + Color colour = sr.getResidueColour(seq, r, finder); - + /* * darker colour for hidden regions */ @@ -1346,9 +1377,9 @@ public abstract class AAStructureBindingModel { colour = Color.GRAY; } - + final String chain = mapping[m].getChain(); - + /* * Just keep incrementing the end position for this colour range * _unless_ colour, PDB model or chain has changed, or there is a @@ -1361,8 +1392,8 @@ public abstract class AAStructureBindingModel { if (startPos != -1) { - addAtomSpecRange(colourMap, lastColour, modelNumber, - startPos, lastPos, lastChain); + addAtomSpecRange(colourMap, lastColour, modelId, startPos, + lastPos, lastChain); } startPos = pos; } @@ -1373,7 +1404,7 @@ public abstract class AAStructureBindingModel // final colour range if (lastColour != null) { - addAtomSpecRange(colourMap, lastColour, modelNumber, startPos, + addAtomSpecRange(colourMap, lastColour, modelId, startPos, lastPos, lastChain); } // break; @@ -1385,8 +1416,38 @@ public abstract class AAStructureBindingModel } /** - * Helper method to add one contiguous range to the AtomSpec model for the given - * value (creating the model if necessary). As used by Jalview, {@code value} is + * todo better refactoring (map lookup or similar to get viewer structure id) + * + * @param pdbfnum + * @param file + * @return + */ + protected String getModelId(int pdbfnum, String file) + { + return String.valueOf(pdbfnum); + } + + /** + * Saves chains, formatted as "pdbId:chainCode", and lookups from this to the + * full PDB file path + * + * @param pdb + * @param file + */ + public void stashFoundChains(StructureFile pdb, String file) + { + for (int i = 0; i < pdb.getChains().size(); i++) + { + String chid = pdb.getId() + ":" + pdb.getChains().elementAt(i).id; + addChainFile(chid, file); + getChainNames().add(chid); + } + } + + /** + * Helper method to add one contiguous range to the AtomSpec model for the + * given value (creating the model if necessary). As used by Jalview, + * {@code value} is *
    *
  • a colour, when building a 'colour structure by sequence' command
    • *
  • a feature value, when building a 'set Chimera attributes from features' @@ -1401,8 +1462,8 @@ public abstract class AAStructureBindingModel * @param chain */ public static final void addAtomSpecRange(Map map, - Object value, - int model, int startPos, int endPos, String chain) + Object value, String model, int startPos, int endPos, + String chain) { /* * Get/initialize map of data for the colour @@ -1413,7 +1474,486 @@ public abstract class AAStructureBindingModel atomSpec = new AtomSpecModel(); map.put(value, atomSpec); } - + atomSpec.addRange(model, startPos, endPos, chain); } + + /** + * Returns the file extension (including '.' separator) to use for a saved + * viewer session file. Default is to return null (not supported), override as + * required. + * + * @return + */ + public String getSessionFileExtension() + { + return null; + } + + /** + * If supported, saves the state of the structure viewer to a temporary file + * and returns the file. Returns null and logs an error on any failure. + * + * @return + */ + public File saveSession() + { + String prefix = getViewerType().toString(); + String suffix = getSessionFileExtension(); + File f = null; + try + { + f = File.createTempFile(prefix, suffix); + saveSession(f); + } catch (IOException e) + { + Console.error(String.format("Error saving %s session: %s", prefix, + e.toString())); + } + + return f; + } + + /** + * Saves the structure viewer session to the given file + * + * @param f + */ + protected void saveSession(File f) + { + StructureCommandI cmd = commandGenerator.saveSession(f.getPath()); + if (cmd != null) + { + executeCommand(cmd, false); + } + } + + /** + * Returns true if the viewer is an external structure viewer for which the + * process is still alive, else false (for Jmol, or an external viewer which + * the user has independently closed) + * + * @return + */ + public boolean isViewerRunning() + { + return false; + } + + /** + * Closes Jalview's structure viewer panel and releases associated resources. + * If it is managing an external viewer program, and {@code forceClose} is + * true, also asks that program to close. + * + * @param forceClose + */ + public void closeViewer(boolean forceClose) + { + getSsm().removeStructureViewerListener(this, this.getStructureFiles()); + releaseUIResources(); + + /* + * end the thread that closes this panel if the external viewer closes + */ + if (externalViewerMonitor != null) + { + externalViewerMonitor.interrupt(); + externalViewerMonitor = null; + } + + stopListening(); + + if (forceClose) + { + StructureCommandI cmd = getCommandGenerator().closeViewer(); + if (cmd != null) + { + executeCommand(cmd, false); + } + } + } + + /** + * Returns the URL of a help page for the structure viewer, or null if none is + * known + * + * @return + */ + public String getHelpURL() + { + return null; + } + + /** + * 
    +   * Helper method to build a map of 
+   *   { featureType, { feature value, AtomSpecModel } }
+   *
    + * + * @param viewPanel + * @return + */ + protected Map> buildFeaturesMap( + AlignmentViewPanel viewPanel) + { + Map> theMap = new LinkedHashMap<>(); + String[] files = getStructureFiles(); + if (files == null) + { + return theMap; + } + + FeatureRenderer fr = viewPanel.getFeatureRenderer(); + if (fr == null) + { + return theMap; + } + + AlignViewportI viewport = viewPanel.getAlignViewport(); + List visibleFeatures = fr.getDisplayedFeatureTypes(); + + /* + * if alignment is showing features from complement, we also transfer + * these features to the corresponding mapped structure residues + */ + boolean showLinkedFeatures = viewport.isShowComplementFeatures(); + List complementFeatures = new ArrayList<>(); + FeatureRenderer complementRenderer = null; + if (showLinkedFeatures) + { + AlignViewportI comp = fr.getViewport().getCodingComplement(); + if (comp != null) + { + complementRenderer = Desktop.getAlignFrameFor(comp) + .getFeatureRenderer(); + complementFeatures = complementRenderer.getDisplayedFeatureTypes(); + } + } + if (visibleFeatures.isEmpty() && complementFeatures.isEmpty()) + { + return theMap; + } + + AlignmentI alignment = viewPanel.getAlignment(); + SequenceI[][] seqs = getSequence(); + + for (int pdbfnum = 0; pdbfnum < files.length; pdbfnum++) + { + String modelId = getModelIdForFile(files[pdbfnum]); + StructureMapping[] mapping = ssm.getMapping(files[pdbfnum]); + + if (mapping == null || mapping.length < 1) + { + continue; + } + + for (int seqNo = 0; seqNo < seqs[pdbfnum].length; seqNo++) + { + for (int m = 0; m < mapping.length; m++) + { + final SequenceI seq = seqs[pdbfnum][seqNo]; + int sp = alignment.findIndex(seq); + StructureMapping structureMapping = mapping[m]; + if (structureMapping.getSequence() == seq && sp > -1) + { + /* + * found a sequence with a mapping to a structure; + * now scan its features + */ + if (!visibleFeatures.isEmpty()) + { + scanSequenceFeatures(visibleFeatures, structureMapping, seq, + theMap, modelId); + } + if (showLinkedFeatures) + { + scanComplementFeatures(complementRenderer, structureMapping, + seq, theMap, modelId); + } + } + } + } + } + return theMap; + } + + /** + * Ask the structure viewer to open a session file. Returns true if + * successful, else false (or not supported). + * + * @param filepath + * @return + */ + public boolean openSession(String filepath) + { + StructureCommandI cmd = getCommandGenerator().openSession(filepath); + if (cmd == null) + { + return false; + } + executeCommand(cmd, true); + // todo: test for failure - how? + return true; + } + + /** + * Scans visible features in mapped positions of the CDS/peptide complement, + * and adds any found to the map of attribute values/structure positions + * + * @param complementRenderer + * @param structureMapping + * @param seq + * @param theMap + * @param modelNumber + */ + protected static void scanComplementFeatures( + FeatureRenderer complementRenderer, + StructureMapping structureMapping, SequenceI seq, + Map> theMap, + String modelNumber) + { + /* + * for each sequence residue mapped to a structure position... + */ + for (int seqPos : structureMapping.getMapping().keySet()) + { + /* + * find visible complementary features at mapped position(s) + */ + MappedFeatures mf = complementRenderer + .findComplementFeaturesAtResidue(seq, seqPos); + if (mf != null) + { + for (SequenceFeature sf : mf.features) + { + String type = sf.getType(); + + /* + * Don't copy features which originated from Chimera + */ + if (JalviewChimeraBinding.CHIMERA_FEATURE_GROUP + .equals(sf.getFeatureGroup())) + { + continue; + } + + /* + * record feature 'value' (score/description/type) as at the + * corresponding structure position + */ + List mappedRanges = structureMapping + .getPDBResNumRanges(seqPos, seqPos); + + if (!mappedRanges.isEmpty()) + { + String value = sf.getDescription(); + if (value == null || value.length() == 0) + { + value = type; + } + float score = sf.getScore(); + if (score != 0f && !Float.isNaN(score)) + { + value = Float.toString(score); + } + Map featureValues = theMap.get(type); + if (featureValues == null) + { + featureValues = new HashMap<>(); + theMap.put(type, featureValues); + } + for (int[] range : mappedRanges) + { + addAtomSpecRange(featureValues, value, modelNumber, range[0], + range[1], structureMapping.getChain()); + } + } + } + } + } + } + + /** + * Inspect features on the sequence; for each feature that is visible, + * determine its mapped ranges in the structure (if any) according to the + * given mapping, and add them to the map. + * + * @param visibleFeatures + * @param mapping + * @param seq + * @param theMap + * @param modelId + */ + protected static void scanSequenceFeatures(List visibleFeatures, + StructureMapping mapping, SequenceI seq, + Map> theMap, String modelId) + { + List sfs = seq.getFeatures().getPositionalFeatures( + visibleFeatures.toArray(new String[visibleFeatures.size()])); + for (SequenceFeature sf : sfs) + { + String type = sf.getType(); + + /* + * Don't copy features which originated from Chimera + */ + if (JalviewChimeraBinding.CHIMERA_FEATURE_GROUP + .equals(sf.getFeatureGroup())) + { + continue; + } + + List mappedRanges = mapping.getPDBResNumRanges(sf.getBegin(), + sf.getEnd()); + + if (!mappedRanges.isEmpty()) + { + String value = sf.getDescription(); + if (value == null || value.length() == 0) + { + value = type; + } + float score = sf.getScore(); + if (score != 0f && !Float.isNaN(score)) + { + value = Float.toString(score); + } + Map featureValues = theMap.get(type); + if (featureValues == null) + { + featureValues = new HashMap<>(); + theMap.put(type, featureValues); + } + for (int[] range : mappedRanges) + { + addAtomSpecRange(featureValues, value, modelId, range[0], + range[1], mapping.getChain()); + } + } + } + } + + /** + * Returns the number of structure files in the structure viewer and mapped to + * Jalview. This may be zero if the files are still in the process of loading + * in the viewer. + * + * @return + */ + public int getMappedStructureCount() + { + String[] files = getStructureFiles(); + return files == null ? 0 : files.length; + } + + /** + * Starts a thread that waits for the external viewer program process to + * finish, so that we can then close the associated resources. This avoids + * leaving orphaned viewer panels in Jalview if the user closes the external + * viewer. + * + * @param p + */ + protected void startExternalViewerMonitor(Process p) + { + externalViewerMonitor = new Thread(new Runnable() + { + + @Override + public void run() + { + try + { + p.waitFor(); + JalviewStructureDisplayI display = getViewer(); + if (display != null) + { + display.closeViewer(false); + } + } catch (InterruptedException e) + { + // exit thread if Chimera Viewer is closed in Jalview + } + } + }); + externalViewerMonitor.start(); + } + + /** + * If supported by the external structure viewer, sends it commands to notify + * model or selection changes to the specified URL (where Jalview has started + * a listener) + * + * @param uri + */ + protected void startListening(String uri) + { + List commands = getCommandGenerator() + .startNotifications(uri); + if (commands != null) + { + executeCommands(commands, false, null); + } + } + + /** + * If supported by the external structure viewer, sends it commands to stop + * notifying model or selection changes + */ + protected void stopListening() + { + List commands = getCommandGenerator() + .stopNotifications(); + if (commands != null) + { + executeCommands(commands, false, null); + } + } + + /** + * If supported by the structure viewer, queries it for all residue attributes + * with the given attribute name, and creates features on corresponding + * residues of the alignment. Returns the number of features added. + * + * @param attName + * @param alignmentPanel + * @return + */ + public int copyStructureAttributesToFeatures(String attName, + AlignmentPanel alignmentPanel) + { + StructureCommandI cmd = getCommandGenerator() + .getResidueAttributes(attName); + if (cmd == null) + { + return 0; + } + List residueAttributes = executeCommand(cmd, true); + + int featuresAdded = createFeaturesForAttributes(attName, + residueAttributes); + if (featuresAdded > 0) + { + alignmentPanel.getFeatureRenderer().featuresAdded(); + } + return featuresAdded; + } + + /** + * Parses {@code residueAttributes} and creates sequence features on any + * mapped alignment residues. Returns the number of features created. + *

    + * {@code residueAttributes} is the reply from the structure viewer to a + * command to list any residue attributes for the given attribute name. Syntax + * and parsing of this is viewer-specific. + * + * @param attName + * @param residueAttributes + * @return + */ + protected int createFeaturesForAttributes(String attName, + List residueAttributes) + { + return 0; + } }