def pdb2gmx_pc(input_structure_pdb_path, output_gro_path, output_top_zip_path, properties, **kwargs): pdb2gmx.Pdb2gmx(input_structure_pdb_path=input_structure_pdb_path, output_gro_path=output_gro_path, output_top_zip_path=output_top_zip_path, properties=properties, **kwargs).launch()
def main(): start_time = time.time() yaml_path = sys.argv[1] system = sys.argv[2] conf = settings.YamlReader(yaml_path, system) workflow_path = conf.properties[system]['workflow_path'] fu.create_dir(os.path.abspath(workflow_path)) out_log, _ = fu.get_logs(path=workflow_path, console=True, level='DEBUG') paths = conf.get_paths_dic() prop = conf.get_prop_dic(global_log=out_log) out_log.info('') out_log.info('_______GROMACS FULL WORKFLOW_______') out_log.info('') out_log.info("Command Executed:") out_log.info(" ".join(sys.argv)) out_log.info('Workflow_path: ' + workflow_path) out_log.info('Config File: ' + yaml_path) out_log.info('System: ' + system) out_log.info('') out_log.info('step1: mmbpdb ------ Get PDB') structure = conf.properties[system].get('initial_structure_pdb_path', None) out_log.info(22 * ' ' + 'Selected PDB structure: ' + structure) out_log.info('step2 and 3: scw --- Check and Repair PDB') fu.create_dir(prop['step3_scw']['path']) paths['step3_scw']['input_pdb_path'] = structure out_log.debug('\nPaths:\n' + str(paths['step3_scw']) + '\nProperties:\n' + str(prop['step3_scw']) + '\n') scwrl.Scwrl4(properties=prop['step3_scw'], **paths['step3_scw']).launch() out_log.info('step4: p2g --------- Create gromacs topology') fu.create_dir(prop['step4_p2g']['path']) out_log.debug('\nPaths:\n' + str(paths['step4_p2g']) + '\nProperties:\n' + str(prop['step4_p2g']) + '\n') pdb2gmx.Pdb2gmx(properties=prop['step4_p2g'], **paths['step4_p2g']).launch() out_log.info('step5: ec ---------- Define box dimensions') fu.create_dir(prop['step5_ec']['path']) out_log.debug('\nPaths:\n' + str(paths['step5_ec']) + '\nProperties:\n' + str(prop['step5_ec']) + '\n') editconf.Editconf(properties=prop['step5_ec'], **paths['step5_ec']).launch() out_log.info('step6: sol --------- Fill the box with water molecules') fu.create_dir(prop['step6_sol']['path']) out_log.debug('\nPaths:\n' + str(paths['step6_sol']) + '\nProperties:\n' + str(prop['step6_sol']) + '\n') solvate.Solvate(properties=prop['step6_sol'], **paths['step6_sol']).launch() out_log.info('step7: gppions ----- Preprocessing: Adding monoatomic ions') fu.create_dir(prop['step7_gppions']['path']) out_log.debug('\nPaths:\n' + str(paths['step7_gppions']) + '\nProperties:\n' + str(prop['step7_gppions']) + '\n') grompp.Grompp(properties=prop['step7_gppions'], **paths['step7_gppions']).launch() out_log.info('step8: gio --------- Running: Adding monoatomic ions') fu.create_dir(prop['step8_gio']['path']) out_log.debug('\nPaths:\n' + str(paths['step8_gio']) + '\nProperties:\n' + str(prop['step8_gio']) + '\n') genion.Genion(properties=prop['step8_gio'], **paths['step8_gio']).launch() out_log.info('Step9: gppndx ------- Preprocessing index creation') fu.create_dir(prop['step9_gppndx']['path']) out_log.debug('\nPaths:\n' + str(paths['step9_gppndx']) + '\nProperties:\n' + str(prop['step9_gppndx']) + '\n') grompp.Grompp(properties=prop['step9_gppndx'], **paths['step9_gppndx']).launch() out_log.info('Step10: make_ndx ---- Create restrain index') fu.create_dir(prop['step10_make_ndx']['path']) out_log.debug('\nPaths:\n' + str(paths['step10_make_ndx']) + '\nProperties:\n' + str(prop['step10_make_ndx']) + '\n') make_ndx.MakeNdx(properties=prop['step10_make_ndx'], **paths['step10_make_ndx']).launch() out_log.info('Step11: genrest - Create restrain topology') fu.create_dir(prop['step11_genrestr']['path']) out_log.debug('\nPaths:\n' + str(paths['step11_genrestr']) + '\nProperties:\n' + str(prop['step11_genrestr']) + '\n') genrestr.Genrestr(properties=prop['step11_genrestr'], **paths['step11_genrestr']).launch() out_log.info('step15: gppmin ------ Preprocessing: minimization') fu.create_dir(prop['step15_gppmin']['path']) out_log.debug('\nPaths:\n' + str(paths['step15_gppmin']) + '\nProperties:\n' + str(prop['step15_gppmin']) + '\n') grompp.Grompp(properties=prop['step15_gppmin'], **paths['step15_gppmin']).launch() out_log.info('step16: mdmin ------- Running: minimization') fu.create_dir(prop['step16_mdmin']['path']) out_log.debug('\nPaths:\n' + str(paths['step16_mdmin']) + '\nProperties:\n' + str(prop['step16_mdmin']) + '\n') mdrun.Mdrun(properties=prop['step16_mdmin'], **paths['step16_mdmin']).launch() out_log.info('Step17: genrestr - Create restrain topology') fu.create_dir(prop['step17_genrestr']['path']) out_log.debug('\nPaths:\n' + str(paths['step17_genrestr']) + '\nProperties:\n' + str(prop['step17_genrestr']) + '\n') genrestr.Genrestr(properties=prop['step17_genrestr'], **paths['step17_genrestr']).launch() out_log.info('step18: gppsa ------- Preprocessing: simulated annealing') fu.create_dir(prop['step18_gppsa']['path']) out_log.debug('\nPaths:\n' + str(paths['step18_gppsa']) + '\nProperties:\n' + str(prop['step18_gppsa']) + '\n') grompp.Grompp(properties=prop['step18_gppsa'], **paths['step18_gppsa']).launch() out_log.info('step19: mdsa -------- Running: simulated annealing') fu.create_dir(prop['step19_mdsa']['path']) out_log.debug('\nPaths:\n' + str(paths['step19_mdsa']) + '\nProperties:\n' + str(prop['step19_mdsa']) + '\n') mdrun.Mdrun(properties=prop['step19_mdsa'], **paths['step19_mdsa']).launch() out_log.info( 'step20: gppnvt_1000 - Preprocessing: nvt constant number of molecules, volume and temp' ) fu.create_dir(prop['step20_gppnvt_1000']['path']) out_log.debug('\nPaths:\n' + str(paths['step20_gppnvt_1000']) + '\nProperties:\n' + str(prop['step20_gppnvt_1000']) + '\n') grompp.Grompp(properties=prop['step20_gppnvt_1000'], **paths['step20_gppnvt_1000']).launch() out_log.info( 'step21: mdnvt_1000 -- Running: nvt constant number of molecules, volume and temp' ) fu.create_dir(prop['step21_mdnvt_1000']['path']) out_log.debug('\nPaths:\n' + str(paths['step21_mdnvt_1000']) + '\nProperties:\n' + str(prop['step21_mdnvt_1000']) + '\n') mdrun.Mdrun(properties=prop['step21_mdnvt_1000'], **paths['step21_mdnvt_1000']).launch() out_log.info('Step22: genrestr - Create restrain topology') fu.create_dir(prop['step22_genrestr']['path']) out_log.debug('\nPaths:\n' + str(paths['step22_genrestr']) + '\nProperties:\n' + str(prop['step22_genrestr']) + '\n') genrestr.Genrestr(properties=prop['step22_genrestr'], **paths['step22_genrestr']).launch() out_log.info( 'step23: gppnvt_800 -- Preprocessing: nvt constant number of molecules, volume and temp' ) fu.create_dir(prop['step23_gppnvt_800']['path']) out_log.debug('\nPaths:\n' + str(paths['step23_gppnvt_800']) + '\nProperties:\n' + str(prop['step23_gppnvt_800']) + '\n') grompp.Grompp(properties=prop['step23_gppnvt_800'], **paths['step23_gppnvt_800']).launch() out_log.info( 'step24: mdnvt_800 --- Running: nvt constant number of molecules, volume and temp' ) fu.create_dir(prop['step24_mdnvt_800']['path']) out_log.debug('\nPaths:\n' + str(paths['step24_mdnvt_800']) + '\nProperties:\n' + str(prop['step24_mdnvt_800']) + '\n') mdrun.Mdrun(properties=prop['step24_mdnvt_800'], **paths['step24_mdnvt_800']).launch() out_log.info('Step25: genrestr - Create restrain topology') fu.create_dir(prop['step25_genrestr']['path']) out_log.debug('\nPaths:\n' + str(paths['step25_genrestr']) + '\nProperties:\n' + str(prop['step25_genrestr']) + '\n') genrestr.Genrestr(properties=prop['step25_genrestr'], **paths['step25_genrestr']).launch() out_log.info( 'step26: gppnpt_500 -- Preprocessing: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step26_gppnpt_500']['path']) out_log.debug('\nPaths:\n' + str(paths['step26_gppnpt_500']) + '\nProperties:\n' + str(prop['step26_gppnpt_500']) + '\n') grompp.Grompp(properties=prop['step26_gppnpt_500'], **paths['step26_gppnpt_500']).launch() out_log.info( 'step27: mdnpt_500 --- Running: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step27_mdnpt_500']['path']) out_log.debug('\nPaths:\n' + str(paths['step27_mdnpt_500']) + '\nProperties:\n' + str(prop['step27_mdnpt_500']) + '\n') mdrun.Mdrun(properties=prop['step27_mdnpt_500'], **paths['step27_mdnpt_500']).launch() out_log.info('Step28: genrestr - Create restrain topology') fu.create_dir(prop['step28_genrestr']['path']) out_log.debug('\nPaths:\n' + str(paths['step28_genrestr']) + '\nProperties:\n' + str(prop['step28_genrestr']) + '\n') genrestr.Genrestr(properties=prop['step28_genrestr'], **paths['step28_genrestr']).launch() out_log.info( 'step29: gppnpt_300 -- Preprocessing: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step29_gppnpt_300']['path']) out_log.debug('\nPaths:\n' + str(paths['step29_gppnpt_300']) + '\nProperties:\n' + str(prop['step29_gppnpt_300']) + '\n') grompp.Grompp(properties=prop['step29_gppnpt_300'], **paths['step29_gppnpt_300']).launch() out_log.info( 'step30: mdnpt_300 --- Running: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step30_mdnpt_300']['path']) out_log.debug('\nPaths:\n' + str(paths['step30_mdnpt_300']) + '\nProperties:\n' + str(prop['step30_mdnpt_300']) + '\n') mdrun.Mdrun(properties=prop['step30_mdnpt_300'], **paths['step30_mdnpt_300']).launch() out_log.info('Step31: genrestr - Create restrain topology') fu.create_dir(prop['step31_genrestr']['path']) out_log.debug('\nPaths:\n' + str(paths['step31_genrestr']) + '\nProperties:\n' + str(prop['step31_genrestr']) + '\n') genrestr.Genrestr(properties=prop['step31_genrestr'], **paths['step31_genrestr']).launch() out_log.info( 'step32: gppnpt_200 -- Preprocessing: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step32_gppnpt_200']['path']) out_log.debug('\nPaths:\n' + str(paths['step32_gppnpt_200']) + '\nProperties:\n' + str(prop['step32_gppnpt_200']) + '\n') grompp.Grompp(properties=prop['step32_gppnpt_200'], **paths['step32_gppnpt_200']).launch() out_log.info( 'step33: mdnpt_200 --- Running: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step33_mdnpt_200']['path']) out_log.debug('\nPaths:\n' + str(paths['step33_mdnpt_200']) + '\nProperties:\n' + str(prop['step33_mdnpt_200']) + '\n') mdrun.Mdrun(properties=prop['step33_mdnpt_200'], **paths['step33_mdnpt_200']).launch() out_log.info('Step34: genrestr - Create restrain topology') fu.create_dir(prop['step34_genrestr']['path']) out_log.debug('\nPaths:\n' + str(paths['step34_genrestr']) + '\nProperties:\n' + str(prop['step34_genrestr']) + '\n') genrestr.Genrestr(properties=prop['step34_genrestr'], **paths['step34_genrestr']).launch() out_log.info( 'step35: gppnpt_100 -- Preprocessing: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step35_gppnpt_100']['path']) out_log.debug('\nPaths:\n' + str(paths['step35_gppnpt_100']) + '\nProperties:\n' + str(prop['step35_gppnpt_100']) + '\n') grompp.Grompp(properties=prop['step35_gppnpt_100'], **paths['step35_gppnpt_100']).launch() out_log.info( 'step36: mdnpt_100 --- Running: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step36_mdnpt_100']['path']) out_log.debug('\nPaths:\n' + str(paths['step36_mdnpt_100']) + '\nProperties:\n' + str(prop['step36_mdnpt_100']) + '\n') mdrun.Mdrun(properties=prop['step36_mdnpt_100'], **paths['step36_mdnpt_100']).launch() out_log.info('Step37: genrestr - Create restrain topology') fu.create_dir(prop['step37_genrestr']['path']) out_log.debug('\nPaths:\n' + str(paths['step37_genrestr']) + '\nProperties:\n' + str(prop['step37_genrestr']) + '\n') genrestr.Genrestr(properties=prop['step37_genrestr'], **paths['step37_genrestr']).launch() out_log.info( 'step38: gppnpt ------ Preprocessing: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step38_gppnpt']['path']) out_log.debug('\nPaths:\n' + str(paths['step38_gppnpt']) + '\nProperties:\n' + str(prop['step38_gppnpt']) + '\n') grompp.Grompp(properties=prop['step38_gppnpt'], **paths['step38_gppnpt']).launch() out_log.info( 'step39: mdnpt ------- Running: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step39_mdnpt']['path']) out_log.debug('\nPaths:\n' + str(paths['step39_mdnpt']) + '\nProperties:\n' + str(prop['step39_mdnpt']) + '\n') mdrun.Mdrun(properties=prop['step39_mdnpt'], **paths['step39_mdnpt']).launch() out_log.info( 'step40: gppmd ------- Preprocessing: Free Molecular dynamics') fu.create_dir(prop['step40_gppmd']['path']) out_log.debug('\nPaths:\n' + str(paths['step40_gppmd']) + '\nProperties:\n' + str(prop['step40_gppmd']) + '\n') grompp.Grompp(properties=prop['step40_gppmd'], **paths['step40_gppmd']).launch() out_log.info('step41: md ---------- Running: Free Molecular dynamics') fu.create_dir(prop['step41_md']['path']) out_log.debug('\nPaths:\n' + str(paths['step41_md']) + '\nProperties:\n' + str(prop['step41_md']) + '\n') mdrun.Mdrun(properties=prop['step41_md'], **paths['step41_md']).launch() fu.remove_temp_files([ '#', '.top', '.plotscript', '.edr', '.xtc', '.itp', '.top', '.log', '.pdb', '.cpt', '.mdp', '.xvg', '.seq' ]) elapsed_time = time.time() - start_time out_log.info('') out_log.info('') out_log.info('Execution sucessful: ') out_log.info(' Workflow_path: ' + workflow_path) out_log.info(' Config File: ' + yaml_path) out_log.info(' System: ' + system) if len(sys.argv) >= 5: out_log.info(' Nodes: ' + sys.argv[4]) out_log.info('') out_log.info('Elapsed time: ' + str(elapsed_time) + ' seconds') out_log.info('')
def main(): start_time = time.time() yaml_path=sys.argv[1] system=sys.argv[2] n_mutations=sys.argv[3] conf = settings.YamlReader(yaml_path, system) workflow_path = conf.properties[system]['workflow_path'] fu.create_dir(os.path.abspath(workflow_path)) out_log, _ = fu.get_logs(path=workflow_path, console=True) paths_glob = conf.get_paths_dic() prop_glob = conf.get_prop_dic() out_log.info('') out_log.info('_______GROMACS FULL WORKFLOW_______') out_log.info('') out_log.info("Command Executed:") out_log.info(" ".join(sys.argv)) out_log.info('Workflow_path: '+workflow_path) out_log.info('Config File: '+yaml_path) out_log.info('System: '+system) out_log.info('Mutations limit: '+n_mutations) if len(sys.argv) >= 5: out_log.info('Nodes: '+sys.argv[4]) out_log.info('') out_log.info( 'step1: mmbpdb -- Get PDB') structure = conf.properties[system].get('initial_structure_pdb_path', None) if structure is None or not os.path.isfile(structure): out_log.info( ' Selected PDB code: ' + prop_glob['step1_mmbpdb']['pdb_code']) fu.create_dir(prop_glob['step1_mmbpdb']['path']) pdb.MmbPdb().get_pdb(prop_glob['step1_mmbpdb']['pdb_code'], paths_glob['step1_mmbpdb']['output_pdb_path']) structure = paths_glob['step1_mmbpdb']['output_pdb_path'] out_log.info( 'step2: mmbuniprot -- Get mutations') mutations = conf.properties.get('input_mapped_mutations_list', None) if mutations is None or len(mutations) < 7: mmbuniprot = uniprot.MmbVariants(prop_glob['step1_mmbpdb']['pdb_code']) mutations = mmbuniprot.get_pdb_variants() if mutations is None or len(mutations) == 0: return else: mutations = [m.strip() for m in conf.properties.get('input_mapped_mutations_list').split(',')] mutations_limit = min(len(mutations), int(n_mutations)) out_log.info('') out_log.info('Number of mutations to be modelled: ' + str(mutations_limit)) rms_list = [] mutations_counter = 0 for mut in mutations: if mutations_counter == mutations_limit: break mutations_counter += 1 paths = conf.get_paths_dic(mut) prop = conf.get_prop_dic(mut, global_log=out_log) out_log.info('') out_log.info('-------------------------') out_log.info(str(mutations_counter) + '/' + str(mutations_limit) + ' ' + mut) out_log.info('-------------------------') out_log.info('') out_log.info('step3: scw ------ Model mutation') fu.create_dir(prop['step3_scw']['path']) paths['step3_scw']['input_pdb_path']=structure scwrl.Scwrl4(properties=prop['step3_scw'], **paths['step3_scw']).launch() out_log.info('step4: p2g ------ Create gromacs topology') fu.create_dir(prop['step4_p2g']['path']) pdb2gmx.Pdb2gmx(properties=prop['step4_p2g'], **paths['step4_p2g']).launch() out_log.info('step5: ec ------- Define box dimensions') fu.create_dir(prop['step5_ec']['path']) editconf.Editconf(properties=prop['step5_ec'], **paths['step5_ec']).launch() out_log.info('step6: sol ------ Fill the box with water molecules') fu.create_dir(prop['step6_sol']['path']) solvate.Solvate(properties=prop['step6_sol'], **paths['step6_sol']).launch() out_log.info('step7: gppions -- Preprocessing: Adding monoatomic ions') fu.create_dir(prop['step7_gppions']['path']) grompp.Grompp(properties=prop['step7_gppions'], **paths['step7_gppions']).launch() out_log.info('step8: gio ------ Running: Adding monoatomic ions') fu.create_dir(prop['step8_gio']['path']) genion.Genion(properties=prop['step8_gio'], **paths['step8_gio']).launch() out_log.info('step9: gppmin --- Preprocessing: Energy minimization') fu.create_dir(prop['step9_gppmin']['path']) grompp.Grompp(properties=prop['step9_gppmin'], **paths['step9_gppmin']).launch() out_log.info('step10: mdmin ---- Running: Energy minimization') fu.create_dir(prop['step10_mdmin']['path']) mdrun.Mdrun(properties=prop['step10_mdmin'], **paths['step10_mdmin']).launch() out_log.info('step11: gppnvt --- Preprocessing: nvt constant number of molecules, volume and temp') fu.create_dir(prop['step11_gppnvt']['path']) grompp.Grompp(properties=prop['step11_gppnvt'], **paths['step11_gppnvt']).launch() out_log.info('step12: mdnvt ---- Running: nvt constant number of molecules, volume and temp') fu.create_dir(prop['step12_mdnvt']['path']) mdrun.Mdrun(properties=prop['step12_mdnvt'], **paths['step12_mdnvt']).launch() out_log.info('step13: gppnpt --- Preprocessing: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step13_gppnpt']['path']) grompp.Grompp(properties=prop['step13_gppnpt'], **paths['step13_gppnpt']).launch() out_log.info('step14: mdnpt ---- Running: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step14_mdnpt']['path']) mdrun.Mdrun(properties=prop['step14_mdnpt'], **paths['step14_mdnpt']).launch() out_log.info('step15: gppeq ---- Preprocessing: 1ns Molecular dynamics Equilibration') fu.create_dir(prop['step15_gppeq']['path']) grompp.Grompp(properties=prop['step15_gppeq'], **paths['step15_gppeq']).launch() out_log.info('step16: mdeq ----- Running: Free Molecular dynamics Equilibration') fu.create_dir(prop['step16_mdeq']['path']) mdrun.Mdrun(properties=prop['step16_mdeq'], **paths['step16_mdeq']).launch() out_log.info('step17: rmsd ----- Computing RMSD') fu.create_dir(prop['step17_rmsd']['path']) rms_list.append(rms.Rms(properties=prop['step17_rmsd'], **paths['step17_rmsd']).launch()) removed_list = fu.remove_temp_files(['#', '.top', '.plotscript', '.edr', '.xtc', '.itp', '.top', '.log', '.pdb', '.cpt', '.mdp', '.xvg', '.grp', '.seq']) xvg_dict=reduce(lambda a, b: dict(a, **b), rms_list) out_log.info('step18: gnuplot ----- Creating RMSD plot') fu.create_dir(prop_glob['step18_gnuplot']['path']) gnuplot.Gnuplot(input_xvg_path_dict=xvg_dict, properties=prop_glob['step18_gnuplot'], **paths_glob['step18_gnuplot']).launch() elapsed_time = time.time() - start_time out_log.info('') out_log.info('') out_log.info('Execution sucessful: ') out_log.info(' Workflow_path: '+workflow_path) out_log.info(' Config File: '+yaml_path) out_log.info(' System: '+system) if len(sys.argv) >= 5: out_log.info(' Nodes: '+sys.argv[4]) out_log.info('') out_log.info('Elapsed time: '+str(elapsed_time)+' seconds') out_log.info('')
def main(): start_time = time.time() yaml_path=sys.argv[1] system=sys.argv[2] n_mutations=sys.argv[3] conf = settings.YamlReader(yaml_path, system) workflow_path = conf.properties[system]['workflow_path'] fu.create_dir(os.path.abspath(workflow_path)) out_log, _ = fu.get_logs(path=workflow_path, console=True) paths_glob = conf.get_paths_dic() prop_glob = conf.get_prop_dic() out_log.info('') out_log.info('_______GROMACS FULL WORKFLOW_______') out_log.info('') out_log.info("Command Executed:") out_log.info(" ".join(sys.argv)) out_log.info('Workflow_path: '+workflow_path) out_log.info('Config File: '+yaml_path) out_log.info('System: '+system) out_log.info('Mutations limit: '+n_mutations) if len(sys.argv) >= 5: out_log.info('Nodes: '+sys.argv[4]) out_log.info('') out_log.info( 'step1: mmbpdb -- Get PDB') structure = conf.properties[system].get('initial_structure_pdb_path', None) if structure is None or not os.path.isfile(structure): out_log.info( ' Selected PDB code: ' + prop_glob['step1_mmbpdb']['pdb_code']) fu.create_dir(prop_glob['step1_mmbpdb']['path']) pdb.MmbPdb().get_pdb(prop_glob['step1_mmbpdb']['pdb_code'], paths_glob['step1_mmbpdb']['output_pdb_path']) structure = paths_glob['step1_mmbpdb']['output_pdb_path'] out_log.info( 'step2: mmbuniprot -- Get mutations') mutations = conf.properties.get('input_mapped_mutations_list', None) if mutations is None or len(mutations) < 7: mmbuniprot = uniprot.MmbVariants(prop_glob['step1_mmbpdb']['pdb_code']) mutations = mmbuniprot.get_pdb_variants() if mutations is None or len(mutations) == 0: return else: mutations = [m.strip() for m in conf.properties.get('input_mapped_mutations_list').split(',')] mutations_limit = min(len(mutations), int(n_mutations)) out_log.info('') out_log.info('Number of mutations to be modelled: ' + str(mutations_limit)) rms_list = [] mutations_counter = 0 for mut in mutations: if mutations_counter == mutations_limit: break mutations_counter += 1 mut = mut if not mut.startswith('*') else mut.replace('*', 'ALL') paths = conf.get_paths_dic(mut) prop = conf.get_prop_dic(mut, global_log=out_log) out_log.info('') out_log.info('-------------------------') out_log.info(str(mutations_counter) + '/' + str(mutations_limit) + ' ' + mut) out_log.info('-------------------------') out_log.info('') out_log.info('step3: scw --------- Model mutation') fu.create_dir(prop['step3_scw']['path']) paths['step3_scw']['input_pdb_path']=structure scwrl.Scwrl4(properties=prop['step3_scw'], **paths['step3_scw']).launch() out_log.info('step4: p2g --------- Create gromacs topology') fu.create_dir(prop['step4_p2g']['path']) pdb2gmx.Pdb2gmx(properties=prop['step4_p2g'], **paths['step4_p2g']).launch() out_log.info('step5: ec ---------- Define box dimensions') fu.create_dir(prop['step5_ec']['path']) editconf.Editconf(properties=prop['step5_ec'], **paths['step5_ec']).launch() out_log.info('step6: sol --------- Fill the box with water molecules') fu.create_dir(prop['step6_sol']['path']) solvate.Solvate(properties=prop['step6_sol'], **paths['step6_sol']).launch() out_log.info('step7: gppions ----- Preprocessing: Adding monoatomic ions') fu.create_dir(prop['step7_gppions']['path']) grompp.Grompp(properties=prop['step7_gppions'], **paths['step7_gppions']).launch() out_log.info('step8: gio --------- Running: Adding monoatomic ions') fu.create_dir(prop['step8_gio']['path']) genion.Genion(properties=prop['step8_gio'], **paths['step8_gio']).launch() out_log.info('Step9: gppndx ------- Preprocessing index creation') fu.create_dir(prop['step9_gppndx']['path']) grompp.Grompp(properties=prop['step9_gppndx'], **paths['step9_gppndx']).launch() out_log.info('Step10: make_ndx ---- Create restrain index') fu.create_dir(prop['step10_make_ndx']['path']) make_ndx.MakeNdx(properties=prop['step10_make_ndx'], **paths['step10_make_ndx']).launch() out_log.info('Step11: ndx2resttop - Create restrain topology') fu.create_dir(prop['step11_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step11_ndx2resttop'], **paths['step11_ndx2resttop']).launch() out_log.info('step12: gppresmin Preprocessing: Mutated residue minimization') fu.create_dir(prop['step12_gppresmin']['path']) grompp.Grompp(properties=prop['step12_gppresmin'], **paths['step12_gppresmin']).launch() out_log.info('step13: mdresmin ---- Running: Mutated residue minimization') fu.create_dir(prop['step13_mdresmin']['path']) mdrun.Mdrun(properties=prop['step13_mdresmin'], **paths['step13_mdresmin']).launch() out_log.info('Step14: ndx2resttop - Create restrain topology') fu.create_dir(prop['step14_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step14_ndx2resttop'], **paths['step14_ndx2resttop']).launch() out_log.info('step15: gppmin Preprocessing: minimization') fu.create_dir(prop['step15_gppmin']['path']) grompp.Grompp(properties=prop['step15_gppmin'], **paths['step15_gppmin']).launch() out_log.info('step16: mdmin ---- Running: minimization') fu.create_dir(prop['step16_mdmin']['path']) mdrun.Mdrun(properties=prop['step16_mdmin'], **paths['step16_mdmin']).launch() out_log.info('Step17: ndx2resttop - Create restrain topology') fu.create_dir(prop['step17_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step17_ndx2resttop'], **paths['step17_ndx2resttop']).launch() out_log.info('step18: gppsa Preprocessing: simulated annealing') fu.create_dir(prop['step18_gppsa']['path']) grompp.Grompp(properties=prop['step18_gppsa'], **paths['step18_gppsa']).launch() out_log.info('step19: mdsa ---- Running: simulated annealing') fu.create_dir(prop['step19_mdsa']['path']) mdrun.Mdrun(properties=prop['step19_mdsa'], **paths['step19_mdsa']).launch() out_log.info('step20: gppnvt_1000 Preprocessing: nvt constant number of molecules, volume and temp') fu.create_dir(prop['step20_gppnvt_1000']['path']) grompp.Grompp(properties=prop['step20_gppnvt_1000'], **paths['step20_gppnvt_1000']).launch() out_log.info('step21: mdnvt_1000 ---- Running: nvt constant number of molecules, volume and temp') fu.create_dir(prop['step21_mdnvt_1000']['path']) mdrun.Mdrun(properties=prop['step21_mdnvt_1000'], **paths['step21_mdnvt_1000']).launch() out_log.info('Step22: ndx2resttop - Create restrain topology') fu.create_dir(prop['step22_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step22_ndx2resttop'], **paths['step22_ndx2resttop']).launch() out_log.info('step23: gppnvt_800 Preprocessing: nvt constant number of molecules, volume and temp') fu.create_dir(prop['step23_gppnvt_800']['path']) grompp.Grompp(properties=prop['step23_gppnvt_800'], **paths['step23_gppnvt_800']).launch() out_log.info('step24: mdnvt_800 ---- Running: nvt constant number of molecules, volume and temp') fu.create_dir(prop['step24_mdnvt_800']['path']) mdrun.Mdrun(properties=prop['step24_mdnvt_800'], **paths['step24_mdnvt_800']).launch() out_log.info('Step25: ndx2resttop - Create restrain topology') fu.create_dir(prop['step25_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step25_ndx2resttop'], **paths['step25_ndx2resttop']).launch() out_log.info('step26: gppnpt_500 Preprocessing: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step26_gppnpt_500']['path']) grompp.Grompp(properties=prop['step26_gppnpt_500'], **paths['step26_gppnpt_500']).launch() out_log.info('step27: mdnpt_500 ---- Running: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step27_mdnpt_500']['path']) mdrun.Mdrun(properties=prop['step27_mdnpt_500'], **paths['step27_mdnpt_500']).launch() out_log.info('Step28: ndx2resttop - Create restrain topology') fu.create_dir(prop['step28_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step28_ndx2resttop'], **paths['step28_ndx2resttop']).launch() out_log.info('step29: gppnpt_300 Preprocessing: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step29_gppnpt_300']['path']) grompp.Grompp(properties=prop['step29_gppnpt_300'], **paths['step29_gppnpt_300']).launch() out_log.info('step30: mdnpt_300 ---- Running: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step30_mdnpt_300']['path']) mdrun.Mdrun(properties=prop['step30_mdnpt_300'], **paths['step30_mdnpt_300']).launch() out_log.info('Step31: ndx2resttop - Create restrain topology') fu.create_dir(prop['step31_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step31_ndx2resttop'], **paths['step31_ndx2resttop']).launch() out_log.info('step32: gppnpt_200 Preprocessing: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step32_gppnpt_200']['path']) grompp.Grompp(properties=prop['step32_gppnpt_200'], **paths['step32_gppnpt_200']).launch() out_log.info('step33: mdnpt_200 ---- Running: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step33_mdnpt_200']['path']) mdrun.Mdrun(properties=prop['step33_mdnpt_200'], **paths['step33_mdnpt_200']).launch() out_log.info('Step34: ndx2resttop - Create restrain topology') fu.create_dir(prop['step34_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step34_ndx2resttop'], **paths['step34_ndx2resttop']).launch() out_log.info('step35: gppnpt_100 Preprocessing: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step35_gppnpt_100']['path']) grompp.Grompp(properties=prop['step35_gppnpt_100'], **paths['step35_gppnpt_100']).launch() out_log.info('step36: mdnpt_100 ---- Running: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step36_mdnpt_100']['path']) mdrun.Mdrun(properties=prop['step36_mdnpt_100'], **paths['step36_mdnpt_100']).launch() out_log.info('Step37: ndx2resttop - Create restrain topology') fu.create_dir(prop['step37_ndx2resttop']['path']) ndx2resttop.Ndx2resttop(properties=prop['step37_ndx2resttop'], **paths['step37_ndx2resttop']).launch() out_log.info('step38: gppnpt - Preprocessing: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step38_gppnpt']['path']) grompp.Grompp(properties=prop['step38_gppnpt'], **paths['step38_gppnpt']).launch() out_log.info('step39: mdnpt ---- Running: npt constant number of molecules, pressure and temp') fu.create_dir(prop['step39_mdnpt']['path']) mdrun.Mdrun(properties=prop['step39_mdnpt'], **paths['step39_mdnpt']).launch() fu.remove_temp_files(['#', '.top', '.plotscript', '.edr', '.xtc', '.itp', '.top', '.log', '.pdb', '.cpt', '.mdp', '.xvg', '.seq']) elapsed_time = time.time() - start_time out_log.info('') out_log.info('') out_log.info('Execution sucessful: ') out_log.info(' Workflow_path: '+workflow_path) out_log.info(' Config File: '+yaml_path) out_log.info(' System: '+system) if len(sys.argv) >= 5: out_log.info(' Nodes: '+sys.argv[4]) out_log.info('') out_log.info('Elapsed time: '+str(elapsed_time)+' seconds') out_log.info('')
def main(): start_time = time.time() yaml_path = sys.argv[1] system = sys.argv[2] conf = settings.YamlReader(yaml_path, system) workflow_path = conf.properties[system]['workflow_path'] fu.create_dir(os.path.abspath(workflow_path)) out_log, _ = fu.get_logs(path=workflow_path, console=True) paths = conf.get_paths_dic() prop = conf.get_prop_dic(global_log=out_log) out_log.info('') out_log.info('_______MD SETUP FULL WORKFLOW_______') out_log.info('') out_log.info('step1: mmbpdb --- Get PDB') structure = conf.properties[system].get('initial_structure_pdb_path', None) if structure is None or not os.path.isfile(structure): out_log.info(' Selected PDB code: ' + prop['step1_mmbpdb']['pdb_code']) fu.create_dir(prop['step1_mmbpdb']['path']) pdb.MmbPdb().get_pdb(prop['step1_mmbpdb']['pdb_code'], paths['step1_mmbpdb']['output_pdb_path']) structure = paths['step1_mmbpdb']['output_pdb_path'] out_log.info('step2: p2g ------ Create gromacs topology') fu.create_dir(prop['step2_p2g']['path']) pdb2gmx.Pdb2gmx(properties=prop['step2_p2g'], **paths['step2_p2g']).launch() out_log.info('step3: ec ------- Define box dimensions') fu.create_dir(prop['step3_ec']['path']) editconf.Editconf(properties=prop['step3_ec'], **paths['step3_ec']).launch() out_log.info('step4: sol ------ Fill the box with water molecules') fu.create_dir(prop['step4_sol']['path']) solvate.Solvate(properties=prop['step4_sol'], **paths['step4_sol']).launch() out_log.info('step5: gppions -- Preprocessing: Adding monoatomic ions') fu.create_dir(prop['step5_gppions']['path']) grompp.Grompp(properties=prop['step5_gppions'], **paths['step5_gppions']).launch() out_log.info('step6: gio ------ Running: Adding monoatomic ions') fu.create_dir(prop['step6_gio']['path']) genion.Genion(properties=prop['step6_gio'], **paths['step6_gio']).launch() out_log.info('step7: gppmin --- Preprocessing: Energy minimization') fu.create_dir(prop['step7_gppmin']['path']) grompp.Grompp(properties=prop['step7_gppmin'], **paths['step7_gppmin']).launch() out_log.info('step8: mdmin ---- Running: Energy minimization') fu.create_dir(prop['step8_mdmin']['path']) mdrun.Mdrun(properties=prop['step8_mdmin'], **paths['step8_mdmin']).launch() out_log.info( 'step9: gppnvt --- Preprocessing: nvt constant number of molecules, volume and temp' ) fu.create_dir(prop['step9_gppnvt']['path']) grompp.Grompp(properties=prop['step9_gppnvt'], **paths['step9_gppnvt']).launch() out_log.info( 'step10: mdnvt ---- Running: nvt constant number of molecules, volume and temp' ) fu.create_dir(prop['step10_mdnvt']['path']) mdrun.Mdrun(properties=prop['step10_mdnvt'], **paths['step10_mdnvt']).launch() out_log.info( 'step11: gppnpt --- Preprocessing: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step11_gppnpt']['path']) grompp.Grompp(properties=prop['step11_gppnpt'], **paths['step11_gppnpt']).launch() out_log.info( 'step12: mdnpt ---- Running: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step12_mdnpt']['path']) mdrun.Mdrun(properties=prop['step12_mdnpt'], **paths['step12_mdnpt']).launch() out_log.info( 'step13: gppeq ---- Preprocessing: Molecular dynamics Equilibration') fu.create_dir(prop['step13_gppeq']['path']) grompp.Grompp(properties=prop['step13_gppeq'], **paths['step13_gppeq']).launch() out_log.info( 'step14: mdeq ----- Running: Molecular dynamics Equilibration') fu.create_dir(prop['step14_mdeq']['path']) mdrun.Mdrun(properties=prop['step14_mdeq'], **paths['step14_mdeq']).launch() #Create setupfiles dir and copy files setupfiles_path = os.path.join(workflow_path, 'setupfiles') fu.create_dir(setupfiles_path) shutil.copy(paths['step14_mdeq']['output_gro_path'], os.path.join(setupfiles_path, 'md_setup.gro')) shutil.copy(paths['step14_mdeq']['output_cpt_path'], os.path.join(setupfiles_path, 'md_setup.cpt')) shutil.copy(paths['step6_gio']['output_top_zip_path'], os.path.join(setupfiles_path, 'md_setup.zip')) removed_list = fu.remove_temp_files([ '#', '.top', '.plotscript', '.edr', '.xtc', '.itp', '.top', '.log', '.pdb', '.cpt', '.mdp' ]) out_log.info('') out_log.info('Removing unwanted files: ') for removed_file in removed_list: out_log.info(' X ' + removed_file) elapsed_time = time.time() - start_time out_log.info('') out_log.info('') out_log.info('Execution sucessful: ') out_log.info(' Workflow_path: ' + workflow_path) out_log.info(' Config File: ' + yaml_path) out_log.info(' System: ' + system) if len(sys.argv) >= 4: out_log.info(' Nodes: ' + sys.argv[3]) out_log.info('') out_log.info('Elapsed time: ' + str(elapsed_time) + ' seconds') out_log.info('')
def main(): start_time = time.time() structure_pdb_path_in = os.path.abspath(sys.argv[1]) structure_pdb_path_out = os.path.abspath(sys.argv[2]) yaml_path = '/home/user/pymdsetup/workflows/conf/conf_mug_refinement.yaml' system = 'mug_pymdsetup' conf = settings.YamlReader(yaml_path, system) workflow_path = conf.properties[system]['workflow_path'] fu.create_dir(os.path.abspath(workflow_path)) out_log, _ = fu.get_logs(path=workflow_path, console=True) paths = conf.get_paths_dic() prop = conf.get_prop_dic(global_log=out_log) #TODO: Source of problems #Change directories always creates problems os.chdir(workflow_path) out_log.info('\n\n_______MUG REFINEMENT_______\n\n') out_log.info('in ----------- Get PDB structure') fu.create_dir(prop['step1_mmbpdb']['path']) shutil.copy(structure_pdb_path_in, paths['step1_mmbpdb']['output_pdb_path']) out_log.info('sed ---------- Replacing atom names') sed_path = opj(workflow_path, 'step2_sed') fu.create_dir(sed_path) sed_pdb_path = opj(sed_path, 'sed_replaced.pdb') shutil.copy(paths['step1_mmbpdb']['output_pdb_path'], sed_pdb_path) cmd = ['sed', '-i', "'s/ O1P / OP1 /g'", sed_pdb_path] sed1_out_log, sed1_err_log = fu.get_logs(path=sed_path, step='step2_sed1') command = cmd_wrapper.CmdWrapper(cmd, sed1_out_log, sed1_err_log) command.launch() cmd = ['sed', '-i', "'s/ O2P / OP2 /g'", sed_pdb_path] sed2_out_log, sed2_err_log = fu.get_logs(path=sed_path, step='step2_sed2') command = cmd_wrapper.CmdWrapper(cmd, sed2_out_log, sed2_err_log) command.launch() out_log.info('pdb2gmx ------ Create gromacs topology') fu.create_dir(prop['step4_p2g']['path']) pdb2gmx.Pdb2gmx(properties=prop['step4_p2g'], input_structure_pdb_path=sed_pdb_path, output_gro_path=paths['step4_p2g']['output_gro_path'], output_top_zip_path=paths['step4_p2g'] ['output_top_zip_path']).launch() out_log.info('editconf ----- Define box dimensions') fu.create_dir(prop['step5_ec']['path']) editconf.Editconf(properties=prop['step5_ec'], **paths['step5_ec']).launch() out_log.info('solvate ------ Fill the box with water molecules') fu.create_dir(prop['step6_sol']['path']) solvate.Solvate(properties=prop['step6_sol'], **paths['step6_sol']).launch() out_log.info('grompp_ions -- Preprocessing: Adding monoatomic ions') fu.create_dir(prop['step7_gppions']['path']) grompp.Grompp(properties=prop['step7_gppions'], **paths['step7_gppions']).launch() out_log.info('genion ------- Running: Adding monoatomic ions') fu.create_dir(prop['step8_gio']['path']) genion.Genion(properties=prop['step8_gio'], **paths['step8_gio']).launch() out_log.info('grompp_min --- Preprocessing: Energy minimization') fu.create_dir(prop['step9_gppmin']['path']) grompp.Grompp(properties=prop['step9_gppmin'], **paths['step9_gppmin']).launch() out_log.info('mdrun_min ---- Running: Energy minimization') fu.create_dir(prop['step10_mdmin']['path']) mdrun.Mdrun(properties=prop['step10_mdmin'], **paths['step10_mdmin']).launch() out_log.info( 'grompp_nvt --- Preprocessing: nvt constant number of molecules, volume and temp' ) fu.create_dir(prop['step11_gppnvt']['path']) grompp.Grompp(properties=prop['step11_gppnvt'], **paths['step11_gppnvt']).launch() out_log.info( 'mdrun_nvt ---- Running: nvt constant number of molecules, volume and temp' ) fu.create_dir(prop['step12_mdnvt']['path']) mdrun.Mdrun(properties=prop['step12_mdnvt'], **paths['step12_mdnvt']).launch() out_log.info( 'grompp_npt --- Preprocessing: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step13_gppnpt']['path']) grompp.Grompp(properties=prop['step13_gppnpt'], **paths['step13_gppnpt']).launch() out_log.info( 'mdrun_npt ---- Running: npt constant number of molecules, pressure and temp' ) fu.create_dir(prop['step14_mdnpt']['path']) mdrun.Mdrun(properties=prop['step14_mdnpt'], **paths['step14_mdnpt']).launch() out_log.info( 'grompp_eq ---- Preprocessing: 100ps Molecular dynamics Equilibration') fu.create_dir(prop['step15_gppeq']['path']) grompp.Grompp(properties=prop['step15_gppeq'], **paths['step15_gppeq']).launch() out_log.info( 'mdrun_eq ----- Running: 100ps Molecular dynamics Equilibration') fu.create_dir(prop['step16_mdeq']['path']) mdrun.Mdrun(properties=prop['step16_mdeq'], **paths['step16_mdeq']).launch() out_log.info('trjconv ------ Extract last snapshot') step17_path = opj(workflow_path, 'step17_trjconv') step17_index = opj(step17_path, 'step17_make_ndx.ndx') step17_pdb = opj(step17_path, 'step17_trjconv.pdb') fu.create_dir(step17_path) #TODO: source of problems # should create a wrapper for the make_ndx tool and call it using subprocess os.system('printf "! \\"Water_and_ions\\" \nq\n" | ' + prop['step16_mdeq']['gmx_path'] + ' make_ndx -f ' + paths['step16_mdeq']['output_gro_path'] + ' -o ' + step17_index + ' > ' + opj(step17_path, 'make_ndx.out') + ' 2> ' + opj(step17_path, 'make_ndx.err')) cmd = [ 'echo', 'Protein_DNA', '|', prop['step16_mdeq']['gmx_path'], "trjconv", "-s", paths['step15_gppeq']['output_tpr_path'], "-f", paths['step16_mdeq']['output_trr_path'], "-o", step17_pdb, "-n", step17_index, "-dump", '1' ] step17_out_log, step17_err_log = fu.get_logs(path=step17_path, step='step17_trjconv') command = cmd_wrapper.CmdWrapper(cmd, step17_out_log, step17_err_log) command.launch() elapsed_time = time.time() - start_time removed_list = fu.remove_temp_files([ '#', '.top', '.plotscript', '.edr', '.xtc', '.itp', '.top', '.log', '.pdb', '.cpt', '.mdp', '.ndx' ]) shutil.copy(step17_pdb, structure_pdb_path_out) out_log.info('') out_log.info('Removing unwanted files: ') for removed_file in removed_list: out_log.info(' X ' + removed_file) out_log.info('\n\nExecution sucessful: ') out_log.info(' Output_pdb_path: ' + structure_pdb_path_out) out_log.info(' Workflow_path: ' + workflow_path) out_log.info(' Config File: ' + yaml_path) out_log.info(' System: ' + system) out_log.info(' Elapsed time: ' + str(elapsed_time) + 'seconds')