Skip to content

Commit

Permalink
NF: Added actor for rhombicuboctahedron, TEST: Added uni tests for rh…
Browse files Browse the repository at this point in the history
…ombicuboctahedron.
  • Loading branch information
ManishReddyR committed Jan 28, 2025
1 parent 27200f2 commit 9ff2329
Show file tree
Hide file tree
Showing 2 changed files with 125 additions and 0 deletions.
66 changes: 66 additions & 0 deletions fury/actor.py
Original file line number Diff line number Diff line change
Expand Up @@ -569,3 +569,69 @@ def icosahedron(
material=material,
enable_picking=enable_picking,
)


def rhombicuboctahedron(
centers,
*,
directions=(0, 0, 0),
colors=(1, 1, 1),
scales=(1, 1, 1),
opacity=None,
material="phong",
enable_picking=True,
):
"""Visualize one or many rhombicuboctahedrons with different features.
Parameters
----------
centers : ndarray, shape (N, 3)
rhombicuboctahedron positions.
directions : ndarray, shape (N, 3), optional
The orientation vector of the rhombicuboctahedron.
colors : ndarray (N,3) or (N, 4) or tuple (3,) or tuple (4,), optional
RGB or RGBA (for opacity) R, G, B and A should be at the range [0, 1].
scales : int or ndarray (N,3) or tuple (3,), optional
The size of the rhombicuboctahedro in each dimension. If a single value is
provided, the same size will be used for all rhombicuboctahedron.
opacity : float, optional
Takes values from 0 (fully transparent) to 1 (opaque).
If both `opacity` and RGBA are provided, the final alpha will be:
final_alpha = alpha_in_RGBA * opacity
material : str, optional
The material type for the rhombicuboctahedrons. Options are 'phong' and 'basic'.
enable_picking : bool, optional
Whether the rhombicuboctahedrons should be pickable in a 3D scene.
Returns
-------
mesh_actor : Actor
A mesh actor containing the generated rhombicuboctahedrons, with the specified
material and properties.
Examples
--------
>>> from fury import window, actor
>>> import numpy as np
>>> scene = window.Scene()
>>> centers = np.random.rand(5, 3) * 10
>>> colors = np.random.rand(5, 3)
>>> rhombicuboctahedron_actor = actor.rhombicuboctahedron(
centers=centers, colors=colors
)
>>> scene.add(rhombicuboctahedron_actor)
>>> show_manager = window.ShowManager(scene=scene, size=(600, 600))
>>> show_manager.start()
"""
vertices, faces = fp.prim_rhombicuboctahedron()
return actor_from_primitive(
vertices,
faces,
centers=centers,
colors=colors,
scales=scales,
directions=directions,
opacity=opacity,
material=material,
enable_picking=enable_picking,
)
59 changes: 59 additions & 0 deletions fury/tests/test_actor.py
Original file line number Diff line number Diff line change
Expand Up @@ -343,3 +343,62 @@ def test_icosahedron():
assert g == 0 and b == 0
assert r == 255
scene.remove(icosahedron_actor_2)


def test_rhombicuboctahedron():
scene = window.Scene()
centers = np.array([[0, 0, 0]])
colors = np.array([[1, 0, 0]])

rhombicuboctahedron_actor = actor.rhombicuboctahedron(
centers=centers, colors=colors
)
scene.add(rhombicuboctahedron_actor)

npt.assert_array_equal(rhombicuboctahedron_actor.local.position, centers[0])

mean_vertex = np.mean(rhombicuboctahedron_actor.geometry.positions.view, axis=0)
npt.assert_array_almost_equal(mean_vertex, centers[0])

assert rhombicuboctahedron_actor.prim_count == 1

window.snapshot(scene=scene, fname="rhombicuboctahedron_test_1.png")

img = Image.open("rhombicuboctahedron_test_1.png")
img_array = np.array(img)

mean_r, mean_g, mean_b, mean_a = np.mean(
img_array.reshape(-1, img_array.shape[2]), axis=0
)

assert mean_r > mean_b and mean_r > mean_g

middle_pixel = img_array[img_array.shape[0] // 2, img_array.shape[1] // 2]
r, g, b, a = middle_pixel
assert r > g and r > b
assert g == b
scene.remove(rhombicuboctahedron_actor)

rhombicuboctahedron_actor_2 = actor.rhombicuboctahedron(
centers=centers, colors=colors, material="basic"
)
scene.add(rhombicuboctahedron_actor_2)
window.snapshot(scene=scene, fname="rhombicuboctahedron_test_2.png")

img = Image.open("rhombicuboctahedron_test_2.png")
img_array = np.array(img)

mean_r, mean_g, mean_b, mean_a = np.mean(
img_array.reshape(-1, img_array.shape[2]), axis=0
)

assert mean_r > mean_b and mean_r > mean_g
assert 0 < mean_r < 255
assert mean_g == 0 and mean_b == 0

middle_pixel = img_array[img_array.shape[0] // 2, img_array.shape[1] // 2]
r, g, b, a = middle_pixel
assert r > g and r > b
assert g == 0 and b == 0
assert r == 255
scene.remove(rhombicuboctahedron_actor_2)

0 comments on commit 9ff2329

Please sign in to comment.