From a82636c0de01bfcb14d4ef403c1692f481cbc237 Mon Sep 17 00:00:00 2001
From: redbirdly <155550833+redbirdly@users.noreply.github.com>
Date: Mon, 24 Jun 2024 13:49:40 +0800
Subject: [PATCH] Add wifi_draw.js

---
 mods/wifi_draw.js | 77 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)
 create mode 100644 mods/wifi_draw.js

diff --git a/mods/wifi_draw.js b/mods/wifi_draw.js
new file mode 100644
index 00000000..5ced23c5
--- /dev/null
+++ b/mods/wifi_draw.js
@@ -0,0 +1,77 @@
+// RedBirdly's mod to draw lines between transmitters/receivers
+// logicgates.js required
+
+let logicReceivers = [];
+let logicTransmitters = [];
+// let receivers = [];
+// let transmitters = [];
+
+function updateLogicLists() {
+    // receivers = [];
+    // transmitters = [];
+    logicReceivers = [];
+	logicTransmitters = [];
+    
+    for (let i = 0; i < currentPixels.length; i++) {
+        const pixel = currentPixels[i];
+        if (pixel.element === "logic_receiver") {
+            logicReceivers.push(pixel);
+        } else if (pixel.element === "logic_transmitter") {
+            logicTransmitters.push(pixel);
+        } /*else if (pixel.element === "receiver") {
+            receivers.push(pixel);
+        } else if (pixel.element === "transmitter") {
+            transmitters.push(pixel);
+        }*/
+    }
+}
+
+// Function to draw lines between linked transmitters and receivers
+function drawLinks() {
+    // Iterate through transmitters and receivers to draw lines for linked channels
+    for (const transmitter of logicTransmitters) {
+        for (const receiver of logicReceivers) {
+            if (transmitter.channel === receiver.channel) {
+                ctx.beginPath();
+                ctx.moveTo(transmitter.x * pixelSize + pixelSizeHalf, transmitter.y * pixelSize + pixelSizeHalf);
+                ctx.lineTo(receiver.x * pixelSize + pixelSizeHalf, receiver.y * pixelSize + pixelSizeHalf);
+                ctx.strokeStyle = "RGBA(255,255,255,0.2)";
+
+                const neighbors = [
+                    { x: transmitter.x, y: transmitter.y - 1 }, // Top
+                    { x: transmitter.x, y: transmitter.y + 1 }, // Bottom
+                    { x: transmitter.x - 1, y: transmitter.y }, // Left
+                    { x: transmitter.x + 1, y: transmitter.y }  // Right
+                ];
+
+                let highlight = false;
+                for (const neighbor of neighbors) {
+                    if (
+                        neighbor.x >= 0 && neighbor.x < width &&
+                        neighbor.y >= 0 && neighbor.y < height
+                    ) {
+                        const neighborPixel = pixelMap[neighbor.x][neighbor.y];
+                        if (neighborPixel && neighborPixel.lstate > 0) {
+                            highlight = true;
+                            break;
+                        }
+                    }
+                }
+
+                if (highlight) {
+                    ctx.strokeStyle = "RGBA(255,200,0,0.4)";
+                }
+                
+                ctx.lineWidth = 2;
+                ctx.stroke();
+            }
+        }
+    }
+}
+
+var originalDrawPixels2 = drawPixels;
+drawPixels = function() {
+    originalDrawPixels2();
+    updateLogicLists();
+    drawLinks();
+};
\ No newline at end of file