in progress
[jalview.git] / forester / java / src / org / forester / evoinference / TestPhylogenyReconstruction.java
index 7c78c5c..f6d3975 100644 (file)
@@ -32,11 +32,14 @@ import java.io.FileInputStream;
 import java.io.StringWriter;
 import java.util.Date;
 import java.util.List;
+import java.util.SortedSet;
 
+import org.forester.archaeopteryx.Archaeopteryx;
 import org.forester.evoinference.distance.NeighborJoining;
 import org.forester.evoinference.distance.NeighborJoiningF;
 import org.forester.evoinference.distance.NeighborJoiningR;
 import org.forester.evoinference.distance.PairwiseDistanceCalculator;
+import org.forester.evoinference.distance.S;
 import org.forester.evoinference.matrix.character.BasicCharacterStateMatrix;
 import org.forester.evoinference.matrix.character.CharacterStateMatrix;
 import org.forester.evoinference.matrix.character.CharacterStateMatrix.BinaryStates;
@@ -74,6 +77,13 @@ public class TestPhylogenyReconstruction {
         //        else {
         //            System.out.println( "  failed." );
         //        }
+        System.out.println( "S" );
+        if ( testS() ) {
+            System.out.println( "  OK." );
+        }
+        else {
+            System.out.println( "  failed." );
+        }
         System.out.println( "NJR" );
         if ( testNeighborJoiningR() ) {
             System.out.println( "  OK." );
@@ -81,7 +91,7 @@ public class TestPhylogenyReconstruction {
         else {
             System.out.println( "  failed." );
         }
-        //timeNeighborJoining();
+       // timeNeighborJoining();
     }
 
     public static boolean test( final File test_dir ) {
@@ -109,6 +119,12 @@ public class TestPhylogenyReconstruction {
             return false;
         }
         System.out.println( "OK." );
+        System.out.print( "  Datastructure S: " );
+        if ( !testS() ) {
+            System.out.println( "failed." );
+            return false;
+        }
+        System.out.println( "OK." );
         System.out.print( "  Neighbor Joining: " );
         if ( !testNeighborJoining() ) {
             System.out.println( "failed." );
@@ -2228,9 +2244,146 @@ public class TestPhylogenyReconstruction {
         return true;
     }
 
+    private static boolean testS() {
+        try {
+            final S s0 = new S();
+            s0.initialize( 1 );
+            s0.addPairing( 0, 1, 0 );
+            s0.addPairing( 7, 8, 0 );
+            s0.addPairing( 4, 55, 0 );
+            s0.addPairing( 2, 3, 0 );
+            s0.addPairing( 4, 5, 0 );
+            s0.addPairing( 5, 6666, 0 );
+            s0.addPairing( 5, 666, 0 );
+            s0.addPairing( 5, 66, 0 );
+            s0.addPairing( 5, 6, 0 );
+            s0.addPairing( 6, 7, 0 );
+            s0.addPairing( 3, 4, 0 );
+            s0.addPairing( 1, 2, 0 );
+            if ( s0.size() != 1 ) {
+                return false;
+            }
+            if ( s0.getS( 0 ).size() != 8 ) {
+                return false;
+            }
+            if ( s0.getValues( 0, 0 ).size() != 1 ) {
+                return false;
+            }
+            if ( s0.getValues( 1, 0 ).size() != 1 ) {
+                return false;
+            }
+            if ( s0.getValues( 2, 0 ).size() != 1 ) {
+                return false;
+            }
+            if ( s0.getValues( 3, 0 ).size() != 1 ) {
+                return false;
+            }
+            if ( s0.getValues( 4, 0 ).size() != 2 ) {
+                return false;
+            }
+            if ( s0.getValues( 5, 0 ).size() != 4 ) {
+                return false;
+            }
+            if ( s0.getValues( 6, 0 ).size() != 1 ) {
+                return false;
+            }
+            if ( s0.getValues( 7, 0 ).size() != 1 ) {
+                return false;
+            }
+            if ( !s0.getValues( 0, 0 ).contains( 1 ) ) {
+                return false;
+            }
+            if ( !s0.getValues( 5, 0 ).contains( 6 ) ) {
+                return false;
+            }
+            if ( !s0.getValues( 5, 0 ).contains( 66 ) ) {
+                return false;
+            }
+            if ( !s0.getValues( 5, 0 ).contains( 666 ) ) {
+                return false;
+            }
+            if ( !s0.getValues( 5, 0 ).contains( 6666 ) ) {
+                return false;
+            }
+            s0.removePairing( 5, 6666, 0 );
+            if ( s0.getValues( 5, 0 ).contains( 6666 ) ) {
+                return false;
+            }
+            s0.removePairing( 5, 666, 0 );
+            if ( s0.getValues( 5, 0 ).contains( 666 ) ) {
+                return false;
+            }
+            s0.removePairing( 5, 66, 0 );
+            if ( s0.getValues( 5, 0 ).contains( 66 ) ) {
+                return false;
+            }
+            if ( s0.getValues( 5, 0 ).size() != 1 ) {
+                return false;
+            }
+            if ( s0.getS( 0 ).size() != 8 ) {
+                return false;
+            }
+            s0.removePairing( 5, 6, 0 );
+            if ( s0.getS( 0 ).size() != 7 ) {
+                return false;
+            }
+            s0.addPairing( 5, 6, 0 );
+            if ( s0.getS( 0 ).size() != 8 ) {
+                return false;
+            }
+            if ( s0.getValues( 5, 0 ).size() != 1 ) {
+                return false;
+            }
+            if ( !s0.getValues( 5, 0 ).contains( 6 ) ) {
+                return false;
+            }
+            s0.addPairing( 5, 403, 0 );
+            if ( s0.getValues( 5, 0 ).size() != 2 ) {
+                return false;
+            }
+            if ( !s0.getValues( 5, 0 ).contains( 403 ) ) {
+                return false;
+            }
+            s0.addPairing( 693, 100, 0 );
+            s0.addPairing( 693, 101, 0 );
+            if ( s0.getValues( 693, 0 ).size() != 2 ) {
+                return false;
+            }
+            s0.addPairing( 2, 33, 0 );
+            s0.addPairing( 2, 333, 0 );
+            final SortedSet<Integer>[] a = s0.toArray( 0 );
+            if ( !a[ 0 ].contains( 1 ) ) {
+                return false;
+            }
+            if ( a[ 0 ].size() != 1 ) {
+                return false;
+            }
+            if ( !a[ 1 ].contains( 2 ) ) {
+                return false;
+            }
+            if ( a[ 1 ].size() != 1 ) {
+                return false;
+            }
+            if ( !a[ 2 ].contains( 3 ) ) {
+                return false;
+            }
+            if ( !a[ 2 ].contains( 33 ) ) {
+                return false;
+            }
+            if ( !a[ 2 ].contains( 333 ) ) {
+                return false;
+            }
+        }
+        catch ( final Exception e ) {
+            e.printStackTrace( System.out );
+            return false;
+        }
+        return true;
+    }
+
     private static boolean testNeighborJoiningR() {
         try {
-            final NeighborJoiningR nj = NeighborJoiningR.createInstance();
+            //            final NeighborJoiningR nj0 = NeighborJoiningR.createInstance();
             //            final BasicSymmetricalDistanceMatrix m0 = new BasicSymmetricalDistanceMatrix( 4 );
             //            m0.setIdentifier( 0, "A" );
             //            m0.setIdentifier( 1, "B" );
@@ -2239,7 +2392,7 @@ public class TestPhylogenyReconstruction {
             //            m0.setRow( "5 ", 1 );
             //            m0.setRow( "3 6 ", 2 );
             //            m0.setRow( "7.5 10.5 5.5", 3 );
-            //            final Phylogeny p0 = nj.execute( m0 );
+            //            final Phylogeny p0 = nj0.execute( m0 );
             //            p0.reRoot( p0.getNode( "D" ) );
             //            //  Archaeopteryx.createApplication( p0 );
             //            if ( isUnequal( p0.getNode( "A" ).getDistanceToParent(), 1 ) ) {
@@ -2272,8 +2425,8 @@ public class TestPhylogenyReconstruction {
             //            m.setIdentifier( 3, "D" );
             //            m.setIdentifier( 4, "E" );
             //            m.setIdentifier( 5, "F" );
-            //            nj = NeighborJoiningR.createInstance();
-            //            final Phylogeny p1 = nj.execute( m );
+            //            final NeighborJoiningR nj1 = NeighborJoiningR.createInstance();
+            //            final Phylogeny p1 = nj1.execute( m );
             //            p1.reRoot( p1.getNode( "F" ) );
             //            Archaeopteryx.createApplication( p1 );
             //            if ( isUnequal( p1.getNode( "A" ).getDistanceToParent(), 1 ) ) {
@@ -2328,10 +2481,30 @@ public class TestPhylogenyReconstruction {
             m.setRow( "1.52430 1.44650 0.59580 0.46310 0.00000 0.34840 0.30830", 4 );
             m.setRow( "1.60430 1.43890 0.61790 0.50610 0.34840 0.00000 0.26920", 5 );
             m.setRow( "1.59050 1.46290 0.55830 0.47100 0.30830 0.26920 0.00000", 6 );
-            final NeighborJoiningR njr = NeighborJoiningR.createInstance( true, 6 );
+            final NeighborJoining nj = NeighborJoining.createInstance( true, 6 );
             //nj = NeighborJoining.createInstance( true, 6 );
-            final Phylogeny p2 = njr.execute( m );
-            //  Archaeopteryx.createApplication( p2 );
+            final Phylogeny pnj = nj.execute( m );
+            Archaeopteryx.createApplication( pnj );
+            //
+            m = new BasicSymmetricalDistanceMatrix( 7 );
+            m.setIdentifier( 0, "Bovine" );
+            m.setIdentifier( 1, "Mouse" );
+            m.setIdentifier( 2, "Gibbon" );
+            m.setIdentifier( 3, "Orang" );
+            m.setIdentifier( 4, "Gorilla" );
+            m.setIdentifier( 5, "Chimp" );
+            m.setIdentifier( 6, "Human" );
+            m.setRow( "0.00000 1.68660 1.71980 1.66060 1.52430 1.60430 1.59050", 0 );
+            m.setRow( "1.68660 0.00000 1.52320 1.48410 1.44650 1.43890 1.46290", 1 );
+            m.setRow( "1.71980 1.52320 0.00000 0.71150 0.59580 0.61790 0.55830", 2 );
+            m.setRow( "1.66060 1.48410 0.71150 0.00000 0.46310 0.50610 0.47100", 3 );
+            m.setRow( "1.52430 1.44650 0.59580 0.46310 0.00000 0.34840 0.30830", 4 );
+            m.setRow( "1.60430 1.43890 0.61790 0.50610 0.34840 0.00000 0.26920", 5 );
+            m.setRow( "1.59050 1.46290 0.55830 0.47100 0.30830 0.26920 0.00000", 6 );
+            final NeighborJoiningR nj2 = NeighborJoiningR.createInstance( true, 6 );
+            //nj = NeighborJoining.createInstance( true, 6 );
+            final Phylogeny p2 = nj2.execute( m );
+              Archaeopteryx.createApplication( p2 );
             p2.reRoot( p2.getNode( "Bovine" ) );
             if ( isUnequal( p2.getNode( "Chimp" ).getDistanceToParent(), 0.151675 ) ) {
                 return false;
@@ -2743,6 +2916,15 @@ public class TestPhylogenyReconstruction {
     }
 
     private static void timeNeighborJoining() {
+        final NeighborJoiningR njr = NeighborJoiningR.createInstance();
+        for( int n = 3; n <= 9; ++n ) {
+            final int x = ( int ) Math.pow( 2, n );
+            final BasicSymmetricalDistanceMatrix mt = new BasicSymmetricalDistanceMatrix( x );
+            mt.randomize( new Date().getTime() );
+            final long start_time = new Date().getTime();
+            njr.execute( mt );
+            System.out.println( "Size: " + x + " -> " + ( new Date().getTime() - start_time ) + "ms" );
+        }
         final NeighborJoiningF njf = NeighborJoiningF.createInstance();
         for( int n = 3; n <= 9; ++n ) {
             final int x = ( int ) Math.pow( 2, n );