Add probability handling and drawing

1 files changed, 13 insertions, 8 deletions

diff --git a/src/sequencer.c b/src/sequencer.c
index 04153aa..00baebb 100644
--- a/src/sequencer.c
+++ b/src/sequencer.c
@@ -71,6 +71,11 @@ find_prev_pattern(void) {
     return idx;
 }
 
+bool
+should_play(u8 prob) {
+    return true;
+}
+
 void
 play_step(void) {
     Pattern *pat = &patterns[current_pattern];
@@ -94,7 +99,7 @@ play_step(void) {
     if (pat->ch1.active) {
         TriggerNote *trig = &pat->ch1.notes[step_counter];
         ChannelSquareParams *params = &pat->ch1.params[step_counter];
-        if (trig->active) {
+        if (trig->active && should_play(params->prob)) {
             if (params->sweep_time == 0) {
                 SOUND_SQUARE1_SWEEP = SOUND_SWEEP_DIR(1);
             } else {
@@ -128,7 +133,7 @@ play_step(void) {
     if (pat->ch2.active) {
         TriggerNote *trig = &pat->ch2.notes[step_counter];
         ChannelSquareParams *params = &pat->ch2.params[step_counter];
-        if (trig->active) {
+        if (trig->active && should_play(params->prob)) {
             SOUND_SQUARE2_CTRL = SOUND_SQUARE_ENV_VOL(params->env_volume)
                 | SOUND_SQUARE_ENV_TIME(params->env_time)
                 | SOUND_SQUARE_ENV_DIR(params->env_direction)
@@ -147,7 +152,7 @@ play_step(void) {
     if (pat->ch3.active) {
         TriggerNote *trig = &pat->ch3.notes[step_counter];
         ChannelWaveParams *params = &pat->ch3.params[step_counter];
-        if (trig->active) {
+        if (trig->active && should_play(params->prob)) {
             switch (params->wave_mode) {
                 case 0: {
                     SOUND_WAVE_MODE = SOUND_WAVE_BANK_SELECT(1);
@@ -195,7 +200,7 @@ play_step(void) {
     if (pat->ch4.active) {
         TriggerNote *trig = &pat->ch4.notes[step_counter];
         ChannelNoiseParams *params = &pat->ch4.params[step_counter];
-        if (trig->active) {
+        if (trig->active && should_play(params->prob)) {
             static const u8 div_freq[] = {
                 7, 6, 5, 4, 7, 6, 5, 4, 7, 6, 5, 4,
                 7, 6, 5, 4, 7, 6, 5, 4, 7, 6, 5, 4,
@@ -778,7 +783,7 @@ set_param_selection_sq1(ChannelSquareParams *params, InputHandler return_handler
                 }
             } break;
             case 3: { params->env_direction ^= 1; } break;
-            case 4: { /* TODO: probability */ } break;
+            case 4: { params->prob = CLAMP(params->prob + inc, 0, PROB_NUM - 1); } break;
             case 5: { params->sweep_number = CLAMP(params->sweep_number + inc, 0, 7); } break;
             case 6: { params->sweep_time = CLAMP(params->sweep_time + inc, 0, 7); } break;
             case 7: { params->sweep_direction ^= 1; } break;
@@ -843,7 +848,7 @@ set_param_selection_sq2(ChannelSquareParams *params, InputHandler return_handler
                 }
             } break;
             case 3: { params->env_direction ^= 1; } break;
-            case 4: { /* TODO: probability */ } break;
+            case 4: { params->prob = CLAMP(params->prob + inc, 0, PROB_NUM - 1); } break;
         }
         redraw_params = true;
         return true;
@@ -907,7 +912,7 @@ set_param_selection_wave(ChannelWaveParams *params, InputHandler return_handler)
             case 1: { params->phase_a = CLAMP(params->phase_a + inc, 0, WAVE_VARS - 1); } break;
             case 2: { params->shape_b = CLAMP(params->shape_a + inc, 0, WAVE_MAX - 1); } break;
             case 3: { params->phase_b = CLAMP(params->phase_b + inc, 0, WAVE_VARS - 1); } break;
-            case 4: { /* TODO: probability */ } break;
+            case 4: { params->prob = CLAMP(params->prob + inc, 0, PROB_NUM - 1); } break;
             case 5: { params->wave_mode = CLAMP(params->wave_mode + inc, 0, 2); } break;
             case 6: { params->wave_volume = CLAMP(params->wave_volume + inc, 0, 4); } break;
         }
@@ -971,7 +976,7 @@ set_param_selection_noise(ChannelNoiseParams *params, InputHandler return_handle
                 }
             } break;
             case 3: { params->env_direction ^= 1; } break;
-            case 4: { /* TODO: probability */ } break;
+            case 4: { params->prob = CLAMP(params->prob + inc, 0, PROB_NUM - 1); } break;
         }
         redraw_params = true;
         return true;