]> git.donarmstrong.com Git - biopieces.git/blobdiff - code_ruby/lib/maasha/seq.rb
removed debug message
[biopieces.git] / code_ruby / lib / maasha / seq.rb
index 1d8f3c00594e9208f434251eb23c342413df9506..7377da533d50ea34fb4dc12c3e2bfb7f05a7e944 100644 (file)
@@ -93,7 +93,7 @@ class Seq
     type     = record[:SEQ_TYPE].to_sym if record[:SEQ_TYPE]
     qual     = record[:SCORES]
 
-    self.new(seq_name, seq, type, qual)
+    self.new(seq_name: seq_name, seq: seq, type: type, qual: qual)
   end
 
   # Class method that generates all possible oligos of a specifed length and type.
@@ -125,16 +125,20 @@ class Seq
     oligos
   end
 
-  # Initialize a sequence object with the following arguments:
-  # - seq_name: Name of the sequence.
-  # - seq: The sequence.
-  # - type: The sequence type - DNA, RNA, or protein
-  # - qual: An Illumina type quality scores string.
-  def initialize(seq_name = nil, seq = nil, type = nil, qual = nil)
-    @seq_name = seq_name
-    @seq      = seq
-    @type     = type
-    @qual     = qual
+  # Initialize a sequence object with the following options:
+  # - :seq_name   Name of the sequence.
+  # - :seq        The sequence.
+  # - :type       The sequence type - DNA, RNA, or protein
+  # - :qual       An Illumina type quality scores string.
+  def initialize(options = {})
+    @seq_name = options[:seq_name]
+    @seq      = options[:seq]
+    @type     = options[:type]
+    @qual     = options[:qual]
+
+    if @seq and @qual and @seq.length != @qual.length
+      raise SeqError, "Sequence length and score length mismatch: #{@seq.length} != #{@qual.length}"
+    end
   end
 
   # Method that guesses and returns the sequence type
@@ -290,8 +294,8 @@ class Seq
     raise SeqError, "Missing seq_name" if self.seq_name.nil? or self.seq_name == ''
     raise SeqError, "Missing seq"      if self.seq.nil?      or self.seq.empty?
 
-    seq_name = self.seq_name.to_s
-    seq      = self.seq.to_s
+    seq_name = self.seq_name
+    seq      = self.seq.dup
 
     unless wrap.nil?
       seq.gsub!(/(.{#{wrap}})/) do |match|
@@ -301,7 +305,7 @@ class Seq
       seq.chomp!
     end
 
-    ">" + seq_name + $/ + seq + $/
+    ">" + seq_name.to_s + $/ + seq + $/
   end
 
   # Method that given a Seq entry returns a FASTQ entry (a string).
@@ -339,7 +343,14 @@ class Seq
 
   # Method to reverse the sequence.
   def reverse
-    Seq.new(self.seq_name, self.seq.reverse, self.type, self.qual ? self.qual.reverse : self.qual)
+    entry = Seq.new(
+      seq_name: self.seq_name,
+      seq:      self.seq.reverse,
+      type:     self.type,
+      qual:     (self.qual ? self.qual.reverse : self.qual)
+    )
+
+    entry
   end
 
   # Method to reverse the sequence.
@@ -353,10 +364,11 @@ class Seq
   def complement
     raise SeqError, "Cannot complement 0 length sequence" if self.length == 0
 
-    entry          = Seq.new
-    entry.seq_name = self.seq_name
-    entry.type     = self.type
-    entry.qual     = self.qual
+    entry = Seq.new(
+      seq_name: self.seq_name,
+      type:     self.type,
+      qual:     self.qual
+    )
 
     if self.is_dna?
       entry.seq = self.seq.tr('AGCUTRYWSMKHDVBNagcutrywsmkhdvbn', 'TCGAAYRWSKMDHBVNtcgaayrwskmdhbvn')
@@ -420,7 +432,12 @@ class Seq
 
   # Method to return a new Seq object with shuffled sequence.
   def shuffle
-    Seq.new(self.seq_name, self.seq.split('').shuffle!.join, self.type, self.qual)
+    Seq.new(
+      seq_name: self.seq_name,
+      seq:      self.seq.split('').shuffle!.join,
+      type:     self.type,
+      qual:     self.qual
+    )
   end
 
   # Method to shuffle a sequence randomly inline.
@@ -452,7 +469,7 @@ class Seq
   # Index method for Seq objects.
   def [](*args)
     entry = Seq.new
-    entry.seq_name = self.seq_name
+    entry.seq_name = self.seq_name.dup if self.seq_name
     entry.seq      = self.seq[*args]
     entry.type     = self.type
     entry.qual     = self.qual[*args] unless self.qual.nil?
@@ -468,41 +485,6 @@ class Seq
     self
   end
 
-  # Method that returns a subsequence of from a given start position
-  # and of a given length.
-  def subseq(start, length = self.length - start)
-    raise SeqError, "subsequence start: #{start} < 0"                                                if start  < 0
-    raise SeqError, "subsequence length: #{length} < 0"                                              if length < 0
-    raise SeqError, "subsequence start + length > Seq.length: #{start} + #{length} > #{self.length}" if start + length > self.length
-
-    if length == 0
-      seq  = ""
-      qual = "" unless self.qual.nil?
-    else
-      stop = start + length - 1
-
-      seq  = self.seq[start .. stop]
-      qual = self.qual[start .. stop] unless self.qual.nil?
-    end
-
-    seq_name = self.seq_name.nil? ? nil : self.seq_name.dup
-
-    Seq.new(seq_name, seq, self.type, qual)
-  end
-
-  # Method that replaces a sequence with a subsequence from a given start position
-  # and of a given length.
-  def subseq!(start, length = self.length - start)
-    s = subseq(start, length)
-
-    self.seq_name = s.seq_name
-    self.seq      = s.seq
-    self.type     = s.type
-    self.qual     = s.qual
-
-    self
-  end
-
   # Method that returns a subsequence of a given length
   # beginning at a random position.
   def subseq_rand(length)
@@ -512,7 +494,7 @@ class Seq
       start = rand(self.length - length + 1)
     end
 
-    self.subseq(start, length)
+    self[start .. start + length]
   end
 
   # Method that returns the residue compositions of a sequence in
@@ -652,22 +634,26 @@ class Seq
     regex_start = Regexp.new(start_codons.join('|'), true)
     regex_stop  = Regexp.new(stop_codons.join('|'), true)
 
-    while pos_beg and pos_beg < self.length - size_min
-      if pos_beg = self.seq.index(regex_start, pos_beg)
-        if pos_end = self.seq.index(regex_stop, pos_beg)
-          length = (pos_end - pos_beg) + 3
+    while pos_beg = self.seq.index(regex_start, pos_beg)
+      pos_end = pos_beg + 3
+       
+      while pos_end = self.seq.index(regex_stop, pos_end)
+        length = (pos_end - pos_beg) + 3
 
-          if (length % 3) == 0
-            if size_min <= length and length <= size_max
-              subseq = self.subseq(pos_beg, length)
+        if (length % 3) == 0
+          if size_min <= length and length <= size_max
+            subseq = self[pos_beg ... pos_beg + length]
 
-              orfs << [subseq, pos_beg, pos_end + 3]
-            end
+            orfs << [subseq, pos_beg, pos_end + 3]
           end
+
+          break
         end
 
-        pos_beg += 1
+        pos_end += 1
       end
+
+      pos_beg += 1
     end
 
     if pick_longest