28745f18c50a48cb50de027cd7eeb1f70bc670cb
[jalview.git] / src / jalview / datamodel / ColumnSelection.java
1 /*
2  * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3  * Copyright (C) $$Year-Rel$$ The Jalview Authors
4  * 
5  * This file is part of Jalview.
6  * 
7  * Jalview is free software: you can redistribute it and/or
8  * modify it under the terms of the GNU General Public License 
9  * as published by the Free Software Foundation, either version 3
10  * of the License, or (at your option) any later version.
11  *  
12  * Jalview is distributed in the hope that it will be useful, but 
13  * WITHOUT ANY WARRANTY; without even the implied warranty 
14  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
15  * PURPOSE.  See the GNU General Public License for more details.
16  * 
17  * You should have received a copy of the GNU General Public License
18  * along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
19  * The Jalview Authors are detailed in the 'AUTHORS' file.
20  */
21 package jalview.datamodel;
22
23 import jalview.util.Comparison;
24 import jalview.util.ShiftList;
25 import jalview.viewmodel.annotationfilter.AnnotationFilterParameter;
26 import jalview.viewmodel.annotationfilter.AnnotationFilterParameter.SearchableAnnotationField;
27
28 import java.util.ArrayList;
29 import java.util.BitSet;
30 import java.util.Collections;
31 import java.util.List;
32 import java.util.Vector;
33
34 /**
35  * Data class holding the selected columns and hidden column ranges for a view.
36  * Ranges are base 1.
37  */
38 public class ColumnSelection
39 {
40   /**
41    * A class to hold an efficient representation of selected columns
42    */
43   private class IntList
44   {
45     /*
46      * list of selected columns (ordered by selection order, not column order)
47      */
48     private List<Integer> order;
49
50     /*
51      * an unmodifiable view of the selected columns list
52      */
53     private List<Integer> _uorder;
54
55     /**
56      * bitfield for column selection - allows quick lookup
57      */
58     private BitSet selected;
59
60     /**
61      * Constructor
62      */
63     IntList()
64     {
65       order = new ArrayList<Integer>();
66       _uorder = Collections.unmodifiableList(order);
67       selected = new BitSet();
68     }
69
70     /**
71      * Copy constructor
72      * 
73      * @param other
74      */
75     IntList(IntList other)
76     {
77       this();
78       if (other != null)
79       {
80         int j = other.size();
81         for (int i = 0; i < j; i++)
82         {
83           add(other.elementAt(i));
84         }
85       }
86     }
87
88     /**
89      * adds a new column i to the selection - only if i is not already selected
90      * 
91      * @param i
92      */
93     void add(int i)
94     {
95       if (!selected.get(i))
96       {
97         order.add(Integer.valueOf(i));
98         selected.set(i);
99       }
100     }
101
102     void clear()
103     {
104       order.clear();
105       selected.clear();
106     }
107
108     void remove(int col)
109     {
110
111       Integer colInt = new Integer(col);
112
113       if (selected.get(col))
114       {
115         // if this ever changes to List.remove(), ensure Integer not int
116         // argument
117         // as List.remove(int i) removes the i'th item which is wrong
118         order.remove(colInt);
119         selected.clear(col);
120       }
121     }
122
123     boolean contains(Integer colInt)
124     {
125       return selected.get(colInt);
126     }
127
128     boolean isEmpty()
129     {
130       return order.isEmpty();
131     }
132
133     /**
134      * Returns a read-only view of the selected columns list
135      * 
136      * @return
137      */
138     List<Integer> getList()
139     {
140       return _uorder;
141     }
142
143     int size()
144     {
145       return order.size();
146     }
147
148     /**
149      * gets the column that was selected first, second or i'th
150      * 
151      * @param i
152      * @return
153      */
154     int elementAt(int i)
155     {
156       return order.get(i);
157     }
158
159     protected boolean pruneColumnList(final List<int[]> shifts)
160     {
161       int s = 0, t = shifts.size();
162       int[] sr = shifts.get(s++);
163       boolean pruned = false;
164       int i = 0, j = order.size();
165       while (i < j && s <= t)
166       {
167         int c = order.get(i++).intValue();
168         if (sr[0] <= c)
169         {
170           if (sr[1] + sr[0] >= c)
171           { // sr[1] -ve means inseriton.
172             order.remove(--i);
173             selected.clear(c);
174             j--;
175           }
176           else
177           {
178             if (s < t)
179             {
180               sr = shifts.get(s);
181             }
182             s++;
183           }
184         }
185       }
186       return pruned;
187     }
188
189     /**
190      * shift every selected column at or above start by change
191      * 
192      * @param start
193      *          - leftmost column to be shifted
194      * @param change
195      *          - delta for shift
196      */
197     void compensateForEdits(int start, int change)
198     {
199       BitSet mask = new BitSet();
200       for (int i = 0; i < order.size(); i++)
201       {
202         int temp = order.get(i);
203
204         if (temp >= start)
205         {
206           // clear shifted bits and update List of selected columns
207           selected.clear(temp);
208           mask.set(temp - change);
209           order.set(i, new Integer(temp - change));
210         }
211       }
212       // lastly update the bitfield all at once
213       selected.or(mask);
214     }
215
216     boolean isSelected(int column)
217     {
218       return selected.get(column);
219     }
220
221     int getMaxColumn()
222     {
223       return selected.length() - 1;
224     }
225
226     int getMinColumn()
227     {
228       return selected.get(0) ? 0 : selected.nextSetBit(0);
229     }
230
231     /**
232      * @return a series of selection intervals along the range
233      */
234     List<int[]> getRanges()
235     {
236       List<int[]> rlist = new ArrayList<int[]>();
237       if (selected.isEmpty())
238       {
239         return rlist;
240       }
241       int next = selected.nextSetBit(0), clear = -1;
242       while (next != -1)
243       {
244         clear = selected.nextClearBit(next);
245         rlist.add(new int[] { next, clear - 1 });
246         next = selected.nextSetBit(clear);
247       }
248       return rlist;
249     }
250
251     @Override
252     public int hashCode()
253     {
254       // TODO Auto-generated method stub
255       return selected.hashCode();
256     }
257
258     @Override
259     public boolean equals(Object obj)
260     {
261       if (obj instanceof IntList)
262       {
263         return ((IntList) obj).selected.equals(selected);
264       }
265       return false;
266     }
267   }
268
269   IntList selection = new IntList();
270
271   /*
272    * list of hidden column [start, end] ranges; the list is maintained in
273    * ascending start column order
274    */
275   Vector<int[]> hiddenColumns;
276
277   /**
278    * Add a column to the selection
279    * 
280    * @param col
281    *          index of column
282    */
283   public void addElement(int col)
284   {
285     selection.add(col);
286   }
287
288   /**
289    * clears column selection
290    */
291   public void clear()
292   {
293     selection.clear();
294   }
295
296   /**
297    * Removes value 'col' from the selection (not the col'th item)
298    * 
299    * @param col
300    *          index of column to be removed
301    */
302   public void removeElement(int col)
303   {
304     selection.remove(col);
305   }
306
307   /**
308    * removes a range of columns from the selection
309    * 
310    * @param start
311    *          int - first column in range to be removed
312    * @param end
313    *          int - last col
314    */
315   public void removeElements(int start, int end)
316   {
317     Integer colInt;
318     for (int i = start; i < end; i++)
319     {
320       colInt = new Integer(i);
321       if (selection.contains(colInt))
322       {
323         selection.remove(colInt);
324       }
325     }
326   }
327
328   /**
329    * Returns a read-only view of the (possibly empty) list of selected columns
330    * <p>
331    * The list contains no duplicates but is not necessarily ordered. It also may
332    * include columns hidden from the current view. To modify (for example sort)
333    * the list, you should first make a copy.
334    * <p>
335    * The list is not thread-safe: iterating over it could result in
336    * ConcurrentModificationException if it is modified by another thread.
337    */
338   public List<Integer> getSelected()
339   {
340     return selection.getList();
341   }
342
343   /**
344    * @return list of int arrays containing start and end column position for
345    *         runs of selected columns ordered from right to left.
346    */
347   public List<int[]> getSelectedRanges()
348   {
349     return selection.getRanges();
350   }
351
352   /**
353    * 
354    * @param col
355    *          index to search for in column selection
356    * 
357    * @return true if col is selected
358    */
359   public boolean contains(int col)
360   {
361     return (col > -1) ? selection.isSelected(col) : false;
362   }
363
364   /**
365    * Answers true if no columns are selected, else false
366    */
367   public boolean isEmpty()
368   {
369     return selection == null || selection.isEmpty();
370   }
371
372   /**
373    * rightmost selected column
374    * 
375    * @return rightmost column in alignment that is selected
376    */
377   public int getMax()
378   {
379     if (selection.isEmpty())
380     {
381       return -1;
382     }
383     return selection.getMaxColumn();
384   }
385
386   /**
387    * Leftmost column in selection
388    * 
389    * @return column index of leftmost column in selection
390    */
391   public int getMin()
392   {
393     if (selection.isEmpty())
394     {
395       return 1000000000;
396     }
397     return selection.getMinColumn();
398   }
399
400   /**
401    * propagate shift in alignment columns to column selection
402    * 
403    * @param start
404    *          beginning of edit
405    * @param left
406    *          shift in edit (+ve for removal, or -ve for inserts)
407    */
408   public List<int[]> compensateForEdit(int start, int change)
409   {
410     List<int[]> deletedHiddenColumns = null;
411     selection.compensateForEdits(start, change);
412
413     if (hiddenColumns != null)
414     {
415       deletedHiddenColumns = new ArrayList<int[]>();
416       int hSize = hiddenColumns.size();
417       for (int i = 0; i < hSize; i++)
418       {
419         int[] region = hiddenColumns.elementAt(i);
420         if (region[0] > start && start + change > region[1])
421         {
422           deletedHiddenColumns.add(region);
423
424           hiddenColumns.removeElementAt(i);
425           i--;
426           hSize--;
427           continue;
428         }
429
430         if (region[0] > start)
431         {
432           region[0] -= change;
433           region[1] -= change;
434         }
435
436         if (region[0] < 0)
437         {
438           region[0] = 0;
439         }
440
441       }
442
443       this.revealHiddenColumns(0);
444     }
445
446     return deletedHiddenColumns;
447   }
448
449   /**
450    * propagate shift in alignment columns to column selection special version of
451    * compensateForEdit - allowing for edits within hidden regions
452    * 
453    * @param start
454    *          beginning of edit
455    * @param left
456    *          shift in edit (+ve for removal, or -ve for inserts)
457    */
458   private void compensateForDelEdits(int start, int change)
459   {
460
461     selection.compensateForEdits(start, change);
462
463     if (hiddenColumns != null)
464     {
465       for (int i = 0; i < hiddenColumns.size(); i++)
466       {
467         int[] region = hiddenColumns.elementAt(i);
468         if (region[0] >= start)
469         {
470           region[0] -= change;
471         }
472         if (region[1] >= start)
473         {
474           region[1] -= change;
475         }
476         if (region[1] < region[0])
477         {
478           hiddenColumns.removeElementAt(i--);
479         }
480
481         if (region[0] < 0)
482         {
483           region[0] = 0;
484         }
485         if (region[1] < 0)
486         {
487           region[1] = 0;
488         }
489       }
490     }
491   }
492
493   /**
494    * Adjust hidden column boundaries based on a series of column additions or
495    * deletions in visible regions.
496    * 
497    * @param shiftrecord
498    * @return
499    */
500   public ShiftList compensateForEdits(ShiftList shiftrecord)
501   {
502     if (shiftrecord != null)
503     {
504       final List<int[]> shifts = shiftrecord.getShifts();
505       if (shifts != null && shifts.size() > 0)
506       {
507         int shifted = 0;
508         for (int i = 0, j = shifts.size(); i < j; i++)
509         {
510           int[] sh = shifts.get(i);
511           // compensateForEdit(shifted+sh[0], sh[1]);
512           compensateForDelEdits(shifted + sh[0], sh[1]);
513           shifted -= sh[1];
514         }
515       }
516       return shiftrecord.getInverse();
517     }
518     return null;
519   }
520
521   /**
522    * removes intersection of position,length ranges in deletions from the
523    * start,end regions marked in intervals.
524    * 
525    * @param shifts
526    * @param intervals
527    * @return
528    */
529   private boolean pruneIntervalVector(final List<int[]> shifts,
530           Vector<int[]> intervals)
531   {
532     boolean pruned = false;
533     int i = 0, j = intervals.size() - 1, s = 0, t = shifts.size() - 1;
534     int hr[] = intervals.elementAt(i);
535     int sr[] = shifts.get(s);
536     while (i <= j && s <= t)
537     {
538       boolean trailinghn = hr[1] >= sr[0];
539       if (!trailinghn)
540       {
541         if (i < j)
542         {
543           hr = intervals.elementAt(++i);
544         }
545         else
546         {
547           i++;
548         }
549         continue;
550       }
551       int endshift = sr[0] + sr[1]; // deletion ranges - -ve means an insert
552       if (endshift < hr[0] || endshift < sr[0])
553       { // leadinghc disjoint or not a deletion
554         if (s < t)
555         {
556           sr = shifts.get(++s);
557         }
558         else
559         {
560           s++;
561         }
562         continue;
563       }
564       boolean leadinghn = hr[0] >= sr[0];
565       boolean leadinghc = hr[0] < endshift;
566       boolean trailinghc = hr[1] < endshift;
567       if (leadinghn)
568       {
569         if (trailinghc)
570         { // deleted hidden region.
571           intervals.removeElementAt(i);
572           pruned = true;
573           j--;
574           if (i <= j)
575           {
576             hr = intervals.elementAt(i);
577           }
578           continue;
579         }
580         if (leadinghc)
581         {
582           hr[0] = endshift; // clip c terminal region
583           leadinghn = !leadinghn;
584           pruned = true;
585         }
586       }
587       if (!leadinghn)
588       {
589         if (trailinghc)
590         {
591           if (trailinghn)
592           {
593             hr[1] = sr[0] - 1;
594             pruned = true;
595           }
596         }
597         else
598         {
599           // sr contained in hr
600           if (s < t)
601           {
602             sr = shifts.get(++s);
603           }
604           else
605           {
606             s++;
607           }
608           continue;
609         }
610       }
611     }
612     return pruned; // true if any interval was removed or modified by
613     // operations.
614   }
615
616   /**
617    * remove any hiddenColumns or selected columns and shift remaining based on a
618    * series of position, range deletions.
619    * 
620    * @param deletions
621    */
622   public void pruneDeletions(ShiftList deletions)
623   {
624     if (deletions != null)
625     {
626       final List<int[]> shifts = deletions.getShifts();
627       if (shifts != null && shifts.size() > 0)
628       {
629         // delete any intervals intersecting.
630         if (hiddenColumns != null)
631         {
632           pruneIntervalVector(shifts, hiddenColumns);
633           if (hiddenColumns != null && hiddenColumns.size() == 0)
634           {
635             hiddenColumns = null;
636           }
637         }
638         if (selection != null && selection.size() > 0)
639         {
640           selection.pruneColumnList(shifts);
641           if (selection != null && selection.size() == 0)
642           {
643             selection = null;
644           }
645         }
646         // and shift the rest.
647         this.compensateForEdits(deletions);
648       }
649     }
650   }
651
652   /**
653    * This Method is used to return all the HiddenColumn regions
654    * 
655    * @return empty list or List of hidden column intervals
656    */
657   public List<int[]> getHiddenColumns()
658   {
659     return hiddenColumns == null ? Collections.<int[]> emptyList()
660             : hiddenColumns;
661   }
662
663   /**
664    * Return absolute column index for a visible column index
665    * 
666    * @param column
667    *          int column index in alignment view (count from zero)
668    * @return alignment column index for column
669    */
670   public int adjustForHiddenColumns(int column)
671   {
672     int result = column;
673     if (hiddenColumns != null)
674     {
675       for (int i = 0; i < hiddenColumns.size(); i++)
676       {
677         int[] region = hiddenColumns.elementAt(i);
678         if (result >= region[0])
679         {
680           result += region[1] - region[0] + 1;
681         }
682       }
683     }
684     return result;
685   }
686
687   /**
688    * Use this method to find out where a column will appear in the visible
689    * alignment when hidden columns exist. If the column is not visible, then the
690    * left-most visible column will always be returned.
691    * 
692    * @param hiddenColumn
693    *          the column index in the full alignment including hidden columns
694    * @return the position of the column in the visible alignment
695    */
696   public int findColumnPosition(int hiddenColumn)
697   {
698     int result = hiddenColumn;
699     if (hiddenColumns != null)
700     {
701       int index = 0;
702       int[] region;
703       do
704       {
705         region = hiddenColumns.elementAt(index++);
706         if (hiddenColumn > region[1])
707         {
708           result -= region[1] + 1 - region[0];
709         }
710       } while ((hiddenColumn > region[1]) && (index < hiddenColumns.size()));
711
712       if (hiddenColumn >= region[0] && hiddenColumn <= region[1])
713        {
714          // Here the hidden column is within a region, so
715          // we want to return the position of region[0]-1, adjusted for any
716          // earlier hidden columns.
717          // Calculate the difference between the actual hidden col position
718          // and region[0]-1, and then subtract from result to convert result from
719          // the adjusted hiddenColumn value to the adjusted region[0]-1 value
720
721         // However, if the region begins at 0 we cannot return region[0]-1
722         // just return 0
723         if (region[0] == 0)
724         {
725           return 0;
726         }
727         else
728         {
729           return result - (hiddenColumn - region[0] + 1);
730         }
731       }
732     }
733     return result; // return the shifted position after removing hidden columns.
734   }
735
736   /**
737    * Use this method to determine where the next hiddenRegion starts
738    * 
739    * @param hiddenRegion
740    *          index of hidden region (counts from 0)
741    * @return column number in visible view
742    */
743   public int findHiddenRegionPosition(int hiddenRegion)
744   {
745     int result = 0;
746     if (hiddenColumns != null)
747     {
748       int index = 0;
749       int gaps = 0;
750       do
751       {
752         int[] region = hiddenColumns.elementAt(index);
753         if (hiddenRegion == 0)
754         {
755           return region[0];
756         }
757
758         gaps += region[1] + 1 - region[0];
759         result = region[1] + 1;
760         index++;
761       } while (index <= hiddenRegion);
762
763       result -= gaps;
764     }
765
766     return result;
767   }
768
769   /**
770    * THis method returns the rightmost limit of a region of an alignment with
771    * hidden columns. In otherwords, the next hidden column.
772    * 
773    * @param index
774    *          int
775    */
776   public int getHiddenBoundaryRight(int alPos)
777   {
778     if (hiddenColumns != null)
779     {
780       int index = 0;
781       do
782       {
783         int[] region = hiddenColumns.elementAt(index);
784         if (alPos < region[0])
785         {
786           return region[0];
787         }
788
789         index++;
790       } while (index < hiddenColumns.size());
791     }
792
793     return alPos;
794
795   }
796
797   /**
798    * This method returns the leftmost limit of a region of an alignment with
799    * hidden columns. In otherwords, the previous hidden column.
800    * 
801    * @param index
802    *          int
803    */
804   public int getHiddenBoundaryLeft(int alPos)
805   {
806     if (hiddenColumns != null)
807     {
808       int index = hiddenColumns.size() - 1;
809       do
810       {
811         int[] region = hiddenColumns.elementAt(index);
812         if (alPos > region[1])
813         {
814           return region[1];
815         }
816
817         index--;
818       } while (index > -1);
819     }
820
821     return alPos;
822
823   }
824
825   public void hideSelectedColumns()
826   {
827     synchronized (selection)
828     {
829       for (int[] selregions : selection.getRanges())
830       {
831         hideColumns(selregions[0], selregions[1]);
832       }
833       selection.clear();
834     }
835
836   }
837
838   /**
839    * Adds the specified column range to the hidden columns
840    * 
841    * @param start
842    * @param end
843    */
844   public void hideColumns(int start, int end)
845   {
846     if (hiddenColumns == null)
847     {
848       hiddenColumns = new Vector<int[]>();
849     }
850
851     /*
852      * traverse existing hidden ranges and insert / amend / append as
853      * appropriate
854      */
855     for (int i = 0; i < hiddenColumns.size(); i++)
856     {
857       int[] region = hiddenColumns.elementAt(i);
858
859       if (end < region[0] - 1)
860       {
861         /*
862          * insert discontiguous preceding range
863          */
864         hiddenColumns.insertElementAt(new int[] { start, end }, i);
865         return;
866       }
867
868       if (end <= region[1])
869       {
870         /*
871          * new range overlaps existing, or is contiguous preceding it - adjust
872          * start column
873          */
874         region[0] = Math.min(region[0], start);
875         return;
876       }
877
878       if (start <= region[1] + 1)
879       {
880         /*
881          * new range overlaps existing, or is contiguous following it - adjust
882          * start and end columns
883          */
884         region[0] = Math.min(region[0], start);
885         region[1] = Math.max(region[1], end);
886
887         /*
888          * also update or remove any subsequent ranges 
889          * that are overlapped
890          */
891         while (i < hiddenColumns.size() - 1)
892         {
893           int[] nextRegion = hiddenColumns.get(i + 1);
894           if (nextRegion[0] > end + 1)
895           {
896             /*
897              * gap to next hidden range - no more to update
898              */
899             break;
900           }
901           region[1] = Math.max(nextRegion[1], end);
902           hiddenColumns.remove(i + 1);
903         }
904         return;
905       }
906     }
907
908     /*
909      * remaining case is that the new range follows everything else
910      */
911     hiddenColumns.addElement(new int[] { start, end });
912   }
913
914   /**
915    * Hides the specified column and any adjacent selected columns
916    * 
917    * @param res
918    *          int
919    */
920   public void hideColumns(int col)
921   {
922     /*
923      * deselect column (whether selected or not!)
924      */
925     removeElement(col);
926
927     /*
928      * find adjacent selected columns
929      */
930     int min = col - 1, max = col + 1;
931     while (contains(min))
932     {
933       removeElement(min);
934       min--;
935     }
936
937     while (contains(max))
938     {
939       removeElement(max);
940       max++;
941     }
942
943     /*
944      * min, max are now the closest unselected columns
945      */
946     min++;
947     max--;
948     if (min > max)
949     {
950       min = max;
951     }
952
953     hideColumns(min, max);
954   }
955
956   /**
957    * Unhides, and adds to the selection list, all hidden columns
958    */
959   public void revealAllHiddenColumns()
960   {
961     if (hiddenColumns != null)
962     {
963       for (int i = 0; i < hiddenColumns.size(); i++)
964       {
965         int[] region = hiddenColumns.elementAt(i);
966         for (int j = region[0]; j < region[1] + 1; j++)
967         {
968           addElement(j);
969         }
970       }
971     }
972
973     hiddenColumns = null;
974   }
975
976   /**
977    * Reveals, and marks as selected, the hidden column range with the given
978    * start column
979    * 
980    * @param start
981    */
982   public void revealHiddenColumns(int start)
983   {
984     for (int i = 0; i < hiddenColumns.size(); i++)
985     {
986       int[] region = hiddenColumns.elementAt(i);
987       if (start == region[0])
988       {
989         for (int j = region[0]; j < region[1] + 1; j++)
990         {
991           addElement(j);
992         }
993
994         hiddenColumns.removeElement(region);
995         break;
996       }
997     }
998     if (hiddenColumns.size() == 0)
999     {
1000       hiddenColumns = null;
1001     }
1002   }
1003
1004   public boolean isVisible(int column)
1005   {
1006     if (hiddenColumns != null)
1007     {
1008       for (int[] region : hiddenColumns)
1009       {
1010         if (column >= region[0] && column <= region[1])
1011         {
1012           return false;
1013         }
1014       }
1015     }
1016
1017     return true;
1018   }
1019
1020   /**
1021    * Copy constructor
1022    * 
1023    * @param copy
1024    */
1025   public ColumnSelection(ColumnSelection copy)
1026   {
1027     if (copy != null)
1028     {
1029       selection = new IntList(copy.selection);
1030       if (copy.hiddenColumns != null)
1031       {
1032         hiddenColumns = new Vector<int[]>(copy.hiddenColumns.size());
1033         for (int i = 0, j = copy.hiddenColumns.size(); i < j; i++)
1034         {
1035           int[] rh, cp;
1036           rh = copy.hiddenColumns.elementAt(i);
1037           if (rh != null)
1038           {
1039             cp = new int[rh.length];
1040             System.arraycopy(rh, 0, cp, 0, rh.length);
1041             hiddenColumns.addElement(cp);
1042           }
1043         }
1044       }
1045     }
1046   }
1047
1048   /**
1049    * ColumnSelection
1050    */
1051   public ColumnSelection()
1052   {
1053   }
1054
1055   public String[] getVisibleSequenceStrings(int start, int end,
1056           SequenceI[] seqs)
1057   {
1058     int i, iSize = seqs.length;
1059     String selections[] = new String[iSize];
1060     if (hiddenColumns != null && hiddenColumns.size() > 0)
1061     {
1062       for (i = 0; i < iSize; i++)
1063       {
1064         StringBuffer visibleSeq = new StringBuffer();
1065         List<int[]> regions = getHiddenColumns();
1066
1067         int blockStart = start, blockEnd = end;
1068         int[] region;
1069         int hideStart, hideEnd;
1070
1071         for (int j = 0; j < regions.size(); j++)
1072         {
1073           region = regions.get(j);
1074           hideStart = region[0];
1075           hideEnd = region[1];
1076
1077           if (hideStart < start)
1078           {
1079             continue;
1080           }
1081
1082           blockStart = Math.min(blockStart, hideEnd + 1);
1083           blockEnd = Math.min(blockEnd, hideStart);
1084
1085           if (blockStart > blockEnd)
1086           {
1087             break;
1088           }
1089
1090           visibleSeq.append(seqs[i].getSequence(blockStart, blockEnd));
1091
1092           blockStart = hideEnd + 1;
1093           blockEnd = end;
1094         }
1095
1096         if (end > blockStart)
1097         {
1098           visibleSeq.append(seqs[i].getSequence(blockStart, end));
1099         }
1100
1101         selections[i] = visibleSeq.toString();
1102       }
1103     }
1104     else
1105     {
1106       for (i = 0; i < iSize; i++)
1107       {
1108         selections[i] = seqs[i].getSequenceAsString(start, end);
1109       }
1110     }
1111
1112     return selections;
1113   }
1114
1115   /**
1116    * return all visible segments between the given start and end boundaries
1117    * 
1118    * @param start
1119    *          (first column inclusive from 0)
1120    * @param end
1121    *          (last column - not inclusive)
1122    * @return int[] {i_start, i_end, ..} where intervals lie in
1123    *         start<=i_start<=i_end<end
1124    */
1125   public int[] getVisibleContigs(int start, int end)
1126   {
1127     if (hiddenColumns != null && hiddenColumns.size() > 0)
1128     {
1129       List<int[]> visiblecontigs = new ArrayList<int[]>();
1130       List<int[]> regions = getHiddenColumns();
1131
1132       int vstart = start;
1133       int[] region;
1134       int hideStart, hideEnd;
1135
1136       for (int j = 0; vstart < end && j < regions.size(); j++)
1137       {
1138         region = regions.get(j);
1139         hideStart = region[0];
1140         hideEnd = region[1];
1141
1142         if (hideEnd < vstart)
1143         {
1144           continue;
1145         }
1146         if (hideStart > vstart)
1147         {
1148           visiblecontigs.add(new int[] { vstart, hideStart - 1 });
1149         }
1150         vstart = hideEnd + 1;
1151       }
1152
1153       if (vstart < end)
1154       {
1155         visiblecontigs.add(new int[] { vstart, end - 1 });
1156       }
1157       int[] vcontigs = new int[visiblecontigs.size() * 2];
1158       for (int i = 0, j = visiblecontigs.size(); i < j; i++)
1159       {
1160         int[] vc = visiblecontigs.get(i);
1161         visiblecontigs.set(i, null);
1162         vcontigs[i * 2] = vc[0];
1163         vcontigs[i * 2 + 1] = vc[1];
1164       }
1165       visiblecontigs.clear();
1166       return vcontigs;
1167     }
1168     else
1169     {
1170       return new int[] { start, end - 1 };
1171     }
1172   }
1173
1174   /**
1175    * Locate the first and last position visible for this sequence. if seq isn't
1176    * visible then return the position of the left and right of the hidden
1177    * boundary region, and the corresponding alignment column indices for the
1178    * extent of the sequence
1179    * 
1180    * @param seq
1181    * @return int[] { visible start, visible end, first seqpos, last seqpos,
1182    *         alignment index for seq start, alignment index for seq end }
1183    */
1184   public int[] locateVisibleBoundsOfSequence(SequenceI seq)
1185   {
1186     int fpos = seq.getStart(), lpos = seq.getEnd();
1187     int start = 0;
1188
1189     if (hiddenColumns == null || hiddenColumns.size() == 0)
1190     {
1191       int ifpos = seq.findIndex(fpos) - 1, ilpos = seq.findIndex(lpos) - 1;
1192       return new int[] { ifpos, ilpos, fpos, lpos, ifpos, ilpos };
1193     }
1194
1195     // Simply walk along the sequence whilst watching for hidden column
1196     // boundaries
1197     List<int[]> regions = getHiddenColumns();
1198     int spos = fpos, lastvispos = -1, rcount = 0, hideStart = seq
1199             .getLength(), hideEnd = -1;
1200     int visPrev = 0, visNext = 0, firstP = -1, lastP = -1;
1201     boolean foundStart = false;
1202     for (int p = 0, pLen = seq.getLength(); spos <= seq.getEnd()
1203             && p < pLen; p++)
1204     {
1205       if (!Comparison.isGap(seq.getCharAt(p)))
1206       {
1207         // keep track of first/last column
1208         // containing sequence data regardless of visibility
1209         if (firstP == -1)
1210         {
1211           firstP = p;
1212         }
1213         lastP = p;
1214         // update hidden region start/end
1215         while (hideEnd < p && rcount < regions.size())
1216         {
1217           int[] region = regions.get(rcount++);
1218           visPrev = visNext;
1219           visNext += region[0] - visPrev;
1220           hideStart = region[0];
1221           hideEnd = region[1];
1222         }
1223         if (hideEnd < p)
1224         {
1225           hideStart = seq.getLength();
1226         }
1227         // update visible boundary for sequence
1228         if (p < hideStart)
1229         {
1230           if (!foundStart)
1231           {
1232             fpos = spos;
1233             start = p;
1234             foundStart = true;
1235           }
1236           lastvispos = p;
1237           lpos = spos;
1238         }
1239         // look for next sequence position
1240         spos++;
1241       }
1242     }
1243     if (foundStart)
1244     {
1245       return new int[] { findColumnPosition(start),
1246           findColumnPosition(lastvispos), fpos, lpos, firstP, lastP };
1247     }
1248     // otherwise, sequence was completely hidden
1249     return new int[] { visPrev, visNext, 0, 0, firstP, lastP };
1250   }
1251
1252   /**
1253    * delete any columns in alignmentAnnotation that are hidden (including
1254    * sequence associated annotation).
1255    * 
1256    * @param alignmentAnnotation
1257    */
1258   public void makeVisibleAnnotation(AlignmentAnnotation alignmentAnnotation)
1259   {
1260     makeVisibleAnnotation(-1, -1, alignmentAnnotation);
1261   }
1262
1263   /**
1264    * delete any columns in alignmentAnnotation that are hidden (including
1265    * sequence associated annotation).
1266    * 
1267    * @param start
1268    *          remove any annotation to the right of this column
1269    * @param end
1270    *          remove any annotation to the left of this column
1271    * @param alignmentAnnotation
1272    *          the annotation to operate on
1273    */
1274   public void makeVisibleAnnotation(int start, int end,
1275           AlignmentAnnotation alignmentAnnotation)
1276   {
1277     if (alignmentAnnotation.annotations == null)
1278     {
1279       return;
1280     }
1281     if (start == end && end == -1)
1282     {
1283       start = 0;
1284       end = alignmentAnnotation.annotations.length;
1285     }
1286     if (hiddenColumns != null && hiddenColumns.size() > 0)
1287     {
1288       // then mangle the alignmentAnnotation annotation array
1289       Vector<Annotation[]> annels = new Vector<Annotation[]>();
1290       Annotation[] els = null;
1291       List<int[]> regions = getHiddenColumns();
1292       int blockStart = start, blockEnd = end;
1293       int[] region;
1294       int hideStart, hideEnd, w = 0;
1295
1296       for (int j = 0; j < regions.size(); j++)
1297       {
1298         region = regions.get(j);
1299         hideStart = region[0];
1300         hideEnd = region[1];
1301
1302         if (hideStart < start)
1303         {
1304           continue;
1305         }
1306
1307         blockStart = Math.min(blockStart, hideEnd + 1);
1308         blockEnd = Math.min(blockEnd, hideStart);
1309
1310         if (blockStart > blockEnd)
1311         {
1312           break;
1313         }
1314
1315         annels.addElement(els = new Annotation[blockEnd - blockStart]);
1316         System.arraycopy(alignmentAnnotation.annotations, blockStart, els,
1317                 0, els.length);
1318         w += els.length;
1319         blockStart = hideEnd + 1;
1320         blockEnd = end;
1321       }
1322
1323       if (end > blockStart)
1324       {
1325         annels.addElement(els = new Annotation[end - blockStart + 1]);
1326         if ((els.length + blockStart) <= alignmentAnnotation.annotations.length)
1327         {
1328           // copy just the visible segment of the annotation row
1329           System.arraycopy(alignmentAnnotation.annotations, blockStart,
1330                   els, 0, els.length);
1331         }
1332         else
1333         {
1334           // copy to the end of the annotation row
1335           System.arraycopy(alignmentAnnotation.annotations, blockStart,
1336                   els, 0,
1337                   (alignmentAnnotation.annotations.length - blockStart));
1338         }
1339         w += els.length;
1340       }
1341       if (w == 0)
1342       {
1343         return;
1344       }
1345
1346       alignmentAnnotation.annotations = new Annotation[w];
1347       w = 0;
1348
1349       for (Annotation[] chnk : annels)
1350       {
1351         System.arraycopy(chnk, 0, alignmentAnnotation.annotations, w,
1352                 chnk.length);
1353         w += chnk.length;
1354       }
1355     }
1356     else
1357     {
1358       alignmentAnnotation.restrict(start, end);
1359     }
1360   }
1361
1362   /**
1363    * Invert the column selection from first to end-1. leaves hiddenColumns
1364    * untouched (and unselected)
1365    * 
1366    * @param first
1367    * @param end
1368    */
1369   public void invertColumnSelection(int first, int width)
1370   {
1371     boolean hasHidden = hiddenColumns != null && hiddenColumns.size() > 0;
1372     for (int i = first; i < width; i++)
1373     {
1374       if (contains(i))
1375       {
1376         removeElement(i);
1377       }
1378       else
1379       {
1380         if (!hasHidden || isVisible(i))
1381         {
1382           addElement(i);
1383         }
1384       }
1385     }
1386   }
1387
1388   /**
1389    * add in any unselected columns from the given column selection, excluding
1390    * any that are hidden.
1391    * 
1392    * @param colsel
1393    */
1394   public void addElementsFrom(ColumnSelection colsel)
1395   {
1396     if (colsel != null && !colsel.isEmpty())
1397     {
1398       for (Integer col : colsel.getSelected())
1399       {
1400         if (hiddenColumns != null && isVisible(col.intValue()))
1401         {
1402           selection.add(col);
1403         }
1404       }
1405     }
1406   }
1407
1408   /**
1409    * set the selected columns the given column selection, excluding any columns
1410    * that are hidden.
1411    * 
1412    * @param colsel
1413    */
1414   public void setElementsFrom(ColumnSelection colsel)
1415   {
1416     selection = new IntList();
1417     if (colsel.selection != null && colsel.selection.size() > 0)
1418     {
1419       if (hiddenColumns != null && hiddenColumns.size() > 0)
1420       {
1421         // only select visible columns in this columns selection
1422         addElementsFrom(colsel);
1423       }
1424       else
1425       {
1426         // add everything regardless
1427         for (Integer col : colsel.getSelected())
1428         {
1429           addElement(col);
1430         }
1431       }
1432     }
1433   }
1434
1435   /**
1436    * Add gaps into the sequences aligned to profileseq under the given
1437    * AlignmentView
1438    * 
1439    * @param profileseq
1440    * @param al
1441    *          - alignment to have gaps inserted into it
1442    * @param input
1443    *          - alignment view where sequence corresponding to profileseq is
1444    *          first entry
1445    * @return new Column selection for new alignment view, with insertions into
1446    *         profileseq marked as hidden.
1447    */
1448   public static ColumnSelection propagateInsertions(SequenceI profileseq,
1449           AlignmentI al, AlignmentView input)
1450   {
1451     int profsqpos = 0;
1452
1453     // return propagateInsertions(profileseq, al, )
1454     char gc = al.getGapCharacter();
1455     Object[] alandcolsel = input.getAlignmentAndColumnSelection(gc);
1456     ColumnSelection nview = (ColumnSelection) alandcolsel[1];
1457     SequenceI origseq = ((SequenceI[]) alandcolsel[0])[profsqpos];
1458     nview.propagateInsertions(profileseq, al, origseq);
1459     return nview;
1460   }
1461
1462   /**
1463    * 
1464    * @param profileseq
1465    *          - sequence in al which corresponds to origseq
1466    * @param al
1467    *          - alignment which is to have gaps inserted into it
1468    * @param origseq
1469    *          - sequence corresponding to profileseq which defines gap map for
1470    *          modifying al
1471    */
1472   public void propagateInsertions(SequenceI profileseq, AlignmentI al,
1473           SequenceI origseq)
1474   {
1475     char gc = al.getGapCharacter();
1476     // recover mapping between sequence's non-gap positions and positions
1477     // mapping to view.
1478     pruneDeletions(ShiftList.parseMap(origseq.gapMap()));
1479     int[] viscontigs = getVisibleContigs(0, profileseq.getLength());
1480     int spos = 0;
1481     int offset = 0;
1482     // input.pruneDeletions(ShiftList.parseMap(((SequenceI[])
1483     // alandcolsel[0])[0].gapMap()))
1484     // add profile to visible contigs
1485     for (int v = 0; v < viscontigs.length; v += 2)
1486     {
1487       if (viscontigs[v] > spos)
1488       {
1489         StringBuffer sb = new StringBuffer();
1490         for (int s = 0, ns = viscontigs[v] - spos; s < ns; s++)
1491         {
1492           sb.append(gc);
1493         }
1494         for (int s = 0, ns = al.getHeight(); s < ns; s++)
1495         {
1496           SequenceI sqobj = al.getSequenceAt(s);
1497           if (sqobj != profileseq)
1498           {
1499             String sq = al.getSequenceAt(s).getSequenceAsString();
1500             if (sq.length() <= spos + offset)
1501             {
1502               // pad sequence
1503               int diff = spos + offset - sq.length() - 1;
1504               if (diff > 0)
1505               {
1506                 // pad gaps
1507                 sq = sq + sb;
1508                 while ((diff = spos + offset - sq.length() - 1) > 0)
1509                 {
1510                   // sq = sq
1511                   // + ((diff >= sb.length()) ? sb.toString() : sb
1512                   // .substring(0, diff));
1513                   if (diff >= sb.length())
1514                   {
1515                     sq += sb.toString();
1516                   }
1517                   else
1518                   {
1519                     char[] buf = new char[diff];
1520                     sb.getChars(0, diff, buf, 0);
1521                     sq += buf.toString();
1522                   }
1523                 }
1524               }
1525               sq += sb.toString();
1526             }
1527             else
1528             {
1529               al.getSequenceAt(s).setSequence(
1530                       sq.substring(0, spos + offset) + sb.toString()
1531                               + sq.substring(spos + offset));
1532             }
1533           }
1534         }
1535         // offset+=sb.length();
1536       }
1537       spos = viscontigs[v + 1] + 1;
1538     }
1539     if ((offset + spos) < profileseq.getLength())
1540     {
1541       // pad the final region with gaps.
1542       StringBuffer sb = new StringBuffer();
1543       for (int s = 0, ns = profileseq.getLength() - spos - offset; s < ns; s++)
1544       {
1545         sb.append(gc);
1546       }
1547       for (int s = 0, ns = al.getHeight(); s < ns; s++)
1548       {
1549         SequenceI sqobj = al.getSequenceAt(s);
1550         if (sqobj == profileseq)
1551         {
1552           continue;
1553         }
1554         String sq = sqobj.getSequenceAsString();
1555         // pad sequence
1556         int diff = origseq.getLength() - sq.length();
1557         while (diff > 0)
1558         {
1559           // sq = sq
1560           // + ((diff >= sb.length()) ? sb.toString() : sb
1561           // .substring(0, diff));
1562           if (diff >= sb.length())
1563           {
1564             sq += sb.toString();
1565           }
1566           else
1567           {
1568             char[] buf = new char[diff];
1569             sb.getChars(0, diff, buf, 0);
1570             sq += buf.toString();
1571           }
1572           diff = origseq.getLength() - sq.length();
1573         }
1574       }
1575     }
1576   }
1577
1578   /**
1579    * 
1580    * @return true if there are columns marked
1581    */
1582   public boolean hasSelectedColumns()
1583   {
1584     return (selection != null && selection.size() > 0);
1585   }
1586
1587   /**
1588    * 
1589    * @return true if there are columns hidden
1590    */
1591   public boolean hasHiddenColumns()
1592   {
1593     return hiddenColumns != null && hiddenColumns.size() > 0;
1594   }
1595
1596   /**
1597    * 
1598    * @return true if there are more than one set of columns hidden
1599    */
1600   public boolean hasManyHiddenColumns()
1601   {
1602     return hiddenColumns != null && hiddenColumns.size() > 1;
1603   }
1604
1605   /**
1606    * mark the columns corresponding to gap characters as hidden in the column
1607    * selection
1608    * 
1609    * @param sr
1610    */
1611   public void hideInsertionsFor(SequenceI sr)
1612   {
1613     List<int[]> inserts = sr.getInsertions();
1614     for (int[] r : inserts)
1615     {
1616       hideColumns(r[0], r[1]);
1617     }
1618   }
1619
1620   public boolean filterAnnotations(Annotation[] annotations,
1621           AnnotationFilterParameter filterParams)
1622   {
1623     // JBPNote - this method needs to be refactored to become independent of
1624     // viewmodel package
1625     this.revealAllHiddenColumns();
1626     this.clear();
1627     int count = 0;
1628     do
1629     {
1630       if (annotations[count] != null)
1631       {
1632
1633         boolean itemMatched = false;
1634
1635         if (filterParams.getThresholdType() == AnnotationFilterParameter.ThresholdType.ABOVE_THRESHOLD
1636                 && annotations[count].value >= filterParams
1637                         .getThresholdValue())
1638         {
1639           itemMatched = true;
1640         }
1641         if (filterParams.getThresholdType() == AnnotationFilterParameter.ThresholdType.BELOW_THRESHOLD
1642                 && annotations[count].value <= filterParams
1643                         .getThresholdValue())
1644         {
1645           itemMatched = true;
1646         }
1647
1648         if (filterParams.isFilterAlphaHelix()
1649                 && annotations[count].secondaryStructure == 'H')
1650         {
1651           itemMatched = true;
1652         }
1653
1654         if (filterParams.isFilterBetaSheet()
1655                 && annotations[count].secondaryStructure == 'E')
1656         {
1657           itemMatched = true;
1658         }
1659
1660         if (filterParams.isFilterTurn()
1661                 && annotations[count].secondaryStructure == 'S')
1662         {
1663           itemMatched = true;
1664         }
1665
1666         String regexSearchString = filterParams.getRegexString();
1667         if (regexSearchString != null
1668                 && !filterParams.getRegexSearchFields().isEmpty())
1669         {
1670           List<SearchableAnnotationField> fields = filterParams
1671                   .getRegexSearchFields();
1672           try
1673           {
1674             if (fields.contains(SearchableAnnotationField.DISPLAY_STRING)
1675                     && annotations[count].displayCharacter
1676                             .matches(regexSearchString))
1677             {
1678               itemMatched = true;
1679             }
1680           } catch (java.util.regex.PatternSyntaxException pse)
1681           {
1682             if (annotations[count].displayCharacter
1683                     .equals(regexSearchString))
1684             {
1685               itemMatched = true;
1686             }
1687           }
1688           if (fields.contains(SearchableAnnotationField.DESCRIPTION)
1689                   && annotations[count].description != null
1690                   && annotations[count].description
1691                           .matches(regexSearchString))
1692           {
1693             itemMatched = true;
1694           }
1695         }
1696
1697         if (itemMatched)
1698         {
1699           this.addElement(count);
1700         }
1701       }
1702       count++;
1703     } while (count < annotations.length);
1704     return false;
1705   }
1706
1707   /**
1708    * Returns a hashCode built from selected columns and hidden column ranges
1709    */
1710   @Override
1711   public int hashCode()
1712   {
1713     int hashCode = selection.hashCode();
1714     if (hiddenColumns != null)
1715     {
1716       for (int[] hidden : hiddenColumns)
1717       {
1718         hashCode = 31 * hashCode + hidden[0];
1719         hashCode = 31 * hashCode + hidden[1];
1720       }
1721     }
1722     return hashCode;
1723   }
1724
1725   /**
1726    * Answers true if comparing to a ColumnSelection with the same selected
1727    * columns and hidden columns, else false
1728    */
1729   @Override
1730   public boolean equals(Object obj)
1731   {
1732     if (!(obj instanceof ColumnSelection))
1733     {
1734       return false;
1735     }
1736     ColumnSelection that = (ColumnSelection) obj;
1737
1738     /*
1739      * check columns selected are either both null, or match
1740      */
1741     if (this.selection == null)
1742     {
1743       if (that.selection != null)
1744       {
1745         return false;
1746       }
1747     }
1748     if (!this.selection.equals(that.selection))
1749     {
1750       return false;
1751     }
1752
1753     /*
1754      * check hidden columns are either both null, or match
1755      */
1756     if (this.hiddenColumns == null)
1757     {
1758       return (that.hiddenColumns == null);
1759     }
1760     if (that.hiddenColumns == null
1761             || that.hiddenColumns.size() != this.hiddenColumns.size())
1762     {
1763       return false;
1764     }
1765     int i = 0;
1766     for (int[] thisRange : hiddenColumns)
1767     {
1768       int[] thatRange = that.hiddenColumns.get(i++);
1769       if (thisRange[0] != thatRange[0] || thisRange[1] != thatRange[1])
1770       {
1771         return false;
1772       }
1773     }
1774     return true;
1775   }
1776
1777   /**
1778    * Updates the column selection depending on the parameters, and returns true
1779    * if any change was made to the selection
1780    * 
1781    * @param markedColumns
1782    *          a set identifying marked columns (base 0)
1783    * @param startCol
1784    *          the first column of the range to operate over (base 0)
1785    * @param endCol
1786    *          the last column of the range to operate over (base 0)
1787    * @param invert
1788    *          if true, deselect marked columns and select unmarked
1789    * @param extendCurrent
1790    *          if true, extend rather than replacing the current column selection
1791    * @param toggle
1792    *          if true, toggle the selection state of marked columns
1793    * 
1794    * @return
1795    */
1796   public boolean markColumns(BitSet markedColumns, int startCol,
1797           int endCol, boolean invert, boolean extendCurrent, boolean toggle)
1798   {
1799     boolean changed = false;
1800     if (!extendCurrent && !toggle)
1801     {
1802       changed = !this.isEmpty();
1803       clear();
1804     }
1805     if (invert)
1806     {
1807       // invert only in the currently selected sequence region
1808       int i = markedColumns.nextClearBit(startCol);
1809       int ibs = markedColumns.nextSetBit(startCol);
1810       while (i >= startCol && i <= endCol)
1811       {
1812         if (ibs < 0 || i < ibs)
1813         {
1814           changed = true;
1815           if (toggle && contains(i))
1816           {
1817             removeElement(i++);
1818           }
1819           else
1820           {
1821             addElement(i++);
1822           }
1823         }
1824         else
1825         {
1826           i = markedColumns.nextClearBit(ibs);
1827           ibs = markedColumns.nextSetBit(i);
1828         }
1829       }
1830     }
1831     else
1832     {
1833       int i = markedColumns.nextSetBit(startCol);
1834       while (i >= startCol && i <= endCol)
1835       {
1836         changed = true;
1837         if (toggle && contains(i))
1838         {
1839           removeElement(i);
1840         }
1841         else
1842         {
1843           addElement(i);
1844         }
1845         i = markedColumns.nextSetBit(i + 1);
1846       }
1847     }
1848     return changed;
1849   }
1850
1851   /**
1852    * Adjusts column selections, and the given selection group, to match the
1853    * range of a stretch (e.g. mouse drag) operation
1854    * <p>
1855    * Method refactored from ScalePanel.mouseDragged
1856    * 
1857    * @param res
1858    *          current column position, adjusted for hidden columns
1859    * @param sg
1860    *          current selection group
1861    * @param min
1862    *          start position of the stretch group
1863    * @param max
1864    *          end position of the stretch group
1865    */
1866   public void stretchGroup(int res, SequenceGroup sg, int min, int max)
1867   {
1868     if (!contains(res))
1869     {
1870       addElement(res);
1871     }
1872
1873     if (res > sg.getStartRes())
1874     {
1875       // expand selection group to the right
1876       sg.setEndRes(res);
1877     }
1878     if (res < sg.getStartRes())
1879     {
1880       // expand selection group to the left
1881       sg.setStartRes(res);
1882     }
1883
1884     /*
1885      * expand or shrink column selection to match the
1886      * range of the drag operation
1887      */
1888     for (int col = min; col <= max; col++)
1889     {
1890       if (col < sg.getStartRes() || col > sg.getEndRes())
1891       {
1892         // shrinking drag - remove from selection
1893         removeElement(col);
1894       }
1895       else
1896       {
1897         // expanding drag - add to selection
1898         addElement(col);
1899       }
1900     }
1901   }
1902 }