"""
Line colormapping
=================

Nearly all objects in Pygfx can have a colormap. And the texcoords
can also be scaled using ``material.maprange``. This is useful
in scientific applications, where texcoords can e.g. mean temperature of force.

The same approach can be used in points and meshes (and images and
volumes, where this concept is commonly known as contrast limits.)
"""

# sphinx_gallery_pygfx_docs = 'screenshot'
# sphinx_gallery_pygfx_test = 'compare'

import numpy as np
from rendercanvas.auto import RenderCanvas, loop
import pygfx as gfx


canvas = RenderCanvas(size=(1000, 1000))
renderer = gfx.WgpuRenderer(canvas)

x = np.linspace(0, 4 * np.pi, 1000)
y = np.sin(x)


positions = np.array([x * 100, y * 100, np.zeros_like(x)], np.float32).T.copy()
mapcoords = y.astype(np.float32) * 8  # so values are in [-8, 8]

geometry = gfx.Geometry(positions=positions, texcoords=mapcoords)

line1 = gfx.Line(
    geometry,
    gfx.LineMaterial(
        thickness=20,
        color_mode="vertex_map",
        map=gfx.cm.viridis,
        maprange=None,  # default (0, 1)
    ),
)

line2 = gfx.Line(
    geometry,
    gfx.LineMaterial(
        thickness=20,
        map=gfx.cm.viridis,
        maprange=(-8, 8),  # full range
    ),
)
line3 = gfx.Line(
    geometry,
    gfx.LineMaterial(
        thickness=20,
        color_mode="vertex_map",
        map=gfx.cm.viridis,
        maprange=(-12, 12),  # zoomed out
    ),
)

line4 = gfx.Line(
    geometry,
    gfx.LineMaterial(
        thickness=20,
        color_mode="vertex_map",
        map=gfx.cm.viridis,
        maprange=(0, 10),  # Only positive half
    ),
)

for line in (line1, line2, line3, line4):
    line.material.thickness_space = "screen"

line1.local.position = 0, 0, 0
line2.local.position = 0, -500, 0
line3.local.position = 0, -1000, 0
line4.local.position = 0, -1500, 0

scene = gfx.Scene()
scene.add(line1, line2, line3, line4)

camera = gfx.OrthographicCamera()
camera.show_object(scene)

controller = gfx.OrbitController(camera, register_events=renderer)

canvas.request_draw(lambda: renderer.render(scene, camera))

if __name__ == "__main__":
    print(__doc__)
    loop.run()