From 5465cae38c1feb8e949ddcb3b4299894344b3053 Mon Sep 17 00:00:00 2001
From: MrJs <30782821+mrjschulte@users.noreply.github.com>
Date: Sun, 25 Feb 2024 07:49:58 -0800
Subject: [PATCH] Create "Checkerboard" node (#2606)

* Create "Checkerboard" node

This node creates a checkerboard with user definable colors and size for the squares.

* Update create_checkerboard.py - update floats to ints

Define ints instead of floats for h/w/size

* Update create_checkerboard.py - Adds RGBA Support

Now with RGBA support

* Update create_checkerboard.py - RGBA tweaks

Forgot a for second color.
---
 .../create_images/create_checkerboard.py      | 72 +++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 backend/src/packages/chaiNNer_standard/image/create_images/create_checkerboard.py

diff --git a/backend/src/packages/chaiNNer_standard/image/create_images/create_checkerboard.py b/backend/src/packages/chaiNNer_standard/image/create_images/create_checkerboard.py
new file mode 100644
index 000000000..2b102e892
--- /dev/null
+++ b/backend/src/packages/chaiNNer_standard/image/create_images/create_checkerboard.py
@@ -0,0 +1,72 @@
+from __future__ import annotations
+
+import numpy as np
+
+import navi
+from nodes.impl.color.color import Color
+from nodes.properties.inputs import ColorInput, NumberInput
+from nodes.properties.outputs import ImageOutput
+
+from .. import create_images_group
+
+
+@create_images_group.register(
+    schema_id="chainner:image:create_checkerboard",
+    name="Create Checkerboard",
+    description="Create a Checkerboard of specified dimensions filled with the given colors for the squares.",
+    icon="MdBorderAll",
+    inputs=[
+        NumberInput("Width", minimum=1, unit="px", default=1024),
+        NumberInput("Height", minimum=1, unit="px", default=1024),
+        ColorInput(
+            "Color 1", channels=[4], default=Color.bgra((0.75, 0.75, 0.75, 1.0))
+        ).with_id(2),
+        ColorInput(
+            "Color 2", channels=[4], default=Color.bgra((0.35, 0.35, 0.35, 1.0))
+        ).with_id(3),
+        NumberInput("Square Size", minimum=1, default=32),
+    ],
+    outputs=[
+        ImageOutput(
+            image_type=navi.Image(
+                width="Input0",
+                height="Input1",
+                channels=4,
+            ),
+        )
+    ],
+)
+def create_checkerboard_node(
+    width: int,
+    height: int,
+    color_1: Color,
+    color_2: Color,
+    square_size: int,
+) -> np.ndarray:
+    img = np.zeros(
+        (height, width, 4), dtype=np.float32
+    )  # Create a new buffer with all zeros
+
+    # Determine the number of squares in each direction
+    num_cols = (width + square_size - 1) // square_size
+    num_rows = (height + square_size - 1) // square_size
+
+    # Fill the checkerboard with alternating squares
+    for i in range(num_rows):
+        for j in range(num_cols):
+            if (i + j) % 2 == 0:
+                img[
+                    max(0, height - (i + 1) * square_size) : max(
+                        0, height - i * square_size
+                    ),
+                    j * square_size : min(width, (j + 1) * square_size),
+                ] = color_1.value
+            else:
+                img[
+                    max(0, height - (i + 1) * square_size) : max(
+                        0, height - i * square_size
+                    ),
+                    j * square_size : min(width, (j + 1) * square_size),
+                ] = color_2.value
+
+    return img