#include "SharedState.h"
#include "LuckyStrategy.h"
#include "ArpStrategy.h"
#include "EuclideanStrategy.h"
#include "MarkovStrategy.h"
#include "CellularAutomataStrategy.h"
#include "LSystemStrategy.h"

// Global state variables
Step sequence[NUM_TRACKS][NUM_STEPS];
Step nextSequence[NUM_TRACKS][NUM_STEPS];
volatile bool sequenceChangeScheduled = false;
volatile bool needsPanic = false;

UIState currentState = UI_MENU_MAIN;

// Menus
MenuItem menuItems[] = {
    { "Main", MENU_ID_GROUP_MAIN, true, true, 0 },
      { "Melody", MENU_ID_MELODY, false, false, 1 },
      { "Scale", MENU_ID_SCALE, false, false, 1 },
      { "Tempo", MENU_ID_TEMPO, false, false, 1 },
      { "Song Mode", MENU_ID_SONG_MODE, false, false, 1 },
    { "Track", MENU_ID_GROUP_TRACK, true, true, 0 },
      { "Track", MENU_ID_TRACK_SELECT, false, false, 1 },
      { "Mute", MENU_ID_MUTE, false, false, 1 },
      { "Flavour", MENU_ID_FLAVOUR, 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 },
      { "Theme 3", MENU_ID_THEME_3, false, false, 1 },
      { "Theme 4", MENU_ID_THEME_4, false, false, 1 },
      { "Theme 5", MENU_ID_THEME_5, false, false, 1 },
      { "Theme 6", MENU_ID_THEME_6, false, false, 1 },
      { "Theme 7", MENU_ID_THEME_7, false, false, 1 },
    { "Setup", MENU_ID_GROUP_SETUP, true, false, 0 },
      { "Channel", MENU_ID_CHANNEL, false, false, 1 },
      { "Reset", MENU_ID_RESET, false, false, 1 }
};
extern const int menuItemsCount = sizeof(menuItems) / sizeof(MenuItem);

bool isItemVisible(int index) {
    if (menuItems[index].indentLevel == 0) return true;
    for (int i = index - 1; i >= 0; i--) {
        if (menuItems[i].indentLevel < menuItems[index].indentLevel) {
            return menuItems[i].expanded;
        }
    }
    return true;
}

int menuSelection = 0;
volatile bool trackMute[NUM_TRACKS];
int randomizeTrack = 0;
volatile int playbackStep = 0;
volatile int midiChannels[NUM_TRACKS];
int scaleNotes[12];
int numScaleNotes = 0;
int melodySeeds[NUM_TRACKS];
volatile int queuedTheme = -1;
volatile int currentThemeIndex = 1;
extern const uint32_t EEPROM_MAGIC = 0x4242424B;

MelodyStrategy* strategies[] = {
    new LuckyStrategy(), new ArpStrategy(), new EuclideanStrategy(),
    new MarkovStrategy(), new CellularAutomataStrategy(), new LSystemStrategy()};
extern const int numStrategies = 6;
int currentStrategyIndices[NUM_TRACKS];

volatile PlayMode playMode = MODE_POLY;
volatile bool mutationEnabled = false;
volatile bool songModeEnabled = false;
volatile int songRepeatsRemaining = 0;
volatile int nextSongRepeats = 0;
volatile bool songModeNeedsNext = false;
volatile bool isPlaying = false;
volatile int tempo = 120; // BPM
volatile unsigned long lastClockTime = 0;
volatile int clockCount = 0;

// Encoder State
volatile int encoderDelta = 0;

// Button State
bool lastButtonState = HIGH;
unsigned long lastDebounceTime = 0;
bool buttonActive = false;
bool buttonConsumed = false;
unsigned long buttonPressTime = 0;