From 3ba4702181a5a4843d9c9f7b4a8888b5661209b3 Mon Sep 17 00:00:00 2001
From: Blaz Kristan <blaz@kristan-sp.si>
Date: Thu, 19 May 2022 18:57:23 +0200
Subject: [PATCH] 2D crazy bees.

---
 wled00/FX.cpp     | 71 ++++++++++++++++++++++++++++++++++++++++++++++-
 wled00/FX.h       |  7 +++--
 wled00/FX_fcn.cpp |  3 +-
 3 files changed, 77 insertions(+), 4 deletions(-)

diff --git a/wled00/FX.cpp b/wled00/FX.cpp
index f36d22e2f..34ab85d71 100644
--- a/wled00/FX.cpp
+++ b/wled00/FX.cpp
@@ -5481,4 +5481,73 @@ uint16_t WS2812FX::mode_2Dspaceships(void) {    //// Space ships by stepko (c)05
 
   setPixels(leds);
   return FRAMETIME;
-}
\ No newline at end of file
+}
+
+/////////////////////////
+//     2D Crazy Bees   //
+/////////////////////////
+//// Crazy bees by stepko (c)12.02.21 [https://editor.soulmatelights.com/gallery/651-crazy-bees], adapted by Blaz Kristan
+#define MAX_BEES 5
+uint16_t WS2812FX::mode_2Dcrazybees(void) {
+  if (!isMatrix) return mode_static(); // not a 2D set-up
+
+  uint16_t width  = SEGMENT.virtualWidth();
+  uint16_t height = SEGMENT.virtualHeight();
+  uint16_t dataSize = sizeof(CRGB) * width * height;
+
+  byte n = MIN(MAX_BEES, (width * height) / 256 + 1);
+
+  if (!SEGENV.allocateData(dataSize)) return mode_static(); //allocation failed
+  CRGB *leds = reinterpret_cast<CRGB*>(SEGENV.data);
+
+  static struct {
+    uint8_t posX, posY, aimX, aimY, hue;
+    int8_t deltaX, deltaY, signX, signY, error;
+    void aimed(uint16_t width, uint16_t height) {
+      randomSeed(millis());
+      aimX = random8(0, width);
+      aimY = random8(0, height);
+      hue = random8();
+      deltaX = abs(aimX - posX);
+      deltaY = abs(aimY - posY);
+      signX = posX < aimX ? 1 : -1;
+      signY = posY < aimY ? 1 : -1;
+      error = deltaX - deltaY;
+    };
+  } bee[MAX_BEES];
+
+  if (SEGENV.call == 0) {
+    fill_solid(leds, CRGB::Black);
+    for (byte i = 0; i < n; i++) {
+      bee[i].posX = random8(0, width);
+      bee[i].posY = random8(0, height);
+      bee[i].aimed(width, height);
+    }
+  }
+
+  fadeToBlackBy(leds, SEGMENT.speed>>4);
+  for (byte i = 0; i < n; i++) {
+    leds[XY(bee[i].aimX + 1, bee[i].aimY)] += CHSV(bee[i].hue, 255, 255);
+    leds[XY(bee[i].aimX, bee[i].aimY + 1)] += CHSV(bee[i].hue, 255, 255);
+    leds[XY(bee[i].aimX - 1, bee[i].aimY)] += CHSV(bee[i].hue, 255, 255);
+    leds[XY(bee[i].aimX, bee[i].aimY - 1)] += CHSV(bee[i].hue, 255, 255);
+    if (bee[i].posX != bee[i].aimX || bee[i].posY != bee[i].aimY) {
+      leds[XY(bee[i].posX, bee[i].posY)] = CHSV(bee[i].hue, 60, 255);
+      int8_t error2 = bee[i].error * 2;
+      if (error2 > -bee[i].deltaY) {
+        bee[i].error -= bee[i].deltaY;
+        bee[i].posX += bee[i].signX;
+      }
+      if (error2 < bee[i].deltaX) {
+        bee[i].error += bee[i].deltaX;
+        bee[i].posY += bee[i].signY;
+      }
+    } else {
+      bee[i].aimed(width, height);
+    }
+  }
+  blur2d(leds, SEGMENT.intensity>>3);
+
+  setPixels(leds);
+  return FRAMETIME;
+}
diff --git a/wled00/FX.h b/wled00/FX.h
index 7b77dcfe2..d716d77f4 100644
--- a/wled00/FX.h
+++ b/wled00/FX.h
@@ -262,8 +262,9 @@
 #define FX_MODE_TARTAN                 139
 #define FX_MODE_WAVERLY                140
 #define FX_MODE_SPACESHIPS             141
+#define FX_MODE_CRAZYBEES              142
 
-#define MODE_COUNT                     142
+#define MODE_COUNT                     143
 
 
 class WS2812FX {
@@ -652,6 +653,7 @@ class WS2812FX {
       _mode[FX_MODE_WAVERLY]                 = &WS2812FX::mode_2DWaverly;
       _mode[FX_MODE_AKEMI]                   = &WS2812FX::mode_2DAkemi;
       _mode[FX_MODE_SPACESHIPS]              = &WS2812FX::mode_2Dspaceships;
+      _mode[FX_MODE_CRAZYBEES]               = &WS2812FX::mode_2Dcrazybees;
 
       _brightness = DEFAULT_BRIGHTNESS;
       currentPalette = CRGBPalette16(CRGB::Black);
@@ -954,7 +956,8 @@ class WS2812FX {
     mode_2Dtartan(void),
     mode_2DWaverly(void),
     mode_2DAkemi(void),
-    mode_2Dspaceships(void);
+    mode_2Dspaceships(void),
+    mode_2Dcrazybees(void);
 
 // end 2D support
 
diff --git a/wled00/FX_fcn.cpp b/wled00/FX_fcn.cpp
index 9cfd8e49c..6d5854a52 100644
--- a/wled00/FX_fcn.cpp
+++ b/wled00/FX_fcn.cpp
@@ -1455,7 +1455,8 @@ const char JSON_mode_names[] PROGMEM = R"=====([
 "2D Sun Radiation@Variance,Brightness;;",
 "2D Tartan@X scale,Y scale;;!",
 "2D Waverly@Fade rate,Sensitivity;;!",
-"2D Spaceships@Fade rate,Blur;!,!,!;!"
+"2D Spaceships@Fade rate,Blur;!,!,!;!",
+"2D Crazy Bees@Fade rate,Blur;;"
 ])=====";
 
 const char JSON_palette_names[] PROGMEM = R"=====([