Spaces:

SmaugC137
/

splasher

Sleeping

splasher / tests /test_projection.py

hf: sync with main — procedural natural-terrain demo scene

5c72cc9 3 months ago

2.56 kB

"""Headless tests for the BEV projection and rectangle selection."""

import numpy as np

from splasher import Grid
from splasher.core.projection import (
    bev_count,
    bev_image,
    bev_max_height,
    cells_in_rect,
    points_in_rect,
)


def _grid():
    return Grid(0.0, 4.0, 0.0, 4.0, 1.0)  # 4x4


def test_bev_max_height_and_empty():
    g = _grid()
    pts = np.array([[0.5, 0.5, 1.0], [0.6, 0.6, 3.0], [3.5, 3.5, 2.0]])
    h = bev_max_height(pts, g)
    assert h.shape == (4, 4)
    assert h[0, 0] == 3.0  # max of the two points in cell (0,0)
    assert h[3, 3] == 2.0
    assert np.isnan(h[2, 2])  # empty cell


def test_bev_count():
    g = _grid()
    pts = np.array([[0.5, 0.5, 0.0], [0.6, 0.6, 0.0]])
    c = bev_count(pts, g)
    assert c[0, 0] == 2.0
    assert np.isnan(c[1, 1])


def test_bev_field_colors_by_feature():
    from splasher.core.projection import bev_field

    g = _grid()
    # columns: [x, y, z, intensity, range] — the range of the last point is NaN (feature absent)
    pts = np.array([
        [0.5, 0.5, 0.0, 10.0, 1.0],
        [0.6, 0.6, 0.0, 20.0, 3.0],
        [3.5, 3.5, 0.0, 5.0, np.nan],
    ])
    names = ["intensity", "range"]

    inten = bev_field(pts, g, "intensity", names)
    assert inten[0, 0] == 15.0            # mean of 10 and 20 in cell (0,0)
    assert inten[3, 3] == 5.0
    assert np.isnan(inten[2, 2])          # empty cell

    rng = bev_field(pts, g, "range", names)
    assert rng[0, 0] == 2.0               # mean of 1 and 3
    assert np.isnan(rng[3, 3])            # only point there has a NaN range → excluded → NaN cell

    # unknown / absent feature falls back to height (max z)
    assert np.array_equal(bev_field(pts, g, "nope", names),
                          bev_field(pts, g, "height", names), equal_nan=True)


def test_bev_image_alpha():
    g = _grid()
    pts = np.array([[0.5, 0.5, 1.0]])
    img = bev_image(bev_max_height(pts, g))
    assert img.shape == (4, 4, 4)
    assert img[0, 0, 3] > 0  # filled cell, opaque
    assert img[2, 2, 3] == 0  # empty cell, transparent


def test_cells_in_rect():
    g = _grid()
    si, sj = cells_in_rect((1.2, 0.5, 2.9, 3.4), g)
    assert (si.start, si.stop) == (0, 4)
    assert (sj.start, sj.stop) == (1, 3)


def test_points_in_rect():
    xy = np.array([[0.0, 0.0], [2.0, 2.0], [5.0, 5.0]])
    mask = points_in_rect(xy, (1.0, 1.0, 3.0, 3.0))
    assert mask.tolist() == [False, True, False]
    # corner order does not matter
    assert points_in_rect(xy, (3.0, 3.0, 1.0, 1.0)).tolist() == [False, True, False]

	"""Headless tests for the BEV projection and rectangle selection."""

	import numpy as np

	from splasher import Grid
	from splasher.core.projection import (
	bev_count,
	bev_image,
	bev_max_height,
	cells_in_rect,
	points_in_rect,
	)


	def _grid():
	return Grid(0.0, 4.0, 0.0, 4.0, 1.0) # 4x4


	def test_bev_max_height_and_empty():
	g = _grid()
	pts = np.array([[0.5, 0.5, 1.0], [0.6, 0.6, 3.0], [3.5, 3.5, 2.0]])
	h = bev_max_height(pts, g)
	assert h.shape == (4, 4)
	assert h[0, 0] == 3.0 # max of the two points in cell (0,0)
	assert h[3, 3] == 2.0
	assert np.isnan(h[2, 2]) # empty cell


	def test_bev_count():
	g = _grid()
	pts = np.array([[0.5, 0.5, 0.0], [0.6, 0.6, 0.0]])
	c = bev_count(pts, g)
	assert c[0, 0] == 2.0
	assert np.isnan(c[1, 1])


	def test_bev_field_colors_by_feature():
	from splasher.core.projection import bev_field

	g = _grid()
	# columns: [x, y, z, intensity, range] — the range of the last point is NaN (feature absent)
	pts = np.array([
	[0.5, 0.5, 0.0, 10.0, 1.0],
	[0.6, 0.6, 0.0, 20.0, 3.0],
	[3.5, 3.5, 0.0, 5.0, np.nan],
	])
	names = ["intensity", "range"]

	inten = bev_field(pts, g, "intensity", names)
	assert inten[0, 0] == 15.0 # mean of 10 and 20 in cell (0,0)
	assert inten[3, 3] == 5.0
	assert np.isnan(inten[2, 2]) # empty cell

	rng = bev_field(pts, g, "range", names)
	assert rng[0, 0] == 2.0 # mean of 1 and 3
	assert np.isnan(rng[3, 3]) # only point there has a NaN range → excluded → NaN cell

	# unknown / absent feature falls back to height (max z)
	assert np.array_equal(bev_field(pts, g, "nope", names),
	bev_field(pts, g, "height", names), equal_nan=True)


	def test_bev_image_alpha():
	g = _grid()
	pts = np.array([[0.5, 0.5, 1.0]])
	img = bev_image(bev_max_height(pts, g))
	assert img.shape == (4, 4, 4)
	assert img[0, 0, 3] > 0 # filled cell, opaque
	assert img[2, 2, 3] == 0 # empty cell, transparent


	def test_cells_in_rect():
	g = _grid()
	si, sj = cells_in_rect((1.2, 0.5, 2.9, 3.4), g)
	assert (si.start, si.stop) == (0, 4)
	assert (sj.start, sj.stop) == (1, 3)


	def test_points_in_rect():
	xy = np.array([[0.0, 0.0], [2.0, 2.0], [5.0, 5.0]])
	mask = points_in_rect(xy, (1.0, 1.0, 3.0, 3.0))
	assert mask.tolist() == [False, True, False]
	# corner order does not matter
	assert points_in_rect(xy, (3.0, 3.0, 1.0, 1.0)).tolist() == [False, True, False]