/*
   Copyright (C) 2002, 2003, 2004, 2005, 2006, 2007, 2008, 2009, 2010,
   2011, 2012 Her Majesty the Queen in Right of Canada (Communications
   Research Center Canada)

   Copyright (C) 2024
   Matthias P. Braendli, matthias.braendli@mpb.li

    http://www.opendigitalradio.org
   */
/*
   This file is part of ODR-DabMux.

   ODR-DabMux 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.

   ODR-DabMux 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 ODR-DabMux.  If not, see <http://www.gnu.org/licenses/>.
*/

#pragma once
#include <iostream>
#include <iterator>
#include <thread>
#include <vector>
#include <chrono>
#include <atomic>
#include "ThreadsafeQueue.h"
#include "dabOutput/dabOutput.h"
#include "edioutput/TagItems.h"
#include "edioutput/TagPacket.h"
#include "edioutput/Transport.h"

// This metadata gets transmitted in the zmq stream
struct metadata_t {
    uint32_t edi_time = 0;
    int16_t utc_offset = 0;
    uint16_t dlfc = 0;
};

struct frame_t {
    // Since a zmq message actually contains 4 frames, the
    // original_zmq_msg is only non-empty for the first of the
    // four calls to Sender::send_edi_frame().
    zmq::message_t original_zmq_message;
    std::vector<uint8_t> data;
    metadata_t metadata;
    std::chrono::steady_clock::time_point received_at;
};

struct zmq_send_config_t {
    std::vector<std::string> urls;
};

class Sender {
    public:
        Sender();
        Sender(const Sender& other) = delete;
        Sender& operator=(const Sender& other) = delete;
        ~Sender();
        void start(const edi::configuration_t& conf,
                const zmq_send_config_t& zmq_conf,
                int delay_ms, bool drop_late_packets);
        void push_frame(frame_t&& frame);
        void print_configuration(void);

    private:
        void send_eti_frame(frame_t& frame);
        void process(void);

        int tist_delay_ms;
        bool drop_late;
        std::atomic<bool> running;
        std::thread process_thread;
        edi::configuration_t edi_conf;
        ThreadsafeQueue<frame_t> frames;

        std::shared_ptr<edi::Sender> edi_sender;

        zmq::context_t zmq_ctx;
        std::vector<zmq::socket_t> zmq_sockets;

        struct buffering_stat_t {
            // Time between when we received the packets and when we transmit packets, in microseconds
            double buffering_time_us = 0.0;
            bool late = false;
        };
        std::vector<buffering_stat_t> buffering_stats;
        size_t num_zmq_send_errors = 0;

};