]> git.donarmstrong.com Git - biopieces.git/blobdiff - code_ruby/lib/maasha/seq/backtrack.rb
fixed unit tests
[biopieces.git] / code_ruby / lib / maasha / seq / backtrack.rb
index 088b596a1e4e1e52784a5bdfeb67920d1fe2ea48..c216ec979e0c1f15e092a1914352baa32583ff80 100644 (file)
@@ -41,39 +41,79 @@ module BackTrack
   MAX_DEL    = 5 # Maximum number of deletions allowed
 
   # ------------------------------------------------------------------------------
-  #   str.scan(pattern[, max_mismatches[, max_insertions[, max_deletions]]])
+  #   str.patmatch(pattern[, start|[start, stop][, max_mis[, max_ins[, max_del]]]])
+  #   -> Match
+  #   str.patmatch(pattern[, start|[start, stop][, max_mis[, max_ins[, max_del]]]]) { |match|
+  #     block
+  #   }
+  #   -> Match
+  #
+  # ------------------------------------------------------------------------------
+  # Method to iterate through a sequence from a given start position to the end of
+  # the sequence or to a given stop position to locate a pattern allowing for a
+  # maximum number of mismatches, insertions, and deletions. Insertions are
+  # nucleotides found in the pattern but not in the sequence. Deletions are
+  # nucleotides found in the sequence but not in the pattern.
+  def patmatch(pattern, start = 0, max_mismatches = 0, max_insertions = 0, max_deletions = 0)
+    self.patscan(pattern, start, max_mismatches, max_insertions, max_deletions) do |m|
+      if block_given?
+        yield m
+      else
+        return m
+      end
+
+      break
+    end
+
+    if block_given?
+      yield nil
+    else
+      return nil
+    end
+  end
+
+  # ------------------------------------------------------------------------------
+  #   str.patscan(pattern[, start|[start, stop][, max_mis[, max_ins[, max_del]]]])
   #   -> Array
-  #   str.scan(pattern[, max_mismatches[, max_insertions[, max_deletions]]]) { |match|
+  #   str.patscan(pattern[, start|[start, stop][, max_mis[, max_ins[, max_del]]]]) { |match|
   #     block
   #   }
   #   -> Match
   #
   # ------------------------------------------------------------------------------
-  # Method to iterate through a sequence to locate pattern matches starting at a
-  # given offset allowing for a maximum number of mismatches, insertions, and
-  # deletions. Insertions are nucleotides found in the pattern but not in the
-  # sequence. Deletions are nucleotides found in the sequence but not in the
-  # pattern. Matches found in block context return the Match object. Otherwise
-  # matches are returned in an Array of Match objects.
-  def patscan(pattern, offset = 0, max_mismatches = 0, max_insertions = 0, max_deletions = 0)
+  # Method to iterate through a sequence from a given start position to the end of
+  # the sequence or to a given stop position to locate a pattern allowing for a
+  # maximum number of mismatches, insertions, and deletions. Insertions are
+  # nucleotides found in the pattern but not in the sequence. Deletions are
+  # nucleotides found in the sequence but not in the pattern. Matches found in
+  # block context return the Match object. Otherwise matches are returned in an
+  # Array of Match objects.
+  def patscan(pattern, start = 0, max_mismatches = 0, max_insertions = 0, max_deletions = 0)
+    if start.is_a? Array
+      start, stop = *start
+    else
+      stop = self.length - 1
+    end
+
     raise BackTrackError, "Bad pattern: #{pattern}"                                          unless pattern.downcase =~ OK_PATTERN
-    raise BackTrackError, "offset: #{offset} out of range (0 ... #{self.length})"            unless (0 ... self.length).include? offset
+    raise BackTrackError, "start: #{start} out of range (0 .. #{self.length - 1})"           unless (0 ... self.length).include? start
+    raise BackTrackError, "stop: #{stop} out of range (0 .. #{self.length - 1})"             unless (0 ... self.length).include? stop
     raise BackTrackError, "max_mismatches: #{max_mismatches} out of range (0 .. #{MAX_MIS})" unless (0 .. MAX_MIS).include? max_mismatches
     raise BackTrackError, "max_insertions: #{max_insertions} out of range (0 .. #{MAX_INS})" unless (0 .. MAX_INS).include? max_insertions
     raise BackTrackError, "max_deletions: #{max_deletions} out of range (0 .. #{MAX_DEL})"   unless (0 .. MAX_DEL).include? max_deletions
 
     matches = []
 
-    if block_given?
-      while result = track_C(self.seq, self.length, pattern, offset, max_mismatches, max_insertions, max_deletions)
-        yield Match.new(result.first, result.last, self.seq[result.first ... result.first + result.last])
-        offset = result.first + 1
-      end
-    else
-      while result = track_C(self.seq, self.length, pattern, offset, max_mismatches, max_insertions, max_deletions)
-        matches << Match.new(result.first, result.last, self.seq[result.first ... result.first + result.last])
-        offset = result.first + 1
+    while result = scan_C(self.seq, pattern, start, stop, max_mismatches, max_insertions, max_deletions)
+      match = Match.new(result.first, result.last, self.seq[result.first ... result.first + result.last])
+
+      if block_given?
+        yield match
+      else
+        matches << match
       end
