bp_bin/mean_vals

   1 #!/usr/bin/env perl
   2
   3 # Copyright (C) 2007-2009 Martin A. Hansen.
   4
   5 # This program is free software; you can redistribute it and/or
   6 # modify it under the terms of the GNU General Public License
   7 # as published by the Free Software Foundation; either version 2
   8 # of the License, or (at your option) any later version.
   9
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14
  15 # You should have received a copy of the GNU General Public License
  16 # along with this program; if not, write to the Free Software
  17 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18
  19 # http://www.gnu.org/copyleft/gpl.html
  20
  21
  22 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> DESCRIPTION <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  23
  24 # Find the mean values in the stream for given keys.
  25
  26 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  27
  28
  29 use warnings;
  30 use strict;
  31 use Maasha::Biopieces;
  32 use Maasha::Calc;
  33
  34
  35 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
  36
  37
  38 my ( $options, $in, $out, $record, $new_record, $key, %sum_hash, %count_hash, $mean, $fh );
  39
  40 $options = Maasha::Biopieces::parse_options(
  41     [
  42         { long => 'no_stream', short => 'x', type => 'flag',   mandatory => 'no', default => undef, allowed => undef, disallowed => undef },
  43         { long => 'data_out',  short => 'o', type => 'file',   mandatory => 'no', default => undef, allowed => undef, disallowed => undef },
  44         { long => 'keys',      short => 'k', type => 'list',   mandatory => 'no', default => undef, allowed => undef, disallowed => undef },
  45         { long => 'list',      short => 'l', type => 'string', mandatory => 'no', default => undef, allowed => undef, disallowed => undef },
  46     ]
  47 );
  48
  49 Maasha::Common::error( qq(both --keys and --list specified) ) if     $options->{ "keys" } and     $options->{ "list" };
  50 Maasha::Common::error( qq(no --keys or --list specified) )    if not $options->{ "keys" } and not $options->{ "list" };
  51
  52 $in  = Maasha::Biopieces::read_stream( $options->{ "stream_in" } );
  53 $out = Maasha::Biopieces::write_stream( $options->{ "stream_out" } );
  54
  55 while ( $record = Maasha::Biopieces::get_record( $in ) )
  56 {
  57     foreach $key ( @{ $options->{ "keys" } } )
  58     {
  59         if ( $record->{ $key } )
  60         {
  61             $sum_hash{ $key } += $record->{ $key };
  62             $count_hash{ $key }++;
  63         }
  64     }
  65
  66     if ( $options->{ 'list' } and $record->{ $options->{ 'list' } } ) {
  67         $record->{ $options->{ 'list' } . "_MEAN" } = sprintf( "%.2f", Maasha::Calc::mean( [ split ";", $record->{ $options->{ 'list' } } ] ) );
  68     }
  69
  70     Maasha::Biopieces::put_record( $record, $out ) if not $options->{ "no_stream" };
  71 }
  72
  73 Maasha::Biopieces::close_stream( $in );
  74 Maasha::Biopieces::close_stream( $out );
  75
  76 $fh = Maasha::Biopieces::write_stream( $options->{ "data_out" } );
  77
  78 foreach $key ( @{ $options->{ "keys" } } )
  79 {
  80     if ( exists $count_hash{ $key } ) {
  81         $mean = sprintf( "%.2f", ( $sum_hash{ $key } / $count_hash{ $key } ) );
  82     } else {
  83         $mean = "N/A";
  84     }
  85
  86     $new_record->{ $key . "_MEAN" } = $mean
  87 }
  88
  89 if ( $options->{ "keys" } and $new_record )
  90 {
  91     $new_record->{ 'REC_TYPE' } = "MEAN";
  92
  93     Maasha::Biopieces::put_record( $new_record, $fh );
  94 }
  95
  96 close $fh;
  97
  98
  99 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
 100
 101
 102 BEGIN
 103 {
 104     Maasha::Biopieces::status_set();
 105 }
 106
 107
 108 END
 109 {
 110     Maasha::Biopieces::status_log();
 111 }
 112
 113
 114 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
 115
 116
 117 __END__