-
Notifications
You must be signed in to change notification settings - Fork 0
/
combine_rtscore_timscore.py
34 lines (22 loc) · 1.02 KB
/
combine_rtscore_timscore.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
import argparse
from src.senpy.sqt import parser as sqt_parser
def parse_args():
# Parse Arguments
_parser = argparse.ArgumentParser(description='Arguments for Ms2 Extractor')
_parser.add_argument('--sqt_rt', required=False, type=str,
help='path to sqt file')
_parser.add_argument('--sqt_tims', required=False, type=str,
help='path to sqt file')
# add the command line args from the stream-engine
return _parser.parse_args()
def combine_sqt(sqt_rt, sqt_tims):
_, s_lines_rt = sqt_parser.read_file(sqt_rt)
_, s_lines_tims = sqt_parser.read_file(sqt_tims)
for s_line_rt, s_line_tim in zip(s_lines_rt, s_lines_tims):
for m_line_rt, m_line_tim in zip(s_line_rt.m_lines, s_line_tim.m_lines):
m_line_tim.predicted_ook0 = m_line_rt.tims_score
sqt_parser.write_file(_, s_lines_tims, sqt_tims + ".comb", version="v2.1.0_ext")
if __name__ == '__main__':
args = parse_args()
print(args)
combine_sqt(args.sqt_rt, args.sqt_tims)