JAL-3400 include sequence name in View | Show Chain menu
[jalview.git] / src / jalview / structures / models / AAStructureBindingModel.java
index ef4fed3..b55885d 100644 (file)
@@ -22,6 +22,7 @@ package jalview.structures.models;
 
 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;
@@ -29,12 +30,14 @@ import jalview.datamodel.AlignmentI;
 import jalview.datamodel.HiddenColumns;
 import jalview.datamodel.PDBEntry;
 import jalview.datamodel.SequenceI;
+import jalview.ext.rbvi.chimera.AtomSpecModel;
+import jalview.ext.rbvi.chimera.ChimeraCommands;
 import jalview.io.DataSourceType;
+import jalview.renderer.seqfeatures.FeatureColourFinder;
 import jalview.schemes.ColourSchemeI;
 import jalview.structure.AtomSpec;
 import jalview.structure.StructureListener;
 import jalview.structure.StructureMapping;
-import jalview.structure.StructureMappingcommandSet;
 import jalview.structure.StructureSelectionManager;
 import jalview.util.Comparison;
 import jalview.util.MessageManager;
@@ -43,7 +46,11 @@ import java.awt.Color;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.BitSet;
+import java.util.Collections;
+import java.util.Iterator;
+import java.util.LinkedHashMap;
 import java.util.List;
+import java.util.Map;
 
 /**
  * 
@@ -101,11 +108,13 @@ public abstract class AAStructureBindingModel
    * a list of chains "pdbid:chainid" to show in the viewer;
    * empty means show all
    */
-  // TODO make private once showStructures() deals with this
+  // TODO make private once deprecated JalviewJmolBinding.centerViewer removed
   protected List<String> chainsToShow;
 
   private boolean hideHiddenRegions;
 
+  protected List<String> chainNames = new ArrayList<>();
+
   /**
    * Data bean class to simplify parameterisation in superposeStructures
    */
@@ -747,13 +756,6 @@ public abstract class AAStructureBindingModel
   }
 
   /**
-   * Returns a list of chains mapped in this viewer.
-   * 
-   * @return
-   */
-  public abstract List<String> getChainNames();
-
-  /**
    * Returns the Jalview panel hosting the structure viewer (if any)
    * 
    * @return
@@ -786,7 +788,7 @@ public abstract class AAStructureBindingModel
 
   public abstract void setBackgroundColour(Color col);
 
-  protected abstract StructureMappingcommandSet[] getColourBySequenceCommands(
+  protected abstract String[] getColourBySequenceCommands(
           String[] files, AlignmentViewPanel avp);
 
   /**
@@ -801,7 +803,7 @@ public abstract class AAStructureBindingModel
           AlignmentViewPanel alignment);
 
   protected abstract void colourBySequence(
-          StructureMappingcommandSet[] colourBySequenceCommands);
+          String[] colourBySequenceCommands);
 
   public abstract void colourByChain();
 
@@ -813,16 +815,24 @@ public abstract class AAStructureBindingModel
    * colours on either change of alignment colours, or change to the visible
    * region of the alignment.
    */
-  public void colourBySequence(AlignmentViewPanel alignmentv)
+  public void updateStructureColours(AlignmentViewPanel alignmentv)
   {
     if (!isLoadingFinished())
     {
       return;
     }
-    // todo: property change event for visibleAlignment
-    // to avoid unnecessary redraws here
-    if (!colourBySequence && !isShowAlignmentOnly())
+
+    /*
+     * if structure is not coloured by sequence, but restricted to the alignment,
+     * then redraw it (but don't recolour it) in case hidden regions have changed
+     * (todo: specific messaging for change of hidden region only)
+     */
+    if (!colourBySequence)
     {
+      if (isShowAlignmentOnly())
+      {
+        showStructures(alignmentv.getAlignViewport(), false);
+      }
       return;
     }
     if (getSsm() == null)
@@ -831,7 +841,7 @@ public abstract class AAStructureBindingModel
     }
     String[] files = getStructureFiles();
 
-    StructureMappingcommandSet[] colourBySequenceCommands = getColourBySequenceCommands(
+    String[] colourBySequenceCommands = getColourBySequenceCommands(
             files, alignmentv);
     colourBySequence(colourBySequenceCommands);
   }
@@ -915,7 +925,7 @@ public abstract class AAStructureBindingModel
     }
     if (!isLoadingFromArchive())
     {
-      colourBySequence(ap);
+      updateStructureColours(ap);
     }
   }
 
@@ -946,4 +956,229 @@ public abstract class AAStructureBindingModel
     }
     return chainsToShow.contains(pdbId + ":" + chainId);
   }
