/*
 * Klang – a node+text-based synthesizer library
 *
 * This file is part of the *wellen* library (https://github.com/dennisppaul/wellen).
 * Copyright (c) 2022 Dennis P Paul.
 *
 * This library 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, version 3.
 *
 * This library 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 program. If not, see <http://www.gnu.org/licenses/>.
 */

/**
 *       [ NODE_MIXER_MULTI_STEREO ]
 *       +-------------------------+
 *       |                         |
 * IN00--| SIGNAL_0       SIGNAL_R |--OUT00
 * IN01--| SIGNAL_1       SIGNAL_L |--OUT01
 *   ... | ...                     |
 * INFF--| SIGNAL_FF               |
 *       |                         |
 *       +-------------------------+
 */

#ifndef NodeMixerMultiStereo_hpp
#define NodeMixerMultiStereo_hpp

#include <vector>

#include "KlangNode.hpp"

using namespace std;

namespace klang {
    class NodeMixerMultiStereo : public Node {
    public:
        static const CHANNEL_ID NUM_CH_IN           = 0xFF;
        static const CHANNEL_ID CH_OUT_SIGNAL_LEFT  = 0;
        static const CHANNEL_ID CH_OUT_SIGNAL_RIGHT = 1;
        static const CHANNEL_ID NUM_CH_OUT          = 2;

        bool connect(Connection* pConnection, CHANNEL_ID pInChannel) {
            add_channel(pInChannel, pConnection);
            return true;
        }

        bool disconnect(CHANNEL_ID pInChannel) {
            if (pInChannel < get_number_of_channels()) {
                mConnection_CH_IN_SIGNAL_and_mix[pInChannel].connection = nullptr;
                return true;
            }
            return false;
        }

        void update(CHANNEL_ID pChannel, float* pAudioBlock) {
            const uint16_t mNumberOfChannels = get_number_of_channels();
            bool           m_has_SIGNAL[mNumberOfChannels];
            uint8_t        mSignalInputCounter = 0;
            for (uint8_t i = 0; i < mNumberOfChannels; ++i) {
                m_has_SIGNAL[i] = (mConnection_CH_IN_SIGNAL_and_mix[i].connection != nullptr);
                mSignalInputCounter += m_has_SIGNAL[i];
            }
            if (is_not_updated() && mSignalInputCounter > 0) {
                mBlock_CH_OUT_SIGNAL_LEFT                   = AudioBlockPool::instance().request();
                mBlock_CH_OUT_SIGNAL_RIGHT                  = AudioBlockPool::instance().request();
                float* mBlockData_CH_OUT_SIGNAL_LEFT  = AudioBlockPool::instance().data(mBlock_CH_OUT_SIGNAL_LEFT);
                float* mBlockData_CH_OUT_SIGNAL_RIGHT = AudioBlockPool::instance().data(mBlock_CH_OUT_SIGNAL_RIGHT);

                AUDIO_BLOCK_ID mBlock_SIGNAL[mNumberOfChannels];
                float*   mBlockData_SIGNAL[mNumberOfChannels];

                for (uint8_t i = 0; i < mNumberOfChannels; ++i) {
                    if (m_has_SIGNAL[i]) {
                        mBlock_SIGNAL[i] = AudioBlockPool::instance().request();
                        if (mBlock_SIGNAL[i] == AudioBlockPool::NO_ID) {
                            // @note(probably ran out of memory blocks @maybe(implement some better error handling))
                            memset(pAudioBlock, 0.0, KLANG_SAMPLES_PER_AUDIO_BLOCK * sizeof(float));
                            return;
                        }
                        mBlockData_SIGNAL[i] = AudioBlockPool::instance().data(mBlock_SIGNAL[i]);
                        mConnection_CH_IN_SIGNAL_and_mix[i].connection->update(mBlock_SIGNAL[i]);
                    }
                }

                for (uint16_t i = 0; i < KLANG_SAMPLES_PER_AUDIO_BLOCK; ++i) {
                    float mSumL = 0.0;
                    float mSumR = 0.0;
                    for (uint8_t j = 0; j < mNumberOfChannels; ++j) {
                        if (m_has_SIGNAL[j]) {
                            const float s  = mBlockData_SIGNAL[j][i] * mConnection_CH_IN_SIGNAL_and_mix[j].mix;
                            const float pR = mConnection_CH_IN_SIGNAL_and_mix[j].pan * 0.5 + 0.5;
                            const float pL = 1.0 - pR;
                            mSumL += s * pL;
                            mSumR += s * pR;
                        }
                    }
                    mBlockData_CH_OUT_SIGNAL_LEFT[i]  = mSumL * fMasterVolume;
                    mBlockData_CH_OUT_SIGNAL_RIGHT[i] = mSumR * fMasterVolume;
                }

                for (uint8_t i = 0; i < mNumberOfChannels; ++i) {
                    if (m_has_SIGNAL[i]) {
                        AudioBlockPool::instance().release(mBlock_SIGNAL[i]);
                    }
                }

                flag_updated();
            }
            if (pChannel == CH_OUT_SIGNAL_LEFT) {
                if (mBlock_CH_OUT_SIGNAL_LEFT != AudioBlockPool::NO_ID) {
                    float* mBlockData_CH_OUT_SIGNAL_LEFT = AudioBlockPool::instance().data(mBlock_CH_OUT_SIGNAL_LEFT);
                    memcpy(pAudioBlock,
                           mBlockData_CH_OUT_SIGNAL_LEFT,
                           sizeof(float) * KLANG_SAMPLES_PER_AUDIO_BLOCK);
                } else {
                    memset(pAudioBlock, 0.0, KLANG_SAMPLES_PER_AUDIO_BLOCK * sizeof(float));
                }
            } else if (pChannel == CH_OUT_SIGNAL_RIGHT) {
                if (mBlock_CH_OUT_SIGNAL_RIGHT != AudioBlockPool::NO_ID) {
                    float* mBlockData_CH_OUT_SIGNAL_RIGHT = AudioBlockPool::instance().data(mBlock_CH_OUT_SIGNAL_RIGHT);
                    memcpy(pAudioBlock,
                           mBlockData_CH_OUT_SIGNAL_RIGHT,
                           sizeof(float) * KLANG_SAMPLES_PER_AUDIO_BLOCK);
                } else {
                    memset(pAudioBlock, 0.0, KLANG_SAMPLES_PER_AUDIO_BLOCK * sizeof(float));
                }
            }
        }

