changed layout of ruby source

[biopieces.git] / bp_bin / read_fastq

diff --git a/bp_bin/read_fastq b/bp_bin/read_fastq
index df2ad6e088d786fa12bce015627d0ce89a3a250e..063edd6ce2883bbe5b46f286c59175c71cf55456 100755 (executable)
--- a/bp_bin/read_fastq
+++ b/bp_bin/read_fastq
@@ -1,6 +1,6 @@
-#!/usr/bin/env perl
+#!/usr/bin/env ruby
 
-# Copyright (C) 2007-2009 Martin A. Hansen.
+# Copyright (C) 2007-2010 Martin A. Hansen.
 
 # This program is free software; you can redistribute it and/or
 # modify it under the terms of the GNU General Public License
@@ -18,88 +18,56 @@
 
 # http://www.gnu.org/copyleft/gpl.html
 
-
-# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> DESCRIPTION <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
-
-# Read FASTQ entries from one or more files.
-
 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
 
+# This program is part of the Biopieces framework (www.biopieces.org).
 
-use warnings;
-use strict;
-use Maasha::Biopieces;
-use Maasha::Filesys;
-use Maasha::Fastq;
+# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>> DESCRIPTION <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
 
+# Read FASTQ entries from one or more files.
 
 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
 
+require 'maasha/biopieces'
+require 'maasha/fastq'
 
-my ( $options, $in, $out, $record, $data_in, $num, $entry );
-
-$options = Maasha::Biopieces::parse_options(
-    [
-        { long => 'data_in',     short => 'i', type => 'files!', mandatory => 'no', default => undef, allowed => undef, disallowed => undef },
-        { long => 'num',         short => 'n', type => 'uint',   mandatory => 'no', default => undef, allowed => undef, disallowed => 0     },
-        { long => 'quality',     short => 'q', type => 'uint',   mandatory => 'no', default => 20,    allowed => undef, disallowed => undef },
-        { long => 'skip_scores', short => 's', type => 'flag',   mandatory => 'no', default => undef, allowed => undef, disallowed => undef },
-    ]   
-);
-
-$in  = Maasha::Biopieces::read_stream( $options->{ "stream_in" } );
-$out = Maasha::Biopieces::write_stream( $options->{ "stream_out" } );
+casts = []
+casts << {:long=>'data_in', :short=>'i', :type=>'files!', :mandatory=>false, :default=>nil, :allowed=>nil, :disallowed=>nil}
+casts << {:long=>'num',     :short=>'n', :type=>'uint',   :mandatory=>false, :default=>nil, :allowed=>nil, :disallowed=>'0'}
+casts << {:long=>'solexa',  :short=>'s', :type=>'flag',   :mandatory=>false, :default=>nil, :allowed=>nil, :disallowed=>nil}
 
-while ( $record = Maasha::Biopieces::get_record( $in ) ) {
-    Maasha::Biopieces::put_record( $record, $out );
-}
+PHRED_SCORES = Regexp.new('[!"#$%&\'()*+,-./0123456789:]')
 
-if ( $options->{ 'data_in' } )
-{
-    $data_in = Maasha::Filesys::files_read_open( $options->{ 'data_in' } );
+bp = Biopieces.new
 
-    $num = 1;
+options = bp.parse(ARGV, casts)
 
-    while ( $entry = Maasha::Fastq::get_entry( $data_in ) ) 
-    {
-        if ( $record = Maasha::Fastq::fastq2biopiece( $entry ) )
-        {
-            if ( not $options->{ 'skip_scores' } )
-            {
-                Maasha::Fastq::lowercase_low_scores( $record->{ 'SEQ' }, $record->{ 'SCORES' }, $options->{ 'quality' } );
-
-                $record->{ 'SCORES' }     =~ s/(.)/ord( $1 ) - 33 . ";"/ge; # http://maq.sourceforge.net/fastq.shtml
-                $record->{ 'SCORE_MEAN' } = sprintf( "%.2f", Maasha::Calc::mean( [ split /;/, $record->{ 'SCORES' } ] ) );
-            }
-
-            Maasha::Biopieces::put_record( $record, $out );
-        }
-
-        last if $options->{ "num" } and $num == $options->{ "num" };
-
-        $num++;
-    }
-
-    close $data_in;
-}
-
-Maasha::Biopieces::close_stream( $in );
-Maasha::Biopieces::close_stream( $out );
-
-
-# >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
+bp.each_record do |record|
+  bp.puts record
+end
 
+num  = 0
+last = false
 
-BEGIN
-{
-    Maasha::Biopieces::status_set();
-}
+if options.has_key? :data_in
+  options[:data_in].each do |file|
+    Fastq.open(file, mode='r') do |fastq|
+      fastq.each do |entry|
+        entry.convert_phred2illumina!  if entry.qual.match PHRED_SCORES
+        entry.convert_solexa2illumina! if options[:solexa]
+        bp.puts entry.to_bp
+        num += 1
 
+        if options.has_key? :num and options[:num] == num
+          last = true
+          break
+        end
+      end
+    end
 
-END
-{
-    Maasha::Biopieces::status_log();
-}
+    break if last
+  end
+end
 
 
 # >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>><<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<