3 // FORESTER -- software libraries and applications
4 // for evolutionary biology research and applications.
6 // Copyright (C) 2008-2009 Christian M. Zmasek
7 // Copyright (C) 2008-2009 Burnham Institute for Medical Research
10 // This library is free software; you can redistribute it and/or
11 // modify it under the terms of the GNU Lesser General Public
12 // License as published by the Free Software Foundation; either
13 // version 2.1 of the License, or (at your option) any later version.
15 // This library is distributed in the hope that it will be useful,
16 // but WITHOUT ANY WARRANTY; without even the implied warranty of
17 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 // Lesser General Public License for more details.
20 // You should have received a copy of the GNU Lesser General Public
21 // License along with this library; if not, write to the Free Software
22 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
24 // Contact: phylosoft @ gmail . com
25 // WWW: www.phylosoft.org/forester
27 package org.forester.evoinference.matrix.distance;
29 import java.io.IOException;
30 import java.io.Writer;
31 import java.text.DecimalFormat;
32 import java.text.NumberFormat;
33 import java.util.StringTokenizer;
35 import org.forester.util.ForesterUtil;
36 import org.forester.util.IllegalFormatUseException;
38 public class BasicSymmetricalDistanceMatrix implements DistanceMatrix {
40 NumberFormat nf1 = NumberFormat.getInstance();
41 private final static NumberFormat PHYLIP_FORMATTER = new DecimalFormat( "0.000000" );
42 final double[][] _values;
43 final String[] _identifiers;
45 public BasicSymmetricalDistanceMatrix( final int size ) {
46 _values = new double[ size ][ size ];
47 _identifiers = new String[ size ];
51 public String getIdentifier( final int i ) {
52 return _identifiers[ i ];
56 public int getIndex( final String identifier ) {
57 for( int i = 0; i < _identifiers.length; i++ ) {
58 if ( getIdentifier( i ).equals( identifier ) ) {
62 throw new IllegalArgumentException( "identifier [" + identifier + "] not found in distance matrix" );
66 public int getSize() {
67 return _values.length;
71 public double getValue( final int col, final int row ) {
73 if ( col >= getSize() ) {
74 throw new IndexOutOfBoundsException( "" );
78 else if ( col > row ) {
79 return _values[ row ][ col ];
81 return _values[ col ][ row ];
84 public void randomize( final long seed ) {
85 final java.util.Random r = new java.util.Random( seed );
86 for( int j = 0; j < getSize(); ++j ) {
87 for( int i = 0; i < j; ++i ) {
88 setValue( i, j, r.nextDouble() );
94 public void setIdentifier( final int i, final String identifier ) {
95 _identifiers[ i ] = identifier;
98 public void setRow( final String s, final int row ) {
99 final StringTokenizer tk = new StringTokenizer( s );
101 while ( tk.hasMoreElements() ) {
102 setValue( i, row, new Double( tk.nextToken() ).doubleValue() );
108 public void setValue( final int col, final int row, final double d ) {
109 if ( ( col == row ) && ( d != 0.0 ) ) {
110 throw new IllegalArgumentException( "attempt to set a non-zero value on the diagonal of a symmetrical distance matrix" );
112 else if ( col > row ) {
113 _values[ row ][ col ] = d;
115 _values[ col ][ row ] = d;
118 public void write( final Writer w ) throws IOException {
120 w.write( getSize() + "" );
121 w.write( ForesterUtil.LINE_SEPARATOR );
122 for( int row = 0; row < getSize(); ++row ) {
123 if ( !ForesterUtil.isEmpty( getIdentifier( row ) ) ) {
124 w.write( ForesterUtil.pad( getIdentifier( row ), 10, ' ', false ).toString() );
129 throw new IllegalFormatUseException( "Phylip format does not allow empty identifiers" );
131 for( int col = 0; col < getSize(); ++col ) {
132 w.write( PHYLIP_FORMATTER.format( getValue( col, row ) ) );
133 if ( col < getSize() - 1 ) {
138 if ( row < getSize() - 1 ) {
139 w.write( ForesterUtil.LINE_SEPARATOR );
144 private StringBuffer toPhylip() {
145 final StringBuffer sb = new StringBuffer();
150 sb.append( getSize() );
151 sb.append( ForesterUtil.LINE_SEPARATOR );
152 for( int row = 0; row < getSize(); ++row ) {
153 if ( !ForesterUtil.isEmpty( getIdentifier( row ) ) ) {
154 sb.append( ForesterUtil.pad( getIdentifier( row ), 10, ' ', false ) );
159 throw new IllegalFormatUseException( "Phylip format does not allow empty identifiers" );
162 for( int col = 0; col < getSize(); ++col ) {
163 sb.append( PHYLIP_FORMATTER.format( getValue( col, row ) ) );
164 if ( col < getSize() - 1 ) {
169 if ( row < getSize() - 1 ) {
170 sb.append( ForesterUtil.LINE_SEPARATOR );
177 public String toString() {
178 return toPhylip().toString();
182 public StringBuffer toStringBuffer( final Format format ) {
187 throw new IllegalArgumentException( "Unknown format:" + format );