+
+  @Override
+  public abstract String[] getStructureFiles();
+
+  /**
+   * Builds a model of residues mapped from sequences to show on structure, taking
+   * into account user choices of
+   * <ul>
+   * <li>which chains are shown</li>
+   * <li>whether all structure is shown, or only that mapped to the alignment</li>
+   * <li>whether hidden regions of the alignment are hidden (excluded) or grayed
+   * out (included)</li>
+   * </ul>
+   * 
+   * @param av
+   * @return
+   */
+  protected AtomSpecModel getShownResidues(AlignViewportI av)
+  {
+    AlignmentI alignment = av.getAlignment();
+    final int width = alignment.getWidth();
+  
+    String[] files = getStructureFiles();
+  
+    AtomSpecModel model = new AtomSpecModel();
+  
+    for (int pdbfnum = 0; pdbfnum < files.length; pdbfnum++)
+    {
+      StructureMapping[] mappings = getSsm().getMapping(files[pdbfnum]);
+  
+      /*
+       * Find the first mapped sequence (if any) for this PDB entry which is in
+       * the alignment
+       */
+      final int seqCountForPdbFile = getSequence()[pdbfnum].length;
+      for (int s = 0; s < seqCountForPdbFile; s++)
+      {
+        for (StructureMapping mapping : mappings)
+        {
+          final SequenceI theSequence = getSequence()[pdbfnum][s];
+          if (mapping.getSequence() == theSequence
+                  && alignment.findIndex(theSequence) > -1)
+          {
+            String chainCd = mapping.getChain();
+            if (!isShowChain(mapping.getPdbId(), chainCd))
+            {
+              continue;
+            }
+            Iterator<int[]> visible;
+            if (isShowAlignmentOnly() && isHideHiddenRegions())
+            {
+              visible = alignment.getHiddenColumns()
+                    .getVisContigsIterator(0, width, true);
+            }
+            else
+            {
+              visible = Collections.singletonList(new int[] { 0, width })
+                      .iterator();
+            }
+            while (visible.hasNext())
+            {
+              int[] visibleRegion = visible.next();
+              int seqStartPos = theSequence.findPosition(visibleRegion[0]);
+              int seqEndPos = theSequence.findPosition(visibleRegion[1]);
+              List<int[]> residueRanges = mapping
+                      .getPDBResNumRanges(seqStartPos, seqEndPos);
+              if (!residueRanges.isEmpty())
+              {
+                for (int[] range : residueRanges)
+                {
+                  model.addRange(pdbfnum, range[0], range[1], chainCd);
+                }
+              }
+            }
+          }
+        }
+      }
+    }
+  
+    return model;
+  }
+
+  /**
+   * Answers a default structure model specification which is simply the string
+   * form of the model number. Override if needed to specify submodels.
+   * 
+   * @param model
+   * @return
+   */
+  public String getModelSpec(int model)
+  {
+    return String.valueOf(model);
+  }
+
+  /**
+   * Build a data structure which records contiguous subsequences for each colour.
+   * From this we can easily generate the Chimera command for colour by sequence.
+   * 
+   * <pre>
+   * Color
+   *     Model number
+   *         Chain
+   *             list of start/end ranges
+   * </pre>
+   * 
+   * Ordering is by order of addition (for colours and positions), natural
+   * ordering (for models and chains)
+   * 
+   * @param viewPanel
+   * @return
+   */
+  public Map<Object, AtomSpecModel> buildColoursMap(
+          AlignmentViewPanel viewPanel)
+  {
+    FeatureRenderer fr = viewPanel.getFeatureRenderer();
+    FeatureColourFinder finder = new FeatureColourFinder(fr);
+    AlignViewportI viewport = viewPanel.getAlignViewport();
+    HiddenColumns cs = viewport.getAlignment().getHiddenColumns();
+    AlignmentI al = viewport.getAlignment();
+    SequenceRenderer sr = getSequenceRenderer(viewPanel);
+    String[] files = getStructureFiles();
+    
+    Map<Object, AtomSpecModel> colourMap = new LinkedHashMap<>();
+    Color lastColour = null;
+  
+    for (int pdbfnum = 0; pdbfnum < files.length; 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++)
+      {
+        for (int sp, m = 0; m < mapping.length; m++)
+        {
+          final SequenceI seq = sequence[pdbfnum][s];
+          if (mapping[m].getSequence() == seq
+                  && (sp = al.findIndex(seq)) > -1)
+          {
+            SequenceI asp = al.getSequenceAt(sp);
+            for (int r = 0; r < asp.getLength(); r++)
+            {
+              // no mapping to gaps in sequence
+              if (Comparison.isGap(asp.getCharAt(r)))
+              {
+                continue;
+              }
+              int pos = mapping[m].getPDBResNum(asp.findPosition(r));
+  
+              if (pos < 1 || pos == lastPos)
+              {
+                continue;
+              }
+  
+              Color colour = sr.getResidueColour(seq, r, finder);
+  
+              /*
+               * hidden regions are shown gray or, optionally, ignored
+               */
+              if (!cs.isVisible(r))
+              {
+                if (hideHiddenRegions)
+                {
+                  continue;
+                }
+                else
+                {
+                  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
+               * gap in the mapped residue sequence
+               */
+              final boolean newColour = !colour.equals(lastColour);
+              final boolean nonContig = lastPos + 1 != pos;
+              final boolean newChain = !chain.equals(lastChain);
+              if (newColour || nonContig || newChain)
+              {
+                if (startPos != -1)
+                {
+                  ChimeraCommands.addAtomSpecRange(colourMap, lastColour,
+                          pdbfnum, startPos,
+                          lastPos, lastChain);
+                }
+                startPos = pos;
+              }
+              lastColour = colour;
+              lastPos = pos;
+              lastChain = chain;
+            }
+            // final colour range
+            if (lastColour != null)
+            {
+              ChimeraCommands.addAtomSpecRange(colourMap, lastColour,
+                      pdbfnum,
+                      startPos, lastPos, lastChain);
+            }
+            // break;
+          }
+        }
+      }
+    }
+    return colourMap;
+  }
+
+  /**
+   * Returns a list of chains mapped in this viewer. Note this list is not
+   * currently scoped per structure.
+   * 
+   * @return
+   */
+  public List<String> getChainNames()
+  {
+    return chainNames;
+  }
 }