"""
Functions for advection
#######################
"""
from pysph.sph.equation import Equation
from numpy import cos
from numpy import pi as M_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(M_PI*t/self.T) * d_u0[d_idx]
d_v[d_idx] = -cos(M_PI*t/self.T) * d_v0[d_idx]