Source code for pysph.sph.misc.advection

"""
Functions for advection
#######################
"""

from pysph.sph.equation import Equation
from numpy import cos, pi

[docs]class Advect(Equation):
[docs] def loop(self, d_idx, d_ax, d_ay, d_u, d_v): d_ax[d_idx] = d_u[d_idx] d_ay[d_idx] = d_v[d_idx]
[docs]class MixingVelocityUpdate(Equation): def __init__(self, dest, sources, T): self.T = T super(MixingVelocityUpdate, self).__init__(dest, sources)
[docs] def loop(self, d_idx, d_u, d_v, d_u0, d_v0, t=0.1): d_u[d_idx] = cos(pi*t/self.T) * d_u0[d_idx] d_v[d_idx] = -cos(pi*t/self.T) * d_v0[d_idx]