From f34f9603583f07ee41bbc373719a5de58bf90fea Mon Sep 17 00:00:00 2001
From: Dejvino <dejvino@gmail.com>
Date: Sat, 7 Mar 2026 21:31:17 +0100
Subject: [PATCH] Chance setting

---
 Persistence.cpp    | 21 +++++++++++++++++++++
 PlaybackThread.cpp |  5 ++++-
 SharedState.cpp    |  4 +++-
 SharedState.h      |  2 ++
 TrackerTypes.h     |  1 +
 UIManager.cpp      | 15 +++++++++++++++
 UIThread.cpp       | 24 ++++++++++++++++++++++++
 7 files changed, 70 insertions(+), 2 deletions(-)

diff --git a/Persistence.cpp b/Persistence.cpp
index 300f875..7822742 100644
--- a/Persistence.cpp
+++ b/Persistence.cpp
@@ -22,6 +22,9 @@ void Persistence::saveSequence(bool quiet) {
   int intensities[NUM_TRACKS];
   for(int i=0; i<NUM_TRACKS; i++) intensities[i] = (int)trackIntensity[i];
   EEPROM.put(addr, intensities); addr += sizeof(intensities);
+  int chances[NUM_TRACKS];
+  for(int i=0; i<NUM_TRACKS; i++) chances[i] = (int)trackChance[i];
+  EEPROM.put(addr, chances); addr += sizeof(chances);
   int steps[NUM_TRACKS];
   for(int i=0; i<NUM_TRACKS; i++) steps[i] = numSteps[i];
   EEPROM.put(addr, steps); addr += sizeof(steps);
@@ -79,6 +82,13 @@ bool Persistence::loadSequence() {
     if (trackIntensity[i] < 1) trackIntensity[i] = 1;
     if (trackIntensity[i] > 10) trackIntensity[i] = 10;
   }
+  int chances[NUM_TRACKS];
+  EEPROM.get(addr, chances); addr += sizeof(chances);
+  for(int i=0; i<NUM_TRACKS; i++) {
+    trackChance[i] = chances[i];
+    if (trackChance[i] < 0) trackChance[i] = 0;
+    if (trackChance[i] > 100) trackChance[i] = 100;
+  }
   int steps[NUM_TRACKS];
   EEPROM.get(addr, steps); addr += sizeof(steps);
   for(int i=0; i<NUM_TRACKS; i++) {
@@ -129,6 +139,9 @@ void Persistence::savePatch(int bank, int slot) {
     int intensities[NUM_TRACKS];
     for(int i=0; i<NUM_TRACKS; i++) intensities[i] = trackIntensity[i];
     EEPROM.put(addr, intensities); addr += sizeof(intensities);
+    int chances[NUM_TRACKS];
+    for(int i=0; i<NUM_TRACKS; i++) chances[i] = trackChance[i];
+    EEPROM.put(addr, chances); addr += sizeof(chances);
     midi.unlock();
     EEPROM.commit();
     ui.showMessage("SAVED!");
@@ -173,6 +186,13 @@ void Persistence::loadPatch(int bank, int slot, Step (*scratchBuffer)[NUM_STEPS]
       if (trackIntensity[i] < 1) trackIntensity[i] = 1;
       if (trackIntensity[i] > 10) trackIntensity[i] = 10;
     }
+    int chances[NUM_TRACKS];
+    EEPROM.get(addr, chances); addr += sizeof(chances);
+    for(int i=0; i<NUM_TRACKS; i++) {
+      trackChance[i] = chances[i];
+      if (trackChance[i] < 0) trackChance[i] = 0;
+      if (trackChance[i] > 100) trackChance[i] = 100;
+    }
     
     if (isPlaying) {
         SequenceGenerator::generateSequenceData(currentThemeIndex, nextSequence);
@@ -189,6 +209,7 @@ void Persistence::factoryReset() {
   ui.showMessage("RESETTING...");
   for(int i=0; i<NUM_TRACKS; i++) {
     trackIntensity[i] = 10;
+    trackChance[i] = 100;
   }
   uint32_t magic = 0;
   EEPROM.put(0, magic);
diff --git a/PlaybackThread.cpp b/PlaybackThread.cpp
index 77179ea..2f9251a 100644
--- a/PlaybackThread.cpp
+++ b/PlaybackThread.cpp
@@ -133,7 +133,10 @@ static void handlePlayback() {
       // If tied to the SAME note, do not retrigger (sustain)
       bool isContinuing = wasTied && (prevNote == currentNote) && !justPanicked;
 
-      if (!trackMute[t] && currentNote != -1 && !isContinuing) {
+      // Check chance
+      bool shouldPlay = (random(100) < trackChance[t]);
+
+      if (!trackMute[t] && currentNote != -1 && !isContinuing && shouldPlay) {
         uint8_t velocity = local_sequence[t][current_step].accent ? 127 : 100;
         midi.sendNoteOn(currentNote, velocity, trackChannel);
       }
diff --git a/SharedState.cpp b/SharedState.cpp
index cfff9e2..6de9022 100644
--- a/SharedState.cpp
+++ b/SharedState.cpp
@@ -20,6 +20,7 @@ UIState currentState = UI_MENU_MAIN;
 bool protectedMode = false;
 
 volatile int trackIntensity[NUM_TRACKS] = {10, 10, 10, 10};
+volatile int trackChance[NUM_TRACKS] = {100, 100, 100, 100};
 // Menus
 
 #define PROG_STEP_ITEMS(n) \
@@ -57,6 +58,7 @@ MenuItem menuItems[] = {
       { "Steps", MENU_ID_STEPS, false, false, 1 },
       { "Flavour", MENU_ID_FLAVOUR, false, false, 1 },
       { "Intensity", MENU_ID_INTENSITY, false, false, 1 },
+      { "Chance", MENU_ID_CHANCE, false, false, 1 },
       { "Mutation", MENU_ID_MUTATION, false, false, 1 },
       { "Theme 1", MENU_ID_THEME_1, false, false, 1 },
       { "Theme 2", MENU_ID_THEME_2, false, false, 1 },
@@ -148,7 +150,7 @@ volatile int currentThemeIndex = 1;
 int globalRoot = 0; // The "Key" of the song
 int currentRoot = 0; // C
 int currentScaleType = 1; // Major
-extern const uint32_t EEPROM_MAGIC = 0x42424253;
+extern const uint32_t EEPROM_MAGIC = 0x42424254;
 
 const ChordProgression progressions[] = {
     { "Off", 0, { {0,0,1} } }, // Dummy
diff --git a/SharedState.h b/SharedState.h
index c75903d..c5ef26a 100644
--- a/SharedState.h
+++ b/SharedState.h
@@ -37,6 +37,7 @@ enum MenuItemID {
     MENU_ID_STEPS,
     MENU_ID_FLAVOUR,
     MENU_ID_INTENSITY,
+    MENU_ID_CHANCE,
     MENU_ID_MUTATION,
     MENU_ID_THEME_1,
     MENU_ID_THEME_2,
@@ -85,6 +86,7 @@ extern volatile int currentThemeIndex;
 extern int globalRoot;
 extern int currentRoot;
 extern volatile int trackIntensity[NUM_TRACKS];
+extern volatile int trackChance[NUM_TRACKS];
 extern int currentScaleType;
 extern ChordProgression currentProgression;
 extern ProgressionOrder progressionOrder;
diff --git a/TrackerTypes.h b/TrackerTypes.h
index f312f2b..705a0a7 100644
--- a/TrackerTypes.h
+++ b/TrackerTypes.h
@@ -41,6 +41,7 @@ enum UIState {
   UI_EDIT_STEPS,
   UI_EDIT_FLAVOUR,
   UI_EDIT_INTENSITY,
+  UI_EDIT_CHANCE,
   UI_SETUP_PLAYMODE_EDIT,
   UI_RANDOMIZE_TRACK_EDIT,
   UI_EDIT_ROOT,
diff --git a/UIManager.cpp b/UIManager.cpp
index 6f2602b..8f840ad 100644
--- a/UIManager.cpp
+++ b/UIManager.cpp
@@ -103,6 +103,9 @@ void UIManager::draw(UIState currentState, int menuSelection,
         case UI_EDIT_INTENSITY:
             drawNumberEditor("SET INTENSITY", trackIntensities[randomizeTrack], 1, 10);
             break;
+        case UI_EDIT_CHANCE:
+            drawNumberEditor("SET CHANCE %", trackChance[randomizeTrack], 0, 100);
+            break;
         case UI_RANDOMIZE_TRACK_EDIT:
             drawEditScreen("SET TRACK", "TRK: ", randomizeTrack + 1);
             break;
@@ -261,6 +264,18 @@ void UIManager::drawMenu(int selection, UIState currentState, int midiChannel, i
                 display.drawRect(barX, barY, maxW + 2, h, color);
                 display.fillRect(barX + 1, barY + 1, (val * maxW) / 10, h - 2, color);
             }
+            else if (id == MENU_ID_CHANCE) {
+                display.print(F(": "));
+                display.print(trackChance[randomizeTrack]);
+                int val = trackChance[randomizeTrack];
+                int barX = display.getCursorX() + 3;
+                int barY = y + 2;
+                int maxW = 20;
+                int h = 5;
+                uint16_t color = (i == selection) ? SSD1306_BLACK : SSD1306_WHITE;
+                display.drawRect(barX, barY, maxW + 2, h, color);
+                display.fillRect(barX + 1, barY + 1, (val * maxW) / 100, h - 2, color);
+            }
             else if (id == MENU_ID_MUTATION) { display.print(F(": ")); display.print(mutationEnabled ? F("ON") : F("OFF")); }
             else if (id == MENU_ID_PROTECTED_MODE) { display.print(F(": ")); display.print(protectedMode ? F("ON") : F("OFF")); }
             
diff --git a/UIThread.cpp b/UIThread.cpp
index 2088ce1..d91ec7e 100644
--- a/UIThread.cpp
+++ b/UIThread.cpp
@@ -162,6 +162,15 @@ static void handleInput() {
           trackIntensity[randomizeTrack] = current;
         }
         break;
+      case UI_EDIT_CHANCE:
+        {
+          int current = trackChance[randomizeTrack];
+          current += delta;
+          if (current < 0) current = 0;
+          if (current > 100) current = 100;
+          trackChance[randomizeTrack] = current;
+        }
+        break;
     }
     if (currentState == UI_RANDOMIZE_TRACK_EDIT) {
         randomizeTrack += (delta > 0 ? 1 : -1);
@@ -254,6 +263,7 @@ static void handleInput() {
               case MENU_ID_MUTE: trackMute[randomizeTrack] = !trackMute[randomizeTrack]; break;
               case MENU_ID_FLAVOUR: currentState = UI_EDIT_FLAVOUR;  break;
               case MENU_ID_INTENSITY: currentState = UI_EDIT_INTENSITY; break;
+              case MENU_ID_CHANCE: currentState = UI_EDIT_CHANCE; break;
               case MENU_ID_MUTATION: mutationEnabled = !mutationEnabled; break;
               
               case MENU_ID_CHANNEL: currentState = UI_SETUP_CHANNEL_EDIT; break;
@@ -355,6 +365,20 @@ static void handleInput() {
               }
               saveSequence(true);
               break;
+            case UI_EDIT_CHANCE:
+              currentState = UI_MENU_MAIN;
+              if (isPlaying) {
+                  int theme = (queuedTheme != -1) ? queuedTheme : currentThemeIndex;
+                  midi.lock();
+                  if (!sequenceChangeScheduled) {
+                      memcpy(nextSequence, sequence, sizeof(sequence));
+                  }
+                  SequenceGenerator::generateTrackData(randomizeTrack, theme, nextSequence);
+                  sequenceChangeScheduled = true;
+                  midi.unlock();
+              }
+              saveSequence(true);
+              break;
             case UI_RANDOMIZE_TRACK_EDIT:
               currentState = UI_MENU_MAIN;
               saveSequence(true);