-
Notifications
You must be signed in to change notification settings - Fork 1
/
feature_types.py
69 lines (61 loc) · 3.46 KB
/
feature_types.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
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
from residue import Residue
def convert_list_to_resobject_list(contact_residues):
resobject_list = []
for residue in contact_residues:
new_residue = Residue(resSeq = residue[1], chain_id = residue[0])
resobject_list.append(new_residue)
return(resobject_list)
#CHOOSE RESIDUES:
helix_residues = {}
helix_residues["tm1"] = list(range(29,61))
helix_residues["icl1"] = list(range(61,67))
helix_residues["tm2"] = list(range(67,97))
helix_residues["ecl1"] = list(range(97,102))
helix_residues["tm3"] = list(range(102,137))
helix_residues["icl2"] = list(range(137,147))
helix_residues["tm4"] = list(range(147,171))
helix_residues["ecl2"] = list(range(171,197))
helix_residues["tm5"] = list(range(197,230))
helix_residues["icl3"] = list(range(230,267))
helix_residues["tm6"] = list(range(267,299))
helix_residues["ecl3"] = list(range(299,305))
helix_residues["tm7"] = list(range(305,330))
cutoff = 1.0
helix_residues["tm8"] = list(range(330,340))
switch_residues = [130, 131, 208, 211, 219, 268, 272, 286, 288, 316, 322, 323, 326]
switch_npxx = [130, 131, 208, 211, 219, 268, 272, 286, 288, 316] + list(range(322,328))
switch_pp_npxx = set(switch_npxx + [51, 79, 106, 113, 118, 121, 130, 131, 132, 141, 158, 208, 211, 219, 268, 272, 282, 285, 286, 288, 316, 318, 319, 320, 323, 326, 282])
tm6_residues = list(range(270,299))
bp_residues = [82, 86, 93, 106, 110, 113, 114, 117, 118, 164, 191, 192, 193, 195, 199, 200, 203, 206, 208, 286, 289, 290, 293, 305, 308, 309, 312, 316]
dihedral_residues = list(set(switch_npxx + tm6_residues))
skip_5_residues = list(range(30,340,5))
skip_3_residues = list(range(30,340,3))
skip5_switches_pp_npxx = list(set(skip_5_residues + list(switch_pp_npxx)))
skip5_switches_pp_npxx_ser = list(set(skip_5_residues + list(switch_pp_npxx) + [207]))
skip3_switches_pp_npxx = list(set(skip_3_residues + list(switch_pp_npxx)))
#print(len(skip5_switches_pp_npxx))
all_residues = list(range(29,340))
tm_residues = helix_residues["tm1"] + helix_residues["tm2"] + helix_residues["tm3"] + helix_residues["tm4"] + helix_residues["tm5"] + helix_residues["tm6"] + helix_residues["tm7"] + helix_residues["tm8"]
sampling_method = "random"
precision = "SP"
#feature_types = "_switches_tm6"
#feature_types = "_switches_npxx_tm6_bp"
#feature_types = "_switches_npxx_tm6_dihedrals_switches_npxx_contact"
#feature_types = "_switches_npxx_tm6_dihedrals_switches_pp_npxx_contact"\
#feature_types = "_skip5_switches_pp_npxx_contact"
#feature_types = "_skip5_switches_pp_npxx_contact_cutoff%dnm" %(int(cutoff))
#feature_types = "_skip3_switches_pp_npxx_contact_cutoff20"
#feature_types = "switches_pp_npxx_contact_cutoff10000"
#feature_types = "skip5_switches_pp_npxx_ser_cutoff%dnm" %(int(cutoff))
#feature_types = "all_residues_under_cutoff%dnm" %(int(cutoff))
#feature_types = "all_residues_under_cutoff%dnm_allframes" %(int(cutoff))
#feature_types = "all_tm_residues_under_cutoff%dnm" %(int(cutoff))
#feature_types = "reimaged_notrajfix_tm_residues_under_cutoff%dnm" %(int(cutoff))
#feature_types = "reimaged_notrajfix_tm_residues_2rh1_3sn6_under_cutoff%dnm" %(int(cutoff))
#feature_types = "reimaged_notrajfix_all_residues_under_cutoff%dnm" %(int(cutoff))
#feature_types = "all_residues_2rh1_3sn6_under_cutoff%dnm" %(int(cutoff))
feature_name = "tm_residues_2rh1_3sn6_under_cutoff%dnm" %(int(cutoff))
#feature_types = "reference_receptors"
contact_residues = [(0, res) for res in tm_residues]
#contact_residues = [(0, res) for res in all_residues]
contact_residues = convert_list_to_resobject_list(contact_residues)