in progress...
[jalview.git] / forester / java / src / org / forester / msa / DeleteableMsa.java
index 17f72bb..5126e47 100644 (file)
 
 package org.forester.msa;
 
-import java.util.HashMap;
 import java.util.List;
 
 import org.forester.sequence.BasicSequence;
-import org.forester.sequence.Sequence;
+import org.forester.sequence.MolecularSequence;
 
 public final class DeleteableMsa extends BasicMsa {
 
-    private int                      _length                 = 0;
-    private int                      _mapped_col_positions[] = null;
-    private int                      _mapped_row_positions[] = null;
-    private HashMap<String, Integer> _seq_id_to_row_map      = null;
-    private int                      _seqs                   = 0;
+    private int _length                 = 0;
+    private int _mapped_col_positions[] = null;
+    private int _mapped_row_positions[] = null;
+    private int _seqs                   = 0;
 
     private DeleteableMsa( final BasicMsa msa ) {
         super( msa );
@@ -48,14 +46,42 @@ public final class DeleteableMsa extends BasicMsa {
         for( int i = 0; i < _mapped_row_positions.length; ++i ) {
             _mapped_row_positions[ i ] = i;
         }
-        _seq_id_to_row_map = new HashMap<String, Integer>();
-        for( int row = 0; row < msa.getNumberOfSequences(); ++row ) {
-            _seq_id_to_row_map.put( msa.getIdentifier( row ), row );
-        }
         _length = msa.getLength();
         _seqs = msa.getNumberOfSequences();
     }
 
+    public final double[] calcGappiness() {
+        final int length = getLength();
+        final double gappiness[] = new double[ length ];
+        final int seqs = getNumberOfSequences();
+        for( int row = 0; row < seqs; ++row ) {
+            for( int col = 0; col < length; ++col ) {
+            }
+        }
+        return gappiness;
+    }
+
+    public static int calcGapSumPerColumn( final Msa msa, final int col ) {
+        int gap_rows = 0;
+        for( int j = 0; j < msa.getNumberOfSequences(); ++j ) {
+            if ( msa.isGapAt( j, col ) ) {
+                gap_rows++;
+            }
+        }
+        return gap_rows;
+    }
+
+    public short determineMaxIdLength() {
+        short max = 0;
+        for( int row = 0; row < getNumberOfSequences(); ++row ) {
+            final short l = ( short ) getIdentifier( row ).length();
+            if ( l > max ) {
+                max = l;
+            }
+        }
+        return max;
+    }
+
     final public void deleteGapColumns( final double max_allowed_gap_ratio ) {
         if ( ( max_allowed_gap_ratio < 0 ) || ( max_allowed_gap_ratio > 1 ) ) {
             throw new IllegalArgumentException( "max allowed gap ration is out of range: " + max_allowed_gap_ratio );
@@ -70,13 +96,13 @@ public final class DeleteableMsa extends BasicMsa {
 
     final public void deleteGapOnlyColumns() {
         for( int col = getLength() - 1; col >= 0; --col ) {
-            if ( MsaMethods.calcGapSumPerColumn( this, col ) == getNumberOfSequences() ) {
+            if ( isAllGap( col ) ) {
                 deleteColumn( col );
             }
         }
     }
 
-    final public void deleteRow( final String id ) {
+    final public MolecularSequence deleteRow( final String id, final boolean return_removed_seq ) {
         int row = -1;
         for( int r = 0; r < getNumberOfSequences(); ++r ) {
             if ( getIdentifier( r ).equals( id ) ) {
@@ -87,11 +113,30 @@ public final class DeleteableMsa extends BasicMsa {
         if ( row < 0 ) {
             throw new IllegalArgumentException( "id [" + id + "] not found" );
         }
+        MolecularSequence s = null;
+        StringBuilder sb = null;
+        if ( return_removed_seq ) {
+            s = getSequence( row );
+            final char[] x = s.getMolecularSequence();
+            sb = new StringBuilder( x.length );
+            for( final char element : x ) {
+                if ( element != MolecularSequence.GAP ) {
+                    sb.append( element );
+                }
+            }
+        }
         deleteRow( row );
+        if ( return_removed_seq ) {
+            return new BasicSequence( new String( s.getIdentifier() ), sb.toString(), s.getType() );
+        }
+        else {
+            return null;
+        }
     }
 
     @Override
     final public String getIdentifier( final int row ) {
+        checkRow( row );
         return super.getIdentifier( _mapped_row_positions[ row ] );
     }
 
@@ -107,45 +152,69 @@ public final class DeleteableMsa extends BasicMsa {
 
     @Override
     final public char getResidueAt( final int row, final int col ) {
+        checkRow( row );
+        checkColumn( col );
         return super.getResidueAt( _mapped_row_positions[ row ], _mapped_col_positions[ col ] );
     }
 
     @Override
-    public Sequence getSequence( final int row ) {
+    public MolecularSequence getSequence( final int row ) {
+        checkRow( row );
         return new BasicSequence( getIdentifier( row ), getSequenceAsString( row ).toString(), getType() );
     }
 
+    final public boolean isAllGap( final int col ) {
+        final int m_col = _mapped_col_positions[ col ];
+        for( int j = 0; j < getNumberOfSequences(); ++j ) {
+            if ( super.getResidueAt( _mapped_row_positions[ j ], m_col ) != MolecularSequence.GAP ) {
+                return false;
+            }
+        }
+        return true;
+    }
+
     @Override
     final public void setIdentifier( final int row, final String id ) {
+        checkRow( row );
         super.setIdentifier( _mapped_row_positions[ row ], id );
     }
 
     @Override
     final public void setResidueAt( final int row, final int col, final char residue ) {
+        checkRow( row );
+        checkColumn( col );
         super.setResidueAt( _mapped_row_positions[ row ], _mapped_col_positions[ col ], residue );
     }
 
-    final private void deleteColumn( final int col ) {
+    final private void checkColumn( final int col ) {
         if ( ( col >= _length ) || ( col < 0 ) ) {
             throw new IllegalArgumentException( "column " + col + " is out of range" );
         }
-        for( int c = col; c < _length - 1; ++c ) {
+    }
+
+    final private void checkRow( final int row ) {
+        if ( ( row >= _seqs ) || ( row < 0 ) ) {
+            throw new IllegalArgumentException( "row " + row + " is out of range" );
+        }
+    }
+
+    final private void deleteColumn( final int col ) {
+        checkColumn( col );
+        for( int c = col; c < ( _length - 1 ); ++c ) {
             _mapped_col_positions[ c ] = _mapped_col_positions[ c + 1 ];
         }
         --_length;
     }
 
     final private void deleteRow( final int row ) {
-        if ( ( row >= _seqs ) || ( row < 0 ) ) {
-            throw new IllegalArgumentException( "row " + row + " is out of range" );
-        }
-        for( int r = row; r < _seqs - 1; ++r ) {
+        checkRow( row );
+        for( int r = row; r < ( _seqs - 1 ); ++r ) {
             _mapped_row_positions[ r ] = _mapped_row_positions[ r + 1 ];
         }
         --_seqs;
     }
 
-    public final static DeleteableMsa createInstance( final List<Sequence> seqs ) {
+    public final static DeleteableMsa createInstance( final List<MolecularSequence> seqs ) {
         return new DeleteableMsa( ( BasicMsa ) BasicMsa.createInstance( seqs ) );
     }