#pragma once

/**
 * @file block_simple.h
 * @brief Simplified DSP block class, with one input and one output
 */

#include "dsp/block.h"

namespace satdump
{
    namespace ndsp
    {
        /**
         * @brief Simplified DSP block implementation, for
         * synchronous blocks that can process one buffer at
         * a time. Takes away the burden of boilerplate code.
         *
         * @param output_buffer_size_ratio defines the allocated
         * output buffer size relative to the input buffer
         * max_size
         */
        template <typename Ti, typename To>
        class BlockSimple : public Block
        {
        protected:
            float output_buffer_size_ratio = 1;

        private:
            bool work()
            {
                DSPBuffer iblk = inputs[0].fifo->wait_dequeue();

                if (iblk.isTerminator())
                {
                    if (iblk.terminatorShouldPropagate())
                        outputs[0].fifo->wait_enqueue(outputs[0].fifo->newBufferTerminator());
                    inputs[0].fifo->free(iblk);
                    return true;
                }

                DSPBuffer oblk = outputs[0].fifo->newBufferSamples(ceil(iblk.max_size * output_buffer_size_ratio), sizeof(To));

                oblk.size = process(iblk.getSamples<Ti>(), iblk.size, oblk.getSamples<To>());

                if (oblk.size > 0)
                    outputs[0].fifo->wait_enqueue(oblk);

                inputs[0].fifo->free(iblk);

                return false;
            }

        public:
            BlockSimple(std::string id, std::vector<BlockIO> in = {}, std::vector<BlockIO> out = {}) : Block(id, in, out) {}

            /**
             * @brief Simplified "work" function, called automatically by work().
             * This takes away all boilerplate work usually needed, making simpler
             * block implementations more trivial to work with.
             *
             * The number of output samples MUST be <= nsamples!
             *
             * @param input input sample buffer
             * @param nsamples number of input samples
             * @param output output sample byffer
             * @return number of output samples (<= nsamples)
             */
            virtual uint32_t process(Ti *input, uint32_t nsamples, To *output) = 0;
        };
    } // namespace ndsp
} // namespace satdump