for (formula, pes_idx, sub_pes_idx), rxn_lst in RUN_PES_DCT.items():
# Print PES form and SUB PES Channels
print('\nRunning PES {}: {}, SUB PES {}'.format(
pes_idx, formula, sub_pes_idx))
for rxn in rxn_lst:
print(' Running Channel {}: {} = {}'.format(
rxn['chn_idx'],
'+'.join(rxn['reacs']),
'+'.join(rxn['prods'])))
esdriver.run(
pes_idx,
rxn_lst,
SPC_DCT,
CLA_DCT,
ES_TSK_LST,
THY_DCT,
RUN_INP_DCT
)
else:
PES_IDX = 0
esdriver.run(
PES_IDX,
RUN_SPC_LST_DCT,
SPC_DCT,
CLA_DCT,
ES_TSK_LST,
THY_DCT,
RUN_INP_DCT
)
def _mechdriver_main(tmp_dir):
""" Copy of MechDriver bin
"""
# print header message and host name (probably combine into one function)
ioprinter.program_header('amech')
ioprinter.random_cute_animal()
ioprinter.host_name()
# parse all of the input
ioprinter.program_header('inp')
inp_strs = ioparser.read_amech_input(tmp_dir)
thy_dct = ioparser.thy.theory_dictionary(inp_strs['thy'])
kmod_dct, smod_dct = ioparser.models.models_dictionary(
inp_strs['mod'], thy_dct)
inp_key_dct = ioparser.run.input_dictionary(inp_strs['run'])
pes_idx_dct, spc_idx_dct = ioparser.run.chem_idxs(inp_strs['run'])
tsk_lst_dct = ioparser.run.tasks(
inp_strs['run'], inp_strs['mech'], thy_dct)
spc_dct, glob_dct = ioparser.spc.species_dictionary(
inp_strs['spc'], inp_strs['dat'], inp_strs['geo'], 'csv')
pes_dct = ioparser.mech.pes_dictionary(
inp_strs['mech'], 'chemkin', spc_dct)
pes_rlst, spc_rlst = ioparser.rlst.run_lst(
pes_dct, spc_dct, pes_idx_dct, spc_idx_dct)
# build the run-save filesystem directories
prefix_fs(inp_key_dct['run_prefix'], inp_key_dct['save_prefix'])
# run drivers requested by user
es_tsks = tsk_lst_dct.get('es')
if es_tsks is not None:
ioprinter.program_header('es')
esdriver.run(
pes_rlst, spc_rlst,
es_tsks,
spc_dct, glob_dct, thy_dct,
inp_key_dct['run_prefix'], inp_key_dct['save_prefix']
)
ioprinter.program_exit('es')
therm_tsks = tsk_lst_dct.get('thermo')
if therm_tsks is not None:
ioprinter.program_header('thermo')
thermodriver.run(
pes_rlst, spc_rlst,
therm_tsks,
kmod_dct, smod_dct,
spc_dct, thy_dct,
inp_key_dct['run_prefix'], inp_key_dct['save_prefix'], tmp_dir
)
ioprinter.program_exit('thermo')
trans_tsks = tsk_lst_dct.get('trans')
if trans_tsks is not None:
ioprinter.program_header('trans')
if pes_dct:
transdriver.run(
pes_rlst, spc_rlst,
trans_tsks,
smod_dct,
spc_dct, thy_dct,
inp_key_dct['run_prefix'], inp_key_dct['save_prefix']
)
ioprinter.program_exit('trans')
ktp_tsks = tsk_lst_dct.get('ktp')
if ktp_tsks is not None:
ioprinter.program_header('ktp')
ktpdriver.run(
pes_rlst,
ktp_tsks,
spc_dct, glob_dct,
kmod_dct, smod_dct,
inp_key_dct['run_prefix'], inp_key_dct['save_prefix'], tmp_dir
)
ioprinter.program_exit('ktp')
proc_tsks = tsk_lst_dct.get('proc')
if proc_tsks is not None:
ioprinter.program_header('proc')
procdriver.run(
pes_rlst, spc_rlst,
proc_tsks,
spc_dct, thy_dct,
kmod_dct, smod_dct,
inp_key_dct['run_prefix'], inp_key_dct['save_prefix'], tmp_dir
)
ioprinter.program_exit('proc')
# exit program
ioprinter.obj('vspace')
ioprinter.program_exit('amech')
PES_DCT = ioparser.mech.pes_dictionary(
INP_STRS['mech'], 'chemkin', SPC_DCT)
PES_RLST, SPC_RLST = ioparser.rlst.run_lst(
PES_DCT, SPC_DCT, PES_IDX_DCT, SPC_IDX_DCT)
# Build the Run-Save Filesystem Directories
prefix_fs(INP_KEY_DCT['run_prefix'], INP_KEY_DCT['save_prefix'])
# Run Drivers Requested by User
ES_TSKS = TSK_LST_DCT.get('es')
if ES_TSKS is not None:
ioprinter.program_header('es')
esdriver.run(
PES_RLST, SPC_RLST,
ES_TSKS,
SPC_DCT, GLOB_DCT, THY_DCT,
INP_KEY_DCT['run_prefix'], INP_KEY_DCT['save_prefix']
)
ioprinter.program_exit('es')
THERM_TSKS = TSK_LST_DCT.get('thermo')
if THERM_TSKS is not None:
ioprinter.program_header('thermo')
thermodriver.run(
PES_RLST, SPC_RLST,
THERM_TSKS,
KMOD_DCT, SMOD_DCT,
SPC_DCT,
INP_KEY_DCT['run_prefix'], INP_KEY_DCT['save_prefix']
)
ioprinter.program_exit('thermo')
print('{}'.format(RUN_INP_DCT['run_prefix']))
print('{}'.format(RUN_INP_DCT['save_prefix']))
# Run the requested drivers: es, thermo, ktp
print('\n\nRunning the requested drivers...')
# need_ts = bool('find_ts' in es_tsk_lst)
if 'es' in RUN_JOBS_LST:
if RUN_OBJ_DCT['pes'] or RUN_OBJ_DCT['pspc']:
# Call ESDriver for spc in each PES
for pes, rxn_lst in RUN_PES_DCT.items():
# Get info for the transition states
ts_dct = loadspc.build_sadpt_dct(rxn_lst, MODEL_DCT, THY_DCT,
ES_TSK_STR, RUN_INP_DCT,
RUN_OPTIONS_DCT, SPC_DCT, {})
SPC_DCT.update(ts_dct)
esdriver.run(rxn_lst, SPC_DCT, ES_TSK_STR, MODEL_DCT, THY_DCT,
RUN_OPTIONS_DCT, RUN_INP_DCT)
else:
# Call ESDriver for all of the species
esdriver.run(RUN_SPC_LST_DCT, SPC_DCT, ES_TSK_STR, MODEL_DCT, THY_DCT,
RUN_OPTIONS_DCT, RUN_INP_DCT)
if 'pf' in RUN_JOBS_LST or 'thermo' in RUN_JOBS_LST:
if RUN_OBJ_DCT['pes'] or RUN_OBJ_DCT['pspc']:
# Call ThermoDriver for spc in each PES
for pes, rxn_lst in RUN_PES_DCT.items():
thermodriver.run(SPC_DCT,
MODEL_DCT,
THY_DCT,
rxn_lst,
RUN_INP_DCT,
ref_scheme='basic',