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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#!/usr/bin/env python
"""
Parser for RNAPlot.
"""
from cogent.parse.record_finder import LabeledRecordFinder
__author__ = "Jeremy Widmann"
__copyright__ = "Copyright 2007-2009 2007, The Cogent Project"
__credits__ = ["Jeremy Widmann"]
__license__ = "GPL"
__version__ = "1.4.1"
__maintainer__ = "Jeremy Widmann"
__email__ = "jeremy.widmann@colorado.edu"
__status__ = "Development"
def get_sequence(lines):
"""Returns sequence string.
"""
seq_line = ''.join(lines)
seq = seq_line.split('(')[1].split(') def')[0].strip('\n\\')
return seq
def get_coordinates(lines):
"""Returns list of coordinates.
"""
coords = []
for l in lines:
if l.startswith('] def'):
break
elif l.startswith('['):
coords.append(map(float,l.strip('\n[]').split()))
return coords
def get_pairs(lines):
"""Returns list of pairing indices.
"""
pairs = []
for l in lines:
if l.startswith('] def'):
break
elif l.startswith('['):
first,second = l.strip('\n[]').split()
pairs.append([int(first)-1,int(second)-1])
return pairs
def RnaPlotParser(lines):
"""Returns sequence, coordinates, and pairing indices.
"""
#Split on sequence block
sequence_finder = LabeledRecordFinder(is_label_line=\
lambda x: x.startswith('/sequence'))
prefix, seq_block = list(sequence_finder(lines))
#split on coordinate block
coordinate_finder = LabeledRecordFinder(is_label_line=\
lambda x: x.startswith('/coor'))
#sequence block is first item in list
sequence_block, coord_block = list(coordinate_finder(seq_block))
#split on pairs block
pairs_finder = LabeledRecordFinder(is_label_line=\
lambda x: x.startswith('/pairs'))
#coordinate block is first item in list
coordinate_block, pairs_block = list(pairs_finder(coord_block))
sequence = get_sequence(sequence_block)
coordinates = get_coordinates(coordinate_block)
pairs = get_pairs(pairs_block)
return sequence, coordinates, pairs
|
