source: rotdif/bin/main_elmdock.py @ 1647

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

update

  • Property svn:executable set to *
File size: 2.7 KB
Line 
1#!/usr/bin/python
2import os
3import glob
4import re
5import shutil
6import sys
7import subprocess
8import json
9import cStringIO
10import shlex 
11import socket
12import time
13import pty
14import select
15import atexit
16import signal
17import functools
18import itertools
19import math
20import pandas as pd
21import numpy as np
22from itertools import groupby
23from matplotlib import cm, colors
24from StringIO import StringIO
25from subprocess import Popen, PIPE, STDOUT
26
27#import our functions
28from run_rotdif_elmdock import rotdif
29from plot_2d_3d import plot_2d_3d
30from plot_exp_elmdock import exp_plot
31from plot_vec import vec_plot
32from plot_iso import iso_plot
33from plot_axi import axi_plot
34from plot_ani import ani_plot
35from plot_diso import diso_plot
36from plot_daxi import daxi_plot
37from plot_dani import dani_plot
38from plot_chi2 import chi2_plot
39from save_elm import save_elm
40from save_dock_elmdock import save_dock
41from indicator_chain import dock
42from StringIO import StringIO
43
44if __name__ == "__main__":
45    color_list=[ '#1f77b4','#ff7f0e','#2ca02c','#d62728', '#9467bd',\
46'#8c564b', '#e377c2', '#7f7f7f', '#bcbd22','#17becf']
47    json_variables = " " 
48    InitialDirectoryStr = os.path.abspath(os.path.dirname(sys.argv[0]))
49    argv_io_string = StringIO(sys.argv[1])
50    json_variables = json.load(argv_io_string)
51    relax_location = json_variables['relax_location']
52    relax_loc = relax_location[0]
53    to_dock = dock(relax_loc)
54    output_res, base_dir, dyna_flag, elm_flag, elmdock_flag = rotdif(to_dock)
55    exp_keys = exp_plot(color_list, dyna_flag, elm_flag, elmdock_flag,relax_loc)[1]
56    outputmd_arr_files = []
57    log_out = [str(base_dir) + "/" + 'elmdock_log.out']
58    if dyna_flag == True:
59        output_res[ 'diso_plot' ] = diso_plot(color_list, exp_keys, to_dock, dyna_flag, elm_flag, elmdock_flag)
60        output_res[ 'daxi_plot' ] = daxi_plot(color_list, exp_keys, to_dock, dyna_flag, elm_flag, elmdock_flag)
61        output_res[ 'dani_plot' ] = dani_plot(color_list,exp_keys, to_dock, dyna_flag, elm_flag, elmdock_flag)
62 
63    if elm_flag == True:
64        save_elm(dyna_flag, elm_flag, elmdock_flag)
65        elm_out = [str(base_dir) + "/" + 'ELM_prediction']
66        output_res[ 'elm_out' ] = elm_out   
67
68    if elmdock_flag == True:
69        save_dock(dyna_flag, elm_flag, elmdock_flag)
70        dock_out = [str(base_dir) + "/" + 'ELMDOCK']     
71        output_res[ 'elmdock_out' ] = dock_out
72        dock_pdb = [str(base_dir) + "/" + 'out_dock.pdb']
73        output_res[ 'pdb'] = dock_pdb
74        view_pdb = str(base_dir) + "/" + 'out_dock.pdb'
75        output_res['outputpdb'] = { "file" : view_pdb, "script" : "ribbon" }
76       
77        #user_pdb = [str(base_dir)+"/"+'diUb_AB.pdb']
78    output_res[ 'outputrotdif' ] = log_out
79    print (json.dumps(output_res))
Note: See TracBrowser for help on using the repository browser.