fastplotlib/examples/mesh/surface_height.py at main · fastplotlib/fastplotlib

35 lines (24 loc) · 805 Bytes

"""

Simple surface

==============

Example showing a surface mesh

"""

# test_example = true

# sphinx_gallery_pygfx_docs = 'screenshot'

import fastplotlib as fpl

import numpy as np

import pygfx as gfx

figure = fpl.Figure(size=(700, 560), cameras="3d", controller_types="orbit")

t = np.linspace(0, 6, 100).astype(np.float32)

x = np.sin(t)

y = np.cos(t * 2)

z = (x.reshape(1, -1) * x.reshape(-1, 1)) * 50 # 100x100

surface = figure[0, 0].add_surface(z, cmap="bwr")

# figure[0, 0].axes.grids.xy.visible = True

figure[0, 0].camera.show_object(surface.world_object, (-2, 2, -3), up=(0, 0, 1))

figure.show()

# NOTE: fpl.loop.run() should not be used for interactive sessions

# See the "JupyterLab and IPython" section in the user guide

if __name__ == "__main__":

print(__doc__)

fpl.loop.run()

Provide feedback