Adjust features when cutting and pasting
[jalview.git] / src / jalview / commands / EditCommand.java
index 72d82aa..f3b178f 100644 (file)
@@ -20,6 +20,8 @@ package jalview.commands;
 \r
 import jalview.datamodel.*;\r
 \r
+import java.util.Hashtable;\r
+\r
 /**\r
  *\r
  * <p>Title: EditCommmand</p>\r
@@ -54,23 +56,6 @@ public class EditCommand implements CommandI
     this.description = description;\r
   }\r
 \r
-  public EditCommand(String description,\r
-                     int command,\r
-                     SequenceI[] seqs,\r
-                     int position,\r
-                     int number,\r
-                     char gapChar)\r
-  {\r
-    this.description = description;\r
-\r
-    if (command==INSERT_GAP || command==DELETE_GAP)\r
-    {\r
-      edits = new Edit[] { new Edit(command, seqs, position, number, gapChar)};\r
-    }\r
-\r
-    performEdit(0);\r
-  }\r
-\r
   public EditCommand( String description,\r
                       int command,\r
                       SequenceI[] seqs,\r
@@ -88,7 +73,7 @@ public class EditCommand implements CommandI
   }\r
 \r
 \r
-  public String getDescription()\r
+  final public String getDescription()\r
   {\r
     return description;\r
   }\r
@@ -98,20 +83,25 @@ public class EditCommand implements CommandI
     return edits==null?0:edits.length;\r
   }\r
 \r
-  public AlignmentI getAlignment()\r
+  final public AlignmentI getAlignment()\r
   {\r
     return edits[0].al;\r
   }\r
 \r
 \r
-  public void appendEdit(int command,\r
+  final public void appendEdit(int command,\r
                          SequenceI[] seqs,\r
                          int position,\r
                          int number,\r
-                         char gapChar,\r
+                         AlignmentI al,\r
                          boolean performEdit)\r
   {\r
-    Edit edit = new Edit(command, seqs, position, number, gapChar);\r
+    Edit edit = new Edit(command, seqs, position, number, al.getGapCharacter());\r
+    if(al.getHeight()==seqs.length)\r
+    {\r
+      edit.al = al;\r
+      edit.fullAlignmentHeight = true;\r
+    }\r
 \r
     if (edits != null)\r
     {\r
@@ -127,7 +117,7 @@ public class EditCommand implements CommandI
       performEdit(edits.length - 1);\r
   }\r
 \r
-  void performEdit(int commandIndex)\r
+  final void performEdit(int commandIndex)\r
   {\r
     int eSize = edits.length;\r
     for (int e = commandIndex; e < eSize; e++)\r
@@ -151,12 +141,12 @@ public class EditCommand implements CommandI
     }\r
   }\r
 \r
-  public void doCommand()\r
+  final public void doCommand()\r
   {\r
     performEdit(0);\r
   }\r
 \r
-  public void undoCommand()\r
+  final public void undoCommand()\r
   {\r
     int e = 0, eSize = edits.length;\r
     for (e = eSize-1; e > -1; e--)\r
@@ -180,7 +170,7 @@ public class EditCommand implements CommandI
     }\r
   }\r
 \r
-  void insertGap(Edit command)\r
+  final void insertGap(Edit command)\r
   {\r
     for(int s=0; s<command.seqs.length; s++)\r
     {\r
@@ -188,14 +178,18 @@ public class EditCommand implements CommandI
                                      command.number,\r
                                      command.gapChar);\r
     }\r
+\r
+    adjustAnnotations(command, true);\r
   }\r
 \r
-  void deleteGap(Edit command)\r
+  final void deleteGap(Edit command)\r
   {\r
     for (int s = 0; s < command.seqs.length; s++)\r
     {\r
       command.seqs[s].deleteChars(command.position, command.position+command.number);\r
     }\r
+\r
+    adjustAnnotations(command, false);\r
   }\r
 \r
   void cut(Edit command)\r
@@ -207,8 +201,25 @@ public class EditCommand implements CommandI
       if(command.seqs[i].getLength()>command.position)\r
       {\r
         command.string[i] = command.seqs[i].getSequence(command.position,\r
-            command.position + command.number).toCharArray();\r
+            command.position + command.number);\r
 \r
+        if(command.seqs[i].getDatasetSequence()!=null\r
+        || command.seqs[i].getSequenceFeatures()!=null)\r
+        {\r
+          for (int s = command.position; s < command.position + command.number; s++)\r
+          {\r
+            if (jalview.schemes.ResidueProperties\r
+                .aaIndex[command.seqs[i].getCharAt(s)] != 23)\r
+            {\r
+              adjustFeatures(command, i,\r
+                             command.seqs[i].findPosition(command.position),\r
+                             command.seqs[i].findPosition(command.position +\r
+                                                          command.number),\r
+                             false);\r
+              break;\r
+            }\r
+          }\r
+        }\r
         command.seqs[i].deleteChars(command.position,\r
                                     command.position + command.number);\r
       }\r
@@ -218,13 +229,19 @@ public class EditCommand implements CommandI
         command.al.deleteSequence(command.seqs[i]);\r
       }\r
     }\r
+\r
+    adjustAnnotations(command, false);\r
   }\r
 \r
   void paste(Edit command)\r
   {\r
     StringBuffer tmp;\r
+    boolean newDSNeeded;\r
+    int start=0, end=0;\r
+\r
     for(int i=0; i<command.seqs.length; i++)\r
     {\r
+      newDSNeeded = false;\r
       if(command.seqs[i].getLength()<1)\r
       {\r
         // ie this sequence was deleted, we need to\r
@@ -235,7 +252,8 @@ public class EditCommand implements CommandI
         else\r
           command.al.addSequence(command.seqs[i]);\r
       }\r
-      tmp = new StringBuffer(command.seqs[i].getSequence());\r
+      tmp = new StringBuffer();\r
+      tmp.append(command.seqs[i].getSequence());\r
 \r
       if(command.string!=null && command.string[i]!=null)\r
       {\r
@@ -251,16 +269,240 @@ public class EditCommand implements CommandI
           }\r
         }\r
         tmp.insert(command.position, command.string[i]);\r
+\r
+        for (int s = 0; s < command.string[i].length; s++)\r
+        {\r
+          if (jalview.schemes.ResidueProperties.aaIndex[command.string[i][s]] != 23)\r
+          {\r
+            newDSNeeded = true;\r
+            start = command.seqs[i].findPosition(command.position);\r
+            end = command.seqs[i].findPosition(command.position+command.number);\r
+            break;\r
+          }\r
+        }\r
         command.string[i] = null;\r
       }\r
+\r
+\r
       command.seqs[i].setSequence(tmp.toString());\r
+\r
+      if(newDSNeeded)\r
+      {\r
+        if (command.seqs[i].getDatasetSequence() != null)\r
+        {\r
+          Sequence ds = new Sequence(command.seqs[i].getName(),\r
+                                     jalview.analysis.AlignSeq.extractGaps(\r
+                                         jalview.util.Comparison.GapChars,\r
+                                         command.seqs[i].getSequenceAsString()\r
+                                     ),\r
+                                     command.seqs[i].getStart(),\r
+                                     command.seqs[i].getEnd());\r
+          ds.setDescription(command.seqs[i].getDescription());\r
+          command.seqs[i].setDatasetSequence(ds);\r
+        }\r
+\r
+        adjustFeatures(command, i, start, end, true);\r
+      }\r
     }\r
 \r
+\r
+    adjustAnnotations(command, true);\r
+\r
     command.string = null;\r
   }\r
 \r
+\r
+  final void adjustAnnotations(Edit command, boolean insert)\r
+  {\r
+\r
+    AlignmentAnnotation [] annotations = null;\r
+\r
+    if (command.fullAlignmentHeight)\r
+    {\r
+      annotations = command.al.getAlignmentAnnotation();\r
+    }\r
+    else\r
+    {\r
+      int aSize = 0;\r
+      AlignmentAnnotation [] tmp;\r
+      for(int s=0; s<command.seqs.length; s++)\r
+      {\r
+        if(command.seqs[s].getAnnotation()==null)\r
+          continue;\r
+\r
+        if (aSize == 0)\r
+            annotations = command.seqs[s].getAnnotation();\r
+        else\r
+        {\r
+          tmp = new AlignmentAnnotation\r
+              [aSize + command.seqs[s].getAnnotation().length];\r
+\r
+          System.arraycopy(annotations,0,tmp,0,aSize);\r
+\r
+          System.arraycopy(command.seqs[s].getAnnotation(),\r
+              0,tmp,aSize,command.seqs[s].getAnnotation().length);\r
+\r
+          annotations = tmp;\r
+        }\r
+\r
+\r
+        aSize = annotations.length;\r
+      }\r
+    }\r
+\r
+    if(annotations==null)\r
+      return;\r
+\r
+\r
+      if(!insert)\r
+        command.deletedAnnotations = new Hashtable();\r
+\r
+      int aSize, tSize;\r
+      Annotation [] temp;\r
+      for (int a = 0; a < annotations.length; a++)\r
+      {\r
+        if(annotations[a].autoCalculated)\r
+        {\r
+          continue;\r
+        }\r
+\r
+        aSize = annotations[a].annotations.length;\r
+        if(insert)\r
+          tSize = aSize + command.number;\r
+        else\r
+          tSize = aSize - command.number;\r
+\r
+        temp = new Annotation[tSize];\r
+\r
+        if(insert)\r
+        {\r
+          if(command.position < annotations[a].annotations.length)\r
+          {\r
+            System.arraycopy(annotations[a].annotations,\r
+                             0, temp, 0, command.position);\r
+\r
+            if (command.deletedAnnotations != null\r
+                &&\r
+                command.deletedAnnotations.containsKey(annotations[a].annotationId))\r
+            {\r
+              Annotation[] restore = (Annotation[])\r
+                  command.deletedAnnotations.get(annotations[a].annotationId);\r
+\r
+              System.arraycopy(restore,\r
+                               0,\r
+                               temp,\r
+                               command.position,\r
+                               command.number);\r
+\r
+            }\r
+\r
+            System.arraycopy(annotations[a].annotations,\r
+                             command.position, temp,\r
+                             command.position + command.number,\r
+                             aSize - command.position);\r
+          }\r
+          else\r
+            temp = annotations[a].annotations;\r
+        }\r
+        else\r
+        {\r
+          if(command.position < annotations[a].annotations.length)\r
+          {\r
+            System.arraycopy(annotations[a].annotations,\r
+                             0, temp, 0, command.position);\r
+\r
+            Annotation[] deleted = new Annotation[command.number];\r
+            System.arraycopy(annotations[a].annotations,\r
+                             command.position, deleted, 0, command.number);\r
+\r
+            command.deletedAnnotations.put(annotations[a].annotationId,\r
+                                           deleted);\r
+\r
+            System.arraycopy(annotations[a].annotations,\r
+                             command.position + command.number,\r
+                             temp, command.position,\r
+                             aSize - command.position - command.number);\r
+          }\r
+          else\r
+            temp = annotations[a].annotations;\r
+        }\r
+\r
+        annotations[a].annotations = temp;\r
+     }\r
+  }\r
+\r
+  final void adjustFeatures(Edit command, int index, int i, int j, boolean insert)\r
+  {\r
+    SequenceI seq = command.seqs[index];\r
+    SequenceI sequence = seq.getDatasetSequence();\r
+    if(sequence==null)\r
+      sequence = seq;\r
+\r
+    if(insert)\r
+    {\r
+      if (command.editedFeatures != null\r
+          && command.editedFeatures.containsKey(seq))\r
+        sequence.setSequenceFeatures(\r
+            (SequenceFeature[]) command.editedFeatures.get(seq)\r
+            );\r
+\r
+      return;\r
+    }\r
+\r
+\r
+    SequenceFeature [] sf = sequence.getSequenceFeatures();\r
+\r
+\r
+    if(sf==null)\r
+    {\r
+      return;\r
+    }\r
+\r
+    SequenceFeature [] oldsf = new SequenceFeature[sf.length];\r
+\r
+    int cSize = j - i;\r
+\r
+    for (int s = 0; s < sf.length; s++)\r
+    {\r
+      SequenceFeature copy = new SequenceFeature(sf[s]);\r
+\r
+      oldsf[s] = copy;\r
+\r
+      if (sf[s].getEnd() < i)\r
+        continue;\r
+\r
+      if (sf[s].getBegin() > j)\r
+      {\r
+        sf[s].setBegin(copy.getBegin() - cSize);\r
+        sf[s].setEnd(copy.getEnd() - cSize);\r
+        continue;\r
+      }\r
+\r
+      if (sf[s].getBegin() >= i)\r
+        sf[s].setBegin(i);\r
+\r
+      if (sf[s].getEnd() < j)\r
+        sf[s].setEnd(j - 1);\r
+\r
+      sf[s].setEnd(sf[s].getEnd() - (cSize));\r
+\r
+      if (sf[s].getBegin() > sf[s].getEnd())\r
+        sequence.deleteFeature(sf[s]);\r
+    }\r
+\r
+    if (command.editedFeatures == null)\r
+      command.editedFeatures = new Hashtable();\r
+\r
+    command.editedFeatures.put(seq, oldsf);\r
+\r
+  }\r
+\r
+\r
   class Edit\r
   {\r
+    boolean fullAlignmentHeight = false;\r
+    Hashtable deletedAnnotations;\r
+    Hashtable editedFeatures;\r
     AlignmentI al;\r
     int command;\r
     char [][] string;\r
@@ -295,11 +537,15 @@ public class EditCommand implements CommandI
       this.position = position;\r
       this.number = number;\r
       this.al = al;\r
+\r
       alIndex = new int[seqs.length];\r
       for(int i=0; i<seqs.length; i++)\r
+      {\r
         alIndex[i] = al.findIndex(seqs[i]);\r
-    }\r
+      }\r
 \r
+      fullAlignmentHeight = (al.getHeight()==seqs.length);\r
+    }\r
   }\r
 \r
 }\r