From 0266f90fc223bd2f84494be161f0d5f33f3059d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Costa?= <jdpc557@gmail.com>
Date: Sat, 28 Dec 2024 16:45:40 +0000
Subject: [PATCH] Add Surface#tabulate helpers

---
 .../minart/graphics/MutableSurface.scala      | 13 ++++++++
 .../eu/joaocosta/minart/graphics/Plane.scala  |  2 +-
 .../minart/graphics/RamSurface.scala          | 31 +++++++++++++++++++
 .../joaocosta/minart/graphics/Surface.scala   | 13 ++++++++
 4 files changed, 58 insertions(+), 1 deletion(-)

diff --git a/core/shared/src/main/scala/eu/joaocosta/minart/graphics/MutableSurface.scala b/core/shared/src/main/scala/eu/joaocosta/minart/graphics/MutableSurface.scala
index 6fb24ef0..3f149d5f 100644
--- a/core/shared/src/main/scala/eu/joaocosta/minart/graphics/MutableSurface.scala
+++ b/core/shared/src/main/scala/eu/joaocosta/minart/graphics/MutableSurface.scala
@@ -169,3 +169,16 @@ trait MutableSurface extends Surface {
     blit(f(this.view).toRamSurface())(0, 0)
   }
 }
+
+object MutableSurface {
+
+  /** Produces a mutable surface containing values of a given function
+    *  over ranges of integer values starting from 0.
+    *
+    *  @param width the surface width
+    *  @param height the surface height
+    *  @param f the function computing the element values
+    */
+  def tabulate(width: Int, height: Int)(f: (Int, Int) => Color): MutableSurface =
+    RamSurface.tabulate(width, height)(f)
+}
diff --git a/core/shared/src/main/scala/eu/joaocosta/minart/graphics/Plane.scala b/core/shared/src/main/scala/eu/joaocosta/minart/graphics/Plane.scala
index 2af4a78c..b939f719 100644
--- a/core/shared/src/main/scala/eu/joaocosta/minart/graphics/Plane.scala
+++ b/core/shared/src/main/scala/eu/joaocosta/minart/graphics/Plane.scala
@@ -226,7 +226,7 @@ trait Plane extends Function2[Int, Int, Color] { outer =>
     * @param height surface height
     */
   final def toRamSurface(width: Int, height: Int): RamSurface =
-    toSurfaceView(width, height).toRamSurface()
+    RamSurface.tabulate(width, height)(getPixel)
 }
 
 object Plane {
diff --git a/core/shared/src/main/scala/eu/joaocosta/minart/graphics/RamSurface.scala b/core/shared/src/main/scala/eu/joaocosta/minart/graphics/RamSurface.scala
index db7ca7c5..0f8998bf 100644
--- a/core/shared/src/main/scala/eu/joaocosta/minart/graphics/RamSurface.scala
+++ b/core/shared/src/main/scala/eu/joaocosta/minart/graphics/RamSurface.scala
@@ -40,3 +40,34 @@ final class RamSurface(val dataBuffer: Vector[Array[Color]]) extends MutableSurf
     }
   }
 }
+
+object RamSurface {
+
+  /** Produces a RAM surface containing values of a given function
+    *  over ranges of integer values starting from 0.
+    *
+    *  @param width the surface width
+    *  @param height the surface height
+    *  @param f the function computing the element values
+    */
+  def tabulate(width: Int, height: Int)(f: (Int, Int) => Color): RamSurface = {
+    val b = Vector.newBuilder[Array[Color]]
+    b.sizeHint(height)
+    var y = 0
+    while (y < height) {
+      if (width <= 0) {
+        b += Array.empty[Color]
+      } else {
+        val array = new Array[Color](width)
+        var x     = 0
+        while (x < width) {
+          array(x) = f(x, y)
+          x += 1
+        }
+        b += array
+      }
+      y += 1
+    }
+    new RamSurface(b.result())
+  }
+}
diff --git a/core/shared/src/main/scala/eu/joaocosta/minart/graphics/Surface.scala b/core/shared/src/main/scala/eu/joaocosta/minart/graphics/Surface.scala
index c23b1ee7..76ea6488 100644
--- a/core/shared/src/main/scala/eu/joaocosta/minart/graphics/Surface.scala
+++ b/core/shared/src/main/scala/eu/joaocosta/minart/graphics/Surface.scala
@@ -82,3 +82,16 @@ trait Surface {
   /** Copies this surface into a new surface stored in RAM. */
   final def toRamSurface(): RamSurface = new RamSurface(getPixels())
 }
+
+object Surface {
+
+  /** Produces a surface containing values of a given function
+    *  over ranges of integer values starting from 0.
+    *
+    *  @param width the surface width
+    *  @param height the surface height
+    *  @param f the function computing the element values
+    */
+  def tabulate(width: Int, height: Int)(f: (Int, Int) => Color): Surface =
+    MutableSurface.tabulate(width, height)(f)
+}