polygons

system/ui/lib/tests/test_shader_polygon.py

@@ -59,7 +59,8 @@ def test_triangulate_preserves_coordinates_in_contiguous_float_buffer(points):
 
 @pytest.mark.parametrize("use_gradient", [False, True])
 @pytest.mark.parametrize("point_count", [4, 5, 66])
-def test_draw_polygon_passes_native_vertices(monkeypatch, use_gradient, point_count):
+def test_draw_polygon_legacy_shader_fallback(monkeypatch, use_gradient, point_count):
+  monkeypatch.setattr(shader_polygon, "USE_VERTEX_GRADIENTS", False)
   points = np.arange(point_count * 4, dtype=np.float64).reshape(-1, 2)[::2]
   rect = rl.Rectangle(0, 0, 100, 100)
   color = rl.Color(10, 20, 30, 40)
@@ -84,3 +85,76 @@ def test_draw_polygon_passes_native_vertices(monkeypatch, use_gradient, point_co
 
   assert calls == (["initialize", ("configure", (state, None, gradient, rect)), ("begin", "shader"), "draw", "end"]
                    if use_gradient else ["draw"])
+
+
+@pytest.mark.parametrize("use_gradient", [False, True])
+@pytest.mark.parametrize("point_count", [4, 5, 66])
+def test_draw_polygon_native_vertex_gradients(monkeypatch, use_gradient, point_count):
+  monkeypatch.setattr(shader_polygon, "USE_VERTEX_GRADIENTS", True)
+  points = np.arange(point_count * 4, dtype=np.float64).reshape(-1, 2)[::2]
+  rect = rl.Rectangle(0, 0, 100, 100)
+  color = rl.Color(10, 20, 30, 40)
+  gradient = Gradient((0, 0), (1, 1), [color, rl.Color(*rl.WHITE)], [0, 1]) if use_gradient else None
+
+  raw_calls = []
+  emitted_vertices = []
+  emitted_colors = []
+  fake_raw = SimpleNamespace(
+    rlBegin=lambda m: raw_calls.append(("rlBegin", m)),
+    rlEnd=lambda: raw_calls.append("rlEnd"),
+    rlColor4ub=lambda r, g, b, a: emitted_colors.append((int(r), int(g), int(b), int(a))),
+    rlVertex2f=lambda x, y: emitted_vertices.append((float(x), float(y))),
+  )
+  monkeypatch.setattr(shader_polygon, "raw_rl", fake_raw)
+
+  draw_calls = []
+  def draw_strip(vertices, count, fill):
+    assert rl.ffi.typeof(vertices) == rl.ffi.typeof("Vector2 *")
+    coordinates = [[vertices[i].x, vertices[i].y] for i in range(count)]
+    np.testing.assert_array_equal(coordinates, triangulate(points))
+    assert fill == color
+    draw_calls.append("draw")
+
+  monkeypatch.setattr(rl, "draw_triangle_strip", draw_strip)
+
+  draw_polygon(rect, points, gradient=gradient, color=None if use_gradient else color)
+
+  if use_gradient:
+    strip = triangulate(points)
+    n = len(strip)
+    assert raw_calls == [("rlBegin", 4), "rlEnd"]
+    expected_vertex_count = (n - 2) * 3
+    assert len(emitted_vertices) == expected_vertex_count
+    assert len(emitted_colors) == expected_vertex_count
+    assert draw_calls == []
+  else:
+    assert draw_calls == ["draw"]
+    assert raw_calls == []
+
+
+def test_vertex_gradient_color_interpolation(monkeypatch):
+  monkeypatch.setattr(shader_polygon, "USE_VERTEX_GRADIENTS", True)
+  points = np.array([
+    [0.0, 100.0],
+    [0.0, 0.0],
+    [10.0, 0.0],
+    [10.0, 100.0],
+  ], dtype=np.float32)
+  rect = rl.Rectangle(0, 0, 10, 100)
+  color_start = rl.Color(255, 0, 0, 255)
+  color_end = rl.Color(0, 255, 0, 128)
+  gradient = Gradient((0.0, 1.0), (0.0, 0.0), [color_start, color_end], [0.0, 1.0])
+
+  emitted_colors = []
+  fake_raw = SimpleNamespace(
+    rlBegin=lambda m: None,
+    rlEnd=lambda: None,
+    rlColor4ub=lambda r, g, b, a: emitted_colors.append((int(r), int(g), int(b), int(a))),
+    rlVertex2f=lambda x, y: None,
+  )
+  monkeypatch.setattr(shader_polygon, "raw_rl", fake_raw)
+
+  draw_polygon(rect, points, gradient=gradient)
+
+  assert (255, 0, 0, 255) in emitted_colors
+  assert (0, 255, 0, 128) in emitted_colors