70 lines
2.7 KiB
C++
70 lines
2.7 KiB
C++
#pragma once
|
|
|
|
#include <memory>
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
|
|
#include "../rtp.hh"
|
|
#include "../socket.hh"
|
|
#include "../queue.hh"
|
|
#include "../util.hh"
|
|
|
|
namespace uvg_rtp {
|
|
|
|
namespace formats {
|
|
|
|
typedef struct media_info {
|
|
uint32_t s_seq;
|
|
uint32_t e_seq;
|
|
size_t npkts;
|
|
size_t size;
|
|
std::map<uint16_t, uvg_rtp::frame::rtp_frame *> fragments;
|
|
} media_info_t;
|
|
|
|
typedef struct media_frame_info {
|
|
std::unordered_map<uint32_t, media_info> frames;
|
|
std::unordered_set<uint32_t> dropped;
|
|
} media_frame_info_t;
|
|
|
|
class media {
|
|
public:
|
|
media(uvg_rtp::socket *socket, uvg_rtp::rtp *rtp_ctx, int flags);
|
|
virtual ~media();
|
|
|
|
/* These two functions are called by media_stream which is self is called by the application.
|
|
* They act as thunks and forward the call to __push_frame() which every media should
|
|
* implement if they require more processing than what the default implementation offers
|
|
*
|
|
* Return RTP_OK on success */
|
|
rtp_error_t push_frame(uint8_t *data, size_t data_len, int flags);
|
|
rtp_error_t push_frame(std::unique_ptr<uint8_t[]> data, size_t data_len, int flags);
|
|
|
|
/* Media-specific packet handler. The default handler, depending on what "flags_" contains,
|
|
* may only return the received RTP packet or it may merge multiple packets together before
|
|
* returning a complete frame to the user.
|
|
*
|
|
* If the implemented media requires more fine-tuned processing, this should be overrode
|
|
*
|
|
* Return RTP_OK if the packet was successfully handled
|
|
* Return RTP_PKT_NOT_HANDLED if the packet is not handled by this handler
|
|
* Return RTP_PKT_MODIFIED if the packet was modified but should be forwarded to other handlers
|
|
* Return RTP_GENERIC_ERROR if the packet was corrupted in some way */
|
|
static rtp_error_t packet_handler(void *arg, int flags, frame::rtp_frame **frame);
|
|
|
|
/* Return pointer to the internal frame info structure which is relayed to packet handler */
|
|
media_frame_info_t *get_hevc_frame_info();
|
|
|
|
protected:
|
|
virtual rtp_error_t __push_frame(uint8_t *data, size_t data_len, int flags);
|
|
|
|
uvg_rtp::socket *socket_;
|
|
uvg_rtp::rtp *rtp_ctx_;
|
|
int flags_;
|
|
uvg_rtp::frame_queue *fqueue_;
|
|
|
|
private:
|
|
media_frame_info_t minfo_;
|
|
};
|
|
};
|
|
};
|