Source code for pyPRISM.closure.PercusYevick

#!python
from __future__ import division,print_function
from pyPRISM.closure.AtomicClosure import AtomicClosure
import numpy as np
[docs]class PercusYevick(AtomicClosure): r'''Percus Yevick closure evaluated in terms of a change of variables **Mathematial Definition** .. math:: c_{\alpha,\beta}(r) = (\exp(-U_{\alpha,\beta}(r)) - 1.0) (1.0 + \gamma_{\alpha,\beta}(r)) .. math:: \gamma_{\alpha,\beta}(r) = h_{\alpha,\beta}(r) - c_{\alpha,\beta}(r) **Variables Definitions** - :math:`h_{\alpha,\beta}(r)` Total correlation function value at distance :math:`r` between sites :math:`\alpha` and :math:`\beta`. - :math:`c_{\alpha,\beta}(r)` Direct correlation function value at distance :math:`r` between sites :math:`\alpha` and :math:`\beta`. - :math:`U_{\alpha,\beta}(r)` Interaction potential value at distance :math:`r` between sites :math:`\alpha` and :math:`\beta`. **Description** The Percus-Yevick (PY) is derived by expanding the exponential of the direct correlation function, :math:`c_{\alpha,\beta}(r)`, in powers of density shift from a reference state. See Reference [1] for a full derivation. The change of variables is necessary in order to use potentials with hard cores in the computational setting. Written in the standard form, this closure diverges with divergent potentials, which makes it impossible to numerically solve. This closure has been shown to be accurate for systems with hard cores (strongly repulsive at short distances) and when the potential is short ranged. References ---------- #. Hansen, J.P.; McDonald, I.R.; Theory of Simple Liquids; Chapter 4, Section 4; 4th Edition (2013), Elsevier [`link <https://www.sciencedirect.com/science/book/9780123870322>`__] Example ------- .. code-block:: python import pyPRISM sys = pyPRISM.System(['A','B']) sys.closure['A','A'] = pyPRISM.closure.PercusYevick() sys.closure['A','B'] = pyPRISM.closure.PercusYevick() sys.closure['B','B'] = pyPRISM.closure.HypernettedChain() # ** finish populating system object ** PRISM = sys.createPRISM() PRISM.solve() '''
[docs] def __init__(self,apply_hard_core=False): '''Contstructor Parameters ---------- apply_hard_core: bool If True, the total correlation function will be assumed to be -1 inside the core (:math:`r_{i,j}<(d_i + d_j)/2.0`) and the closure will not be applied in this region. ''' self.potential = None self.value = None self.sigma = None self.apply_hard_core=apply_hard_core
def __repr__(self): return '<AtomicClosure: PercusYevick>'
[docs] def calculate(self,r,gamma): '''Calculate direct correlation function based on supplied :math:`\gamma` Arguments --------- r: np.ndarray array of real-space values associated with :math:`\gamma` gamma: np.ndarray array of :math:`\gamma` values used to calculate the direct correlation function ''' assert self.potential is not None,'Potential for this closure is not set!' assert len(gamma) == len(self.potential),'Domain mismatch!' if self.apply_hard_core: assert self.sigma is not None, 'If apply_hard_core=True, sigma parameter must be set!' # apply hard core condition self.value = -1 - gamma # calculate closure outside hard core mask = r>self.sigma self.value[mask] = (np.exp(-self.potential[mask])-1.0)*(1.0+gamma[mask]) else: self.value = (np.exp(-self.potential)-1.0)*(1.0+gamma) return self.value
[docs]class PY(PercusYevick): '''Alias of PercusYevick''' pass