49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
|
from python.ste import *
|
||
|
|
from python.helpers import *
|
||
|
|
|
||
|
|
# Simulation parameter
|
||
|
|
motion = 'IsotropicAngle'
|
||
|
|
distribution = 'Delta'
|
||
|
|
parameter = {
|
||
|
|
"angle": [3, 10, 30, 109.4],
|
||
|
|
}
|
||
|
|
|
||
|
|
parameter = prepare_rw_parameter(parameter)
|
||
|
|
|
||
|
|
fig_tau_cc, ax_tau_cc = plt.subplots()
|
||
|
|
ax_tau_cc.set_title('tau_cc')
|
||
|
|
|
||
|
|
fig_beta_cc, ax_beta_cc = plt.subplots()
|
||
|
|
ax_beta_cc.set_title('beta_cc')
|
||
|
|
|
||
|
|
fig_finfty_cc, ax_finfty_cc = plt.subplots()
|
||
|
|
ax_finfty_cc.set_title('f_infty_cc')
|
||
|
|
|
||
|
|
fig_tau_ss, ax_tau_ss = plt.subplots()
|
||
|
|
ax_tau_ss.set_title('tau_cc')
|
||
|
|
|
||
|
|
fig_beta_ss, ax_beta_ss = plt.subplots()
|
||
|
|
ax_beta_ss.set_title('beta_cc')
|
||
|
|
|
||
|
|
fig_finfty_ss, ax_finfty_ss = plt.subplots()
|
||
|
|
ax_finfty_ss.set_title('f_infty_cc')
|
||
|
|
|
||
|
|
for variation in parameter:
|
||
|
|
print(f"\nRun RW for {motion}/{distribution} with arguments {variation}\n")
|
||
|
|
run_sims(motion, distribution, ste=True, spectrum=True, **variation)
|
||
|
|
|
||
|
|
conf_file = find_config_file(variation)
|
||
|
|
print(conf_file)
|
||
|
|
|
||
|
|
vary_string, tau_cc, beta_cc, finfty_cc, tau_ss, beta_ss, finfty_ss = fit_and_save_ste(conf_file, plot_decays=False, verbose=False)
|
||
|
|
|
||
|
|
ax_tau_cc.semilogy(*tau_cc.T, label=vary_string)
|
||
|
|
ax_beta_cc.plot(*beta_cc.T, label=vary_string)
|
||
|
|
ax_finfty_cc.plot(*finfty_cc.T, label=vary_string)
|
||
|
|
ax_tau_ss.semilogy(*tau_ss.T, label=vary_string)
|
||
|
|
ax_beta_ss.plot(*beta_ss.T, label=vary_string)
|
||
|
|
ax_finfty_ss.plot(*finfty_ss.T, label=vary_string)
|
||
|
|
|
||
|
|
for ax in [ax_tau_cc, ax_beta_cc, ax_finfty_cc, ax_tau_ss, ax_beta_ss, ax_finfty_ss]:
|
||
|
|
ax.legend()
|
||
|
|
plt.show()
|