rtmsReceiver.hpp

Go to the documentation of this file.

 
#ifndef RTMS_RECEIVER_HPP_
#define RTMS_RECEIVER_HPP_
 
#define BOOST_BIND_NO_PLACEHOLDERS
 
#include <boost/asio.hpp>
#include <boost/array.hpp>
#include <boost/atomic.hpp>
 
#include <mudpiif/MudpiPacket.hpp>
#include <mudpiif/MudpiReceiver.hpp>
#include <hrtcpif/HrtcpPayloadPacket.hpp>
#include <hrtcpif/HrtcpLeaderPacket.hpp>
 
#include <ifw/rtmslib/logger.hpp>
 
namespace ifw::rtmstools {
 
    inline std::string GetThreadName() {
        char thread_name[32];
        pthread_getname_np(pthread_self(), thread_name, 32);
        return thread_name;
    }
 
    constexpr int MAX_PACKETS = 1000;
    constexpr int PACKET_SIZE = 9600;
 
    void PrintPacketInfo(mudpiif::mudpiPacketJumbo mudpi_packet,
                         hrtcpif::hrtcpLeaderPacket& hrtcp_packet,
                         int pdu_size);
 
    class RtmsReceiver {
      public:
        std::string        m_address;
        int                m_port;
        std::string        m_net_if;
        int                m_width;
        int                m_height;
        int                m_bpp;
        long unsigned int  m_prev_frames_expected;
        long unsigned int  m_prev_frames_received;
        bool               m_execute;
        int                m_rcv_buf_factor{3};
        double             m_last_lost_packets_log{0.0};
        double             m_last_diff_sample_id_log{0.0};
        int32_t            m_current_sample_id{-1};
 
        unsigned int                 m_sample{0};
        bool                         m_io_service_running{false};
        boost::asio::io_service      m_io_service;
        boost::asio::ip::udp::socket m_socket{m_io_service};
 
        // Packet size defined assuming a Jumbo packet (9600).
        boost::array<char, PACKET_SIZE> m_recv_buffer;
        boost::asio::ip::udp::endpoint  m_remote_endpoint;
 
        struct timeval     m_start, m_end;
        int                m_num_frames = 0;
 
        log4cplus::Logger  m_logger;
 
        RtmsReceiver(std::string address,
                     int port,
                     std::string net_if,
                     int width,
                     int height,
                     int bpp,
                     const int rcv_buf_factor=3);
 
        virtual ~RtmsReceiver();
 
        int GetSize() const;
 
        int GetWidth() const;
        int GetHeight() const;
 
        void ReceivePacket(const boost::system::error_code& error, size_t bytes_transferred);
 
        virtual void HandleLeaderPacket();
 
        virtual void HandlePayloadPacket(mudpiif::mudpiPacketJumbo& packet);
 
        virtual void HandleTrailerPacket();
 
        virtual void HandleLostPackets();
 
        virtual void HandleWrongFrameId();
 
        virtual void Wait();
 
        void StartReceiver();
 
        void StopReceiver();
 
        int32_t GetCurrentSampleId() const;
 
    };
}
 
#endif  // RTMS_RECEIVER_HPP_