Water Wave

wave_coord

from pynodes import *
from pynodes.math import *

@tree
def wave_coord(mesh: Mesh, h: Float = 2, Ω: Float = 2, θ: Float = 0):

    d = mesh.position.length

    with frame("z = h * sin(d * Ω - θ) / d"):

        z = h * sin(d * Ω - θ) / d

    return CombineXYZ(mesh.position.x, mesh.position.y, z)

grid_vertex_select

@tree
def grid_vertex_select(grid: Mesh, slice: Boolean):

    with frame("Selection"):

        index = grid.index

        with frame("Index >= 2000"):

            selection1 = grid[20000:].selection

        with frame("Index % 2 == 0 if slice"):

            selection2 = grid[Boolean.true.switch(slice, (index % 2) == 0)].selection

    return selection1, selection2

water_wave

@tree
def water_wave(
    l: Integer = ("Grid Size", 20),
    n: Integer = ("Grid Subdivision", 200),
    h: Float = ("Height", 2, -10, 10),
    Ω: Float = 2,
    θ: Float = ("Phase", 0),
    slice: Boolean = False
):

    grid = MeshGrid(l, l, n, n).mesh

    θ = sin(SceneTime().seconds) * tau

    pos = wave_coord(grid, h, Ω, θ)

    sel1, sel2 = grid_vertex_select(grid, slice)

    grid[sel1].set_position(pos)

    return grid.set_shade_smooth(slice)