def test_get_value_3d_nd(
ray_start, ray_direction, expected_value, expected_index
):
vertices = np.array(
[
[0, 3, 0, 0],
[0, 3, 0, 3],
[0, 3, 3, 0],
[0, 5, 0, 0],
[0, 5, 0, 3],
[0, 5, 3, 0],
[0, 2, 50, 50],
[0, 2, 50, 100],
[0, 2, 100, 50],
]
)
faces = np.array([[0, 1, 2], [3, 4, 5], [6, 7, 8]])
values = np.array([1, 2, 3, 1, 2, 3, 1, 2, 3])
surface_layer = Surface((vertices, faces, values))
surface_layer._slice_dims([0, 0, 0, 0], ndisplay=3)
value, index = surface_layer.get_value(
position=ray_start,
view_direction=ray_direction,
dims_displayed=[1, 2, 3],
world=False,
)
assert index == expected_index
np.testing.assert_allclose(value, expected_value)
class Surface2DSuite:
"""Benchmarks for the Surface layer with 2D data"""
params = [2**i for i in range(4, 18, 2)]
def setup(self, n):
np.random.seed(0)
self.data = (
np.random.random((n, 2)),
np.random.randint(n, size=(n, 3)),
np.random.random(n),
)
self.layer = Surface(self.data)
def time_create_layer(self, n):
"""Time to create an image layer."""
Surface(self.data)
def time_refresh(self, n):
"""Time to refresh view."""
self.layer.refresh()
def time_set_view_slice(self, n):
"""Time to set view slice."""
self.layer._set_view_slice()
def time_update_thumbnail(self, n):
"""Time to update thumbnail."""
self.layer._update_thumbnail()
def time_get_value(self, n):
"""Time to get current value."""
self.layer.get_value((0, ) * 2)
def mem_layer(self, n):
"""Memory used by layer."""
return self.layer
def mem_data(self, n):
"""Memory used by raw data."""
return self.data
