/*
 ==============================================================================
 This file is part of the IEM plug-in suite.
 Author: Daniel Rudrich
 Copyright (c) 2017 - Institute of Electronic Music and Acoustics (IEM)
 https://iem.at

 The IEM plug-in suite is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 The IEM plug-in suite is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this software.  If not, see <https://www.gnu.org/licenses/>.
 ==============================================================================
 */

#pragma once

#include "../../resources/AudioProcessorBase.h"
#include "../JuceLibraryCode/JuceHeader.h"

#include "../../resources/Conversions.h"
#include "../../resources/Quaternion.h"
#include "../../resources/ReferenceCountedMatrix.h"

#include "../../resources/ht-api-juce/supperware/Tracker.h"
#include "../../resources/ht-api-juce/supperware/midi/midi.h"

#define ProcessorClass SceneRotatorAudioProcessor

//==============================================================================
class SceneRotatorAudioProcessor
    : public AudioProcessorBase<IOTypes::Ambisonics<>, IOTypes::Ambisonics<>, true>,
      private juce::MidiMessageCollector,
      private juce::Timer,
      private Midi::TrackerDriver::Listener
{
public:
    constexpr static int numberOfInputChannels = 64;
    constexpr static int numberOfOutputChannels = 64;
    //==============================================================================
    SceneRotatorAudioProcessor();
    ~SceneRotatorAudioProcessor();

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

    void processBlock (juce::AudioSampleBuffer&, juce::MidiBuffer&) override;

    //==============================================================================
    juce::AudioProcessorEditor* createEditor() override;
    bool hasEditor() 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& parameterID, float newValue) override;
    void updateBuffers() override; // use this to implement a buffer update method

    //======= Parameters ===========================================================
    std::vector<std::unique_ptr<juce::RangedAudioParameter>> createParameterLayout();

    //======= OSC ==================================================================
    inline const bool interceptOSCMessage (juce::OSCMessage& message) override;

    //==============================================================================
    inline void updateQuaternions();
    inline void updateEuler();

    void rotateBuffer (juce::AudioBuffer<float>* bufferToRotate,
                       const int nChannels,
                       const int samples);
    void calcRotationMatrix (const int order);

    //======= MIDI Connection ======================================================
    enum class MidiScheme
    {
        none = 0,
        mrHeadTrackerYprDir,
        mrHeadTrackerYprInv,
        mrHeadTrackerQuaternions,
        supperwareQuaternions
    };

    const juce::StringArray midiSchemeNames { "none (link only)",
                                              "MrHT YPR Direct",
                                              "MrHT YPR Inverse",
                                              "MrHT Quaternions",
                                              "Supperware" };

    const juce::Identifier midiSchemeIdentifieres[5] { "none",
                                                       "MrHT_YprDir",
                                                       "MrHT_YprInv",
                                                       "MrHT_Quat",
                                                       "Sup_Quat" };

    juce::MidiDeviceInfo getCurrentMidiDeviceInfo();
    void openMidiInput (juce::MidiDeviceInfo midiDevice,
                        bool forceUpdatingCurrentMidiDeviceName = false);
    void openMidiInput (juce::String midiDeviceName,
                        bool forceUpdatingCurrentMidiDeviceName = false);
    void closeMidiInput();

    const juce::StringArray getMidiSchemes() { return midiSchemeNames; };
    MidiScheme getCurrentMidiScheme() { return currentMidiScheme; };
    void setMidiScheme (MidiScheme newMidiScheme);

    //======= Supperware ===========================================================
    void trackerMidiConnectionChanged (Midi::State newState) override;
    void trackerOrientationQ (float inp_qw, float inp_qx, float inp_qy, float inp_qz) override;

    //==============================================================================
    // Flags for editor
    juce::Atomic<bool> deviceHasChanged = false;
    juce::Atomic<bool> showMidiOpenError = false;
    juce::Atomic<bool> schemeHasChanged = false;

private:
    //==============================================================================

    juce::Atomic<bool> usingYpr = true;

    // list of used audio parameters
    std::atomic<float>* orderSetting;
    std::atomic<float>* useSN3D;

    std::atomic<float>* yaw;
    std::atomic<float>* pitch;
    std::atomic<float>* roll;
    std::atomic<float>* qw;
    std::atomic<float>* qx;
    std::atomic<float>* qy;
    std::atomic<float>* qz;
    std::atomic<float>* invertYaw;
    std::atomic<float>* invertPitch;
    std::atomic<float>* invertRoll;
    std::atomic<float>* invertQuaternion;
    std::atomic<float>* rotationSequence;

    juce::Atomic<bool> updatingParams { false };
    juce::Atomic<bool> rotationParamsHaveChanged { true };

    juce::AudioBuffer<float> copyBuffer;

    juce::OwnedArray<juce::dsp::Matrix<float>> orderMatrices;
    juce::OwnedArray<juce::dsp::Matrix<float>> orderMatricesCopy;

    double P (int i,
              int l,
              int a,
              int b,
              juce::dsp::Matrix<float>& R1,
              juce::dsp::Matrix<float>& Rlm1);
    double U (int l, int m, int n, juce::dsp::Matrix<float>& Rone, juce::dsp::Matrix<float>& Rlm1);
    double V (int l, int m, int n, juce::dsp::Matrix<float>& Rone, juce::dsp::Matrix<float>& Rlm1);
    double W (int l, int m, int n, juce::dsp::Matrix<float>& Rone, juce::dsp::Matrix<float>& Rlm1);

    void timerCallback() override;

    // ============ MIDI Device Connection ======================
    // MrHeadTracker 14-bit MIDI Data
    int yawLsb = 0, pitchLsb = 0, rollLsb = 0;
    int qwLsb = 0, qxLsb = 0, qyLsb = 0, qzLsb = 0;

    std::unique_ptr<juce::MidiInput> midiInput;
    juce::MidiDeviceInfo currentMidiDeviceInfo;
    MidiScheme currentMidiScheme = MidiScheme::none;
    juce::CriticalSection changingMidiDevice;

    // TrackerDriver for Supperware head tracker
    Midi::TrackerDriver trackerDriver;
    Midi::State supMidiState = Midi::State::Unavailable;
    //==============================================================================
    JUCE_DECLARE_NON_COPYABLE_WITH_LEAK_DETECTOR (SceneRotatorAudioProcessor)
};