        void set_master(float pValue) {
            fMasterVolume = pValue;
        }

        float get_master() {
            return fMasterVolume;
        }

        void set_mix(uint8_t pChannel, float pValue) {
            mConnection_CH_IN_SIGNAL_and_mix[pChannel].mix = pValue;
        }

        float get_mix(uint8_t pChannel) {
            return mConnection_CH_IN_SIGNAL_and_mix[pChannel].mix;
        }

        void set_pan(uint8_t pChannel, float pValue) {
            mConnection_CH_IN_SIGNAL_and_mix[pChannel].pan = pValue;
        }

        float get_pan(uint8_t pChannel) {
            return mConnection_CH_IN_SIGNAL_and_mix[pChannel].pan;
        }

        void set_command(KLANG_CMD_TYPE pCommand, KLANG_CMD_TYPE* pPayLoad) {
            switch (pCommand) {
                case KLANG_SET_MIX_F32:
                    set_mix(static_cast<uint8_t>(pPayLoad[0]), KlangMath::FLOAT_32(pPayLoad, 1));
                    break;
                case KLANG_SET_PAN_I8_F32:
                    set_pan(static_cast<uint8_t>(pPayLoad[0]), KlangMath::FLOAT_32(pPayLoad, 1));
                    break;
            }
        }

    private:
        AUDIO_BLOCK_ID mBlock_CH_OUT_SIGNAL_LEFT  = AudioBlockPool::NO_ID;
        AUDIO_BLOCK_ID mBlock_CH_OUT_SIGNAL_RIGHT = AudioBlockPool::NO_ID;

        struct MixConnectionStruct {
            Connection* connection;
            float mix;
            float pan;
        };
        vector<MixConnectionStruct> mConnection_CH_IN_SIGNAL_and_mix;
        float                 fMasterVolume = 1.0;

        void add_channel(uint32_t pChannel, Connection* pConnection) {
            if (pChannel >= mConnection_CH_IN_SIGNAL_and_mix.size()) {
                mConnection_CH_IN_SIGNAL_and_mix.resize(pChannel + 1);
            }
            MixConnectionStruct mc;
            mc.connection                              = pConnection;
            mc.mix                                     = 1.0;
            mc.pan                                     = 0.0;
            mConnection_CH_IN_SIGNAL_and_mix[pChannel] = mc;
        }

        uint16_t get_number_of_channels() {
            return mConnection_CH_IN_SIGNAL_and_mix.size();
        }
    };
}  // namespace klang

#endif /* NodeMixerMultiStereo_hpp */