2
|
1 #!/usr/bin/env python
|
|
2 """A simple wrapper script to call MIRA4's mirabait and collect its output.
|
|
3 """
|
4
|
4
|
|
5 from __future__ import print_function
|
|
6
|
2
|
7 import os
|
4
|
8 import shutil
|
|
9 import subprocess
|
2
|
10 import sys
|
|
11 import time
|
|
12
|
4
|
13 WRAPPER_VER = "0.0.11" # Keep in sync with the XML file
|
2
|
14
|
|
15
|
|
16 def get_version(mira_binary):
|
|
17 """Run MIRA to find its version number"""
|
|
18 # At the commend line I would use: mira -v | head -n 1
|
|
19 # however there is some pipe error when doing that here.
|
|
20 cmd = [mira_binary, "-v"]
|
|
21 try:
|
4
|
22 child = subprocess.Popen(cmd, universal_newlines=True,
|
2
|
23 stdout=subprocess.PIPE,
|
|
24 stderr=subprocess.STDOUT)
|
4
|
25 except Exception as err:
|
2
|
26 sys.stderr.write("Error invoking command:\n%s\n\n%s\n" % (" ".join(cmd), err))
|
|
27 sys.exit(1)
|
|
28 ver, tmp = child.communicate()
|
|
29 del child
|
4
|
30 # Workaround for -v not working in mirabait 4.0RC4
|
2
|
31 if "invalid option" in ver.split("\n", 1)[0]:
|
|
32 for line in ver.split("\n", 1):
|
|
33 if " version " in line:
|
|
34 line = line.split()
|
4
|
35 return line[line.index("version") + 1].rstrip(")")
|
|
36 sys.exit("Could not determine MIRA version:\n%s" % ver)
|
2
|
37 return ver.split("\n", 1)[0]
|
|
38
|
4
|
39
|
|
40 if "MIRA4" in os.environ:
|
2
|
41 mira_path = os.environ["MIRA4"]
|
4
|
42 mira_binary = os.path.join(mira_path, "mirabait")
|
|
43 if not os.path.isfile(mira_binary):
|
|
44 sys.exit("Missing mirabait under $MIRA4, %r\nFolder contained: %s"
|
|
45 % (mira_binary, ", ".join(os.listdir(mira_path))))
|
|
46 else:
|
|
47 sys.stderr.write("DEBUG: Since $MIRA4 is not set, assuming mira binaries are on $PATH.\n")
|
|
48 mira_path = None
|
|
49 mira_binary = "mirabait"
|
|
50
|
2
|
51 mira_ver = get_version(mira_binary)
|
|
52 if not mira_ver.strip().startswith("4.0"):
|
4
|
53 sys.exit("This wrapper is for MIRA V4.0, not:\n%s" % mira_ver)
|
2
|
54 if "-v" in sys.argv or "--version" in sys.argv:
|
4
|
55 print("%s, MIRA wrapper version %s" % (mira_ver, WRAPPER_VER))
|
2
|
56 sys.exit(0)
|
|
57
|
|
58
|
|
59 format, output_choice, strand_choice, kmer_length, min_occurance, bait_file, in_file, out_file = sys.argv[1:]
|
|
60
|
|
61 if format.startswith("fastq"):
|
|
62 format = "fastq"
|
|
63 elif format == "mira":
|
|
64 format = "maf"
|
|
65 elif format != "fasta":
|
4
|
66 sys.exit("Was not expected format %r" % format)
|
2
|
67
|
|
68 assert out_file.endswith(".dat")
|
|
69 out_file_stem = out_file[:-4]
|
|
70
|
|
71 cmd_list = [mira_binary, "-f", format, "-t", format,
|
|
72 "-k", kmer_length, "-n", min_occurance,
|
|
73 bait_file, in_file, out_file_stem]
|
|
74 if output_choice == "pos":
|
|
75 pass
|
|
76 elif output_choice == "neg":
|
4
|
77 # Invert the selection...
|
2
|
78 cmd_list.insert(1, "-i")
|
|
79 else:
|
4
|
80 sys.exit("Output choice should be 'pos' or 'neg', not %r" % output_choice)
|
2
|
81 if strand_choice == "both":
|
|
82 pass
|
|
83 elif strand_choice == "fwd":
|
4
|
84 # Ingore reverse strand...
|
2
|
85 cmd_list.insert(1, "-r")
|
|
86 else:
|
4
|
87 sys.exit("Strand choice should be 'both' or 'fwd', not %r" % strand_choice)
|
2
|
88
|
|
89 cmd = " ".join(cmd_list)
|
4
|
90 # print cmd
|
2
|
91 start_time = time.time()
|
|
92 try:
|
4
|
93 # Run MIRA
|
|
94 child = subprocess.Popen(cmd_list, universal_newlines=True,
|
2
|
95 stdout=subprocess.PIPE,
|
|
96 stderr=subprocess.STDOUT)
|
4
|
97 except Exception as err:
|
2
|
98 sys.stderr.write("Error invoking command:\n%s\n\n%s\n" % (cmd, err))
|
|
99 sys.exit(1)
|
4
|
100 # Use .communicate as can get deadlocks with .wait(),
|
2
|
101 stdout, stderr = child.communicate()
|
4
|
102 assert stderr is None # Due to way we ran with subprocess
|
2
|
103 run_time = time.time() - start_time
|
|
104 return_code = child.returncode
|
4
|
105 print("mirabait took %0.2f minutes" % (run_time / 60.0))
|
2
|
106
|
|
107 if return_code:
|
|
108 sys.stderr.write(stdout)
|
4
|
109 sys.exit("Return error code %i from command:\n%s" % (return_code, cmd),
|
2
|
110 return_code)
|
|
111
|
4
|
112 # Capture output
|
2
|
113 out_tmp = out_file_stem + "." + format
|
|
114 if not os.path.isfile(out_tmp):
|
|
115 sys.stderr.write(stdout)
|
4
|
116 sys.exit("Missing output file from mirabait: %s" % out_tmp)
|
2
|
117 shutil.move(out_tmp, out_file)
|