source: rotdif/bin/plot_diso.py @ 1541

Last change on this file since 1541 was 1541, checked in by yuexi, 3 years ago

plot diso

  • Property svn:executable set to *
File size: 3.7 KB
Line 
1#!/usr/bin/python
2##order in dynamics output is s2, tauloc,rex,s2_fast
3from preprocess_split import split_data
4
5def diso_plot(color_list, exp_keys, in_dock, dyna_flag, elm_flag, elmdock_flag):
6    diso_output = split_data(dyna_flag, elm_flag, elmdock_flag)[3]
7    diso_data = []
8    if in_dock == True:
9        for i in range(len(exp_keys)):
10            diso_tmp = [
11       {
12         "x" : diso_output[0][i],
13         "y" : diso_output[1][i],
14         "mode" : "lines+markers",
15         "marker" : {
16           "color" : color_list[i],
17           "size": 8
18         },
19         "line" : {
20         "color" : color_list[i],
21         "width" : 1
22         },
23         "name": str(exp_keys[i])
24       },
25       {
26         "x" : diso_output[0][i],
27         "y" : diso_output[2][i],
28         "type" : "bar",
29         "marker": {"color": color_list[i]},
30         "showlegend": False,
31         "xaxis": "x2",
32         "yaxis": "y2"
33       },
34       {
35         "x" : diso_output[0][i],
36         "y" : diso_output[3][i],
37         "type" : "bar",
38         "marker": {"color": color_list[i]},
39         "showlegend": False,
40         "xaxis" : "x3",
41         "yaxis" : "y3",
42         "type"  : "bar"
43       },
44       {
45         "x": diso_output[0][i],
46         "y": diso_output[4][i],
47         "xaxis": "x4",
48         "yaxis": "y4",
49         "mode": "lines+markers",
50         "showlegend": False,
51         "marker": {
52           "color" : color_list[i],
53           "size" : 8
54         },
55         "line": {
56           "color": color_list[i],
57           "width":1
58         }
59       }   
60       ]
61            diso_data.extend(diso_tmp)
62
63    else:
64        diso_data = [
65       {
66         "x" : diso_output[0],
67         "y" : diso_output[1],
68         "mode" : "lines+markers",
69         "marker" : {
70           "color" : color_list[3],
71           "size": 8
72         }, 
73         "line" : { 
74         "color" : color_list[3],
75         "width" : 1 
76         }, 
77         "name": "All Frequencies"
78       }, 
79       {   
80         "x" : diso_output[0],
81         "y" : diso_output[2],
82         "type" : "bar",
83         "marker": {"color": color_list[3]},
84         "showlegend": False,
85         "xaxis": "x2",
86         "yaxis": "y2"
87       }, 
88       {   
89         "x" : diso_output[0],
90         "y" : diso_output[3],
91         "type" : "bar",
92         "marker": {"color": color_list[3]},
93         "showlegend": False,
94         "xaxis" : "x3",
95         "yaxis" : "y3",
96         "type"  : "bar"
97       }, 
98       {   
99         "x": diso_output[0],
100         "y": diso_output[4],
101         "xaxis": "x4",
102         "yaxis": "y4",
103         "mode": "lines+markers",
104         "showlegend": False,
105         "marker": {
106           "color" : color_list[3],
107           "size" : 8
108         }
109        }]   
110
111    diso_plotly ={
112   "data" : diso_data,
113   "layout" : {
114       "title" : "Dynamics: Isotropic Model",
115       "xaxis" : {
116         "title" : "residue number",
117         "domain" : [0,0.45]
118       },
119       "yaxis" : {
120         "title" : "S^2",
121         "domain" : [0.65,1]
122       },
123       "xaxis2" : { 
124         "title" : "residue number",
125         "domain": [0.55,1],
126         "anchor": "y2"
127       }, 
128       "yaxis2" : { 
129         "title" : "tau_loc(ns)",
130         "domain": [0,0.35],
131         "anchor": "x2"
132       },
133       "xaxis3" : {
134         "title": "residue number",
135         "domain": [0,0.45],
136         "anchor": "y3"   
137       },
138       "yaxis3" : {
139         "title": "Rex(1/s)",
140         "domain": [0,0.35],
141         "anchor": "x3"   
142       },
143       "xaxis4" : {
144         "title": "residue number",
145         "domain": [0.55,1],
146         "anchor": "y4"
147       },
148       "yaxis4" : {
149         "title": "S^2_fast",
150         "domain": [0.65,1],
151         "anchor": "x4"
152       }
153   }
154}
155    return diso_plotly
156
Note: See TracBrowser for help on using the repository browser.