31 lines
937 B
Python
31 lines
937 B
Python
"""
|
|
Calculating the ISF of Water
|
|
=======================================================
|
|
|
|
In this example the ISF of water oxygens is calculated for a bulk simulation.
|
|
Additionally a KWW function is fitted to the results.
|
|
"""
|
|
from functools import partial
|
|
import matplotlib.pyplot as plt
|
|
from scipy.optimize import curve_fit
|
|
import mdevaluate as md
|
|
import tudplot
|
|
|
|
OW = md.open('/data/niels/sim/water/bulk/260K', trajectory='out/*.xtc').subset(atom_name='OW')
|
|
t, S = md.correlation.shifted_correlation(
|
|
partial(md.correlation.isf, q=22.7),
|
|
OW,
|
|
average=True
|
|
)
|
|
# Only include data-points of the alpha-relaxation for the fit
|
|
mask = t > 3e-1
|
|
fit, cov = curve_fit(md.functions.kww, t[mask], S[mask])
|
|
tau = md.functions.kww_1e(*fit)
|
|
|
|
tudplot.activate()
|
|
plt.figure()
|
|
plt.plot(t, S, '.', label='ISF of Bulk Water')
|
|
plt.plot(t, md.functions.kww(t, *fit), '-', label=r'KWW, $\tau$={:.2f}ps'.format(tau))
|
|
plt.xscale('log')
|
|
plt.legend()
|