Source code for gamdpy.interactions.potential_functions.apply_shifted_potential_cutoff

import numpy as np
import numba
import math
from numba import cuda



def apply_shifted_potential_cutoff(pair_potential: callable) -> callable:
    """ Apply shifted potential cutoff to a pair-potential function

        If the input pair potential is :math:`u(r)`,
        then the shifted potential is :math:`u(r) - u(r_{c})`, where :math:`r_c` is the cutoff distance.
        Calls the original potential function twice, avoiding changes to params.

        Parameters
        ----------

        pair_potential : callable
            A function that calculates a pair-potential:
            `u, s, umm =  pair_potential(dist, params)`

        Returns
        -------

        pair_potential : callable
            A function where shifted potentia cutoff is applied to the original function.

        Example
        -------

        Example demonstrating how to set up the Lennard-Jones 12-6 potential, :func:`~gamdpy.LJ_12_6`,
        truncated and shifted to zero at the cutoff distance of 2.5.

        >>> import gamdpy as gp
        >>> pair_func = gp.apply_shifted_force_cutoff(gp.LJ_12_6)
        >>> A12, A6, cut = 1.0, 1.0, 2.5
        >>> pair_pot = gp.PairPotential(pair_func, params=[A12, A6, cut], max_num_nbs=1000)
        >>> interactions = [pair_pot, ]  # List of interactions only containing the pair potential

    """
    pair_pot = numba.njit(pair_potential)

    @numba.njit
    def potential(dist, params): # pragma: no cover
        cut = params[-1]
        u, s, umm = pair_pot(dist, params)
        u_cut, s_cut, umm_cut = pair_pot(cut, params)
        u -= u_cut
        return u, s, umm

    return potential