283 lines
11 KiB
C++
283 lines
11 KiB
C++
#include "UIManager.h"
|
|
#include "config.h"
|
|
#include "SharedState.h"
|
|
|
|
// --- HARDWARE CONFIGURATION ---
|
|
#define SCREEN_WIDTH 128
|
|
#define SCREEN_HEIGHT 64
|
|
#define OLED_RESET -1
|
|
#define SCREEN_ADDRESS 0x3C
|
|
#define PIN_NEOPIXEL 16
|
|
#define NUM_PIXELS 64
|
|
|
|
UIManager ui;
|
|
|
|
UIManager::UIManager()
|
|
: display(SCREEN_WIDTH, SCREEN_HEIGHT, &Wire, OLED_RESET),
|
|
pixels(NUM_PIXELS, PIN_NEOPIXEL, NEO_GRB + NEO_KHZ800)
|
|
{
|
|
}
|
|
|
|
void UIManager::begin() {
|
|
// Setup Display
|
|
Wire.begin();
|
|
if(!display.begin(SSD1306_SWITCHCAPVCC, SCREEN_ADDRESS)) {
|
|
Serial.println(F("SSD1306 allocation failed"));
|
|
for(;;);
|
|
}
|
|
display.clearDisplay();
|
|
display.display();
|
|
|
|
// Setup NeoPixel Matrix
|
|
pixels.begin();
|
|
pixels.setBrightness(40);
|
|
pixels.clear();
|
|
pixels.show();
|
|
}
|
|
|
|
void UIManager::showMessage(const char* msg) {
|
|
display.clearDisplay();
|
|
display.setCursor(10, 25);
|
|
display.setTextColor(SSD1306_WHITE);
|
|
display.setTextSize(2);
|
|
display.print(msg);
|
|
display.display();
|
|
delay(500);
|
|
display.setTextSize(1);
|
|
}
|
|
|
|
void UIManager::draw(UIState currentState, int menuSelection,
|
|
int midiChannel, int tempo, MelodyStrategy* currentStrategy,
|
|
int queuedTheme, int currentThemeIndex,
|
|
int numScaleNotes, const int* scaleNotes, int melodySeed,
|
|
bool mutationEnabled, bool songModeEnabled,
|
|
const Step sequence[][NUM_STEPS], int playbackStep, bool isPlaying,
|
|
int randomizeTrack, const bool* trackMute) {
|
|
|
|
display.clearDisplay();
|
|
display.setTextSize(1);
|
|
display.setTextColor(SSD1306_WHITE);
|
|
display.setCursor(0, 0);
|
|
|
|
switch(currentState) {
|
|
case UI_MENU_MAIN:
|
|
drawMenu(menuSelection, currentState, midiChannel, tempo, currentStrategy->getName(), queuedTheme, currentThemeIndex, numScaleNotes, scaleNotes, melodySeed, mutationEnabled, songModeEnabled, randomizeTrack, trackMute);
|
|
break;
|
|
case UI_SETUP_CHANNEL_EDIT:
|
|
display.println(F("SET MIDI CHANNEL"));
|
|
display.drawLine(0, 8, 128, 8, SSD1306_WHITE);
|
|
display.setCursor(20, 25);
|
|
display.setTextSize(2);
|
|
display.print(F("CH: "));
|
|
if (midiChannel < 10) display.print(F(" "));
|
|
display.print(midiChannel);
|
|
display.setTextSize(1);
|
|
display.setCursor(0, 50);
|
|
display.println(F(" (Press to confirm)"));
|
|
break;
|
|
case UI_EDIT_TEMPO:
|
|
display.println(F("SET TEMPO"));
|
|
display.drawLine(0, 8, 128, 8, SSD1306_WHITE);
|
|
display.setCursor(20, 25);
|
|
display.setTextSize(2);
|
|
display.print(F("BPM: "));
|
|
display.print(tempo);
|
|
display.setTextSize(1);
|
|
display.setCursor(0, 50);
|
|
display.println(F(" (Press to confirm)"));
|
|
break;
|
|
case UI_EDIT_FLAVOUR:
|
|
display.println(F("SET FLAVOUR"));
|
|
display.drawLine(0, 8, 128, 8, SSD1306_WHITE);
|
|
display.setCursor(20, 25);
|
|
display.setTextSize(2);
|
|
display.print(currentStrategy->getName());
|
|
display.setTextSize(1);
|
|
display.setCursor(0, 50);
|
|
display.println(F(" (Press to confirm)"));
|
|
break;
|
|
case UI_RANDOMIZE_TRACK_EDIT:
|
|
display.println(F("SET TRACK"));
|
|
display.drawLine(0, 8, 128, 8, SSD1306_WHITE);
|
|
display.setCursor(20, 25);
|
|
display.setTextSize(2);
|
|
display.print(F("TRK: "));
|
|
display.print(randomizeTrack + 1);
|
|
display.setTextSize(1);
|
|
display.setCursor(0, 50);
|
|
display.println(F(" (Press to confirm)"));
|
|
break;
|
|
}
|
|
display.display();
|
|
}
|
|
|
|
void UIManager::drawMenu(int selection, UIState currentState, int midiChannel, int tempo, const char* flavourName,
|
|
int queuedTheme, int currentThemeIndex, int numScaleNotes,
|
|
const int* scaleNotes, int melodySeed, bool mutationEnabled,
|
|
bool songModeEnabled, int randomizeTrack, const bool* trackMute) {
|
|
|
|
// Calculate visual cursor position and scroll offset
|
|
int visualCursor = 0;
|
|
for(int i=0; i<selection; i++) {
|
|
if(isItemVisible(i)) visualCursor++;
|
|
}
|
|
|
|
const int MAX_LINES = 7; // No title, so we have more space
|
|
int startVisualIndex = 0;
|
|
if (visualCursor >= MAX_LINES) {
|
|
startVisualIndex = visualCursor - (MAX_LINES - 1);
|
|
}
|
|
|
|
int currentVisualIndex = 0;
|
|
int y = 0;
|
|
|
|
for (int i = 0; i < menuItemsCount; i++) {
|
|
if (!isItemVisible(i)) continue;
|
|
|
|
if (currentVisualIndex >= startVisualIndex) {
|
|
if (y > 55) break;
|
|
|
|
if (i == selection) {
|
|
display.fillRect(0, y, 128, 9, SSD1306_WHITE);
|
|
display.setTextColor(SSD1306_BLACK, SSD1306_WHITE);
|
|
} else {
|
|
display.setTextColor(SSD1306_WHITE);
|
|
}
|
|
|
|
int x = 2 + (menuItems[i].indentLevel * 6);
|
|
display.setCursor(x, y + 1);
|
|
|
|
if (menuItems[i].isGroup) {
|
|
display.print(menuItems[i].expanded ? F("v ") : F("> "));
|
|
}
|
|
|
|
display.print(menuItems[i].label);
|
|
|
|
MenuItemID id = menuItems[i].id;
|
|
|
|
if (id == MENU_ID_CHANNEL) {
|
|
display.print(F(": ")); display.print(midiChannel);
|
|
}
|
|
|
|
// Dynamic values
|
|
if (id == MENU_ID_MELODY) {
|
|
display.print(F(": ")); display.print(melodySeed);
|
|
} else if (id == MENU_ID_SCALE) {
|
|
display.print(F(": "));
|
|
if (numScaleNotes > 0) {
|
|
const char* noteNames[] = {"C", "C#", "D", "D#", "E", "F", "F#", "G", "G#", "A", "A#", "B"};
|
|
for (int j = 0; j < min(numScaleNotes, 6); j++) {
|
|
display.print(noteNames[scaleNotes[j]]);
|
|
if (j < min(numScaleNotes, 6) - 1) display.print(F(" "));
|
|
}
|
|
}
|
|
} else if (id == MENU_ID_TEMPO) { display.print(F(": ")); display.print(tempo); }
|
|
else if (id == MENU_ID_SONG_MODE) { display.print(F(": ")); display.print(songModeEnabled ? F("ON") : F("OFF")); }
|
|
else if (id == MENU_ID_TRACK_SELECT) { display.print(F(": ")); display.print(randomizeTrack + 1); }
|
|
else if (id == MENU_ID_MUTE) { display.print(F(": ")); display.print(trackMute[randomizeTrack] ? F("YES") : F("NO")); }
|
|
else if (id == MENU_ID_FLAVOUR) { display.print(F(": ")); display.print(flavourName); }
|
|
else if (id == MENU_ID_MUTATION) { display.print(F(": ")); display.print(mutationEnabled ? F("ON") : F("OFF")); }
|
|
|
|
if (id >= MENU_ID_THEME_1 && id <= MENU_ID_THEME_7) {
|
|
int themeIdx = id - MENU_ID_THEME_1 + 1;
|
|
if (queuedTheme == themeIdx) display.print(F(" [NEXT]"));
|
|
if (currentThemeIndex == themeIdx) display.print(F(" *"));
|
|
}
|
|
|
|
y += 9;
|
|
}
|
|
|
|
currentVisualIndex++;
|
|
}
|
|
}
|
|
|
|
uint32_t UIManager::getNoteColor(int note, bool dim) {
|
|
if (note == -1) return 0;
|
|
uint16_t hue = 30000 + (note % 12) * 3628;
|
|
return Adafruit_NeoPixel::ColorHSV(hue, 255, dim ? 10 : 50);
|
|
}
|
|
|
|
int UIManager::getPixelIndex(int x, int y) {
|
|
return y * 8 + x;
|
|
}
|
|
|
|
void UIManager::updateLeds(const Step sequence[][NUM_STEPS], int playbackStep, bool isPlaying,
|
|
UIState currentState, bool songModeEnabled,
|
|
int songRepeatsRemaining, bool sequenceChangeScheduled, PlayMode playMode,
|
|
int selectedTrack, int numScaleNotes, const int* scaleNotes, const bool* trackMute) {
|
|
pixels.clear();
|
|
|
|
const uint32_t COLOR_PLAYHEAD = pixels.Color(0, 255, 0);
|
|
const uint32_t COLOR_PLAYHEAD_DIM = pixels.Color(0, 32, 0);
|
|
const uint32_t COLOR_MUTED_PLAYHEAD = pixels.Color(0, 0, 255);
|
|
const uint32_t COLOR_CURSOR = pixels.Color(255, 255, 255);
|
|
const uint32_t COLOR_CURSOR_DIM = pixels.Color(32, 0, 0);
|
|
|
|
if(playMode == MODE_POLY) {
|
|
for(int t=0; t<NUM_TRACKS; t++) {
|
|
for(int s=0; s<NUM_STEPS; s++) {
|
|
int row = t * 2 + (s / 8);
|
|
int col = s % 8;
|
|
|
|
uint32_t color = 0;
|
|
int note = sequence[t][s].note;
|
|
|
|
if (note != -1) {
|
|
color = getNoteColor(note, !sequence[t][s].accent);
|
|
}
|
|
|
|
if (isPlaying && s == playbackStep) {
|
|
if (trackMute[t]) {
|
|
color = COLOR_MUTED_PLAYHEAD;
|
|
} else {
|
|
color = (note != -1) ? COLOR_PLAYHEAD : COLOR_PLAYHEAD_DIM;
|
|
}
|
|
}
|
|
|
|
pixels.setPixelColor(getPixelIndex(col, row), color);
|
|
}
|
|
}
|
|
} else {
|
|
// --- Mono Mode (original) ---
|
|
const Step* trackSequence = sequence[selectedTrack];
|
|
for (int s = 0; s < NUM_STEPS; s++) {
|
|
int x = s % 8;
|
|
int yBase = (s / 8) * 4;
|
|
uint32_t color = 0, dimColor = 0;
|
|
bool isCursorHere = (isPlaying && s == playbackStep);
|
|
if (trackSequence[s].note != -1) {
|
|
color = getNoteColor(trackSequence[s].note, trackSequence[s].tie);
|
|
dimColor = getNoteColor(trackSequence[s].note, true);
|
|
}
|
|
uint32_t c[4] = {0};
|
|
if (trackSequence[s].note != -1) {
|
|
int octave = trackSequence[s].note / 12;
|
|
if (octave > 4) { c[0] = color; if (trackSequence[s].accent) c[1] = dimColor; }
|
|
else if (octave < 4) { c[2] = color; if (trackSequence[s].accent) c[1] = dimColor; }
|
|
else { c[1] = color; if (trackSequence[s].accent) { c[0] = dimColor; c[2] = dimColor; } }
|
|
}
|
|
uint32_t cursorColor = pixels.Color(0, 0, 50);
|
|
if (isPlaying) {
|
|
cursorColor = pixels.Color(0, 50, 0);
|
|
if (songModeEnabled && s >= 8) {
|
|
int repeats = min(songRepeatsRemaining, 8);
|
|
if (x >= (8 - repeats)) cursorColor = (songRepeatsRemaining == 1 && x == 7 && (millis()/250)%2) ? pixels.Color(255, 200, 0) : pixels.Color(100, 220, 40);
|
|
}
|
|
}
|
|
|
|
if (cursorColor != 0) {
|
|
if (isCursorHere) {
|
|
for(int i=0; i<4; i++) {
|
|
if (c[i] == 0) c[i] = cursorColor;
|
|
}
|
|
} else {
|
|
uint8_t r = (uint8_t)(cursorColor >> 16), g = (uint8_t)(cursorColor >> 8), b = (uint8_t)cursorColor;
|
|
c[3] = pixels.Color(r/5, g/5, b/5);
|
|
}
|
|
}
|
|
for(int i=0; i<4; i++) pixels.setPixelColor(getPixelIndex(x, yBase + i), c[i]);
|
|
}
|
|
}
|
|
if (sequenceChangeScheduled && (millis() / 125) % 2) pixels.setPixelColor(NUM_PIXELS - 1, pixels.Color(127, 50, 0));
|
|
pixels.show();
|
|
} |