/* ------------------------------------------------------------------
 * Copyright (C) 2017 AVT GmbH - Fabien Vercasson
 * Copyright (C) 2019 Matthias P. Braendli
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 * -------------------------------------------------------------------
 */


/*! \section AVT Input
 *
 * This input comunicate with AVT encoder
 *
 * The encoded frames and reassambled
 * The PAD Frames are sent to the encoder for insertion.
 * The encoder is remotely controled to set bitrate and audio mode.
 *
 */

#pragma once

#include "Socket.h"
#include "OrderedQueue.h"
#include <cstdint>
#include <cstdio>
#include <string>
#include <queue>
#include <vector>
#include <chrono>

#define DEF_BR  64

// The enum values folown the AVT messages definitions.
enum {
    AVT_Mono            = 0,
    AVT_Mono_SBR,
    AVT_Stereo,
    AVT_Stereo_SBR,
    AVT_Stereo_SBR_PS
};

enum {
    AVT_MonoMode_LR2    = 0,
    AVT_MonoMode_L,
    AVT_MonoMode_R
};

enum {
  AVT_DAC_32            = 0,
  AVT_DAC_48
};


class AVTInput
{
    public:
        AVTInput(const std::string& input_uri, const std::string& output_uri, uint32_t pad_port,
                size_t jitterBufferSize = 40);

        /*! Open the file and prepare the wav decoder.
         *
         * \return nonzero on error
         */
        int prepare(void);

        /*! Inform class and remove encoder about the bitrate and audio mode
         *
         * \return nonzero on error
         */
        int setDabPlusParameters(int bitrate, int channels, int sample_rate, bool sbr, bool ps);

        /*! Read incoming frames from the encoder, reorder and reassemble then into DAB+ superframes
         *! Give the next reassembled audio frame (120ms for DAB+)
         *
         * \return the size of the frame or 0 if none are available yet
         */
        size_t getNextFrame(std::vector<uint8_t> &buf, std::chrono::system_clock::time_point& ts);

        /*! Store a new PAD frame.
         *! Frames are sent to the encoder on request
         */
        void pushPADFrame(const uint8_t* buf, size_t size);

        /* \return true if PAD Frame queue is full */
        bool padQueueFull();


    private:
        std::string _input_uri;
        std::string _output_uri;
        uint32_t _pad_port;
        size_t _jitterBufferSize;

        Socket::UDPSocket _input_socket;
        Socket::UDPSocket _output_socket;
        Socket::UDPPacket _output_packet;
        Socket::UDPSocket _input_pad_socket;
        Socket::UDPPacket _pad_packet;
        OrderedQueue _ordered;
        std::queue<std::vector<uint8_t> > _padFrameQueue;

        int32_t _subChannelIndex = DEF_BR/8;
        int32_t _bitRate = DEF_BR * 1000;
        int32_t _audioMode = AVT_Mono;
        int32_t _monoMode = AVT_MonoMode_LR2;
        int32_t _dac = AVT_DAC_48;
        size_t _dab24msFrameSize = DEF_BR*3;
        bool _frameAligned = false;
        std::vector<uint8_t> _currentFrame;
        int32_t _nbFrames = 0;
        int32_t _expectedFrameIndex = 0;
        int32_t _previousRtpIndex = -1;
        std::chrono::system_clock::time_point _frameZeroTimestamp;
        size_t _currentFrameSize = 0;

        bool _parseURI(const char* uri, std::string& address, long& port);
        int _openSocketSrv(Socket::UDPSocket* socket, const char* uri);
        int _openSocketCli();

        void _sendCtrlMessage();
        void _sendPADFrame();
        void _interpretMessage(const uint8_t *data, size_t size);
        bool _checkMessage();
        void _purgeMessages();

        /*! Test Bytes 1,2,3 for STI detection */
        bool _isSTI(const uint8_t* buf);

        /*! Find and extract the DAB frame from UDP/RTP/STI received frame
         * \param   frameNumber will contain the frameNumber
         * \param   dataSize will contain the actual DAB frame size
         * \return  Pointer to first byte of the DAB frame, or NULL if not found
         */
        const uint8_t* _findDABFrameFromUDP(const uint8_t* buf, size_t size,
                                    int32_t& frameNumber, size_t& dataSize);

        /*! Read and store one frame from encoder
         *
         * \return true if a data has been received
         */
        bool _readFrame();

        /*! Output info about received frames*/
        enum _frameType {
            _typeSTI,
            _typeSTIRTP,
            _typeCantExtract
        };
        _frameType _lastInfoFrameType;
        size_t _lastInfoSize = 0;
        size_t _infoNbFrame = 0;
        void _info(_frameType type, size_t size);
};