3 """@namespace IMP.isd.TALOSReader
4 Classes to handle TALOS files or folders.
7 from __future__
import print_function
16 """ reads a TALOS file, or a TALOS folder, and stores the data """
18 def __init__(self, sequence, detailed_input, keep_all=False,
19 sequence_match=(1, 1)):
20 """start the TALOSReader
21 sequence : a dictionary of sequence number keys and 3-letter code
23 detailed_input : True if the input will be either predAll.tab or the
24 pred/res???.tab files. False if it's pred.tab
25 keep_all : whether to keep outliers or not, when detailed_input==True.
26 sequence_match : in the form (talos_no, sequence_no), assigns a
27 correspondence between residue numberings.
29 self.detailed_input = detailed_input
31 self.keep_all = keep_all
32 self.sequence = sequence
33 self.offset = sequence_match[1] - sequence_match[0]
36 """in the case of a list of predictions for one residue, add an entry to
39 'num' : number of predictions
40 'phi' : the list of predictions for phi
44 if resno
not in self.data:
46 'full':
True,
'num': len(phi),
'phi': phi,
'psi': psi}
48 raise RuntimeError(
"would overwrite data for residue %d" % resno)
51 """in the case of a single (average) prediction output by talos for a
52 given residue, add an entry to data which is:
54 'num' : the number of matches this average was calculated from
55 'phi' : a tuple in the form (mean, error)
60 if resno
not in self.data:
68 raise RuntimeError(
"would overwrite data for residue %d" % resno)
70 def _read_one_residue(self, fname):
72 resno = int(os.path.basename(fname)[3:6]) + self.offset
79 if tokens[1] ==
'RESNAMES':
80 check_residue(self.sequence[resno], tokens[3])
82 if not tokens[0].isdigit():
84 if float(tokens[4]) < 0.999
and not self.keep_all:
86 phi.append(float(tokens[1]) * 2 * pi / 360.)
87 psi.append(float(tokens[2]) * 2 * pi / 360.)
90 def _read_predAll(self, fname):
95 if len(tokens) == 0
or not tokens[0].isdigit():
98 resno = int(tokens[1]) + self.offset
104 resname = tokens[2][1]
105 check_residue(self.sequence[resno], resname)
106 if float(tokens[6]) < 0.999
and not self.keep_all:
108 phi.append(float(tokens[3]) * 2 * pi / 360.)
109 psi.append(float(tokens[4]) * 2 * pi / 360.)
111 def _read_observations(self, fname):
112 if fname.endswith(
'predAll.tab'):
113 self._read_predAll(fname)
115 self._read_one_residue(fname)
117 def _read_averages(self, fname):
120 tokens = line.split()
121 if not tokens[0].isdigit():
123 resno = int(tokens[0]) + self.offset
124 check_residue(resno, tokens[1])
125 phi, psi, dphi, dpsi = [2 * pi * float(a) / 360.
for a
in tokens[2:6]]
132 "reads a TALOS file and returns data. See add_datum methods."
133 if self.detailed_input:
134 self._read_observations(fname)
136 self._read_averages(fname)
142 if __name__ ==
'__main__':
145 sequence = read_sequence_file(
'seq.dat', sequence_match=(1, 5))
148 data = reader.get_data()
def add_full_datum
in the case of a list of predictions for one residue, add an entry to data which is: 'full' : always ...
def __init__
start the TALOSReader sequence : a dictionary of sequence number keys and 3-letter code values...
reads a TALOS file, or a TALOS folder, and stores the data
def read
reads a TALOS file and returns data.
def add_mean_datum
in the case of a single (average) prediction output by talos for a given residue, add an entry to dat...