DOCS: updated examples and tests to adapt to impostor sphere

Author: m-agour

docs/examples/viz_billboard_spheres.py

@@ -34,7 +34,11 @@
 # These are actual 3D mesh spheres with vertices and faces.
 
 geometry_spheres = actor.sphere(
-    centers=centers + np.array([[0, -2, 0]]), colors=colors, radii=radii, opacity=1
+    centers=centers + np.array([[0, -2, 0]]),
+    colors=colors,
+    radii=radii,
+    opacity=1,
+    impostor=False,
 )
 
 ###############################################################################

fury/actor/tests/test_curved.py

@@ -22,9 +22,7 @@
 def test_sphere():
     centers = np.array([[0, 0, 0]])
     colors = np.array([[1, 0, 0]])
-    validate_actors(
-        centers=centers, colors=colors, actor_type="sphere", impostor=False
-    )
+    validate_actors(centers=centers, colors=colors, actor_type="sphere", impostor=False)
 
 
 def test_cylinder():

fury/wgsl/billboard_render.wgsl

@@ -49,8 +49,17 @@ fn vs_main(in: VertexInput) -> Varyings {
     varyings.world_pos = vec3<f32>(world_pos);
 
     // Load color if available - colors are duplicated 6x like positions
-    let color = load_s_colors(billboard_index * 6);
-    varyings.color = vec4<f32>(color, 1.0);
+    $$ if color_buffer_channels == 4
+    varyings.color = vec4<f32>(load_s_colors(billboard_index * 6));
+    $$ elif color_buffer_channels == 3
+    varyings.color = vec4<f32>(load_s_colors(billboard_index * 6), 1.0);
+    $$ elif color_buffer_channels == 2
+    let cvalue = load_s_colors(billboard_index * 6);
+    varyings.color = vec4<f32>(cvalue.r, cvalue.r, cvalue.r, cvalue.g);
+    $$ elif color_buffer_channels == 1
+    let cvalue = load_s_colors(billboard_index * 6);
+    varyings.color = vec4<f32>(cvalue, cvalue, cvalue, 1.0);
+    $$ endif
 
     return varyings;
 }

fury/wgsl/billboard_sphere_render.wgsl

@@ -49,8 +49,17 @@ fn vs_main(in: VertexInput) -> Varyings {
     var varyings: Varyings;
     varyings.position = vec4<f32>(clip_pos);
     varyings.world_pos = vec3<f32>(world_pos);
-    let color = load_s_colors(billboard_index * 6);
-    varyings.color = vec4<f32>(color, 1.0);
+    $$ if color_buffer_channels == 4
+    varyings.color = vec4<f32>(load_s_colors(billboard_index * 6));
+    $$ elif color_buffer_channels == 3
+    varyings.color = vec4<f32>(load_s_colors(billboard_index * 6), 1.0);
+    $$ elif color_buffer_channels == 2
+    let cvalue = load_s_colors(billboard_index * 6);
+    varyings.color = vec4<f32>(cvalue.r, cvalue.r, cvalue.r, cvalue.g);
+    $$ elif color_buffer_channels == 1
+    let cvalue = load_s_colors(billboard_index * 6);
+    varyings.color = vec4<f32>(cvalue, cvalue, cvalue, 1.0);
+    $$ endif
     varyings.texcoord_vert = vec2<f32>(tex_coord);
     varyings.billboard_center = vec3<f32>(world_center.x, world_center.y, world_center.z);
     varyings.billboard_right = vec3<f32>(cam_right.x, cam_right.y, cam_right.z);