compactor work

[jalview.git] / forester / java / src / org / forester / msa_compactor / MsaProperties.java

diff --git a/forester/java/src/org/forester/msa_compactor/MsaProperties.java b/forester/java/src/org/forester/msa_compactor/MsaProperties.java
index 57cc40c..1abc638 100644 (file)
--- a/forester/java/src/org/forester/msa_compactor/MsaProperties.java
+++ b/forester/java/src/org/forester/msa_compactor/MsaProperties.java
@@ -29,36 +29,62 @@ import org.forester.msa.MsaMethods;
 
 public final class MsaProperties {
 
-    final private double _average_identity_ratio;
+    final private double _entropy21;
+    final private double _entropy7;
     final private double _gap_ratio;
     final private int    _length;
     final private int    _number_of_sequences;
-
-    public MsaProperties( final Msa msa ) {
-        _number_of_sequences = msa.getNumberOfSequences();
-        _length = msa.getLength();
-        _gap_ratio = MsaMethods.calcGapRatio( msa );
-        _average_identity_ratio = MsaMethods.calculateIdentityRatio( 0, msa.getLength() - 1, msa ).arithmeticMean();
-    }
+    final private double _avg_number_of_gaps;
+    final private String _removed_seq;
 
     public MsaProperties( final int number_of_sequences,
                           final int length,
                           final double gap_ratio,
-                          final double average_identity_ratio ) {
+                          final double entropy7,
+                          final double entropy21,
+                          final double avg_number_of_gaps,
+                          final String removed_seq ) {
         _number_of_sequences = number_of_sequences;
         _length = length;
         _gap_ratio = gap_ratio;
-        _average_identity_ratio = average_identity_ratio;
+        _entropy7 = entropy7;
+        _entropy21 = entropy21;
+        _avg_number_of_gaps = avg_number_of_gaps;
+        _removed_seq = removed_seq;
     }
 
-    public final double getAverageIdentityRatio() {
-        return _average_identity_ratio;
+    public MsaProperties( final Msa msa, final String removed_seq, final boolean calculate_normalized_shannon_entropy ) {
+        _number_of_sequences = msa.getNumberOfSequences();
+        _length = msa.getLength();
+        _gap_ratio = MsaMethods.calcGapRatio( msa );
+        _removed_seq = removed_seq;
+        _avg_number_of_gaps = MsaMethods.calcNumberOfGapsStats( msa ).arithmeticMean();
+        if ( calculate_normalized_shannon_entropy ) {
+            _entropy7 = MsaMethods.calcNormalizedShannonsEntropy( 7, msa );
+            _entropy21 = MsaMethods.calcNormalizedShannonsEntropy( 21, msa );
+        }
+        else {
+            _entropy7 = -1;
+            _entropy21 = -1;
+        }
+    }
+
+    public final double getEntropy21() {
+        return _entropy21;
+    }
+
+    public final double getEntropy7() {
+        return _entropy7;
     }
 
     public final double getGapRatio() {
         return _gap_ratio;
     }
 
+    public final double getAvgNumberOfGaps() {
+        return _avg_number_of_gaps;
+    }
+    
     public final int getLength() {
         return _length;
     }
@@ -66,4 +92,8 @@ public final class MsaProperties {
     public final int getNumberOfSequences() {
         return _number_of_sequences;
     }
+
+    public final String getRemovedSeq() {
+        return _removed_seq;
+    }
 }