/*
  ==============================================================================

    This file contains the basic framework code for a JUCE plugin processor.

  ==============================================================================
*/

#pragma once

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

//==============================================================================
/**
*/
class NeuralSynthAudioProcessor : public juce::AudioProcessor,
    private juce::AudioProcessorValueTreeState::Listener
{
public:
    //==============================================================================
    NeuralSynthAudioProcessor();
    ~NeuralSynthAudioProcessor() override;

    //==============================================================================
    void prepareToPlay(double sampleRate, int samplesPerBlock) override;
    void releaseResources() override;
    bool isBusesLayoutSupported(const BusesLayout& layouts) const;

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

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

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

    //==============================================================================
    const juce::String getName() const override;

    bool acceptsMidi() const override;
    bool producesMidi() const override;
    bool isMidiEffect() const override;
    double getTailLengthSeconds() const override;

    //==============================================================================
    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;

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

    //==============================================================================
    void parameterChanged(const juce::String& id, float newValue) override;

    void buildParams(std::vector<std::unique_ptr<juce::RangedAudioParameter>>& params, const std::string& paramGroup);

    juce::MidiMessageCollector& getMidiMessageCollector() noexcept { return midiMessageCollector; }

    juce::MidiMessageCollector midiMessageCollector;
    juce::AudioProcessorValueTreeState parameters;
    juce::AudioProcessorValueTreeState::ParameterLayout createParameterLayout();

    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; }

private:
    //==============================================================================
    JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR(NeuralSynthAudioProcessor)

    NeuralAudioEngine audioEngine;
    AudioBufferQueue<float> audioBufferQueue;

    AudioBufferQueue<float> chorusBufferQueue;
    AudioBufferQueue<float> delayBufferQueue;
    AudioBufferQueue<float> reverbBufferQueue;

    AudioBufferQueue<float> flangerBufferQueue;
    AudioBufferQueue<float> distortionBufferQueue;
    AudioBufferQueue<float> filterBufferQueue;

    ScopeDataCollector<float> scopeDataCollector{ audioBufferQueue };

    NeuralSharedParams sp;
};