diff find_in_reference.py @ 2:c4fd2ea4f988

Add the option to test the reversed sequence and the DNA reverse complement of the sequence (ignored if the sequence cannot be interpreted as DNA)
author Jim Johnson <jj@umn.edu>
date Thu, 13 Nov 2014 14:09:50 -0600
parents e83e0ce8fb68
children 2429b413d90a
line wrap: on
line diff
--- a/find_in_reference.py	Wed Aug 13 15:01:33 2014 -0500
+++ b/find_in_reference.py	Thu Nov 13 14:09:50 2014 -0600
@@ -42,12 +42,28 @@
   parser.add_option('-C','--reference_column', dest='reference_column', default=None, help='The column for the value in the reference file. (first column = 1, default to last column)')
   parser.add_option( '-I', '--case_insensitive', dest='ignore_case', action="store_true", default=False, help='case insensitive' )
   parser.add_option( '-R', '--reverse_find', dest='reverse_find', action="store_true", default=False, help='find the reference string in the input string' )
+  parser.add_option( '-B', '--test_reverse', dest='test_reverse', action="store_true", default=False, help='Also search for reversed input string in reference' )
+  parser.add_option( '-D', '--test_dna_reverse_complement', dest='test_reverse_comp', action="store_true", default=False, help='Also search for the DNA reverse complement of input string' )
   parser.add_option( '-k', '--keep', dest='keep', action="store_true", default=False, help='' )
   parser.add_option( '-a', '--annotation_columns', dest='annotation_columns', default=None, help='If string is found, add these columns from reference' )
   parser.add_option( '-s', '--annotation_separator', dest='annotation_separator', default=';', help='separator character between annotations from different lines' )
   parser.add_option( '-S', '--annotation_col_sep', dest='annotation_col_sep', default=',', help='separator character between annotation column from the same line' )
   parser.add_option( '-d', '--debug', dest='debug', action='store_true', default=False, help='Turn on wrapper debugging to stdout'  )
   (options, args) = parser.parse_args()
+
+  revcompl = lambda x: ''.join([{'A':'T','C':'G','G':'C','T':'A','a':'t','c':'g','g':'c','t':'a','N':'N','n':'n'}[B] for B in x][::-1])
+  def test_rcomplement(seq, target):
+    if options.test_reverse_comp:
+      try:
+        comp = revcompl(seq)
+        return comp in target
+      except:
+        pass
+    return False
+
+  def test_reverse(seq,target):
+    return options.test_reverse and seq and seq[::-1] in target
+  
   # Input files
   if options.input != None:
     try:
@@ -116,7 +132,7 @@
         target = target_string if not options.reverse_find else search_string
         if options.debug: 
           print >> sys.stderr, "in: %s %s %s" % (search,search in target,target)
-        if search in target:
+        if search in target or test_reverse(search,target) or test_rcomplement(search,target):
           found = True
           if annotate:
             annotation = options.annotation_col_sep.join([fields[i] for i in annotation_columns])