in progress
[jalview.git] / forester / java / src / org / forester / application / mcc.java
1 // $Id:
2 // FORESTER -- software libraries and applications
3 // for evolutionary biology research and applications.
4 //
5 // Copyright (C) 2012 Christian M. Zmasek
6 // Copyright (C) 2012 Sanford Burnham Medical Research Institute
7 // All rights reserved
8 //
9 // This library is free software; you can redistribute it and/or
10 // modify it under the terms of the GNU Lesser General Public
11 // License as published by the Free Software Foundation; either
12 // version 2.1 of the License, or (at your option) any later version.
13 //
14 // This library is distributed in the hope that it will be useful,
15 // but WITHOUT ANY WARRANTY; without even the implied warranty of
16 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17 // Lesser General Public License for more details.
18 //
19 // You should have received a copy of the GNU Lesser General Public
20 // License along with this library; if not, write to the Free Software
21 // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
22 //
23 // Contact: phylosoft @ gmail . com
24 // WWW: www.phylosoft.org/forester
25
26 package org.forester.application;
27
28 import java.io.File;
29 import java.io.FileInputStream;
30
31 import org.forester.io.parsers.GeneralMsaParser;
32 import org.forester.msa.Msa;
33 import org.forester.msa.MsaMethods;
34 import org.forester.util.BasicDescriptiveStatistics;
35 import org.forester.util.CommandLineArguments;
36 import org.forester.util.DescriptiveStatistics;
37 import org.forester.util.ForesterUtil;
38
39 public class mcc {
40
41     final static private String HELP_OPTION_1 = "help";
42     final static private String HELP_OPTION_2 = "h";
43     final static private String FROM_OPTION   = "f";
44     final static private String TO_OPTION     = "t";
45     final static private String STEP_OPTION   = "s";
46     final static private String WINDOW_OPTION = "w";
47     final static private String PRG_NAME      = "mcc";
48     final static private String PRG_DESC      = "msa consensus conservation";
49     final static private String PRG_VERSION   = "1.00";
50     final static private String PRG_DATE      = "2012.05.18";
51     final static private String E_MAIL        = "phylosoft@gmail.com";
52     final static private String WWW           = "www.phylosoft.org/forester/";
53
54     public static void main( final String args[] ) {
55         try {
56             final CommandLineArguments cla = new CommandLineArguments( args );
57             if ( cla.isOptionSet( HELP_OPTION_1 ) || cla.isOptionSet( HELP_OPTION_2 ) || ( args.length != 3 ) ) {
58                 printHelp();
59                 System.exit( 0 );
60             }
61             final File in = cla.getFile( 0 );
62             int from = 0;
63             int to = 0;
64             int window = 0;
65             int step = 0;
66             if ( cla.isOptionSet( FROM_OPTION ) && cla.isOptionSet( TO_OPTION ) ) {
67                 from = cla.getOptionValueAsInt( FROM_OPTION );
68                 to = cla.getOptionValueAsInt( TO_OPTION );
69             }
70             else if ( cla.isOptionSet( STEP_OPTION ) && cla.isOptionSet( WINDOW_OPTION ) ) {
71                 step = cla.getOptionValueAsInt( STEP_OPTION );
72                 window = cla.getOptionValueAsInt( WINDOW_OPTION );
73             }
74             else {
75                 printHelp();
76                 System.exit( 0 );
77             }
78             Msa msa = null;
79             msa = GeneralMsaParser.parse( new FileInputStream( in ) );
80             if ( cla.isOptionSet( FROM_OPTION ) ) {
81                 singleCalc( in, from, to, msa );
82             }
83             else {
84                 windowedCalcs( window, step, msa );
85             }
86         }
87         catch ( final Exception e ) {
88             ForesterUtil.fatalError( PRG_NAME, e.getMessage() );
89         }
90     }
91
92     private static void printHelp() {
93         ForesterUtil.printProgramInformation( PRG_NAME, PRG_DESC, PRG_VERSION, PRG_DATE, E_MAIL, WWW, ForesterUtil.getForesterLibraryInformation() );
94         System.out.println( "Usage:" );
95         System.out.println();
96         System.out.println( PRG_NAME + " <options> <msa input file>" );
97         System.out.println();
98         System.out.println( " options: " );
99         System.out.println();
100         System.out.println( "   -" + FROM_OPTION + "=<integer>: from (msa column)" );
101         System.out.println( "   -" + TO_OPTION + "=<integer>: to (msa column)" );
102         System.out.println( "    or" );
103         System.out.println( "   -" + WINDOW_OPTION + "=<integer>: window size (msa columns)" );
104         System.out.println( "   -" + STEP_OPTION + "=<integer>: step size (msa columns)" );
105         System.out.println();
106         System.out.println();
107         System.out.println();
108     }
109
110     private static void windowedCalcs( int window, int step, final Msa msa ) {
111         if ( window < 1 ) {
112             window = 1;
113         }
114         if ( step < 1 ) {
115             step = 1;
116         }
117         final double id_ratios[] = new double[ msa.getLength() ];
118         for( int i = 0; i <= msa.getLength() - 1; ++i ) {
119             id_ratios[ i ] = MsaMethods.calculateIdentityRatio( msa, i );
120         }
121         String min_pos = "";
122         String max_pos = "";
123         double min = 1;
124         double max = 0;
125         for( int i = 0; i <= msa.getLength() - 1; i += step ) {
126             int to = i + window - 1;
127             if ( to > ( msa.getLength() - 1 ) ) {
128                 to = msa.getLength() - 1;
129             }
130             final DescriptiveStatistics stats = calc( i, to, id_ratios );
131             final double mean = stats.arithmeticMean();
132             final String pos = i + "-" + to;
133             System.out.print( pos );
134             System.out.print( ":\t" );
135             System.out.print( mean );
136             if ( stats.getN() > 2 ) {
137                 System.out.print( "\t" );
138                 System.out.print( stats.median() );
139                 System.out.print( "\t" );
140                 System.out.print( stats.sampleStandardDeviation() );
141             }
142             System.out.println();
143             if ( mean > max ) {
144                 max = mean;
145                 max_pos = pos;
146             }
147             if ( mean < min ) {
148                 min = mean;
149                 min_pos = pos;
150             }
151         }
152         System.out.println( "Min: " + min_pos + ": " + min );
153         System.out.println( "Max: " + max_pos + ": " + max );
154     }
155
156     private static void singleCalc( final File in, int from, int to, final Msa msa ) {
157         if ( from < 0 ) {
158             from = 0;
159         }
160         if ( to > ( msa.getLength() - 1 ) ) {
161             to = msa.getLength() - 1;
162         }
163         final DescriptiveStatistics stats = calc( from, to, msa );
164         System.out.println( in.toString() + ": " + from + "-" + to + ":" );
165         System.out.println();
166         System.out.println( stats.toString() );
167     }
168
169     private static DescriptiveStatistics calc( final int from, final int to, final Msa msa ) {
170         final DescriptiveStatistics stats = new BasicDescriptiveStatistics();
171         for( int c = from; c <= to; ++c ) {
172             stats.addValue( MsaMethods.calculateIdentityRatio( msa, c ) );
173         }
174         return stats;
175     }
176
177     private static DescriptiveStatistics calc( final int from, final int to, final double id_ratios[] ) {
178         final DescriptiveStatistics stats = new BasicDescriptiveStatistics();
179         for( int c = from; c <= to; ++c ) {
180             stats.addValue( id_ratios[ c ] );
181         }
182         return stats;
183     }
184 }