Add tabulate to backend surfaces

JD557 · Jan 5, 2025 · f0ddadd · f0ddadd
1 parent a4da678
commit f0ddadd
Show file tree

Hide file tree

Showing 3 changed files with 70 additions and 0 deletions.
diff --git a/backend/js/src/main/scala/eu/joaocosta/minart/backend/ImageDataOpaqueSurface.scala b/backend/js/src/main/scala/eu/joaocosta/minart/backend/ImageDataOpaqueSurface.scala
@@ -80,6 +80,28 @@ final class ImageDataOpaqueSurface(val data: ImageData) extends MutableSurface {
 
 object ImageDataOpaqueSurface {
 
+  /** Produces an opaque ImageData backed 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): ImageDataOpaqueSurface = {
+    val surface =
+      ImageDataOpaqueSurface.fromImage(new Image(width, height))
+    var y = 0
+    while (y < height) {
+      var x = 0
+      while (x < width) {
+        surface.unsafePutPixel(x, y, f(x, y))
+        x += 1
+      }
+      y += 1
+    }
+    surface
+  }
+
   /** Loads an ImageDataOpaqueSurface from an offscreen canvas
     *
     * @param canvas offscreen canvas

diff --git a/backend/js/src/main/scala/eu/joaocosta/minart/backend/ImageDataSurface.scala b/backend/js/src/main/scala/eu/joaocosta/minart/backend/ImageDataSurface.scala
@@ -78,6 +78,28 @@ final class ImageDataSurface(val data: ImageData) extends MutableSurface {
 
 object ImageDataSurface {
 
+  /** Produces an ImageData backed 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): ImageDataSurface = {
+    val surface =
+      ImageDataSurface.fromImage(new Image(width, height))
+    var y = 0
+    while (y < height) {
+      var x = 0
+      while (x < width) {
+        surface.unsafePutPixel(x, y, f(x, y))
+        x += 1
+      }
+      y += 1
+    }
+    surface
+  }
+
   /** Loads an ImageDataOpaqueSurface from an offscreen canvas
     *
     * @param canvas offscreen canvas

diff --git a/backend/jvm/src/main/scala/eu/joaocosta/minart/backend/BufferedImageSurface.scala b/backend/jvm/src/main/scala/eu/joaocosta/minart/backend/BufferedImageSurface.scala
@@ -67,3 +67,29 @@ final class BufferedImageSurface(val bufferedImage: BufferedImage) extends Mutab
         super.blit(that, blendMode)(x, y, cx, cy, cw, ch)
     }
 }
+
+object BufferedImageSurface {
+
+  /** Produces a BufferedImage backed 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): BufferedImageSurface = {
+    val surface =
+      new BufferedImageSurface(new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB))
+    var y = 0
+    while (y < height) {
+      var x = 0
+      while (x < width) {
+        surface.unsafePutPixel(x, y, f(x, y))
+        x += 1
+      }
+      y += 1
+    }
+    surface
+  }
+
+}