#pragma once

#include <JuceHeader.h>
#include "AudioBufferQueue.h"
#include "AudioEngine.h"
#include "ScopeDataCollector.h"
#include "NeuralSharedParams.h"

//==============================================================================
// Processor
class NeuralSynthAudioProcessor : public juce::AudioProcessor,
                                  private juce::AudioProcessorValueTreeState::Listener
{
public:
    struct PresetDefinition
    {
        juce::String category;
        juce::String name;
        int wtBankIndex;
        int wt2BankIndex { -1 };
        std::vector<std::pair<juce::String, float>> parameterValues;
    };

    NeuralSynthAudioProcessor();
    ~NeuralSynthAudioProcessor() override;

    // AudioProcessor overrides
    void prepareToPlay(double sampleRate, int samplesPerBlock) override;
    void releaseResources() override;

#ifndef JucePlugin_PreferredChannelConfigurations
    bool isBusesLayoutSupported(const BusesLayout& layouts) const override;
#endif

    void processBlock(juce::AudioBuffer<float>&, juce::MidiBuffer&) override;

    // Editor
    juce::AudioProcessorEditor* createEditor() override;
    bool hasEditor() const override;

    // Info
    const juce::String getName() const override;
    bool acceptsMidi() const override;
    bool producesMidi() const override;
    bool isMidiEffect() const override;
    double getTailLengthSeconds() const override;

    // Programs
    int getNumPrograms() override;
    int getCurrentProgram() override;
    void setCurrentProgram(int index) override;
    const juce::String getProgramName(int index) override;
    void changeProgramName(int index, const juce::String& newName) override;

    // State
    void getStateInformation(juce::MemoryBlock& destData) override;
    void setStateInformation(const void* data, int sizeInBytes) override;

    // Parameters
    void parameterChanged(const juce::String& id, float newValue) override;
    void buildParams(std::vector<std::unique_ptr<juce::RangedAudioParameter>>& params,
                     const std::string& paramGroup);
    juce::AudioProcessorValueTreeState::ParameterLayout createParameterLayout();

    void applyPreset(int index);
    const std::vector<PresetDefinition>& getFactoryPresets() const noexcept { return factoryPresets; }
    int getCurrentPresetIndex() const noexcept { return currentPresetIndex; }

    // Utilities
    juce::MidiMessageCollector& getMidiMessageCollector() noexcept { return midiMessageCollector; }
    AudioBufferQueue<float>&    getAudioBufferQueue()    noexcept { return audioBufferQueue; }

    AudioBufferQueue<float>& getChorusAudioBufferQueue()     noexcept { return chorusBufferQueue; }
    AudioBufferQueue<float>& getDelayAudioBufferQueue()      noexcept { return delayBufferQueue; }
    AudioBufferQueue<float>& getReverbAudioBufferQueue()     noexcept { return reverbBufferQueue; }
    AudioBufferQueue<float>& getFlangerAudioBufferQueue()    noexcept { return flangerBufferQueue; }
    AudioBufferQueue<float>& getDistortionAudioBufferQueue() noexcept { return distortionBufferQueue; }
    AudioBufferQueue<float>& getFilterAudioBufferQueue()     noexcept { return filterBufferQueue; }

    // Public members (by JUCE convention)
    juce::MidiMessageCollector midiMessageCollector;
    juce::AudioProcessorValueTreeState parameters;

private:
    JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (NeuralSynthAudioProcessor)

    // ---- IMPORTANT ORDER FIX ----
    // Objects are constructed in THIS order. 'sp' must come BEFORE audioEngine.
    NeuralSharedParams sp;                 // <— construct first
    NeuralAudioEngine  audioEngine;        // needs a valid reference to 'sp'

    // Meter/scope queues
    AudioBufferQueue<float> audioBufferQueue;
    AudioBufferQueue<float> chorusBufferQueue;
    AudioBufferQueue<float> delayBufferQueue;
    AudioBufferQueue<float> reverbBufferQueue;
    AudioBufferQueue<float> flangerBufferQueue;
    AudioBufferQueue<float> distortionBufferQueue;
    AudioBufferQueue<float> filterBufferQueue;

    // Scope collector (uses audioBufferQueue, so declare after it)
    ScopeDataCollector<float> scopeDataCollector { audioBufferQueue };

    std::vector<PresetDefinition> factoryPresets;
    int currentPresetIndex { -1 };
    bool presetChangeInProgress { false };

    static std::vector<PresetDefinition> makeFactoryPresets();
    void setParameterValue(const juce::String& paramID, float value);

    juce::dsp::Limiter<float> outputLimiter;
};