comparison experiments/scripts/cnbh-syllables/run_training_and_testing/randomize_snrs.py @ 144:f8ace1ee8782

- Support for training on random SNRs

+#!/usr/bin/env python
+# encoding: utf-8
+"""
+randomize_snrs.py
+
+Created by Thomas Walters on 2010-11-06.
+"""
+
+import sys
+import getopt
+import re
+import random
+
+help_message = '''
+Replace a string in each of two files with a string randomly selected from a list.
+'''
+
+replacement_strings = ('snr_0dB',
+                       'snr_3dB',
+                       'snr_6dB',
+                       'snr_9dB',
+                       'snr_12dB',
+                       'snr_15dB',
+                       'snr_18dB',
+                       'snr_21dB',
+                       'snr_24dB',
+                       'snr_27dB',
+                       'snr_30dB',
+                       'snr_33dB',
+                       'snr_36dB',
+                       'snr_39dB',
+                       'snr_42dB',
+                       'snr_45dB',
+                       'clean')
+
+class Usage(Exception):
+  def __init__(self, msg):
+    self.msg = msg
+
+
+def main(argv=None):
+  if argv is None:
+    argv = sys.argv
+  try:
+    try:
+      opts, args = getopt.getopt(argv[1:], "hs:f:m:o:p:", ["help", "string_to_replace=", "input_script_file=", "input_mlf=", "output_script_file=", "output_mlf="])
+    except getopt.error, msg:
+      raise Usage(msg)
+  
+    # option processing
+    string_to_replace = ""
+    script_file = ""
+    ml_file = ""
+    output_script_file = ""
+    output_ml_file = ""
+    for option, value in opts:
+      if option in ("-h", "--help"):
+        raise Usage(help_message)
+      if option in ("-s", "--string_to_replace"):
+        string_to_replace = value
+      if option in ("-f", "--input_script_file"):
+        script_file = value
+      if option in ("-o", "--outut_script_file"):
+        output_script_file = value
+      if option in ("-p", "--output_mlf"):
+        output_ml_file = value
+      if option in ("-m", "--input_mlf"):
+        ml_file = value
+      
+  except Usage, err:
+    print >> sys.stderr, sys.argv[0].split("/")[-1] + ": " + str(err.msg)
+    print >> sys.stderr, "\t for help use --help"
+    return 2
+  
+  script = open(script_file, 'r')
+  mlf = open(ml_file, 'r')
+  out_script = open(output_script_file, 'w')
+  out_mlf = open(output_ml_file, 'w')
+  out_mlf.write(mlf.readline())
+  for l in script:
+    replacement = random.choice(replacement_strings)
+    out_script.write(re.sub(string_to_replace, replacement, l))
+    out_mlf.write(re.sub(string_to_replace, replacement, mlf.readline()))
+    out_mlf.write(mlf.readline())
+    out_mlf.write(mlf.readline())
+    out_mlf.write(mlf.readline())
+    out_mlf.write(mlf.readline())
+    
+
+
+if __name__ == "__main__":
+  sys.exit(main())