+
+      start = result.first + 1
     end
 
     return matches.empty? ? nil : matches unless block_given?
@@ -82,8 +122,6 @@ module BackTrack
   private
 
   inline do |builder|
-    builder.add_static "id_seq", 'rb_intern("@seq")', "ID"
-
     # Macro for matching nucleotides including ambiguity codes.
     builder.prefix %{
       #define MATCH(A,B) ((bitmap[A] & bitmap[B]) != 0)
@@ -115,15 +153,17 @@ module BackTrack
 
     # Backtrack algorithm for matching a pattern (p) starting in a sequence (s) allowing for mis
     # mismatches, ins insertions and del deletions. ss is the start of the sequence, used only for
-    # reporting the match endpoints.
+    # reporting the match endpoints. State is used to avoid ins followed by del and visa versa which
+    # are nonsense.
     builder.prefix %{
       unsigned int backtrack(
-        char *ss,   // Sequence start
-        char *s,    // Sequence
-        char *p,    // Pattern
-        int   mis,  // Max mismatches
-        int   ins,  // Max insertions
-        int   del   // Max deletions
+        char         *ss,     // Sequence start
+        char         *s,      // Sequence
+        char         *p,      // Pattern
+        unsigned int  mis,    // Max mismatches
+        unsigned int  ins,    // Max insertions
+        unsigned int  del,    // Max deletions
+        int           state   // Last event: mis, ins or del
       )
       {
         unsigned int r = 0;
@@ -134,9 +174,9 @@ module BackTrack
           return (unsigned int) (s - ss);
         else
         {
-          if (mis && *s && *p && (r = backtrack(ss, s + 1, p + 1, mis - 1, ins, del))) return r;
-          if (ins && *p &&       (r = backtrack(ss, s, p + 1, mis, ins - 1, del)))     return r;
-          if (del && *s &&       (r = backtrack(ss, s + 1, p, mis, ins, del - 1)))     return r;
+          if (mis && *s && *p &&            (r = backtrack(ss, s + 1, p + 1, mis - 1, ins, del, 0))) return r;
+          if (ins && *p && (state != -1) && (r = backtrack(ss, s, p + 1, mis, ins - 1, del, 1)))     return r;
+          if (del && *s && (state != 1)  && (r = backtrack(ss, s + 1, p, mis, ins, del - 1, -1)))    return r;
         }
 
         return 0;
@@ -146,35 +186,36 @@ module BackTrack
     # Find pattern (p) in a sequence (s) starting at pos, with at most mis mismatches, ins
     # insertions and del deletions.
     builder.c %{
-      VALUE track_C(
-        VALUE _s,     // Sequence
-        VALUE _len,   // Sequence length
-        VALUE _p,     // Pattern
-        VALUE _pos,   // Start position
-        VALUE _mis,   // Maximum mismatches
-        VALUE _ins,   // Maximum insertions
-        VALUE _del    // Maximum deletions
+      VALUE scan_C(
+        VALUE _s,       // Sequence
+        VALUE _p,       // Pattern
+        VALUE _start,   // Search postition start
+        VALUE _stop,    // Search position stop
+        VALUE _mis,     // Maximum mismatches
+        VALUE _ins,     // Maximum insertions
+        VALUE _del      // Maximum deletions
       )
       {
-        char         *s   = StringValuePtr(_s);
-        char         *p   = StringValuePtr(_p);
-        unsigned int  len = FIX2UINT(_len);
-        unsigned int  pos = FIX2UINT(_pos);
-        unsigned int  mis = FIX2UINT(_mis);
-        unsigned int  ins = FIX2UINT(_ins);
-        unsigned int  del = FIX2UINT(_del);
-
-        char         *ss = s;
-        unsigned int  e  = 0;
+        char         *s     = StringValuePtr(_s);
+        char         *p     = StringValuePtr(_p);
+        unsigned int  start = FIX2UINT(_start);
+        unsigned int  stop  = FIX2UINT(_stop);
+        unsigned int  mis   = FIX2UINT(_mis);
+        unsigned int  ins   = FIX2UINT(_ins);
+        unsigned int  del   = FIX2UINT(_del);
+
+        char         *ss    = s;
+        int           state = 0;
+        unsigned int  e     = 0;
         VALUE         tuple;
 
-        if (pos < len)
+        if (start <= stop)
         {
-          s += pos;
+          s += start;
 
           while (*s)
           {
-            if ((e = backtrack(ss, s, p, mis, ins, del)))
+            if ((e = backtrack(ss, s, p, mis, ins, del, state)))
             {
               tuple = rb_ary_new();
               rb_ary_push(tuple, INT2FIX((int) (s - ss)));